HLT/TPCLib/AliHLTTPCRawSpacePointContainer.h

   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>
  19 using namespace std;
  20
  21 /**
  22  * @class AliHLTTPCRawSpacePointContainer
  23  * Handler class for HLT TPCS raw space point data blocks.
  24  *
  25  * @ingroup alihlt_tpc
  26  */
  27 class 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;
  49   virtual float GetMaxSignal(AliHLTUInt32_t clusterID) const;
  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
  81   virtual int Write(AliHLTUInt8_t* outputPtr, AliHLTUInt32_t size,
  82                     vector<AliHLTComponentBlockData>& outputBlocks,
  83                     const char* option="") const;
  84
  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
 123 ostream& operator<<(ostream &out, const AliHLTTPCRawSpacePointContainer::AliHLTTPCRawSpacePointProperties& p);
 124
 125 #endif