Source code for pipeline.hif.tasks.refant.qa

import collections

import pipeline.infrastructure.logging as logging
import pipeline.infrastructure.pipelineqa as pqa
import pipeline.infrastructure.utils as utils
from . import referenceantenna

LOG = logging.get_logger(__name__)


[docs]class RefantQAHandler(pqa.QAPlugin): result_cls = referenceantenna.RefAntResults child_cls = None
[docs] def handle(self, context, result): vis = result.inputs['vis'] ms = context.observing_run.get_ms(vis) if result._refant != '': qa_score = 1.0 longmsg = 'Reference antenna for %s was selected successfully' % ms.basename shortmsg = 'Refant OK' else: qa_score = 0.0 longmsg = 'Could not select reference antenna for %s' % ms.basename shortmsg = 'No refant' origin = pqa.QAOrigin(metric_name='RefantQAHandler', metric_score=bool(qa_score), metric_units='Reference antenna was identified') scores = [pqa.QAScore(qa_score, longmsg=longmsg, shortmsg=shortmsg, vis=ms.basename, origin=origin)] result.qa.pool[:] = scores
[docs]class RefantListQAHandler(pqa.QAPlugin): result_cls = collections.Iterable child_cls = referenceantenna.RefAntResults
[docs] def handle(self, context, result): # collate the QAScores from each child result, pulling them into our # own QAscore list collated = utils.flatten([r.qa.pool for r in result]) result.qa.pool[:] = collated