ը fdZddlmZddlZddlZddlZddlZddlZddlm Z ddl m Z ddl m Z ddlmZ e ddlddlddldadad ad Zd Z dZ! e"e#e$e%e$Z&e&ea'e"e#e$e$Z(e(e a)GddZ*y#e$reZYcwxYw#eeef$r-Zed j5eej6 YdZ[dZ[wwxYw#e$rYWwxYw)a @package vdigit.wxdisplay @brief wxGUI vector digitizer (display driver) Code based on wxVdigit C++ component from GRASS 6.4.0 (gui/wxpython/vdigit). Converted to Python in 2010/12-2011/01. List of classes: - wxdisplay::DisplayDriver (C) 2007-2016 by the GRASS Development Team This program is free software under the GNU General Public License (>=v2). Read the file COPYING that comes with GRASS for details. @author Martin Landa )print_functionN)Debug) UserSettings) DecodeString)Rect)*zwxdigit.py: {})filectrYtjjdk\rt |j }tj |tjzn t|td|zz a y)zRedirect stderr r) logsys version_infomajorrdatawriteoslinesepprint last_error)msgtypes 1/usr/lib/grass83/gui/wxpython/vdigit/wxdisplay.py print_errorr2sV    ! !Q &sxx(C # "# c #)J c>trtj|y y)zRedirect progress infor)progressSetValue)values rprint_progressr!As"%   rc0t}|ddk7r|dz }da|S)N.r )r)rets r GetLastErrorr&Ls% C 2w#~ s J JrceZdZdZdZdZdZd#dZdZdZ dZ d$d Z d Z d Z d$d Zd ZdZd%dZd&dZd'dZdZd&dZd(dZd)dZdZdZdZdZd*dZdZd+dZdZd,dZ d Z!d!Z"d"Z#y)- DisplayDriverc,|a|atdtjdk7r$t j t jdttttd|_ d|_ d|_||_||_||_|j%|_||_t|_d|_d|_t/|_t3|_dt7t7t7d|_id dd ddd d ddd d ddd dddd dddd dddd dddd dddd dddd dddd dddd dddd dddd dddd dd|_|j=d|_d|_ |jCy)aDisplay driver used by vector digitizer :param device: wx.PseudoDC device where to draw vector objects :param deviceTmp: wx.PseudoDC device where to draw temporary vector objects :param mapOng: Map Object (render.Map) :param windiow: parent window for dialogs :param glog: logging device (None to discard messages) :param gprogress: progress bar device (None to discard message) wxvdigitwin32CNFTr#)fieldcatsidsidsDupl highlight highlightDupl)enabledcolorpointline boundaryNo boundaryOne boundaryTwo centroidIn centroidOut centroidDupnodeOnenodeTwovertexarea direction lineWidth)"rr G_gisinitrplatformlocale setlocale LC_NUMERICG_set_error_routineerrfuncG_set_percent_routineperfuncmapInfo poMapInfois3DdcdcTmpmapObjGetCurrentRegionregionwindow firstNode lastNodeIdVect_new_line_structpoPointsVect_new_cats_structpoCatslistselectedsettings_resetTopology _drawSelected _drawSegmentsUpdateSettings)selfdevice deviceTmprQrTglog gprogresss r__init__zDisplayDriver.__init__as* <<7 "   V.. 4G$g&    --/  -. *, F6v     >  6  5  ed;  ut<   ut<  ed;  ut<  ut<  548  548  %$7  5  UT: !  ( "" rctt|jr|jt |j t |jy)zClose currently open vector mapN)G_unset_error_routineG_unset_percent_routinerMCloseMapVect_destroy_line_structrXVect_destroy_cats_structrZrbs r__del__zDisplayDriver.__del__s6! >> MMO / -rc ,ddddddddddddd |_y)zReset topology dictr) r1r5r6r7r8r9r:r;r<r=r>r?N)topologyrns rr^zDisplayDriver._resetTopologys1   rcBt|jd|jd}|jd|jjdz |zz }|jd|jjdz |zz}t ||z |z t ||z |z fS)a!Conversion from geographic coordinates (east, north) to screen (x, y) .. todo:: 3D stuff... :param east: east coordinate :param north: north coordinate :param elev: elevation :return: x, y screen coordinates (integer) ewresnsrescenter_eastingcenter_northing)maxrSrQwidthheightint)rbeastnorthelevmap_reswns r _cell2PixelzDisplayDriver._cell2Pixelsdkk'*DKK,@A KK( )T[[->->-Bg,M M KK) *dkk.@.@1.D-O OD1H'(#q5yG.C*DDDrc|r|sy|j|j|z |j|j|z|j|j|j|j|z |j|j|zy)a(Draw cross symbol of given size to device content Used for points, nodes, vertices :param pdc: PseudoDC where to draw :param point: coordinates of center :param size: size of the cross symbol :return: 0 on success :return: -1 on failure r#r)DrawLinexy)rbpdcr5sizes r _drawCrosszDisplayDriver._drawCrosssc% UWWt^UWWeggneggF UWWeggnegguww~Frc  |jr |jsytjdd|j|j |j d}|j tk(r|jt|j|jrS|j}tj}tj}d}|jdxxdz cc<|js}tA|j D]>}|jB|}|d kDr|jE||dz }|jG||@y|d kDrK|jHr>d|_%|j dzdz |_d}d }||j dz krtjL|jB|jN|jB|jP}tjL|jB|dzjN|jB|dzjP} |jE||j+||jS|dz tU|jN|jPd d |jS|tU|jN|jP| jN|jNz | jP|jPz |jW|jN|jP| jN| jP|dz }|dz }||j dz kr|jS|dz tU|jB|j dz jN|jB|j dz jPd d ytY} tA|j D]J}|jB|}| j[tjL|jN|jPLt]| dkr%|j^jatcdy|j tk(r|je| y|jg| y)zDraw given object to the device The object is defined as robject() from vedit.h. :param robj: object to be rendered :return: 1 on success :return: -1 on failure (vector feature marked as dead, etc.) r#z)_drawObject(): line=%d type=%d npoints=%dNr1r2r3r4rBrr FrvTzPWARNING: Zero-length line or boundary drawing skipped. Use v.clean to remove it.)4rOrPrrfidrnpoints TYPE_AREA _isSelectedVect_get_area_centroidrMwxTRANSPARENT_PENTRANSPARENT_BRUSHrqr_r] _isDuplicatedPenSOLID _definePenSetPenSetBrush TYPE_POINTTYPE_CENTROIDINTYPE_CENTROIDOUTTYPE_CENTROIDDUP TYPE_NODEONE TYPE_NODETWO TYPE_VERTEXrUrVranger5SetIdrr` fisrtNodePointrr SetIdBoundsrrr[appendlenrr_ DrawPolygon DrawLines) rbrobjbrushrpendcIdip point_beg point_endpointss r _drawObjectzDisplayDriver._drawObjects wwdjj 7 HH II LL   99 !d&6&6 "4>>488 <' **C$$C((ED MM+ &! + &%% YY) #(8(8(B**C}}_-i8T=O=O>ffMM/27;MM+.HH ffMM+. k0JBHHD MM+ &! + &%%''C3JCD 3  LL  99             ax99 +DYY,"=>~~ ).#4<<( (JJqM!8IIdOAIDQ'  (axD..!%"&,,"2Q"6$,,** "A$**Q-// JI "AE):)<)O>Q>Q RIIIdOJJsOOOD1Hd9;; QPQ.RSOO%KK%KK%KK)++5%KK)++5 LLikk9;; TFAAID%$,,**&1H 4<r?r@islerA)rAr@rr)r@rr4rBN)r TYPE_LINETYPE_BOUNDARYNOTYPE_BOUNDARYTWOTYPE_BOUNDARYONErrrrrrr TYPE_ISLETYPE_DIRECTIONrqrrBrushr]rrr)rbrtypekeyrrs rrzDisplayDriver._definePenzs J C i C o %C & &C & &C o %C & &C & &C l "C l "C k !C i C i C n $C 3 3 MM# ! #  " "$$Cf}s!3G!r?r@rA)r] DRAW_POINT DRAW_LINEDRAW_BOUNDARYNODRAW_BOUNDARYTWODRAW_BOUNDARYONEDRAW_CENTROIDINDRAW_CENTROIDOUTDRAW_CENTROIDDUP DRAW_NODEONE DRAW_NODETWO DRAW_VERTEX DRAW_AREADRAW_DIRECTIONrbr%s r _getDrawFlagzDisplayDriver._getDrawFlagsq == !) , : C ==  + 9 C == &y 1 ? "C == ' 2 # #C == ' 2 # #C == &y 1 ? "C == ' 2 # #C == ' 2 # #C == #I . < C == #I . < C == "9 - ; C ==  + 9 C == %i 0 > !C rc(||jdvryy)zCheck if vector object selected? :param line: feature id :return: True if vector object is selected :return: False if vector object is not selected r/TFr\)rbr6forces rrzDisplayDriver._isSelecteds 4==' 'rc$||jdvS)zCheck for already marked duplicates :param line: feature id :return: True line already marked as duplicated :return: False not duplicated r0rrbr6s rrzDisplayDriver._isDuplicatedst}}Y///rct}|jd|_|jd|_|jd|_|jd|_t |_t |_|S)zHGet bound_box() from current region :return: bound_box rser) bound_boxrSNSEWPORT_DOUBLE_MAXTB)rbboxs r _getRegionBoxzDisplayDriver._getRegionBoxsa k C  C  C  C    rctjdd||jr|jr |jsyt |jt |j|j|jd|jd|jj|jjt|jd|jdj}|j|jj!|jj!t#|j$D],}|j&|j}|j)|.|jj+|jj+d|j,d<t/|j,d <y ) zDraw content of the vector map to the device :param force: force drawing :type force: bool :return: number of drawn features :return: -1 on error rz!DisplayDriver.DrawMap(): force=%dr#rurwrtrsr-r.N)rrrMrOrPVedit_render_mapbyrefrrrSrQryrzrxcontentsr^ BeginDrawingrnitemsitemr EndDrawingr\r[)rbrrlistrrs rDrawMapzDisplayDriver.DrawMaps^  !8%@~~TWWDJJ NN $$$& '     KK( ) KK) * KK   KK    G$dkk'&: ;   (    !u||$ #A::a=))D   T " #  "$ g $ frcd}dtfdtfdtfdtffD](}t j dd|ddg s!||d z}*|S) z_Get type(s) to be selected Used by SelectLinesByBox() and SelectLineByPoint() rr5r6centroidboundaryvdigit selectTyper3grouprsubkeyr)GV_POINTGV_LINE GV_CENTROID GV_BOUNDARYrGet)rbftypefeatures r_getSelectTypezDisplayDriver._getSelectType%sq  h  W   %  %  $G L'!*i9P# $ rc@|dkDr|t|jkryy)zCheck if feature id is valid :param line: feature id :return: True valid feature id :return: False invalid rTF)Vect_get_num_linesrMrs r _validLinezDisplayDriver._validLine8s" !8 24>> BBrNcD|du}|s |j}|sy|r%||_d|_t|jd<t }|d\}}|d\} } t } t| ||dt| | |dt| | | dt| || dt| ||d|s|j}t|| dd|||j} | j} tjdd| t| D]}| j|}t!j"dd d rd}|j%|syt'||j(d||j(j}t|j*D]-}t-|j.||j0|| r+d }n|s|j3|s|jd j5||jd j7|t9| t;|| S)aSelect vector objects by given bounding box If line id is already in the list of selected lines, then it will be excluded from this list. :param bbox: bounding box definition :param ltype: feature type or None for default :param drawSeg: True to draw segments of line :param poMapInfo: use external Map_info, None for self.poMapInfo :return: number of selected features :return: None on error NTr.rrz)DisplayDriver.SelectLinesByBox() num = %dr selectInsider3rFr/)rMr`r_r[r\ Vect_new_listrWVect_append_pointrVect_select_lines_by_polygonrn_valuesrrrr rrrVect_read_linerXn_pointsVect_point_in_polyrrrrremoverlVect_destroy_list)rbbboxltypedrawSegrM thisMapInfopoListx1y1x2y2poBboxflistnlinesrr6insiderrs rSelectLinesByBoxzDisplayDriver.SelectLinesByBoxEs 4' I !(D !%D %)FDMM& !aBaB%'&"b#.&"b#.&"b#.&"b#.&"b#.'')E$Y4O !@&Iv 2A;;q>DhN9Ut,y$--tD//v/A-fhhqk688A;O!& ##D) e$++D1 e$++D1' 2* !(&! rc |du}|s |j}|sdddS|rd|_t}tdt |dzD]}t ||t |t|d|d||t |dk(s;t||}|j|s|jdj|n|jdj|||dcSdddS)Nr#)r@rTrrr/) rMr_rrVect_get_num_areasVect_get_area_boxrVect_point_in_arearrr\rr )rbr5rMrrr@rs rSelectAreaByPointzDisplayDriver.SelectAreaByPoints4' IB/ / !%D k!/ :Q>? >* 6{{1~;&MM%(//5 6    5>>* 6{{1~;&t/A/A$/GMM%(//5 6 '" "&D #rxx288.LMMrc@t}|D]}t|||S)zGenerate from list struct_ilist)rVect_list_append)rbplistilistvals r _listToIListzDisplayDriver._listToILists) )C UC ( ) rcd|r|j|S|j|jdS)zGet list of selected objects as struct_ilist Returned IList must be freed by Vect_destroy_list(). :return: struct_ilist r/)r1r\)rbr/s rGetSelectedIListzDisplayDriver.GetSelectedIList s2 $$U+ +  u!566rc|r|jdSt}|js|jd|St |jddkDrwt |j |jd|jdd|jj}tdd|jzD]}|j||S)zGet ids of selected objects :param grassId: True for feature id, False for PseudoDC id :return: list of ids of selected vector objects r/rrNrv) r\r[r`rrrrMrXrrr)rbgrassIddc_idsrrs r GetSelectedzDisplayDriver.GetSelecteds ==' '!! MM!  u% & * 4>>4==$ e@TUV@W X]]++F1a&//12 ! a  ! rc|rd|_nd|_||jd<|dkDr||jd<t|jd<t|j}t d|dzD]}t |j|st|jd|j|}|ttzzsMd}|jj}t d|jD],}|jdD]} |j|| k(sd},.|s|jdj|y||jd<g|jd<y) zSet selected vector objects :param list: of ids (None to unselect features) :param layer: layer number for features selected based on category number TFr-rr.r/rN)r_r\r[rrMrVect_line_aliverrZ GV_POINTSGV_LINESrn_catscatr) rbr/layerrr6r r*r.rr=s r SetSelectedzDisplayDriver.SetSelected-sE !%D !&D !& g 19$'DMM& !#'6DMM% (7Fa!, 6&t~~t<&t~~tT[[$OX!56{{++q$++."A#}}V4"88A;#-$(E!"" MM%(//5! 6$$'DMM% $&DMM& !rc &t}t|jddk7s |js|Sd}|jdd}|j |syt |j |j|j|}d}d}d}|jj} t| jD]} t|d|dd| j| | j| | j| d} | dk(r| }| }n || kDr| }| }|j!| j| | j| | j| \} } t#| | dd}|j$j'|||dz }||j)kDr|S|dzdz}|j+|||k(r|j+dn|j+|dz || jdz dz|zk(r|j+d|S|j+|dz|S)zGet PseudoDC vertex id of selected line Set bounding box for vertices of line. :param pos: position :return: id of center, left and right vertex :return: 0 no line found :return: -1 on error r/rrr#rrv)r[rr\r`rrrMrXrZrrrVect_points_distancerrzrrrOrr!r)rbposreturnIdstartIdr6rminDistGidDCidridxdistvxvyrects rGetSelectedVertexzDisplayDriver.GetSelectedVertexTs6 t}}U# $ )1C1CO}}U#A&t$t~~t}}dkk4P'') C'AAVXXc]FHHSM688C=RSDaxT>"GC%%fhhsmVXXc]FHHSMRFBB1%D GG  d + AID! $ T&&( (OQw{  7? OOB  OOD1H % FOOa'1,w6 6 OOB  OOD1H %rc"t}t}d}t|j}|jdD]}t |j|}|dkDr&||kr!t |j|t |s"Dt|j|t |se|r!tt |t |d}tt |t ||j|j|j|jfS)zQGet minimal region extent of selected features :return: n,s,w,e Tr/rF)rrrMr\Vect_get_centroid_arearrVect_get_line_box Vect_box_copyVect_box_extendrrrr)rb regionBoxlineBox setRegionnareasr6r@s rGetRegionSelectedzDisplayDriver.GetRegionSelecteds K + #DNN3MM%( BD)$..$?DaxDFN(uW~N(uW~NeI.g?! i 0%.A B {{IKKikkAArc$t||_y)zmDraw selected features :param flag: True to draw selected features :type flag: bool N)boolr_)rbflags r DrawSelectedzDisplayDriver.DrawSelecteds "$Zrc|jsy|jjjtk(r/t |jt t |jt|j}|`dx|_|_|S)z[Close vector map :return: 0 on success :return: non-zero on error rN) rMrmode GV_MODE_RWVect_build_partial GV_BUILD_NONE Vect_build Vect_closerLrs rrkzDisplayDriver.CloseMapsh ~~ >> " " ' ': 5 t~~} = t~~ &( L(,, rc>tjdd||||js)t|_t |j|_|r|rt }nt}n|rt}nt}||j ||}|dk(r|`dx|_|_n|dkrtj|jtd|ztdtjtjztj ztj"z}|j%}|tj&k7r|`dx|_|_nt)|j |rt+|j d t-|j |_|j S) aOpen vector map by the driver :param name: name of vector map to be open :type name: str :param mapset: name of mapset where the vector map lives :tryp mapset: str :param update: True to open vector map in update mode :type update: bool :param tmp: True to open temporary vector map :type tp: bool :return: map_info :return: None on error rz5DisplayDriver.OpenMap(): name=%s mapset=%s updated=%dr#NrvzTopology for vector map <%s> is not available. Topology is required by digitizer. Do you want to rebuild topology (takes some time) and open the vector map for editing?zTopology missing)parentmessagecaptionstyleT)rrrLMap_infopointerrMVect_open_tmp_updateVect_open_updateVect_open_tmp_old Vect_open_oldr MessageDialogrTrYES_NO YES_DEFAULT ICON_QUESTIONCENTRE ShowModalID_YESrbVect_set_updated Vect_is_3drN)rbnamemapsetupdatetmpopen_fnr%dlgs rOpenMapzDisplayDriver.OpenMapsS  C     ||#:DL$T\\2DN .*+'dnndF3 "9 ,0 0DNT\ 1W""{{#  ,-ii"..023C3CCbiiO C--/CbiiL0444>>*  T^^T 2t~~. ~~rc|jsyt}t|jt||j|j |j |j|j|jfS)zVGet bounding box of (opened) vector map layer :return: (w,s,b,e,n,t) N) rMrVect_get_map_boxrrrrrrr)rbr s rGetMapBoundingBoxzDisplayDriver.GetMapBoundingBoxsR ~~{t5vvtvvtvvtvvtvvtvv==rc t}|jjD]1}|dk(r/tt j ddd|j|<8t jt j dd|dgdt j dd|dgdt j dd|dgd |}|d k(r||j|<|d k(r2tt j dd d |j|d <n3tt j dd|d g|j|d <||j|d<4y)zUpdate display driver settings .. todo:: map units :param alpha: color value for alpha channel rBrr rsymbolr4rrrvr1r2 checkForDuplr3N) dictr]keysr{rrrColourrZ)rbalphar4rs rrazDisplayDriver.UpdateSettings%sd==%%'% 0Ck!%( $$8WU& c"II  xXsGnU  xXsGnU  xXsGnU Ek!%* c"o%04 $$&N91 c"9- 15 $$&Hc9=M1 c"9- +0DMM# w 'K% 0rcB|jj|_y)z1Update geographical region used by display driverN)rQrRrSrns r UpdateRegionzDisplayDriver.UpdateRegionUskk224 rcb|tjd|d}|;tjd|d}|!tjd|d}|dk(rdnd}|dkr"|jd |jd zd z }|dk(r+t|jd |jd }||zS|S) zReturn threshold value in map units :param type: snapping mode (node, vertex) :param value: threshold to be set up :param units: units (0 for screen pixels, 1 for map units) :return: threshold value rr runitunitsz screen pixelsrrrtrsg@)rrrSrx)rbrr rress rr!zDisplayDriver.GetThresholdYs = $$8gNE = $$8fME}$((xT'R"o51 19[[)DKK,@@CGE A:dkk'*DKK,@AC3;  rcV|jsyt}t}t}t|jd<t t |jdD]+}|jd|}|j|r(t|j|d||jdD]}||k(s|j|rt|j|d|t||tsD||vrPt||<||j||j|f|jdj|||j||j|f|jdj|.t|t||S)z3Return ids of (selected) duplicated vector featuresNr0r/)rMrrWr[r\rrrrVect_line_check_duplicate WITHOUT_Zr _getCatStringrl)rbr/APointsBPointsrline1line2s rr$zDisplayDriver.GetDuplicatesvso~~ f&(&(#'6 i s4==/01 ;AMM%(+E!!%( 4>>7D% @u- ;E>T%7%7%>t~~weD,WgyI|!%AA ud.@.@.G&HI i077>FMM5$*<*catsStrlcs rrzDisplayDriver._getCatStringst~~tT[[$?{{##6t{{# 0AJJqMEH$"&& UO " "488A; /  0 MM(+ >DAq AsxxC '<#==G >rcd}|D]X}|j|r|jdj||jddsE|j |sWd}Z|r|j t |jdS)zNUnselect vector features :param lines: list of feature id(s) Fr/r2r3T)rr\r r]rr$r)rblinesrr6s rUnSelectzDisplayDriver.UnSelects   $D% e$++D1}}_-i8T=O=OPT=U#  $     4=='((r))F)NFN)N)NN)T)r#)TF))snappingNN)$__name__ __module__ __qualname__rgror^rrrrrrrrrrrrrr+r1r3r7r?rNrXr\rkr~rrarr!r$rrrrr(r(`sIV . "E&(I*V3j#J 0 ('T& EN,2ZNx 70%'NAFB8(*FP >.0`5:$L")rr()+__doc__ __future__rrErrrr core.debugr core.settingsr core.gcmdr gui_core.wrapr WindowsError NameErrorOSError grass.lib.gisgrass.lib.vectorgrass.lib.vedit ImportError TypeErrorrrformatstderrrrrrr!r& CFUNCTYPE UNCHECKEDc_intStringerrtyperIpertyperKr(rrrrs && &"7"!         %(&%8Gk"G %(%0Gn%G ])])yL \9-7  ! !! $3::667d  s:BB/CBBC'#CCCC