ը f/TdZddlZddlZddlZddlZddlmZ ddlmZdZ ddl m Z ddlmZddlmZddlmZdd lmZGd d ZGd d ZGddZGddZd"dZdZdZdZd#dZdZ dZ!dZ"dZ#dZ$d$dZ%d$dZ&dZ'dZ(d Z)d!Z*y#e $rdZ YwxYw)%a @package animation.utils @brief Miscellaneous functions and enum classes Classes: - utils::TemporalMode - utils::TemporalType - utils::Orientation - utils::ReplayMode (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 Perasova N) cpu_count)ImageTF)encode) EmptyBitmap) GExceptionceZdZdZdZy) TemporalModeN)__name__ __module__ __qualname__TEMPORAL NONTEMPORAL0/usr/lib/grass83/gui/wxpython/animation/utils.pyr r )s HKrr ceZdZdZdZy) TemporalTyper r N)r r rABSOLUTERELATIVErrrrr.s HHrrceZdZdZdZy) Orientationr r N)r r rFORWARDBACKWARDrrrrr3s GHrrceZdZdZdZdZy) ReplayModer r N)r r rONESHOTREVERSEREPEATrrrrr8sGG Frrctj|}|jddk\r5|jdd\}}|||vr|St t d|ztj j}|D]&}||jvs|||vs|dz|zcSt t d|z)zwChecks if space time dataset exists and completes missing mapset. Raises GException if dataset doesn't exist. @rr z"Space time dataset <%s> not found.) tgis tlist_groupedfindsplitr_get_tgis_c_library_interfaceavailable_mapsetskeys) timeseriesetype trastDict nameShortmapsetmapsetss rvalidateTimeseriesNamer2>s ""5)Isq &,,S!4 6  &) ) QCDzQR R//1CCEG1 Y^^% %Yv..!C'&001 Q; not found.FT) grass list_groupedr&r'appendrr(six iteritems) namesr-mapDictnewNamesnamer/r0foundmapNamess rvalidateMapNamesr?Ts   'GHB 99S>Q  $ 3 2 IvGFO+% #8!9D!@AAE$'MM'$: 9 8# EOOD3J$78 9 #8!9D!@AAB Orcg}tj||}|jddd}g}|r|D]}|j|d|S)zhReturns list of maps registered in dataset. Can throw ScriptError if the dataset doesn't exist. idN start_time)columnswhereorder)r$ open_old_stdsget_registered_mapsr6)r,r-timeseriesMapssprowsrows rgetRegisteredMapsrLosaN  J .B ! !$d, ! ODN  -C  ! !#d) , - rcdx}}d|vr|jd\}}d|vr|jd\}}|r|dz|z}||fS)a=Checks whether map name contains layer and returns map name with mapset (when there was mapset) and layer (can be None). >>> getNameAndLayer('name:2@mapset') ('name@mapset', '2') >>> getNameAndLayer('name@mapset') ('name@mapset', None) >>> getNameAndLayer('name:2') ('name', '2') Nr#:)r')r<r0layers rgetNameAndLayerrP~sYFU d{zz# f d{jjo e czF" ;rcttttd}|r%|D]\}}tj||}|j}|j d}|dj ||dj ||j dd}|r{g} |dj t||D]9} |d k(r| j} n| j} | j| ;|d j t| |d j d|dj d"t|dd kDrttd t|dd kDrttd t|dd kDrttdt|d d kDrttd|rt} |D]} | j t| t| d kDrttd|r4t| dt|ddk7rttd|rt dS|rt|ddSy)a8Checks whether time series (map series and stds) are compatible, which means they have equal number of maps ad times (in case of stds). This is needed within layer list, not within the entire animation tool. Throws GException if these are incompatible. :return: number of maps for animation )count temporalTypemapTypemapTimesrrTrSNrB)rDrErRabsoluterUr z=The number of maps in space-time datasets has to be the same.zPThe temporal type (absolute/relative) of space-time datasets has to be the same.zHThe map type (point/interval) of space-time datasets has to be the same.zHThe temporal extents of maps in space-time datasets have to be the same.zEThe number of maps to animate has to be the same for each map series.z[The number of maps to animate has to be the same as the number of maps in temporal dataset.)setr$rF get_map_timeget_initial_valuesaddget_registered_maps_as_objectslenget_absolute_timeget_relative_timer6tuplerr(list) mapSeriesListtimeseriesListtimeseriesInfostdsr-rIrTtempTyperJtimesrKtimerR mapSeriess rcheckSeriesCompatibilityris5E N) 2KD%##D%0Boo'G,,.q1H 9 % ) )' 2 > * . .x 8444|4TDw'++CI6'C:-"446"446LL& ' z*..uU|<z*..t4w'++D1) 2, >' "#a' P Q   >. )*Q. &    >) $%) &    >* %&* '   & &I IIc)n % & u:>4  d5k!n^G5L0Ma0PPJ E{1~N7+,Q//rcz|dt|dz }|dt|dz }||krD|}t|d|zdz}t|d|zdz}d}t|d|z dz dz}nC|}t|d|zdz}t|d|zdz}d}t|d|z dz dz}|||||dS)aIFits source rectangle into destination rectangle by scaling and centering. >>> ComputeScaledRect(sourceSize = (10, 40), destSize = (100, 50)) {'height': 50, 'scale': 1.25, 'width': 13, 'x': 44, 'y': 0} :param sourceSize: size of source rectangle :param destSize: size of destination rectangle rr g?g@)widthheightxyscale)floatint) sourceSizedestSizeratio1ratio2rorkrlrmrns rComputeScaledRectrvsa[5A/ /F a[5A/ /F JqME)C/0Z]U*S01  !v%,s2 3JqME)C/0Z]U*S01  !u$+c1 2f11u MMrctjtdd}|j||j |\}}t|dz|dz}|j ||j tj|jtj||jtj||j|j|dd|j tj|S)z'Renders text with given font to bitmap.r r )wxMemoryDCrSetFont GetTextExtent SelectObjectSetBackgroundModeSOLIDSetTextBackgroundColourSetTextForegroundClearDrawText NullBitmap)textfontbgcolorfgcolordcwhbmps r RenderTextrs [R( )BJJt  D !DAq a!eQU #COOC"G,-G,-HHJKKaOOBMM" Jrctjd|j|jf}|j t |j |S)zConverts wx.Image to PIL imageRGB)rnewGetWidth GetHeight frombytesbytesGetData)imagepilImages r WxImageToPilr sCyy!15??3D EFH uU]]_-. Orcdj|}|r%|tt|jz }t j t |jS)zCReturns a hash from command given as a list and a region as a dict.r(joinstrsorteditemshashlibsha1r hexdigest)cmdregionr<s rHashCmdr'sH 88C=D  F6<<>*++ <<t % / / 11rcdj|Dcgc] }|D]}| c}}}|r%|tt|jz }t j t |jScc}}w)z:Returns a hash from list of commands and regions as dicts.;r)cmdsrsublistitemr<s rHashCmdsr/sf 88Bg'B$TBTB CD  F6<<>*++ <<t % / / 11CsA< cbtjj|t||dz|zS)z>> interpolate(0, 10, 5) [0, 2.5, 5.0, 7.5, 10] >>> interpolate(10, 0, 5) [10, 7.5, 5.0, 2.5, 0] r )rpr6)startendrRstepvaluess r interpolaters %K5+ +D F s{ck MM% TMEck uEk MM% TMEEkEAI& MM# Mr)strds)NN)ppm)+__doc__rryrr7multiprocessingrPILrhasPIL ImportErrorgrass.temporaltemporalr$ grass.scriptscriptr4grass.script.utilsr gui_core.wrapr core.gcmdrr rrrr2r?rLrPrirvrrrrrrrrrrrrrrs&  % F%%     K,6 ,Z0zN<"22I K $*N$M  FsBB'&B'