VertexSamplerSpec.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 "VertexSamplerSpec.h"
 
#include <iostream>
#include <fstream>
#include <string>
 
#include <stdexcept>
 
#include "Framework/CallbackService.h"
#include "Framework/ConfigParamRegistry.h"
#include "Framework/ControlService.h"
#include "Framework/DataProcessorSpec.h"
#include "Framework/Lifetime.h"
#include "Framework/Output.h"
#include "Framework/Task.h"
#include "Framework/Logger.h"
 
#include "MathUtils/Cartesian.h"
#include "DataFormatsMCH/ROFRecord.h"
 
namespace o2
{
namespace mch
{
 
using namespace std;
using namespace o2::framework;
 
class VertexSamplerSpec
{
 public:
  //_________________________________________________________________________________________________
  void init(framework::InitContext& ic)
  {
    LOG(info) << "initializing vertex sampler";
 
    auto inputFileName = ic.options().get<std::string>("infile");
    if (!inputFileName.empty()) {
      mInputFile.open(inputFileName, ios::binary);
      if (!mInputFile.is_open()) {
        throw invalid_argument("Cannot open input file" + inputFileName);
      }
    }
 
    auto stop = [this]() {
      LOG(info) << "stop vertex sampler";
      if (mInputFile.is_open()) {
        this->mInputFile.close();
      }
    };
    ic.services().get<CallbackService>().set<CallbackService::Id::Stop>(stop);
  }
 
  //_________________________________________________________________________________________________
  void run(framework::ProcessingContext& pc)
  {
 
    // get the track ROFs for each event in the TF
    auto rofs = pc.inputs().get<gsl::span<ROFRecord>>("rofs");
 
    // create the output message to hold vertices
    auto& vertices = pc.outputs().make<std::vector<math_utils::Point3D<double>>>(OutputRef{"vertices"});
 
    // read the vertex of each event or set it to (0,0,0)
    if (mInputFile.is_open()) {
      int event(-1);
      VertexStruct vtx{};
      vertices.reserve(rofs.size());
      for (const auto& rof : rofs) {
        mInputFile.read(reinterpret_cast<char*>(&event), sizeof(int));
        if (mInputFile.fail() || event != rof.getBCData().orbit) {
          throw out_of_range(std::string("missing vertex for event ") + rof.getBCData().orbit);
        }
        mInputFile.read(reinterpret_cast<char*>(&vtx), sizeof(VertexStruct));
        vertices.emplace_back(vtx.x, vtx.y, vtx.z);
      }
    } else {
      vertices.resize(rofs.size(), math_utils::Point3D<double>(0., 0., 0.));
    }
  }
 
 private:
  struct VertexStruct {
    double x;
    double y;
    double z;
  };
 
  std::ifstream mInputFile{}; 
};
 
//_________________________________________________________________________________________________
o2::framework::DataProcessorSpec getVertexSamplerSpec(const char* specName)
{
  return DataProcessorSpec{
    specName,
    Inputs{InputSpec{"rofs", "MCH", "TRACKROFS", 0, Lifetime::Timeframe},
           // track and cluster messages are there just to keep things synchronized
           InputSpec{"tracks", "MCH", "TRACKS", 0, Lifetime::Timeframe},
           InputSpec{"clusters", "MCH", "TRACKCLUSTERS", 0, Lifetime::Timeframe}},
    Outputs{OutputSpec{{"vertices"}, "MCH", "VERTICES", 0, Lifetime::Timeframe}},
    AlgorithmSpec{adaptFromTask<VertexSamplerSpec>()},
    Options{{"infile", VariantType::String, "", {"input filename"}}}};
}
 
} // end namespace mch
} // end namespace o2