ParticleInjectorOG/ParticleInjector.h

Go to the documentation of this file.

#ifndef _ParticleInjectorOG_ParticleInjector_h_
#define _ParticleInjectorOG_ParticleInjector_h_

#include <fwk/VModule.h>

#include <utl/Particle.h>

#include <string>
#include <vector>


class TF1;
namespace sdet {
  class Station; 
}
namespace evt {
  class Event;
}
namespace sevt {
  class SEvent;
}
namespace utl {
  class RandomEngine;
}

namespace ParticleInjectorOG {

  class ParticleInjector : public fwk::VModule {

  public:
    ParticleInjector();
    virtual ~ParticleInjector();

    fwk::VModule::ResultFlag Init();
    fwk::VModule::ResultFlag Run(evt::Event& theEvent);
    fwk::VModule::ResultFlag Finish();

  private:
    // switches
    bool fUseSingleTank;
    bool fUseDiscreteDirection;
    bool fUseDiscreteSpectrum;
    bool fUseInjectAllParticles;
    bool fUseSinglePosition;
    bool fUseDiscreteTime;

    unsigned int fNumberOfParticles;
    unsigned int fSingleTankID;

    double fX, fY, fZ;

    std::vector<double> fDiscreteAzimuth;
    std::vector<double> fDiscreteZenith;
    std::vector<double> fDiscreteParticleTime;
    std::vector<double> fDiscreteMuonFlux;
    std::vector<double> fDiscreteElectronFlux;
    std::vector<double> fDiscretePhotonFlux;
    std::vector<double> fDiscreteMuonEnergy;
    std::vector<double> fDiscreteElectronEnergy;
    std::vector<double> fDiscretePhotonEnergy;

    std::string fContinuousZenithString;
    std::string fContinuousAzimuthString;
    std::string fContinuousTimeString;
    std::string fMuonSpectrumString;
    std::string fElectronSpectrumString;
    std::string fPhotonSpectrumString;

    TF1* fContinuousZenithDistribution;
    TF1* fContinuousAzimuthDistribution;
    TF1* fTimeSpectrum;
    TF1* fMuonSpectrum;
    TF1* fElectronSpectrum;
    TF1* fPhotonSpectrum;

    double fMuonEnergyMin;
    double fMuonEnergyMax;
    double fElectronEnergyMin;
    double fElectronEnergyMax;
    double fPhotonEnergyMin;
    double fPhotonEnergyMax;
    double fMuonSpectrumMax;
    double fElectronSpectrumMax;
    double fPhotonSpectrumMax;

    utl::Particle::Type fParticleType;

    utl::RandomEngine* fRandomEngine;

    void InjectParticles();

    double GetAzimuth();
    double GetZenith();
    double GetEnergy();
    double GetTime();

    void SetPosition(double& x, double& y, double& z);
    double GetDiscreteEnergy(std::vector<double>& flux, 
                             std::vector<double>& energy,
                             double max);

    const sdet::Station* fCurrentDetectorStation;
    sevt::SEvent* fSEvent;

    REGISTER_MODULE("ParticleInjectorOG", ParticleInjector);

  };

}


#endif // _ParticleInjectorOG_ParticleInjector_h_ 

// Configure (x)emacs for this file ...
// Local Variables:
// mode: c++
// compile-command: "make -C .. -k"
// End: