First version, stand alone detector

[u/mrichter/AliRoot.git] / HLT / comp / AliL3Compress.h

#ifndef AliL3_Compress
#define AliL3_Compress

#include "AliL3RootTypes.h"
#include "AliL3Models.h"
#include "AliL3DigitData.h"

class AliL3TrackArray;

class AliL3Compress {
  
 private:
  AliL3TrackArray *fTracks; //!
  AliL3RandomDigitData *fDigits; //!
  AliL3RandomDigitData **fDPt; //!
  Int_t fNDigits;
  Int_t fNUsed;
  Int_t fMaxDigits;
  
  Int_t fNumPadBits;
  Int_t fNumTimeBits;
  Int_t fNumChargeBits;
  Int_t fNumShapeBits;
  Int_t fSlice;
  Int_t fPatch;
  Char_t fPath[100];
  Bool_t fWriteShape;
  
  void CreateDigitArray(Int_t maxnumber);
  void CreateDigits(Int_t row,Int_t npads,Float_t pad,Float_t time,Int_t charge,Float_t ywidth,Float_t zwidth);
  void QSort(AliL3RandomDigitData **a, Int_t first, Int_t last);
  Int_t ComparePoints(Int_t row,UShort_t pad,UShort_t time);
  Int_t CompareDigits(AliL3RandomDigitData *a,AliL3RandomDigitData *b);

 public:
  AliL3Compress();
  AliL3Compress(Int_t slice,Int_t patch,Char_t *path="./",Bool_t writeshape=kFALSE);
  virtual ~AliL3Compress();
  
  void SetBitNumbers(Int_t pad,Int_t time,Int_t charge,Int_t shape);
  void WriteFile(AliL3TrackArray *tracks);
  void ReadFile(Char_t which);
  void CompressFile();
  void ExpandFile();
  void RestoreData(Char_t which='u');
  void WriteRestoredData();
  void WriteRootFile(Char_t *newrootfile);
  void PrintDigits(Int_t padrow=-1);
  void PrintCompRatio();
  
  AliL3TrackArray *GetTracks() {return fTracks;}
  
  ClassDef(AliL3Compress,1) 

};

inline Int_t  AliL3Compress::ComparePoints(Int_t row,UShort_t pad,UShort_t time)
{
  if(fNUsed >= fNDigits) return 0;
  
  if(fDPt[fNUsed]->fRow != row) return 0;
  
  
  if(fDPt[fNUsed]->fPad < pad) return 1;
  if(fDPt[fNUsed]->fPad == pad && fDPt[fNUsed]->fTime < time) return 1;
  
  //if(fDPt[fNUsed]->fPad == pad && fDPt[fNUsed]->fTime == time) return 2;
  
  return 0;

}

inline Int_t AliL3Compress::CompareDigits(AliL3RandomDigitData *a,AliL3RandomDigitData *b)
{
  if(a->fRow < b->fRow) return -1;
    
  if(a->fPad==b->fPad && a->fTime == b->fTime && a->fRow == b->fRow) return 0;
  
  if(a->fPad<b->fPad && a->fRow == b->fRow) return -1;
  if(a->fPad==b->fPad && a->fTime<b->fTime && a->fRow == b->fRow) return -1;
  
  return 1;
}

#endif