fixed evolution of raw pet data header so it handles unrecognised tags

Author: tclose

medimages4tests/dummy/dicom/base.py

@@ -109,6 +109,7 @@ def evolve_header(
         keyword arguments containing values to update in the header
     """
     hdr = deepcopy(dicom_header)
+    [getattr(hdr, a) for a in dir(hdr)]  # Ensure data dict keys are loaded
     if first_name or last_name:
         if not first_name or not last_name:
             try:
@@ -130,7 +131,11 @@ def evolve_header(
                 continue
             raise ValueError(f"Did not find tag corresponding to keyword {key}")
         hex_tag = format(tag_decimal, "08x").upper()
-        elem = hdr[hex_tag]["Value"]
+        tag = hdr[hex_tag]
+        try:
+            elem = tag["Value"]
+        except KeyError:
+            continue
         assert isinstance(elem, list) and len(elem) == 1
         nested_elem = elem[0]
         if isinstance(nested_elem, dict) and list(nested_elem.keys()) == ["Alphabetic"]:

medimages4tests/dummy/raw/pet/siemens/biograph_vision/vr20b/pet_calibration/header.py

@@ -4,8 +4,8 @@
 )
 
 
-def get_image_header(out_dir, **kwargs):
-    hdr = evolve_header(constant_hdr, skip_unknown=True, **kwargs)
+def get_image_header(out_dir, skip_unknown=True, **kwargs):
+    hdr = evolve_header(constant_hdr, skip_unknown=skip_unknown, **kwargs)
     return generate_dicom(out_dir, num_vols, hdr, collated_data, {})
 
 

medimages4tests/dummy/raw/pet/siemens/biograph_vision/vr20b/pet_countrate/header.py

@@ -4,8 +4,8 @@
 )
 
 
-def get_image_header(out_dir, **kwargs):
-    hdr = evolve_header(constant_hdr, skip_unknown=True, **kwargs)
+def get_image_header(out_dir, skip_unknown=True, **kwargs):
+    hdr = evolve_header(constant_hdr, skip_unknown=skip_unknown, **kwargs)
     return generate_dicom(out_dir, num_vols, hdr, collated_data, {})
 
 

medimages4tests/dummy/raw/pet/siemens/biograph_vision/vr20b/pet_listmode/header.py

@@ -4,8 +4,8 @@
 )
 
 
-def get_image_header(out_dir, **kwargs):
-    hdr = evolve_header(constant_hdr, skip_unknown=True, **kwargs)
+def get_image_header(out_dir, skip_unknown=True, **kwargs):
+    hdr = evolve_header(constant_hdr, skip_unknown=skip_unknown, **kwargs)
     return generate_dicom(out_dir, num_vols, hdr, collated_data, varying_hdr)