:Qg*2ddlZddlZddZ ddZdZdZy)Nctj|}tj|tj}|r:tj|dtj tj ||}||fStjtj|dtj tj ||}||fS)a\Thresholds an image using OpenCV's adaptiveThreshold. Parameters ---------- imagename : string Path to image file. process_background : bool, optional (default: False) Whether or not to process lines that are in background. blocksize : int, optional (default: 15) Size of a pixel neighborhood that is used to calculate a threshold value for the pixel: 3, 5, 7, and so on. For more information, refer `OpenCV's adaptiveThreshold `_. c : int, optional (default: -2) Constant subtracted from the mean or weighted mean. Normally, it is positive but may be zero or negative as well. For more information, refer `OpenCV's adaptiveThreshold `_. Returns ------- img : object numpy.ndarray representing the original image. threshold : object numpy.ndarray representing the thresholded image. ) cv2imreadcvtColorCOLOR_BGR2GRAYadaptiveThresholdADAPTIVE_THRESH_GAUSSIAN_C THRESH_BINARYnpinvert) imagenameprocess_background blocksizecimggray thresholds U/var/www/html/answerous/venv/lib/python3.12/site-packages/camelot/image_processing.pyadaptive_thresholdrs8 **Y C <<S// 0D)) #s55s7H7H)UV   >)) IIdO   * *        >cg}|dk(r9|jd|z}tjtjd|f}nK|dk(r9|jd|z}tjtj|df}n | t d|Tt j |j}|D]} | \} } } } d|| | | z| | | zf<t j||}tj|}tj||}tj|||} tj|jt jtjtj\}}}|D]x}tj|\} } } } | | | z}}| | | z}}|dk(r"|j!||zdz|||zdz|fR|dk(sX|j!|||zdz|||zdzfz||fS#t$rVtj|jt jtjtj\}}YwxYw)aiFinds horizontal and vertical lines by applying morphological transformations on an image. Parameters ---------- threshold : object numpy.ndarray representing the thresholded image. regions : list, optional (default: None) List of page regions that may contain tables of the form x1,y1,x2,y2 where (x1, y1) -> left-top and (x2, y2) -> right-bottom in image coordinate space. direction : string, optional (default: 'horizontal') Specifies whether to find vertical or horizontal lines. line_scale : int, optional (default: 15) Factor by which the page dimensions will be divided to get smallest length of lines that should be detected. The larger this value, smaller the detected lines. Making it too large will lead to text being detected as lines. iterations : int, optional (default: 0) Number of times for erosion/dilation is applied. For more information, refer `OpenCV's dilate `_. Returns ------- dmask : object numpy.ndarray representing pixels where vertical/horizontal lines lie. lines : list List of tuples representing vertical/horizontal lines with coordinates relative to a left-top origin in image coordinate space. verticalr horizontalz6Specify direction as either 'vertical' or 'horizontal') iterations)shapergetStructuringElement MORPH_RECT ValueErrorr zerosmultiplyerodedilate findContoursastypeuint8 RETR_EXTERNALCHAIN_APPROX_SIMPLE boundingRectappend)rregions direction line_scalerlinessizeel region_maskregionxywhdmask_contoursrx1x2y1y2s r find_linesr@6sLL EJq!Z/  & &s~~4y A l "q!Z/  & &s~~ay A  QRRhhy/  2FJAq!Q01KAE 1q1u9, - 2KK ;7  )R(I 9b)I JJy" left-top, w -> width and h -> height in image coordinate space. T)keyreverseN ) rr&r'r r(r)r*r!sorted contourArea approxPolyDPr+r,) rrmask__r;contrc_polyr5r6r7r8s r find_contoursrMs$ j D ++ KK !3#4#4c6M6M HbhCOOTB3BGH D "!!!Q-%%f- 1a Q1aL!" K  '' KK !3#4#4c6M6M " sAC AD*)D*ctj||}i}|D]}|\}}}} |||| z|||zf} tj| j tj tj tj\} } } t| dkrg} | D]K}tj|\}}}}|d|z|zdzz|d|z|zdzz}}| j||fM| |||| z||z|f<|S#t$rVtj| j tj tj tj\} } YwxYw)aFinds joints/intersections present inside each table boundary. Parameters ---------- contours : list List of tuples representing table boundaries. Each tuple is of the form (x, y, w, h) where (x, y) -> left-top, w -> width and h -> height in image coordinate space. vertical : object numpy.ndarray representing pixels where vertical lines lie. horizontal : object numpy.ndarray representing pixels where horizontal lines lie. Returns ------- tables : dict Dict with table boundaries as keys and list of intersections in that boundary as their value. Keys are of the form (x1, y1, x2, y2) where (x1, y1) -> lb and (x2, y2) -> rt in image coordinate space. r) r r#rr&r'r( RETR_CCOMPr*r!lenr+r,)r;rrjointstablesrr5r6r7r8roirJjc joint_coordsjjxjyjwjhc1c2s r find_jointsr^so.[[: .F F 4 1aQQYAE )* )) 288$cnnc6M6MJBB r7a<   *A --a0NBB!b&2+!++Q!b&2+!1C-CB   R ) *(41q5!a%#$'4* M %% 288$cnnc6M6MFB sAC??AEE)F)Nrr_r)rnumpyr rr@rMr^rrrcs, ,`PQOd%P.r