o Wg#@spdZddlmZmZddlmZdZddZdd Zdd d Z d dZ ddZ dddZ dddZ dddZd S)z]Utility functions useful for writers or QR Code objects. DOES NOT belong to the public API. )chainrepeat)consts)get_default_border_size get_borderget_symbol_sizecheck_valid_scalecheck_valid_bordermatrix_to_lines matrix_itermatrix_iter_verbosecCs |\}}|dkr||krdSdS)z Returns the default border size (quiet zone) for the provided version. :param tuple(int, int) matrix_size: Tuple of width and height of the matrix. :rtype: int ) matrix_sizewidthheightrrH/var/www/rescue_company/venv/lib/python3.10/site-packages/segno/utils.pyrsrcCs|dur|St|S)a& Returns `border` if not ``None``, otherwise the default border size for the provided QR Code. :param tuple(int, int) matrix_size: Tuple of width and height of the matrix. :param border: The size of the quiet zone or ``None``. :type border: int or None :rtype: int Nr)rborderrrrr s rNcCs@|durt|}|\}}|d|7}|d|7}||||fS)a Returns the symbol size (width x height) with the provided border and scaling factor. :param tuple(int, int) matrix_size: Tuple of width and height of the matrix. :param scale: Indicates the size of a single module (default: 1). The size of a module depends on the used output format; i.e. in a PNG context, a scaling factor of 2 indicates that a module has a size of 2 x 2 pixel. Some outputs (i.e. SVG) accept floating point values. :type scale: int or float :param int border: The border size or ``None`` to specify the default quiet zone (4 for QR Codes, 2 for Micro QR Codes). :rtype: tuple (width, height) Nrr)rscalerrrrrrr.s   rcCs|dkr td|ddS)z Raises a :py:exc:`ValueError` iff `scale` is negative or zero. :param scale: Scaling factor. :type scale: float or int rz.The scale must not be negative or zero. Got: ""N) ValueError)rrrrr Fsr cCs4|durt||ks|dkrtd|ddSdS)z Raises a :py:exc:`ValueError` iff `border` is negative. :param int border: Indicating the size of the quiet zone. Nrz8The border must not a non-negative integer value. Got: "r)intr)rrrrr Qsr c cs||8}d}|D]:}||}}||7}|D]}||kr)|s)||f||ffV|}|d7}|s3|d7}|}q|rC||f||ffVd}q dS)a Converts the `matrix` into an iterable of ((x1, y1), (x2, y2)) tuples which represent a sequence (horizontal line) of dark modules. The path starts at the 1st row of the matrix and moves down to the last row. :param matrix: An iterable of bytearrays. :param x: Initial position on the x-axis. :param y: Initial position on the y-axis. :param incby: Value to move along the y-axis (default: 1). :rtype: iterable of (x1, y1), (x2, y2) tuples rrNr) matrixxyincbylast_bitrowx1x2bitrrrr [s&   r c #st|ttt||}|\}dg}t| |t| ||}}|D]/}d|kr:|kr@nn||n|ttfdd|D} tdD]} | VqWq.dS)al Returns an iterator / generator over the provided matrix which includes the border and the scaling factor. If either the `scale` or `border` value is invalid, a :py:exc:`ValueError` is raised. :param matrix: An iterable of bytearrays. :param tuple(int, int) matrix_size: Tuple of width and height of the matrix. :param int scale: The scaling factor (default: ``1``). :param int border: The border size or ``None`` to specify the default quiet zone (4 for QR Codes, 2 for Micro QR Codes). :raises: :py:exc:`ValueError` if an illegal scale or border value is provided rc3s:|]}td|krkrnn|ndVqdS)rNr.0jrrrrr s8zmatrix_iter..N) r rr rrangetupler from_iterabler) rrrrr border_row width_range height_rangeir!srr)rr zs  "$ r c #sddlm}t|ttt||}|\ko%dk|jddd|fddt| |t| |}}|D]t t fdd |D}t d D]}|VqkqTd S) a Returns an iterator / generator over the provided matrix which includes the border and the scaling factor. This iterator / generator returns different values for dark / light modules and therefor the different parts (like the finder patterns, alignment patterns etc.) are distinguishable. If this information isn't necessary, use the :py:func:`matrix_iter()` function because it is much cheaper and faster. If either the `scale` or `border` value is invalid, a py:exc:`ValueError` is raised. :param matrix: An iterable of bytearrays. :param tuple(int, int) matrix_size: Tuple of width and height of the matrix. :param int scale: The scaling factor (default: ``1``). :param int border: The border size or ``None`` to specify the default quiet zone (4 for QR Codes, 2 for Micro QR Codes). :raises: :py:exc:`ValueError` if an illegal scale or border value is provided r)encoderF)reserve_regions add_timingcspd|kr kr5ntjSd|krkr5ntjS||}sz||}|dkr8tjtjf|Srmdkrm|dkrQd|krPdksend|kr_dkrmnn |dkrmtjtjf|S|dkrz|dkrztjSs|dkrd|krdksn|dkrd|krdksnr|dkr|dks|dkr|dkrtjtjf|S|dkr|dks؈s|d ks|dkr|dks؈s|dkrtjtj f|S|dkr|dkss|dkss|dkr|dkrtj tj f|S|dkr|dks*s|dks*s-|dkr-|dkr-tj Stj tjf|StjS) Nrr)  )rTYPE_ALIGNMENT_PATTERN_LIGHTTYPE_ALIGNMENT_PATTERN_DARKTYPE_VERSION_LIGHTTYPE_VERSION_DARKTYPE_DARKMODULETYPE_TIMING_LIGHTTYPE_TIMING_DARKTYPE_FORMAT_LIGHTTYPE_FORMAT_DARKTYPE_FINDER_PATTERN_LIGHTTYPE_FINDER_PATTERN_DARKTYPE_SEPARATORTYPE_DATA_LIGHTTYPE_DATA_DARKTYPE_QUIET_ZONE)r2r(val alignment_val)alignment_matrixris_micro is_squarerrrrget_bitsF""   &(H    (z$matrix_iter_verbose..get_bitc3s |] }t|VqdS)Nr%r&)rSr2rrrr+sz&matrix_iter_verbose..N) segnor4r rr r make_matrixadd_alignment_patternsr,r-rr.r) rrrrr4r0r1r!r3r) rPrSrr2rQrRrrrrr s&   "& r )rN)r)__doc__ itertoolsrrr__all__rrrr r r r r rrrrs