added basic IO functions

bhklab · Jan 16, 2020 · dfb11dd · dfb11dd
1 parent 2afb431
commit dfb11dd
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/imgtools/__init__.py b/imgtools/__init__.py
diff --git a/imgtools/io.py b/imgtools/io.py
@@ -0,0 +1,18 @@
+import SimpleITK as sitk
+from pydicom import dcmread
+
+
+def read_dicom_series(path):
+    reader = sitk.ImageSeriesReader()
+    dicom_names = reader.GetGDCMSeriesFileNames(path)
+    reader.SetFileNames(dicom_names)
+    return reader.Execute()
+
+
+def read_image(path):
+    return sitk.ReadImage(path)
+
+
+def read_dicom_rtstruct(path):
+    return pydicom.dcmread(path)
+