added storing dictionary in break dyanamic alignment with CT

amypet/align_brkdyn_ct.py

@@ -19,7 +19,7 @@
 
 
 
-def align_break_petct(niidat, cts, Cnt, qcpth=None, refpetidx=None):
+def align_break_petct(niidat, cts, Cnt, qcpth=None, refpetidx=None, use_stored=False):
     '''
     Align PET images in break dynamic acquisition
     using the CT for aligning all frames within each acquisition
@@ -39,6 +39,23 @@ def align_break_petct(niidat, cts, Cnt, qcpth=None, refpetidx=None):
                 used as reference instead.
     '''
 
+    # > output path from the input dictionary
+    opth = niidat['outpath'].parent
+
+    # > output path for alignment of CTs
+    algnFpth = opth/(f'aligned_full')
+    affsF = algnFpth/'combined_affines'
+
+    nimpa.create_dir(algnFpth)
+    nimpa.create_dir(affsF)
+
+    # > output dictionary file
+    fout = algnFpth/'aligned_full_output.npy'
+    if use_stored and fout.is_file():
+        outdct = np.load(fout, allow_pickle=True)
+        outdct = outdct.item()
+        return outdct
+
     # > number of frames for break dynamic acquisitions A and B
     nfrmA = len(niidat['series'][0])
     nfrmB = len(niidat['series'][1])
@@ -65,14 +82,7 @@ def align_break_petct(niidat, cts, Cnt, qcpth=None, refpetidx=None):
     # > alignment results
     algn_frm = [None, None]
 
-    opth = niidat['outpath'].parent
-
-    # > output path for alignment of CTs
-    algnFpth = opth/(f'aligned_full')
-    affsF = algnFpth/'combined_affines'
 
-    nimpa.create_dir(algnFpth)
-    nimpa.create_dir(affsF)
 
     if qcpth is None:
         qcpth = opth/'QC-output'
@@ -343,5 +353,7 @@ def align_break_petct(niidat, cts, Cnt, qcpth=None, refpetidx=None):
     outdct['faffines'] = faffines
     outdct['fnii_com'] = algn_frm[0]['fnii_com'] + algn_frm[1]['fnii_com']
 
+    np.save(fout, outdct)
+
     return outdct