o j2@sdZddlZddlmZddlmZGdddZGdd d eZGd d d eZGd d d eZ GdddeZ GdddZ GdddZ dS)u Objects representing PDF path (stroke and filling) extracted from pdf drawings and annotations. Data structure based on results of ``page.get_drawings()``:: { 'color': (x,x,x) or None, # stroke color 'fill' : (x,x,x) or None, # fill color 'width': float, # line width 'closePath': bool, # whether to connect last and first point 'rect' : rect, # page area covered by this path 'items': [ # list of draw commands: lines, rectangle or curves. ("l", p1, p2), # a line from p1 to p2 ("c", p1, p2, p3, p4), # cubic Bézier curve from p1 to p4, p2 and p3 # are the control points ("re", rect), # a rect represented with two diagonal points ("qu", quad) # a quad represented with four corner points ], ... } References: - https://pymupdf.readthedocs.io/en/latest/page.html#Page.get_drawings - https://pymupdf.readthedocs.io/en/latest/faq.html#extracting-drawings .. note:: The coordinates extracted by ``page.get_drawings()`` is based on **real** page CS, i.e. with rotation considered. This is different from ``page.get_text('rawdict')``. N) rgb_value) constantsc@*eZdZdZddZdedefddZdS) Segmentz9A segment of path, e.g. a line or a rectangle or a curve.cCs|dd|_dSNpoints)selfitemr T/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/pdf2docx/shape/Path.py__init__&szSegment.__init__widthcolorcCsgSNr )r rrr r r to_strokes)szSegment.to_strokesN__name__ __module__ __qualname____doc__rfloatlistrr r r rr$src@s.eZdZdZeddZdedefddZdS) Lz(Line path with source ``("l", p1, p2)``.cCs8|jd\}}|jd\}}||d||ddS)zLength of line.rrrg?r )r x0y0x1y1r r rlength/szL.lengthrrcCs,g}||jd|jd|t|d|S)aConvert to stroke dict. Args: width (float): Specify width for the stroke. color (list): Specify color for the stroke. Returns: list: A list of ``Stroke`` dicts. .. note:: A line corresponds to one stroke, but considering the consistence, the return stroke dict is append to a list. So, the length of list is always 1. rrstartendrr)appendr r)r rrstrokesr r rr7sz L.to_strokesN) rrrrpropertyr rrrr r r rr,s  rc@r) Rz'Rect path with source ``("re", rect)``.cCs8|d\}}}}||f||f||f||f||fg|_dSrr )r r rrrrr r rrRs z R.__init__rrcCsLg}tt|jdD]}||j||j|d|dt|dq |S)aAConvert each edge to stroke dict. Args: width (float): Specify width for the stroke. color (list): Specify color for the stroke. Returns: list: A list of ``Stroke`` dicts. .. note:: One Rect path is converted to a list of 4 stroke dicts. r@r!)rangelenr r$r)r rrr%ir r rr[s   z R.to_strokesNrr r r rr'Ps r'c@seZdZdZddZdS)Qz'Quad path with source ``("qu", quad)``.cCs$|d\}}}}|||||g|_dSrr )r r ZulurZlllrr r rrusz Q.__init__N)rrrrrr r r rr,ss r,c@seZdZdZdS)Cz8Bezier curve path with source ``("c", p1, p2, p3, p4)``.N)rrrrr r r rr/|sr/c@sxeZdZdZddefddZddZedd Zed d Z ed d Z eddZ de defddZ defddZdS)Segmentsz,A sub-path composed of one or more segments.FitemscCsg|_|D]=}|ddkr|jt|q|ddkr%|jt|q|ddkr4|jt|q|ddkrB|jt|q|rgd|jdjd|jdjdf}t|}|jdkri|j|dSdSdS)NrlcrequMbP?) _instancesr$rr/r'r,r r )r r1 close_pathr liner r rrs"zSegments.__init__cCsdd|jDS)Ncss|]}|VqdSrr ).0instancer r r sz$Segments.__iter__..)r8r r r r__iter__szSegments.__iter__cCs g}|jD]}||jq|S)zConnected points of segments.)r8extendr )r r segmentr r rr s zSegments.pointscCs"|j}|dkp|j|tjkS)zCISO-oriented criterion: the ratio of real area to bbox exceeds 0.9.r)bboxget_areaarearZ FACTOR_MOST)r Z bbox_arear r ris_iso_orienteds zSegments.is_iso_orientedc Cs|j}|d|d}}t|d|dt|d|ddkr$dSd}tt|dD]}||\}}||d\}} ||| ||7}q.t|dS)zCalculate segments area with Green formulas. Note the boundary of Bezier curve is simplified with its control points. * https://en.wikipedia.org/wiki/Shoelace_formula rr6rr7r()r absr)r*) r r r"r#rDr+rrrrr r rrDs,  z Segments.areacCs~|j}t|dddd}t|dddd}t|dddd}t|dddd}tt|d t|d t|d t|d S) zCalculate segments bbox. cS|dSNrr pointr r rzSegments.bbox..)keyrcSrHrr rJr r rrLrMrcSrHrIr rJr r rrLrMcSrHrr rJr r rrLrMr)r minmaxfitzRectround)r r rrrrr r rrBs z Segments.bboxrrcC&g}|jD] }||||q|S)zConvert each segment to a ``Stroke`` dict. Args: width (float): Specify stroke width. color (list): Specify stroke color. Returns: list: A list of ``Stroke`` dicts. )r8r@r)r rrr%rAr r rrs zSegments.to_strokescCst|jt|dS)zConvert segment closed area to a ``Fill`` dict. Args: color (list): Specify fill color. Returns: dict: ``Fill`` dict. )rBr)rrBr)r rr r rto_fills zSegments.to_fillN)F)rrrrrrr?r&r rErDrBrrrUr r r rr0s     r0c@s~eZdZdZdefddZeddZeddZ ed d Z ed d Z d dZ de defddZdefddZddZdS)Pathz=Path extracted from PDF, consist of one or more ``Segments``.rawcCs||_|d|_|jr dn|dd}g|_t|_|dd}||dD]'}t ||}|j ||j}| dkrH|| | ||f7}|j|O_q(d S) zInit path in real page CS. Args: raw (dict): Raw dict extracted with `PyMuPDF`, see link https://pymupdf.readthedocs.io/en/latest/page.html#Page.get_drawings typeT closePathFrrFr1rN) rW path_typeis_fillgetr1rQrRrB_group_segmentsr0r$rC)r rWr9wsegmentsSrectr r rrs      z Path.__init__cCsgg}}d}|D]=}|ddvr1|d|d}}|r ||kr&||n|||g}|}q |ddvrF|r@||g}||gq |rN|||S)zGroup connected segments. Args: items (dict): Raw dict extracted from ``page.get_drawings()``. Returns: list: A list of segments list. Nr)r2r3rr6)r4r5)r$)r1r_Z segments_listcursorr r"r#r r rr]s$        zPath._group_segmentscC d|jvS)NsrZr>r r r is_stroke2 zPath.is_strokecCrc)Nfrer>r r rr[5rgz Path.is_fillcCs|jD]}|js dSqdS)z@It is iso-oriented when all contained segments are iso-oriented.FT)r1rE)r r_r r rrE8s zPath.is_iso_orientedcCsdg}d}|jr|jd|}|jdd}|||||jr0|jd|}||||S)zkConvert path to ``Shape`` raw dicts. Returns: list: A list of ``Shape`` dict. )rrrrrrFfill)rfrWr\r@ _to_strokesr[ _to_fills)r Z iso_shapeswhiteZ stroke_colorrZ fill_colorr r r to_shapes@szPath.to_shapesrrcCrT)zmConvert path to ``Stroke`` raw dicts. Returns: list: A list of ``Stroke`` dict. )r1r@r)r rrr%r_r r rrjWs zPath._to_strokescCs$g}|jD] }|||q|S)zConvert path to ``Fill`` raw dicts. Returns: list: A list of ``Fill`` dict. .. note:: The real filling area of this path may be not a rectangle. )r1r$rU)r rZfillsr_r r rrkcs zPath._to_fillscCs8|jdgD]J}|ddkr||d|dq|ddkr(||dq|ddkr6||dq|ddkrM||d|d|d |d qtd ||j|jd d |jdd |jdd |jdd|jdd|jddd|jvrt|jdnd|jdd|jdd|jddd d S)z Plot path for debug purpose. Args: canvas: ``PyMuPDF`` drawing canvas by ``page.new_shape()``. Reference: https://pymupdf.readthedocs.io/en/latest/faq.html#extracting-drawings r1rr2rrr4r5r3zunhandled drawingriNrdasheseven_oddFrYlineJoinlineCaprstroke_opacity fill_opacity) rirrprqrYrrrsrrtru) rWr\Z draw_lineZ draw_rectZ draw_quadZ draw_bezier ValueErrorfinishrP)r Zcanvasr r r rplotrs,    "           z Path.plotN)rrrrdictr staticmethodr]r&rfr[rErmrrrjrkrxr r r rrVs ,    rV) rrQZ common.sharercommonrrrr'r,r/r0rVr r r rs  $# g