U :vh/@sdZddlmZddlmZmZmZmZmZddl m Z m Z m Z m Z ddlmZmZmZerhddlmZGdd d eZGd d d eZGd d d eZGdddeZGdddeZGdddeZGdddeZdS)z0 Handles the creation of patterns and gradients )ABC)ListOptional TYPE_CHECKINGTupleUnion) DeviceCMYK DeviceGray DeviceRGBconvert_to_device_color)NamePDFArray PDFObject)FPDFcs6eZdZdZeddfdd ZeddZZS)Patternz Represents a PDF Pattern object. Currently, this class supports only "shading patterns" (pattern_type 2), using either a linear or radial gradient. Tiling patterns (pattern_type 1) are not yet implemented. )LinearGradientRadialGradient)shadingcs$ttd|_d|_||_dS)Nr)super__init__r typeZ pattern_type_shading)selfr __class__0/tmp/pip-unpacked-wheel-dvf6lv8i/fpdf/pattern.pyrs  zPattern.__init__cCs|jjdS)N 0 R)rget_shading_objectidrrrrrszPattern.shading) __name__ __module__ __qualname____doc__rrpropertyr __classcell__rrrrrsrcs eZdZdZfddZZS) Type2FunctionzTransition between 2 colorscs`td|_d|_dddd|jDd|_dddd|jDd|_d |_dS) Nr[0 1][ css|]}|dVqdS.2fNr.0crrr ,sz)Type2Function.__init__..]css|]}|dVqdSr-rr/rrrr2-sr) rr function_typedomainjoincolorsZc0Zc1n)rZcolor_1Zcolor_2rrrr's    zType2Function.__init__)r#r$r%r&rr(rrrrr)$sr)cs,eZdZdZfddZeddZZS) Type3FunctionzWhen multiple colors are used, a type 3 function is necessary to stitch type 2 functions together and define the bounds between each color transitioncsbtd|_d|_||_dddd|Dd|_dddd|Dd|_d |_dS) Nr*r+r,css|]}|dVqdSr-r)r0boundrrrr2;sz)Type3Function.__init__..r3css|] }dVqdS)z0 1Nr)r0_rrrr2<sr) rrr4r5 _functionsr6boundsencoder8)r functionsr>rrrr5s zType3Function.__init__cCsdddd|jDdS)Nr+r,css|]}|jdVqdS)rN)r!)r0frrrr2Asz*Type3Function.functions..r3)r6r=r"rrrr@?szType3Function.functions)r#r$r%r&rr'r@r(rrrrr91s r9csHeZdZeeeeeefe e eee e fe e dfdd ZZS)Shading shading_type background color_spacecoordsfunction extend_before extend_aftercs~t||_|r2dddd|jDdnd|_t||_||_|j d|_ d|r`dndd|rndndd|_ dS) Nr+r,css|]}|dVqdSr-rr/rrrr2Rsz#Shading.__init__..r3rtruefalse) rrrDr6r7rEr rFrGr!rHextend)rrDrErFrGrHrIrJrrrrEs   zShading.__init__)r#r$r%intrrr r r strrr)r9boolrr(rrrrrBDs rBc@sFeZdZddZeeeefdddZddZ dd Z d d Z d S) Gradientcs|\__d_|r@t|ttttfr6t |nt |_jr^jj j jkr^t d|_ |_|rr|nfddttjdD_tjtjdkrt d_t_d_d_d_dS)NzDThe background color must be of the same color space as the gradientcs"g|]}|dtjdqS)r)lenr7)r0ir"rr osz%Gradient.__init__..rz>Bounds array length must be two less than the number of colorsr)_convert_colorsrFr7rE isinstancerOr r r r rr# ValueErrorrIrJrangerRr>_generate_functionsr@rpattern_shading_objectrGrD)rr7rErIrJr>rr"rr]s4   zGradient.__init__)returncCsg}t|dkrtdt}|D]@}t|ttttfr@t|nt|}| || t |j q"t|dkr|| |fSd|krtdg}|D]2}t|tr| t|j|j|jq| |qd|fS)Nr(A gradient must have at least two colorsrr z'Can't mix CMYK with other color spaces.r )rRrWsetrVrOr r r r appendaddrr#popg)clsr7Z color_listZ color_spacescolorZ current_colorZ convertedrrrrU{s*       zGradient._convert_colorscCst|jdkrtdt|jdkr)rZnumber_of_colorsr@rSrrrrYs "zGradient._generate_functionsc Cs:|js4t|j|j|jt|j|jd|j|j d|_|jS)NrC) r[rBrDrErFrrGr@rIrJr"rrrr s zGradient.get_shading_objectcCs|jS)N)rZr"rrr get_patternszGradient.get_patternN) r#r$r%r classmethodrrOrrUrYr rfrrrrrQ\s   rQc s8eZdZddeeeeeeeeed fdd ZZS)rNFr) fpdffrom_xfrom_yto_xto_yr7rIrJr>c sJt|||| | |j||j|g} fdd| D|_d|_dS)a A shading pattern that creates a linear (axial) gradient in a PDF. The gradient is defined by two points: (from_x, from_y) and (to_x, to_y), along which the specified colors are interpolated. Optionally, you can set a background color, extend the gradient beyond its start or end, and specify custom color stop positions via `bounds`. Args: fpdf (FPDF): The FPDF instance used for PDF generation. from_x (int or float): The x-coordinate of the starting point of the gradient, in user space units. from_y (int or float): The y-coordinate of the starting point of the gradient, in user space units. to_x (int or float): The x-coordinate of the ending point of the gradient, in user space units. to_y (int or float): The y-coordinate of the ending point of the gradient, in user space units. colors (List[str or Tuple[int, int, int]]): A list of colors along which the gradient will be interpolated. Colors may be given as hex strings (e.g., "#FF0000") or (R, G, B) tuples. background (str or Tuple[int, int, int], optional): A background color to use if the gradient does not fully cover the region it is applied to. Defaults to None (no background). extend_before (bool, optional): Whether to extend the first color beyond the starting point (from_x, from_y). Defaults to False. extend_after (bool, optional): Whether to extend the last color beyond the ending point (to_x, to_y). Defaults to False. bounds (List[float], optional): An optional list of floats in the range (0, 1) that represent gradient stops for color transitions. The number of bounds should be two less than the number of colors (for multi-color gradients). Defaults to None, which evenly distributes color stops. csg|]}j|dqSr.kr/rhrrrTsz+LinearGradient.__init__..rNrrhrGrD) rrhrirjrkrlr7rErIrJr>rGrrprrs.zLinearGradient.__init__)NFFNr#r$r%rNrrPrr(rrrrrs rc s<eZdZddeeeeeeeeeeed fdd ZZS)rNFr) rhstart_circle_xstart_circle_ystart_circle_radius end_circle_x end_circle_yend_circle_radiusr7rIrJr>c sNt|| | | | |j|||j||g} fdd| D|_d|_dS)a A shading pattern that creates a radial (or circular/elliptical) gradient in a PDF. The gradient is defined by two circles (start and end). Colors are blended from the start circle to the end circle, forming a radial gradient. You can optionally set a background color, extend the gradient beyond its circles, and provide custom color stop positions via `bounds`. Args: fpdf (FPDF): The FPDF instance used for PDF generation. start_circle_x (int or float): The x-coordinate of the inner circle's center, in user space units. start_circle_y (int or float): The y-coordinate of the inner circle's center, in user space units. start_circle_radius (int or float): The radius of the inner circle, in user space units. end_circle_x (int or float): The x-coordinate of the outer circle's center, in user space units. end_circle_y (int or float): The y-coordinate of the outer circle's center, in user space units. end_circle_radius (int or float): The radius of the outer circle, in user space units. colors (List[str or Tuple[int, int, int]]): A list of colors along which the gradient will be interpolated. Colors may be given as hex strings (e.g., "#FF0000") or (R, G, B) tuples. background (str or Tuple[int, int, int], optional): A background color to display if the gradient does not fully cover the region it's applied to. Defaults to None (no background). extend_before (bool, optional): Whether to extend the gradient beyond the start circle. Defaults to False. extend_after (bool, optional): Whether to extend the gradient beyond the end circle. Defaults to False. bounds (List[float], optional): An optional list of floats in the range (0, 1) that represent gradient stops for color transitions. The number of bounds should be one less than the number of colors (for multi-color gradients). Defaults to None, which evenly distributes color stops. csg|]}j|dqSrmrnr/rprrrT$sz+RadialGradient.__init__..r:Nrq)rrhrtrurvrwrxryr7rErIrJr>rGrrprrs2zRadialGradient.__init__)NFFNrsrrrrrs" rN)r&abcrtypingrrrrrZdrawingr r r r Zsyntaxr rrrhrrr)r9rBrQrrrrrrs   W5