Processors.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_ZDC_FAST_SIMULATIONS_PROCESSORS_H
#define O2_ZDC_FAST_SIMULATIONS_PROCESSORS_H
 
#if __has_include(<onnxruntime/core/session/onnxruntime_cxx_api.h>)
#include <onnxruntime/core/session/onnxruntime_cxx_api.h>
#else
#include <onnxruntime_cxx_api.h>
#endif
#include <optional>
#include <vector>
 
namespace o2::zdc::fastsim::processors
{
 
class StandardScaler
{
 public:
  StandardScaler() = default;
  ~StandardScaler() = default;
 
  [[nodiscard]] std::optional<std::vector<float>> scale(const std::vector<float>& data) const;
 
  [[nodiscard]] std::optional<std::vector<std::vector<float>>> scale_batch(
    const std::vector<std::vector<float>>& data) const;
 
  bool setScales(const std::vector<float>& means, const std::vector<float>& scales);
 
 private:
  std::vector<float> mMeans;
  std::vector<float> mScales;
};
 
std::vector<int> readClassifier(const Ort::Value& value, size_t batchSize);
 
std::vector<std::array<long, 5>> calculateChannels(const Ort::Value& value, size_t batchSize);
 
} // namespace o2::zdc::fastsim::processors
#endif // O2_ZDC_FAST_SIMULATIONS_PROCESSORS_H