Skip to content
Snippets Groups Projects
Commit 9080c3fd authored by Mattia Mancini's avatar Mattia Mancini
Browse files

Refactor find skymodel cal

Former-commit-id: 33381e1d
Former-commit-id: 26f182cb
parent 337f613d
No related branches found
No related tags found
No related merge requests found
...@@ -7,12 +7,14 @@ baseCommand: ...@@ -7,12 +7,14 @@ baseCommand:
- python3 - python3
inputs: inputs:
- id: msin - id: msin
type: Directory type: Directory[]
doc: MS containing the calibrator doc: MS containing the calibrator
inputBinding: inputBinding:
position: 0 position: 0
- id: skymodels - id: skymodels
type: Directory type:
- Directory?
- File?
doc: Directory containing the sky models doc: Directory containing the sky models
- id: skymodels_extension - id: skymodels_extension
type: string? type: string?
...@@ -23,6 +25,9 @@ arguments: ...@@ -23,6 +25,9 @@ arguments:
- '-c' - '-c'
- | - |
import sys import sys
import os
import shutil
null = None
from unittest.mock import MagicMock from unittest.mock import MagicMock
sys.modules['lofarpipe.support.data_map'] = MagicMock() sys.modules['lofarpipe.support.data_map'] = MagicMock()
...@@ -31,16 +36,26 @@ arguments: ...@@ -31,16 +36,26 @@ arguments:
from find_skymodel_cal import main as find_skymodel from find_skymodel_cal import main as find_skymodel
import json import json
mss = sys.argv[1:] mss = sys.argv[1:]
skymodels = "$(inputs.skymodels.path)" print(mss)
skymodels = $(inputs.skymodels)
extension = "$(inputs.skymodels_extension)" extension = "$(inputs.skymodels_extension)"
if skymodels is None:
skymodels = "/data/skymodels"
else:
skymodels = skymodels["path"]
output = {} output = {}
if extension != 'null': if extension != 'null':
output = find_skymodel(mss, skymodels, './', extension) output = find_skymodel(mss, skymodels, './', extension)
else: else:
output = find_skymodel(mss, skymodels, './') output = find_skymodel(mss, skymodels, './')
cwl_output = {'class': 'File', 'path': output['SkymodelCal']} skymodel_path = output['SkymodelCal']
skymodel_name = output['SkymodelName']
skymodel_path = shutil.copy(skymodel_path, os.getcwd())
cwl_output = {"file":{'class': 'File', 'path': skymodel_path},
"skymodel_name": skymodel_name}
with open('./out.json', 'w') as fp: with open('./out.json', 'w') as fp:
json.dump(cwl_output, fp) json.dump(cwl_output, fp)
outputs: outputs:
...@@ -49,11 +64,17 @@ outputs: ...@@ -49,11 +64,17 @@ outputs:
outputBinding: outputBinding:
loadContents: true loadContents: true
glob: 'out.json' glob: 'out.json'
outputEval: $(JSON.parse(self[0].contents)) outputEval: $(JSON.parse(self[0].contents).file)
- id: model_name
type: string
outputBinding:
loadContents: true
glob: 'out.json'
outputEval: $(JSON.parse(self[0].contents).skymodel_name)
requirements: requirements:
- class: InlineJavascriptRequirement - class: InlineJavascriptRequirement
hints: hints:
DockerRequirement: DockerRequirement:
dockerPull: lofareosc/prefactor-ci:master dockerPull: lofareosc/prefactor:latest
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment