BoundRandomValues.hpp

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file BoundRandomValues.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 <array>

namespace ActsFatras {

struct RectangleRandom {
  double xmax;
  double ymax;

  RectangleRandom(double xmax_, double ymax_) : xmax(xmax_), ymax(ymax_) {}

  Acts::Vector2 operator()(double r0, double r1) const {
    return {(2 * r0 - 1) * xmax, (2 * r1 - 1) * ymax};
  }
};

struct TrapezoidRandom {
  double xminy;
  double xmaxy;
  double ymin;
  double ymax;

  TrapezoidRandom(double xminy_, double xmaxy_, double ymax_)
      : xminy(xminy_), xmaxy(xmaxy_), ymin(-ymax_), ymax(ymax_) {}

  TrapezoidRandom(double xminy_, double xmaxy_, double ymin_, double ymax_)
      : xminy(xminy_), xmaxy(xmaxy_), ymin(ymin_), ymax(ymax_) {}

  Acts::Vector2 operator()(double r0, double r1) const {
    double y = ymin + (ymax - ymin) * r1;
    double xmax = xminy + (xmaxy - xminy) / (ymax - ymin) * (y - ymin);
    double x = (2 * r0 - 1) * xmax;
    return {x, y};
  }
};

struct DiscRandom {
  double rmin;
  double rmax;
  double phimin;
  double phimax;

  DiscRandom(double rmin_, double rmax_, double phimin_, double phimax_)
      : rmin(rmin_), rmax(rmax_), phimin(phimin_), phimax(phimax_) {}

  Acts::Vector2 operator()(double r0, double r1) const {
    double r = rmin + (rmax - rmin) * r0;
    double phi = phimin + (phimax - phimin) * r1;
    return {r * std::cos(phi), r * std::sin(phi)};
  }
};

struct AnnulusRandom {
  double rmin;
  double rmax;
  double phimins;   // strip system
  double phimaxs;   // strip system
  double originxs;  // strip system origin x
  double originys;  // strip system origin y

  AnnulusRandom(double rmin_, double rmax_, double phimins_, double phimaxs_,
                double originxs_, double originys_)
      : rmin(rmin_),
        rmax(rmax_),
        phimins(phimins_),
        phimaxs(phimaxs_),
        originxs(originxs_),
        originys(originys_) {}

  Acts::Vector2 operator()(double r0, double r1) const {
    double r = rmin + (rmax - rmin) * r0;
    double phi = phimins + (phimaxs - phimins) * r1;
    return {r * std::cos(phi), r * std::sin(phi)};
  }
};

}  // namespace ActsFatras