U ©È•dPã@sJddlZddlmZddd„Zddd„Zddd „Zdd d „Zddd„ZdS)éN©rréÿéc Csì|j\}}}||dks<|||ks<||dks<|||kr@dSt||ƒt||ƒf} t||ƒt||ƒf} t||ƒt||ƒf} t||ƒt||ƒf} t || | ||¡t || | ||¡t || | ||¡t || | ||¡dS©ú{ Draws a square on the image :param image: :param x: :param y: :param color: :param thickness: rN©ÚshapeÚintÚcv2Úline) ÚimageÚxÚyÚsizeÚcolorÚ thicknessÚhÚwÚ_ÚtlÚtrÚbrÚbl©rú5/home/daniela/catkin_ws/src/hpe/scripts/utils/draw.pyÚ drawSquare2Ds 0rc Cs¸|j\}}}||dks<|||ks<||dks<|||kr@dSt||ƒt|ƒf} t||ƒt|ƒf} t|ƒt||ƒf} t|ƒt||ƒf} t || | ||¡t || | ||¡dSrr) r r rrrrrrrÚleftÚrightÚtopÚbottomrrrÚ drawCross2D!s 0r c CsÈ|j\}}}||dks<|||ks<||dks<|||kr@dSt||ƒt||ƒf} t||ƒt||ƒf} t||ƒt||ƒf} t||ƒt||ƒf} t || | ||¡t || | ||¡dSrr) r r rrrrrrrZtop_leftZ top_rightZ bottom_leftZ bottom_rightrrrÚdrawDiagonalCross2D:s 0r!cCs||f}t |||||¡dS)rN)r Úcircle)r r rrrrZcenter_coordinatesrrrÚ drawCircleSsr#çš™™™™™¹?Fc CsT|dkrx|j|||g||g||gdd}|j||g|||g||gdd}|j||g||g|||gdd} ||| gS|dd |||g¡|dd ||g¡|ddj||gd|dd ||g¡|dd |||g¡|ddj||gd|d d ||g¡|d d ||g¡|d dj|||gddS) NFÚr)rÚgÚbr)Úzsré)ÚplotÚ set_xdataÚ set_ydataÚset_3d_properties) ÚaxÚhandlesÚxcÚycÚzcrÚupdateZhandle_xZhandle_yZhandle_zrrrÚdraw3Dcoordinatesystemrs """ r4)rr)rr)rr)rr)rrrr$F) r Úmatplotlib.pyplotÚpyplotÚpltrr r!r#r4rrrrÚs