PWGHF: add converter for 3-prong ML table in D2H reduced data format (AliceO2Group#8006)

* PWGHF: add converter for 3-prong ML table in D2H reduced data format

* Please consider the following formatting changes

* Remove whitespace added by the bot

* Add missing comma

* Add missing iterator

* Update PWGHF/D2H/TableProducer/converterReduced3ProngsMl.cxx

Co-authored-by: Vít Kučera <[email protected]>

* Update PWGHF/D2H/TableProducer/converterReduced3ProngsMl.cxx

Co-authored-by: Vít Kučera <[email protected]>

* Update PWGHF/D2H/TableProducer/converterReduced3ProngsMl.cxx

Co-authored-by: Vít Kučera <[email protected]>

* Update PWGHF/D2H/TableProducer/CMakeLists.txt

Co-authored-by: Vít Kučera <[email protected]>

---------

Co-authored-by: ALICE Action Bot <[email protected]>
Co-authored-by: Vít Kučera <[email protected]>

Author: 3 people

PWGHF/D2H/DataModel/ReducedDataModel.h

@@ -221,14 +221,21 @@ DECLARE_SOA_TABLE(HfRed3ProngsCov, "AOD", "HFRED3PRONGSCOV", //! Table with 3pro
                   HFTRACKPARCOV_COLUMNS,
                   o2::soa::Marker<2>);
 
-DECLARE_SOA_TABLE(HfRed3ProngsMl, "AOD", "HFRED3PRONGML", //! Table with 3prong candidate ML scores
+DECLARE_SOA_TABLE(HfRed3ProngsMl_000, "AOD", "HFRED3PRONGML", //! Table with 3prong candidate ML scores
                   hf_charm_cand_reduced::MlScoreBkgMassHypo0,
                   hf_charm_cand_reduced::MlScorePromptMassHypo0,
-                  hf_charm_cand_reduced::MlScoreNonpromptMassHypo0,
-                  hf_charm_cand_reduced::MlScoreBkgMassHypo1,
-                  hf_charm_cand_reduced::MlScorePromptMassHypo1,
-                  hf_charm_cand_reduced::MlScoreNonpromptMassHypo1,
-                  o2::soa::Marker<1>);
+                  hf_charm_cand_reduced::MlScoreNonpromptMassHypo0);
+
+DECLARE_SOA_TABLE_VERSIONED(HfRed3ProngsMl_001, "AOD", "HFRED3PRONGML", 1, //! Table with 3prong candidate ML scores (format for 2 mass hypotheses needed for Ds and Lc)
+                            hf_charm_cand_reduced::MlScoreBkgMassHypo0,
+                            hf_charm_cand_reduced::MlScorePromptMassHypo0,
+                            hf_charm_cand_reduced::MlScoreNonpromptMassHypo0,
+                            hf_charm_cand_reduced::MlScoreBkgMassHypo1,
+                            hf_charm_cand_reduced::MlScorePromptMassHypo1,
+                            hf_charm_cand_reduced::MlScoreNonpromptMassHypo1,
+                            o2::soa::Marker<1>);
+
+using HfRed3ProngsMl = HfRed3ProngsMl_001;
 
 // Beauty candidates prongs
 namespace hf_cand_b0_reduced

PWGHF/D2H/TableProducer/CMakeLists.txt

@@ -60,3 +60,9 @@ o2physics_add_dpl_workflow(data-creator-charm-reso-reduced
                     PUBLIC_LINK_LIBRARIES O2Physics::AnalysisCore O2Physics::EventFilteringUtils
                     COMPONENT_NAME Analysis)
 
+# Converters
+
+o2physics_add_dpl_workflow(converter-reduced-3-prongs-ml
+                    SOURCES converterReduced3ProngsMl.cxx
+                    PUBLIC_LINK_LIBRARIES O2Physics::AnalysisCore
+                    COMPONENT_NAME Analysis)

PWGHF/D2H/TableProducer/converterReduced3ProngsMl.cxx

@@ -0,0 +1,41 @@
+// Copyright 2019-2020 CERN and copyright holders of ALICE O2.
+// See https://alice-o2.web.cern.ch/copyright for details of the copyright holders.
+// All rights not expressly granted are reserved.
+//
+// This software is distributed under the terms of the GNU General Public
+// License v3 (GPL Version 3), copied verbatim in the file "COPYING".
+//
+// In applying this license CERN does not waive the privileges and immunities
+// granted to it by virtue of its status as an Intergovernmental Organization
+// or submit itself to any jurisdiction.
+
+/// \file converterReduced3ProngsMl.cxx
+/// \brief Task for conversion of HfRed3ProngsMl to version 001
+///
+/// \author Fabrizio Grosa <[email protected]>, CERN
+
+#include "Framework/runDataProcessing.h"
+#include "Framework/AnalysisTask.h"
+#include "Framework/AnalysisDataModel.h"
+
+#include "PWGHF/D2H/DataModel/ReducedDataModel.h"
+
+using namespace o2;
+using namespace o2::framework;
+
+// Swaps covariance matrix elements if the data is known to be bogus (collision_000 is bogus)
+struct HfConverterReduced3ProngsMl {
+  Produces<aod::HfRed3ProngsMl_001> ml3Prongs;
+
+  void process(aod::HfRed3ProngsMl_000::iterator const& mlScoreTable)
+  {
+    ml3Prongs(mlScoreTable.mlScoreBkgMassHypo0(), mlScoreTable.mlScorePromptMassHypo0(), mlScoreTable.mlScoreNonpromptMassHypo0(), -1.f, -1.f, -1.f);
+  }
+};
+
+WorkflowSpec defineDataProcessing(ConfigContext const& cfgc)
+{
+  return WorkflowSpec{
+    adaptAnalysisTask<HfConverterReduced3ProngsMl>(cfgc),
+  };
+}