ը fU ZdZddlZddlZddlZddlmZddlmcmZddl m Z ddl m Z ddl mZmZddlmZddlmZdd lmZdd lmZddlmZed ed  eded ededed eded eddZGddej:Zy)ae @package wxplot.base @brief Base classes for iinteractive plotting using PyPlot Classes: - base::PlotIcons - base::BasePlotFrame (C) 2011 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)randint)ICONDIR) UserSettings) TextDialog OptDialog)Map)MetaIcon) BaseIcons)MenushowzDraw/re-draw plot)imglabelzlayer-raster-profilez.Draw transect in map display window to profilesettingsstatszPlot statisticssavezSave profile data to CSV filequit)drawtransectoptions statisticsrrceZdZdZddej ddej gfdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZ dZ!y) BasePlotFramez#Abstract PyPlot display frame classNiic tjj||ftj||d|||_||_t |_||_i|_d|_ tjtjtjtjtjd|_tj tjd|_|j%tj&t(j*j-t.dtj0|j3dd|_|j4j7d d gt9j:||_|j>|j<_ |j<jBjEtjF|jH|j<jBjEtjJ|jLg|_'d|_d|_(d|_)d|_*|jW|jYy) N)idsizestyle)soliddotz long-dashz short-dashzdot-dash) transparentrz grass.icor)numberr)-wxFrame__init__ID_ANYparent_gifacer rasterListrasterplottypeSOLIDDOT LONG_DASH SHORT_DASHDOT_DASH linestyledict TRANSPARENT ptfilldictSetIconIconospathjoinrBITMAP_TYPE_ICOCreateStatusBar statusbarSetStatusWidthsplot PlotCanvasclientDrawPointLabelpointLabelFunccanvasBind EVT_LEFT_DOWNOnMouseLeftDown EVT_MOTIONOnMotionplotlistptitlexlabelylabelCentreOnScreen_createColorDict)selfr)gifacerrr+kwargss ,/usr/lib/grass83/gui/wxpython/wxplot/base.pyr'zBasePlotFrame.__init__3s $W2994uWPVW  5$  XX66--  +-..288L RWWRWW\\';?ASASTU --Qa-@ &&Bx0 ood+ &*%8%8 "  0 0$2F2FG  t}}=       c`i|_tjtjD]}|dk(r t tj|ddz}t tj|ddz}t tj|ddz}|||df|j|<y)zkCreate color dictionary to return wx.Colour tuples for assigning colors to images in imagery groupswhiterr!N) colorDictsixiterkeysgrass named_colorsint)rPclrrgbs rSrOzBasePlotFrame._createColorDictss<< 2 23 1Cg~E&&s+A.45AE&&s+A.45AE&&s+A.45A#$aC.DNN3   1rTc||_i|_i|jd<tj|jd|jdd<t j |jddd|jddd|jddd|jddd|_i|jd<tj|jd|jd<t|jdd }td |jd jd D|jdd <|jd k(rtj|jd|jd<t|jdd}td|jd jd D|jdd<tj|jd|jd<t|jdd}td|jd jd D|jdd<i|jd<tj|jd|jdd<d|jdd<i|jd<tj|jd|jdd<d|jdd<tj|jd|jd<d|_ d|_ d|j_|jddddk(rd|j_n$|jddd|j_|jddddk(rd|j_y|jddd|j_y)z"Initialize options for entire plotfontgroupkeyprop defaultSizefamilyrweightr,pcolorc32K|]}t|ywNr^.0colvals rS z-BasePlotFrame.InitPlotOpts..s4 "CK4 z(),profilemarkercolorc32K|]}t|ywrnrorps rSrsz-BasePlotFrame.InitPlotOpts..s7 &F 7rtgridc32K|]}t|ywrnrorps rSrsz-BasePlotFrame.InitPlotOpts..s1 "CK1 rtx-axisNaxisy-axislegendF)FFtypecustommin)r- propertiesrGetr%Fontwx_fontstrtuplestripsplitzoomdragrAlogScalexSpecySpec)rPr-colstrs rS InitPlotOptszBasePlotFrame.InitPlotOptss  "$*6*:*:--V+ 'ww OOF #F +M : OOF #F +H 5 OOF #F +G 4 OOF #F +H 5   %'!$0$4$44==h$W!T__X.x89.34 &,ll4&8&>&>s&C4 / !(+ ==I %(4(8(8mm)DOOH %27;&>s&C1 , (%'!,8,<,<--X- !&)-1!&)$&!,8,<,<--X- !&)-1!&)$0$4$44==h$W!   .  ??8 $V ,V 4 @ %DKK  $ 9& A& IDKK  ??8 $V ,V 4 @ %DKK  $ 9& A& IDKK rTc i}tj|jd|jd<|D]}|j |} t j |}|jd|j|<i||<d||d<|ddvr |d||d<|||d<g||d<d||d <|d ||d <|jdd |jdd ||d <nd ||d <|jdd .|jdd dk7r|jdd ||d <nd||d <|t|jkry|dk(rP|jdd|jdd||d<M|j|j|||d<q|j|j|||d<tdd}tdd}tdd}|||df||d<|S#YxYw)z3Initialize or update raster dictionary for plottingr,rerunitsz(none)z"none"rNplegenddatalistNplinedatatypepwidthrXpstylerrrlrW) rrr-rindexr\ raster_infor,len colorListrYr) rPr+r-rdictr`idxretrbras rSInitRasterOptszBasePlotFrame.InitRasterOptssO$0$4$44==h$W!3 4A""1%C ''* "__X6DKKNE!H "E!HW 7|#AA$'La!"#E!HY #%E!HZ $E!HW #&z?E!HZ x(2>%)__X%>x%Ha"%&a")(3?OOH-h72=%)__X%>x%Ha"%,a"S((!8x0:F-1__X-Fx-Pa*-1^^DNN3 8E""5)C ((q2((q2 ".!1!1I"DKK E%L E%LO E%LO')E%LOG $')E%LOG $G}$BB+/=e Q(G}$BB+/=e Q(as#A&/%(..2Ea2HH %L #(*E%L $*.z*:E%LOJ '*.z*:E%LOJ ' )'2>OOH-g6"<(,(A'(Je W%(-e W%x(1=(,(A'(Je W%()e W%)'2>OOH-g6"<(,(A'(Je W%(/e W%c$..)))-s8K)Le X&AsOAsOAsO*+Q3e X&}> 8@ s s 0I##I(c|jj|j|jddd|j_|jddd|j_|j |j_|j|j_ |jddddk(rd|j_ n$|jddd|j_ |jd dddk(rd|j_ n$|jd dd|j_ |jddddk(rf|jddd|jddd kr;|jddd|jddd f|jdd <nd |jdd <|jd dddk(rf|jd dd|jd dd kr;|jd dd|jd dd f|jd d <nd |jd d <|jddd r#d |jdd <d|j_ |jd dd r#d |jd d <d|j_ |jddd |jd dd f|j_ |jdd|j_ tjtj |jddd|jddd|jdddd}||j_|jddd|j_|jdd|j_y )zSet plot and text optionsrdrh titleSizeaxisSizer|rrrr~maxr}NlogrzenabledrxrrXr!rW)colour legendSizer)rASetFontrr fontSizeTitle fontSizeAxisr enableZoomr enableDragrrr enableGridr%PenColourgridPenfontSizeLegend enableLegend)rPgridpens rS SetGraphStylezBasePlotFrame.SetGraphStyleMs DLL)$(OOF$;F$CK$P !#'??6#:6#B:#N  !% !%  ??8 $V ,V 4 @ %DKK  $ 9& A& IDKK  ??8 $V ,V 4 @ %DKK  $ 9& A& IDKK  OOH %f -f 5 A)&1%8ooh'/67)&1%8)&1%81DOOH %f - 15DOOH %f - OOH %f -f 5 A)&1%8ooh'/67)&1%8)&1%81DOOH %f - 15DOOH %f - ??8 $V ,U 304DOOH %f - %DKK  ??8 $V ,U 304DOOH %f - %DKK  OOH %f -e 4 OOH %f -e 4  "&!8!C &&99'03'03'03  &  &*__V%|j\}}tj||j|||_|jddddk(rd|j _n$|jddd|j _|jddddk(rd|j _n$|jddd|j _|j j|j|jdd|jddy) z1Draw line and point plot from list plot elements.r|rhrrrr~r}N) _getPlotLabelsr? PlotGraphicsrKrrArrDraw)rPrJrLrMs rSDrawPlotzBasePlotFrame.DrawPlots,,.%%h VVL ??8 $V ,V 4 @ %DKK  $ 9& A& IDKK  ??8 $V ,V 4 @ %DKK  $ 9& A& IDKK   II OOH %f - OOH %f - rTc|jtjtj|j tj tjtj |d\}}|j|dz |dz dd|d\}}|d}|d}|d} d|| |||fz} |j| ||d zy ) akThis is the function that defines how the pointLabels are plotted dc - DC that will be passed mDataDict - Dictionary of data that you want to use for the pointLabel As an example I have decided I want a box at the curve point with some text information about the curve plotted below. Any wxDC method can be used. scaledXY pointXYcurveNumpIndexrz(Crv# %i, '%s', Pt. (%.2f,%.2f), PtInd %irXN) SetPenr%rBLACKSetBrushBrushr. DrawRectangleDrawText) rPdc mDataDictsxsypxpycNumpntInrss rSrBzBasePlotFrame.DrawPointLabels "&&"# BHHRXXrxx01:&B aaR09%B$(#8$ 6$BPU9V V Ar26"rTcd|_d|_|j|j_|j|j_y)z#Enable zooming and disable draggingTFN)rrrArrrPevents rSOnZoomzBasePlotFrame.OnZooms0  !% !% rTcd|_d|_|j|j_|j|j_y)z#Enable dragging and disable zoomingFTN)rrrArrrs rSOnDragzBasePlotFrame.OnDrags0  !% !% rTc|j|jj|jjy)z/Redraw the plot window. Unzoom to original sizeN) UpdateLabelsrAResetRedrawrs rSOnRedrawzBasePlotFrame.OnRedraws.   rTc8|jjy)zErase the plot windowN)rAClearrs rSOnErasezBasePlotFrame.OnErases rTc8|jjy)zSave plot to graphics fileN)rASaveFilers rS SaveToFilezBasePlotFrame.SaveToFiles rTc|jtdd|jj|zz|j y)NzLeft Mouse Down at Point:z (%.4f, %.4f)) SetStatusText_rA_getXYSkiprs rSrGzBasePlotFrame.OnMouseLeftDowns>  ) *_t{{?Q?QRW?X-X X   rTc|jjrg|jj|jj|d}|gk7r,|\}}}}}}|||||d} |jj | |j y)z,Indicate when mouse is outside the plot areaT) pointScaled)rrrrrN)rAenablePointLabelGetClosestPointrUpdatePointLabelr) rPrdlstrrrrrdistancers rSrIzBasePlotFrame.OnMotions ;; ' ';;.. ""5)t/DrzHLE&&'8X!)$$& (   ,,Y7 rTc*tj}t}tj|tjt d}|j ||jtj|j|tj|tjt d}|j ||jtj|j||j||jy)z$Popup menu for plot and text options Text settings Plot settingsN) r%GetMousePositionr MenuItemr(r AppendItemrEEVT_MENUPlotText PlotOptions PopupMenuDestroy)rPrpointpoptsettextsetgrids rSPlotOptionsMenuzBasePlotFrame.PlotOptionsMenus##%v++dBIIq/AB   "++t}}g6++dBIIq/AB   "++t//9 t rTctj|tdtdtjtjz}|j |j y)z)Creates a 'not functional' message dialogz"This feature is not yet functionalzUnder Construction)r)messagecaptionrN)r% MessageDialogrOKICON_INFORMATION ShowModalrrPdlgs rS NotFunctionalzBasePlotFrame.NotFunctionalsM:;*+%%"---    rTcd}|j}|jdddr||}|j}|jdddr||}||fS)Ncd|zdzS)Nzlog( z ))txts rSrz)BasePlotFrame._getPlotLabels..logsS=4' 'rTr|rhrr~)rLrrM)rPrxys rSrzBasePlotFrame._getPlotLabelssb ( KK ??8 $V ,U 3AA KK ??8 $V ,U 3AA!t rTc|j|_|j|_|j|_|jr|j|j_|j dy)z(Custom text settings for histogram plot.N)r)rKrLrMr?titlerrs rS OnPlotTextzBasePlotFrame.OnPlotText+sGjj jj jj 99!jjDIIO D !rTcb|j\}}|jj|j|jddd|j_|jddd|j_|jr#||j_||j_ yy)Nrdrhrr) rrArrrrrr?xLabelyLabel)rPrrs rSrzBasePlotFrame.UpdateLabels6s""$1 DLL)$(OOF$;F$CK$P !#'??6#:6#B:#N  99 DII  DII  rTc .t||jtj|jt d}|j }|tjk(s&|tjk(s|tjk(r|jyy)z9Set custom text values for profile title and axis labels.rr)rQrr-rN) rr*r%r(r-rrID_SAVEID_OK ID_CANCELrrPrrbtnvals rSrzBasePlotFrame.PlotTextAsj<<yy]]O$   RZZ 6RXX#52<<9O KKM:PrTc Lt||jtj|jt d}|j }|tjk(s&|tjk(s|tjk(r|j|jy)zSet various profile options, including: line width, color, style; marker size, color, fill, and style; grid and legend options. Calls OptDialog class. rr!N) rr*r%r(r-rrr"r#r$rUpdater%s rSrzBasePlotFrame.PlotOptionsOso <<yy]]O$   RZZ 6RXX#52<<9O KKM rTctj}t}td|jftd|j ftd|j ffD]\\}}tj|tj|}|j||jtj||^|j||jy)zPrint options and output menuz Page setupz Print previewz Print displayN)r%rr r OnPageSetupOnPrintPreview OnDoPrintrr(rrErrr)rPrr printmenurhandleritems rS PrintMenuzBasePlotFrame.PrintMenubs##%F |_d.. /  !4!4 5   0  2NE7 ;;y"))U;D   & IIbkk7D 1 2 y!rTc8|jjyrn)rA PageSetuprs rSr*zBasePlotFrame.OnPageSetupts rTc8|jjyrn)rA PrintPreviewrs rSr+zBasePlotFrame.OnPrintPreviewws   "rTc8|jjyrn)rAPrintoutrs rSr,zBasePlotFrame.OnDoPrintzs rTc&|jdy)NT)Closers rSOnQuitzBasePlotFrame.OnQuit}s 4rT)"__name__ __module__ __qualname____doc__r%SizeDEFAULT_FRAME_STYLEr'rOrrrrrrBrrrrrrGrIr rrrrrrr0r*r+r,r9rrTrSrr0s- RWWS# $$ > @ 1CJJ;zIVLH\ *#.++  .$   " ! &$ #rTr)r=r8rZr%randomr wx.lib.plotlibr?core.globalvarr core.settingsrwxplot.dialogsrr core.renderr icons.iconr gui_core.toolbarsr gui_core.wrapr grass.scriptscriptr\r PlotIconsr&rrrTrSrMs" "&0' q)<'= > "@A$wa0A.BC q)H'I J f   N BHHN rT