Skip to content

Dicom SR

DICOM SR stuff

SRBuilder

This allows the creation of a DICOM SR file from scratch. Lots of dummy values are added to the produced DICOMSR in order to fully follow the DICOM standard while allowing the user to save data that may not be related to a treatment. The steps for the basic usage of this class are simply to create a DICOMSRBuilder object, add content sequences with the method add_content_sequence and build the DICOMSR with the method build (which applies metadata with the method _apply_metadata_to_sr).

Source code in pygrpm/dicom/sr/builder.py
 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
 59
 60
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
class SRBuilder:
    """
    This allows the creation of a DICOM SR file from scratch. Lots of dummy values are added
    to the produced DICOMSR in order to fully follow the DICOM standard while allowing the
    user to save data that may not be related to a treatment. The steps for the basic usage
    of this class are simply to create a DICOMSRBuilder object, add content sequences with the
    method add_content_sequence and build the DICOMSR with the method build (which applies metadata
    with the method _apply_metadata_to_sr).
    """

    def __init__(
            self,
            dicom_to_refer: Union[str, pydicom.FileDataset,
                                  List[Union[str, pydicom.FileDataset]]] = None,
            purpose_of_reference_code: Union[Dict, List[Dict]] = None,
            completion_flag: bool = True
    ):
        """
        :param dicom_to_refer: File path or pydicom.FileDataset object or list of those objects of
         the source DICOM file to use to assemble the DICOMSR
         object Dicom files directly are also accepted
        :param purpose_of_reference_code: Dictionary or list of directory with information tags on
         why the target file is used to generate the DICOM SR. For example:
         {
            "CodeValue": "SEG",
            "CodeMeaning": "Segmentation",
            "CodingSchemeDesignator": "DCM",
         }
        completion_flag: Boolean indicating if the dicom SR contains all relevant information
        about its content, as requested by the DICOM standard. The completion flag should
        always be COMPLETE, but this parameter allows users to choose UNCOMPLETE if
        the SR isn't exhaustive.
        """
        if dicom_to_refer is not None:
            # Ensure that the dicom_to_refer is a list, and that the purpose_of_reference_code is
            # also a list of the same length
            if isinstance(dicom_to_refer, list) and isinstance(purpose_of_reference_code, list):
                assert len(dicom_to_refer) == len(purpose_of_reference_code), \
                    "dicom_to_refer and purpose_of_reference_code don't have the same length"

                self.dicom_to_refer = [read_dicom(d) for d in dicom_to_refer]
                self.purpose_of_reference_code = purpose_of_reference_code

            elif not isinstance(dicom_to_refer, list) and \
                    not isinstance(purpose_of_reference_code, list):
                self.dicom_to_refer = [read_dicom(dicom_to_refer)]
                self.purpose_of_reference_code = [purpose_of_reference_code]

            else:
                raise TypeError(
                    "dicom_to_refer and purpose_of_reference_code both have to be a list "
                    "or neither of them should be a list."
                )

        else:
            self.dicom_to_refer = None
            self.purpose_of_reference_code = None

        self.dicom_sr = None
        self.content_sequence = None
        self.completion_flag = completion_flag

    def add_content_sequence(self, content_sequence) -> None:
        """
        This method allow the user to add a content sequence before building the DICOMSR.

        It contains:

        ***Required Elements***
        - A single Concept Name Code Sequence (0040,A043)
        - A Value
        - The ValueType (text, codes, etc.)

        ***Optionnal Elements***
        - References to images, waveforms or other composite objects
        - Relationships to other Items (by nested content sequences or by reference)


        Format Example:
        {
            "ValueType": "TEXT",
            "Value": "This is some text",
            "ConceptNameCodeSequence": {
                "CodeValue": "1233",
                "CodeMeaning": "Test",
                "CodingSchemeDesignator": "DCM",
            },
            "SomeOtherTag" : "Some additionnal information"
        }

        :param content_sequence: List containing the content_sequence in the appropriate format.
        """
        if self._is_content_sequence_valid(content_sequence):
            self.content_sequence = content_sequence

    def build(self) -> None:
        """
        This method builds the actual DICOMSR. In here, all the information
        related to the DICOM file format is set before actually adding the content
        of the DICOM using _apply_metadata_to_sr().
        """
        if self.content_sequence is None:
            raise ValueError('Empty content sequence')

        # Inspired by https://pydicom.github.io/pydicom/stable/auto_examples/
        # input_output/plot_write_dicom.html
        basic_text_sr_class_uid = "1.2.840.10008.5.1.4.1.1.88.11"
        # Radiation therapy dose storage class
        instance_uid = str(generate_uid())

        self.dicom_sr = pydicom.FileDataset(
            None, {}, file_meta=None, preamble=b"\0" * 128
        )

        # Set the transfer syntax
        self.dicom_sr.is_little_endian = True
        self.dicom_sr.is_implicit_VR = True

        self.dicom_sr.SOPClassUID = basic_text_sr_class_uid
        self.dicom_sr.SOPInstanceUID = instance_uid

        self._apply_metadata_to_sr()

        # Apply file meta
        file_meta = FileMetaDataset()
        file_meta.FileMetaInformationVersion = b"\x00\x01"
        file_meta.MediaStorageSOPClassUID = basic_text_sr_class_uid
        file_meta.MediaStorageSOPInstanceUID = instance_uid
        file_meta.ImplementationClassUID = PYDICOM_IMPLEMENTATION_UID
        file_meta.TransferSyntaxUID = "1.2.840.10008.1.2"

        file_meta.is_little_endian = True
        file_meta.is_implicit_VR = False

        pydicom.dataset.validate_file_meta(file_meta, enforce_standard=True)

        # file_meta.FileMetaInformationGroupLength = 0
        # Write the File Meta Information Group elements
        # first write into a buffer to avoid seeking back, that can be
        # expansive and is not allowed if writing into a zip file
        buffer = DicomBytesIO()
        buffer.is_little_endian = True
        buffer.is_implicit_VR = False
        pydicom.filewriter.write_dataset(buffer, file_meta)

        # CODE FROM THE PYDICOM LIB:
        # If FileMetaInformationGroupLength is present it will be the first written
        #   element and we must update its value to the correct length.
        # Update the FileMetaInformationGroupLength value, which is the number
        #   of bytes from the end of the FileMetaInformationGroupLength element
        #   to the end of all the File Meta Information elements.
        # FileMetaInformationGroupLength has a VR of 'UL' and so has a value
        #   that is 4 bytes fixed. The total length of when encoded as
        #   Explicit VR must therefore be 12 bytes.
        file_meta.FileMetaInformationGroupLength = buffer.tell() - 12
        del buffer

        self.dicom_sr.file_meta = file_meta

    def _first_level_check_validity(self, content_sequence):
        """
        Submethod used for content_sequence validation
        when on tag's first level
        """
        if content_sequence["ValueType"] != "CONTAINER":
            _logger.warning("First level ValueType should be CONTAINER")
            raise ValueError
        if content_sequence["ContinuityOfContent"] not in [
            "SEPARATE",
            "CONTINUOUS",
        ]:
            _logger.warning(
                "First level ContinuityOfContent should be SEPARATE or CONTINUOUS"
            )
            raise ValueError
        if not isinstance(content_sequence["Value"], list):
            _logger.warning("%s Value should be a CONTAINER (list)", content_sequence)
            raise ValueError
        if "RelationshipType" in content_sequence.keys():
            _logger.warning("First level should not have a RelationshipType")
            raise ValueError

        for container in content_sequence["Value"]:
            if not self._is_content_sequence_valid(container, False):
                return False

        return True

    def _check_container_validity(self, content_sequence):
        """
        Submethod used for content_sequence validation
        when it matches the CONTAINER type
        """
        if content_sequence["ContinuityOfContent"] not in [
            "SEPARATE",
            "CONTINUOUS",
        ]:
            _logger.warning(
                "%s ContinuityOfContent must be SEPARATE or CONTINUOUS",
                content_sequence,
            )
            raise ValueError

        if not isinstance(content_sequence["Value"], list):
            _logger.warning("%s Value should be a CONTAINER (list)", content_sequence)
            raise ValueError

        for container in content_sequence["Value"]:
            if not self._is_content_sequence_valid(container, False):
                return False

        return True

    def _check_code_validity(self, content_sequence):
        """
        Submethod used for content_sequence validation
        when it matches the CODE type
        """
        if not self._is_code_sequence_valid(content_sequence["ConceptCodeSequence"]):
            return False

        if "Value" in content_sequence.keys():
            if not isinstance(content_sequence["Value"], list):
                _logger.warning(
                    "%s Value should be a CONTAINER or a Code (list)",
                    content_sequence,
                )
                raise ValueError

            for container in content_sequence["Value"]:
                if not self._is_content_sequence_valid(container, False):
                    return False

        return True

    @staticmethod
    def _is_code_sequence_valid(concept_name_code_sequence):
        """
        Submethod used for content_sequence validation
        Considers code_sequence requirements
        """
        try:
            if not isinstance(concept_name_code_sequence, dict):
                _logger.warning(
                    """\
                    %s should be a dict with keys: CodeValue,\
                    CodeMeaning and CodingSchemeDesignator
                    """,
                    concept_name_code_sequence,
                )
                raise ValueError
            if not isinstance(concept_name_code_sequence["CodeValue"], str):
                _logger.warning(
                    "%s CodeValue should be a int in str", concept_name_code_sequence
                )
                raise ValueError
            if not isinstance(concept_name_code_sequence["CodeMeaning"], str):
                _logger.warning(
                    "%s CodeMeaning should be a string", concept_name_code_sequence
                )
                raise ValueError

            if not isinstance(
                    concept_name_code_sequence["CodingSchemeDesignator"], str
            ):
                _logger.warning(
                    "%s CodingSchemeDesignator should be a string",
                    concept_name_code_sequence,
                )

                raise ValueError

            return True
        except (ValueError, KeyError, TypeError):
            return False

    def _is_content_sequence_valid(self, content_sequence, first_level=True):
        """
        General method for content_sequence's structure validation
        TODO - Currently untreated Value_Types:
        - NUM
        - PNAME
        - DATE
        - DATETIME
        - UIDREF
        - IMAGE
        - COMPOSITE
        - SCOORD
        - TCOORD
        """
        check_value = True

        try:
            if first_level:
                check_value = self._first_level_check_validity(content_sequence)

            else:
                if not isinstance(content_sequence["RelationshipType"], str):
                    _logger.warning(
                        "%s RelationshipType should be a string", content_sequence
                    )
                    raise ValueError

                if content_sequence["ValueType"] not in SR_TYPING:
                    _logger.warning(
                        "%s CodeValue should be in {SR_TYPING}", content_sequence
                    )
                    raise ValueError

                value_type = content_sequence["ValueType"]
                if value_type == "TEXT":
                    if not isinstance(content_sequence["Value"], str):
                        _logger.warning("%s Value should be a string", content_sequence)
                        raise ValueError

                elif value_type == "CONTAINER":
                    check_value = self._check_container_validity(content_sequence)

                elif value_type == "CODE":
                    check_value = self._check_code_validity(content_sequence)

            if not self._is_code_sequence_valid(
                    content_sequence["ConceptNameCodeSequence"]
            ):
                check_value = False

            return check_value

        except (KeyError, ValueError):
            return False

    def _apply_metadata_to_sr(self) -> None:
        """
        TODO: Add Referenced Request Sequence
        TODO: Add Performed ProcedureCode Sequence

        This method adds all the content to the DICOM. Here is a list of all the tags that
        were filled with default value:

        PatientID = "Dose%date%_%time" +
        PatientName = "Unknown^Unknown"
        PatientBirthDate = current_date
        PatientSex = "O"
        StudyDate = current_date
        StudyTime = current_time
        ReferringPhysicianName = "Unknown^Unknown"
        StudyID = ""
        StudyDescription = ''
        InstitutionName = ""

        SeriesDescription = ""
        SeriesDate = current_date
        SeriesTime = current_time
        OperatorsName = "Unknown^Unknown"
        SeriesNumber = 1

        InstanceCreationDate = current_date
        InstanceCreationTime = current_time

        Manufacturer = '"
        InstanceNumber = 1
        ContentDate = current_date
        ContentTime = current_time
        PhotometricInterpretation = 'MONOCHROME2'

        See https://dicom.innolitics.com/ciods/rt-dose/
        """
        current_date = datetime.now()

        # Patient Layer
        self.dicom_sr.PatientID = (
                "SR"
                + current_date.strftime("%Y%m%d")
                + "_"
                + current_date.strftime("%H%M%S.%f")
        )
        self.dicom_sr.PatientName = "Unknown^Unknown"
        self.dicom_sr.PatientBirthDate = current_date.strftime("%Y%m%d")
        self.dicom_sr.PatientSex = "O"

        # Study layer

        self.dicom_sr.StudyDate = current_date.strftime("%Y%m%d")
        self.dicom_sr.StudyTime = current_date.strftime("%H%M%S.%f")
        self.dicom_sr.AccessionNumber = ""
        self.dicom_sr.ReferringPhysicianName = "Unknown^Unknown"
        self.dicom_sr.StudyInstanceUID = generate_uid()
        self.dicom_sr.StudyID = ""
        self.dicom_sr.StudyDescription = ""
        self.dicom_sr.InstitutionName = ""

        # Series Layer

        self.dicom_sr.SeriesDescription = ""
        self.dicom_sr.SeriesDate = current_date.strftime("%Y%m%d")
        self.dicom_sr.SeriesTime = current_date.strftime(
            "%H%M%S.%f"
        )  # long format with micro seconds
        self.dicom_sr.Modality = "SR"
        self.dicom_sr.SeriesInstanceUID = generate_uid()
        self.dicom_sr.SeriesNumber = 1

        # Instance Layer

        self.dicom_sr.ContentDate = current_date.strftime("%Y%m%d")
        self.dicom_sr.ContentTime = current_date.strftime("%H%M%S.%f")
        self.dicom_sr.InstanceNumber = 1
        self.dicom_sr.InstanceCreationDate = current_date.strftime("%Y%m%d")
        self.dicom_sr.InstanceCreationTime = current_date.strftime(
            "%H%M%S.%f"
        )  # long format with micro seconds

        if self.dicom_to_refer is not None:
            self._make_referenced_instances()
            self._adapt_sr_to_existing_study()

        if self.completion_flag:
            self.dicom_sr.CompletionFlag = "COMPLETE"
        else:
            self.dicom_sr.CompletionFlag = "UNCOMPLETE"

        self.dicom_sr.VerificationFlag = "UNVERIFIED"
        self.dicom_sr.ValueType = self.content_sequence["ValueType"]
        self.dicom_sr.ContinuityOfContent = self.content_sequence["ContinuityOfContent"]
        self.dicom_sr.add(
            make_concept_name_code_sequence(
                self.content_sequence["ConceptNameCodeSequence"]["CodeValue"],
                self.content_sequence["ConceptNameCodeSequence"]["CodeMeaning"],
                self.content_sequence["ConceptNameCodeSequence"][
                    "CodingSchemeDesignator"
                ],
            )
        )
        for content in self.content_sequence["Value"]:
            try:
                self.dicom_sr.ContentSequence += [make_content_sequence(content)]

            except AttributeError:
                self.dicom_sr.add(
                    pydicom.DataElement(
                        (0x0040, 0xA730), "SQ", [make_content_sequence(content)]
                    )
                )

    def _make_referenced_instances(self):
        self.dicom_sr.add(
            make_referenced_instance_sequence(
                [dicom.SOPClassUID for dicom in self.dicom_to_refer],
                [dicom.SOPInstanceUID for dicom in self.dicom_to_refer],
                [purpose["CodeValue"] for purpose in self.purpose_of_reference_code],
                [purpose["CodeMeaning"] for purpose in self.purpose_of_reference_code],
                [purpose["CodingSchemeDesignator"] for purpose in self.purpose_of_reference_code],
            )
        )

    def _adapt_sr_to_existing_study(self):
        """
        Retrieve the Patient and Study level information from the
        first referenced instance (dicom_to_refer) and apply it to this SR.
        """
        dicom_to_refer = self.dicom_to_refer[0]

        # Patient Layer
        self.dicom_sr.PatientID = dicom_to_refer.PatientID
        self.dicom_sr.PatientName = dicom_to_refer.PatientName
        self.dicom_sr.PatientBirthDate = dicom_to_refer.PatientBirthDate
        self.dicom_sr.PatientSex = dicom_to_refer.PatientSex

        # Study layer
        self.dicom_sr.StudyDate = getattr(dicom_to_refer, "StudyDate", "")
        self.dicom_sr.StudyTime = getattr(dicom_to_refer, "StudyTime", "")
        self.dicom_sr.AccessionNumber = getattr(dicom_to_refer, "AccessionNumber", "")

        if hasattr(dicom_to_refer, 'InstitutionName'):
            self.dicom_sr.InstitutionName = dicom_to_refer.InstitutionName

        if hasattr(dicom_to_refer, 'InstitutionalDepartmentalName'):
            self.dicom_sr.InstitutionalDepartmentalName = \
                dicom_to_refer.InstitutionalDepartmentalName

        self.dicom_sr.ReferringPhysicianName = getattr(dicom_to_refer, "ReferringPhysicianName", "")
        self.dicom_sr.StudyInstanceUID = dicom_to_refer.StudyInstanceUID
        self.dicom_sr.StudyID = dicom_to_refer.StudyID

        if hasattr(dicom_to_refer, 'StudyDescription'):
            self.dicom_sr.StudyDescription = dicom_to_refer.StudyDescription

__init__(dicom_to_refer=None, purpose_of_reference_code=None, completion_flag=True)

:param dicom_to_refer: File path or pydicom.FileDataset object or list of those objects of the source DICOM file to use to assemble the DICOMSR object Dicom files directly are also accepted :param purpose_of_reference_code: Dictionary or list of directory with information tags on why the target file is used to generate the DICOM SR. For example: { "CodeValue": "SEG", "CodeMeaning": "Segmentation", "CodingSchemeDesignator": "DCM", } completion_flag: Boolean indicating if the dicom SR contains all relevant information about its content, as requested by the DICOM standard. The completion flag should always be COMPLETE, but this parameter allows users to choose UNCOMPLETE if the SR isn't exhaustive.

Source code in pygrpm/dicom/sr/builder.py
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
def __init__(
        self,
        dicom_to_refer: Union[str, pydicom.FileDataset,
                              List[Union[str, pydicom.FileDataset]]] = None,
        purpose_of_reference_code: Union[Dict, List[Dict]] = None,
        completion_flag: bool = True
):
    """
    :param dicom_to_refer: File path or pydicom.FileDataset object or list of those objects of
     the source DICOM file to use to assemble the DICOMSR
     object Dicom files directly are also accepted
    :param purpose_of_reference_code: Dictionary or list of directory with information tags on
     why the target file is used to generate the DICOM SR. For example:
     {
        "CodeValue": "SEG",
        "CodeMeaning": "Segmentation",
        "CodingSchemeDesignator": "DCM",
     }
    completion_flag: Boolean indicating if the dicom SR contains all relevant information
    about its content, as requested by the DICOM standard. The completion flag should
    always be COMPLETE, but this parameter allows users to choose UNCOMPLETE if
    the SR isn't exhaustive.
    """
    if dicom_to_refer is not None:
        # Ensure that the dicom_to_refer is a list, and that the purpose_of_reference_code is
        # also a list of the same length
        if isinstance(dicom_to_refer, list) and isinstance(purpose_of_reference_code, list):
            assert len(dicom_to_refer) == len(purpose_of_reference_code), \
                "dicom_to_refer and purpose_of_reference_code don't have the same length"

            self.dicom_to_refer = [read_dicom(d) for d in dicom_to_refer]
            self.purpose_of_reference_code = purpose_of_reference_code

        elif not isinstance(dicom_to_refer, list) and \
                not isinstance(purpose_of_reference_code, list):
            self.dicom_to_refer = [read_dicom(dicom_to_refer)]
            self.purpose_of_reference_code = [purpose_of_reference_code]

        else:
            raise TypeError(
                "dicom_to_refer and purpose_of_reference_code both have to be a list "
                "or neither of them should be a list."
            )

    else:
        self.dicom_to_refer = None
        self.purpose_of_reference_code = None

    self.dicom_sr = None
    self.content_sequence = None
    self.completion_flag = completion_flag

add_content_sequence(content_sequence)

This method allow the user to add a content sequence before building the DICOMSR.

It contains:

Required Elements - A single Concept Name Code Sequence (0040,A043) - A Value - The ValueType (text, codes, etc.)

Optionnal Elements - References to images, waveforms or other composite objects - Relationships to other Items (by nested content sequences or by reference)

Format Example: { "ValueType": "TEXT", "Value": "This is some text", "ConceptNameCodeSequence": { "CodeValue": "1233", "CodeMeaning": "Test", "CodingSchemeDesignator": "DCM", }, "SomeOtherTag" : "Some additionnal information" }

:param content_sequence: List containing the content_sequence in the appropriate format.

Source code in pygrpm/dicom/sr/builder.py
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
def add_content_sequence(self, content_sequence) -> None:
    """
    This method allow the user to add a content sequence before building the DICOMSR.

    It contains:

    ***Required Elements***
    - A single Concept Name Code Sequence (0040,A043)
    - A Value
    - The ValueType (text, codes, etc.)

    ***Optionnal Elements***
    - References to images, waveforms or other composite objects
    - Relationships to other Items (by nested content sequences or by reference)


    Format Example:
    {
        "ValueType": "TEXT",
        "Value": "This is some text",
        "ConceptNameCodeSequence": {
            "CodeValue": "1233",
            "CodeMeaning": "Test",
            "CodingSchemeDesignator": "DCM",
        },
        "SomeOtherTag" : "Some additionnal information"
    }

    :param content_sequence: List containing the content_sequence in the appropriate format.
    """
    if self._is_content_sequence_valid(content_sequence):
        self.content_sequence = content_sequence

build()

This method builds the actual DICOMSR. In here, all the information related to the DICOM file format is set before actually adding the content of the DICOM using _apply_metadata_to_sr().

Source code in pygrpm/dicom/sr/builder.py
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
def build(self) -> None:
    """
    This method builds the actual DICOMSR. In here, all the information
    related to the DICOM file format is set before actually adding the content
    of the DICOM using _apply_metadata_to_sr().
    """
    if self.content_sequence is None:
        raise ValueError('Empty content sequence')

    # Inspired by https://pydicom.github.io/pydicom/stable/auto_examples/
    # input_output/plot_write_dicom.html
    basic_text_sr_class_uid = "1.2.840.10008.5.1.4.1.1.88.11"
    # Radiation therapy dose storage class
    instance_uid = str(generate_uid())

    self.dicom_sr = pydicom.FileDataset(
        None, {}, file_meta=None, preamble=b"\0" * 128
    )

    # Set the transfer syntax
    self.dicom_sr.is_little_endian = True
    self.dicom_sr.is_implicit_VR = True

    self.dicom_sr.SOPClassUID = basic_text_sr_class_uid
    self.dicom_sr.SOPInstanceUID = instance_uid

    self._apply_metadata_to_sr()

    # Apply file meta
    file_meta = FileMetaDataset()
    file_meta.FileMetaInformationVersion = b"\x00\x01"
    file_meta.MediaStorageSOPClassUID = basic_text_sr_class_uid
    file_meta.MediaStorageSOPInstanceUID = instance_uid
    file_meta.ImplementationClassUID = PYDICOM_IMPLEMENTATION_UID
    file_meta.TransferSyntaxUID = "1.2.840.10008.1.2"

    file_meta.is_little_endian = True
    file_meta.is_implicit_VR = False

    pydicom.dataset.validate_file_meta(file_meta, enforce_standard=True)

    # file_meta.FileMetaInformationGroupLength = 0
    # Write the File Meta Information Group elements
    # first write into a buffer to avoid seeking back, that can be
    # expansive and is not allowed if writing into a zip file
    buffer = DicomBytesIO()
    buffer.is_little_endian = True
    buffer.is_implicit_VR = False
    pydicom.filewriter.write_dataset(buffer, file_meta)

    # CODE FROM THE PYDICOM LIB:
    # If FileMetaInformationGroupLength is present it will be the first written
    #   element and we must update its value to the correct length.
    # Update the FileMetaInformationGroupLength value, which is the number
    #   of bytes from the end of the FileMetaInformationGroupLength element
    #   to the end of all the File Meta Information elements.
    # FileMetaInformationGroupLength has a VR of 'UL' and so has a value
    #   that is 4 bytes fixed. The total length of when encoded as
    #   Explicit VR must therefore be 12 bytes.
    file_meta.FileMetaInformationGroupLength = buffer.tell() - 12
    del buffer

    self.dicom_sr.file_meta = file_meta

make_sr(content_sequence, dicom_to_refer=None, purpose_of_reference_code=None, completion_flag=True)

Creates a DICOM SR object

Args: content_sequence: Content sequence, ex. {'ValueType': ..., 'Value': 'some text', 'ConceptNameCodeSequence': { 'CodeValue': '1233', 'CodeMeaning': 'Test', 'CodingSchemeDesignator': 'DCM', }, 'SomeOtherTag' : 'Some additional information'} dicom_to_refer: File path or pydicom.FileDataset object or list of those objects of the source DICOM file to use to assemble the DICOMSR object purpose_of_reference_code: Dictionary or list of directory with information tags on why the target file is used to generate the DICOM SR. For example: { "CodeValue": "SEG", "CodeMeaning": "Segmentation", "CodingSchemeDesignator": "DCM", } completion_flag: Boolean indicating if the dicom SR contains all relevant information about its content, as requested by the DICOM standard. The completion flag should always be COMPLETE, but this parameter allows users to choose UNCOMPLETE if the SR isn't exhaustive.

Returns: A DICOM SR object (pydicom.FileDataset)

Source code in pygrpm/dicom/sr/sr.py
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
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
def make_sr(
        content_sequence: Dict,
        dicom_to_refer: Union[
            str, pydicom.FileDataset,
            List[Union[str, pydicom.FileDataset]]
        ] = None,
        purpose_of_reference_code: Union[Dict, List[Dict]] = None,
        completion_flag: bool = True) -> pydicom.FileDataset:
    """Creates a DICOM SR object

    Args:
        content_sequence: Content sequence, ex.
            {'ValueType': ...,
             'Value': 'some text', 'ConceptNameCodeSequence': {
                'CodeValue': '1233',
                'CodeMeaning': 'Test',
                'CodingSchemeDesignator': 'DCM',
            },
            'SomeOtherTag' : 'Some additional information'}
        dicom_to_refer: File path or pydicom.FileDataset object or list of those objects of
            the source DICOM file to use to assemble the DICOMSR object
        purpose_of_reference_code: Dictionary or list of directory with information tags on
            why the target file is used to generate the DICOM SR. For example:
            {
               "CodeValue": "SEG",
               "CodeMeaning": "Segmentation",
               "CodingSchemeDesignator": "DCM",
            }
        completion_flag: Boolean indicating if the dicom SR contains all relevant information
            about its content, as requested by the DICOM standard. The completion flag should
            always be COMPLETE, but this parameter allows users to choose UNCOMPLETE if
            the SR isn't exhaustive.

    Returns:
        A DICOM SR object (pydicom.FileDataset)
    """
    if dicom_to_refer is not None and purpose_of_reference_code is None:
        # If the dicom_to_refer is not None, the purpose_of_reference_code has to be provided.
        # If not provided, use the default

        # Ensure that dicom_to_refer is a list
        if not isinstance(dicom_to_refer, list):
            dicom_to_refer = [dicom_to_refer]

        # Ensure that dicom_to_refer are pydicom.FileDataset
        dicom_to_refer = [utils.read_dicom(dicom) for dicom in dicom_to_refer]

        purpose_of_reference_code = _make_default_purpose_of_reference_codes(dicom_to_refer)

    # Building DICOM SR file based on the study
    sr_builder = SRBuilder(
        dicom_to_refer=dicom_to_refer,
        purpose_of_reference_code=purpose_of_reference_code,
        completion_flag=completion_flag
    )

    # Adding content_sequence, populating DICOM SR file an extracting
    sr_builder.add_content_sequence(content_sequence)
    sr_builder.build()

    finalize_dicom_sr = sr_builder.dicom_sr

    # Returning the DiCOM SR File in DICOM format
    return finalize_dicom_sr

make_sr_from_text(text, dicom_to_refer=None, purpose_of_reference_code=None, completion_flag=True)

Create a DICOM SR file containing the text given in parameter.

Args: text: string or list of strings of the text to add dicom_to_refer: File path or pydicom.FileDataset object or list of those objects of the source DICOM file to use to assemble the DICOMSR object purpose_of_reference_code: Dictionary or list of directory with information tags on why the target file is used to generate the DICOM SR. For example: { "CodeValue": "SEG", "CodeMeaning": "Segmentation", "CodingSchemeDesignator": "DCM", } completion_flag: Boolean indicating if the dicom SR contains all relevant information about its content, as requested by the DICOM standard. The completion flag should always be COMPLETE, but this parameter allows users to choose UNCOMPLETE if the SR isn't exhaustive.

Returns: A DICOM SR object (pydicom.FileDataset)

Source code in pygrpm/dicom/sr/sr.py
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
def make_sr_from_text(
        text: Union[str, List[str]],
        dicom_to_refer: [
            str, pydicom.FileDataset,
            List[Union[str, pydicom.FileDataset]]
        ] = None,
        purpose_of_reference_code: Union[Dict, List[Dict]] = None,
        completion_flag: bool = True) -> pydicom.FileDataset:
    """Create a DICOM SR file containing the text given in parameter.

    Args:
        text: string or list of strings of the text to add
        dicom_to_refer: File path or pydicom.FileDataset object or list of those objects of
            the source DICOM file to use to assemble the DICOMSR object
        purpose_of_reference_code: Dictionary or list of directory with information tags on
            why the target file is used to generate the DICOM SR. For example:
            {
               "CodeValue": "SEG",
               "CodeMeaning": "Segmentation",
               "CodingSchemeDesignator": "DCM",
            }
        completion_flag: Boolean indicating if the dicom SR contains all relevant information
            about its content, as requested by the DICOM standard. The completion flag should
            always be COMPLETE, but this parameter allows users to choose UNCOMPLETE if
            the SR isn't exhaustive.

    Returns:
        A DICOM SR object (pydicom.FileDataset)
    """
    # If a string is received, make a list of one string

    if isinstance(text, str):
        text_list = []
        text_list.append(text)
        text = text_list
    # Create the value list of the content sequence
    value_list = []
    for text_item in text:
        value_list.append({
            "RelationshipType": "HAS PROPERTIES",
            "ValueType": "TEXT",
            "ConceptNameCodeSequence": {
                "CodeValue": "113012",
                "CodeMeaning": "Key Object Description",
                "CodingSchemeDesignator": "DCM",
            },
            "Value": text_item,
        })
    # Create the content sequence
    content_sequence = {
        "ValueType": "CONTAINER",
        "ConceptNameCodeSequence": {
            "CodeValue": "DOC",
            "CodeMeaning": "Document",
            "CodingSchemeDesignator": "DCM",
        },
        "ContinuityOfContent": "SEPARATE",
        "Value": value_list,
    }

    return make_sr(
        content_sequence=content_sequence,
        dicom_to_refer=dicom_to_refer,
        purpose_of_reference_code=purpose_of_reference_code,
        completion_flag=completion_flag
    )