Qmax for merged clusters fixed
[u/mrichter/AliRoot.git] / HLT / TPCLib / AliHLTTPCRawSpacePointContainer.h
CommitLineData
71510a33 1//-*- Mode: C++ -*-
2// $Id$
3#ifndef ALIHLTTPCRAWSPACEPOINTCONTAINER_H
4#define ALIHLTTPCRAWSPACEPOINTCONTAINER_H
5//* This file is property of and copyright by the ALICE HLT Project *
6//* ALICE Experiment at CERN, All rights reserved. *
7//* See cxx source for full Copyright notice *
8
9/// @file AliHLTTPCRawSpacePointContainer.h
10/// @author Matthias Richter
11/// @date 2011-08-01
12/// @brief Helper class for handling of HLT TPC raw cluster data blocks
13/// @note Class is a duplicate of AliHLTTPCRawSpacePointContainer and should
14/// be merged with it in a generic way
15
16#include "AliHLTSpacePointContainer.h"
17#include "AliHLTTPCRawCluster.h"
18#include <map>
19using namespace std;
20
21/**
22 * @class AliHLTTPCRawSpacePointContainer
23 * Handler class for HLT TPCS raw space point data blocks.
24 *
25 * @ingroup alihlt_tpc
26 */
27class AliHLTTPCRawSpacePointContainer : public AliHLTSpacePointContainer
28{
29 public:
30 /// standard constructor
31 AliHLTTPCRawSpacePointContainer();
32 /// copy constructor
33 AliHLTTPCRawSpacePointContainer(const AliHLTTPCRawSpacePointContainer& c);
34 /// assignment operator
35 AliHLTTPCRawSpacePointContainer& operator=(const AliHLTTPCRawSpacePointContainer& c);
36 /// destructor
37 ~AliHLTTPCRawSpacePointContainer();
38
39 virtual bool Check(AliHLTUInt32_t clusterID) const;
40 virtual int GetClusterIDs(vector<AliHLTUInt32_t>& tgt) const;
41 virtual const vector<AliHLTUInt32_t>* GetClusterIDs(AliHLTUInt32_t mask);
42 virtual float GetX(AliHLTUInt32_t clusterID) const;
43 virtual float GetXWidth(AliHLTUInt32_t clusterID) const;
44 virtual float GetY(AliHLTUInt32_t clusterID) const;
45 virtual float GetYWidth(AliHLTUInt32_t clusterID) const;
46 virtual float GetZ(AliHLTUInt32_t clusterID) const;
47 virtual float GetZWidth(AliHLTUInt32_t clusterID) const;
48 virtual float GetCharge(AliHLTUInt32_t clusterID) const;
5ecd199d 49 virtual float GetMaxSignal(AliHLTUInt32_t clusterID) const;
71510a33 50 virtual float GetPhi(AliHLTUInt32_t clusterID) const;
51
52 /// add input block to the collection
53 virtual int AddInputBlock(const AliHLTComponentBlockData* pDesc);
54
55 /// clear the object and reset pointer references
56 virtual void Clear(Option_t * option ="");
57
58 /// print information
59 virtual void Print(ostream& out, Option_t *option="") const;
60
61 /// create a collection of clusters for a space point mask
62 virtual AliHLTSpacePointContainer* SelectByMask(AliHLTUInt32_t mask, bool bAlloc=false) const;
63
64 /// create a collection of clusters for a specific track
65 virtual AliHLTSpacePointContainer* SelectByTrack(int trackId, bool bAlloc=false) const;
66
67 /// create a collection of clusters for a specific MC track
68 virtual AliHLTSpacePointContainer* SelectByMC(int mcId, bool bAlloc=false) const;
69
70 /// create a collection of all used clusters
71 virtual AliHLTSpacePointContainer* UsedClusters(bool bAlloc=false) const;
72
73 /// create a collection of all unused clusters
74 virtual AliHLTSpacePointContainer* UnusedClusters(bool bAlloc=false) const;
75
76 virtual int MarkUsed(const AliHLTUInt32_t* clusterIDs, int arraySize);
77 virtual int SetTrackID(int trackID, const AliHLTUInt32_t* clusterIDs, int arraySize);
78 virtual int GetTrackID(AliHLTUInt32_t clusterID) const;
79 virtual int SetMCID(int clusterID, const AliHLTUInt32_t* clusterIDs, int arraySize);
80
c2b3c776 81 virtual int Write(AliHLTUInt8_t* outputPtr, AliHLTUInt32_t size,
82 vector<AliHLTComponentBlockData>& outputBlocks,
83 const char* option="") const;
84
71510a33 85 class AliHLTTPCRawSpacePointProperties {
86 public:
87 AliHLTTPCRawSpacePointProperties();
88 AliHLTTPCRawSpacePointProperties(const AliHLTTPCRawCluster* pCluster);
89 AliHLTTPCRawSpacePointProperties(const AliHLTTPCRawSpacePointProperties& src);
90 AliHLTTPCRawSpacePointProperties& operator=(const AliHLTTPCRawSpacePointProperties& src);
91
92 ~AliHLTTPCRawSpacePointProperties() {}
93
94 const AliHLTTPCRawCluster* Data() const {return fCluster;}
95 bool IsUsed() const {return fUsed;}
96 void MarkUsed(bool used=true) {fUsed=used;}
97 int GetTrackId() const {return fTrackId;}
98 void SetTrackId(int trackId) {fTrackId=trackId;}
99 int GetMCId() const {return fMCId;}
100 void SetMCId(int mcId) {fMCId=mcId;}
101
102 void Print(ostream& out, Option_t *option="") const;
103
104 private:
105 const AliHLTTPCRawCluster* fCluster; //! transient
106 bool fUsed; //! transient
107 int fTrackId; //! track id from reconstruction
108 int fMCId; //! MC id
109 };
110
111 protected:
112
113 private:
114 /// map of clusters
115 std::map<AliHLTUInt32_t, AliHLTTPCRawSpacePointProperties> fClusters; //!
116
117 /// map of cluster id collection for different masks
118 std::map<AliHLTUInt32_t, vector<AliHLTUInt32_t>*> fSelections; //!
119
120 ClassDef(AliHLTTPCRawSpacePointContainer, 0)
121};
122
123ostream& operator<<(ostream &out, const AliHLTTPCRawSpacePointContainer::AliHLTTPCRawSpacePointProperties& p);
124
125#endif