ը fHbdZddlZddlZddlZddlZddlmZddlmZ ddlm Z ddl m Z mZddlmZddlmZmZdd lmZdd lmZdd lmZe\ZZGd d ej8Zy)aX @package rdigit.controller @brief rdigit controller for drawing and rasterizing Classes: - controller::RDigitController (C) 2014 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 Anna Petrasova N)NewEvent)coreraster)CalledModuleError ScriptError)Signal)GErrorGMessage) UserSettings)gThread)NewRasterDialogceZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZddZdZdZdZ ddZdZdZdZdZdZdZdZy)RDigitControllerzsController object for raster digitizer. Inherits from EvtHandler to be able to send wx events from thraed. cztjj|||_||_t |_d|_d|_d|_ d|_ d|_ d|_ d|_ d|_g|_d|_d|_tj$|_d|_d|_d|_d|_t0j2j5|_d|_d|_t=d|_t=d|_ t=d|_!t=d|_"t=d |_#y) zConstructs controller :param giface: grass interface object :param mapWindow: instance of BufferedMapWindow NFdareaz!RDigitController:newRasterCreatedz"RDigitController:newFeatureCreatedz$RDigitController:uploadMapCategorieszRDigitController:quitDigitizerz!RDigitController:showNotification)$wx EvtHandler__init___giface _mapWindowr _thread _editedRaster_backgroundRaster_backupRasterName_editOldRaster_mapType_areas_lines_points_all_drawing_runningGREEN _drawColor_drawTransparency _graphicsType_currentCellValue_currentWidthValueosenvironcopy_env _oldMouseUse _oldCursorr newRasterCreatednewFeatureCreateduploadMapCategories quitDigitizershowNotification)selfgiface mapWindows 2/usr/lib/grass83/gui/wxpython/rdigit/controller.pyrzRDigitController.__init__(s t$ #y !!%!%$       ((!$#!%"&JJOO%  !''J K!'(L!M$**P#Q #$DE &'J Kcp|jjj|j|jjj|j |jj j|j|jjtjyN) r mouseLeftDownconnect_start mouseLeftUp _addPoint mouseRightUp_finishUnbindrEVT_CONTEXT_MENUr6s r9 _connectAllzRDigitController._connectAllcsn %%--dkk: ##++DNN; $$,,T\\: r223r:c|jjj|j|jjj|j |jj j|j|jjtj|jjyr<) rr= disconnectr?r@rArBrCBindrrE OnContextMenurFs r9_disconnectAllzRDigitController._disconnectAllisz %%00= ##..t~~> $$// = R00$//2O2OPr:c|jry|js!t|jt dy|j s|j dk(rJ|jjg}|jdd|jj|n|j dk(rJ|jjg}|jdd|jj|nX|j d k(rI|jjg}|jdd|jj|d |_yy) zStart digitizing a new object. :param x: x coordinate in map units :param y: y coordinate in map units Nz"Please select first the raster mapparentmessager)coordspenNamepen1linepointT)r$rr r_r#r(rAddItemSetPropertyValr"appendr r!)r6xyitems r9r?zRDigitController._startos == !! 2V0W  }}!!V+{{**"*5##Iv6   &##v-{{**"*5##Iv6   &##w.||++2+6##Iv6   & DMr:c|jry|jsy|jdk(rh|jj d}|j ||ggz}|j ||jjtdn|jdk(rh|jj d}|j ||ggz}|j ||jjtdnM|jdk(r>|jj d}|j ||g|j|jj|jj|jj|jj|jj!y) zxAdd point to an object. :param x: x coordinate in map units :param y: y coordinate in map units NrzRight click to finish area)textrTzRight click to finish linerU)r$r#r(rGetItem GetCoords SetCoordsr5emitrVr r!rCr ClearLinesDrawRefresh)r6rZr[rrQrTrUs r9rAzRDigitController._addPointsg == }}     ';;&&r*D^^%!Q0F NN6 "  ! ! & &A.J,K & L   6 );;&&r*D^^%!Q0F NN6 "  ! ! & &A.J,K & L   7 *LL((,E OOQF # LLN ""$    !r:c@|jry|jdk(r|jjd}nW|jdk(r|jjd}n,|jdk(r|j jd}nyd|_|jdd|jd |jd |jd |j|jd |j|jj|jj|jj|jj|j j|jj!y) zFinish digitizing a new object and redraws. Saves current cell value and buffer width for that object. :param x: x coordinate in map units :param y: y coordinate in map units NrUr^rrTF brushNamedone cellValue widthValue)r$r(r!r`rr r#rX AddPropertyr)r*r2rcrrdrerf)r6r\s r9rCzRDigitController._finishs8 ==     (<<''+D   6 );;&&r*D   6 );;&&r*D   K0 % & K)?)?@ L$*A*AB ##% ""$    !r:c`|jr-|r+|j|k7r|jr|j|jr|s|jj |jj |jj d|jj d<|j|jj|j|j|jj d<||_y|j|r|jd|jj d<d|jj d<tjdd tj |j_|jjd ||_y) zSelects method (area/line/point) for drawing. Connects and disconnects signal to allow other tools in map toolbar to work. pdcbeginenduseNrTboxredcolourwidthstylepencil)r(r#rCrrdpdcTmpmouserLSetNamedCursorr0r/rGrPen SHORT_DASHpen)r6 drawingTypes r9 SelectTypezRDigitController.SelectTypes@   ""k1  LLN   k OO & &4??+A+A & B+/??+@+@+IDOO ! !% (    ! OO * *4?? ;+/+<+..s)<)<)>r:callableondoneN)rr MessageDialogrrVYES_NO ShowModalID_YESr#rCrRun _exportRasterr4rc)r6dlgs` r9StopzRDigitController.Stop s   ""01+, C }}"))+==LLN   !//>! ""'')    # # %r:cjrjjjjfdy)z#Saves current edits to a raster mapc$jSr<)_updaters r9rz'RDigitController.Save..;s dllnr:rN)r#rCrrrrFs`r9SavezRDigitController.Save5s6 == LLN ''0L  r:cNt|jr|jjd}|jj ||j j ||j j |d|_|jjdyy)z:Undo a change, goes object back (finished or not finished)r^FrN) lenr"popr DeleteItemr r!r#rr)r6removeds r9UndozRDigitController.Undo>sx tyy>iimmB'G KK " "7 + KK " "7 + LL # #G ,!DM OO % %U % 3 r:c* |jr$tjddd|jd|jj |jj |jjd|jjd<|jr|j|jj|j|jj|j|jj|j|jjd |rI|jj|j |j"|jjd <y y #t$rY_wxYw) zlCleans up drawing, temporary maps. :param restore: if restore previous cursor, mouse['use'] g.removerfTtypeflagsnamequietrnrprqFrrrN)rgcore run_commandrrrdr{r|r(rLUnregisterGraphicsToDrawrr r!rr}r0r/)r6restores r9CleanUpzRDigitController.CleanUpJs1 %%!!!//  ""t'='=">'+'<' !!!/  OO * *4?? ;+/+<+After selecting old raster, creates a backup copy for editing.z2Failed to create backup copy of edited raster map.rNF)mapdatatypeT) _backupRasterrr rrVrgrast raster_inforr)r6rs r9 SelectOldMapzRDigitController.SelectOldMapwso    t $"))d3J? "  NO   sA)A,+A,Nc|r |j|||yt |j}|j |jtjk(rQ |j|j|j|j |jy|jy#t$r#t|jt dYywxYw#t$r3t|jt dY|jywxYw#|jwxYw)aAfter selecting new raster, shows dialog to choose name, background map and type of the new map. :params standalone, mapName, bgMap, mapType: if digitizer is launched as standalone module :param bool standalone: if digitizer is launched as standalone module :param str mapName: edited raster map name :param str bgMap: background raster map name :param str mapType: raster map type CELL, FCELL, DCELL )mapName backgroundMapmapTypez Failed to create new raster map.rNFT)rO) _createNewMaprr rrVrCenterOnParentrrID_OK GetMapNameGetBackgroundMapName GetMapTypeDestroy)r6 standalonerbgMaprrs r9 SelectNewMapzRDigitController.SelectNewMaps&  ""#"'##!9C    }}"((* "&& # 0&)&>&>&@ # 0'KKM 9 ??@A  "#!# !"D E!KKM!KKMs5C!>C1)C.-C.1)D-D0,D--D00Ec ^|jdd}|jdd}dddd}|r|}nd} tjdj|||| d d |r||_t j d ||jd |dk(rTt jdd|d j}|r+|jj|jd|j||dzt jdz}||_||_|j"j|t j} |j$j&j| d| d| dd|jdddy#t$rtwxYw)z@Creates a new raster map based on specified background and type.@rintfloatdouble)CELLFCELLDCELLznull()z{name} = {mtype}({back}))rmtypebackT)exp overwriterr.colorsrrrrz r.describe1n)rrr )valuesMAPSET)rGISDBASE LOCATION_NAMEnewr)grassdblocationmapsetactionrelementN)splitrmapcalcformatrrr read_commandstripr3rcrrrgisenvrrr1rgrassdbChanged) r6rrrr backgroundtypesrrrs r9rzRDigitController._createNewMaps}}S!!$"((-a0 8D DD  MM.55U7^$6  )6&!!D1G1Gtf$"//$Dd$eg0055V\\$=O5P 4 czELLN844!  """- ##((:&O,(# 3" ) !   s B2FF,c|jdd}|dzttjz} t j d||gd||_y#t $rtwxYw)zCreates a temporary backup raster necessary for undo behavior. :param str name: name of raster map for which we create backup rr _backupcopy_g.copyT)rrN) rstrr+getpidrrrrr)r6rbackups r9rzRDigitController._backupRastersl zz#q!&RYY[)99    hf~T J"(!   s AA+c|j|jr |jryd|_t|jdkr_|j}d|jvr|jj dd}t jd|j|gddndttjz}g}g}d}dx}}tt|jdtd  }tj|||jdj!d }|jdj!d }|jD];} | j!d r|| j!d k7s|| j!d k7r|r1|j#|||j$|} |j'| g}|j)| ||j#|| j!d |j$|} |j'| g}n|j)| || j!d }| j!d }|dz }tt|j|td  }tj||>|r>|j#| j!d |j$|} |j'| t jd |ddd|jgz|jdd|j*t jddd||gzd |j,ry|j.s@t1j2ddd} | sd} t jd| |jdyt jd|j|j.dy#t4$r*d|_t7|j8tdYywxYw)aURasterizes digitized features. Uses r.in.poly and r.grow for buffering features. Creates separate raster maps depending on common cell values and buffering width necessary to keep the order of editing. These rasters are then patched together. Sets default color table for the newly digitized raster. NTrrr)rrrtmp_rdigit_rast_zRasterizing...)rangerr_rjrkzr.patchr^)inputoutputrrenvrrrr rasterLayer colorTable selection)groupkeysubkeyrainbowr)rrrrFz7Failed to set default color table for edited raster maprN) _setRegionrr$rr"rrrrrr+rupdateProgressrVr PostEventGetPropertyVal _rasterizerrY _writeItemr.rrr Getrr r) r6r tempRasterr_rastersToPatchi lastCellValuelastWidthValueevtr\outtables r9rzRDigitController._exportRasters !!T]]  tyy>A $$Cd(((((..s3A6   ..4  ,c"))+.>>JDNA-1 1MN $))nAA6F4GC LLs # IIaL77 DM!YYq\88FN  (&&|4!T%8%8%EE%)<)<\)JJ"oo .$--'--c2!OOD$///++L9 " C #))#.DOOD$/ $ 3 3K @ !%!4!4\!BQ$dii.:J8K T3'= (>oo$--l;T]]J%%c*   $TrT*d.D.D-EE))II     #zl2   ""))$(('\+%E!!e1C1C4!!**11  ! !DM ST  s; OA O-O0O54O5c >|j}|dk(r|g}|jd}dj|}|D]2}|dj|Dcgc] }t |c}z }|dz }4|dj|z }|j |y cc}w) z.Writes digitized features in r.in.poly format.Prjz{vtype} )vtype rz= {cellValue} )rjN)rarrjoinrrY) r6r\r'r_rQrjrecordcoordcs r9 _writeFeaturezRDigitController._writeFeaturems! C<XF'' 4 ##%#0 E chh61A67 7F dNF  #**Y*?? F 7sB c*||jjvr|j|d|y||jjvr|j|d|y||jjvr|j|d|yy)NA)r'r_Lr&)r GetAllItemsr-r r!)r6r\r_s r9rzRDigitController._writeItem{s 4;;**, ,   t3T  : T[[,,. .   t3T  : T\\--/ /   t3T  :0r:c dttjddz}tjdd}|j dj ||j|rQ|dz}tjd |j||d d tjd ||d |d |jn/tjd |j||d |jtj|j|S)aPerforms the actual rasterization using r.in.poly and buffering with r.grow if required. :param str text: string in r.in.poly format :param float bufferDist: buffer distance in map units :param str mapType: CELL, FCELL, DCELL :param str tempRaster: name of temporary raster used in computation :return: output raster map name as a result of digitization rZNwF)modedeleterg@z r.in.polyT)r r type_rrzr.growm)r r rradiusrr)r r r7rr)ruuiduuid4tempfileNamedTemporaryFilewriter)closerrrr.r+unlink)r6r_ bufferDistrrr  asciiFiles r9rzRDigitController._rasterizess4::<(!,,//SG  $(  # J   nn!      !II    nnII   )..! r:c^tj|j|jd<y)z%Set region according input raster mapr GRASS_REGIONN)r region_envrr.rFs r9rzRDigitController._setRegions!$)$4$4DrTsT  $&(;*&& *&.j##Q Tr}}Q Tr: