CosmicsGenerator.h

Go to the documentation of this file.

// SPDX-License-Identifier: LGPL-3.0-or-later
// SPDX-FileCopyrightText: Copyright CERN for the benefit of the SHiP
// Collaboration
 
// -------------------------------------------------------------------
// -----       CosmicsGenerator source file for SHiP             -----
// -----      Version as of 15.09.15  by Martin Franke           -----
// -----       mailto: mfranke(at)physik.hu-berlin.de            -----
// -------------------------------------------------------------------
 
#ifndef SHIPGEN_COSMICSGENERATOR_H_
#define SHIPGEN_COSMICSGENERATOR_H_
 
#include "Generator.h"
#include "TF1.h"
#include "TH1.h"
#include "TMath.h"
#include "TROOT.h"
#include "TRandom3.h"
 
class FairPrimaryGenerator;
 
class Co3Rng {
 public:
  Co3Rng() {
    rng = new TRandom3(gRandom->GetSeed());
    fTheta = new TF1("f2", "cos(x)*cos(x)", 0, TMath::Pi() / 2);
    fTheta->SetNpx(10);
    fSpectrumH =
        new TF1("f4", "1400*TMath::Power(x,-2.7)*(1/(1+x/115)+0.054/(1+x/850))",
                100, 1000);  // momentum above 100GeV
  };
  virtual ~Co3Rng() {
    delete rng;
    delete fTheta;
    delete fSpectrumH;
  };
  double Uniform(Float_t min, Float_t max) { return rng->Uniform(min, max); };
  TF1* fSpectrumH;
  TF1* fTheta;
  double fSpectrumL(double theta, double minE,
                    Bool_t generateP);  // momentum below 100GeV
 private:
  TRandom3* rng;  
};
 
class CosmicsGenerator : public SHiP::Generator {
 public:
  CosmicsGenerator() {};
  ~CosmicsGenerator() override {
    delete fRandomEngine;
    std::cout << nInside << " events have been generated." << std::endl;
    std::cout << "There is a total of " << nInside << "/" << nTest
              << " muons that passed close enough to the detector."
              << std::endl;
    std::cout << "Including the given weight this corresponds to ";
    std::cout << weighttest / FluxIntegral / xdist / zdist * 10000
              << " spills (1 spill = " << xdist * zdist * FluxIntegral / 10000;
    std::cout << " real cosmic muons = " << n_EVENTS << " simulated events)."
              << std::endl;
  };
 
  using SHiP::Generator::Init;
  Bool_t ReadEvent(FairPrimaryGenerator*) override;
  //  virtual Bool_t Init(); //!
  virtual Bool_t Init(Bool_t largeMom);
  Bool_t Init(const char* inFile) override { return Init(inFile, 0); };
 
  Bool_t Init(const char* inFile, int startEvent) override {
    LOG(warning) << "Init with files not implemented for CosmicsGenerator.";
    return false;
  };
 
  double z0, yBox, xBox, zBox, xdist, zdist, minE;
  int n_EVENTS;
 
 private:
  Co3Rng* fRandomEngine;  
 
 protected:
  double P, px, py, pz, x, y, z, weighttest, weight, mass, FluxIntegral, theta;
  int PID, nInside, nTest;  
  Bool_t high;
 
  void GenerateDynamics();
  Bool_t DetectorBox();
};
 
#endif  // SHIPGEN_COSMICSGENERATOR_H_ /* !PNDCoGENERATOR_H */