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 <Math/Transform3D.h>
#include <Math/Vector3D.h>
#include <Math/Point3D.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 event() override;
 
  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;     
    ROOT::Math::Transform3D m_transform; 
    TF1* m_customDistribution = 0; 
    // data store objects
    StoreArray<MCParticle> m_MCParticles; 
    StoreArray<TOPSimCalPulse> m_simCalPulses; 
    bool isInsideSlit(const ROOT::Math::XYZPoint& point, const ROOT::Math::XYZVector& direction) const;
 
    ROOT::Math::XYZVector getDirectionGaussian() const;
 
    ROOT::Math::XYZVector getDirectionUniform() const;
 
    ROOT::Math::XYZVector getDirectionLambertian() const;
 
    ROOT::Math::XYZVector getDirectionUniformRingAngle() const;
 
    ROOT::Math::XYZVector getDirectionCustom() const;
 
  };
 
} // Belle2 namespace