TOFCalibWriterSpec.cxx

Go to the documentation of this file.

// 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.
 
 
#include "TOFWorkflowIO/TOFCalibWriterSpec.h"
#include "Framework/ConfigParamRegistry.h"
#include "Framework/ControlService.h"
#include "DPLUtils/MakeRootTreeWriterSpec.h"
#include "Headers/DataHeader.h"
#include <SimulationDataFormat/MCCompLabel.h>
#include <SimulationDataFormat/MCTruthContainer.h>
#include "TTree.h"
#include "TBranch.h"
#include "TFile.h"
#include "ReconstructionDataFormats/MatchInfoTOF.h"
#include "DataFormatsTOF/CalibInfoTOF.h"
#include "DataFormatsTOF/Cluster.h"
#include "DataFormatsTOF/Diagnostic.h"
 
using namespace o2::framework;
 
namespace o2
{
namespace tof
{
 
template <typename T>
using BranchDefinition = MakeRootTreeWriterSpec::BranchDefinition<T>;
using CalibInfosType = std::vector<o2::dataformats::CalibInfoTOF>;
using CalibDiaType = o2::tof::Diagnostic;
DataProcessorSpec getTOFCalibWriterSpec(const char* outdef, bool toftpc, bool addDia, bool onlyDia)
{
  if (!addDia) {
    onlyDia = false;
  }
  // A spectator for logging
  auto logger = [](CalibInfosType const& indata) {
    LOG(debug) << "RECEIVED MATCHED SIZE " << indata.size();
  };
  // A spectator for logging
  auto loggerDia = [](CalibDiaType const& indata) {
    LOG(debug) << "DIAGNOSTIC WORD SIZE " << indata.size();
  };
  o2::header::DataDescription ddCalib{"CALIBDATA"};
  o2::header::DataDescription ddCalibDia{"DIAFREQ"};
  return MakeRootTreeWriterSpec("TOFCalibWriter",
                                outdef,
                                "calibTOF",
                                BranchDefinition<CalibInfosType>{InputSpec{"input", o2::header::gDataOriginTOF, ddCalib, 0},
                                                                 "TOFCalibInfo",
                                                                 "calibinfo-branch-name",
                                                                 !onlyDia,
                                                                 logger},
                                BranchDefinition<CalibDiaType>{InputSpec{"inputDia", o2::header::gDataOriginTOF, ddCalibDia, 0},
                                                               "TOFDiaInfo",
                                                               "calibdia-branch-name",
                                                               addDia,
                                                               loggerDia})();
}
 
} // namespace tof
} // namespace o2