[u/mrichter/AliRoot.git] / PWGLF / RESONANCES / AliRsnMiniValue.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 "AliLog.h"

#include "AliRsnMiniPair.h"
#include "AliRsnMiniEvent.h"
#include "AliRsnMiniParticle.h"

#include "AliRsnMiniValue.h"

ClassImp(AliRsnMiniValue)

//_____________________________________________________________________________
AliRsnMiniValue::AliRsnMiniValue(EType type, Bool_t useMC) :
   TNamed(ValueName(type, useMC), ""),
   fType(type),
   fUseMCInfo(useMC)
{
//
// Constructor
//
}

//_____________________________________________________________________________
AliRsnMiniValue::AliRsnMiniValue(const AliRsnMiniValue &copy) :
   TNamed(copy),
   fType(copy.fType),
   fUseMCInfo(copy.fUseMCInfo)
{
//
// Copy constructor
//
}

//_____________________________________________________________________________
AliRsnMiniValue &AliRsnMiniValue::operator=(const AliRsnMiniValue &copy)
{
//
// Assignment operator.
// Works like copy constructor.
//
   TNamed::operator=(copy);
   if (this == &copy)
      return *this;
   fType = copy.fType;
   fUseMCInfo = copy.fUseMCInfo;

   return (*this);
}

//_____________________________________________________________________________
const char *AliRsnMiniValue::TypeName(EType type)
{
//
// This method returns a string to give a name to each possible
// computation value.
//

   switch (type) {
      case kVz:           return "EventVz";
      case kMult:         return "EventMult";
      case kPlaneAngle:   return "EventPlane";
      case kLeadingPt:    return "EventLeadingPt";
      case kPt:           return "Pt";
      case kPz:           return "Pz";
      case kInvMass:      return "InvMass";
      case kInvMassRes:   return "InvMassResolution";
      case kInvMassDiff:  return "InvMassDifference";
      case kEta:          return "Eta";
      case kMt:           return "Mt";
      case kY:            return "Y";
      case kPtRatio:      return "PtRatio";
      case kDipAngle:     return "DipAngle";
      case kCosThetaStar: return "CosThetaStar";
      case kAngleLeading: return "AngleToLeading";
      case kFirstDaughterPt: return "FirstDaughterPt";
      case kSecondDaughterPt: return "SecondDaughterPt";
      case kFirstDaughterP: return "FirstDaughterP";
      case kSecondDaughterP: return "SecondDaughterP";
      case kDCAproduct:   return "DaughterDCAproduct";
      case kFirstDaughterDCA: return "FirstDaughterDCA";
      case kSecondDaughterDCA: return "SecondDaughterDCA";
      case kNSisters:     return "NumberOfSisters";
      default:            return "Undefined";
   }
}

//_____________________________________________________________________________
Float_t AliRsnMiniValue::Eval(AliRsnMiniPair *pair, AliRsnMiniEvent *event)
{
//
// Evaluation of the required value.
// In this implementation, fills the member 4-vectors with data
// coming from the object passed as argument, and then returns the value
//

   if (!pair && fType > kEventCuts) {
      AliError("Null pair passed!");
      return 1E20;
   }

   // compute value depending on types in the enumeration
   // if the type does not match any available choice, or if
   // the computation is not doable due to any problem
   // (not initialized support object, wrong values, risk of floating point errors)
   // the method returng kFALSE and sets the computed value to a meaningless number
   Double_t p3[3]= {0.,0.,0.};
   AliRsnMiniParticle *l;
   TLorentzVector v;
   switch (fType) {
         // ---- event values -------------------------------------------------------------------------
      case kVz:
         return event->Vz();
      case kMult:
         return event->Mult();
      case kPlaneAngle:
         return event->Angle();
      case kLeadingPt:
         l = event->LeadingParticle();
         if (l) {
            l->Set4Vector(v,-1.0,fUseMCInfo);
            return v.Pt();
         }
         return 0.0;
      case kPt:
         return pair->Pt(fUseMCInfo);
      case kInvMass:
         return pair->InvMass(fUseMCInfo);
      case kEta:
         return pair->Eta(fUseMCInfo);
      case kInvMassRes:
         return pair->InvMassRes();
      case kInvMassDiff:
         return pair->InvMassDiff();
      case kMt:
         return pair->Mt(fUseMCInfo);
      case kY:
         return pair->Y(fUseMCInfo);
      case kPtRatio:
         return pair->PtRatio(fUseMCInfo);
      case kDipAngle:
         return pair->DipAngle(fUseMCInfo);
      case kCosThetaStar:
         return pair->CosThetaStar(fUseMCInfo);
      case kAngleLeading:
         l = event->LeadingParticle();
         if (l) {
            l->Set4Vector(v,-1.0,fUseMCInfo);
            Double_t angle = v.Phi() - pair->Sum(fUseMCInfo).Phi();

            //return angle w.r.t. leading particle in the range -pi/2, 3/2pi
            while (angle >= 1.5 * TMath::Pi()) angle -= 2 * TMath::Pi();
            while (angle < -0.5 * TMath::Pi()) angle += 2 * TMath::Pi();
            return angle;
         }
//         AliWarning("This method is not yet implemented");
         return 1E20;
      case kFirstDaughterPt:
         return pair->DaughterPt(0,fUseMCInfo);
      case kSecondDaughterPt:
         return pair->DaughterPt(1,fUseMCInfo);
      case kFirstDaughterP:
         pair->DaughterPxPyPz(0,fUseMCInfo, p3);
         return TMath::Sqrt(p3[0]*p3[0]+p3[1]*p3[1]+p3[2]*p3[2]);
      case kSecondDaughterP:
         pair->DaughterPxPyPz(1,fUseMCInfo, p3);
         return TMath::Sqrt(p3[0]*p3[0]+p3[1]*p3[1]+p3[2]*p3[2]);
      case kDCAproduct:
         return pair->DCAProduct();
      case kFirstDaughterDCA:
         return pair->DaughterDCA(0);
      case kSecondDaughterDCA:
         return pair->DaughterDCA(1);
      case kNSisters:
         return pair->NSisters();
      default:
         AliError("Invalid value type");
         return 1E20;
   }
}
Commit	Line	Data
03d23846	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 "AliLog.h"
	43
	44	#include "AliRsnMiniPair.h"
	45	#include "AliRsnMiniEvent.h"
4fb0dfa3	46	#include "AliRsnMiniParticle.h"
03d23846	47
	48	#include "AliRsnMiniValue.h"
	49
	50	ClassImp(AliRsnMiniValue)
	51
	52	//_____________________________________________________________________________
	53	AliRsnMiniValue::AliRsnMiniValue(EType type, Bool_t useMC) :
	54	TNamed(ValueName(type, useMC), ""),
	55	fType(type),
	56	fUseMCInfo(useMC)
	57	{
	58	//
	59	// Constructor
	60	//
	61	}
	62
	63	//_____________________________________________________________________________
61f275d1	64	AliRsnMiniValue::AliRsnMiniValue(const AliRsnMiniValue &copy) :
03d23846	65	TNamed(copy),
	66	fType(copy.fType),
	67	fUseMCInfo(copy.fUseMCInfo)
	68	{
	69	//
	70	// Copy constructor
	71	//
	72	}
	73
	74	//_____________________________________________________________________________
61f275d1	75	AliRsnMiniValue &AliRsnMiniValue::operator=(const AliRsnMiniValue &copy)
03d23846	76	{
	77	//
	78	// Assignment operator.
	79	// Works like copy constructor.
	80	//
03d23846	81	TNamed::operator=(copy);
e6f3a909	82	if (this == &copy)
61f275d1	83	return *this;
03d23846	84	fType = copy.fType;
	85	fUseMCInfo = copy.fUseMCInfo;
	86
	87	return (*this);
	88	}
	89
	90	//_____________________________________________________________________________
61f275d1	91	const char *AliRsnMiniValue::TypeName(EType type)
03d23846	92	{
	93	//
	94	// This method returns a string to give a name to each possible
	95	// computation value.
	96	//
	97
	98	switch (type) {
	99	case kVz: return "EventVz";
	100	case kMult: return "EventMult";
	101	case kPlaneAngle: return "EventPlane";
	102	case kLeadingPt: return "EventLeadingPt";
	103	case kPt: return "Pt";
	104	case kPz: return "Pz";
	105	case kInvMass: return "InvMass";
	106	case kInvMassRes: return "InvMassResolution";
2d21288f	107	case kInvMassDiff: return "InvMassDifference";
03d23846	108	case kEta: return "Eta";
	109	case kMt: return "Mt";
	110	case kY: return "Y";
	111	case kPtRatio: return "PtRatio";
	112	case kDipAngle: return "DipAngle";
	113	case kCosThetaStar: return "CosThetaStar";
	114	case kAngleLeading: return "AngleToLeading";
4fb0dfa3	115	case kFirstDaughterPt: return "FirstDaughterPt";
	116	case kSecondDaughterPt: return "SecondDaughterPt";
	117	case kFirstDaughterP: return "FirstDaughterP";
	118	case kSecondDaughterP: return "SecondDaughterP";
213adb92	119	case kDCAproduct: return "DaughterDCAproduct";
	120	case kFirstDaughterDCA: return "FirstDaughterDCA";
	121	case kSecondDaughterDCA: return "SecondDaughterDCA";
088ca370	122	case kNSisters: return "NumberOfSisters";
03d23846	123	default: return "Undefined";
	124	}
	125	}
	126
	127	//_____________________________________________________________________________
	128	Float_t AliRsnMiniValue::Eval(AliRsnMiniPair pair, AliRsnMiniEvent event)
	129	{
	130	//
	131	// Evaluation of the required value.
	132	// In this implementation, fills the member 4-vectors with data
	133	// coming from the object passed as argument, and then returns the value
	134	//
	135
	136	if (!pair && fType > kEventCuts) {
	137	AliError("Null pair passed!");
	138	return 1E20;
	139	}
61f275d1	140
03d23846	141	// compute value depending on types in the enumeration
	142	// if the type does not match any available choice, or if
	143	// the computation is not doable due to any problem
	144	// (not initialized support object, wrong values, risk of floating point errors)
	145	// the method returng kFALSE and sets the computed value to a meaningless number
4fb0dfa3	146	Double_t p3[3]= {0.,0.,0.};
	147	AliRsnMiniParticle *l;
	148	TLorentzVector v;
03d23846	149	switch (fType) {
61f275d1	150	// ---- event values -------------------------------------------------------------------------
03d23846	151	case kVz:
	152	return event->Vz();
	153	case kMult:
	154	return event->Mult();
	155	case kPlaneAngle:
	156	return event->Angle();
	157	case kLeadingPt:
4fb0dfa3	158	l = event->LeadingParticle();
	159	if (l) {
	160	l->Set4Vector(v,-1.0,fUseMCInfo);
	161	return v.Pt();
	162	}
03d23846	163	return 0.0;
e6952ec7	164	case kPt:
03d23846	165	return pair->Pt(fUseMCInfo);
	166	case kInvMass:
	167	return pair->InvMass(fUseMCInfo);
	168	case kEta:
	169	return pair->Eta(fUseMCInfo);
	170	case kInvMassRes:
	171	return pair->InvMassRes();
2d21288f	172	case kInvMassDiff:
2d21288f	173	return pair->InvMassDiff();
03d23846	174	case kMt:
	175	return pair->Mt(fUseMCInfo);
	176	case kY:
	177	return pair->Y(fUseMCInfo);
	178	case kPtRatio:
	179	return pair->PtRatio(fUseMCInfo);
	180	case kDipAngle:
	181	return pair->DipAngle(fUseMCInfo);
	182	case kCosThetaStar:
	183	return pair->CosThetaStar(fUseMCInfo);
	184	case kAngleLeading:
4fb0dfa3	185	l = event->LeadingParticle();
	186	if (l) {
	187	l->Set4Vector(v,-1.0,fUseMCInfo);
	188	Double_t angle = v.Phi() - pair->Sum(fUseMCInfo).Phi();
	189
	190	//return angle w.r.t. leading particle in the range -pi/2, 3/2pi
	191	while (angle >= 1.5 * TMath::Pi()) angle -= 2 * TMath::Pi();
	192	while (angle < -0.5 * TMath::Pi()) angle += 2 * TMath::Pi();
	193	return angle;
	194	}
	195	// AliWarning("This method is not yet implemented");
	196	return 1E20;
397c0062	197	case kFirstDaughterPt:
	198	return pair->DaughterPt(0,fUseMCInfo);
	199	case kSecondDaughterPt:
	200	return pair->DaughterPt(1,fUseMCInfo);
	201	case kFirstDaughterP:
4fb0dfa3	202	pair->DaughterPxPyPz(0,fUseMCInfo, p3);
397c0062	203	return TMath::Sqrt(p3[0]p3[0]+p3[1]p3[1]+p3[2]*p3[2]);
397c0062	204	case kSecondDaughterP:
4fb0dfa3	205	pair->DaughterPxPyPz(1,fUseMCInfo, p3);
397c0062	206	return TMath::Sqrt(p3[0]p3[0]+p3[1]p3[1]+p3[2]*p3[2]);
213adb92	207	case kDCAproduct:
	208	return pair->DCAProduct();
	209	case kFirstDaughterDCA:
	210	return pair->DaughterDCA(0);
	211	case kSecondDaughterDCA:
	212	return pair->DaughterDCA(1);
088ca370	213	case kNSisters:
088ca370	214	return pair->NSisters();
03d23846	215	default:
	216	AliError("Invalid value type");
	217	return 1E20;
	218	}
	219	}