o
    }ck                     @   sB   d dl Z d dlZd dlZd dlmZ d dlmZ G dd dZdS )    N)Alfit)Dofrec                   @   s8   e Zd Zdd Zdd Zdd Zddd	Zdd
dZdS )Tifinc                 C   s.   || _ t| j | _t| j | _d| _d | _d S )NF)er   alfitr   dofre
do_verbosedofs)selferimp r   !/home/ernad/ernad/python/tifin.py__init__   s   zTifin.__init__c                 C   s0   d}|D ]}||vrq||| ||  7 }q|S )z)--> cross product for two alfits by papidr   r   )r
   fits1fits2outfitr   r   r   cropro   s   zTifin.croproc                 C   sf   | j d urdS | j }|d u rtdtjd t  td| d ddd t|| _ td	dd
 dS )NFz-tifin: there are no dofres. I can't continue.)fileztifin loads z ... T )flushenddone)r   )	r	   r   fufiprintsysstderrquitfilerload)r
   r   r   r   r   	load_dofs   s   

zTifin.load_dofsFc           
      C   s   | j  s| jdu r|   d}i }i }|D ]}d|d }|| | }|||< ||7 }qt|}	|D ]3}t|| |	 ||< | jdusL| j  sh|| jvr]| jr\td| d q5|| | j|  ||< q5|S )z$parts copied from vemli's feed_papidNr       '' has no dof.)	r   
is_testingr	   r    countmathsqrtr   r   )
r
   fitsno_dofssumftefsr   r   lengthadd	root_sumfr   r   r   beaks,   s*   


zTifin.beaksc                 C   s   | j  s| jdu r|   d}i }i }|D ]}|| ||< ||| 7 }qt|}|D ]3}t|| | ||< | jdusC| j  s_|| jvrT| jrStd| d q,|| | j|  ||< q,|S )z=same as beaks, but takes no accounts of word count in featureNr   r#   r$   )r   r%   r	   r    r'   r(   r   r   )r
   r)   r*   r+   r,   r   r   r/   r   r   r   bunksH   s&   

zTifin.bunksN)F)__name__
__module____qualname__r   r   r    r0   r1   r   r   r   r   r      s    	
r   )r'   r   r   r   r   r   r   r   r   r   r   r   <module>   s   