Skip to content

Commit

Permalink
Remove the unused upload outputs.
Browse files Browse the repository at this point in the history
  • Loading branch information
FredLoney committed Jun 23, 2017
1 parent b2a43dd commit 11027e9
Showing 1 changed file with 17 additions and 21 deletions.
38 changes: 17 additions & 21 deletions qipipe/pipeline/staging.py
Original file line number Diff line number Diff line change
Expand Up @@ -222,20 +222,20 @@ def _create_workflow(self, is_multi_volume=True):
stg_inputs = stg_fields + iter_fields + ['opts']
stg_xfc = Function(input_names=stg_inputs, output_names=['out_file'],
function=stage_volume)
stg_node = pe.Node(stg_xfc, name='stage_volume')
stg_node.inputs.opts = self._child_options()
stage = pe.Node(stg_xfc, name='stage_volume')
stage.inputs.opts = self._child_options()
for fld in stg_fields:
workflow.connect(input_spec, fld, stg_node, fld)
workflow.connect(input_spec, fld, stage, fld)
for fld in iter_fields:
workflow.connect(iter_volume, fld, stg_node, fld)
workflow.connect(iter_volume, fld, stage, fld)

# Collect the 3D volume files.
collect_xfc = IdentityInterface(fields=['volume_files'])
collect_vols = pe.JoinNode(
collect_xfc, joinsource='iter_volume',
joinfield='volume_files', name='collect_volumes'
)
workflow.connect(stg_node, 'out_file', collect_vols, 'volume_files')
workflow.connect(stage, 'out_file', collect_vols, 'volume_files')

# Upload the processed DICOM and NIfTI files.
# The upload out_files output is the volume files.
Expand All @@ -244,16 +244,15 @@ def _create_workflow(self, is_multi_volume=True):
['project', 'dcm_dir', 'volume_files', 'time_series']
)
upload_xfc = Function(input_names=upload_fields,
output_names=['out_files'],
output_names=[],
function=upload)
upload_node = pe.Node(upload_xfc, name='upload')
upload_node.inputs.project = self.project
workflow.connect(input_spec, 'subject', upload_node, 'subject')
workflow.connect(input_spec, 'session', upload_node, 'session')
workflow.connect(input_spec, 'scan', upload_node, 'scan')
workflow.connect(input_spec, 'dest', upload_node, 'dcm_dir')
workflow.connect(collect_vols, 'volume_files',
upload_node, 'volume_files')
upload = pe.Node(upload_xfc, name='_upload')
upload.inputs.project = self.project
workflow.connect(input_spec, 'subject', upload, 'subject')
workflow.connect(input_spec, 'session', upload, 'session')
workflow.connect(input_spec, 'scan', upload, 'scan')
workflow.connect(input_spec, 'dest', upload, 'dcm_dir')
workflow.connect(collect_vols, 'volume_files', upload, 'volume_files')
if is_multi_volume:
# Merge the volumes.
merge_vols_xfc = MergeNifti(out_format=SCAN_TS_BASE)
Expand All @@ -263,10 +262,10 @@ def _create_workflow(self, is_multi_volume=True):
workflow.connect(collect_vols, 'volume_files',
merge_vols, 'in_files')
workflow.connect(merge_vols, 'out_file',
upload_node, 'time_series')
upload, 'time_series')
self.logger.debug('Connected staging to scan time series merge.')
else:
upload_node.inputs.time_series = None
upload.inputs.time_series = None
self.logger.debug('Connected scan time series merge to upload.')

# The output is the 4D time series and 3D NIfTI volume image files.
Expand Down Expand Up @@ -561,8 +560,8 @@ def stage_volume(collection, subject, session, scan, volume, in_files,
return out_file


def upload(project, subject, session, scan, dcm_dir, volume_files,
time_series=None):
def _upload(project, subject, session, scan, dcm_dir, volume_files,
time_series=None):
"""
Uploads the staged files in *dcm_dir* as follows:
* the processed DICOM ``.dcm.gz`` files are uploaded to the
Expand All @@ -578,7 +577,6 @@ def upload(project, subject, session, scan, dcm_dir, volume_files,
:param volume_files: the 3D scan volume files
:param time_series: the 4D scan time series file, if the scan is
multi-volume
:return: the 3D volume image NIfTI files
"""
import os
import glob
Expand Down Expand Up @@ -636,8 +634,6 @@ def upload(project, subject, session, scan, dcm_dir, volume_files,
_logger.debug("Uploaded %d %s %s scan %d staged NIfTI files to"
" XNAT." % (nii_file_cnt, subject, session, scan))

return vol_files


def volume_format(collection):
"""
Expand Down

0 comments on commit 11027e9

Please sign in to comment.