[u/mrichter/AliRoot.git] / PWGLF / RESONANCES / AliRsnMiniMonitor.cxx

/**************************************************************************
 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 *                                                                        *
 * Author: The ALICE Off-line Project.                                    *
 * Contributors are mentioned in the code where appropriate.              *
 *                                                                        *
 * Permission to use, copy, modify and distribute this software and its   *
 * documentation strictly for non-commercial purposes is hereby granted   *
 * without fee, provided that the above copyright notice appears in all   *
 * copies and that both the copyright notice and this permission notice   *
 * appear in the supporting documentation. The authors make no claims     *
 * about the suitability of this software for any purpose. It is          *
 * provided "as is" without express or implied warranty.                  *
 **************************************************************************/

////////////////////////////////////////////////////////////////////////////////
//
//  This class contains all code which is used to compute any of the values
//  which can be of interest within a resonance analysis. Besides the obvious
//  invariant mass, it allows to compute other utility values on all possible
//  targets, in order to allow a wide spectrum of binning and checks.
//  When needed, this object can also define a binning in the variable which
//  it is required to compute, which is used for initializing axes of output
//  histograms (see AliRsnFunction).
//  The value computation requires this object to be passed the object whose
//  informations will be used. This object can be of any allowed input type
//  (track, pair, event), then this class must inherit from AliRsnTarget.
//  Then, when value computation is attempted, a check on target type is done
//  and computation is successful only if expected target matches that of the
//  passed object.
//  In some cases, the value computation can require a support external object,
//  which must then be passed to this class. It can be of any type inheriting
//  from TObject.
//
//  authors: A. Pulvirenti (alberto.pulvirenti@ct.infn.it)
//           M. Vala (martin.vala@cern.ch)
//
////////////////////////////////////////////////////////////////////////////////

#include "Riostream.h"

#include "TH1.h"
#include "TH2.h"
#include "TH3.h"

#include "AliLog.h"
#include "AliPID.h"
#include "AliPIDResponse.h"

#include "AliRsnDaughter.h"
#include "AliRsnEvent.h"

#include "AliRsnMiniMonitor.h"

ClassImp(AliRsnMiniMonitor)

//__________________________________________________________________________________________________
AliRsnMiniMonitor::AliRsnMiniMonitor() :
   TNamed(),
   fType(kTypes),
   fCutID(-1),
   fCharge(0),
   fListID(-1),
   fList(0x0)
{
//
// Dummy constructor
//
}

//__________________________________________________________________________________________________
AliRsnMiniMonitor::AliRsnMiniMonitor(const char *name, EType type, Int_t cutID) :
   TNamed(name, ""),
   fType(type),
   fCutID(cutID),
   fCharge(0),
   fListID(-1),
   fList(0x0)
{
//
// Default constructor
//
}

//__________________________________________________________________________________________________
AliRsnMiniMonitor::AliRsnMiniMonitor(const AliRsnMiniMonitor &copy) :
   TNamed(copy),
   fType(copy.fType),
   fCutID(copy.fCutID),
   fCharge(copy.fCharge),
   fListID(copy.fListID),
   fList(copy.fList)
{
//
// Copy constructor
//
}

//__________________________________________________________________________________________________
AliRsnMiniMonitor &AliRsnMiniMonitor::operator=(const AliRsnMiniMonitor &copy)
{
//
// Assignment operator
//

   TNamed::operator=(copy);
   if (this == &copy)
      return *this;
   fType = copy.fType;
   fCutID = copy.fCutID;
   fCharge = copy.fCharge;
   fListID = copy.fListID;
   fList = copy.fList;

   return (*this);
}

//__________________________________________________________________________________________________
Bool_t AliRsnMiniMonitor::Init(const char *name, TList *list)
{
//
// Initialize this output histogram and put into the passed list
//

   // name
   TString sname(name);
   sname += '_';
   sname += GetName();

   // check list
   fList = list;
   if (!list) {
      AliError("No list!");
      return kFALSE;
   }

   // reset histogram
   TH1 *histogram = 0x0;

   switch (fType) {
      case kTrackPt:
         sname += "_TrackPt_cut";
         sname += fCutID;
         histogram = new TH1F(sname.Data(), "", 100, 0.0, 10.0);
         break;
      case kdEdxTPCvsP:
         sname += "_TPCsignal";
         histogram = new TH2F(sname.Data(), "", 500, 0.0, 5.0, 1000, 0.0, 1000.0);
         break;
      case ktimeTOFvsPPion:
         sname += "_TOFsignalPi";
         histogram = new TH2F(sname.Data(), "", 500, 0.0, 5.0, 1000, 0.0, 200000.0);
      case ktimeTOFvsPKaon:
         sname += "_TOFsignalK";
         histogram = new TH2F(sname.Data(), "", 500, 0.0, 5.0, 1000, 0.0, 200000.0);
         break;
      case ktimeTOFvsPProton:
         sname += "_TOFsignalP";
         histogram = new TH2F(sname.Data(), "", 500, 0.0, 5.0, 1000, 0.0, 200000.0);
         break;
      default:
         AliError("Wrong enum type");
         return kFALSE;
   }

   // add to list
   if (histogram && fList) {
      histogram->Sumw2();
      fList->Add(histogram);
      fListID = fList->IndexOf(histogram);
      AliInfo(Form("Histogram '%s' added to list in slot #%d", histogram->GetName(), fListID));
      return kTRUE;
   }

   return kFALSE;
}

//_____________________________________________________________________________
Bool_t AliRsnMiniMonitor::Fill(AliRsnDaughter *track, AliRsnEvent *event)
{
//
// Fill the histogram
//

   // retrieve object from list
   if (!fList) {
      AliError("List pointer is NULL");
      return kFALSE;
   }
   TObject *obj = fList->At(fListID);
   if (!obj) {
      AliError("List object is NULL");
      return kFALSE;
   }

   Double_t valueX, valueY;
   AliVTrack *vtrack = track->Ref2Vtrack();

   AliPIDResponse *pid = event->GetPIDResponse();

   switch (fType) {
      case kTrackPt:
         if (!vtrack) {
            AliWarning("Required vtrack for this value");
            return kFALSE;
         }
         if (fCharge == '+' && vtrack->Charge() <= 0) return kFALSE;
         if (fCharge == '-' && vtrack->Charge() >= 0) return kFALSE;
         if (fCharge == '0' && vtrack->Charge() != 0) return kFALSE;
         valueX = vtrack->Pt();
         ((TH1F *)obj)->Fill(valueX);
         return kTRUE;
      case kdEdxTPCvsP:
         if (!vtrack) {
            AliWarning("Required vtrack for this value");
            return kFALSE;
         }
         valueX = vtrack->GetTPCmomentum();
         valueY = vtrack->GetTPCsignal();
         ((TH2F *)obj)->Fill(valueX, valueY);
         return kTRUE;
      case ktimeTOFvsPPion:
         if (!vtrack) {
            AliWarning("Required vtrack for this value");
            return kFALSE;
         }
         valueX = vtrack->P();
         //valueY = vtrack->GetTOFsignal();
         valueY = 1E20;
         if (pid) valueY = pid->NumberOfSigmasTOF(vtrack, AliPID::kPion);
         ((TH2F *)obj)->Fill(valueX, valueY);
         return kTRUE;
      case ktimeTOFvsPKaon:
         if (!vtrack) {
            AliWarning("Required vtrack for this value");
            return kFALSE;
         }
         valueX = vtrack->P();
         //valueY = vtrack->GetTOFsignal();
         valueY = 1E20;
         if (pid) valueY = pid->NumberOfSigmasTOF(vtrack, AliPID::kKaon);
         ((TH2F *)obj)->Fill(valueX, valueY);
         return kTRUE;
      case ktimeTOFvsPProton:
         if (!vtrack) {
            AliWarning("Required vtrack for this value");
            return kFALSE;
         }
         valueX = vtrack->P();
         //valueY = vtrack->GetTOFsignal();
         valueY = 1E20;
         if (pid) valueY = pid->NumberOfSigmasTOF(vtrack, AliPID::kProton);
         ((TH2F *)obj)->Fill(valueX, valueY);
         return kTRUE;
      default:
         AliError("Invalid value type");
         return kFALSE;
   }
}
Commit	Line	Data
	1	/**************************************************************************
	2	* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	3	* *
	4	* Author: The ALICE Off-line Project. *
	5	* Contributors are mentioned in the code where appropriate. *
	6	* *
	7	* Permission to use, copy, modify and distribute this software and its *
	8	* documentation strictly for non-commercial purposes is hereby granted *
	9	* without fee, provided that the above copyright notice appears in all *
	10	* copies and that both the copyright notice and this permission notice *
	11	* appear in the supporting documentation. The authors make no claims *
	12	* about the suitability of this software for any purpose. It is *
	13	* provided "as is" without express or implied warranty. *
	14	**************************************************************************/
	15
	16	////////////////////////////////////////////////////////////////////////////////
	17	//
	18	// This class contains all code which is used to compute any of the values
	19	// which can be of interest within a resonance analysis. Besides the obvious
	20	// invariant mass, it allows to compute other utility values on all possible
	21	// targets, in order to allow a wide spectrum of binning and checks.
	22	// When needed, this object can also define a binning in the variable which
	23	// it is required to compute, which is used for initializing axes of output
	24	// histograms (see AliRsnFunction).
	25	// The value computation requires this object to be passed the object whose
	26	// informations will be used. This object can be of any allowed input type
	27	// (track, pair, event), then this class must inherit from AliRsnTarget.
	28	// Then, when value computation is attempted, a check on target type is done
	29	// and computation is successful only if expected target matches that of the
	30	// passed object.
	31	// In some cases, the value computation can require a support external object,
	32	// which must then be passed to this class. It can be of any type inheriting
	33	// from TObject.
	34	//
	35	// authors: A. Pulvirenti (alberto.pulvirenti@ct.infn.it)
	36	// M. Vala (martin.vala@cern.ch)
	37	//
	38	////////////////////////////////////////////////////////////////////////////////
	39
	40	#include "Riostream.h"
	41
	42	#include "TH1.h"
	43	#include "TH2.h"
	44	#include "TH3.h"
	45
	46	#include "AliLog.h"
	47	#include "AliPID.h"
	48	#include "AliPIDResponse.h"
	49
	50	#include "AliRsnDaughter.h"
	51	#include "AliRsnEvent.h"
	52
	53	#include "AliRsnMiniMonitor.h"
	54
	55	ClassImp(AliRsnMiniMonitor)
	56
	57	//__________________________________________________________________________________________________
	58	AliRsnMiniMonitor::AliRsnMiniMonitor() :
	59	TNamed(),
	60	fType(kTypes),
	61	fCutID(-1),
	62	fCharge(0),
	63	fListID(-1),
	64	fList(0x0)
	65	{
	66	//
	67	// Dummy constructor
	68	//
	69	}
	70
	71	//__________________________________________________________________________________________________
	72	AliRsnMiniMonitor::AliRsnMiniMonitor(const char *name, EType type, Int_t cutID) :
	73	TNamed(name, ""),
	74	fType(type),
	75	fCutID(cutID),
	76	fCharge(0),
	77	fListID(-1),
	78	fList(0x0)
	79	{
	80	//
	81	// Default constructor
	82	//
	83	}
	84
	85	//__________________________________________________________________________________________________
	86	AliRsnMiniMonitor::AliRsnMiniMonitor(const AliRsnMiniMonitor &copy) :
	87	TNamed(copy),
	88	fType(copy.fType),
	89	fCutID(copy.fCutID),
	90	fCharge(copy.fCharge),
	91	fListID(copy.fListID),
	92	fList(copy.fList)
	93	{
	94	//
	95	// Copy constructor
	96	//
	97	}
	98
	99	//__________________________________________________________________________________________________
	100	AliRsnMiniMonitor &AliRsnMiniMonitor::operator=(const AliRsnMiniMonitor &copy)
	101	{
	102	//
	103	// Assignment operator
	104	//
	105
	106	TNamed::operator=(copy);
	107	if (this == &copy)
	108	return *this;
	109	fType = copy.fType;
	110	fCutID = copy.fCutID;
	111	fCharge = copy.fCharge;
	112	fListID = copy.fListID;
	113	fList = copy.fList;
	114
	115	return (*this);
	116	}
	117
	118	//__________________________________________________________________________________________________
	119	Bool_t AliRsnMiniMonitor::Init(const char name, TList list)
	120	{
	121	//
	122	// Initialize this output histogram and put into the passed list
	123	//
	124
	125	// name
	126	TString sname(name);
	127	sname += '_';
	128	sname += GetName();
	129
	130	// check list
	131	fList = list;
	132	if (!list) {
	133	AliError("No list!");
	134	return kFALSE;
	135	}
	136
	137	// reset histogram
	138	TH1 *histogram = 0x0;
	139
	140	switch (fType) {
	141	case kTrackPt:
	142	sname += "_TrackPt_cut";
	143	sname += fCutID;
	144	histogram = new TH1F(sname.Data(), "", 100, 0.0, 10.0);
	145	break;
	146	case kdEdxTPCvsP:
	147	sname += "_TPCsignal";
	148	histogram = new TH2F(sname.Data(), "", 500, 0.0, 5.0, 1000, 0.0, 1000.0);
	149	break;
	150	case ktimeTOFvsPPion:
	151	sname += "_TOFsignalPi";
	152	histogram = new TH2F(sname.Data(), "", 500, 0.0, 5.0, 1000, 0.0, 200000.0);
	153	case ktimeTOFvsPKaon:
	154	sname += "_TOFsignalK";
	155	histogram = new TH2F(sname.Data(), "", 500, 0.0, 5.0, 1000, 0.0, 200000.0);
	156	break;
	157	case ktimeTOFvsPProton:
	158	sname += "_TOFsignalP";
	159	histogram = new TH2F(sname.Data(), "", 500, 0.0, 5.0, 1000, 0.0, 200000.0);
	160	break;
	161	default:
	162	AliError("Wrong enum type");
	163	return kFALSE;
	164	}
	165
	166	// add to list
	167	if (histogram && fList) {
	168	histogram->Sumw2();
	169	fList->Add(histogram);
	170	fListID = fList->IndexOf(histogram);
	171	AliInfo(Form("Histogram '%s' added to list in slot #%d", histogram->GetName(), fListID));
	172	return kTRUE;
	173	}
	174
	175	return kFALSE;
	176	}
	177
	178	//_____________________________________________________________________________
	179	Bool_t AliRsnMiniMonitor::Fill(AliRsnDaughter track, AliRsnEvent event)
	180	{
	181	//
	182	// Fill the histogram
	183	//
	184
	185	// retrieve object from list
	186	if (!fList) {
	187	AliError("List pointer is NULL");
	188	return kFALSE;
	189	}
	190	TObject *obj = fList->At(fListID);
	191	if (!obj) {
	192	AliError("List object is NULL");
	193	return kFALSE;
	194	}
	195
	196	Double_t valueX, valueY;
	197	AliVTrack *vtrack = track->Ref2Vtrack();
	198
	199	AliPIDResponse *pid = event->GetPIDResponse();
	200
	201	switch (fType) {
	202	case kTrackPt:
	203	if (!vtrack) {
	204	AliWarning("Required vtrack for this value");
	205	return kFALSE;
	206	}
	207	if (fCharge == '+' && vtrack->Charge() <= 0) return kFALSE;
	208	if (fCharge == '-' && vtrack->Charge() >= 0) return kFALSE;
	209	if (fCharge == '0' && vtrack->Charge() != 0) return kFALSE;
	210	valueX = vtrack->Pt();
	211	((TH1F *)obj)->Fill(valueX);
	212	return kTRUE;
	213	case kdEdxTPCvsP:
	214	if (!vtrack) {
	215	AliWarning("Required vtrack for this value");
	216	return kFALSE;
	217	}
	218	valueX = vtrack->GetTPCmomentum();
	219	valueY = vtrack->GetTPCsignal();
	220	((TH2F *)obj)->Fill(valueX, valueY);
	221	return kTRUE;
	222	case ktimeTOFvsPPion:
	223	if (!vtrack) {
	224	AliWarning("Required vtrack for this value");
	225	return kFALSE;
	226	}
	227	valueX = vtrack->P();
	228	//valueY = vtrack->GetTOFsignal();
	229	valueY = 1E20;
	230	if (pid) valueY = pid->NumberOfSigmasTOF(vtrack, AliPID::kPion);
	231	((TH2F *)obj)->Fill(valueX, valueY);
	232	return kTRUE;
	233	case ktimeTOFvsPKaon:
	234	if (!vtrack) {
	235	AliWarning("Required vtrack for this value");
	236	return kFALSE;
	237	}
	238	valueX = vtrack->P();
	239	//valueY = vtrack->GetTOFsignal();
	240	valueY = 1E20;
	241	if (pid) valueY = pid->NumberOfSigmasTOF(vtrack, AliPID::kKaon);
	242	((TH2F *)obj)->Fill(valueX, valueY);
	243	return kTRUE;
	244	case ktimeTOFvsPProton:
	245	if (!vtrack) {
	246	AliWarning("Required vtrack for this value");
	247	return kFALSE;
	248	}
	249	valueX = vtrack->P();
	250	//valueY = vtrack->GetTOFsignal();
	251	valueY = 1E20;
	252	if (pid) valueY = pid->NumberOfSigmasTOF(vtrack, AliPID::kProton);
	253	((TH2F *)obj)->Fill(valueX, valueY);
	254	return kTRUE;
	255	default:
	256	AliError("Invalid value type");
	257	return kFALSE;
	258	}
	259	}