o h@sdddlZddlmZmZddZddZGdddeZGd d d eZGd d d eZeegZ dS) N)ImageRGBImageBuffercCs0zddl}W|Styddlm}Y|Sw)Nrcv2)r ImportErrorcvrrU/var/www/html/ndineBlogger/venv/lib/python3.10/site-packages/willow/plugins/opencv.py_cv2s  r cCs ddl}|S)Nrnumpyr rrr _numpysr c@sXeZdZddZeddZejddZejddZ ejd d Z ejd d Z d S)BaseOpenCVImagecCs||_||_dSN)imagesize)selfrrrrr __init__s zBaseOpenCVImage.__init__cCs tdSr)r clsrrr checks zBaseOpenCVImage.checkcCs|jSr)rrrrr get_sizeszBaseOpenCVImage.get_sizecCdS)Nrrrrr get_frame_count!zBaseOpenCVImage.get_frame_countcCrNFrrrrr has_alpha&rzBaseOpenCVImage.has_alphacCrrrrrrr has_animation+rzBaseOpenCVImage.has_animationN) __name__ __module__ __qualname__r classmethodrr operationrrrrrrrr rs    rcs6eZdZefddZeeeddZZ S)OpenCVColorImagecsttdSr)superrr r __class__rr r2s  zOpenCVColorImage.checkcCsRt}t}|j|j|jd}||jd|jdd}|||j}|||jS)z_ Converts a Color Image buffer into a numpy array suitable for use with OpenCV )dtyperr) r r frombufferdatauint8reshapercvtColor COLOR_RGB2BGR)r image_bufferr rrrrr from_buffer_rgb7s  z OpenCVColorImage.from_buffer_rgb) r r!r"r#rrconverter_fromrr2 __classcell__rrr'r r%1s r%c@sXeZdZdZdZdZdZejddZ ejddd Z d d Z e e ed d ZdS)OpenCVGrayscaleImagerr*g?)(r6cCs>t}t}||jddd}|durgS||d}|S)zQ Find interesting features of an image suitable for cropping to. g{Gz?g?N))r r goodFeaturesToTrackrr.tolist)rr rpointsrrr detect_featuresLsz$OpenCVGrayscaleImage.detect_features haarcascade_frontalface_alt2.xmlcCsNt}||}||}||j}|||j|j|j|j }dd|DS)z| Run OpenCV face detection on the image. Returns a list of coordinates representing a box around each face. cSs8g|]}|d|d|d|d|d|dfqS)rrr9r*r).0facerrr lsz5OpenCVGrayscaleImage.detect_faces..) r _find_cascadeCascadeClassifier equalizeHistrdetectMultiScaleface_haar_scaleface_min_neighborsface_haar_flags face_min_size)rcascade_filenamercascadeequalised_imagefacesrrr detect_faces\s   z!OpenCVGrayscaleImage.detect_facescCs0tj|stjtjtjtd|}|S)z{ Find the requested OpenCV cascade file. If a relative path was provided, check local cascades directory. z data/cascades)ospathisabsjoindirname__file__)rrJrrr rBvs z"OpenCVGrayscaleImage._find_cascadecCs"t}||j|j}|||jS)zF Convert OpenCVColorImage to an OpenCVGrayscaleImage. )r r/rCOLOR_BGR2GRAYr)r colour_imagerrrrr from_colors zOpenCVGrayscaleImage.from_colorN)r>)r r!r"rHrGrFrIrr$r=rNrBr#r3r%rWrrrr r5Fs   r5) rO willow.imagerrr r rr%r5willow_image_classesrrrr s G