// -------------------------------------------------------------------------
// -----                 CbmDYMuonGenerator header file                 -----
// -----          Created on 27/01/2015 by Parta Pratim Bhaduri      -----
// -------------------------------------------------------------------------

/** CbmDYMuonGenerator.h
 *@author Dr. Partha Pratim Bhaduri <partha.bhaduri@vecc.gov.in>

 *
 The CbmDYMuonGenerator reads the DY output file (ROOT format)
 and inserts the tracks into the FairStack via the FairPrimaryGenerator.
 Derived from FairGenerator.
**/


#ifndef CBM_DYMUONGENERATOR_H
#define CBM_DYMUONGENERATOR_H


#include "FairGenerator.h"

class TClonesArray;
class TFile;
class TTree;
class TBranch;
class TVirtualMCStack;
class FairPrimaryGenerator;



class CbmDYMuonGenerator : public FairGenerator
{

  public:

    /** Default constructor (should not be used) **/
    CbmDYMuonGenerator();


    /** Standard constructor
     ** @param fileName The input (PLUTO) file name
     **/
    CbmDYMuonGenerator(const Char_t* fileName);


    /** Destructor **/
    virtual ~CbmDYMuonGenerator();


    /** Reads on event from the input file and pushes the tracks onto
     ** the stack. Abstract method in base class.
     ** @param primGen  pointer to the FairPrimaryGenerator
     **/
    virtual Bool_t ReadEvent(FairPrimaryGenerator* primGen);



  private:

    Int_t iEvent;      //! Event number
    const Char_t* fFileName;   //! Input file name
    TFile* fInputFile;        //! Pointer to input file
    TTree* fInputTree;        //! Pointer to input tree
    //  TClonesArray* fParticles;  //! Particle array from PLUTO

    TBranch        *b_Run;   //!
    TBranch        *b_Event;   //!
    TBranch        *b_mass;   //!
    TBranch        *b_trackNo;   //!
    TBranch        *b_Px;   //!
    TBranch        *b_Py;   //!
    TBranch        *b_Pz;   //!
    TBranch        *b_En;   //!

    Int_t           Run;
    Int_t           Event;
    Int_t           trackNo;
    Double_t MDY;
    Double_t        Px[2];   //[trackNo]
    Double_t        Py[2];   //[trackNo]
    Double_t        Pz[2];   //[trackNo]
    Double_t        En[2];   //[trackNo]


    /** Private method CloseInput. Just for convenience. Closes the
     ** input file properly. Called from destructor and from ReadEvent. **/
    void CloseInput();

    CbmDYMuonGenerator(const CbmDYMuonGenerator&);
    CbmDYMuonGenerator& operator=(const CbmDYMuonGenerator&);

    ClassDef(CbmDYMuonGenerator,1);

};

#endif