ը fHdZddlZddlZddlZddlZddlZddlZddlmcm Z ddl m Z ddl mZmZddlmZmZddlmZddlmZmZddlmZmZmZmZGdd eZGd d eZy) aZ @package wxplot.profile @brief Profiling using PyPlot Classes: - profile::ProfileFrame - profile::ProfileToolbar (C) 2011-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 Michael Barton, Arizona State University N) BasePlotFrame PlotIcons) BaseToolbar BaseIcons) StockCursor)ProfileRasterDialogPlotStatsFrame) RunCommandGWarningGErrorGMessageceZdZdZej dddfdZdZdZdZ d Z d Z d Z d Z d ZdZdZdZdZy) ProfileFramezNMainframe for displaying profile of one or more raster maps. Uses wx.lib.plot.iiNc tj|f|||d|||_|jjj |j g|_t||_tjdk7r|j|j|jtdd|vr|nd|_|g|_n||_d|_d|_g|_d|_d |_td |_gd |_|j/t1|jd kDr,|j3|j|j |_ni|_|j-|jdk7rtd |jz|_ntd|_td|_|j;t<j>|j@|jCy)N)parentgifacesizerdarwinzGRASS Profile Analysis Tooldegreemetersprofileg Profile of)blueredgreenyellowmagentacyanaquablackgreyorangebrownpurplevioletindigorz Distance (%s)zDistance along transectz Cell values)"r__init__ controllertransectChangedconnect SetTransecttransectProfileToolbartoolbarsysplatform SetToolBarSetTitle__units rasterListplottypecoordstrseglistppointstransect_lengthptitle colorList _initOptslenInitRasterOptsrasterxlabelylabelBindwx EVT_CLOSE OnCloseWindow SetGraphStyle)selfrrr*unitsrr7kwargss //usr/lib/grass83/gui/wxpython/wxplot/profile.pyr)zProfileFrame.__init__&s tWF6WPVW$ ''//0@0@A %T2 <<8 # OODLL ) a567'u4e(   DO(DO!    " o  "   1 $--doot}}MDKDK ;; "t{{b'8O,t{{:DK56DK &  ",, 2 23 c&|jdy)zInitialize plot optionsrN) InitPlotOptsrJs rMr?zProfileFrame._initOptsrs )$rNc\||_|r|jdy|jdyN)r.OnCreateProfileOnErase)rJcoordss rMr-zProfileFrame.SetTransectvs&    & LL rNc|jjr|jj|jj|jj |jj y)z(Draws transect to profile in map displayN)r*IsActiveStopStartrSetFocusRaiserJevents rMOnDrawTransectzProfileFrame.OnDrawTransect}sR ?? # # % OO "   rNct|}|j|jtjk(r~|j |_|j |j |j|_t|jdkDr*t|j dkDr|jd|jy)zSelect raster map(s) to profilerrN)r^) rCenterOnParent ShowModalrFID_OKr7rAr8rBr@r.rTDestroy)rJr^dlgs rMOnSelectRasterzProfileFrame.OnSelectRasters!.  ==?bhh &!nnDO--doot}}MDK4==!A%#doo*>*B$$4$0 rNc d}d}d|_dx}}tj}d}t|jdkDr|jD]p}|d|dcxkr|dkrnn|d|dcxkr|d ksnd }|jdk(rd |d|dfz|_Td |j|d|dfz|_r|st t d |t|jdk(ryt d|_t|jdkDrg|_ |jD]}td|d|jdd |d|dfz}|jdjdd} | | dk(rXt| } |rN|rLtjtj ||dz dtj ||dz dz}||z }||_|jj%|| f|d}|d} |jj'dd|_d} t+j,|j.D]} g|j.| d<|j1| |j} t| dkDs@| |j.| d<|j.| ddk7r*|xj(d|j.| d| fzz c_| dz } |xjd| jddzz c_|jj3d|_|j(dk(rt d|_y|j(j3d|_y#YPxYw)zjCreate coordinate string for profiling. Create segment list for transect segment markers. rrNTwesnFz%f,%fz%s,%f,%fz:Not all points of profile lie inside computational region.)messagerrzr.what)rreadmap coordinates|*datalistrKz%s (%d),z %s ,@,z Raster values)r9grassregionr@r.r r5r7r=r:r splitlinessplitfloatmathsqrtpowr<appendpoprDsixiterkeysrBCreateDatalistrstrip) rJdistcumdistlasteast lastnorthry insideRegionpointretvalirrus rM SetupProfilezProfileFrame.SetupProfilesm  ##9 t}}  ! U3K58:vc{:s uQx>6#;>#(L==B&$+uQxq.B$BDM$.$--q5QR81T$TDM U VW  t 1 $  o  t}}  !DL % * '58U1X*> > nn&q)//4Q7;#*Cj 99(U1X"5:((Ia$81=>D4(/$ ##WcN3 8!!H = %B    # dkk* 9A)+DKKN: &**1dmm'*b0KK:Q1H!0L#LLKQ wa88  9kk((- ;;" O,DK++,,S1DK9 s M99M>c g}tj}tt|dt|d}d}|j|z dkDr|jdz }n|}t d||||ddd}|sgS|j D]f} | jjd \} } | | d k(s| dk(s | | d k(s| dk(r>t| } t| } |j| | fh|S) ztBuild a list of distance, value pairs for points along transect Uses r.profile to obtain the data. nsresewresriz r.profilenanT)rinputrp resolutionnullquietrn r) rxryminr|r<r rzstripr{r) rJrBrVrurycurr_res transect_rec transect_resrlinerelevs rMrzProfileFrame.CreateDatalists uVG_-uVG_/EF   ( *S 0//#5L#L #  INN$ *D++C0JD$ 2:5=<2:5=;D;D OOT4L ) *rNct|jdk(st|jdk(r|tj|t dt dtj tjztjz}|j|jy|jttj|j|j}|j!|y)aLMain routine for creating a profile. Uses r.profile to create a list of distance,cell value pairs. This is passed to plot to create a line graph of the profile. If the profile transect is in multiple segments, these are drawn as points. Profile transect is drawn, using methods in mapdisp.py rz>You must draw a transect to profile in the map display window.zNothing to profile)rrmcaptionstyleN)r@r.r7rF MessageDialogr5OKICON_INFORMATIONCENTRErbrd SetCursorr CURSOR_ARROWrCreatePlotListDrawPlot)rJr^reps rMrTzProfileFrame.OnCreateProfile-s t}}  "c$//&:a&?""T./eeb111BII= C MMO KKM  {2??34     ! arNc ng|_t|jdkDrtj|jd|j ddzt j|j ddd|j ddd|j dddd|j dd |j|j dd |j dd  |_ |jj|j|jD]}t j|j|d d|j|d d|j|d dd}tj|j|d||j|d|j|j|d|j|d|j|d<|jj|j|dt|jdkDr |jSy)zjCreate a plot data list from transect datalist and transect segment endpoint coordinates. rrmarkerlegendcolorrkrtrfilltype)rcolourr fillstylerpcolorrupwidthpstyleplegend)rwidthrrplineN)plotlistr@r:plot PolyMarker propertiesrFColour ptfilldictr;rr7rBPolyLine linestyledict)rJrcols rMrzProfileFrame.CreatePlotListHs   t|| q ?? T__X6x@@yyOOH-g6q9OOH-g6q9OOH-g6q9  __X.v6//$//(*CF*KLx08 DL MM  . :A)) Ax(+ Ax(+ Ax(+ C '+mm Az*kk!nX.((Q)AB{{1~i0 'DKKN7 # MM Q!8 9 :" t}}  !== rNcf|j|j}|j|y)z%Update profile after changing optionsN)rIrr)rJrs rMUpdatezProfileFrame.Updatexs(     ! arNc tj|tdtjtdtj }g}|j tjk(r|j}|jD]`}|j|dzt|jddzdztjj|drtj|td|dztd tj tj"ztj$z }|j tj&k7r!|j)|j+ t-|dd }|j2|dD]0} |j5dt7| dt7| dfz2|j9c|j+|r)tdt;|dj=|fz} n td} t?|| y#t.$r9}t1|td |d|fz |j+Yd}~yd}~wwxYw)z!Save r.profile data to a csv filez9Choose prefix for file(s) where to save profile values...z#Comma separated value (*.csv)|*.csv)rrm defaultDirwildcardrr5rvz.csvz=File <%s> already exists. Do you want to overwrite this file?zOverwrite file?)rmrrrhz0Unable to open file <%s> for writing. Reason: %s)rrmNruz %.6f,%.6f rrkz%d files created: %s zNo files generated.) rF FileDialogr5osgetcwdFD_SAVErbrcGetPathr7rstrreplacepathexistsrYES_NO YES_DEFAULT ICON_QUESTIONID_YESrrdopenIOErrorr rBwriter|closer@joinr ) rJr^repfilerrdlgOvfdridatapairrms rMSaveProfileToFilezProfileFrame.SaveProfileToFile~s!mmQRyy{<=**   ==?bhh &;;=D__"  TCZ#aiiS.A*BBVKL77>>%),,, !B! ) !$ !""3 4 ii"..82;K;KK E(BII5    eBi-B!% Az :WHHH]eHQK.@%QR BT-UUVW E" H /0CJ %@P3QQG-.Gg.- # !R!!9a.!)KKM s I J.JJcFg}td}tj|jD]} |j dd}d|z}d|j|dD}t j |t j}|d|jzz }|dt j|zz }|d t j|zz }|d t j|zz }|d t j|zz }|d t j|zz }|d t j|zz }t j|t j|z } |d| zz }|dt j|zz }|dt j |zz }|d|j"zz }|j%|t'|t(j*||} | j-t(j.k(r| j1yy#YxYw)z-Displays regression information in messageboxzStatistics for Profile(s)rvrzProfile of %s c3&K|] }|d yw)rkN).0rs rM z'ProfileFrame.OnStats..sEQAaDEsruzn: %f z minimum: %f z maximum: %f z range: %f z mean: %f zstandard deviation: %f z variance: %f zcoefficient of variation: %f zsum: %f z median: %f zdistance along transect: %f )idrmtitleN)r5rrrBr{numpyfromiterr|raminamaxptpmeanstdvarsummedianr<rr rFID_ANYShowID_CLOSErd) rJr^rmrrraststatstriterableacvstatss rMOnStatszProfileFrame.OnStatss-.dkk* A wws|A-4E$++a.*DENN8U[[99qvv--?UZZ]::?UZZ]::=599Q<77<%**Q-775 ! DD+eiil::YYq\EJJqM1;b@@;155>ELLO;;|jdtdjftd|jjfdt djft d|jj fddt djft d|jjfdtdjftd|jjfdtdjftd|jjfd td jftd |jjfd td jftd |jjfdd t d jft d |jjfdt djft d|jjfdtdjftd|jjfdtdjftd|jjfddt djft d|jj fdt djft d|jj"ffS)z Toolbar data addrasteraddRastr.rSdrawerasedragpanzoomzoomInunzoomzoomBack statisticsdatasavesaveimagesaveFileprintsettingsoptionsquit)_getToolbarDatarlabelrrfrr_rTrUOnDragOnZoomOnRedrawrr SaveToFile PrintMenuPlotOptionsMenuOnQuitrQs rMrzProfileToolbar._toolbarDatas##!)I"6"<"<=i(KK..  :!6!r/sV" 04%><<x=xv X [X rN