[u/mrichter/AliRoot.git] / PWG2 / RESONANCES / AliRsnAnalysisKStarKpi.h

//
// Class AliRsnAnalysisKStarKpi
//
// Virtual Class derivated from AliRsnVAnalysisTaskSE which will be base class
// for all RSN SE tasks
//
// authors: Martin Vala (martin.vala@cern.ch)
//          Alberto Pulvirenti (alberto.pulvirenti@ct.infn.it)
//
#ifndef AliRsnAnalysisKStarKpi_H
#define AliRsnAnalysisKStarKpi_H

#include "TClonesArray.h"

#include "AliRsnVAnalysisTaskSE.h"

#include "AliRsnDaughter.h"
#include "AliRsnMother.h"
#include "AliRsnPairDef.h"
#include "AliRsnEvent.h"
#include "AliRsnCutSet.h"

class AliRsnPIDDefESD;

class AliRsnAnalysisKStarKpi : public AliRsnVAnalysisTaskSE
{
  public:
  
    AliRsnAnalysisKStarKpi(const char *name = "RSNphi", Bool_t useKine = kFALSE);
    AliRsnAnalysisKStarKpi(const AliRsnAnalysisKStarKpi& copy);
    AliRsnAnalysisKStarKpi& operator=(const AliRsnAnalysisKStarKpi& copy);
    virtual ~AliRsnAnalysisKStarKpi() {}

    virtual void            RsnUserCreateOutputObjects();
    virtual void            RsnUserExec(Option_t*);
    virtual void            RsnTerminate(Option_t*);
    virtual Bool_t          EventProcess();
    
    AliRsnCutSet*           GetEventCuts()          {return &fCutEvent;}
    AliRsnCutSet*           GetCommonDaughterCuts() {return &fCutTrackCommon;}
    AliRsnCutSet*           GetKaonDaughterCuts()   {return &fCutTrackKaon;}
    AliRsnCutSet*           GetPionDaughterCuts()   {return &fCutTrackPion;}
    AliRsnCutSet*           GetMotherCuts()         {return &fCutPair;}
    
    void                    AddFunction(AliRsnFunction* const fcn);
    TList*                  GenerateHistograms();

  private:
  
    TArrayI                 fGoodK;             // indexes of good tracks (kaons)
    TArrayI                 fGoodPi;            // indexes of good tracks (pions)
    AliRsnDaughter          fKaon;              // daughter data-member (pion)
    AliRsnDaughter          fPion;              // daughter data-member (pion)
    AliRsnMother            fMother;            // mother data-member (to save memory)
    AliRsnPairDef           fPairDef;           // resonance decay tree (fixed)
    
    AliRsnCutSet            fCutEvent;          // cut set for events
    AliRsnCutSet            fCutTrackCommon;    // cut set for tracks (common)
    AliRsnCutSet            fCutTrackKaon;      // cut set for tracks (only pos)
    AliRsnCutSet            fCutTrackPion;      // cut set for tracks (only neg)
    AliRsnCutSet            fCutPair;           // cut set for pairs

    TClonesArray            fFuncPM;            // collection of functions for unlike-sign (K+ pi-)
    TClonesArray            fFuncMP;            // collection of functions for unlike-sign (K- pi+)
    TClonesArray            fFuncPP;            // collection of functions for like-sign ++
    TClonesArray            fFuncMM;            // collection of functions for like-sign --
    TClonesArray            fFuncTruePM;        // collection of functions for unlike-sign true pairs (K+ pi-)
    TClonesArray            fFuncTrueMP;        // collection of functions for unlike-sign true pairs (K- pi+)
    
    TList                  *fOutList;           // list of output events

    ClassDef(AliRsnAnalysisKStarKpi, 1)
};

#endif
Commit	Line	Data
b0d3d6b9	1	//
	2	// Class AliRsnAnalysisKStarKpi
	3	//
	4	// Virtual Class derivated from AliRsnVAnalysisTaskSE which will be base class
	5	// for all RSN SE tasks
	6	//
	7	// authors: Martin Vala (martin.vala@cern.ch)
	8	// Alberto Pulvirenti (alberto.pulvirenti@ct.infn.it)
	9	//
	10	#ifndef AliRsnAnalysisKStarKpi_H
	11	#define AliRsnAnalysisKStarKpi_H
	12
	13	#include "TClonesArray.h"
	14
	15	#include "AliRsnVAnalysisTaskSE.h"
	16
	17	#include "AliRsnDaughter.h"
	18	#include "AliRsnMother.h"
	19	#include "AliRsnPairDef.h"
	20	#include "AliRsnEvent.h"
	21	#include "AliRsnCutSet.h"
	22
	23	class AliRsnPIDDefESD;
	24
	25	class AliRsnAnalysisKStarKpi : public AliRsnVAnalysisTaskSE
	26	{
	27	public:
	28
	29	AliRsnAnalysisKStarKpi(const char *name = "RSNphi", Bool_t useKine = kFALSE);
	30	AliRsnAnalysisKStarKpi(const AliRsnAnalysisKStarKpi& copy);
	31	AliRsnAnalysisKStarKpi& operator=(const AliRsnAnalysisKStarKpi& copy);
	32	virtual ~AliRsnAnalysisKStarKpi() {}
	33
	34	virtual void RsnUserCreateOutputObjects();
	35	virtual void RsnUserExec(Option_t*);
	36	virtual void RsnTerminate(Option_t*);
	37	virtual Bool_t EventProcess();
	38
	39	AliRsnCutSet* GetEventCuts() {return &fCutEvent;}
	40	AliRsnCutSet* GetCommonDaughterCuts() {return &fCutTrackCommon;}
	41	AliRsnCutSet* GetKaonDaughterCuts() {return &fCutTrackKaon;}
	42	AliRsnCutSet* GetPionDaughterCuts() {return &fCutTrackPion;}
	43	AliRsnCutSet* GetMotherCuts() {return &fCutPair;}
	44
	45	void AddFunction(AliRsnFunction* const fcn);
	46	TList* GenerateHistograms();
	47
	48	private:
	49
	50	TArrayI fGoodK; // indexes of good tracks (kaons)
	51	TArrayI fGoodPi; // indexes of good tracks (pions)
	52	AliRsnDaughter fKaon; // daughter data-member (pion)
	53	AliRsnDaughter fPion; // daughter data-member (pion)
	54	AliRsnMother fMother; // mother data-member (to save memory)
	55	AliRsnPairDef fPairDef; // resonance decay tree (fixed)
	56
	57	AliRsnCutSet fCutEvent; // cut set for events
	58	AliRsnCutSet fCutTrackCommon; // cut set for tracks (common)
	59	AliRsnCutSet fCutTrackKaon; // cut set for tracks (only pos)
	60	AliRsnCutSet fCutTrackPion; // cut set for tracks (only neg)
	61	AliRsnCutSet fCutPair; // cut set for pairs
	62
	63	TClonesArray fFuncPM; // collection of functions for unlike-sign (K+ pi-)
	64	TClonesArray fFuncMP; // collection of functions for unlike-sign (K- pi+)
65	TClonesArray fFuncPP; // collection of functions for like-sign ++
66	TClonesArray fFuncMM; // collection of functions for like-sign --
67	TClonesArray fFuncTruePM; // collection of functions for unlike-sign true pairs (K+ pi-)
68	TClonesArray fFuncTrueMP; // collection of functions for unlike-sign true pairs (K- pi+)
69
70	TList *fOutList; // list of output events
71
72	ClassDef(AliRsnAnalysisKStarKpi, 1)
73	};
74
75	#endif