[u/mrichter/AliRoot.git] / PWGCF / FEMTOSCOPY / AliFemto / AliFemtoCutMonitorParticlePID.cxx

////////////////////////////////////////////////////////////////////////////////
//                                                                            //
// AliFemtoCutMonitorParticlePID - the cut monitor for particles to study     //
// various aspects of the PID determination                                   //
//                                                                            //
////////////////////////////////////////////////////////////////////////////////
#include "AliFemtoCutMonitorParticlePID.h"
#include "AliFemtoModelHiddenInfo.h"
#include <TH1D.h>
#include <TH2D.h>
#include <TList.h>
#include <TMath.h>

AliFemtoCutMonitorParticlePID::AliFemtoCutMonitorParticlePID():
  fTPCdEdx(0),
  fTOFParticle(0),
  fTOFTime(0x0),
    fTOFNSigma(0),
    fTPCNSigma(0),
    fTPCTOFNSigma(0)
{
  // Default constructor
    fTPCdEdx =  new TH2D("TPCdEdx", "TPC dEdx vs. momentum", 100, 0.0, 5.0, 250, 0.0, 500.0);
    fTOFTime = new TH2D("TOFTime", "TOF Time vs. momentum", 100, 0.1, 5.0, 400, -4000.0, 4000.0);
    fTOFNSigma = new TH2D("TOFNSigma","TOF NSigma vs. momentum", 100, 0.0, 5.0, 100, -5.0, 5.0);
    fTPCNSigma = new TH2D("TPCNSigma","TPC NSigma vs. momentum", 100, 0.0, 5.0, 100, -5.0, 5.0);
    fTPCTOFNSigma = new TH2D("TPCTOFNSigma","TPC & TOF NSigma vs. momentum", 100, 0.0, 5.0, 100, 0.0, 10.0);

}

AliFemtoCutMonitorParticlePID::AliFemtoCutMonitorParticlePID(const char *aName, Int_t aTOFParticle):
  AliFemtoCutMonitor(),
  fTPCdEdx(0),
  fTOFParticle(aTOFParticle),
  fTOFTime(0x0),
    fTOFNSigma(0),
    fTPCNSigma(0),
    fTPCTOFNSigma(0)
{
  // Normal constructor
  char name[200];
  snprintf(name, 200, "TPCdEdx%s", aName);
    // fTPCdEdx = new TH2D(name, "TPC dEdx vs. momentum", 100, 0.0, 6.0, 250, 0.0, 500.0);
    fTPCdEdx = new TH2D(name, "TPC dEdx vs. momentum", 200, 0.1, 4.0, 250, 0.0, 500.0);

  snprintf(name, 200, "TOFTime%s", aName);
    fTOFTime = new TH2D(name, "TOF Time vs. momentum", 100, 0.1, 5.0, 400, -4000.0, 4000.0);

    snprintf(name, 200, "TOFNSigma%s", aName);
    fTOFNSigma = new TH2D(name,"TOF NSigma vs. momentum", 100, 0.0, 5.0, 100, -5.0, 5.0);

    snprintf(name, 200, "TPCNSigma%s", aName);
    fTPCNSigma = new TH2D(name,"TPC NSigma vs. momentum", 100, 0.0, 5.0, 100, -5.0, 5.0);

    snprintf(name, 200, "TPCTOFNSigma%s", aName);
    fTPCTOFNSigma = new TH2D(name,"TPC & TOF NSigma vs. momentum", 100, 0.0, 5.0, 100, 0.0, 10.0);

}

AliFemtoCutMonitorParticlePID::AliFemtoCutMonitorParticlePID(const AliFemtoCutMonitorParticlePID &aCut):
  AliFemtoCutMonitor(),
  fTPCdEdx(0),
  fTOFParticle(0),
  fTOFTime(0x0),
    fTOFNSigma(0),
    fTPCNSigma(0),
    fTPCTOFNSigma(0)

{
  // copy constructor
  if (fTPCdEdx) delete fTPCdEdx;
  fTPCdEdx = new TH2D(*aCut.fTPCdEdx);

  if (fTOFTime) delete fTOFTime;
  fTOFTime = new TH2D(*aCut.fTOFTime);

    if (fTOFNSigma) delete fTOFNSigma;
    fTOFNSigma= new TH2D(*aCut.fTOFNSigma);

    if (fTPCNSigma) delete fTPCNSigma;
    fTPCNSigma= new TH2D(*aCut.fTPCNSigma);

    if (fTPCTOFNSigma) delete fTPCTOFNSigma;
    fTPCTOFNSigma= new TH2D(*aCut.fTPCTOFNSigma);
}

AliFemtoCutMonitorParticlePID::~AliFemtoCutMonitorParticlePID()
{
  // Destructor
  delete fTPCdEdx;
  delete fTOFTime;
    delete fTOFNSigma;
    delete fTPCNSigma;
    delete fTPCTOFNSigma;

}

AliFemtoCutMonitorParticlePID& AliFemtoCutMonitorParticlePID::operator=(const AliFemtoCutMonitorParticlePID& aCut)
{
  // assignment operator
  if (this == &aCut)
    return *this;

  if (fTPCdEdx) delete fTPCdEdx;
  fTPCdEdx = new TH2D(*aCut.fTPCdEdx);

  if (fTOFTime) delete fTOFTime;
  fTOFTime = new TH2D(*aCut.fTOFTime);

    if(fTOFNSigma) delete fTOFNSigma;
    fTOFNSigma = new TH2D(*aCut.fTOFNSigma);

    if(fTPCNSigma) delete fTPCNSigma;
    fTPCNSigma = new TH2D(*aCut.fTPCNSigma);

    if(fTPCTOFNSigma) delete fTPCTOFNSigma;
    fTPCTOFNSigma = new TH2D(*aCut.fTPCTOFNSigma);

  return *this;
}

AliFemtoString AliFemtoCutMonitorParticlePID::Report(){
  // Prepare report from the execution
  string stemp = "*** AliFemtoCutMonitorParticlePID report";
  AliFemtoString returnThis = stemp;
  return returnThis;
}

void AliFemtoCutMonitorParticlePID::Fill(const AliFemtoTrack* aTrack)
{
  // Fill in the monitor histograms with the values from the current track
  float tMom = aTrack->P().Mag();
  float tdEdx = aTrack->TPCsignal();
  float tTOF = 0.0;
  //  short tchg = aTrack->Charge();
  if (fTOFParticle == 0) tTOF = aTrack->TOFpionTime();
  if (fTOFParticle == 1) tTOF = aTrack->TOFkaonTime();
  if (fTOFParticle == 2) tTOF = aTrack->TOFprotonTime();

  fTPCdEdx->Fill(tMom, tdEdx);
  fTOFTime->Fill(tMom, tTOF);

  //  float vp= aTrack->VTOF();
  //     if (vp > 0.) {
  //         fTOFTime->Fill(tMom, tTOF);
  //         if (fTOFParticle == 0) fTOFNSigma->Fill(tMom, aTrack->NSigmaTOFPi());
  //         if (fTOFParticle == 1) fTOFNSigma->Fill(tMom, aTrack->NSigmaTOFK());
  //         if (fTOFParticle == 2) fTOFNSigma->Fill(tMom, aTrack->NSigmaTOFP());
  // }

    if (fTOFParticle == 0) fTOFNSigma->Fill(tMom, aTrack->NSigmaTOFPi());
    if (fTOFParticle == 1) fTOFNSigma->Fill(tMom, aTrack->NSigmaTOFK());
    if (fTOFParticle == 2) fTOFNSigma->Fill(tMom, aTrack->NSigmaTOFP());

    if (fTOFParticle == 0) fTPCNSigma->Fill(tMom, aTrack->NSigmaTPCPi());
    if (fTOFParticle == 1) fTPCNSigma->Fill(tMom, aTrack->NSigmaTPCK());
    if (fTOFParticle == 2) fTPCNSigma->Fill(tMom, aTrack->NSigmaTPCP());

    if (fTOFParticle == 0) fTPCTOFNSigma->Fill(tMom, TMath::Hypot( aTrack->NSigmaTPCPi(), aTrack->NSigmaTOFPi() ) );
    if (fTOFParticle == 1) fTPCTOFNSigma->Fill(tMom, TMath::Hypot( aTrack->NSigmaTPCK(), aTrack->NSigmaTOFK() ) );
    if (fTOFParticle == 2) fTPCTOFNSigma->Fill(tMom, TMath::Hypot( aTrack->NSigmaTPCP(), aTrack->NSigmaTOFP() ) );

}

void AliFemtoCutMonitorParticlePID::Write()
{
  // Write out the relevant histograms
  fTPCdEdx->Write();
  fTOFTime->Write();
    fTOFNSigma->Write();
    fTPCNSigma->Write();
    fTPCTOFNSigma->Write();

}

TList *AliFemtoCutMonitorParticlePID::GetOutputList()
{
  TList *tOutputList = new TList();
  tOutputList->Add(fTPCdEdx);
  tOutputList->Add(fTOFTime);
    tOutputList->Add(fTOFNSigma);
    tOutputList->Add(fTPCNSigma);
    tOutputList->Add(fTPCTOFNSigma);

  return tOutputList;
}
Commit	Line	Data
76ce4b5b	1	////////////////////////////////////////////////////////////////////////////////
	2	// //
	3	// AliFemtoCutMonitorParticlePID - the cut monitor for particles to study //
	4	// various aspects of the PID determination //
	5	// //
	6	////////////////////////////////////////////////////////////////////////////////
	7	#include "AliFemtoCutMonitorParticlePID.h"
	8	#include "AliFemtoModelHiddenInfo.h"
	9	#include <TH1D.h>
	10	#include <TH2D.h>
	11	#include <TList.h>
	12	#include <TMath.h>
	13
	14	AliFemtoCutMonitorParticlePID::AliFemtoCutMonitorParticlePID():
	15	fTPCdEdx(0),
	16	fTOFParticle(0),
	17	fTOFTime(0x0),
ceeb8ab7	18	fTOFNSigma(0),
	19	fTPCNSigma(0),
	20	fTPCTOFNSigma(0)
76ce4b5b	21	{
76ce4b5b	22	// Default constructor
acb5803e	23	fTPCdEdx = new TH2D("TPCdEdx", "TPC dEdx vs. momentum", 100, 0.0, 5.0, 250, 0.0, 500.0);
	24	fTOFTime = new TH2D("TOFTime", "TOF Time vs. momentum", 100, 0.1, 5.0, 400, -4000.0, 4000.0);
	25	fTOFNSigma = new TH2D("TOFNSigma","TOF NSigma vs. momentum", 100, 0.0, 5.0, 100, -5.0, 5.0);
	26	fTPCNSigma = new TH2D("TPCNSigma","TPC NSigma vs. momentum", 100, 0.0, 5.0, 100, -5.0, 5.0);
	27	fTPCTOFNSigma = new TH2D("TPCTOFNSigma","TPC & TOF NSigma vs. momentum", 100, 0.0, 5.0, 100, 0.0, 10.0);
ceeb8ab7	28
76ce4b5b	29	}
	30
	31	AliFemtoCutMonitorParticlePID::AliFemtoCutMonitorParticlePID(const char *aName, Int_t aTOFParticle):
	32	AliFemtoCutMonitor(),
	33	fTPCdEdx(0),
	34	fTOFParticle(aTOFParticle),
	35	fTOFTime(0x0),
ceeb8ab7	36	fTOFNSigma(0),
	37	fTPCNSigma(0),
	38	fTPCTOFNSigma(0)
76ce4b5b	39	{
	40	// Normal constructor
	41	char name[200];
	42	snprintf(name, 200, "TPCdEdx%s", aName);
130ef00d	43	// fTPCdEdx = new TH2D(name, "TPC dEdx vs. momentum", 100, 0.0, 6.0, 250, 0.0, 500.0);
130ef00d	44	fTPCdEdx = new TH2D(name, "TPC dEdx vs. momentum", 200, 0.1, 4.0, 250, 0.0, 500.0);
76ce4b5b	45
76ce4b5b	46	snprintf(name, 200, "TOFTime%s", aName);
acb5803e	47	fTOFTime = new TH2D(name, "TOF Time vs. momentum", 100, 0.1, 5.0, 400, -4000.0, 4000.0);
ceeb8ab7	48
ceeb8ab7	49	snprintf(name, 200, "TOFNSigma%s", aName);
acb5803e	50	fTOFNSigma = new TH2D(name,"TOF NSigma vs. momentum", 100, 0.0, 5.0, 100, -5.0, 5.0);
ceeb8ab7	51
ceeb8ab7	52	snprintf(name, 200, "TPCNSigma%s", aName);
acb5803e	53	fTPCNSigma = new TH2D(name,"TPC NSigma vs. momentum", 100, 0.0, 5.0, 100, -5.0, 5.0);
ceeb8ab7	54
ceeb8ab7	55	snprintf(name, 200, "TPCTOFNSigma%s", aName);
acb5803e	56	fTPCTOFNSigma = new TH2D(name,"TPC & TOF NSigma vs. momentum", 100, 0.0, 5.0, 100, 0.0, 10.0);
76ce4b5b	57
76ce4b5b	58	}
	59
	60	AliFemtoCutMonitorParticlePID::AliFemtoCutMonitorParticlePID(const AliFemtoCutMonitorParticlePID &aCut):
	61	AliFemtoCutMonitor(),
	62	fTPCdEdx(0),
	63	fTOFParticle(0),
	64	fTOFTime(0x0),
ceeb8ab7	65	fTOFNSigma(0),
	66	fTPCNSigma(0),
	67	fTPCTOFNSigma(0)
	68
76ce4b5b	69	{
	70	// copy constructor
	71	if (fTPCdEdx) delete fTPCdEdx;
	72	fTPCdEdx = new TH2D(*aCut.fTPCdEdx);
	73
	74	if (fTOFTime) delete fTOFTime;
	75	fTOFTime = new TH2D(*aCut.fTOFTime);
	76
ceeb8ab7	77	if (fTOFNSigma) delete fTOFNSigma;
	78	fTOFNSigma= new TH2D(*aCut.fTOFNSigma);
	79
	80	if (fTPCNSigma) delete fTPCNSigma;
	81	fTPCNSigma= new TH2D(*aCut.fTPCNSigma);
	82
	83	if (fTPCTOFNSigma) delete fTPCTOFNSigma;
	84	fTPCTOFNSigma= new TH2D(*aCut.fTPCTOFNSigma);
76ce4b5b	85	}
	86
	87	AliFemtoCutMonitorParticlePID::~AliFemtoCutMonitorParticlePID()
	88	{
	89	// Destructor
	90	delete fTPCdEdx;
	91	delete fTOFTime;
ceeb8ab7	92	delete fTOFNSigma;
	93	delete fTPCNSigma;
	94	delete fTPCTOFNSigma;
	95
76ce4b5b	96	}
	97
	98	AliFemtoCutMonitorParticlePID& AliFemtoCutMonitorParticlePID::operator=(const AliFemtoCutMonitorParticlePID& aCut)
	99	{
	100	// assignment operator
91bce940	101	if (this == &aCut)
76ce4b5b	102	return *this;
	103
	104	if (fTPCdEdx) delete fTPCdEdx;
	105	fTPCdEdx = new TH2D(*aCut.fTPCdEdx);
	106
	107	if (fTOFTime) delete fTOFTime;
	108	fTOFTime = new TH2D(*aCut.fTOFTime);
91bce940	109
ceeb8ab7	110	if(fTOFNSigma) delete fTOFNSigma;
	111	fTOFNSigma = new TH2D(*aCut.fTOFNSigma);
	112
	113	if(fTPCNSigma) delete fTPCNSigma;
	114	fTPCNSigma = new TH2D(*aCut.fTPCNSigma);
	115
	116	if(fTPCTOFNSigma) delete fTPCTOFNSigma;
	117	fTPCTOFNSigma = new TH2D(*aCut.fTPCTOFNSigma);
	118
76ce4b5b	119	return *this;
	120	}
	121
91bce940	122	AliFemtoString AliFemtoCutMonitorParticlePID::Report(){
76ce4b5b	123	// Prepare report from the execution
91bce940	124	string stemp = "*** AliFemtoCutMonitorParticlePID report";
76ce4b5b	125	AliFemtoString returnThis = stemp;
91bce940	126	return returnThis;
76ce4b5b	127	}
	128
	129	void AliFemtoCutMonitorParticlePID::Fill(const AliFemtoTrack* aTrack)
	130	{
	131	// Fill in the monitor histograms with the values from the current track
	132	float tMom = aTrack->P().Mag();
	133	float tdEdx = aTrack->TPCsignal();
	134	float tTOF = 0.0;
	135	// short tchg = aTrack->Charge();
	136	if (fTOFParticle == 0) tTOF = aTrack->TOFpionTime();
	137	if (fTOFParticle == 1) tTOF = aTrack->TOFkaonTime();
	138	if (fTOFParticle == 2) tTOF = aTrack->TOFprotonTime();
	139
	140	fTPCdEdx->Fill(tMom, tdEdx);
	141	fTOFTime->Fill(tMom, tTOF);
	142
64536eaf	143	// float vp= aTrack->VTOF();
	144	// if (vp > 0.) {
	145	// fTOFTime->Fill(tMom, tTOF);
	146	// if (fTOFParticle == 0) fTOFNSigma->Fill(tMom, aTrack->NSigmaTOFPi());
	147	// if (fTOFParticle == 1) fTOFNSigma->Fill(tMom, aTrack->NSigmaTOFK());
	148	// if (fTOFParticle == 2) fTOFNSigma->Fill(tMom, aTrack->NSigmaTOFP());
	149	// }
	150
	151	if (fTOFParticle == 0) fTOFNSigma->Fill(tMom, aTrack->NSigmaTOFPi());
	152	if (fTOFParticle == 1) fTOFNSigma->Fill(tMom, aTrack->NSigmaTOFK());
	153	if (fTOFParticle == 2) fTOFNSigma->Fill(tMom, aTrack->NSigmaTOFP());
ceeb8ab7	154
	155	if (fTOFParticle == 0) fTPCNSigma->Fill(tMom, aTrack->NSigmaTPCPi());
	156	if (fTOFParticle == 1) fTPCNSigma->Fill(tMom, aTrack->NSigmaTPCK());
	157	if (fTOFParticle == 2) fTPCNSigma->Fill(tMom, aTrack->NSigmaTPCP());
	158
91bce940	159	if (fTOFParticle == 0) fTPCTOFNSigma->Fill(tMom, TMath::Hypot( aTrack->NSigmaTPCPi(), aTrack->NSigmaTOFPi() ) );
	160	if (fTOFParticle == 1) fTPCTOFNSigma->Fill(tMom, TMath::Hypot( aTrack->NSigmaTPCK(), aTrack->NSigmaTOFK() ) );
	161	if (fTOFParticle == 2) fTPCTOFNSigma->Fill(tMom, TMath::Hypot( aTrack->NSigmaTPCP(), aTrack->NSigmaTOFP() ) );
ceeb8ab7	162
76ce4b5b	163	}
	164
	165	void AliFemtoCutMonitorParticlePID::Write()
	166	{
	167	// Write out the relevant histograms
	168	fTPCdEdx->Write();
	169	fTOFTime->Write();
ceeb8ab7	170	fTOFNSigma->Write();
	171	fTPCNSigma->Write();
	172	fTPCTOFNSigma->Write();
	173
76ce4b5b	174	}
	175
	176	TList *AliFemtoCutMonitorParticlePID::GetOutputList()
	177	{
	178	TList *tOutputList = new TList();
	179	tOutputList->Add(fTPCdEdx);
	180	tOutputList->Add(fTOFTime);
ceeb8ab7	181	tOutputList->Add(fTOFNSigma);
	182	tOutputList->Add(fTPCNSigma);
	183	tOutputList->Add(fTPCTOFNSigma);
	184
76ce4b5b	185	return tOutputList;
76ce4b5b	186	}