[u/mrichter/AliRoot.git] / PWG2 / FEMTOSCOPY / AliFemto / AliTwoTrackRes.h

////////////////////////////////////////////////////////////////////////////////
//345678901234567890123456789012345678901234567890123456789012345678901234567890
//       1         2         3         4         5         6         7         8
//
// Class AliTwoTrackRes
// J. Mercado <mercado@physi.uni-heidelberg.de> Last modified: 20.01.2011
//
////////////////////////////////////////////////////////////////////////////////

#ifndef ALITWOTRACKRES_H
#define ALITWOTRACKRES_H

#include <TNtuple.h>
#include <TObject.h>
#include <TFile.h>
#include <TTree.h>
#include "TMath.h"
#include "TBits.h"
#include "TVector3.h"
#include "TLorentzVector.h"
#include "AliAnalysisTask.h"
#include "AliESD.h"
#include "AliESDEvent.h"
#include "AliESDtrack.h"
#include "AliESDtrackCuts.h"
#include "AliExternalTrackParam.h"

//______________________________________________________________________________
class AliTwoTrackRes : public AliAnalysisTask {

public:
  
  AliTwoTrackRes() : AliAnalysisTask("",""), fChain(0), fESDEvent(0), fOutContainer(0), 
    fTrackCuts(0), fNTuple1(0), fNTuple2(0), fP1(), fP2(), fPb1(), fPb2(), fP(), 
    fQ(), fTpcEnt1(), fTpcEnt2(), fTpcDist(), fOutFilename() {}
  AliTwoTrackRes(const char *name);
  AliTwoTrackRes(const AliTwoTrackRes& aTwoTrackRes);
  virtual ~AliTwoTrackRes();

  AliTwoTrackRes& operator=(const AliTwoTrackRes& aTwoTrackRes);

  virtual void ConnectInputData(Option_t *);
  virtual void CreateOutputObjects();
  virtual void Exec(Option_t *opt = "");
  virtual void Terminate(Option_t *opt = "");

  void   SetTr1(double pt1, double eta1, double phi1, double m);
  void   SetTr2(double pt2, double eta2, double phi2, double m);
  void   SetTpcEnt1(double x1, double y1, double z1);
  void   SetTpcEnt2(double x2, double y2, double z2);
  double Qinv2()         {fQ = fP2 - fP1; return -1.*fQ.M2();}
  double Qinv()          {return TMath::Sqrt(TMath::Abs(Qinv2()));}
  double MinDist(AliExternalTrackParam *trk1, AliExternalTrackParam *trk2);
  int    GetNSha(TBits cl, TBits sh);
  double Corr(TBits cl1, TBits cl2, TBits sh1, TBits sh2);
  double Qfac(TBits cl1, TBits cl2, TBits sh1, TBits sh2);
  double RotTr2Phi();
  double Dist()    {fTpcDist = fTpcEnt2 - fTpcEnt1; return fTpcDist.Mag();}
  double DEta()    const {return TMath::Abs(fP2.Eta()-fP1.Eta());}
  double DTheta()  const {return fP2.Theta()-fP1.Theta();}
  double DPhi()    const {return TVector2::Phi_mpi_pi(fP2.Phi()-fP1.Phi());}
  void   NoSwap()        {fPb1 = fP1; fPb2 = fP2;}
  void   Swap()          {fPb1 = fP2; fPb2 = fP1;}
  void   FillNTuple1(double minsep, double sep, double corr, double qf, 
		     int ns1, int ns2);
  void   FillNTuple2(double minsep, double sep, double corr, double qf, 
		     int ns1, int ns2);
  void   SetOutfile(char *outfil) {fOutFilename = outfil;}

private:

  TTree           *fChain;        // Chain of ESD trees
  AliESDEvent     *fESDEvent;     // Leaves
  TObjArray       *fOutContainer; // Output data container
  AliESDtrackCuts *fTrackCuts;    // Track cuts
  TNtuple         *fNTuple1;      // True pairs
  TNtuple         *fNTuple2;      // Mixed pairs
  TLorentzVector  fP1;            // Four-momentum of track 1 (lab)
  TLorentzVector  fP2;            // Four-momentum of track 2 (lab)
  TLorentzVector  fPb1;           // Buffer single-track 1 for swapping
  TLorentzVector  fPb2;           // Buffer single-track 2 for swapping
  TLorentzVector  fP;             // Total four-momentum (lab)
  TLorentzVector  fQ;             // Four-momentum difference (lab)
  TVector3        fTpcEnt1;       // Nominal TPC entrance point track 1
  TVector3        fTpcEnt2;       // Nominal TPC entrance point track 2
  TVector3        fTpcDist;       // Nominal TPC entrance separation 
  TString         fOutFilename;   // Output filename

  ClassDef(AliTwoTrackRes, 0);
};
#endif

//______________________________________________________________________________
// EOF
Commit	Line	Data
bd82160e	1	////////////////////////////////////////////////////////////////////////////////
	2	//345678901234567890123456789012345678901234567890123456789012345678901234567890
	3	// 1 2 3 4 5 6 7 8
	4	//
1b35c600	5	// Class AliTwoTrackRes
1b35c600	6	// J. Mercado <mercado@physi.uni-heidelberg.de> Last modified: 20.01.2011
bd82160e	7	//
	8	////////////////////////////////////////////////////////////////////////////////
	9
	10	#ifndef ALITWOTRACKRES_H
	11	#define ALITWOTRACKRES_H
	12
	13	#include <TNtuple.h>
	14	#include <TObject.h>
	15	#include <TFile.h>
	16	#include <TTree.h>
	17	#include "TMath.h"
	18	#include "TBits.h"
	19	#include "TVector3.h"
	20	#include "TLorentzVector.h"
	21	#include "AliAnalysisTask.h"
	22	#include "AliESD.h"
	23	#include "AliESDEvent.h"
	24	#include "AliESDtrack.h"
	25	#include "AliESDtrackCuts.h"
	26	#include "AliExternalTrackParam.h"
	27
	28	//______________________________________________________________________________
	29	class AliTwoTrackRes : public AliAnalysisTask {
	30
	31	public:
	32
	33	AliTwoTrackRes() : AliAnalysisTask("",""), fChain(0), fESDEvent(0), fOutContainer(0),
	34	fTrackCuts(0), fNTuple1(0), fNTuple2(0), fP1(), fP2(), fPb1(), fPb2(), fP(),
1b35c600	35	fQ(), fTpcEnt1(), fTpcEnt2(), fTpcDist(), fOutFilename() {}
bd82160e	36	AliTwoTrackRes(const char *name);
	37	AliTwoTrackRes(const AliTwoTrackRes& aTwoTrackRes);
	38	virtual ~AliTwoTrackRes();
	39
	40	AliTwoTrackRes& operator=(const AliTwoTrackRes& aTwoTrackRes);
	41
	42	virtual void ConnectInputData(Option_t *);
	43	virtual void CreateOutputObjects();
	44	virtual void Exec(Option_t *opt = "");
	45	virtual void Terminate(Option_t *opt = "");
	46
	47	void SetTr1(double pt1, double eta1, double phi1, double m);
	48	void SetTr2(double pt2, double eta2, double phi2, double m);
	49	void SetTpcEnt1(double x1, double y1, double z1);
	50	void SetTpcEnt2(double x2, double y2, double z2);
	51	double Qinv2() {fQ = fP2 - fP1; return -1.*fQ.M2();}
	52	double Qinv() {return TMath::Sqrt(TMath::Abs(Qinv2()));}
	53	double MinDist(AliExternalTrackParam trk1, AliExternalTrackParam trk2);
	54	int GetNSha(TBits cl, TBits sh);
	55	double Corr(TBits cl1, TBits cl2, TBits sh1, TBits sh2);
	56	double Qfac(TBits cl1, TBits cl2, TBits sh1, TBits sh2);
	57	double RotTr2Phi();
	58	double Dist() {fTpcDist = fTpcEnt2 - fTpcEnt1; return fTpcDist.Mag();}
	59	double DEta() const {return TMath::Abs(fP2.Eta()-fP1.Eta());}
	60	double DTheta() const {return fP2.Theta()-fP1.Theta();}
	61	double DPhi() const {return TVector2::Phi_mpi_pi(fP2.Phi()-fP1.Phi());}
	62	void NoSwap() {fPb1 = fP1; fPb2 = fP2;}
	63	void Swap() {fPb1 = fP2; fPb2 = fP1;}
	64	void FillNTuple1(double minsep, double sep, double corr, double qf,
	65	int ns1, int ns2);
	66	void FillNTuple2(double minsep, double sep, double corr, double qf,
	67	int ns1, int ns2);
1b35c600	68	void SetOutfile(char *outfil) {fOutFilename = outfil;}
bd82160e	69
	70	private:
	71
	72	TTree *fChain; // Chain of ESD trees
	73	AliESDEvent *fESDEvent; // Leaves
	74	TObjArray *fOutContainer; // Output data container
	75	AliESDtrackCuts *fTrackCuts; // Track cuts
	76	TNtuple *fNTuple1; // True pairs
	77	TNtuple *fNTuple2; // Mixed pairs
	78	TLorentzVector fP1; // Four-momentum of track 1 (lab)
	79	TLorentzVector fP2; // Four-momentum of track 2 (lab)
	80	TLorentzVector fPb1; // Buffer single-track 1 for swapping
	81	TLorentzVector fPb2; // Buffer single-track 2 for swapping
	82	TLorentzVector fP; // Total four-momentum (lab)
	83	TLorentzVector fQ; // Four-momentum difference (lab)
	84	TVector3 fTpcEnt1; // Nominal TPC entrance point track 1
	85	TVector3 fTpcEnt2; // Nominal TPC entrance point track 2
	86	TVector3 fTpcDist; // Nominal TPC entrance separation
1b35c600	87	TString fOutFilename; // Output filename
bd82160e	88
	89	ClassDef(AliTwoTrackRes, 0);
	90	};
	91	#endif
	92
	93	//______________________________________________________________________________
	94	// EOF
	95
1b35c600	96