ը fbdZddlmZddlZddlZddlZddlmZ ddlm Z ddlZ dd l e dd ldd ldd ldd ldd ldd ldd lmZdd l m!Z!dd l"m#Z#ddl$m%Z%ddl&m'Z'ddl(m)Z*da+da,dZ-dZ. e/e0e1e2e1Z3e3e-a4e/e0e1e1Z5e5e.a6Gdde7Z8Gdde7Z9Gdde9Z:y#e $r"e dZ e de zejYwxYw#e$r-Ze d j'eejYdZ[dZ[wwxYw#e$reZYwxYw#e eef$r.Ze d j'eejYdZ[dZ[wwxYw#e$rYwxYw)a_ @package nviz.wxnviz @brief wxGUI 3D view mode (ctypes-based classes) This module implements 3D visualization mode for map display (ctypes required). List of classes: - wxnviz::Nviz - wxnviz::Texture - wxnviz::ImageTexture - wxnviz::TextTexture (C) 2008-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 Martin Landa (Google SoC 2008/2010) @author Pythonized by Glynn Clements @author Anna Kratochvilova (Google SoC 2011) )print_functionN)sqrt)matrixzThis module requires the NumPy module, which could not be imported. It probably is not installed (it's not part of the standard Python distribution). See the Numeric Python site (http://numpy.scipy.org) for information on downloading source or binaries.z wxnviz.py: )file)*z wxnviz.py: {})Debug)autoCropImageFromFile) DecodeString)wxPythonPhoenix)RectctrHtjjdk\rt |j }tj |yt|y)zRedirect stderrr)logsys version_infomajorr datawriteprint)msgtypes ,/usr/lib/grass83/gui/wxpython/nviz/wxnviz.py print_errorrLsE    ! !Q &sxx(C #  c ctrBtjdk(stjdtj|yt |y)zRedirect progress infodr)progressGetRangeSetRangeSetValuer)values rprint_progressr"YsE  "c)   c "%   e rceZdZdZdZdZdydZdZdZdZ dZ dyd Z d Z d Z d Zd ZdZdZdZdZdZdZdzdZdZdZdZdZdZdZdZdZdZdZ dZ!d Z"d!Z#d"Z$d#Z%d$Z&d%Z'd&Z(d'Z)d(Z*d)Z+d*Z,d+Z-d,Z.d-Z/d.Z0d/Z1d0Z2d1Z3d2Z4d3Z5d4Z6d5Z7 d{d7Z8 d|d8Z9d9Z:d:Z;d;Zd}d>Z?d?Z@d@ZAdAZBdBZCdCZDdDZEdEZFdFZGdGZHdHZIdIZJdJZKdKZLdLZMdMZNdNZOdOZPdPZQdQZRdRZSdSZTdTZUdUZVdVZWdWZXdXZYdYZZdZZ[d[Z\d\Z]d]Z^d^Z_d_Z`d`ZadaZbdbZcdcZdd~ddZedeZfdfZgddgZhdhZidiZjdjZkdkZldlZmdmZndydnZodydoZpddpZqdqZrdrZsdsZtdtZuduZvdvZwdwZxdxZyy6)Nvizcx|a|atdttt t |jt|_ t|j|_ t|_ t|j|_dx|_|_d|_t%j&ddy)zpInitialize Nviz class instance :param glog: logging area :param gprogress: progressbar wxnvizFz Nviz::Nviz()N)rr G_gisinitG_set_error_routineerrfuncG_set_percent_routineperfuncInitnv_datadata_objpointerrColors color_objcolorwidthheight showLightrr)selfglog gprogresss r__init__z Nviz.__init__ps(G$g&   DMM* T^^, #%% T[ !^$rcBtt|`|`d|_y)zDestroy Nviz class instanceN)G_unset_error_routineG_unset_percent_routinerr0rr8s r__del__z Nviz.__del__s! I Mrctjdk7r$tjtjdt t tttty)zInitialize windowwin32CN) rplatformlocale setlocale LC_NUMERICG_unset_windowRast_unset_windowRast__init_window GS_libinit GVL_libinitGVL_init_regionr?s rr.z Nviz.InitsF <<7 "   V.. 4  rct||z|_t||z|_tjdd|j|jt |j|jS)zGL canvas resized :param width: window width :param height: window height :return: 1 on success :return: 0 on failure (window resized by default to 20x20 px) rz(Nviz::ResizeWindow(): width=%d height=%d)intr5r6rrNviz_resize_window)r8r5r6scales r ResizeWindowzNviz.ResizeWindowsY' &5.)  94::t{{ "$**dkk::rc,t|jS)z;Get longest dimension, used for initial size of north arrow)Nviz_get_longdimrr?s r GetLongDimzNviz.GetLongDim **rct}t|j|t}t}t}t t |t |t |t jdd|j|j|j||j|j|jfS)zlSet default view (based on loaded data) :return: z-exag value, default, min and max height r(z1Nviz::SetViewDefault(): hdef=%f, hmin=%f, hmax=%f) Nviz_get_exagNviz_change_exagrc_doubleNviz_get_exag_heightbyrefrrr!)r8z_exaghdefhminhmaxs rSetViewDefaultzNviz.SetViewDefaults F+zzzU4[%+uT{C ? JJ JJ JJ   DJJ ;;rc t|t||t|t|t j dd|||||y)zChange view settings :param x,y: position :param height: :param persp: perpective :param twist: rz:Nviz::SetView(): x=%f, y=%f, height=%f, persp=%f, twist=%fN)Nviz_set_viewpoint_heightNviz_set_viewpoint_positionNviz_set_viewpoint_twistNviz_set_viewpoint_persprr)r8xyr6persptwists rSetViewz Nviz.SetViewsG "&)#Aq) ' ' H    rct}t}t}tt|tt|t||j|j|jfSN)rZNviz_get_viewpoint_heightr\Nviz_get_viewpoint_positionr!)r8rgrhhs rGetViewpointPositionzNviz.GetViewpointPositionsN J J J!%(+#E!HeAh7!''**rctt||zt||ztjdd||z||zy)zALook here feature :param x,y: screen coordinates rzNviz::LookHere(): x=%f, y=%fN)Nviz_look_hererOrr)r8rgrhrQs rLookHerez Nviz.LookHeres9 s1u9~s1u9~6 !3QYE JrcPttdtjddy)z*Center view at center of displayed surfacer'rzNviz::LookAtCenter()N)Nviz_set_focus_mapMAP_OBJ_UNDEFINEDrrr?s r LookAtCenterzNviz.LookAtCenters,b1 !+,rcBtjddt|jrtt }t }t }t |jt |t |t ||j|j|jfSy)z Get focusrzNviz::GetFocus())r'r'r')rrNviz_has_focusrc_floatNviz_get_focusr\r!r8rgrhzs rGetFocusz Nviz.GetFocussl !'( $)) $ A A A 499eAha%( C77AGGQWW, ,rc`tjddt|j|||y)z Set focusrzNviz::SetFocus()N)rrNviz_set_focusrr}s rSetFocusz Nviz.SetFocuss# !'(tyy!Q*rctjddtdz}tt ||d|d|dfS)z Get viewdirrzNviz::GetViewdir()rr()rrr{GS_get_viewdirr\)r8dirs r GetViewdirzNviz.GetViewdirsA !)*{ouSz"1vs1vs1v%%rctjdd|||fztdz}t|||fD] \}}|||< t t |y)z Set viewdirrz$Nviz::SetViewdir(): x=%f, y=%f, z=%fN)rrr{ enumerateGS_set_viewdirr\)r8rgrhr~ricoords r SetViewdirzNviz.SetViewdirsZ !;q!QiGH{o!1a), HAuCF uSz"rc\tjdd|t|j|S)zKSet z-exag value :param z_exag: value :return: 1 rzNviz::SetZExag(): z_exag=%f)rrrYr)r8r]s rSetZExagz Nviz.SetZExags&  !2F; 622rctjdd|t|jdd|rt |j|yt |jy)zDraw canvas Draw quick mode: - DRAW_QUICK_SURFACE - DRAW_QUICK_VLINES - DRAW_QUICK_VPOINTS - DRAW_QUICK_VOLUME :param quick: if true draw in wiremode :param quick_mode: quick mode rzNviz::Draw(): quick=%dr'N)rrNviz_draw_cplanerNviz_draw_quick Nviz_draw_all)r8quick quick_modes rDrawz Nviz.Draw"sB  !-u5B+  DIIz 2 $)) $rcltjddtt|jy)z)Erase map display (with background color)r(zNviz::EraseMap()N)rrGS_clearNviz_get_bgcolorrr?s rEraseMapz Nviz.EraseMap7s# !'(!$)),-rct|jtt|jt dt t |jt|jtjddy)zInitialize viewwhiter(zNviz::InitView()N) Nviz_init_datarNviz_set_surface_attr_defaultNviz_set_bgcolorNviz_color_from_strrrNviz_init_viewrrr?s rInitViewz Nviz.InitView<sY tyy! &'$7$@A!$)),-tyy! !'(rcBt|jt|y)zISet background color :param str color_str: color string N)rrr)r8 color_strs r SetBgColorzNviz.SetBgColorLs $7 $BCrc t|j|||||t|j||t|j|t |dt |dt |dt |j||y)a Change lighting settings :param x,y,z: position :param color: light color (as string) :param bright: light brightness :param ambient: light ambient :param w: local coordinate (default to 0) :param lid: light id rr(rN)Nviz_set_light_positionrNviz_set_light_brightNviz_set_light_colorrONviz_set_light_ambient) r8rgrhr~r4brightambientwlids rSetLightz Nviz.SetLightSsk  31a;diif5 IIsCaM3uQx=#eAh-  tyy#w7rc xt|d}|ttd|yttt ||d|j }|rat|d}|!ttd|t|yt|tttt ||d|j nd|r1t|tttdt||j n1t|tttt ||d|j ttdtj dd|||S) aLoad raster map (surface) :param name: raster map name :param color_name: raster map for color (None for color_value) :param color_value: color string (named color or RGB triptet) :return: object id :return: -1 on failure NzRaster map <%s> not foundr'r(z$Nviz::LoadRaster(): name=%s -> id=%d)G_find_raster2 G_warning_Nviz_new_map_obj MAP_OBJ_SURFG_fully_qualified_namerGS_delete_surface Nviz_set_attr ATT_COLORMAP_ATT CONST_ATTrrvrwrrr8name color_name color_valuemapsetids r LoadSurfacezNviz.LoadSurfaceds b) > a34d ; 0v>TYY  #J3F~!78*E!"% &z6:   #K0   &tV4   ,b1 !;T2F rc ttd||j}t|ttt dt ||jttdtjdd||S)zAdd new constant surfaceNr'r(zNviz::AddConstant(): id=%d) rrrrrrrrvrwrr)r8r!r4rs r AddConstantzNviz.AddConstants_ lD% C       & II  ,b1 !126 rcht|sytjdd|t|dkryy)zsUnload surface :param id: surface id :return: 1 on success :return: 0 on failure rr(zNviz::UnloadSurface(): id=%d)GS_surf_existsrrrr8rs r UnloadSurfacezNviz.UnloadSurfaces3b! !3R8 R 1 $rcd}tdk(rNttdd|j}t }t t |}t|dtdt|d}|ttd||r'ttt||d|j}n&ttt||d|j}tj dd ||||fS) zLoad vector map overlay :param name: vector map name :param points: if true load 2d points rather then 2d lines :return: object id, id of base surface (or -1 if it is not loaded) :return: -1 on failure r'rNrrzVector map <%s> not foundr(z$Nviz::LoadVector(): name=%s -> id=%d) GS_num_surfsrrrc_intGS_get_surf_listr\GS_set_att_const ATT_TRANSPG_find_vector2rr MAP_OBJ_SITEr MAP_OBJ_VECTrr)r8rpointsbaseIdnsurf surf_listrrs r LoadVectorzNviz.LoadVectors >Q %lD#tyyIFGE(u6I Yq\:s ;b) > a34d ; !4T6BCB"4T6BCB  !;T2F6zrctjdd||rt|syt|dkryyt |syt |dkryy)zUnload vector set :param id: vector set id :param points: vector points or lines set :return: 1 on success :return: 0 on failure r(zNviz::UnloadVector(): id=%dr)rrGP_site_existsGP_delete_siteGV_vect_existsGV_delete_vector)r8rrs r UnloadVectorzNviz.UnloadVectorsX  !2B7 !"%b!A% ""%#a'rcPt||}tjdd||||S)zCheck if surface is selected (currently unused) :param vid: vector id :param sid: surface id :return: True if selected :return: False if not selected r(z selected=%d)GV_surf_is_selectedrr)r8vidsidselecteds rVectorSurfaceSelectedzNviz.VectorSurfaceSelecteds2'sC0 J     rc Xt|d}|ttd|yttt ||d|j }|rat|d}|!ttd|t|yt|tttt ||d|j nd|r1t|tttdt||j n1t|tttt ||d|j tjdd|||S) aLoad 3d raster map (volume) :param name: 3d raster map name :param color_name: 3d raster map for color (None for color_value) :param color_value: color string (named color or RGB triptet) :return: object id :return: -1 on failure rNz3d raster map <%s> not foundr'rrr(z$Nviz::LoadVolume(): name=%s -> id=%d)G_find_raster3drrr MAP_OBJ_VOLrrGVL_delete_volrrrrrrrrs r LoadVolumezNviz.LoadVolumes!r* > a67 > /f=sDII  $Z4F~!:;ZHr" &z6:   #K0   &tV4    !;T2F rcht|sytjdd|t|dkryy)zqUnload volume :param id: volume id :return: 1 on success :return: 0 on failure rr(zNviz::UnloadVolume(): id=%d)GVL_vol_existsrrrrs r UnloadVolumezNviz.UnloadVolumeWs3b! !2B7 "  !rc2|j|t||S)aSet surface topography :param id: surface id :param map: if true use map otherwise constant :param value: map name of value :return: 1 on success :return: -1 surface not found :return: -2 setting attributes failed )SetSurfaceAttrATT_TOPOr8rmapr!s rSetSurfaceTopozNviz.SetSurfaceTopoi""2xe<>rc2|j|t||S)aSet surface shininess :param id: surface id :param map: if true use map otherwise constant :param value: map name of value :return: 1 on success :return: -1 surface not found :return: -2 setting attributes failed )r ATT_SHINErs rSetSurfaceShinezNviz.SetSurfaceShinerrc2|j|t||S)a!Set surface emission (currently unused) :param id: surface id :param map: if true use map otherwise constant :param value: map name of value :return: 1 on success :return: -1 surface not found :return: -2 setting attributes failed )rATT_EMITrs rSetSurfaceEmitzNviz.SetSurfaceEmitrrc .t|sy|r$t|t|t|d|j}nC|t k(r t |}n t|}t|t|td||j}tjdd|||||dkryy) a3Set surface attribute :param id: surface id :param attr: attribute desc :param map: if true use map otherwise constant :param value: map name of value :return: 1 on success :return: -1 surface not found :return: -2 setting attributes failed r'rNrz8Nviz::SetSurfaceAttr(): id=%d, attr=%d, map=%d, value=%srr() rrrrrrrfloatrrr)r8rattrrr!retvals rrzNviz.SetSurfaceAttrsb! L$diiXCy )%0ElL$ 4diiXC F       7rc.|j|tS)zUnset surface mask :param id: surface id :return: 1 on success :return: -1 surface not found :return: -2 setting attributes failed :return: -1 on failure )UnsetSurfaceAttrrrs rUnsetSurfaceMaskzNviz.UnsetSurfaceMasks$$R22rc.|j|tS)zUnset surface transparency :param id: surface id :return: 1 on success :return: -1 surface not found :return: -2 setting attributes failed )r rrs rUnsetSurfaceTranspzNviz.UnsetSurfaceTransps$$R44rc.|j|tS)zUnset surface emission (currently unused) :param id: surface id :return: 1 on success :return: -1 surface not found :return: -2 setting attributes failed )r rrs rUnsetSurfaceEmitzNviz.UnsetSurfaceEmits$$R22rczt|sytjdd||t|t|}|dkryy)zUnset surface attribute :param id: surface id :param attr: attribute descriptor :return: 1 on success :return: -1 surface not found :return: -2 setting attributes failed r'rz(Nviz::UnsetSurfaceAttr(): id=%d, attr=%drrr()rrrNviz_unset_attrr)r8rrr s rr zNviz.UnsetSurfaceAttrs=b! !?TJb,5 7rctjdd||||dkDr t|syt|||||dkryyt ||||y)aSet surface resolution :param id: surface id :param fine: x/y fine resolution :param coarse: x/y coarse resolution :return: 1 on success :return: -1 surface not found :return: -2 setting attributes failed rz0Nviz::SetSurfaceRes(): id=%d, fine=%d, coarse=%drr'rr()rrrGS_set_drawresGS_setall_drawres)r8rfinecoarses r SetSurfaceReszNviz.SetSurfaceRess`  A2tV  6!"%b$ff=A dD&& 9rctjdd|||dkDrt|syt||dkryyt |dkryy)aSet draw style Draw styles: - DM_GOURAUD - DM_FLAT - DM_FRINGE - DM_WIRE - DM_COL_WIRE - DM_POLY - DM_WIRE_POLY - DM_GRID_WIRE - DM_GRID_SURF :param id: surface id (<= 0 for all) :param style: draw style :return: 1 on success :return: -1 surface not found :return: -2 setting attributes failed rz(Nviz::SetSurfaceStyle(): id=%d, style=%drr'rr()rrrGS_set_drawmodeGS_setall_drawmode)r8rstyles rSetSurfaceStylezNviz.SetSurfaceStyle6sQ*  !?UK 6!"%r5)A- e $q (rc2tjdd||t|}|dkDrt|syt ||yt }t t|}t|jD]}||}t ||t|d}y)aHSet color of wire .. todo:: all :param id: surface id (< 0 for all) :param color_str: color string (R:G:B) :return: 1 on success :return: -1 surface not found :return: -2 setting attributes failed :return: 1 on success :return: 0 on failure rz%Nviz::SetWireColor(): id=%d, color=%srr'Nr() rrrrGS_set_wire_colorrrr\ranger!G_free)r8rrr4nsurfsrrs r SetWireColorzNviz.SetWireColor[s  !rcDt|syt|t|S)zSet z mode (use z coordinate or not) :param id: volume id :param zMode: bool :return: -1 on failure :return: 0 when no 3d :return: 1 on success r')r GP_set_zmoderO)r8rzModes rSetVectorPointZModezNviz.SetVectorPointZModes b!BE ++rct|sy|t|}|dks||k7ryt|dkryt|}t||dz t|S)zAdd new isosurface :param id: volume id :param level: isosurface level (topography) :param isosurf_id: isosurface id :return: -1 on failure :return: 1 on success r'rr()rGVL_isosurf_num_isosurfsGVL_isosurf_addGVL_isosurf_set_att_constr)r8rlevel isosurf_idnum nisosurfss r AddIsosurfacezNviz.AddIsosurfacesab!  !*2.CQw*+ 2  "-R0 (Y]HeLLrc~t|sy|t|}|dks||k7ryt|dkryt|S)zAdd new slice :param id: volume id :param slice_id: slice id :return: -1 on failure :return: number of slices r'r)rGVL_slice_num_slices GVL_slice_add)r8rslice_idrqs rAddSlicez Nviz.AddSlicesJb!  &r*CQw(c/  q #B''rc^t|sy|t|kDryt||}|dkryy)zDelete isosurface :param id: volume id :param isosurf_id: isosurface id :return: 1 on success :return: -1 volume not found :return: -2 isosurface not found :return: -3 on failure r'rrr5r()rrlGVL_isosurf_del)r8rrpr s rDeleteIsosurfacezNviz.DeleteIsosurfaces9b! 04 4b*- 7rc^t|sy|t|kDryt||}|dkryy)zDelete slice :param id: volume id :param slice_id: slice id :return: 1 on success :return: -1 volume not found :return: -2 slice not found :return: -3 on failure r'rrr5r()rru GVL_slice_del)r8rrwr s r DeleteSlicezNviz.DeleteSlice(s9b! *2. .B) 7rc|t|sy|t|kDry|r t||}n t||}|dkryy)a1Move isosurface up/down in the list :param id: volume id :param isosurf_id: isosurface id :param up: if true move up otherwise down :return: 1 on success :return: -1 volume not found :return: -2 isosurface not found :return: -3 on failure r'rrr5r()rrlGVL_isosurf_move_upGVL_isosurf_move_down)r8rrpupr s rMoveIsosurfacezNviz.MoveIsosurface@sHb! 04 4 %b*5C'J7C 7rc|t|sy|t|kDry|r t||}n t||}|dkryy)a Move slice up/down in the list :param id: volume id :param slice_id: slice id :param up: if true move up otherwise down :return: 1 on success :return: -1 volume not found :return: -2 slice not found :return: -3 on failure r'rrr5r()rruGVL_slice_move_upGVL_slice_move_down)r8rrwrr s r MoveSlicezNviz.MoveSlice\sHb! *2. . #B1C%b(3C 7rc4|j||t||S)abSet isosurface level :param id: volume id :param isosurf_id: isosurface id (0 - MAX_ISOSURFS) :param map: if true use map otherwise constant :param value: map name of value :return: 1 on success :return: -1 volume not found :return: -2 isosurface not found :return: -3 on failure )SetIsosurfaceAttrrr8rrprr!s rSetIsosurfaceTopozNviz.SetIsosurfaceTopox%%b*hUKKrc4|j||t||S)abSet isosurface color :param id: volume id :param isosurf_id: isosurface id (0 - MAX_ISOSURFS) :param map: if true use map otherwise constant :param value: map name of value :return: 1 on success :return: -1 volume not found :return: -2 isosurface not found :return: -3 on failure )rrrs rSetIsosurfaceColorzNviz.SetIsosurfaceColor%%b*ieLLrc4|j||td|S)aSet isosurface mask .. todo:: invert :param id: volume id :param isosurf_id: isosurface id (0 - MAX_ISOSURFS) :param invert: true for invert mask :param value: map name to be used for mask :return: 1 on success :return: -1 volume not found :return: -2 isosurface not found :return: -3 on failure T)rr)r8rrprr!s rSetIsosurfaceMaskzNviz.SetIsosurfaceMasks %%b*heLLrc4|j||t||S)aiSet isosurface transparency :param id: volume id :param isosurf_id: isosurface id (0 - MAX_ISOSURFS) :param map: if true use map otherwise constant :param value: map name of value :return: 1 on success :return: -1 volume not found :return: -2 isosurface not found :return: -3 on failure )rrrs rSetIsosurfaceTranspzNviz.SetIsosurfaceTransps%%b*j#uMMrc4|j||t||S)afSet isosurface shininess :param id: volume id :param isosurf_id: isosurface id (0 - MAX_ISOSURFS) :param map: if true use map otherwise constant :param value: map name of value :return: 1 on success :return: -1 volume not found :return: -2 isosurface not found :return: -3 on failure )rrrs rSetIsosurfaceShinezNviz.SetIsosurfaceShinerrc4|j||t||S)axSet isosurface emission (currently unused) :param id: volume id :param isosurf_id: isosurface id (0 - MAX_ISOSURFS) :param map: if true use map otherwise constant :param value: map name of value :return: 1 on success :return: -1 volume not found :return: -2 isosurface not found :return: -3 on failure )rrrs rSetIsosurfaceEmitzNviz.SetIsosurfaceEmitrrc t|sy|t|dz kDry|rt||||}n.|tk(r t |}n t |}t ||||}tjdd||||||dkryy)aSet isosurface attribute :param id: volume id :param isosurf_id: isosurface id (0 - MAX_ISOSURFS) :param attr: attribute desc :param map: if true use map otherwise constant :param value: map name of value :return: 1 on success :return: -1 volume not found :return: -2 isosurface not found :return: -3 setting attributes failed r'r(rrzGNviz::SetIsosurfaceAttr(): id=%d, isosurf=%d, attr=%d, map=%s, value=%sr) rrlGVL_isosurf_set_att_maprrrrnrr)r8rrprrr!r r s rrzNviz.SetIsosurfaceAttrsb! 04q8 8 )"j$FCy )%0El+B D#FC  (       7rc0|j||tS)aUnset isosurface mask :param id: volume id :param isosurf_id: isosurface id (0 - MAX_ISOSURFS) :return: 1 on success :return: -1 volume not found :return: -2 isosurface not found :return: -3 setting attributes failed )UnsetIsosurfaceAttrrr8rrps rUnsetIsosurfaceMaskzNviz.UnsetIsosurfaceMask''JAArc0|j||tS)aUnset isosurface transparency :param id: volume id :param isosurf_id: isosurface id (0 - MAX_ISOSURFS) :return: 1 on success :return: -1 volume not found :return: -2 isosurface not found :return: -3 setting attributes failed )rrrs rUnsetIsosurfaceTranspzNviz.UnsetIsosurfaceTransps''J CCrc0|j||tS)a*Unset isosurface emission (currently unused) :param id: volume id :param isosurf_id: isosurface id (0 - MAX_ISOSURFS) :return: 1 on success :return: -1 volume not found :return: -2 isosurface not found :return: -3 setting attributes failed )rrrs rUnsetIsosurfaceEmitzNviz.UnsetIsosurfaceEmitrrct|sy|t|dz kDrytjdd|||t |||}|dkryy)a1Unset surface attribute :param id: surface id :param isosurf_id: isosurface id (0 - MAX_ISOSURFS) :param attr: attribute descriptor :return: 1 on success :return: -1 volume not found :return: -2 isosurface not found :return: -2 on failure r'r(rrz7Nviz::UnsetSurfaceAttr(): id=%d, isosurf_id=%d, attr=%dr)rrlrrGVL_isosurf_unset_att)r8rrprr s rrzNviz.UnsetIsosurfaceAttr*s]b! 04q8 8 E     $B D9 7rc@t|syt||}|dkryy)zSet draw mode for isosurfaces :param id: isosurface id :param mode: isosurface draw mode :return: 1 on success :return: -1 volume set not found :return: -2 on failure r'rrr()rGVL_isosurf_set_drawmoder8rmoder s rSetIsosurfaceModezNviz.SetIsosurfaceModeKs(b!&r40 7rc@t|syt||}|dkryy)zSet draw mode for slices :param id: slice id :param mode: slice draw mode :return: 1 on success :return: -1 volume set not found :return: -2 on failure r'rrr()rGVL_slice_set_drawmoders r SetSliceModezNviz.SetSliceMode_s(b!$R. 7rcDt|syt||||}|dkryy)zSet draw resolution for isosurfaces :param id: isosurface id :param res: resolution value :return: 1 on success :return: -1 volume set not found :return: -2 on failure r'rrr()rGVL_isosurf_set_drawresr8rresr s rSetIsosurfaceReszNviz.SetIsosurfaceResss,b!%b#sC8 7rcDt|syt||||}|dkryy)zSet draw resolution for slices :param id: slice id :param res: resolution value :return: 1 on success :return: -1 volume set not found :return: -2 on failure r'rrr()rGVL_slice_set_drawresrs r SetSliceReszNviz.SetSliceRess,b!#BS#6 7rc lt|sy|t|kDryt||||||||| } | dkryy)a/Set slice position :param id: volume id :param slice_id: slice id :param x1,x2,y1,y2,z1,z2: slice coordinates :param dir: axis :return: 1 on success :return: -1 volume not found :return: -2 slice not found :return: -3 on failure r'rrr()rruGVL_slice_set_pos) r8rrwx1x2y1y2z1z2rr s rSetSlicePositionzNviz.SetSlicePositionsGb! *2. .Hb"b"b"cJ 7rc`t|sy|t|kDryt|||}|dkryy)aSet slice transparency :param id: volume id :param slice_id: slice id :param value: transparency value (0 - 255) :return: 1 on success :return: -1 volume not found :return: -2 slice not found :return: -3 on failure r'rrr()rruGVL_slice_set_transp)r8rrwr!r s rSetSliceTranspzNviz.SetSliceTransps;b! *2. ."2x7 7rcft|sy|t|dz kDryt|||}|dkryy)aSet inout mode :param id: volume id :param isosurf_id: isosurface id :param inout: mode true/false :return: 1 on success :return: -1 volume set not found :return: -2 isosurface not found :return: -3 on failure r'r(rrr5)rrlGVL_isosurf_set_flags)r8rrpinoutr s rSetIsosurfaceInOutzNviz.SetIsosurfaceInOuts@b! 04q8 8#B E: 7rc`t|sgSttt}}}t|t|t|t|t j dd||j |j |j |j |j |j gS)zGet volume position :param id: volume id :return: x,y,z :return: zero-length vector on error rz2Nviz::GetVolumePosition(): id=%d, x=%f, y=%f, z=%f)rr{ GVL_get_transr\rrr!r's rGetVolumePositionzNviz.GetVolumePositionsb!I)WY a1b%(E!HeAh7 @  GG GG GG  !''**rclt|sytjdd||||t||||y)zSet volume position :param id: volume id :param x,y,z: translation values :return: 1 on success :return: -1 volume not found :return: -2 setting position failed r'rz2Nviz::SetVolumePosition(): id=%d, x=%f, y=%f, z=%fr()rrr GVL_set_transr's rSetVolumePositionzNviz.SetVolumePositions9b! !I2qRSUVWb!Q"rcvt|sytjdd||t|t |y)zDisplay volume wire box :param id: volume id :param ifBox: True to draw wire box, False otherwise :type ifBox: bool :return: 1 on success :return: -1 volume not found r'rz)Nviz::SetVolumeDrawBox(): id=%d, ifBox=%dr()rrrGVL_set_draw_wirerO)r8rifBoxs rSetVolumeDrawBoxzNviz.SetVolumeDrawBoxs3b! !@"eL"c%j)rc,t|jSrm)Nviz_get_current_cplanerr?s rGetCPlaneCurrentzNviz.GetCPlaneCurrent&s&tyy11rc,t|jS)z Returns number of cutting planes)Nviz_num_cplanesrr?s rGetCPlanesCountzNviz.GetCPlanesCount)rVrc ttt}}}t|j}t|j|t |t |t ||j |j |j fS)z4Returns rotation parameters of current cutting plane)r{rrNviz_get_cplane_rotationr\r!r8rgrhr~currents rGetCPlaneRotationzNviz.GetCPlaneRotation-s\)WY a1)$))4 GU1XuQxqRww((rc ttt}}}t|j}t|j|t |t |t ||j |j |j fS)z7Returns translation parameters of current cutting plane)r{rrNviz_get_cplane_translationr\r!rs rGetCPlaneTranslationzNviz.GetCPlaneTranslation6s])WY a1)$))4#DIIwa%(ERSHUww((rct|j}t|j||||t|jddy)zSSet current clip plane rotation :param x,y,z: rotation parameters r'N)rrNviz_set_cplane_rotationrrs rSetCPlaneRotationzNviz.SetCPlaneRotation?s6 *$))4 GQ1=B+rct|j}t|j||||t|jddt j dd||||y)zYSet current clip plane translation :param x,y,z: translation parameters r'rz5Nviz::SetCPlaneTranslation(): id=%d, x=%f, y=%f, z=%fN)rrNviz_set_cplane_translationrrrrs rSetCPlaneTranslationzNviz.SetCPlaneTranslationHsS *$))4#DIIw1a@B+ FQRTUWX rct|j}t|j|||}|r0t|jdd|j \}}}|||fSy)Nr')NNN)rrNviz_set_cplane_hererr)r8rgrhrr r~s rSetCPlaneInteractivelyzNviz.SetCPlaneInteractivelyTsY)$))4"499gq!<  TYYB ///1GAq!a7N#rc0t|j|y)zKSelect cutting plane :param index: index of cutting plane N)Nviz_on_cplanerr8indexs r SelectCPlanezNviz.SelectCPlane^s tyy%(rc0t|j|y)zMUnselect cutting plane :param index: index of cutting plane N)Nviz_off_cplanerrs rUnselectCPlanezNviz.UnselectCPlanees  5)rc0t|j|y)z^Select current cutting plane :param index: type of fence - from 0 (off) to 4 N)Nviz_set_fence_colorrrs r SetFenceColorzNviz.SetFenceColorls TYY.rc,t|jS)z Get xy range)Nviz_get_xyrangerr?s r GetXYRangezNviz.GetXYRangesrVrctt}}t|jt|t||j|jfS)z Get z range)r{Nviz_get_zrangerr\r!)r8minmaxs r GetZRangezNviz.GetZRangews:9giS 5:uSz:yy#))##rc|j}|j}|j||tt |j |j dd|dk(r t|n t||j||y)zSave current GL screen to ppm/tif file :param filename: file name :param width: image width :param height: image height :param itype: image type ('ppm' or 'tif') Fr'ppmN) r5r6rRrrrr GS_write_ppm GS_write_tif)r8filenamer5r6itype widthOrig heightOrigs r SaveToFilezNviz.SaveToFile}smJJ [[  %(!$)),- % E>  "  " )Z0rcH|jrt|jyy)zDraw lighting modelN)r7Nviz_draw_modelrr?s rDrawLightingModelzNviz.DrawLightingModels >> DII & rc.t|jy)z Draw fringeN)Nviz_draw_fringerr?s r DrawFringezNviz.DrawFringes#rc t|ddzt|dzdzt|dz}t|j|t||t |t |t |t |y)zSet fringe :param sid: surface id :param color: color :param elev: elevation (height) :param nw,ne,sw,se: fringe edges (turn on/off) r:r(rN)strNviz_set_fringerrrO) r8rr4elevnwneswsescolors r SetFringezNviz.SetFringesnU1X$s58}4s:Sq]J II   '  G G G G rc,t|jS)zDraw north arrow)Nviz_draw_arrowrr?s r DrawArrowzNviz.DrawArrowstyy))rc Ft|j|||t|S)zSet north arrow from canvas coordinates :param sx,sy: canvas coordinates :param size: arrow length :param color: arrow color )Nviz_set_arrowrr)r8sxsyrBr4s rSetArrowz Nviz.SetArrows!diiR7J57QRRrc.t|jy)zDelete north arrowN)Nviz_delete_arrowrr?s r DeleteArrowzNviz.DeleteArrows$))$rc Ht|j||||t|S)zSet scale bar from canvas coordinates :param sx,sy: canvas coordinates :param id: scale bar id :param size: scale bar length :param color: scale bar color )Nviz_set_scalebarrr)r8rrrrBr4s r SetScalebarzNviz.SetScalebars(! IIr2r4))rZGS_get_rotation_matrixr\rC)r8r returnMatrixrKs rGetRotationMatrixzNviz.GetRotationMatrix2sBR-"uV}-  &D    % &rctdz}tt|D] }||||< tt |y)zSet rotation matrixr>N)rZr!lenGS_set_rotation_matrixr\)r8rmtrxrs rSetRotationMatrixzNviz.SetRotationMatrix;s?2  s6{# AQiDG uT{+rcDt|j|jyrm) Nviz_set_2Dr5r6r?s rStart2Dz Nviz.Start2DBsDJJ ,rctdz}t|D] \}}|||< tdz}t|d|d<t|d|d<t |j |||y)zFly through the scene :param flyInfo: fly parameters :param mode: 0 or 1 for different fly behaviour :param exagInfo: parameters changing fly speed rrmoverturnr(N)r{rrrONviz_flythroughr)r8flyInforexagInfoflyrrKexags r FlyThroughzNviz.FlyThroughEss{o ) GAtCF  }hv&'Qhv&'Q 3d3r)r(rA)NFNNN)NFNrm)ror)FFFFT)z__name__ __module__ __qualname__r;r@r.rRrUrarkrqrtrxrrrrrrrrrrrrrrrrrrrrrrrrrr rrr rrr$r(r+r/r3r9r=rDrGrJrMrSrYr\r_rcrfrjrsrxr{r~rrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrr r#r&r)r5r<rPrSrVrYr]rbrernrrr$r$os%6 ; +<2 ,+K- + &#3%*. ) D8"CJ$$ D0&;z$ = >> ? > =%N 3 5 3,4#J"H+2&@$,,"H$, :/6  OF>BA2$$, ,M4(,0088 L MM$ N M L,\ B D BB((((442+2&$2+)),  $)*/+$ 1*' $ (*S%  -,6( .0*,- 4rr$cTeZdZdZdZdZdZdZdZdZ dZ d Z d Z dd Z d Zy )Texturez!Class representing OpenGL texturecx||_t||_|jjx|_|_|jj x|_|_||_ ||_ d|_ |jjs|jj|jt}t!t#||j$|_|j&|jks|j&|jkrd|_y|j+|_y)zLoad image to texture :param filepath: path to image file :param overlayId: id of overlay (1 for legend, 101 and more for text) :param coords: image coordinates TN)pathr imageGetWidthr5 orig_width GetHeightr6 orig_heightrcoordsactiveHasAlpha InitAlphaResizerNviz_get_max_texturer\r!maxSize textureIdLoad)r8filepath overlayIdr~rs rr;zTexture.__init__Xs *84 '+zz':':'<< T_)-)=)=)?? d&  zz""$ JJ " 'U7^,}} <<$** $ t{{(B!DN!YY[DNrc|jrt|jtj|jy)zDelete textureN)rNviz_del_texturegrass try_removerxr?s rr@zTexture.__del__xs' >> T^^ , #rcdx}}|jtd|kDr|dz }|jtd|kDr|jtd|kDr|dz }|jtd|kDr|jj td|td|fd|jj |_|jj |_y)zResize image to match 2^nr(rr@)rBposN)r5powr6ryrrzr|)r8nrIs rrzTexture.Resize~s Ajj3q!9$ FAjj3q!9$kkC1I% FAkkC1I% Aq 3q!956BZZ((* jj**, rc|jjrd}nd}||jz|jz}|jdz }t |z}d|jz|jz}|j|jz}t j t|dz|jj}|jjr{tr;t j t|dz|jj}n:t j t|dz|jj}tjt|jD]} t|jD]} || || z |jzzdzdz|| | |jzz|zdz<|| || z |jzzdzdz|| | |jzz|zdz<|| || z |jzzdzdz|| | |jzz|zdz<|jjs| || z |jzz|| | |jzz|zdz<tjt!||j|j|jj} | S)zLoad image to texturer?rr(Brr)ryrr5r6c_ubytestructunpackr GetDatar GetAlpha GetAlphaDatawxBeginBusyCursorr! EndBusyCursorNviz_load_image) r8 bytesPerPixelbytesrev_valimbytes3bytes1 imageData alphaDatarjrs rrz Texture.Loads ::   MM *T[[8++/o TZZ$++-dkk)MM#f+"3TZZ5G5G5IJ ::   "MM#f+*;TZZ=P=P=RS "MM#f+*;TZZ=T=T=VW  t{{# A4::& ?H'A+33q81<@ADJJ&-7!;<@I'A+33q81<@ADJJ&-7!;<@I'A+33q81<@ADJJ&-7!;<::&&(CLgkTZZ77DBA N*m;a?@    RT[[$**:M:M:O P rct|jd|jd|j|j|jy)zDraw texture as an imagerr(N)Nviz_draw_imager~r5r6rr?s rrz Texture.Draws1 KKNDKKNDJJ T^^ rct|jd|jd|j|j}|j |||j ||S)Nrr()r r~r{r}Inflate ContainsXY)r8rgrhradiuscopys rHitTestzTexture.HitTestsKDKKNDKKNDOOTEUEUV VV$q!$$rc`|jdxx|z cc<|jdxx|z cc<y)zMove texture on the screenrr(N)r~)r8rErFs r MoveTexturezTexture.MoveTextures& A" A"rc||d|jdz }|d|jdz }|j||y)zSet coordinatesrr(N)r~r)r8r~rErFs r SetCoordszTexture.SetCoordss> AYQ ' AYQ ' R rc|jS)zReturns image id.)rr?s rGetIdz Texture.GetIds wwrc||_yrmr)r8rs r SetActivezTexture.SetActives  rc|jSrmrr?s rIsActivezTexture.IsActives {{rNrp)rqrrrs__doc__r;r@rrrrrrrrrrtrrrvrvUs>+)@$ -'R %  ! rrvc"eZdZdZdZdZdZy) ImageTexturez5Class representing OpenGL texture as an overlay imagecDtj||||||_y)zLoad image to texture :param filepath: path to image file :param overlayId: id of overlay (1 for legend) :param coords: image coordinates :param cmd: d.legend command )rrr~N)rvr;cmd)r8rrr~rs rr;zImageTexture.__init__s" IfUrc|jS)zReturns overlay command.)rr?s rGetCmdzImageTexture.GetCmds xxrcht|jt|jk(Srm)sortedr)r8rKs r CorrespondszImageTexture.Correspondss#dkkm$t{{}(===rN)rqrrrsrr;rrrtrrrrs? >rr);r __future__rrrErmathrnumpyr ImportErrorrrrstderrrctypesKeyErroreformat WindowsError NameErrorOSError grass.lib.gisgrass.lib.raster3dgrass.lib.vectorgrass.lib.ogsfgrass.lib.nvizgrass.lib.raster TypeError core.debugr core.utilsr core.gcmdr core.globalvarr gui_core.wrapr grass.scriptscriptrrrrr" CFUNCTYPE UNCHECKEDrStringerrtyper+pertyper-objectr$rvrrtrrrs0&  0 66$"  ","*        %(&%8Gk"G %(%0Gn%G c46c4L?{f{|>7>cE0   C -# CJJ/06 /  ##**556 L \9-6 /  ##**556Z  sbCC8D-D;=/E4$C54C58D*=#D%%D*-D87D8;E1#E,,E14E<;E<