[u/mrichter/AliRoot.git] / PWGHF / hfe / AliHFEreducedEventCreatorESD.h

/**************************************************************************
* 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.                  *
**************************************************************************/
//
// Debug tree to look at the distribution of the variable we are cutting on
//
//
#ifndef ALIHFEREDUCEDEVENTCREATORESD_H
#define ALIHFEREDUCEDEVENTCREATORESD_H

#include "AliAnalysisTaskSE.h"

class TString;
class TTree;
class AliAnalysisUtils;
class AliPIDResponse;
class AliHFEcuts;
class AliHFEextraCuts;
class AliHFEpidTPC;
class AliTRDTriggerAnalysis;
class AliHFEsignalCuts;
class AliHFEreducedEvent;
class AliHFEV0taginfo;

class AliHFEreducedEventCreatorESD : public AliAnalysisTaskSE{
  public:
    AliHFEreducedEventCreatorESD();
    AliHFEreducedEventCreatorESD(const char *name);
    virtual ~AliHFEreducedEventCreatorESD();

    virtual void UserCreateOutputObjects();
    virtual void UserExec(Option_t *);
    virtual void Terminate(Option_t *);

    // Setters for cuts
    void SetMinNclustersTPC(Int_t mincl) { fNclustersTPC = mincl; };
    void SetMinNclustersTPCPID(Int_t mincl) { fNclustersTPCPID = mincl; };
    void SetMinNclustersITS(Int_t mincl) { fNclustersITS = mincl; };
    void SetRemoveFirstEventFromChunk() { fRemoveFirstEvent = kTRUE; }
    void SetFlagPileupEvents() { fFlagPileupEvents = kTRUE; }
    void SetSelectSignalOnly(Bool_t select = kTRUE) { fSelectSignalOnly = select; } 
    AliHFEpidTPC *GetTPCResponse() { return fTPCpid; }

    Bool_t IsTOFmismatch(const AliVTrack *const track, const AliPIDResponse *const pid) const;
    
  private:
    AliHFEreducedEventCreatorESD(const AliHFEreducedEventCreatorESD &);
    AliHFEreducedEventCreatorESD &operator=(const AliHFEreducedEventCreatorESD &);
    
    TTree *fHFEtree;                  // HFE tree 
    AliAnalysisUtils *fAnalysisUtils; // Analysis Utils
    AliHFEreducedEvent *fHFEevent;    // hfe event
    AliHFEcuts *fTrackCuts;           // Track
    AliHFEextraCuts *fExtraCuts;      // HFE IP info
    AliHFEsignalCuts *fSignalCuts;    // Signal Cuts
    AliHFEpidTPC *fTPCpid;            // TPC PID
    AliHFEV0taginfo *fV0Tagger;       // Tags v0 tracks per Event
    AliTRDTriggerAnalysis *fTRDTriggerAnalysis; //! TRD Trigger Analysis
    Int_t fEventNumber;               // Event Number
    Int_t fNclustersTPC;              // Min Number of clusters in TPC
    Int_t fNclustersTPCPID;           // Min Number of clusters for TPC PID
    Int_t fNclustersITS;              // Min Number of clusters in ITS
    Bool_t fRemoveFirstEvent;         // Remove first event from chunk
    Bool_t fFlagPileupEvents;         // Flag pileup events
    Bool_t fSelectSignalOnly;         // Select signal-only tracks
    
    ClassDef(AliHFEreducedEventCreatorESD, 1)
};
#endif
Commit	Line	Data
3513afb7	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	// Debug tree to look at the distribution of the variable we are cutting on
	17	//
	18	//
	19	#ifndef ALIHFEREDUCEDEVENTCREATORESD_H
	20	#define ALIHFEREDUCEDEVENTCREATORESD_H
	21
	22	#include "AliAnalysisTaskSE.h"
	23
	24	class TString;
	25	class TTree;
	26	class AliAnalysisUtils;
	27	class AliPIDResponse;
	28	class AliHFEcuts;
	29	class AliHFEextraCuts;
	30	class AliHFEpidTPC;
4437a0d2	31	class AliTRDTriggerAnalysis;
3513afb7	32	class AliHFEsignalCuts;
3513afb7	33	class AliHFEreducedEvent;
ff8249bd	34	class AliHFEV0taginfo;
3513afb7	35
	36	class AliHFEreducedEventCreatorESD : public AliAnalysisTaskSE{
	37	public:
	38	AliHFEreducedEventCreatorESD();
	39	AliHFEreducedEventCreatorESD(const char *name);
	40	virtual ~AliHFEreducedEventCreatorESD();
	41
	42	virtual void UserCreateOutputObjects();
	43	virtual void UserExec(Option_t *);
	44	virtual void Terminate(Option_t *);
	45
	46	// Setters for cuts
	47	void SetMinNclustersTPC(Int_t mincl) { fNclustersTPC = mincl; };
	48	void SetMinNclustersTPCPID(Int_t mincl) { fNclustersTPCPID = mincl; };
	49	void SetMinNclustersITS(Int_t mincl) { fNclustersITS = mincl; };
	50	void SetRemoveFirstEventFromChunk() { fRemoveFirstEvent = kTRUE; }
ff8249bd	51	void SetFlagPileupEvents() { fFlagPileupEvents = kTRUE; }
ff8249bd	52	void SetSelectSignalOnly(Bool_t select = kTRUE) { fSelectSignalOnly = select; }
3513afb7	53	AliHFEpidTPC *GetTPCResponse() { return fTPCpid; }
	54
	55	Bool_t IsTOFmismatch(const AliVTrack const track, const AliPIDResponse const pid) const;
	56
	57	private:
	58	AliHFEreducedEventCreatorESD(const AliHFEreducedEventCreatorESD &);
	59	AliHFEreducedEventCreatorESD &operator=(const AliHFEreducedEventCreatorESD &);
	60
	61	TTree *fHFEtree; // HFE tree
	62	AliAnalysisUtils *fAnalysisUtils; // Analysis Utils
	63	AliHFEreducedEvent *fHFEevent; // hfe event
	64	AliHFEcuts *fTrackCuts; // Track
	65	AliHFEextraCuts *fExtraCuts; // HFE IP info
	66	AliHFEsignalCuts *fSignalCuts; // Signal Cuts
	67	AliHFEpidTPC *fTPCpid; // TPC PID
ff8249bd	68	AliHFEV0taginfo *fV0Tagger; // Tags v0 tracks per Event
4437a0d2	69	AliTRDTriggerAnalysis *fTRDTriggerAnalysis; //! TRD Trigger Analysis
3513afb7	70	Int_t fEventNumber; // Event Number
	71	Int_t fNclustersTPC; // Min Number of clusters in TPC
	72	Int_t fNclustersTPCPID; // Min Number of clusters for TPC PID
	73	Int_t fNclustersITS; // Min Number of clusters in ITS
	74	Bool_t fRemoveFirstEvent; // Remove first event from chunk
ff8249bd	75	Bool_t fFlagPileupEvents; // Flag pileup events
ff8249bd	76	Bool_t fSelectSignalOnly; // Select signal-only tracks
3513afb7	77
3513afb7	78	ClassDef(AliHFEreducedEventCreatorESD, 1)
ff8249bd	79	};
3513afb7	80	#endif
	81
	82