ը f%}dZddlZddlZddlZddlZddlmZmZddlm Z m Z ddl m Z ddl mZddlmZddlmZmZmZmZdd lmZmZddlmcmZdd lmZdd lm Z Gd d Z!GddZ"GddZ#dZ$dZ%dZ&GddZ'Gdde'Z(Gdde'Z)GddZ*dZ+d"dZ,dZ-d Z.e/d!k(re.yy)#a @package animation.provider @brief Animation files and bitmaps management Classes: - mapwindow::BitmapProvider - mapwindow::BitmapRenderer - mapwindow::BitmapComposer - mapwindow::DictRefCounter - mapwindow::MapFilesPool - mapwindow::BitmapPool - mapwindow::CleanUp (C) 2013 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)ProcessQueue) GException DecodeString) UserSettings)Debug)autoCropImageFromFile)HashCmdHashCmdsGetFileFromCmdGetFileFromCmds) EmptyBitmapBitmapFromImage)cmdlist_to_tuple)Signalc`eZdZdZ ddZddZdZdZdZdZ d Z dd Z d Z d Z d ZdZy)BitmapProviderzClass for management of image files and bitmaps. There is one instance of this class in the application. It handles both 2D and 3D animations. cJ||_||_||_||_||_g|_g|_g|_g|_d|_ g|_ g|_ t|j|j|j|j|_ t|j|j|j|j|j|_t!d|_t!d|_t!d|_t!d|_t!d|_t!d|_t!d|_|jj&j1|j&|jj(j1|j(y)NzBitmapProvider.renderingStartedz!BitmapProvider.compositionStartedz!BitmapProvider.renderingContinuesz#BitmapProvider.compositionContinuesz BitmapProvider.renderingFinishedz"BitmapProvider.compositionFinishedzBitmapProvider.mapsLoaded) _bitmapPool _mapFilesPool imageWidth imageHeight_tempDir _uniqueCmds_cmdsForComposition _opacities_cmds3D _regionFor3D_regions_regionsForUniqueCmdsBitmapRenderer _rendererBitmapComposer _composerrrenderingStartedcompositionStartedrenderingContinuescompositionContinuesrenderingFinishedcompositionFinished mapsLoadedconnect)self bitmapPool mapFilesPooltempDirrrs 3/usr/lib/grass83/gui/wxpython/animation/provider.py__init__zBitmapProvider.__init__0sc&)  ' #%    %'"'    t@P@P ( MM       OO     !''H I"()L"M"()L"M$*+P$Q!!'(J!K#)*N#O  !<= ))11$2I2IJ ++33D4M4MNNc$tjddjt||jj ||j j ||jj ||jy)aSets commands to be rendered with opacity levels. Applies to 2D mode. :param cmdsForComposition: list of lists of command lists [[['d.rast', 'map=elev_2001'], ['d.vect', 'map=points']], # g.pnmcomp [['d.rast', 'map=elev_2002'], ['d.vect', 'map=points']], ...] :param opacities: list of opacity values :param regions: list of regions z!BitmapProvider.SetCmds: {n} lists)nN) rmsgformatlenrextendrr_getUniqueCmds)r-cmdsForComposition opacitiesregionss r1SetCmdszBitmapProvider.SetCmdsZsq  299C@RDctjdd|jrt|j|j D]\}}|j t||=t|j|jD]\}}|jt||=g|_g|_g|_ g|_g|_|jrg|_ d|_ yy)z[Unloads currently loaded data. Needs to be called before setting new data. r5zBitmapProvider.UnloadN)rr7rrIrr rr rrr rrr)r-rRrDrQs r1UnloadzBitmapProvider.Unloads  !,-  # #"4#3#3T5O5OP = V&&wsF';< =$'t'?'?#O @$$Xgv%>? @!D ')D $ DODM)+D & <<DL $D  r3cvd}t||D]\}}t|j||}|s^tjj |r?|j jt|||j|jfk(r}|dz }tjddj||S)zDetermines how many files will be rendered. :param uniqueCmds: list of commands which are to be rendered :param force: if forced rerendering :param regions: list of regions assigned to the commands rrGz3BitmapProvider._dryRender: {c} files to be renderedrA)rIr rospathexistsrGetSizer rrrr7r8)r- uniqueCmdsr>forcecountrRrDfilenames r1 _dryRenderzBitmapProvider._dryRendersz73 KC%dmmS&AHGGNN8,&&..wsF/CDOOT%5%567 QJE   DKKeKT  r3c>d}t||D]e\}}|sYt|||jvrA|jt||j|j|j fk(ra|dz }gt jddj||S)aDetermines how many lists of (commands) files will be composed (with g.pnmcomp). :param cmdLists: list of commands lists which are to be composed :param regions: list of regions assigned to the commands :param force: if forced rerendering rrGr5z4BitmapProvider._dryCompose: {c} files to be composedrA) rIr rr[rrrr7r8)r-cmdListsr>r]r^rQrDs r1 _dryComposezBitmapProvider._dryComposes"8W5 OGVWf-1A1AA$$Xgv%>?GGIOOT%5%567 QJE   ELLuLU  r3c`tjddj|||g}g}|jr6|j |j|j |j |j rC|j |j |j dgt|j z|j|||}|jj|d|jvs|r-t|j|j|jd<|jj!|||j"|||}|j$j|s|j&jy|j(r|j+|j(|j,|}|j.j||j0j3|j(|j,|j4||||j6j|j rS|j D]D}t9j:t=|j>|d|jtA|gd<F|j&jy) anLoads data, both 2D and 3D. In case of 2D, it creates composites, even when there is only 1 layer to compose (to be changed for speedup) :param force: if True reload all data, otherwise only missing data :param bgcolor: background color as a tuple of 3 values 0 to 255 :param nprocs: number of procs to be used for rendering r5z7BitmapProvider.Load: force={f}, bgcolor={b}, nprocs={n})fbr6N)r]r^) regionFor3Dbgcolorr]nprocs)rir]rj)!rr7r8rr:r rr9r`r%emitrcreateNoDataBitmaprrr"Renderrr)r+rrcrr&r$Composerr*wxBitmapr rr ) r-r]rirjrCr>r^okrRs r1LoadzBitmapProvider.LoadsC   117%7f1U     KK(( ) NN455 6 << KK % NND6C $55 6gU; """/ t'' '5%7!1!1&D  T "^^ " "  )) #  ##% OO "   # #$$(($--u%E  # # ( (u ( 5 NN " "((  #   $ $ ) ) + <<|| :<))"4==#t<;  3%!67  r3ctjdd|jj|jj y)z&Requests to stop rendering/compositionr5z#BitmapProvider.RequestStopRenderingN)rr7r"RequestStopRenderingr$RequestStopComposingr-s r1rtz#BitmapProvider.RequestStopRenderings1 !:; ++- ++-r3cd |j|}|S#t$r|jd}Y|SwxYw)zReturns bitmap with given key or 'no data' bitmap if no such key exists. :param dataId: name of bitmap N)rKeyError)r-dataIdbitmaps r1 GetBitmapzBitmapProvider.GetBitmapsD  ,%%f-F  ,%%d+F  ,s //ctjddj||||c|_|_|x|j _|j _|x|j _|j _y)z.Sets size when size of related window changes.z.BitmapProvider.WindowSizeChanged: w={w}, h={h})whN)rr7r8rrr$r")r-widthheights r1WindowSizeChangedz BitmapProvider.WindowSizeChanged$sg < C Cev C V -26))@EE!DNN$=BHH"T^^%?r3ctjddj|tjd\}}t j |t|j|j|ddtjd d t|zt|}t|d fi|d \}}}|d k(rtt|St j|t!|) zCreates raster legend with d.legend :param cmd: d.legend command as a list :return: bitmap with legend r}z#BitmapProvider.LoadOverlay: cmd={c}rA.png)suffixT)rrr transparentrirGzRender raster legend r)rr7r8tempfilemkstemprXclose_setEnvironmentrrstrr read2_commandrr remover)r-rR fileHandlerr_cmdTuple returncodestdoutmessagess r1 LoadOverlayzBitmapProvider.LoadOverlay0s  !:AACAHI ( 0 0 ? X  OO        !,s8}<=#C('4Xa['PHQK'P$ FH ?"#8#BC C IIh X& &r3)N)F)rr)__name__ __module__ __qualname____doc__r2r?rEr;rUr`rcrrrtr{rrr3r1rr)sNNQ(OT(# &%(22?B.  I'r3rc"eZdZdZdZdZdZy)r!z.Class which renders 2D and 3D images to files.cx||_||_||_||_t d|_d|_d|_y)Nz!BitmapRenderer.renderingContinuesF)rrrrrr'_stopRendering _isRendering)r-r/r0rrs r1r2zBitmapRenderer.__init__Rs>) $&"()L"M#!r3c tjddd}d}g} g} g} g} t||D]\} }| ddk(rd}t|j| |}|swt j j|rX|jjt| ||j|jfk(r||jt| |<| j| |ft| }d}d|_| D]\} }|dz }t!}| ddk(r7t#t$|j|j|j| |||f }n6t#t&|j|j|j| |||f }|j)| j|| j|| j| |f|dz }||k(s||k(rt+t| D]}| |j-| |j/}||jt| |d| |d<|jj1t| |d| |d|j|jfd}g} g} g} |j2j5|t7d  |j8sd|_d}nd|_| S) aRenders all maps and stores files. :param cmdList: list of rendering commands to run :param regions: regions for 2D rendering assigned to commands :param regionFor3D: region for setting 3D view :param bgcolor: background color as a tuple of 3 values 0 to 255 :param force: if True reload all data, otherwise only missing data :param nprocs: number of procs to be used for rendering rWzBitmapRenderer.Renderrz m.nviz.imageNFTrGtargetargszRendering map layerscurrenttext)rr7rIr rrXrYrZrr[r rrrJr9rrrRenderProcess3DRenderProcess2DstartrangejoingetSetSizer'rk_r)r-rQr>rhrir]rjr^ proc_count proc_list queue_listcmd_listfilteredCmdListrRrDr_mapNumstoppedqpis r1rmzBitmapRenderer.Render\s  !,-   w0 2KC1v'%dmmS&AHGGNN8,&&..wsF/CDOOT%5%567 opacityListrir]rjr^rrr cmd_listsfilteredCmdListsrQrDnumrrrr_s r1rnzBitmapComposer.Composes  !-.    "8W5 7OGVWf-1A1AA$$Xgv%>?GGIOOT%5%567 ?C>N>NWf-?  '6!:;  # #Wf$5 6 7"# /4 OGV QJEA'OO$$MM  A GGI   a   Q    gv. / !OJV#u|s9~. ,AaL%%')!}002H'/ OOT-=-=DV(($Yq\!_il1oF,BHHX,>?(($Yq\!_il1oF (+ ,     % % * *A&=$> + ""&+#!m4 l"r3c,|jrd|_yy)z!Requests to stop the composition.TN)rrrvs r1ruz#BitmapComposer.RequestStopComposing7rr3N)rrrrr2rnrurr3r1r#r#sL "]"~'r3r#ct|||}d}t||||||r&tjdi|tj d<t |} t| dfi| d\} } } | dk7r`tjd| z|jd|rtj jdt j|y|rtj jd|j|y) aRender raster or vector files as ppm image and write the resulting ppm filename in the provided file queue :param imageWidth: image width :param imageHeight: image height :param tempDir: directory for rendering :param cmd: d.rast/d.vect command as a list :param region: region as a dict or None :param bgcolor: background color as a tuple of 3 values 0 to 255 :param fileQueue: the inter process communication queue storing the file name of the image Tr GRASS_REGIONrrGRendering failed: Nr) r rgcore region_envrXenvironrrwarningputpopr) rrr0rRrDri fileQueuer_ transparencyrrrrs r1rr=sgsF3HLK|W%*%5%5%?%? >"$H#0!#L #L JQ +h67 d  JJNN> * (  ~& MM(r3ct||d}tjdddi|tjd<t j ddt|zt|}tjj|d|dd<d ||fz|dd <d |dd <d j|D cgc] } t| c} x|dd<}t|dfi|d\} } } | dk7rItjd| z|jdtjjdytjjd|j|ycc} w)aRenders image with m.nviz.image and writes the resulting ppm filename in the provided file queue :param imageWidth: image width :param imageHeight: image height :param tempDir: directory for rendering :param cmd: m.nviz.image command as a list :param region: region as a dict :param bgcolor: background color as a tuple of 3 values 0 to 255 :param fileQueue: the inter process communication queue storing the file name of the image Nregion3dTrrGzRender image to file routputz%d,%dsizeppmr8:rirr)r rrrXrrr7rrrYsplitextrrrrr) rrr0rRrDrirr_rpartrrrs r1rrcs@gsD1H!&!1!1!J4!J6!JBJJ~ IIa(3x=89$HGG,,X6q9HQK!Z$==HQK!HQK'*xxw0OtT0O'PPHQK W#0!#L #L JQ +h67 d ~&JJNN>" MM(1Ps/E%c g}g} |D];} |jt|| || jt|| |d=t|||} t||| d||D cgc] } t | }} dj |D cgc] } t | c} }t ddddj t|zddj t| zddj t|z|||| \}}}|d k7r?tjd |z|jd tj| y |j| y cc} wcc} w) a-Performs the composition of image ppm files and writes the resulting ppm filename in the provided file queue :param imageWidth: image width :param imageHeight: image height :param tempDir: directory for rendering :param cmdList: list of d.rast/d.vect commands :param region: region as a dict or None :param opacites: list of opacities :param bgcolor: background color as a tuple of 3 values 0 to 255 :param fileQueue: the inter process communication queue storing the file name of the image pgmFrrz g.pnmcompTz%s,) overwriteinputmaskopacityrirrrrzRendering composite failed: N) rJr r rrrrreversedrrrrXr)rrr0rQrDr=rirmapsmasksrRr_oprrrrs r1rrsQ" D EB N7C89 ^GS&%@ABw8HKug$--RR-I-hhg6dD 67G#0SXXhtn-- CHHXe_- -sxx 344 $ JQ 5@A d ( MM().6s &E! E&c:eZdZdZdZdZdZdZdZdZ dZ y ) DictRefCounterzBase class storing map files/bitmaps (emulates dictionary). Counts the references to know which files/bitmaps to delete. c i|_i|_yr) dictionaryreferenceCountrvs r1r2zDictRefCounter.__init__s r3c |j|Srrr-keys r1 __getitem__zDictRefCounter.__getitem__ss##r3c||j|<||jvrd|j|<n|j|xxdz cc<tjddj |y)NrGr}z*DictRefCounter.__setitem__: +1 for key {k}k)rrrr7r8)r-rvalues r1 __setitem__zDictRefCounter.__setitem__sa$ d)) )'(D   $    $ ) $ !AHH3HOPr3c||jvSrrrs r1 __contains__zDictRefCounter.__contains__sdoo%%r3c~|j|xxdzcc<tjddj|y)NrGr}z*DictRefCounter.__delitem__: -1 for key {k}r)rrr7r8rs r1 __delitem__zDictRefCounter.__delitem__s5 C A%  !AHH3HOPr3c6|jjSr)rkeysrvs r1rzDictRefCounter.keyss##%%r3ctjdd|jjj D]2}||j |dks|j|=|j |=4y)z+Clears items which are not needed any more.rzDictRefCounter.ClearNr)rr7rcopyrrrs r1ClearzDictRefCounter.Clearsh !+,??'')..0 1C&&s+q0,++C0  1r3N) rrrrr2rrrrrrrr3r1rrs,!$Q&Q&1r3rc(eZdZdZdZdZdZdZy) MapFilesPoolz Stores rendered images as files.c<tj|i|_yr)rr2rrvs r1r2zMapFilesPool.__init__s% r3c"||j|<yrr)r-rrs r1rzMapFilesPool.SetSizes #r3c |j|Srrrs r1r[zMapFilesPool.GetSizesyy~r3ctjddt|jj D]}|j |dkst jj|j|\}}t j|j||dk(rt j|dz|j|=|j |=|j|=y)zVRemoves files which are not needed anymore. Removes both ppm and pgm. rzMapFilesPool.Clearrz.ppmz.pgmN) rr7rHrrrrXrYrrr)r-rnameexts r1rzMapFilesPool.Clears  !)*,,./ #C""3'1,GG,,T__S-AB c $//#./&=IIdVm,OOC('',IIcN #r3N)rrrrr2rr[rrr3r1rrs*#r3rceZdZdZdZy) BitmapPoolz+Class storing bitmaps (emulates dictionary)c.tj|yr)rr2rvs r1r2zBitmapPool.__init__s%r3N)rrrrr2rr3r1r r s 5&r3r ceZdZdZdZdZy)CleanUpz&Responsible for cleaning up the files.c||_yr)r)r-r0s r1r2zCleanUp.__init__ s  r3cddl}tjj|jrM |j |jt jddj|jyy#t$r;tjtdj|jYywxYw)Nrr}zCleanUp: removed directory {t})tzDirectory {t} not removed.) shutilrXrYrZrrmtreerr7r8OSErrorrrr)r-rs r1__call__zCleanUp.__call__ s 77>>$-- ( W dmm, !=DDt}}DUV ) W a <=DDt}}DUV WsA A<jJKycc}w)Nr) LayerListLayer) AnimLayer)layerListToCmdsMatrixstrdsJRzd.rastzmap=elev_2007_1mvectorbuildings_2009_approx)zd.vectzmap=buildings_2009_approxz color=grey2z /tmp/testrr)rrc`tjjdj|S)NzTotal number of maps: {c} rAsysrwriter8rgs r1ztest..s$cjj&&'D'K'Ke'K'TUr3c`tjjdj|S)NzCurrent number: {c} rArLrs r1rOztest..s&cjj.. # * *W * 5 r3c`tjjdj|S)Nz'Composition: total number of maps: {c} rArLrgs r1rOztest..s&cjj&& 6 = = = F r3c`tjjdj|S)Nz!Composition: Current number: {c} rArLrs r1rOztest..s&cjj.. 0 7 7' 7 B r3c@tjjdS)NzMaps loading finished )rMrrNrr3r1rOztest..sCJJ$4$45N$Or3) r)rirjr)&rcore.layerlistrBrCanimation.datarDanimation.utilsrEgrass.temporaltemporalinitmapTyperrRAddLayerrr rrXrYrZrmkdirrr%r,r'r&r(r+r?roApprrrSaveFilerBITMAP_TYPE_PNG)rrBrCrDrEtgis layerListlayerbPoolr/r0prov cmdMatrixapprs r1testrjbs/(5!IIK I KEEMEJ-.EI u GEEM(EJEEIEM u LE>LG ww~~g gHHW %w3TW XD!!U ##   ##   %%   OOOP%i0ILL BuU]]BC &&(CIInQI/zz|Y ? #J   WcFl CREWEW XY Cs*I __main__)zNo data)0rrXrMrormultiprocessingrr core.gcmdrr core.settingsr core.debugr core.utilsr rYr r r r gui_core.wraprrgrass.script.corescriptcorergrass.script.taskrgrass.pydispatch.signalrrr!r#rrrrrr r rrlrrjrrr3r1rws* *.&,NN6!!.*c'c'L x'x'vp'p'f#L!H0f%1%1P#>#<&&WW"4@DE@Y` zFr3