DigitizationData.hpp

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file DigitizationData.hpp

// This file is part of the Acts project.
//
// Copyright (C) 2020 CERN for the benefit of the Acts project
//
// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/.

#pragma once

#include "Acts/Definitions/Algebra.hpp"
#include "Acts/Definitions/TrackParametrization.hpp"

#include <unordered_set>
#include <utility>

namespace ActsFatras {

using Cell = std::pair<unsigned int, Acts::ActsScalar>;

template <typename signal_t, size_t kSize>
struct Channel {
  std::array<Cell, kSize> cellId;
  signal_t value = 0;
  std::unordered_set<unsigned int> links = {};

  Channel(std::array<Cell, kSize> cellId_, signal_t value_,
          std::unordered_set<unsigned int> links_ = {})
      : cellId(cellId_), value(value_), links(std::move(links_)) {}

  Channel() = delete;
};

template <typename signal_t, size_t kSize>
struct Cluster {
  using Scalar = Acts::ActsScalar;
  using ParametersVector = Acts::ActsVector<kSize>;
  using CovarianceMatrix = Acts::ActsSquareMatrix<kSize>;

  ParametersVector parameters = ParametersVector::Zero();
  CovarianceMatrix covariance = CovarianceMatrix::Zero();
  std::array<unsigned int, kSize> clusterSize;
  std::vector<Channel<signal_t, kSize>> channels;

  template <typename parameters_t, typename covariance_t>
  Cluster(const Eigen::MatrixBase<parameters_t>& p,
          const Eigen::MatrixBase<covariance_t>& c,
          std::array<unsigned int, kSize> cSize,
          std::vector<Channel<signal_t, kSize>> cChannels)
      : parameters(p), covariance(c), clusterSize(cSize), channels(cChannels) {}

  Cluster() = delete;
};

}  // namespace ActsFatras