o
    nĎiY                  
   @  s  d Z ddlmZ ddlmZ ddlmZ ddlmZ ddl	m
Z
 ddlmZ ddlmZmZ dd	lZdd
lmZ dd	lZeeZG dd deZdZdZdddddddddd	ZefddZG dd deZdZeeZ dZ!ee!Z"dZ#ee#Z$d Z%ee%Z&d!Z'ee'Z(G d"d# d#eZ)G d$d% d%e)Z*G d&d' d'e)Z+G d(d) d)Z,d*d+ Z-d,d- Z.d.d/ Z/ed0krdd	l0Z0dd	l1Z1e02e13 j4 d	S d	S )1a  ttLib/sfnt.py -- low-level module to deal with the sfnt file format.

Defines two public classes:

- SFNTReader
- SFNTWriter

(Normally you don't have to use these classes explicitly; they are
used automatically by ttLib.TTFont.)

The reading and writing of sfnt files is separated in two distinct
classes, since whenever the number of tables changes or whenever
a table's length changes you need to rewrite the whole file anyway.
    )annotations)KeysView)BytesIO)SimpleNamespace)Tag)sstruct)
TTLibErrorTTLibFileIsCollectionErrorN)OrderedDictc                   @  sd   e Zd Zdd ZdddZdddZeZdddZd ddZd!ddZ	d"ddZ
dd Zdd ZdS )#
SFNTReaderc                 O  s^   |r*| t u r*|d }|d t|d}|d |dkr*ddlm} t|S t| S )zmReturn an instance of the SFNTReader sub-class which is compatible
        with the input file type.
        r      wOF2)WOFF2Reader)r   seekr   readfontTools.ttLib.woff2r   object__new__)clsargskwargsinfilesfntVersionr    r   P/home/jeff/fluffinator/venv/lib/python3.10/site-packages/fontTools/ttLib/sfnt.pyr   !   s   



zSFNTReader.__new__r   c                 C  s  || _ || _d | _d | _t| _| j d | j d| _| j d | jdkrft	| j }|j
}d|  kr9|k sBn td|d  || _
| j |j|  | j t}t|tkr^tdtt||  n:| jdkrd| _t| _| j t}t|tkrtd	tt||  n| j t}t|tkrtd
tt||  t| j| _| jdvrtdi }t| jD ]}|  }	|	| j  t|	j}
|	||
< qtt| dd d| _| jdkrt| | _d S d S )Nr   r   s   ttcfz2specify a font number between 0 and %d (inclusive)   'Not a Font Collection (not enough data)   wOFFwoffz!Not a WOFF font (not enough data)z1Not a TrueType or OpenType font (not enough data))   OTTOtruez1Not a TrueType or OpenType font (bad sfntVersion)c                 S  s
   | d j S )Nr   )offset)ir   r   r   <lambda>`   s   
 z%SFNTReader.__init__.<locals>.<lambda>)key) filecheckChecksumsflavor
flavorDataSFNTDirectoryEntryDirectoryEntryr   r   r   readTTCHeadernumFontsr	   offsetTablesfntDirectorySizelenr   r   unpacksfntDirectoryFormatWOFFDirectoryEntrywoffDirectorySizewoffDirectoryFormatr   range	numTablesfromFiletagr
   sorteditemstablesWOFFFlavorData)selfr'   r(   
fontNumberheaderr.   datar=   r$   entryr:   r   r   r   __init__2   s^   






zSFNTReader.__init__r:   str | bytesreturnboolc                 C  s
   || j v S Nr=   r?   r:   r   r   r   has_keyf      
zSFNTReader.has_keyKeysView[Tag]c                 C  s
   | j  S rH   )r=   keysr?   r   r   r   rN   k   rL   zSFNTReader.keysbytesc                 C  s   | j t| }|| j}| jrF|dkr%t|dd d |dd  }nt|}| jdkr;||jks9J d| |S ||jkrFtd| |S )zFetch the raw table data.headN             r   zbad checksum for '%s' table)	r=   r   loadDatar'   r(   calcChecksumcheckSumlogwarning)r?   r:   rC   rB   checksumr   r   r   __getitem__n   s   "

zSFNTReader.__getitem__Nonec                 C  s   | j t|= d S rH   )r=   r   rJ   r   r   r   __delitem__   s   zSFNTReader.__delitem__c                 C  s   | j   d S rH   )r'   closerO   r   r   r   r^         zSFNTReader.closec                 C  s@   t | jtr	| jS | j }|d= | jj|d< | j |d< |S )Nr'   	_filename_filepos)
isinstancer'   r   __dict__copynametellr?   stater   r   r   __getstate__   s   
zSFNTReader.__getstate__c                 C  s<   d|vrt |dd| _| j|d | j| d S )Nr'   r`   rbra   )openpopr'   r   rc   updaterg   r   r   r   __setstate__   s   zSFNTReader.__setstate__N)r   r   )r:   rE   rF   rG   )rF   rM   )r:   rE   rF   rP   )r:   rE   rF   r\   )rF   r\   )__name__
__module____qualname__r   rD   rK   __contains__rN   r[   r]   r^   ri   rn   r   r   r   r   r       s    

4



	r      Fr         rR   
         2   d   )	r      rt   r   ru   rs      rR   	   c                 C  s`   d|  kr
dksn t d| tr|dkr"ddlm} || |S ddlm} || t| dS )aE  Compress 'data' to Zlib format. If 'USE_ZOPFLI' variable is True,
    zopfli is used instead of the zlib module.
    The compression 'level' must be between 0 and 9. 1 gives best speed,
    9 gives best compression (0 gives no compression at all).
    The default value is a compromise between speed and compression (6).
    r   r}   zBad compression level: %s)compress)numiterations)
ValueError
USE_ZOPFLIzlibr~   zopfli.zlibZOPFLI_LEVELS)rB   levelr~   r   r   r   r~      s   
r~   c                   @  s\   e Zd Zdd Z			dddZdd Zd	d
 Zdd Zdd Zdd Z	dd Z
dd ZdS )
SFNTWriterc                 O  sb   d}|rd|v r|d }n|rt |dkr|d }| tu r,|dkr,ddlm} t|S t| S )zpReturn an instance of the SFNTWriter sub-class which is compatible
        with the specified 'flavor'.
        Nr)   rt   woff2r   )WOFF2Writer)r1   r   r   r   r   r   )r   r   r   r)   r   r   r   r   r      s   


zSFNTWriter.__new__r    Nc                 C  s   || _ || _t|| _|| _|| _| jdkr*t| _t| _	t
| _d| _t|t  | _n$| jr4J d| j t| _t| _	t| _ddlm} ||d\| _| _| _| j  | _| j| j	 || jj  | _| j | j | j d| j| j     t | _d S )Nr   wOFFUnknown flavor '%s'r   getSearchRange       )r'   r8   r   r   r)   r*   r6   directoryFormatr5   directorySizer4   r,   	signaturer0   sfntDirectoryEntrySizeorigNextTableOffsetr3   r+   fontTools.ttLibr   searchRangeentrySelector
rangeShiftrf   directoryOffset
formatSizenextTableOffsetr   writer
   r=   )r?   r'   r8   r   r)   r*   r   r   r   r   rD      s<   



zSFNTWriter.__init__c                 C  s$   || j v rtd| || j |< d S )Ncannot rewrite '%s' table)r=   r   )r?   r:   rC   r   r   r   setEntry  s   
zSFNTWriter.setEntryc                 C  s   || j v rtd| |  }||_| j|_|dkr2t|dd d |dd  |_|| _d|_	nt||_|
| j| | jdkrS| j|_|  j|jd	 d
@ 7  _| j|jd	 d
@  | _| jd| j| j    | j| j ksvJ | || dS )zWrite raw table data to disk.r   rQ   NrR   rS   rT   Tr   rt   r   )r=   r   r,   r:   r   r#   rV   rW   	headTableuncompressedsaveDatar'   r)   r   
origOffset
origLengthlengthr   rf   r   )r?   r:   rB   rC   r   r   r   __setitem__  s$   
"

zSFNTWriter.__setitem__c                 C  s
   | j | S rH   rI   rJ   r   r   r   r[   1  rL   zSFNTWriter.__getitem__c           
      C  sz  t | j }t|| jkrtd| jt|f | jdkrd| _d| _d| _	|  j	dt| 7  _	|D ]\}}|  j	|j
d d@ 7  _	q4| jrK| jnt }|jd	ura|jd	ura|j| _|j| _nt| d
rvtd| jdd \| _| _nd | _| _|jrt|j| _| jdd | j | _t|j}t|| _| j| n	d | _ | _| _|jr| jdd | j }|d d@ }| jd||   | j | _t|j| _| j|j nd | _| _| jdd | j | _n| jrJ d| j 	 t !| j"| }| j| j#| j$  d}	|D ]\}}|dkrd}	||%  }q|	r.| &| | j| j# | j| d	S )zTAll tables must have been written to disk. Now write the
        directory.
        z-wrong number of tables; expected %d, found %dr   r   r   rT   r   rt   r   Nr   z>HHr   rR   r{   r   r   rQ   r   )'r;   r=   r<   r1   r8   r   r)   r   reservedtotalSfntSizer   r*   r>   majorVersionminorVersionhasattrstructr2   r   metaDatametaOrigLengthr'   r   rf   
metaOffsetr~   
metaLengthr   privData
privOffset
privLengthr   r   packr   r   r   toStringwriteMasterChecksum)
r?   r=   r:   rC   rB   compressedMetaDataoff	paddedOff	directoryseenHeadr   r   r   r^   4  sn   







zSFNTWriter.closec                 C  s   g }| j  D ]}|| j | j q| jtkrWddlm} || jd\| _	| _
| _tt| }t| j  }|D ]\}}t }|j|_|j|_|j|_|j|_||  }q9tt| j t  }|t|kshJ |t| t|d@ }	d|	 d@ }
|
S )Nr   r   r       l   /ac )r=   rN   appendrW   r,   r+   r   r   r8   r   r   r   r   r   r3   r;   r<   r:   r   r#   r   r   r   r0   r1   r   rV   sum)r?   r   	checksumsr:   r   r=   rC   	sfntEntrydirectory_endrZ   checksumadjustmentr   r   r   _calcMasterChecksum{  s.   
zSFNTWriter._calcMasterChecksumc                 C  s:   |  |}| j| jd jd  | jtd| d S )NrQ   rR   z>L)r   r'   r   r=   r#   r   r   r   )r?   r   r   r   r   r   r     s   
zSFNTWriter.writeMasterChecksumc                 C  s   dS NFr   rO   r   r   r   reordersTables     zSFNTWriter.reordersTables)r    NN)ro   rp   rq   r   rD   r   r   r[   r^   r   r   r   r   r   r   r   r      s    
1G r   a  
		> # big endian
		TTCTag:                  4s # "ttcf"
		Version:                 L  # 0x00010000 or 0x00020000
		numFonts:                L  # number of fonts
		# OffsetTable[numFonts]: L  # array with offsets from beginning of file
		# ulDsigTag:             L  # version 2.0 only
		# ulDsigLength:          L  # version 2.0 only
		# ulDsigOffset:          L  # version 2.0 only
z
		> # big endian
		sfntVersion:    4s
		numTables:      H    # number of tables
		searchRange:    H    # (max2 <= numTables)*16
		entrySelector:  H    # log2(max2 <= numTables)
		rangeShift:     H    # numTables*16-searchRange
zc
		> # big endian
		tag:            4s
		checkSum:       L
		offset:         L
		length:         L
ab  
		> # big endian
		signature:      4s   # "wOFF"
		sfntVersion:    4s
		length:         L    # total woff file size
		numTables:      H    # number of tables
		reserved:       H    # set to 0
		totalSfntSize:  L    # uncompressed size
		majorVersion:   H    # major version of WOFF file
		minorVersion:   H    # minor version of WOFF file
		metaOffset:     L    # offset to metadata block
		metaLength:     L    # length of compressed metadata
		metaOrigLength: L    # length of uncompressed metadata
		privOffset:     L    # offset to private data block
		privLength:     L    # length of private data block
z
		> # big endian
		tag:            4s
		offset:         L
		length:         L    # compressed length
		origLength:     L    # original length
		checkSum:       L    # original checksum
c                   @  sT   e Zd Zdd Zdd Zdd Zdd Zd	d
 Zdd Zdd Z	dd Z
dd ZdS )r,   c                 C  s
   d| _ d S r   )r   rO   r   r   r   rD     rL   zDirectoryEntry.__init__c                 C  s   t | j|| j|  d S rH   )r   r2   formatr   r   )r?   r'   r   r   r   r9     s   zDirectoryEntry.fromFilec                 C  s   t | j||  d S rH   )r   r2   r   )r?   strr   r   r   
fromString  s   zDirectoryEntry.fromStringc                 C  s   t | j| S rH   )r   r   r   rO   r   r   r   r     r_   zDirectoryEntry.toStringc                 C  s6   t | drd| jj| jt| f S d| jjt| f S )Nr:   z<%s '%s' at %x>z
<%s at %x>)r   	__class__ro   r:   idrO   r   r   r   __repr__  s   
zDirectoryEntry.__repr__c                 C  sD   | | j || j}t|| jksJ t| jdr | |}|S )N
decodeData)r   r#   r   r   r1   r   r   r   r?   r'   rB   r   r   r   rU     s   
zDirectoryEntry.loadDatac                 C  s:   t | jdr| |}t|| _|| j || d S )N
encodeData)r   r   r   r1   r   r   r#   r   r   r   r   r   r     s
   

zDirectoryEntry.saveDatac                 C     |S rH   r   )r?   rawDatar   r   r   r   
  r   zDirectoryEntry.decodeDatac                 C  r   rH   r   )r?   rB   r   r   r   r     r   zDirectoryEntry.encodeDataN)ro   rp   rq   rD   r9   r   r   r   rU   r   r   r   r   r   r   r   r,     s    r,   c                   @  s   e Zd ZeZeZdS )r+   N)ro   rp   rq   sfntDirectoryEntryFormatr   r   r   r   r   r   r   r+     s    r+   c                      s4   e Zd ZeZeZ fddZdd Zdd Z	  Z
S )r4   c                   s&   t t|   ttdst| _d S d S )NzlibCompressionLevel)superr4   rD   r   ZLIB_COMPRESSION_LEVELr   rO   r   r   r   rD     s   

zWOFFDirectoryEntry.__init__c                 C  sL   dd l }| j| jkr|}|S | j| jk sJ ||}t|| jks$J |S Nr   )r   r   r   
decompressr1   )r?   r   r   rB   r   r   r   r   %  s   
zWOFFDirectoryEntry.decodeDatac                 C  sR   t || _| jst|| j}| jst || jkr |}| j| _|S |}t || _|S rH   )r1   r   r   r~   r   r   )r?   rB   compressedDatar   r   r   r   r   0  s   

zWOFFDirectoryEntry.encodeData)ro   rp   rq   woffDirectoryEntryFormatr   woffDirectoryEntrySizer   rD   r   r   __classcell__r   r   r   r   r4     s    r4   c                   @  s"   e Zd ZdZdddZdd ZdS )r>   r   Nc                 C  s   d | _ d | _d | _d | _|r`|j | _ |j| _|jrA|j|j |j|j}t	||jks0J | 
|}t	||jks>J || _|jrb|j|j |j|j}t	||jks[J || _d S d S d S rH   )r   r   r   r   r   r'   r   r   r   r1   _decompressr   r   r   )r?   readerr   rB   r   r   r   rD   A  s*   


zWOFFFlavorData.__init__c                 C  s   dd l }||S r   )r   r   )r?   r   r   r   r   r   r   V  s   
zWOFFFlavorData._decompressrH   )ro   rp   rq   FlavorrD   r   r   r   r   r   r>   >  s    
r>   c                 C  s   t | d }|r| dd|  7 } d}d}|d dksJ tdt | |D ]}| |||  }tdt |d  |}|t| d@ }q$|S )a,  Calculate the checksum for an arbitrary block of data.

    If the data length is not a multiple of four, it assumes
    it is to be padded with null byte.

            >>> print(calcChecksum(b"abcd"))
            1633837924
            >>> print(calcChecksum(b"abcdxyz"))
            3655064932
    r   r   r   i   >%dLr   )r1   r7   r   r2   r   )rB   	remaindervalue	blockSizer$   blocklongsr   r   r   rV   \  s   rV   c                 C  s   |  d | t}t|tkrtdt }tt|| |j	dkr'td|j
dks8|j
dks8J d|j
 td|j | |jd	 |_|j
dkrN	 |S )
Nr   r   ttcfzNot a Font Collection   i   zunrecognized TTC version 0x%08xr   r   )r   r   ttcHeaderSizer1   r   r   r   r2   ttcHeaderFormatTTCTagVersionr   r.   r/   )r'   rB   r?   r   r   r   r-   t  s"   



r-   c                 C  sf   t  }d|_d|_||_| d | tt| | 	 }| t
jd|j gdg|j R   |S )Nr   r   r   r   )r   r   r   r.   r   r   r   r   r   rf   r   )r'   r.   r?   r#   r   r   r   writeTTCHeader  s   
&r   __main__)5__doc__
__future__r   collections.abcr   ior   typesr   fontTools.misc.textToolsr   fontTools.miscr   r   r   r	   r   collectionsr
   logging	getLoggerro   rX   r   r   r   r   r   r~   r   r   calcsizer   r3   r0   r   r   r6   r5   r   r   r,   r+   r4   r>   rV   r-   r   sysdoctestexittestmodfailedr   r   r   r   <module>   sf    
  _

	


	)(