WriteExcel 0.2.0

data/lib/WriteExcel/storage_lite.rb

@@ -0,0 +1,590 @@
+#require 'tempfile'
+#require 'stringio'
+
+class OLEStorageLite
+  PPS_TYPE_ROOT   = 5
+  PPS_TYPE_DIR    = 1
+  PPS_TYPE_FILE   = 2
+  DATA_SIZE_SMALL = 0x1000
+  LONG_INT_SIZE   = 4
+  PPS_SIZE        = 0x80
+
+  def asc2ucs(str)
+    str.split(//).join("\0") + "\0"
+  end
+
+  def ucs2asc(str)
+    ary = str.unpack('v*').map { |s| [s].pack('c')}
+    ary.join('')
+  end
+
+  def localDate2OLE(localtime)
+    return "\x00" * 8 unless localtime
+  
+    # Convert from localtime (actually gmtime) to seconds.
+    args = localtime.reverse
+    args[0] += 1900   # year
+    args[1] += 1      # month
+    time = Time.gm(*args)
+    
+    # Add the number of seconds between the 1601 and 1970 epochs.
+    time = time.to_i + 11644473600
+  
+    # The FILETIME seconds are in units of 100 nanoseconds.
+    nanoseconds = time * 1E7
+  
+    # Pack the total nanoseconds into 64 bits...
+    hi, lo = nanoseconds.divmod 1 << 32
+  
+    oletime = [lo, hi].pack("VV")
+    return oletime
+  end
+end
+
+class OLEStorageLitePPS < OLEStorageLite
+  attr_accessor :no, :name, :type, :prev_pps, :next_pps, :dir_pps
+  attr_accessor :time_1st, :time_2nd, :start_block, :size, :data, :child
+  attr_reader   :pps_file
+
+  def initialize(iNo, sNm, iType, iPrev, iNext, iDir,
+                 raTime1st, raTime2nd, iStart, iSize, sData, raChild)
+    @no          = iNo
+    @name        = sNm
+    @type        = iType
+    @prev_pps    = iPrev
+    @next_pps    = iNext
+    @dir_pps     = iDir
+    @time_1st    = raTime1st
+    @time_2nd    = raTime2nd
+    @start_block = iStart
+    @size        = iSize
+    @data        = sData
+    @child       = raChild
+    @pps_file    = nil
+  end
+
+  def _datalen
+    return 0 if @data.nil?
+    if @pps_file
+      return @pps_file.lstat.size
+    else
+      return @data.size
+    end
+  end
+  protected :_datalen
+
+  def _makeSmallData(aList, rh_info)
+    file = rh_info[:fileh]
+    iSmBlk = 0
+    sRes = ''
+  
+    aList.each do |pps|
+      #1. Make SBD, small data string
+      if pps.type == PPS_TYPE_FILE
+        next if pps.size <= 0
+        if pps.size < rh_info[:small_size]
+          iSmbCnt  = pps.size / rh_info[:small_block_size]
+          iSmbCnt += 1 if pps.size % rh_info[:small_block_size] > 0
+          #1.1 Add to SBD
+          0.upto(iSmbCnt-1-1) do |i|
+            file.write([i + iSmBlk+1].pack("V"))
+          end
+          file.write([-2].pack("V"))
+          
+          #1.2 Add to Data String(this will be written for RootEntry)
+          #Check for update
+          if pps.pps_file
+            pps.pps_file.seek(0) #To The Top
+            while sBuff = pps.pps_file.read(4096)
+              sRes << sBuff
+            end
+          else 
+            sRes << pps.data
+          end
+          if pps.size % rh_info[:small_block_size] > 0
+            cnt = rh_info[:small_block_size] - (pps.size % rh_info[:small_block_size])
+            sRes << "\0" * cnt
+          end
+          #1.3 Set for PPS
+          pps.start_block = iSmBlk
+          iSmBlk += iSmbCnt
+        end
+      end
+    end
+    iSbCnt = rh_info[:big_block_size] / LONG_INT_SIZE
+    file.write([-1].pack("V") * (iSbCnt - (iSmBlk % iSbCnt))) if iSmBlk % iSbCnt > 0
+    #2. Write SBD with adjusting length for block
+    return sRes
+  end
+  private :_makeSmallData
+
+  def _savePpsWk(rh_info)
+    #1. Write PPS
+    file = rh_info[:fileh]
+    file.write(
+        @name                          +
+        ("\x00" * (64 - @name.length)) +       #64
+        [@name.length + 2].pack("v")   +       #66
+        [@type].pack("c")              +       #67
+        [0x00].pack("c")               + #UK   #68
+        [@prev_pps].pack("V")          + #Prev #72
+        [@next_pps].pack("V")          + #Next #76
+        [@dir_pps].pack("V")           + #Dir  #80
+        "\x00\x09\x02\x00"             +       #84
+        "\x00\x00\x00\x00"             +       #88
+        "\xc0\x00\x00\x00"             +       #92
+        "\x00\x00\x00\x46"             +       #96
+        "\x00\x00\x00\x00"             +       #100
+        localDate2OLE(@time_1st)       +       #108
+        localDate2OLE(@time_2nd)                #116
+      )
+    if @start_block != 0
+      file.write([@start_block].pack('V'))
+    else
+      file.write([0].pack('V'))
+    end
+    if @size != 0                               #124
+      file.write([@size].pack('V'))
+    else
+      file.write([0].pack('V'))
+    end
+    file.write([0].pack('V'))                   #128
+  end
+  protected :_savePpsWk
+end
+
+class OLEStorageLitePPSRoot < OLEStorageLitePPS
+  def initialize(raTime1st, raTime2nd, raChild)
+    super(
+      nil,
+      asc2ucs('Root Entry'),
+      PPS_TYPE_ROOT,
+      nil,
+      nil,
+      nil,
+      raTime1st,
+      raTime2nd,
+      nil,
+      nil,
+      nil,
+      raChild)
+  end
+
+  def save(sFile, bNoAs = nil, rh_info = nil)
+    #0.Initial Setting for saving
+    rh_info = Hash.new unless rh_info
+    if rh_info[:big_block_size]
+      rh_info[:big_block_size] = 2 ** adjust2(rh_info[:big_block_size])
+    else
+      rh_info[:big_block_size] = 2 ** 9
+    end
+    if rh_info[:small_block_size]
+      rh_info[:small_block_size] = 2 ** adjust2(rh_info[:small_block_size])
+    else
+      rh_info[:small_block_size] = 2 ** 6
+    end
+    rh_info[:small_size] = 0x1000
+    rh_info[:pps_size]   = 0x80
+
+    close_file = true
+
+    #1.Open File
+    #1.1 sFile is Ref of scalar
+    if sFile.kind_of?(String)
+      rh_info[:fileh] = open(sFile, "wb")
+    else
+      rh_info[:fileh] = sFile.binmode
+    end
+
+    iBlk = 0
+    #1. Make an array of PPS (for Save)
+    aList=[]
+    if bNoAs
+      _savePpsSetPnt2([self], aList, rh_info)
+    else
+      _savePpsSetPnt([self], aList, rh_info)
+    end
+    iSBDcnt, iBBcnt, iPPScnt = _calcSize(aList, rh_info)
+
+    #2.Save Header
+    _saveHeader(rh_info, iSBDcnt, iBBcnt, iPPScnt)
+
+    #3.Make Small Data string (write SBD)
+    # Small Datas become RootEntry Data
+    @data = _makeSmallData(aList, rh_info)
+
+    #4. Write BB
+    iBBlk = iSBDcnt
+    _saveBigData(iBBlk, aList, rh_info)
+
+    #5. Write PPS
+    _savePps(aList, rh_info)
+
+    #6. Write BD and BDList and Adding Header informations
+    _saveBbd(iSBDcnt, iBBcnt, iPPScnt, rh_info)
+
+    #7.Close File
+    return rh_info[:fileh].close if close_file != 0
+  end
+
+  def _calcSize(aList, rh_info)
+    #0. Calculate Basic Setting
+    iSBDcnt, iBBcnt, iPPScnt = [0,0,0]
+    iSmallLen = 0
+    iSBcnt = 0
+    aList.each do |pps|
+      if pps.type == PPS_TYPE_FILE
+        pps.size = pps._datalen      #Mod
+        if pps.size < rh_info[:small_size]
+          iSBcnt += pps.size / rh_info[:small_block_size]
+          iSBcnt += 1 if pps.size % rh_info[:small_block_size] > 0
+        else
+          iBBcnt += pps.size / rh_info[:big_block_size]
+          iBBcnt += 1 if pps.size % rh_info[:big_block_size] > 0
+        end
+      end
+    end
+    iSmallLen = iSBcnt * rh_info[:small_block_size]
+    iSlCnt   = rh_info[:big_block_size] / LONG_INT_SIZE
+    iSBDcnt  = iSBcnt / iSlCnt
+    iSBDcnt += 1 if iSBcnt % iSlCnt > 0
+    iBBcnt  += iSmallLen / rh_info[:big_block_size]
+    iBBcnt  += 1 if iSmallLen % rh_info[:big_block_size] > 0
+    iCnt     = aList.size
+    iBdCnt   = rh_info[:big_block_size] / PPS_SIZE
+    iPPScnt  = iCnt / iBdCnt
+    iPPScnt += 1 if iCnt % iBdCnt > 0
+    return [iSBDcnt, iBBcnt, iPPScnt]
+  end
+  private :_calcSize
+
+  def _adjust2(i2)
+    iWk = Math.log(i2)/Math.log(2)
+    return iWk > Integer(iWk) ? Integer(iWk) + 1 : iWk
+  end
+  private :_adjust2
+
+  def _saveHeader(rh_info, iSBDcnt, iBBcnt, iPPScnt)
+    file = rh_info[:fileh]
+  
+    #0. Calculate Basic Setting
+    iBlCnt = rh_info[:big_block_size] / LONG_INT_SIZE
+    i1stBdL = (rh_info[:big_block_size] - 0x4C) / LONG_INT_SIZE
+    i1stBdMax = i1stBdL * iBlCnt  - i1stBdL
+    iBdExL = 0
+    iAll = iBBcnt + iPPScnt + iSBDcnt
+    iAllW = iAll
+    iBdCntW  = iAllW / iBlCnt
+    iBdCntW += 1 if iAllW % iBlCnt > 0
+    iBdCnt = 0
+    #0.1 Calculate BD count
+    iBlCnt -= 1 #the BlCnt is reduced in the count of the last sect is used for a pointer the next Bl
+    iBBleftover = iAll - i1stBdMax
+    if iAll >i1stBdMax
+      while true
+        iBdCnt       = iBBleftover / iBlCnt
+        iBdCnt      += 1 if iBBleftover % iBlCnt > 0
+        iBdExL       = iBdCnt / iBlCnt
+        iBdExL      += 1 if iBdCnt % iBlCnt > 0
+        iBBleftover += iBdExL
+        break if iBdCnt == iBBleftover / iBlCnt + (iBBleftover % iBlCnt > 0 ? 1 : 0)
+      end
+    end
+    iBdCnt += i1stBdL
+    #print "iBdCnt = iBdCnt \n"
+  
+    #1.Save Header
+    file.write(
+        "\xD0\xCF\x11\xE0\xA1\xB1\x1A\xE1" +
+        "\x00\x00\x00\x00" * 4             +
+        [0x3b].pack("v")                   +
+        [0x03].pack("v")                   +
+        [-2].pack("v")                     +
+        [9].pack("v")                      +
+        [6].pack("v")                      +
+        [0].pack("v")                      +
+        "\x00\x00\x00\x00" * 2             +
+        [iBdCnt].pack("V")                 +
+        [iBBcnt+iSBDcnt].pack("V")         + #ROOT START
+        [0].pack("V")                      +
+        [0x1000].pack("V")                 +
+        [0].pack("V")                      + #Small Block Depot
+        [1].pack("V")
+      )
+    #2. Extra BDList Start, Count
+    if iAll <= i1stBdMax
+      file.write(
+          [-2].pack("V")                   + #Extra BDList Start
+          [0].pack("V")                      #Extra BDList Count
+        )
+    else
+      file.write(
+          [iAll + iBdCnt].pack("V")        +
+          [iBdExL].pack("V")
+        )
+    end
+  
+    #3. BDList
+      cnt = i1stBdL
+      cnt = iBdCnt if iBdCnt < i1stBdL
+      0.upto(cnt-1) do |i|
+        file.write([iAll + i].pack("V"))
+      end
+      file.write([-1].pack("V") * (i1stBdL - cnt)) if cnt < i1stBdL
+  end
+  private :_saveHeader
+
+  def _saveBigData(iStBlk, aList, rh_info)
+    iRes = 0
+    file = rh_info[:fileh]
+  
+    #1.Write Big (ge 0x1000) Data into Block
+    aList.each do |pps|
+      if pps.type != PPS_TYPE_DIR
+        #print "PPS: pps DEF:", defined(pps->{Data}), "\n"
+        pps.size = pps._datalen   #Mod
+        if (pps.size >= rh_info[:small_size]) ||
+           ((pps.type == PPS_TYPE_ROOT) && !pps.data.nil?)
+          #1.1 Write Data
+          #Check for update
+          if pps.pps_file
+            iLen = 0
+            pps.pps_file.seek(0, 0) #To The Top
+            while sBuff = pps.pps_file.read(4096)
+              iLen += sBuff.length
+              file.write(sBuff)           #Check for update
+            end
+          else
+            file.write(pps.data)
+          end
+          if pps.size % rh_info[:big_block_size] > 0
+            file.write(
+              "\x00" *
+               (rh_info[:big_block_size] -
+                    (pps.size % rh_info[:big_block_size]))
+              )
+          end
+          #1.2 Set For PPS
+          pps.start_block = iStBlk
+          iStBlk += pps.size / rh_info[:big_block_size]
+          iStBlk += 1 if pps.size % rh_info[:big_block_size] > 0
+        end
+      end
+    end
+  end
+  
+  def _savePps(aList, rh_info)
+    #0. Initial
+    file = rh_info[:fileh]
+    #2. Save PPS
+    aList.each do |oItem|
+      oItem._savePpsWk(rh_info)
+    end
+    #3. Adjust for Block
+    iCnt = aList.size
+    iBCnt = rh_info[:big_block_size] / rh_info[:pps_size]
+    if iCnt % iBCnt > 0
+      file.write("\x00" * ((iBCnt - (iCnt % iBCnt)) * rh_info[:pps_size]))
+    end
+    return (iCnt / iBCnt) + ((iCnt % iBCnt) > 0 ? 1: 0)
+  end
+  private :_savePps
+
+  def _savePpsSetPnt(pps_array, aList, rh_info)
+    #1. make Array as Children-Relations
+    #1.1 if No Children
+bpp=1
+    if pps_array.nil? || pps_array.size == 0
+        return 0xFFFFFFFF
+    #1.2 Just Only one
+    elsif pps_array.size == 1
+      aList << pps_array[0]
+      pps_array[0].no = aList.size - 1
+      pps_array[0].prev_pps = 0xFFFFFFFF
+      pps_array[0].next_pps = 0xFFFFFFFF
+      pps_array[0].dir_pps  = _savePpsSetPnt(pps_array[0].child, aList, rh_info)
+      return pps_array[0].no
+    #1.3 Array
+    else
+      iCnt = pps_array.size
+      #1.3.1 Define Center
+      iPos = Integer(iCnt / 2.0)     #$iCnt
+
+      aList.push(pps_array[iPos])
+      pps_array[iPos].no = aList.size - 1
+      
+      aWk = pps_array.dup
+      #1.3.2 Devide a array into Previous,Next
+      aPrev = aWk[0, iPos]
+      aWk[0..iPos-1] = nil
+      aNext = aWk[1, iCnt - iPos - 1]
+      aWk[1..(1 + iCnt - iPos -1 -1)] = nil
+      pps_array[iPos].prev_pps = _savePpsSetPnt(aPrev, aList, rh_info)
+      pps_array[iPos].next_pps = _savePpsSetPnt(aNext, aList, rh_info)
+      pps_array[iPos].dir_pps  = _savePpsSetPnt(pps_array[iPos].child, aList, rh_info)
+      return pps_array[iPos].no
+    end
+  end
+  private :_savePpsSetPnt
+  
+  def _savePpsSetPnt2(pps_array, aList, rh_info)
+    #1. make Array as Children-Relations
+    #1.1 if No Children
+    if pps_array.nil? || pps_array.size == 0
+        return 0xFFFFFFFF
+    #1.2 Just Only one
+    elsif pps_array.size == 1
+      aList << pps_array[0]
+      pps_array[0].no = aList.size - 1
+      pps_array[0].prev_pps = 0xFFFFFFFF
+      pps_array[0].next_pps = 0xFFFFFFFF
+      pps_array[0].dir_pps  = _savePpsSetPnt2(pps_array[0].child, aList, rh_info)
+      return pps_array[0].no
+    #1.3 Array
+    else
+      iCnt = pps_array.size
+      #1.3.1 Define Center
+      iPos = 0  #int($iCnt/ 2);     #$iCnt
+  
+      aWk = pps_array.dup
+      aPrev = aWk[1, 1]
+      aWk[1..1] = nil
+      aNext = aWk[1..aWk.size]      #, $iCnt - $iPos -1);
+      pps_array[iPos].prev_pps = _savePpsSetPnt2(pps_array, aList, rh_info)
+      aList.push(pps_array[iPos])
+      pps_array[iPos].no = aList.size
+  
+      #1.3.2 Devide a array into Previous,Next
+      pps_array[iPos].next_pps = _savePpsSetPnt2(aNext, aList, rh_info)
+      pps_array[iPos].dir_pps  = _savePpsSetPnt2(pps_array[iPos].child, aList, rh_info)
+      return pps_array[iPos].no
+    end
+  end
+  private :_savePpsSetPnt2
+
+  def _saveBbd(iSbdSize, iBsize, iPpsCnt, rh_info)
+    file = rh_info[:fileh]
+    #0. Calculate Basic Setting
+    iBbCnt    = rh_info[:big_block_size] / LONG_INT_SIZE
+    iBlCnt    = iBbCnt - 1
+    i1stBdL   = (rh_info[:big_block_size] - 0x4C) / LONG_INT_SIZE
+    i1stBdMax = i1stBdL * iBbCnt  - i1stBdL
+    iBdExL    = 0
+    iAll      = iBsize + iPpsCnt + iSbdSize
+    iAllW     = iAll
+    iBdCntW   = iAllW / iBbCnt
+    iBdCntW  += 1 if iAllW % iBbCnt > 0
+    iBdCnt    = 0
+    #0.1 Calculate BD count
+    iBBleftover = iAll - i1stBdMax
+    if iAll >i1stBdMax
+      while true
+        iBdCnt  = iBBleftover / iBlCnt
+        iBdCnt += 1 if iBBleftover % iBlCnt > 0
+
+        iBdExL  = iBdCnt / iBlCnt
+        iBdExL += 1 if iBdCnt % iBlCnt > 0
+        iBBleftover = iBBleftover + iBdExL
+        break if iBdCnt == (iBBleftover / iBlCnt + ((iBBleftover % iBlCnt) > 0 ? 1: 0))
+      end
+    end
+    iAllW  += iBdExL
+    iBdCnt += i1stBdL
+    #print "iBdCnt = iBdCnt \n"
+  
+    #1. Making BD
+    #1.1 Set for SBD
+    if iSbdSize > 0
+      0.upto(iSbdSize-1-1) do |i|
+        file.write([i + 1].pack('V'))
+      end
+      file.write([-2].pack('V'))
+    end
+    #1.2 Set for B
+    0.upto(iBsize-1-1) do |i|
+      file.write([i + iSbdSize + 1].pack('V'))
+    end
+    file.write([-2].pack('V'))
+  
+    #1.3 Set for PPS
+    0.upto(iPpsCnt-1-1) do |i|
+      file.write([i+iSbdSize+iBsize+1].pack("V"))
+    end
+    file.write([-2].pack('V'))
+    #1.4 Set for BBD itself ( 0xFFFFFFFD : BBD)
+    0.upto(iBdCnt-1) do |i|
+      file.write([0xFFFFFFFD].pack("V"))
+    end
+    #1.5 Set for ExtraBDList
+    0.upto(iBdExL-1) do |i|
+      file.write([0xFFFFFFFC].pack("V"))
+    end
+    #1.6 Adjust for Block
+    if (iAllW + iBdCnt) % iBbCnt > 0
+      file.write([-1].pack('V') *  (iBbCnt - ((iAllW + iBdCnt) % iBbCnt)))
+    end
+    #2.Extra BDList
+    if iBdCnt > i1stBdL
+      iN  = 0
+      iNb = 0
+      i1stBdL.upto(iBdCnt-1) do |i|
+        if iN >= iBbCnt-1
+          iN   = 0
+          iNb += 1
+          file.write([iAll+iBdCnt+iNb].pack("V"))
+        end
+        file.write([iBsize+iSbdSize+iPpsCnt+i].pack("V"))
+        iN += 1
+      end
+      if (iBdCnt-i1stBdL) % (iBbCnt-1) > 0
+        file.write([-1].pack("V") * ((iBbCnt-1) - ((iBdCnt-i1stBdL) % (iBbCnt-1))))
+      end
+      file.write([-2].pack('V'))
+    end
+  end
+end
+
+class OLEStorageLitePPSFile < OLEStorageLitePPS
+  def initialize(sNm, data = '')
+    super(
+        nil,
+        sNm || '',
+        PPS_TYPE_FILE,
+        nil,
+        nil,
+        nil,
+        nil,
+        nil,
+        nil,
+        nil,
+        data || '',
+        nil
+      )
+  end
+
+  def set_file(sFile = '')
+    if sFile.nil? or sFile == ''
+      @pps_file = Tempfile.new('OLEStorageLitePPSFile')
+    elsif sFile.kind_of?(IO) || sFile.kind_of?(StringIO)
+      @pps_file = sFile
+    elsif sFile.kind_of?(String)
+      #File Name
+      @pps_file = open(sFile, "r+")
+      return nil unless @pps_file
+    else
+      return nil
+    end
+    @pps_file.seek(0, IO::SEEK_END)
+    @pps_file.binmode
+  end
+
+  def append (data)
+    return if data.nil?
+    if @pps_file
+      @pps_file << data
+      @pps_file.flush
+    else
+      @data << data
+    end
+  end
+end