kh[3dZddlZddlmZddlmZGddZGdd eZGd d eZGd d eZ GddeZ GddZ GddZ y)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&eZdZdZdZdedefdZy)Segmentz9A segment of path, e.g. a line or a rectangle or a curve.c|dd|_yNpoints)selfitems O/var/www/teggl/fontify/venv/lib/python3.12/site-packages/pdf2docx/shape/Path.py__init__zSegment.__init__'s12h widthcolorcgSN)r rrs r to_strokeszSegment.to_strokes*s"9rN__name__ __module__ __qualname____doc__rfloatlistrrrrrr%sC=ur>|sBrr>ceZdZdZddefdZdZedZedZ edZ edZ d e d efd Z d efd Zy )Segmentsz,A sub-path composed of one or more segments.itemsclg|_|D]}|ddk(r%|jjt|0|ddk(r%|jjt|]|ddk(r%|jjt ||ddk(s|jjt ||rpd|jdj d|jdj df}t|}|jdkDr|jj|yyy)NrlcrequMbP?) _instancesr+r r>r/r7r r&)r rA close_pathrlines rrzSegments.__init__s BDaC$//"8"84"AaC$//"8"84"AaD$//"8"84"AaD$//"8"84"A  B ,33B79K9R9RST9UVDT7D{{4!7!7!= rc(d|jDS)Nc3 K|]}|ywrr).0instances r z$Segments.__iter__..sIXIs )rIr s r__iter__zSegments.__iter__sIIIrcbg}|jD]}|j|j|S)zConnected points of segments.)rIextendr )r r segments rr zSegments.pointss1 *G MM'.. ) * rc|jj}|dk(xs |j|z tjk\S)zCISO-oriented criterion: the ratio of real area to bbox exceeds 0.9.r)bboxget_areaarear FACTOR_MOST)r bbox_areas ris_iso_orientedzSegments.is_iso_orienteds9II&&( !|Ityy2I4I4IIIrc"|j}|d|d}}t|d|dz t|d|dz zdkDryd}tt|dz D]#}||\}}||dz\}} ||| z||zz z }%t|dz S)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 rrGr rHr2)r absr3r4) r r r)r*rYr5r"r#r$r%s rrYz Segments.areasAYr s uQxA E!HSVO 4 4T 9s6{1}% "AAYFBAaC[FB BrEBrEM !D " 48}rc "|j}t|dd}t|dd}t|dd}t|dd}tjt |dt |dt |dt |dS) zCalculate segments bbox. c |dSNrrpoints rzSegments.bbox.. 58r)keyrc |dSr rrcs rrezSegments.bbox..rfrr c |dSrbrrcs rrezSegments.bbox..rfrc |dSr rrcs rrezSegments.bbox..rfrr)r minmaxfitzRectround)r r r"r#r$r%s rrWz Segments.bboxs 3 4Q 7 3 4Q 7 3 4Q 7 3 4Q 7yy "aL%A,b! eBlD Drrrcng}|jD]#}|j|j||%|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. )rIrTr)r rrr,rUs rrzSegments.to_strokess: =G NN7--eU; < =rcDt|jt|dS)zConvert segment closed area to a ``Fill`` dict. Args: color (list): Specify fill color. Returns: dict: ``Fill`` dict. )rWr)rrWr)r rs rto_fillzSegments.to_fills!$))_u%  rN)F)rrrrrrrRr-r r\rYrWrrrrrrrr@r@s6 >T >JJJ * D D u D   D  rr@ceZdZdZdefdZedZedZ edZ edZ dZ d e d efd Zd efd Zd Zy)Pathz=Path extracted from PDF, consist of one or more ``Segments``.rawc||_|d|_|jrdn|jdd}g|_t j |_|jdd}|j|dD]h}t||}|jj||j}|jdk(r || | ||fz }|xj|zc_jy ) 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 closePathFrr^rArN) ru path_typeis_fillgetrArmrnrW_group_segmentsr@r+rX)r rurJwsegmentsSrects rrz Path.__init__sV"\\Tsww{E/J  IIK GGGS !,,S\: H:.A JJ  a 66D}}!4QBAq>#94 II I rc&gg}}d}|D]r}|ddvr:|d|d}}|r||k(r|j|n|j||g}|}D|ddvsL|r|j|g}|j|gt|r|j||S)zGroup connected segments. Args: items (dict): Raw dict extracted from ``page.get_drawings()``. Returns: list: A list of segments list. Nr)rCrDr rG)rErF)r+)rAr~ segments_listcursorrr)r*s rr|zPath._group_segmentss#%b- -DAw*$!!Wd2hs 5&=OOD)"((2 $vHaL(!((2!H$$dV,3 -8 ]))(3rcd|jvS)NsryrQs r is_strokezPath.is_stroke3s #t~~ 55rcd|jvS)NfrrQs rrzz Path.is_fill6s!T^^33rcB|jD]}|jryy)z@It is iso-oriented when all contained segments are iso-oriented.FT)rAr\)r r~s rr\zPath.is_iso_oriented9s%  :H++E :rcdg}|jrY|jjdd}|jjdd}|j|j |||j r<|jjdd}|j|j ||S)zkConvert path to ``Shape`` raw dicts. Returns: list: A list of ``Shape`` dict. rNrr^fill)rrur{rT _to_strokesrz _to_fills)r iso_shapes stroke_colorr fill_colors r to_shapeszPath.to_shapesAs   >>88<<6LHHLL#.E   d..ulC D <<fd3J   dnnZ8 9rrrcng}|jD]#}|j|j||%|S)zmConvert path to ``Stroke`` raw dicts. Returns: list: A list of ``Stroke`` dict. )rArTr)r rrr,r~s rrzPath._to_strokesWs:   >H NN8..ue< = >rclg}|jD]"}|j|j|$|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. )rAr+rr)r rfillsr~s rrzPath._to_fillscs8  2H LL))%0 1 2 rc|jjdgD]}|ddk(r|j|d|d$|ddk(r|j|dA|ddk(r|j |d^|ddk(r!|j |d|d|d |d t d ||j|jjd d |jjdd |jjdd |jjdd|jjdd|jjddd|jvrt|jdnd|jjdd|jjdd|jjdd y )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 rArrCr rrErFrDzunhandled drawingrNrdasheseven_oddFrxlineJoinlineCaprstroke_opacity fill_opacity) rrrrrxrrrrr) rur{ draw_line draw_rect draw_quad draw_bezier ValueErrorfinishrl)r canvasrs rplotz Path.plotrsHHLL"- r@rtrrrrsj: $==  F  F++  d d Nmmr