Coverage for python/lsst/analysis/tools/tasks/diffimTaskDetectorVisitMetricsAnalysis.py: 45%
29 statements
« prev ^ index » next coverage.py v7.4.4, created at 2024-04-20 03:58 -0700
« prev ^ index » next coverage.py v7.4.4, created at 2024-04-20 03:58 -0700
1# This file is part of analysis_tools.
2#
3# Developed for the LSST Data Management System.
4# This product includes software developed by the LSST Project
5# (https://www.lsst.org).
6# See the COPYRIGHT file at the top-level directory of this distribution
7# for details of code ownership.
8#
9# This program is free software: you can redistribute it and/or modify
10# it under the terms of the GNU General Public License as published by
11# the Free Software Foundation, either version 3 of the License, or
12# (at your option) any later version.
13#
14# This program is distributed in the hope that it will be useful,
15# but WITHOUT ANY WARRANTY; without even the implied warranty of
16# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
17# GNU General Public License for more details.
18#
19# You should have received a copy of the GNU General Public License
20# along with this program. If not, see <https://www.gnu.org/licenses/>.
21from __future__ import annotations
23__all__ = ("DiffimDetectorVisitMetricsAnalysisConfig", "DiffimDetectorVisitMetricsAnalysisTask")
25import pandas as pd
26from lsst.pipe.base import NoWorkFound, connectionTypes
28from ..interfaces import AnalysisBaseConfig, AnalysisBaseConnections, AnalysisPipelineTask
31class DiffimDetectorVisitMetricsAnalysisConnections(
32 AnalysisBaseConnections,
33 dimensions=("visit", "band", "detector"),
34):
35 metadataSubtract = connectionTypes.Input(
36 doc="Task metadata from image differencing",
37 name="subtractImages_metadata",
38 storageClass="TaskMetadata",
39 dimensions=("visit", "band", "detector"),
40 )
41 metadataDetect = connectionTypes.Input(
42 doc="Task metadata from DIA detection and measurement",
43 name="detectAndMeasure_metadata",
44 storageClass="TaskMetadata",
45 dimensions=("visit", "band", "detector"),
46 )
49class DiffimDetectorVisitMetricsAnalysisConfig(
50 AnalysisBaseConfig, pipelineConnections=DiffimDetectorVisitMetricsAnalysisConnections
51):
52 pass
55class DiffimDetectorVisitMetricsAnalysisTask(AnalysisPipelineTask):
56 ConfigClass = DiffimDetectorVisitMetricsAnalysisConfig
57 _DefaultName = "DiffimDetectorVisitMetricsAnalysis"
59 def runQuantum(self, butlerQC, inputRefs, outputRefs):
60 inputs = butlerQC.get(inputRefs)
61 detectTaskName = inputRefs.metadataDetect.datasetType.name
62 detectTaskName = detectTaskName[: detectTaskName.find("_")]
63 metadata = inputs["metadataDetect"].metadata[detectTaskName].to_dict()
64 if not metadata:
65 raise NoWorkFound("No metadata entries for detectAndMeasure.")
66 inputs.pop("metadataDetect")
67 subtractTaskName = inputRefs.metadataSubtract.datasetType.name
68 subtractTaskName = subtractTaskName[: subtractTaskName.find("_")]
69 metadata |= inputs["metadataSubtract"].metadata[subtractTaskName].to_dict()
70 inputs.pop("metadataSubtract")
71 df = pd.DataFrame(metadata)
73 inputs["data"] = df
74 outputs = self.run(**inputs)
75 butlerQC.put(outputs, outputRefs)