MuDISGenerator.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
 
#ifndef SHIPGEN_MUDISGENERATOR_H_
#define SHIPGEN_MUDISGENERATOR_H_
 
#include "FairLogger.h"  // for FairLogger, MESSAGE_ORIGIN
#include "Generator.h"
#include "TClonesArray.h"
#include "TF1.h"  // for TF1
#include "TROOT.h"
#include "TTree.h"  // for TTree
#include "TVector3.h"
#include "vector"
 
class FairPrimaryGenerator;
 
class MuDISGenerator : public SHiP::Generator {
 public:
  MuDISGenerator();
 
  ~MuDISGenerator() override;
 
  using SHiP::Generator::Init;
  Bool_t ReadEvent(FairPrimaryGenerator*) override;
  Bool_t Init(const char*, int) override;
  Bool_t Init(const char*) override;
  Int_t GetNevents();
 
  void SetPositions(Double_t z_start, Double_t z_end) {
    startZ = z_start;
    endZ = z_end;
  }
 
 protected:
  Double_t startZ, endZ;
  TClonesArray* iMuon;
  TClonesArray* dPart;
  TClonesArray* dPartSoft;
  FairLogger* fLogger;  
  TFile* fInputFile;
  TTree* fTree;
  int fNevents;
  int fn;
  bool fFirst;
};
#endif  // SHIPGEN_MUDISGENERATOR_H_