Unloading volume raycasting when closing a project

Thiago Franco de Moraes
1 parent 4e0ea000
Showing 1 changed file with 19 additions and 1 deletions Show diff stats
invesalius/data/volume.py
@@ -131,11 +131,29 @@ class Volume():
             self.plane.DestroyObjs()
             del self.plane
             self.plane = 0
-            
+
         if self.exist:
             self.exist = None
             Publisher.sendMessage('Remove surface actor from viewer', self.volume)
             Publisher.sendMessage('Disable volume cut menu')
+            Publisher.sendMessage('Unload volume', self.volume)
+
+            del self.image
+            del self.imagedata
+            del self.final_imagedata
+            del self.volume
+            del self.color_transfer
+            del self.opacity_transfer_func
+            del self.volume_properties
+            del self.volume_mapper
+            self.volume = None
+            self.exist = False
+            self.loaded_image = False
+            self.image = None
+            self.final_imagedata = None
+            self.opacity_transfer_func = None
+            self.color_transfer = None
+            Publisher.sendMessage('Render volume viewer')
  
     def OnLoadVolume(self, pubsub_evt):
         label = pubsub_evt.data
...	...	@@ -131,11 +131,29 @@ class Volume():
131	131	self.plane.DestroyObjs()
132	132	del self.plane
133	133	self.plane = 0
134		-
	134	+
135	135	if self.exist:
136	136	self.exist = None
137	137	Publisher.sendMessage('Remove surface actor from viewer', self.volume)
138	138	Publisher.sendMessage('Disable volume cut menu')
	139	+ Publisher.sendMessage('Unload volume', self.volume)
	140	+
	141	+ del self.image
	142	+ del self.imagedata
	143	+ del self.final_imagedata
	144	+ del self.volume
	145	+ del self.color_transfer
	146	+ del self.opacity_transfer_func
	147	+ del self.volume_properties
	148	+ del self.volume_mapper
	149	+ self.volume = None
	150	+ self.exist = False
	151	+ self.loaded_image = False
	152	+ self.image = None
	153	+ self.final_imagedata = None
	154	+ self.opacity_transfer_func = None
	155	+ self.color_transfer = None
	156	+ Publisher.sendMessage('Render volume viewer')
139	157
140	158	def OnLoadVolume(self, pubsub_evt):
141	159	label = pubsub_evt.data
...	...