nviz2/wxGUI: minor workspace-related fixes

git-svn-id: https://svn.osgeo.org/grass/grass/trunk@32324 15284696-431f-4ddb-bdfa-cd5b030d7da7

gui/wxpython/gui_modules/nviz.py

@@ -2554,7 +2554,11 @@ class NvizToolWindow(wx.Frame):
         self.UpdateVectorShow(vecType, checked)
         
         if checked:
-            id = data[vecType]['object']['id']
+            try:
+                id = data[vecType]['object']['id']
+            except KeyError:
+                return
+
             self.mapWindow.SetLayerData(item, id, vecType)
         
             # update properties

gui/wxpython/gui_modules/workspace.py

@@ -422,6 +422,9 @@ class WriteWorkspaceFile(object):
 
         @param data Nviz layer properties
         """
+        if not data.has_key('object'): # skip disabled
+            return
+
         self.indent += 4
         for attrb in data.iterkeys():
             if len(data[attrb]) < 1: # skip empty attributes
@@ -485,7 +488,7 @@ class WriteWorkspaceFile(object):
         self.indent -= 4
 
     def __writeNvizVector(self, data):
-        """Save Nviz vector layer properties to workspace
+        """Save Nviz vector layer properties (lines/points) to workspace
 
         @param data Nviz layer properties
         """
@@ -493,16 +496,22 @@ class WriteWorkspaceFile(object):
         for attrb in data.iterkeys():
             if len(data[attrb]) < 1: # skip empty attributes
                 continue
+
+            if not data[attrb].has_key('object'): # skip disabled
+                continue
+            
             self.file.write('%s<%s>\n' % (' ' * self.indent, attrb))
             self.indent += 4
             for name in data[attrb].iterkeys():
+                if name == 'object':
+                    continue
                 if name == 'mode':
                     self.file.write('%s<%s type="%s">\n' % (' ' * self.indent, name,
                                                           data[attrb][name]['type']))
                     if data[attrb][name]['type'] == 'surface':
                         self.indent += 4
                         self.file.write('%s<map>%s</map>\n' % (' ' * self.indent,
-                                                               data[attrb][name]['map']))
+                                                               data[attrb][name]['surface']))
                         self.indent -= 4
                     self.file.write('%s</%s>\n' % ((' ' * self.indent, name)))
                 else: