de73de4e072df87a99026b1d9b5ecd29cf19344e
[u/mrichter/AliRoot.git] / HLT / comp / AliL3DataCompressor.h
1 // @(#) $Id$
2
3 #ifndef AliL3_DataCompressor
4 #define AliL3_DataCompressor
5
6 class AliL3SpacePointData;
7 class AliL3Benchmark;
8 class AliL3TrackArray;
9 class AliL3Track;
10
11 #ifdef use_root
12 class TH2F;
13 class TFile;
14 #endif
15
16 struct TempCluster {
17   Float_t pad;
18   Float_t time;
19   Float_t sigmaY2;
20   Float_t sigmaZ2;
21   Int_t charge;
22   Int_t padrow;
23 };
24
25 class AliL3DataCompressor {
26   
27  private:
28   AliL3Benchmark *fBenchmark;    //!
29   AliL3TrackArray *fInputTracks; //!
30   AliL3SpacePointData *fClusters[36][6]; //!
31   ofstream *fCompRatioFile;      //!
32 #ifdef use_root
33   TFile *fOutputFile;            //!
34 #else
35   FILE *fOutputFile;
36 #endif
37   
38   UInt_t fNcl[36][6];
39   
40   static Int_t fNumPadBits;
41   static Int_t fNumTimeBits;
42   static Int_t fNumChargeBits;
43   static Int_t fNumShapeBits;
44   
45   static Float_t fXYResidualStep1;
46   static Float_t fXYResidualStep2;
47   static Float_t fXYResidualStep3;
48   static Float_t fZResidualStep1;
49   static Float_t fZResidualStep2;
50   static Float_t fZResidualStep3;
51   static Float_t fXYWidthStep;
52   static Float_t fZWidthStep;
53   static Int_t fClusterCharge;
54   
55   void SelectRemainingClusters();
56   void ExpandTrackData(AliL3TrackArray *tracks);
57   void ReadUncompressedData(TempCluster **clusters,Int_t *ncl,const Int_t maxpoints);
58   void ReadRemaining(TempCluster **clusters,Int_t *ncl,const Int_t maxpoints);
59   void QSort(TempCluster **a, Int_t first, Int_t last);
60   Int_t Compare(TempCluster *a,TempCluster *b);
61   void OpenOutputFile();
62   void CloseOutputFile();
63   
64  protected:
65   Char_t fPath[1024];            //!
66   Int_t fEvent;
67   
68   Bool_t fWriteClusterShape;
69   Bool_t fKeepRemaining;
70   Bool_t fSinglePatch;
71   
72  public:
73   AliL3DataCompressor();
74   AliL3DataCompressor(Char_t *path,Bool_t keep,Bool_t writeshape);
75   virtual ~AliL3DataCompressor();
76   
77   virtual void LoadData(Int_t event,Bool_t sp=kTRUE);
78   virtual void FillData(Int_t minhits,Bool_t expand);
79   virtual void WriteRemaining(Bool_t select);
80   void CompressAndExpand();
81   void RestoreData(Bool_t remaining_only=kFALSE);
82   void DoBench(Char_t *fname="benchmark");
83   
84   void SetBitNumbers(Int_t pad,Int_t time,Int_t charge,Int_t shape);
85   void SetTransverseResolutions(Float_t res1,Float_t res2,Float_t res3,Float_t width=0.005);
86   void SetLongitudinalResolutions(Float_t res1,Float_t res2,Float_t res3,Float_t width=0.005);
87   
88   static const Int_t GetNPadBits() {return fNumPadBits;}
89   static const Int_t GetNTimeBits() {return fNumTimeBits;}
90   static const Int_t GetNChargeBits() {return fNumChargeBits;}
91   static const Int_t GetNShapeBits() {return fNumShapeBits;}
92   static const Float_t GetXYWidthStep() {return fXYWidthStep;}
93   static const Float_t GetZWidthStep() {return fZWidthStep;}
94   static const Int_t GetClusterCharge() {return fClusterCharge;}
95   static const Float_t GetXYResidualStep(Int_t row);
96   static const Float_t GetZResidualStep(Int_t row);
97
98
99   ClassDef(AliL3DataCompressor,1) 
100
101 };
102
103 #endif