Update devices when updating volumes

eigenvivek · Feb 8, 2024 · 3019b25 · 3019b25
1 parent b142d27
commit 3019b25
Show file tree

Hide file tree

Showing 3 changed files with 42 additions and 0 deletions.
diff --git a/diffdrr/_modidx.py b/diffdrr/_modidx.py
@@ -116,6 +116,8 @@
                                                                              'diffdrr/renderers.py'),
                                    'diffdrr.renderers.Trilinear.dims': ('api/renderers.html#trilinear.dims', 'diffdrr/renderers.py'),
                                    'diffdrr.renderers.Trilinear.forward': ('api/renderers.html#trilinear.forward', 'diffdrr/renderers.py'),
+                                   'diffdrr.renderers.Trilinear.spacing': ('api/renderers.html#trilinear.spacing', 'diffdrr/renderers.py'),
+                                   'diffdrr.renderers.Trilinear.volume': ('api/renderers.html#trilinear.volume', 'diffdrr/renderers.py'),
                                    'diffdrr.renderers._get_alpha_minmax': ('api/renderers.html#_get_alpha_minmax', 'diffdrr/renderers.py'),
                                    'diffdrr.renderers._get_alphas': ('api/renderers.html#_get_alphas', 'diffdrr/renderers.py'),
                                    'diffdrr.renderers._get_index': ('api/renderers.html#_get_index', 'diffdrr/renderers.py'),

diff --git a/diffdrr/renderers.py b/diffdrr/renderers.py
@@ -23,6 +23,8 @@ def volume(self):
     @volume.setter
     def volume(self, volume):
         self._volume = volume
+        self.device = volume.device
+        self.dtype = volume.dtype
 
     @property
     def spacing(self):
@@ -162,6 +164,24 @@ def __init__(
         self.device = volume.device
         self.dtype = volume.dtype
 
+    @property
+    def volume(self):
+        return self._volume
+
+    @volume.setter
+    def volume(self, volume):
+        self._volume = volume
+        self.device = volume.device
+        self.dtype = volume.dtype
+
+    @property
+    def spacing(self):
+        return self._spacing
+
+    @spacing.setter
+    def spacing(self, spacing):
+        self._spacing = spacing
+
     @property
     def dims(self):
         return torch.tensor(self.volume.shape, device=self.device, dtype=self.dtype) + 1

diff --git a/notebooks/api/01_renderers.ipynb b/notebooks/api/01_renderers.ipynb
@@ -99,6 +99,8 @@
     "    @volume.setter\n",
     "    def volume(self, volume):\n",
     "        self._volume = volume\n",
+    "        self.device = volume.device\n",
+    "        self.dtype = volume.dtype\n",
     "\n",
     "    @property\n",
     "    def spacing(self):\n",
@@ -279,6 +281,24 @@
     "        self.dtype = volume.dtype\n",
     "\n",
     "    @property\n",
+    "    def volume(self):\n",
+    "        return self._volume\n",
+    "\n",
+    "    @volume.setter\n",
+    "    def volume(self, volume):\n",
+    "        self._volume = volume\n",
+    "        self.device = volume.device\n",
+    "        self.dtype = volume.dtype\n",
+    "\n",
+    "    @property\n",
+    "    def spacing(self):\n",
+    "        return self._spacing\n",
+    "\n",
+    "    @spacing.setter\n",
+    "    def spacing(self, spacing):\n",
+    "        self._spacing = spacing\n",
+    "\n",
+    "    @property\n",
     "    def dims(self):\n",
     "        return torch.tensor(self.volume.shape, device=self.device, dtype=self.dtype) + 1\n",
     "\n",