o 0j . @sddlZddlZddlZddlZddlZddlmZmZddlZ ddl m Z m Z ddl mZddlmZmZmZddlmZdd lmZmZdd lmZmZmZmZmZmZmZed r`ddl Z ed Gd d d eZ!ed  ddee"e"fde j#de$de%de j#f ddZ&dS)N)DictTuple)Image ImageDraw)logging)class_requires_depsfunction_requires_depsis_dep_available) PINGFANG_FONT) BaseCVResult JsonMixin)crop_white_areadraw_box_txt_finedraw_formula_module env_validgenerate_pdf_filegenerate_tex_filepdf2imgzopencv-contrib-pythonc@sTeZdZdZdeeejffddZdeeeffddZdeeeffddZ d S) FormulaRecognitionResultzFormula Recognition Resultreturnc% Cs0t|dddddddddf}i}|d}|dr*|jd$i|dj|d}t|dkr;|jd |d<ztWntjy^}zt d ||d <|WYd}~Sd}~wwt|dkrt | d }|d dd}t |}|dur|\}} } } || | | ||| f}t|}|j\} } ddg| dg| | gd| gg}zPt|j||dd}t|}|j\}}|}t|| | }|||ftj}|jt|d}td ||fd}||d|||jddf||d <|WStjy}zt d||d <|WYd}~Sd}~ww|j|j} } |}t j| | dft jdd}tdt|}|d tt|d D]}|d |}|dt|d}|d}t |\}}}} zdt!ddt!ddt!ddf}!|||||| || g}t |"ddg}dd|#D}"|j$|"|!dd|j$||!dt%| | f||dd}#t |t j&"d }"t'(|#|"gd!|!d"t')||#}WqHtjy}z t dWYd}~qHd}~wwt*||d#}td t| d| fd}$|$|dd| | f|$t|| d| d| f|$|d <|S)%z Converts the internal data to a PIL Image with detection and recognition results. Returns: Dict[str, Image.Image]: An image with detection boxes, texts, and scores blended on it. doc_preprocessor_resZ output_imgNmodel_settingsuse_doc_preprocessorlayout_det_resrreszPlease refer to 2.3 Formula Recognition Pipeline Visualization in Formula Recognition Pipeline Tutorial to install the LaTeX rendering engine at first.Zformula_res_imgRGBformula_res_list rec_formulaF)is_debug r$r$)rrz3Syntax error detected in formula, rendering failed.r Zdtyper$formula_region_iddt_polyscSsg|]\}}||fqSr)).0xyr)r)w/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddlex/inference/pipelines/formula_recognition/result.py sz4FormulaRecognitionResult._to_img..)outlinewidth)fill)rr(Tr3g?r))+rZ fromarrayupdateZimglenr subprocessCalledProcessErrorrwarningnparrayconvertrsizerintresizeZLANCZOSr1newZpasteheightcopyZonesuint8randomseedrZDrawrangestrlistrandintZreshapetolistZpolygondraw_box_formula_fineZint32cv2Z polylinesZ bitwise_andblend)%selfimageZ res_img_dictrrer Zxywhr+r,whZ image_widthZ image_heightboxZ img_formula render_width render_height resize_height resize_widthZnew_image_widthZ new_imageZimg_leftZ img_rightZ draw_lefttnoZ formula_resformular'x1y1Zx2y2colorZptsimg_right_textZimg_showr)r)r-_to_img/s*                   z FormulaRecognitionResult._to_imgcOsi}|d|d<|d|d<|d|d<|ddr#|djd|d<t|ddkr4|djd|d<g|d <tt|d D]-}|d |d |d |d d }d |d |vrf|d |d f|d <|d |q@tj|g|Ri|S)axConverts the instance's attributes to a dictionary and then to a string. Args: *args: Additional positional arguments passed to the base class method. **kwargs: Additional keyword arguments passed to the base class method. Returns: Dict[str, str]: A dictionary with the instance's attributes converted to strings. input_path page_indexrrrrrrrr r&r r&r')rFr5rEappendr_to_strrMargskwargsdatarWZrec_formula_dictr)r)r-rcs$    z FormulaRecognitionResult._to_strcOsi}|d|d<t|d|d<|d|d<|ddr%|djd|d<t|ddkr6|djd|d<g|d <tt|d D]-}|d |d |d |d d }d |d |vrh|d |d f|d <|d |qBtj|g|Ri|S)a[ Converts the object's data to a JSON dictionary. Args: *args: Positional arguments passed to the JsonMixin._to_json method. **kwargs: Keyword arguments passed to the JsonMixin._to_json method. Returns: Dict[str, str]: A dictionary containing the object's data in JSON format. r_r`rrrrrrrr r&rar')rFr5rErbr_to_jsonrdr)r)r-rhs$   z!FormulaRecognitionResult._to_jsonN) __name__ __module__ __qualname____doc__rrFrr^rcrhr)r)r)r-r+s j rFimg_sizerRrXr!rc Cs tt|dd|ddd|dd|ddd}tt|dd|ddd|dd|ddd}t}tj|d}tj|d}tj|d} t||tj |rnt |||d} tj |r}t || d d } | dur| j dd \} } |} | | | }t | t|t| f} | j dd \} } tddg|dg||gd|gg}tj|tjd }t ||}tj| tjd } t j| ||t jt jd d}nt||dtj}|WdS1swYdS)zdraw box formula for pipelinerr r(r3ztemp.texztemp.pdfztemp.jpgNF)Z is_paddingrr%r#)flagsZ borderModeZ borderValuezRendering Failed)r=mathsqrttempfileTemporaryDirectoryospathjoinrexistsrrshaperKr>r9Zfloat32r:ZgetPerspectiveTransformrBZwarpPerspectiveZ INTER_NEARESTZBORDER_CONSTANTrr )rmrRrXr!Z box_heightZ box_widthtdZ tex_file_pathZ pdf_file_pathZ img_file_pathZ formula_imgZ formula_hZ formula_wrUrVZpts1Zpts2Mr]r)r)r-rJsX<<        $rJ)F)'rorsrCr6rqtypingrrnumpyr9ZPILrrutilsrZ utils.depsrr r Z utils.fontsr Z common.resultr rZ!models.formula_recognition.resultrrrrrrrrKrr=ZndarrayrFboolrJr)r)r)r-s>  $ 0