ParticleGun.h

/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/
 
#pragma once
 
#include <mdst/dataobjects/MCParticleGraph.h>
 
#include <vector>
 
namespace Belle2 {
  class ParticleGun {
  public:
    enum EDistribution {
      c_fixedValue,
      c_uniformDistribution,
      c_uniformPtDistribution,
      c_uniformCosDistribution,
      c_uniformLogDistribution,
      c_uniformLogPtDistribution,
      c_normalDistribution,
      c_normalPtDistribution,
      c_normalCosDistribution,
      c_discreteSpectrum,
      c_discretePtSpectrum,
      c_inversePtDistribution,
      c_polylineDistribution,
      c_polylinePtDistribution,
      c_polylineCosDistribution
    };
 
    struct Parameters {
      EDistribution momentumDist = c_fixedValue;
      EDistribution phiDist = c_fixedValue;
      EDistribution thetaDist = c_fixedValue;
      EDistribution xVertexDist = c_fixedValue;
      EDistribution yVertexDist = c_fixedValue;
      EDistribution zVertexDist = c_fixedValue;
      EDistribution timeDist = c_fixedValue;
      double nTracks;
      std::vector<int> pdgCodes = {};
      std::vector<double> momentumParams = {0};
      std::vector<double> phiParams = {0};
      std::vector<double> thetaParams = {0};
      std::vector<double> xVertexParams = {0};
      std::vector<double> yVertexParams = {0};
      std::vector<double> zVertexParams = {0};
      std::vector<double> timeParams = {0};
 
      bool independentVertices = false;
      bool varyNumberOfTracks = true;
      bool fixedMomentumPerEvent = false;
 
    };
 
    ParticleGun(): m_params() {}
 
    ~ParticleGun() {}
 
    bool generateEvent(MCParticleGraph& graph);
 
    bool setParameters(const Parameters& parameters);
 
  protected:
    double generateValue(EDistribution dist, const std::vector<double>& params);
 
    Parameters m_params;
  };
 
} //end namespace Belle2