-/**************************************************************************
- * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
- * See cxx source for full Copyright notice *
- **************************************************************************/
-
-//-------------------------------------------------------------------------
-// Class AliRsnAnalysis
-// Reconstruction and analysis of K* Rsn
-// ........................................
-// ........................................
-// ........................................
-// ........................................
-//
-// author: A. Pulvirenti (email: alberto.pulvirenti@ct.infn.it)
-//-------------------------------------------------------------------------
+//
+// *** Class AliRsnPair ***
+//
+// TODO
+//
+// authors: A. Pulvirenti (email: alberto.pulvirenti@ct.infn.it)
+// M. Vala (email: martin.vala@cern.ch)
+//
#ifndef ALIRSNPAIR_H
#define ALIRSNPAIR_H
-#include <TNamed.h>
-#include "AliRsnPID.h"
+#include "TNamed.h"
-class TH1D;
-class TRefArray;
-class TObjArray;
-class AliRsnEvent;
-class AliRsnDaughter;
-class AliRsnDaughterCut;
-class AliRsnDaughterCutPair;
+#include "AliRsnPairDef.h"
+#include "AliRsnCutManager.h"
+#include "AliRsnMother.h"
-class AliRsnPair : public TNamed
-{
+class TList;
+class AliRsnPair : public TNamed {
public:
-
- AliRsnPair();
- AliRsnPair(const char *name, const char *title,
- Int_t nbins, Double_t min, Double_t max,
- Double_t ptmin = 0., Double_t ptmax = 0., Double_t dmax = 0.);
- AliRsnPair(const AliRsnPair ©);
- const AliRsnPair& operator=(const AliRsnPair ©);
- virtual ~AliRsnPair() {Clear();}
- virtual void Clear(Option_t *option = "");
-
- /* getters */
- TH1D* GetHistogram() {return fHistogram;}
- Char_t GetCharge(Int_t i) const {if (i>=0&&i<2) return fCharge[i]; else return 0;}
- AliRsnPID::EType GetParticle(Int_t i) const {if (i>=0&&i<2) return fType[i]; else return AliRsnPID::kUnknown;}
- Double_t GetMass(Int_t i) const {if (i>=0&&i<2) return fMass[i]; else return 0.0;}
- Bool_t StoreOnlyTruePairs() const {return fStoreOnlyTrue;}
- Bool_t IsForMixing() const {return fForMixing;}
-
- /* setters */
- void SetMass(Int_t i, Double_t value) {if (i>=0&&i<2) fMass[i] = value;}
- void SetTrueMotherPDG(Int_t pdg) {fTrueMotherPDG = pdg;}
- void SetPair(Char_t charge1, AliRsnPID::EType pid1, Char_t charge2, AliRsnPID::EType pid2);
- void SetPtBin(Double_t min, Double_t max) {fPtMin = min; fPtMax = max;}
- void SetImpactMax(Double_t max) {fVtMax = max;}
- void SetStoreOnlyTrue(Bool_t doit = kTRUE) {fStoreOnlyTrue = doit;}
- void SetForMixing(Bool_t doit = kTRUE) {fForMixing = doit;}
-
- /* working parameters */
- void AddCutPair(AliRsnDaughterCutPair *cut);
- void AddCutSingle(Int_t i, AliRsnDaughterCut *cut);
- Stat_t Process(AliRsnEvent *event1, AliRsnEvent *event2 = 0, Bool_t usePID = kTRUE);
-
-private:
- /* private functions */
- void InitHistogram(Int_t nbins, Double_t min, Double_t max);
- Bool_t SingleCutCheck(Int_t ipart, AliRsnDaughter *track) const;
- Bool_t PairCutCheck(AliRsnDaughter *track1, AliRsnDaughter *track2) const;
- Stat_t Fill(TRefArray *list1, TRefArray *list2, Bool_t skipSameIndex = kTRUE);
+ AliRsnPair(const char *name = "default", AliRsnPairDef *def = 0);
+ AliRsnPair(const AliRsnPair ©);
+ AliRsnPair& operator=(const AliRsnPair&);
+ ~AliRsnPair();
+
+ // getters
+ Bool_t IsOnlyTrue() const {return fOnlyTrue;}
+ Bool_t IsCheckDecay() const {return fCheckDecay;}
+ Bool_t IsMixed() const {return fIsMixed;}
+ Int_t GetCount() const {return fCount;}
+ AliRsnPairDef* GetPairDef() {return fPairDef;}
+ AliRsnCutManager* GetCutManager() {return &fCutManager;}
+ AliRsnMother* GetMother() {return &fMother;}
+
+ // shortcuts to data-member getters
+ AliRsnCutSet* GetCommonDaughterCuts() {return fCutManager.GetCommonDaughterCuts();}
+ AliRsnCutSet* GetDaughter1Cuts() {return fCutManager.GetDaughter1Cuts();}
+ AliRsnCutSet* GetDaughter2Cuts() {return fCutManager.GetDaughter2Cuts();}
+ AliRsnCutSet* GetMotherCuts() {return fCutManager.GetMotherCuts();}
+
+ // setters (not for all members)
+ void SetOnlyTrue(Bool_t onlyTrue = kTRUE) {fOnlyTrue = onlyTrue;}
+ void SetCheckDecay(Bool_t check = kTRUE) {fCheckDecay = check;}
+ void SetMixed(Bool_t doit = kTRUE) {fIsMixed = doit;}
+ void SetCount(Int_t count) {fCount = count;}
+ void ResetCount() {fCount = 0;}
+
+ // methods
+ Bool_t Fill(AliRsnDaughter *d0, AliRsnDaughter *d1, Bool_t refFirst = kTRUE);
+ virtual void Print(Option_t *option = "") const;
+ virtual void Compute();
+ virtual void Init(const char *prefix, TList *list);
+
+protected:
+
+ Bool_t fOnlyTrue; // select true pairs only?
+ Bool_t fCheckDecay; // is the decay channel correct in a true pair?
+ Bool_t fIsMixed; // is this an event-mixing?
+ Int_t fCount; // counter incremented for each added pair
- /* flags */
- Bool_t fForMixing; // flag is true for objects created for event mixing
- Bool_t fStoreOnlyTrue; // output = only spectra of true pairs
-
- /* parameters */
- Int_t fTrueMotherPDG; // PDG code of true mother (if known)
- Double_t fMass[2]; // nominal mass of particles
- Char_t fCharge[2]; // charge of particles
- AliRsnPID::EType fType[2]; // particles types
-
- /* basic cuts */
- Double_t fPtMin; // minimum allowed pt for the pair
- Double_t fPtMax; // maximum allowed pt for the pair
- Double_t fVtMax; // maximum transverse impact parameter for each track
-
- /* cuts */
- TObjArray *fCutsSingle[2]; // single-particle cuts
- TObjArray *fCutsPair; // pair cuts
-
- /* output */
- TH1D *fHistogram; // invariant mass distribution
-
- /* ROOT dictionary */
- ClassDef(AliRsnPair, 1)
+ AliRsnPairDef *fPairDef; // pair definition (particles, charges)
+ AliRsnCutManager fCutManager; // collection of all cuts
+ AliRsnMother fMother; // mother candidate (to avoid creating it continuously)
+
+private:
+
+ ClassDef(AliRsnPair, 2)
};
#endif
+
git://git.uio.no / u / mrichter / AliRoot.git / blobdiff