PndCorrDistGenerator.h

Go to the documentation of this file.

//****************************************************************************
//*                   This file is part of PandaRoot.                        *
//*                                                                          *
//*            PandaRoot is distributed under the terms of the               *
//*              GNU General Public License (GPL) version 3,                 *
//*                 copied verbatim in the file "LICENSE".                   *
//*                                                                          *
//*  Copyright (C) 2006 - 2024 FAIR GmbH and copyright holders of PandaRoot  *
//*     The copyright holders are listed in the file "COPYRIGHTHOLDERS".     *
//*               The authors are listed in the file "AUTHORS".              *
//****************************************************************************

// -------------------------------------------------------------------------
// -----               PndCorrDistGenerator header file                -----
// -----               Created 23/05/07  by Aida Galoyan               -----
// -------------------------------------------------------------------------

#ifndef PNDCORRDISTGENERATOR_H
#define PNDCORRDISTGENERATOR_H 1

#include "FairGenerator.h"
#include "THParticle.h"

// using namespace std;

class TClonesArray;
class TFile;
class TH2F;
class TTree;
class TVirtualMCStack;
class FairPrimaryGenerator;

class PndCorrDistGenerator : public FairGenerator {

 public:
  PndCorrDistGenerator();

  PndCorrDistGenerator(const Char_t *fileName);

  virtual ~PndCorrDistGenerator();

  virtual Bool_t ReadEvent(FairPrimaryGenerator *primGen);
  void SetParam() { fParam = kTRUE; };
  void SetThetaRange(Double_t thetLow = 0., Double_t thetHigh = 0.)
  {
    fTheLow = thetLow;
    fTheHigh = thetHigh;
  };

 private:
  Int_t iEvent;             
  const Char_t *fFileName;  
  TFile *fInputFile;        
  TH2F *fInputHist;         
  TClonesArray *fParticles; 
  Int_t fPdgType;
  Bool_t fParam;
  Double_t fTheLow, fTheHigh;

  void CloseInput();

  Double_t MaxBoltDistP(Double_t MeanP);

  Double_t MeanMomentum(Double_t thet);

  ClassDef(PndCorrDistGenerator, 3);
};

#endif