NuclearInteractionParameters.hpp

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

// This file is part of the Acts project.
//
// Copyright (C) 2021 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/Common.hpp"

#include <vector>

namespace ActsFatras {
namespace detail {

struct NuclearInteractionParameters {
  using CumulativeDistribution =
      std::pair<std::vector<float>, std::vector<uint32_t>>;
  using Distributions = std::vector<CumulativeDistribution>;
  using PdgMap =
      std::vector<std::pair<int, std::vector<std::pair<int, float>>>>;

  struct ParametersWithFixedMultiplicity {
    ParametersWithFixedMultiplicity() = default;

    ParametersWithFixedMultiplicity(
        Distributions& momenta, Acts::ActsDynamicVector& eValMom,
        Acts::ActsDynamicVector& eVecMom, Acts::ActsDynamicVector& meanMom,
        Distributions& invariantMasses, Acts::ActsDynamicVector& eValIM,
        Acts::ActsDynamicVector& eVecIM, Acts::ActsDynamicVector& meanIM)
        : validParametrisation(true),
          momentumDistributions(momenta),
          eigenvaluesMomentum(eValMom),
          meanMomentum(meanMom),
          invariantMassDistributions(invariantMasses),
          eigenvaluesInvariantMass(eValIM),
          meanInvariantMass(meanIM) {
      const unsigned int sizeMom = eigenvaluesMomentum.size();
      eigenvectorsMomentum.resize(sizeMom, sizeMom);
      for (unsigned int i = 0; i < sizeMom; i++) {
        for (unsigned int j = 0; j < sizeMom; j++) {
          eigenvectorsMomentum(i, j) = eVecMom[i * sizeMom + j];
        }
      }

      const unsigned int sizeInvMass = eigenvaluesInvariantMass.size();
      eigenvectorsInvariantMass.resize(sizeInvMass, sizeInvMass);
      for (unsigned int i = 0; i < sizeInvMass; i++) {
        for (unsigned int j = 0; j < sizeInvMass; j++) {
          eigenvectorsInvariantMass(i, j) = eVecIM[i * sizeInvMass + j];
        }
      }
    }

    bool validParametrisation = false;

    Distributions momentumDistributions;
    Acts::ActsDynamicVector eigenvaluesMomentum;
    Acts::ActsDynamicMatrix eigenvectorsMomentum;
    Acts::ActsDynamicVector meanMomentum;

    Distributions invariantMassDistributions;
    Acts::ActsDynamicVector eigenvaluesInvariantMass;
    Acts::ActsDynamicMatrix eigenvectorsInvariantMass;
    Acts::ActsDynamicVector meanInvariantMass;
  };

  float momentum = 0;
  float softInteractionProbability = 0;
  PdgMap pdgMap;
  CumulativeDistribution nuclearInteractionProbability;
  CumulativeDistribution softMultiplicity;
  CumulativeDistribution hardMultiplicity;
  std::vector<ParametersWithFixedMultiplicity> softKinematicParameters;
  std::vector<ParametersWithFixedMultiplicity> hardKinematicParameters;
};

using NuclearInteractionParametrisation =
    std::vector<std::pair<float, NuclearInteractionParameters>>;
using MultiParticleNuclearInteractionParametrisation =
    std::vector<std::pair<int, NuclearInteractionParametrisation>>;
}  // namespace detail
}  // namespace ActsFatras