home *** CD-ROM | disk | FTP | other *** search
/ GameStar 2006 January / Gamestar_80_2006-01_dvd.iso / Dema / Civilization4 / data1.cab / Civ4DemoComponent / Assets / Python / System / binhex.pyc (.txt) < prev    next >
Encoding:
Python Compiled Bytecode  |  2005-11-09  |  14.9 KB  |  554 lines
  1. # Source Generated with Decompyle++
  2. # File: in.pyc (Python 2.4)
  3.  
  4. '''Macintosh binhex compression/decompression.
  5.  
  6. easy interface:
  7. binhex(inputfilename, outputfilename)
  8. hexbin(inputfilename, outputfilename)
  9. '''
  10. import sys
  11. import os
  12. import struct
  13. import binascii
  14. __all__ = [
  15.     'binhex',
  16.     'hexbin',
  17.     'Error']
  18.  
  19. class Error(Exception):
  20.     pass
  21.  
  22. (_DID_HEADER, _DID_DATA, _DID_RSRC) = range(3)
  23. REASONABLY_LARGE = 32768
  24. LINELEN = 64
  25. RUNCHAR = chr(144)
  26. if os.name == 'mac':
  27.     import macfs
  28.     import MacOS
  29.     
  30.     try:
  31.         openrf = MacOS.openrf
  32.     except AttributeError:
  33.         openrf = open
  34.  
  35.     
  36.     def FInfo():
  37.         return macfs.FInfo()
  38.  
  39.     
  40.     def getfileinfo(name):
  41.         finfo = macfs.FSSpec(name).GetFInfo()
  42.         (dir, file) = os.path.split(name)
  43.         fp = open(name, 'rb')
  44.         fp.seek(0, 2)
  45.         dlen = fp.tell()
  46.         fp = openrf(name, '*rb')
  47.         fp.seek(0, 2)
  48.         rlen = fp.tell()
  49.         return (file, finfo, dlen, rlen)
  50.  
  51.     
  52.     def openrsrc(name, *mode):
  53.         if not mode:
  54.             mode = '*rb'
  55.         else:
  56.             mode = '*' + mode[0]
  57.         return openrf(name, mode)
  58.  
  59. else:
  60.     
  61.     class FInfo:
  62.         
  63.         def __init__(self):
  64.             self.Type = '????'
  65.             self.Creator = '????'
  66.             self.Flags = 0
  67.  
  68.  
  69.     
  70.     def getfileinfo(name):
  71.         finfo = FInfo()
  72.         fp = open(name)
  73.         data = open(name).read(256)
  74.         for c in data:
  75.             if not c.isspace():
  76.                 if c < ' ' or ord(c) > 127:
  77.                     break
  78.                     continue
  79.         else:
  80.             finfo.Type = 'TEXT'
  81.         fp.seek(0, 2)
  82.         dsize = fp.tell()
  83.         fp.close()
  84.         (dir, file) = os.path.split(name)
  85.         file = file.replace(':', '-', 1)
  86.         return (file, finfo, dsize, 0)
  87.  
  88.     
  89.     class openrsrc:
  90.         
  91.         def __init__(self, *args):
  92.             pass
  93.  
  94.         
  95.         def read(self, *args):
  96.             return ''
  97.  
  98.         
  99.         def write(self, *args):
  100.             pass
  101.  
  102.         
  103.         def close(self):
  104.             pass
  105.  
  106.  
  107.  
  108. class _Hqxcoderengine:
  109.     '''Write data to the coder in 3-byte chunks'''
  110.     
  111.     def __init__(self, ofp):
  112.         self.ofp = ofp
  113.         self.data = ''
  114.         self.hqxdata = ''
  115.         self.linelen = LINELEN - 1
  116.  
  117.     
  118.     def write(self, data):
  119.         self.data = self.data + data
  120.         datalen = len(self.data)
  121.         todo = (datalen // 3) * 3
  122.         data = self.data[:todo]
  123.         self.data = self.data[todo:]
  124.         if not data:
  125.             return None
  126.         
  127.         self.hqxdata = self.hqxdata + binascii.b2a_hqx(data)
  128.         self._flush(0)
  129.  
  130.     
  131.     def _flush(self, force):
  132.         first = 0
  133.         while first <= len(self.hqxdata) - self.linelen:
  134.             last = first + self.linelen
  135.             self.ofp.write(self.hqxdata[first:last] + '\n')
  136.             self.linelen = LINELEN
  137.             first = last
  138.         self.hqxdata = self.hqxdata[first:]
  139.         if force:
  140.             self.ofp.write(self.hqxdata + ':\n')
  141.         
  142.  
  143.     
  144.     def close(self):
  145.         if self.data:
  146.             self.hqxdata = self.hqxdata + binascii.b2a_hqx(self.data)
  147.         
  148.         self._flush(1)
  149.         self.ofp.close()
  150.         del self.ofp
  151.  
  152.  
  153.  
  154. class _Rlecoderengine:
  155.     '''Write data to the RLE-coder in suitably large chunks'''
  156.     
  157.     def __init__(self, ofp):
  158.         self.ofp = ofp
  159.         self.data = ''
  160.  
  161.     
  162.     def write(self, data):
  163.         self.data = self.data + data
  164.         if len(self.data) < REASONABLY_LARGE:
  165.             return None
  166.         
  167.         rledata = binascii.rlecode_hqx(self.data)
  168.         self.ofp.write(rledata)
  169.         self.data = ''
  170.  
  171.     
  172.     def close(self):
  173.         if self.data:
  174.             rledata = binascii.rlecode_hqx(self.data)
  175.             self.ofp.write(rledata)
  176.         
  177.         self.ofp.close()
  178.         del self.ofp
  179.  
  180.  
  181.  
  182. class BinHex:
  183.     
  184.     def __init__(self, .2, ofp):
  185.         (name, finfo, dlen, rlen) = .2
  186.         if type(ofp) == type(''):
  187.             ofname = ofp
  188.             ofp = open(ofname, 'w')
  189.             if os.name == 'mac':
  190.                 fss = macfs.FSSpec(ofname)
  191.                 fss.SetCreatorType('BnHq', 'TEXT')
  192.             
  193.         
  194.         ofp.write('(This file must be converted with BinHex 4.0)\n\n:')
  195.         hqxer = _Hqxcoderengine(ofp)
  196.         self.ofp = _Rlecoderengine(hqxer)
  197.         self.crc = 0
  198.         if finfo is None:
  199.             finfo = FInfo()
  200.         
  201.         self.dlen = dlen
  202.         self.rlen = rlen
  203.         self._writeinfo(name, finfo)
  204.         self.state = _DID_HEADER
  205.  
  206.     
  207.     def _writeinfo(self, name, finfo):
  208.         nl = len(name)
  209.         if nl > 63:
  210.             raise Error, 'Filename too long'
  211.         
  212.         d = chr(nl) + name + '\x00'
  213.         d2 = finfo.Type + finfo.Creator
  214.         d3 = struct.pack('>h', finfo.Flags)
  215.         d4 = struct.pack('>ii', self.dlen, self.rlen)
  216.         info = d + d2 + d3 + d4
  217.         self._write(info)
  218.         self._writecrc()
  219.  
  220.     
  221.     def _write(self, data):
  222.         self.crc = binascii.crc_hqx(data, self.crc)
  223.         self.ofp.write(data)
  224.  
  225.     
  226.     def _writecrc(self):
  227.         self.ofp.write(struct.pack('>h', self.crc))
  228.         self.crc = 0
  229.  
  230.     
  231.     def write(self, data):
  232.         if self.state != _DID_HEADER:
  233.             raise Error, 'Writing data at the wrong time'
  234.         
  235.         self.dlen = self.dlen - len(data)
  236.         self._write(data)
  237.  
  238.     
  239.     def close_data(self):
  240.         if self.dlen != 0:
  241.             raise Error, 'Incorrect data size, diff=%r' % (self.rlen,)
  242.         
  243.         self._writecrc()
  244.         self.state = _DID_DATA
  245.  
  246.     
  247.     def write_rsrc(self, data):
  248.         if self.state < _DID_DATA:
  249.             self.close_data()
  250.         
  251.         if self.state != _DID_DATA:
  252.             raise Error, 'Writing resource data at the wrong time'
  253.         
  254.         self.rlen = self.rlen - len(data)
  255.         self._write(data)
  256.  
  257.     
  258.     def close(self):
  259.         if self.state < _DID_DATA:
  260.             self.close_data()
  261.         
  262.         if self.state != _DID_DATA:
  263.             raise Error, 'Close at the wrong time'
  264.         
  265.         if self.rlen != 0:
  266.             raise Error, 'Incorrect resource-datasize, diff=%r' % (self.rlen,)
  267.         
  268.         self._writecrc()
  269.         self.ofp.close()
  270.         self.state = None
  271.         del self.ofp
  272.  
  273.  
  274.  
  275. def binhex(inp, out):
  276.     '''(infilename, outfilename) - Create binhex-encoded copy of a file'''
  277.     finfo = getfileinfo(inp)
  278.     ofp = BinHex(finfo, out)
  279.     ifp = open(inp, 'rb')
  280.     while None:
  281.         d = ifp.read(128000)
  282.         if not d:
  283.             break
  284.         
  285.     ofp.close_data()
  286.     ifp.close()
  287.     ifp = openrsrc(inp, 'rb')
  288.     while None:
  289.         d = ifp.read(128000)
  290.         if not d:
  291.             break
  292.         
  293.     ofp.close()
  294.     ifp.close()
  295.  
  296.  
  297. class _Hqxdecoderengine:
  298.     '''Read data via the decoder in 4-byte chunks'''
  299.     
  300.     def __init__(self, ifp):
  301.         self.ifp = ifp
  302.         self.eof = 0
  303.  
  304.     
  305.     def read(self, totalwtd):
  306.         '''Read at least wtd bytes (or until EOF)'''
  307.         decdata = ''
  308.         wtd = totalwtd
  309.         while wtd > 0:
  310.             if self.eof:
  311.                 return decdata
  312.             
  313.             wtd = ((wtd + 2) // 3) * 4
  314.             data = self.ifp.read(wtd)
  315.             while None:
  316.                 
  317.                 try:
  318.                     (decdatacur, self.eof) = binascii.a2b_hqx(data)
  319.                 except binascii.Incomplete:
  320.                     pass
  321.  
  322.                 newdata = self.ifp.read(1)
  323.                 if not newdata:
  324.                     raise Error, 'Premature EOF on binhex file'
  325.                 
  326.                 data = data + newdata
  327.             decdata = decdata + decdatacur
  328.             wtd = totalwtd - len(decdata)
  329.             if not decdata and not (self.eof):
  330.                 raise Error, 'Premature EOF on binhex file'
  331.                 continue
  332.         return decdata
  333.  
  334.     
  335.     def close(self):
  336.         self.ifp.close()
  337.  
  338.  
  339.  
  340. class _Rledecoderengine:
  341.     '''Read data via the RLE-coder'''
  342.     
  343.     def __init__(self, ifp):
  344.         self.ifp = ifp
  345.         self.pre_buffer = ''
  346.         self.post_buffer = ''
  347.         self.eof = 0
  348.  
  349.     
  350.     def read(self, wtd):
  351.         if wtd > len(self.post_buffer):
  352.             self._fill(wtd - len(self.post_buffer))
  353.         
  354.         rv = self.post_buffer[:wtd]
  355.         self.post_buffer = self.post_buffer[wtd:]
  356.         return rv
  357.  
  358.     
  359.     def _fill(self, wtd):
  360.         self.pre_buffer = self.pre_buffer + self.ifp.read(wtd + 4)
  361.         if self.ifp.eof:
  362.             self.post_buffer = self.post_buffer + binascii.rledecode_hqx(self.pre_buffer)
  363.             self.pre_buffer = ''
  364.             return None
  365.         
  366.         mark = len(self.pre_buffer)
  367.         if self.pre_buffer[-3:] == RUNCHAR + '\x00' + RUNCHAR:
  368.             mark = mark - 3
  369.         elif self.pre_buffer[-1] == RUNCHAR:
  370.             mark = mark - 2
  371.         elif self.pre_buffer[-2:] == RUNCHAR + '\x00':
  372.             mark = mark - 2
  373.         elif self.pre_buffer[-2] == RUNCHAR:
  374.             pass
  375.         else:
  376.             mark = mark - 1
  377.         self.post_buffer = self.post_buffer + binascii.rledecode_hqx(self.pre_buffer[:mark])
  378.         self.pre_buffer = self.pre_buffer[mark:]
  379.  
  380.     
  381.     def close(self):
  382.         self.ifp.close()
  383.  
  384.  
  385.  
  386. class HexBin:
  387.     
  388.     def __init__(self, ifp):
  389.         if type(ifp) == type(''):
  390.             ifp = open(ifp)
  391.         
  392.         while None:
  393.             ch = ifp.read(1)
  394.             if not ch:
  395.                 raise Error, 'No binhex data found'
  396.             
  397.             if ch == '\r':
  398.                 continue
  399.             
  400.             if ch == ':':
  401.                 break
  402.             
  403.             if ch != '\n':
  404.                 dummy = ifp.readline()
  405.                 continue
  406.         hqxifp = _Hqxdecoderengine(ifp)
  407.         self.ifp = _Rledecoderengine(hqxifp)
  408.         self.crc = 0
  409.         self._readheader()
  410.  
  411.     
  412.     def _read(self, len):
  413.         data = self.ifp.read(len)
  414.         self.crc = binascii.crc_hqx(data, self.crc)
  415.         return data
  416.  
  417.     
  418.     def _checkcrc(self):
  419.         filecrc = struct.unpack('>h', self.ifp.read(2))[0] & 65535
  420.         self.crc = self.crc & 65535
  421.         if filecrc != self.crc:
  422.             raise Error, 'CRC error, computed %x, read %x' % (self.crc, filecrc)
  423.         
  424.         self.crc = 0
  425.  
  426.     
  427.     def _readheader(self):
  428.         len = self._read(1)
  429.         fname = self._read(ord(len))
  430.         rest = self._read(1 + 4 + 4 + 2 + 4 + 4)
  431.         self._checkcrc()
  432.         type = rest[1:5]
  433.         creator = rest[5:9]
  434.         flags = struct.unpack('>h', rest[9:11])[0]
  435.         self.dlen = struct.unpack('>l', rest[11:15])[0]
  436.         self.rlen = struct.unpack('>l', rest[15:19])[0]
  437.         self.FName = fname
  438.         self.FInfo = FInfo()
  439.         self.FInfo.Creator = creator
  440.         self.FInfo.Type = type
  441.         self.FInfo.Flags = flags
  442.         self.state = _DID_HEADER
  443.  
  444.     
  445.     def read(self, *n):
  446.         if self.state != _DID_HEADER:
  447.             raise Error, 'Read data at wrong time'
  448.         
  449.         if n:
  450.             n = n[0]
  451.             n = min(n, self.dlen)
  452.         else:
  453.             n = self.dlen
  454.         rv = ''
  455.         while len(rv) < n:
  456.             rv = rv + self._read(n - len(rv))
  457.         self.dlen = self.dlen - n
  458.         return rv
  459.  
  460.     
  461.     def close_data(self):
  462.         if self.state != _DID_HEADER:
  463.             raise Error, 'close_data at wrong time'
  464.         
  465.         if self.dlen:
  466.             dummy = self._read(self.dlen)
  467.         
  468.         self._checkcrc()
  469.         self.state = _DID_DATA
  470.  
  471.     
  472.     def read_rsrc(self, *n):
  473.         if self.state == _DID_HEADER:
  474.             self.close_data()
  475.         
  476.         if self.state != _DID_DATA:
  477.             raise Error, 'Read resource data at wrong time'
  478.         
  479.         if n:
  480.             n = n[0]
  481.             n = min(n, self.rlen)
  482.         else:
  483.             n = self.rlen
  484.         self.rlen = self.rlen - n
  485.         return self._read(n)
  486.  
  487.     
  488.     def close(self):
  489.         if self.rlen:
  490.             dummy = self.read_rsrc(self.rlen)
  491.         
  492.         self._checkcrc()
  493.         self.state = _DID_RSRC
  494.         self.ifp.close()
  495.  
  496.  
  497.  
  498. def hexbin(inp, out):
  499.     '''(infilename, outfilename) - Decode binhexed file'''
  500.     ifp = HexBin(inp)
  501.     finfo = ifp.FInfo
  502.     if not out:
  503.         out = ifp.FName
  504.     
  505.     if os.name == 'mac':
  506.         ofss = macfs.FSSpec(out)
  507.         out = ofss.as_pathname()
  508.     
  509.     ofp = open(out, 'wb')
  510.     while None:
  511.         d = ifp.read(128000)
  512.         if not d:
  513.             break
  514.         
  515.     ofp.close()
  516.     ifp.close_data()
  517.     d = ifp.read_rsrc(128000)
  518.     if d:
  519.         ofp = openrsrc(out, 'wb')
  520.         ofp.write(d)
  521.         while None:
  522.             d = ifp.read_rsrc(128000)
  523.             if not d:
  524.                 break
  525.             
  526.         ofp.close()
  527.     
  528.     if os.name == 'mac':
  529.         nfinfo = ofss.GetFInfo()
  530.         nfinfo.Creator = finfo.Creator
  531.         nfinfo.Type = finfo.Type
  532.         nfinfo.Flags = finfo.Flags
  533.         ofss.SetFInfo(nfinfo)
  534.     
  535.     ifp.close()
  536.  
  537.  
  538. def _test():
  539.     if os.name == 'mac':
  540.         (fss, ok) = macfs.PromptGetFile('File to convert:')
  541.         if not ok:
  542.             sys.exit(0)
  543.         
  544.         fname = fss.as_pathname()
  545.     else:
  546.         fname = sys.argv[1]
  547.     binhex(fname, fname + '.hqx')
  548.     hexbin(fname + '.hqx', fname + '.viahqx')
  549.     sys.exit(1)
  550.  
  551. if __name__ == '__main__':
  552.     _test()
  553.  
  554.