AAFHInterface.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
#ifndef GENERATORS_AAFH_INTERFACE_H
#define GENERATORS_AAFH_INTERFACE_H
 
#include <vector>
#include <mdst/dataobjects/MCParticleGraph.h>
 
 
namespace Belle2 {
  class AAFHInterface {
  public:
    enum EMode {
      c_MuonParticle = 1,
      c_MuonMuon = 2,
      c_ElectonMuon = 3,
      c_ElectronParticle = 4,
      c_ElectonElectron = 5,
    };
 
    enum ERejection {
      c_Once = 1,
      c_Twice = 2,
    };
 
    AAFHInterface() = default;
    void setMaxTries(int tries);
    void setGeneratorWeights(const std::vector<double>& weights);
    void setMaxWeights(double subgeneratorWeight, double finalWeight);
    void setSupressionLimits(std::vector<double> limits);
    void setMinimumMass(double minMass) { m_minMass = minMass; }
    void setParticle(const std::string& particle)
    {
      m_particle = particle;
    }
 
    int getMaxTries() const;
    std::vector<double> getGeneratorWeights() const;
    double getMaxSubGeneratorWeight() const;
    double getMaxFinalWeight() const;
    std::vector<double> getSuppressionLimits() const;
    double getMinimumMass() const { return m_minMass; }
    std::string getParticle() const { return m_particle; }
    double getTotalCrossSection() const;
    double getTotalCrossSectionError() const;
 
    void initialize(double beamEnergy, EMode mode, ERejection rejection);
    void generateEvent(MCParticleGraph& mpg);
    void finish();
  private:
    void updateParticle(MCParticleGraph::GraphParticle& p, double* q, int pdg, bool isInitial = false);
    double m_minMass {0};
    std::string m_particle {"tau-"};
    int m_particlePDG {0};
  };
 
} //Belle2 namespace
#endif // GENERATORS_AAFH_INTERFACE_H