CRY.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 <framework/gearbox/Unit.h>
 
#include <cry/CRYGenerator.h>
#include <cry/CRYSetup.h>
#include <VecGeom/volumes/UnplacedVolume.h>
 
namespace Belle2 {
  class MCParticleGraph;
  class CRY final {
  public:
 
    CRY() = default;
 
    ~CRY() = default;
 
    void init();
 
    void generateEvent(MCParticleGraph& mcGraph);
 
    void setCosmicDataDir(const std::string& cosmicdatadir) { m_cosmicDataDir = cosmicdatadir; }
 
    void setAcceptance(const std::vector<double>& size) { m_acceptSize = size; }
 
    void setTimeOffset(double timeoffset) { m_timeOffset = timeoffset; }
 
    void setKineticEnergyThreshold(double kineticenergythreshold) { m_kineticEnergyThreshold = kineticenergythreshold; }
 
    void setDate(const std::string& date) { m_date = date; }
 
    void setMaxTrials(int maxtrials) { m_maxTrials = maxtrials; }
 
    void setBoxLength(double length) { m_subboxLength = length / Unit::m; }
 
    void setReturnGammas(bool gammas) { m_returnGammas = gammas; }
 
    void setReturnKaons(bool kaons) { m_returnKaons = kaons; }
 
    void setReturnPions(bool pions) { m_returnPions = pions; }
 
    void setReturnProtons(bool protons) { m_returnProtons = protons; }
 
    void setReturnNeutrons(bool neutrons) { m_returnNeutrons = neutrons; }
 
    void setReturnElectrons(bool electrons) { m_returnElectrons = electrons; }
 
    void setReturnMuons(bool muons) { m_returnMuons = muons; }
 
    void term();
 
  protected:
    std::string m_cosmicDataDir;  
    int m_subboxLength{100}; 
    std::vector<double> m_acceptSize; 
    double m_timeOffset{0};  
    double m_kineticEnergyThreshold{0};  
    std::string m_date{"1-1-2019"}; 
    int m_maxTrials{0};   
    int m_totalTrials{0}; 
    bool m_returnGammas{true}; 
    bool m_returnKaons{true}; 
    bool m_returnPions{true}; 
    bool m_returnProtons{true}; 
    bool m_returnNeutrons{true}; 
    bool m_returnElectrons{true}; 
    bool m_returnMuons{true}; 
    std::unique_ptr<CRYSetup> m_crySetup;   
    std::unique_ptr<CRYGenerator> m_cryGenerator;   
    std::unique_ptr<vecgeom::VUnplacedVolume> m_world; 
    std::unique_ptr<vecgeom::VUnplacedVolume> m_acceptance; 
  };
} // end namespace Belle2