FileReaderWorkflow.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 "TPCReaderWorkflow/ClusterReaderSpec.h"
#include "TPCReaderWorkflow/TriggerReaderSpec.h"
#include "TPCReaderWorkflow/TrackReaderSpec.h"
#include "TPCWorkflow/ClusterSharingMapSpec.h"
#include "DetectorsRaw/HBFUtilsInitializer.h"
#include "Framework/CallbacksPolicy.h"
#include "Algorithm/RangeTokenizer.h"
 
#include "SimulationDataFormat/IOMCTruthContainerView.h"
#include "SimulationDataFormat/ConstMCTruthContainer.h"
#include "SimulationDataFormat/MCCompLabel.h"
 
void customize(std::vector<o2::framework::CallbacksPolicy>& policies)
{
  o2::raw::HBFUtilsInitializer::addNewTimeSliceCallback(policies);
}
 
// add workflow options, note that customization needs to be declared before
// including Framework/runDataProcessing
void customize(std::vector<o2::framework::ConfigParamSpec>& workflowOptions)
{
  using namespace o2::framework;
 
  std::vector<ConfigParamSpec> options{
    {"input-type", VariantType::String, "clusters", {"clusters, tracks"}},
    {"disable-mc", VariantType::Bool, false, {"disable sending of MC information"}},
    {"configKeyValues", VariantType::String, "", {"semicolon separated key=value strings"}}};
  o2::raw::HBFUtilsInitializer::addConfigOption(options);
  std::swap(workflowOptions, options);
}
 
#include "Framework/runDataProcessing.h" // the main driver
 
using namespace o2::framework;
 
enum struct InputType { Clusters,
                        Tracks
};
 
const std::unordered_map<std::string, InputType> InputMap{
  {"clusters", InputType::Clusters},
  {"tracks", InputType::Tracks}};
 
WorkflowSpec defineDataProcessing(ConfigContext const& cfgc)
{
  WorkflowSpec specs;
 
  auto inputType = cfgc.options().get<std::string>("input-type");
  bool doMC = not cfgc.options().get<bool>("disable-mc");
 
  std::vector<InputType> inputTypes;
  try {
    inputTypes = o2::RangeTokenizer::tokenize<InputType>(inputType, [](std::string const& token) { return InputMap.at(token); });
  } catch (std::out_of_range&) {
    throw std::invalid_argument(std::string("invalid input type: ") + inputType);
  }
  auto isEnabled = [&inputTypes](InputType type) {
    return std::find(inputTypes.begin(), inputTypes.end(), type) != inputTypes.end();
  };
 
  if (isEnabled(InputType::Clusters)) {
    specs.emplace_back(o2::tpc::getClusterReaderSpec(doMC));
    if (!getenv("DPL_DISABLE_TPC_TRIGGER_READER") || atoi(getenv("DPL_DISABLE_TPC_TRIGGER_READER")) != 1) {
      specs.emplace_back(o2::tpc::getTPCTriggerReaderSpec());
    }
  }
 
  if (isEnabled(InputType::Tracks)) {
 
    specs.push_back(o2::tpc::getTPCTrackReaderSpec(doMC));
  }
 
  if (isEnabled(InputType::Tracks) && isEnabled(InputType::Clusters)) { // create shared clusters and occupancy maps
    specs.emplace_back(o2::tpc::getClusterSharingMapSpec());
  }
 
  o2::raw::HBFUtilsInitializer hbfIni(cfgc, specs);
  return std::move(specs);
}