ColumnDataSpecsUtils.h

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.
 
 
#ifndef O2_MID_COLUMNDATASPECSUTILS_H
#define O2_MID_COLUMNDATASPECSUTILS_H
 
#include <array>
#include <memory>
#include <string>
#include <string_view>
#include <vector>
#include <gsl/span>
#include "fmt/format.h"
 
#include "Framework/DataAllocator.h"
#include "Framework/InputRecordWalker.h"
#include "Framework/InputSpec.h"
#include "Framework/OutputSpec.h"
#include "Framework/ProcessingContext.h"
#include "SimulationDataFormat/MCTruthContainer.h"
#include "DataFormatsMID/ColumnData.h"
#include "DataFormatsMID/ROFRecord.h"
#include "DataFormatsMID/MCLabel.h"
 
namespace o2
{
namespace mid
{
namespace specs
{
 
std::vector<framework::InputSpec> buildInputSpecs(std::string_view dataBind, std::string_view dataDesc, bool useMC);
 
std::vector<framework::InputSpec> buildInputSpecs(std::string_view dataBind, std::string_view dataDesc, std::string_view rofDesc, std::string_view labelsDesc, bool useMC);
 
std::vector<framework::OutputSpec> buildOutputSpecs(std::string_view bind, std::string_view description);
 
std::vector<framework::OutputSpec> buildStandardOutputSpecs(std::string_view dataBind, std::string_view dataDesc, bool useMC);
 
template <typename T>
std::array<gsl::span<const T>, NEvTypes> getInput(framework::ProcessingContext& pc, std::string_view bind)
{
  std::array<gsl::span<const T>, 3> data;
  for (auto const& inputRef : framework::InputRecordWalker(pc.inputs())) {
    auto const* dh = framework::DataRefUtils::getHeader<o2::header::DataHeader*>(inputRef);
    auto subSpecIdx = static_cast<size_t>(dh->subSpecification);
    if (framework::DataRefUtils::match(inputRef, bind.data())) {
      data[subSpecIdx] = pc.inputs().get<gsl::span<T>>(inputRef);
    }
  }
  return data;
}
 
std::vector<framework::Output> buildOutputs(std::vector<framework::OutputSpec> outputSpecs);
 
std::array<gsl::span<const ColumnData>, NEvTypes> getData(framework::ProcessingContext& pc, std::string_view dataBind);
 
gsl::span<const ColumnData> getData(framework::ProcessingContext& pc, std::string_view dataBind, EventType eventType);
 
std::array<gsl::span<const ROFRecord>, NEvTypes> getRofs(framework::ProcessingContext& pc, std::string_view dataBind);
 
gsl::span<const ROFRecord> getRofs(framework::ProcessingContext& pc, std::string_view dataBind, EventType eventType);
 
std::unique_ptr<const o2::dataformats::MCTruthContainer<MCLabel>> getLabels(framework::ProcessingContext& pc, std::string_view dataBind);
 
} // namespace specs
} // namespace mid
} // namespace o2
 
#endif // O2_MID_COLUMNDATASPECSUTILS_H