PyEQD2/pyeqd2.py at master · chris201706/PyEQD2 · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
'''
Summary:    Import/Export DICOM RT-DOSE, convert to EQD2, plot dose distribution and legend. Object-Oriented.
'''


### Preamble ###
from corefunctions import *
from plotting import *


### Class Definition ###
class RTDose:
    def __init__(self, filepath):
        self.dcmfile = import_dcm(filepath)
        self.name, self.ext = os.path.splitext(os.path.basename(filepath))
        if is_modality(self.dcmfile,'RTDOSE'):
            self.pixelarray, self.shape = read_dose(self.dcmfile)
            self.pixelspacing = self.dcmfile.PixelSpacing
            self.raw_maxval = np.amax(self.pixelarray)
            self.precision = self.dcmfile.BitsAllocated
            self.signedness = self.dcmfile.PixelRepresentation # Cf. [A] for PixelRepresentation: 0 is unsigned, 1 is signed
            print("--New instance of Class RTDose has been initiated.")
        else:
            raise ValueError("Input file is not of modality 'RTDOSE'.(Returning False.)")
    def make_eqd2(self, fn, abratio):
        self.pixelarray, self.dcmfile.DoseGridScaling = dcm_to_eqd2(self.pixelarray, self.dcmfile.DoseGridScaling, self.precision, self.signedness, self.raw_maxval, fn, abratio)
    def multiply(self, factor):
        self.pixelarray, self.dcmfile.DoseGridScaling = dcm_multiply(self.pixelarray, self.dcmfile.DoseGridScaling, self.precision, self.signedness, self.raw_maxval, factor)
    def make_physical(self): # required for RayStation 6
        original_dosetype = self.dcmfile.DoseType
        try:
            self.dcmfile.DoseType = 'PHYSICAL'
            print("--Successfully changed the DICOM file's DoseType from / to:\n %s\n %s" % (original_dosetype, self.dcmfile.DoseType))
        except:
            raise RuntimeError("An error occured while changing the DICOM file's DoseType.")
    def plot(self):
        rtdose_plot(self.pixelarray, self.name, newjet)
    def plot_legend(self):
        legend(newjet)
        pass
    def export(self):
        export_dcm(self.dcmfile, self.pixelarray, self.name) # simply exports to current directory (where plots and legends are also saved -- consistency!)

''' Test
dicompath = '/Users/macuser/Downloads/testfile.dcm'
a = RTDose(dicompath)
a.name
a.plot_legend()
a.plot()
a.make_eqd2(8,3)
a.multiply(8)
a.make_EQD2(1,3)
a.export()
\Test '''


### References ###
# [A] http://dicomiseasy.blogspot.de/2012/08/chapter-12-pixel-data.html