a #Dfu@sdddlmZddlZddlZddlZddlZddlm Z d ddZ dddZ dd d Z dd d Z dS)N)utilscCs<|dur(tttjd}t|d}tjj||d}|S)Nzmpl-data/fonts/ttfzDejaVuSansMono.ttf)fontsize)ospjoindirname matplotlib__file__PILZ ImageFontZtruetype)r font_pathZ fonts_pathrr ]/nfs/NAS7/SABIOD/METHODE/ermites/ermites_venv/lib/python3.9/site-packages/imgviz/draw/text.py _get_font s  rc Csxt||d}d}d}|D]R}|dkr,d}t|drN||dd\}}n||\}}t||}||7}q||fS)zGet text size (height and width). Parameters ---------- text: str Text. size: int Pixel font size. Returns ------- height: int Text height. width: int Text width. )r r getbboxrN)r splitlineshasattrrgetsizemax) textrr r text_widthZ text_heightline line_widthZ line_heightr r r text_sizes     rrrrcCs(t|}t||||||dt|S)aDraw text on numpy array with Pillow. Parameters ---------- src: numpy.ndarray Input image. yx: (2,) array-like Left top point of the text. text: str Text to draw. size: int Text size in pixel. color: (3,) array-like Text RGB color in uint8. Default is (0, 0, 0), which is black. font_path: str Default font is DejaVuSansMono in matplotlib. Returns ------- dst: numpy.ndarray Output image. )imgyxrrcolorr )rZnumpy_to_pillowtext_Zpillow_to_numpy)srcrrrrr dstr r rr7s rc CsBtj|}|\}}t|}t||d} |j||f||| ddS)N)rr )Zxyrfillr)r Z ImageDrawtuplerr) rrrrrr Zdrawy1x1rr r rr Us   r )N)N)rN)rN)os.pathpathrr Z PIL.Imager Z PIL.ImageDrawZ PIL.ImageFontrrrrrr r r r rs    $