[u/mrichter/AliRoot.git] / HBTAN / AliHBTReaderPPprod.h

#ifndef ALIHBTREADERPPPROD_H
#define ALIHBTREADERPPPROD_H

#include "AliHBTReader.h"
#include "AliHBTReaderTPC.h"

//This reader reads tracks AliTPCtracks.root
//                  particles form tpc_good_tracks 
//I am aware that this file is temporary however we do not have any other PID yet
//Piotr.Skowronski@cern.ch

#include <TString.h>
class TFile;

class AliHBTReaderPPprod: public AliHBTReader
{
  public:
    AliHBTReaderPPprod(const Char_t* trackfilename = "AliTPCtracks.root",
                       const Char_t* clusterfilename = "AliTPCclusters.root", 
                       const Char_t* goodtracksfilename = "good_tracks_tpc",
                       const Char_t* galicefilename = "");

    virtual ~AliHBTReaderPPprod();
    
    Int_t Read(AliHBTRun* particles, AliHBTRun *tracks); //reads tracks and particles and puts them in runs
    
    AliHBTEvent* GetParticleEvent(Int_t); //returns pointer to event with particles 
    AliHBTEvent* GetTrackEvent(Int_t);//returns pointer to event with particles 
    Int_t GetNumberOfPartEvents(); //returns number of particle events
    Int_t GetNumberOfTrackEvents();//returns number of track events
    
  protected:
    //in the future this class is will read global tracking

    
    Int_t OpenFiles(); //opens files to be read
    void CloseFiles(); //close files
    
    AliHBTRun* fParticles; //!simulated particles
    AliHBTRun* fTracks; //!reconstructed tracks (particles)

    TString fTrackFileName; //name of the file with tracks
    TString fClusterFileName;//name of the file with clusters
    TString fGAliceFileName;//name of the file with galice.root
    TString fGoodTPCTracksFileName; //name of text file with good tracks
    
    TFile *fTracksFile; //file with tracks
    TFile *fClustersFile;//file with clusters
    
    Bool_t fIsRead; //flag indicating if the data are already read
    
    
  private:
  public:
    ClassDef(AliHBTReaderPPprod,1)
};


class AliGoodTracksPP
 { 
   //container for good tracks
   //this class is for internal use only
   
   friend class AliHBTReaderPPprod;
   
   private:
     AliGoodTracksPP(const TString& infilename = TString("good_tracks_tpc")); //constructor
     ~AliGoodTracksPP(); //dctor
   
     const GoodTrack& GetTrack(Int_t event, Int_t n) const; //returns reference to the nth good track in event "event"

     Int_t  fNevents;  //Number of events
     Int_t* fGoodInEvent; //Numbers of good track in event
     struct GoodTrack **fData;
 };


#endif
Commit	Line	Data
1b446896	1	#ifndef ALIHBTREADERPPPROD_H
	2	#define ALIHBTREADERPPPROD_H
	3
	4	#include "AliHBTReader.h"
	5	#include "AliHBTReaderTPC.h"
	6
	7	//This reader reads tracks AliTPCtracks.root
	8	// particles form tpc_good_tracks
	9	//I am aware that this file is temporary however we do not have any other PID yet
	10	//Piotr.Skowronski@cern.ch
	11
	12	#include <TString.h>
	13	class TFile;
	14
	15	class AliHBTReaderPPprod: public AliHBTReader
	16	{
	17	public:
	18	AliHBTReaderPPprod(const Char_t* trackfilename = "AliTPCtracks.root",
	19	const Char_t* clusterfilename = "AliTPCclusters.root",
	20	const Char_t* goodtracksfilename = "good_tracks_tpc",
	21	const Char_t* galicefilename = "");
	22
	23	virtual ~AliHBTReaderPPprod();
	24
	25	Int_t Read(AliHBTRun* particles, AliHBTRun *tracks); //reads tracks and particles and puts them in runs
	26
	27	AliHBTEvent* GetParticleEvent(Int_t); //returns pointer to event with particles
	28	AliHBTEvent* GetTrackEvent(Int_t);//returns pointer to event with particles
	29	Int_t GetNumberOfPartEvents(); //returns number of particle events
	30	Int_t GetNumberOfTrackEvents();//returns number of track events
	31
	32	protected:
	33	//in the future this class is will read global tracking
	34
	35
	36	Int_t OpenFiles(); //opens files to be read
	37	void CloseFiles(); //close files
	38
	39	AliHBTRun* fParticles; //!simulated particles
	40	AliHBTRun* fTracks; //!reconstructed tracks (particles)
	41
	42	TString fTrackFileName; //name of the file with tracks
	43	TString fClusterFileName;//name of the file with clusters
	44	TString fGAliceFileName;//name of the file with galice.root
	45	TString fGoodTPCTracksFileName; //name of text file with good tracks
	46
	47	TFile *fTracksFile; //file with tracks
	48	TFile *fClustersFile;//file with clusters
	49
	50	Bool_t fIsRead; //flag indicating if the data are already read
	51
	52
	53	private:
	54	public:
	55	ClassDef(AliHBTReaderPPprod,1)
	56	};
	57
	58
	59	class AliGoodTracksPP
	60	{
	61	//container for good tracks
	62	//this class is for internal use only
	63
	64	friend class AliHBTReaderPPprod;
65
66	private:
67	AliGoodTracksPP(const TString& infilename = TString("good_tracks_tpc")); //constructor
68	~AliGoodTracksPP(); //dctor
69
70	const GoodTrack& GetTrack(Int_t event, Int_t n) const; //returns reference to the nth good track in event "event"
71
72	Int_t fNevents; //Number of events
73	Int_t* fGoodInEvent; //Numbers of good track in event
74	struct GoodTrack **fData;
75	};
76
77
78	#endif