OpticalGunModule.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/core/Module.h>
#include <string>
#include <TVector3.h>
#include <TRotation.h>
#include <TF1.h>
 
// DataStore
#include <framework/datastore/StoreArray.h>
#include <mdst/dataobjects/MCParticle.h>
#include <top/dataobjects/TOPSimCalPulse.h>
 
namespace Belle2 {
  class OpticalGunModule : public Module {
 
  public:
 
    OpticalGunModule();
 
    virtual ~OpticalGunModule();
 
    virtual void initialize() override;
 
    virtual void beginRun() override;
 
    virtual void event() override;
 
    virtual void endRun() override;
 
    virtual void terminate() override;
 
    void printModuleParams() const;
 
  private:
 
    // steering parameters
    double m_x = 0;          
    double m_y = 0;          
    double m_z = 0;          
    double m_diameter = 0;   
    double m_maxAlpha = 0;     
    double m_minAlpha = 0;      
    double m_na = 0;         
    double m_wavelength = 0; 
    double m_phi = 0;        
    double m_theta = 0;      
    double m_psi = 0;        
    double m_startTime = 0;  
    double m_pulseWidth = 0; 
    double m_numPhotons = 0; 
    int m_slotID = 0;         
    double m_slitDX = 0;     
    double m_slitDY = 0;     
    double m_slitX0 = 0;     
    double m_slitY0 = 0;     
    double m_slitZ = 0;      
    std::string
    m_angularDistribution; 
    // others
    double m_cosMinAlpha =
      0;   
    double m_cosMaxAlpha =
      0;   
    double m_energy = 0;     
    TVector3 m_translate;  
    TRotation m_rotate;    
    TRotation m_rotateBar; 
    TF1* m_customDistribution = 0; 
    // data store objects
    StoreArray<MCParticle> m_MCParticles; 
    StoreArray<TOPSimCalPulse> m_simCalPulses; 
    bool isInsideSlit(const TVector3& point, const TVector3& direction) const;
 
    TVector3 getDirectionGaussian() const;
 
    TVector3 getDirectionUniform() const;
 
    TVector3 getDirectionLambertian() const;
 
    TVector3 getDirectionUniformRingAngle() const;
 
    TVector3 getDirectionCustom() const;
 
  };
 
} // Belle2 namespace