HLT/src/AliL3MemHandler.h

   1 #ifndef ALIL3MEMHANDLER_H
   2 #define ALIL3MEMHANDLER_H
   3
   4 #include "AliL3RootTypes.h"
   5 #include <stdio.h>
   6 #include <stdlib.h>
   7 #include "AliL3DigitData.h"
   8 #include "AliL3Transform.h"
   9 class AliL3SpacePointData;
  10 class AliL3DigitRowData;
  11 class AliL3TrackSegmentData;
  12 class AliL3TrackArray;
  13 class AliL3RandomPointData;
  14
  15 class AliL3MemHandler{
  16  private:
  17   AliL3Transform *fTransformer;//!
  18   FILE *fInBinary;//!
  19   FILE *fOutBinary;//!
  20   Byte_t *fPt;//!
  21   UInt_t fSize;
  22
  23   AliL3RandomDigitData **fDPt;//!
  24   AliL3RandomDigitData *fDigits;//!
  25   Bool_t IsRandom;
  26   Int_t fNRandom;
  27   Int_t fNGenerate;
  28   Int_t fNUsed;
  29   Int_t fNDigits;
  30
  31   void Write(UInt_t *comp, UInt_t & index, UInt_t & subindex, UShort_t value);
  32   UShort_t Read(UInt_t *comp, UInt_t & index, UInt_t & subindex);
  33   UShort_t Test(UInt_t *comp, UInt_t index, UInt_t subindex);
  34
  35   void DigitizePoint(Int_t row,Int_t pad, Int_t time,Int_t charge);
  36   void QSort(AliL3RandomDigitData **a, Int_t first, Int_t last);
  37   Int_t ComparePoints(UInt_t row,UShort_t pad,UShort_t time);
  38   Int_t CompareDigits(AliL3RandomDigitData *a,AliL3RandomDigitData *b);
  39   void AddData(AliL3DigitData *data,UInt_t & ndata,
  40                       UInt_t row,UShort_t pad,UShort_t time,UShort_t charge);
  41   void AddRandom(AliL3DigitData *data,UInt_t & ndata);
  42   void MergeDataRandom(AliL3DigitData *data,UInt_t & ndata,
  43                       UInt_t row,UShort_t pad,UShort_t time,UShort_t charge);
  44   void AddDataRandom(AliL3DigitData *data,UInt_t & ndata,
  45                       UInt_t row,UShort_t pad,UShort_t time,UShort_t charge);
  46
  47  protected:
  48   Int_t fRowMin;
  49   Int_t fRowMax;
  50   Int_t fSlice;
  51   Int_t fPatch;
  52
  53   Int_t fEtaMinTimeBin[176];
  54   Int_t fEtaMaxTimeBin[176];
  55
  56
  57  public:
  58   AliL3MemHandler();
  59   virtual ~AliL3MemHandler();
  60
  61   void Reset(){CloseBinaryInput();CloseBinaryOutput();Free();}
  62   void Init(AliL3Transform *t){fTransformer = t;}
  63   void Init(Int_t s,Int_t p,const Int_t* row){fSlice=s;fPatch=p;fRowMin=row[0];fRowMax=row[1]; ResetROI();}
  64
  65   Bool_t SetBinaryInput(char *name);
  66   Bool_t SetBinaryInput(FILE *file);
  67   void CloseBinaryInput();
  68
  69   Bool_t SetBinaryOutput(char *name);
  70   Bool_t SetBinaryOutput(FILE *file);
  71   void CloseBinaryOutput();
  72
  73   //Random cluster
  74   void SetRandomCluster(Int_t maxnumber);
  75   void SetRandomSeed(UInt_t seed){srand(seed);}
  76   void SetRandomSeed();
  77
  78   void ResetRandom(){fNDigits = 0; fNUsed = 0;}
  79   void Generate(Int_t row);
  80   void SetNGenerate(Int_t number){(number>fNRandom)?fNGenerate=fNRandom:fNGenerate = number;}
  81
  82   void SetROI(Float_t *eta,Int_t *slice);
  83   void ResetROI();
  84
  85   //Digit IO
  86   Bool_t Memory2Binary(UInt_t nrow,AliL3DigitRowData *data);
  87   Bool_t Binary2Memory(UInt_t & nrow,AliL3DigitRowData *data);
  88
  89   Int_t Memory2CompMemory(UInt_t nrow,AliL3DigitRowData *data,UInt_t *comp);
  90   Int_t CompMemory2Memory(UInt_t nrow,AliL3DigitRowData *data,UInt_t *comp);
  91   Bool_t CompMemory2CompBinary(UInt_t nrow,UInt_t *comp, UInt_t size=0);
  92   Bool_t CompBinary2CompMemory(UInt_t & nrow,UInt_t *comp);
  93
  94   AliL3DigitRowData *CompBinary2Memory(UInt_t & nrow);
  95   Bool_t Memory2CompBinary(UInt_t nrow,AliL3DigitRowData *data);
  96   UInt_t GetNRow(UInt_t *comp,UInt_t size);
  97
  98   //Point IO
  99   Bool_t Memory2Binary(UInt_t npoint,AliL3SpacePointData *data);
 100   Bool_t Binary2Memory(UInt_t & npoint,AliL3SpacePointData *data);
 101   Bool_t Transform(UInt_t npoint,AliL3SpacePointData *data,
 102                               Int_t slice, AliL3Transform* trans);
 103   static void UpdateRowPointer(AliL3DigitRowData *&tempPt);
 104
 105   //Track IO
 106   Bool_t Memory2Binary(UInt_t ntrack,AliL3TrackSegmentData *data);
 107   Bool_t Binary2Memory(UInt_t & ntrack,AliL3TrackSegmentData *data);
 108   Bool_t TrackArray2Binary(AliL3TrackArray *array);
 109   Bool_t Binary2TrackArray(AliL3TrackArray *array);
 110   Bool_t TrackArray2Memory(UInt_t & ntrack,AliL3TrackSegmentData *data,
 111                                              AliL3TrackArray *array);
 112   Bool_t Memory2TrackArray(UInt_t ntrack,AliL3TrackSegmentData *data,
 113                                              AliL3TrackArray *array);
 114   Bool_t Memory2TrackArray(UInt_t ntrack,AliL3TrackSegmentData *data,
 115                            AliL3TrackArray *array,Int_t slice, AliL3Transform* trans);
 116
 117
 118   //Memory Allocation
 119   UInt_t GetAllocatedSize(){return fSize;}
 120   UInt_t GetFileSize();
 121   UInt_t GetMemorySize(UInt_t nrow,UInt_t *comp);
 122   UInt_t GetCompMemorySize(UInt_t nrow,AliL3DigitRowData *data);
 123   UInt_t GetRandomSize();
 124
 125   Byte_t *Allocate(UInt_t size);
 126   Byte_t *Allocate();  // allocate size of Binary Input File
 127   Byte_t *Allocate(AliL3TrackArray *array);
 128   Byte_t *GetDataPointer(UInt_t &size) {size = fSize; return fPt;}
 129   void   Free();
 130
 131   ClassDef(AliL3MemHandler,1) // Memory handler class
 132 };
 133
 134 inline Int_t  AliL3MemHandler::ComparePoints(UInt_t row,UShort_t pad,UShort_t time){
 135   if(fNUsed>=fNDigits) return -2;
 136
 137   if(pad==fDPt[fNUsed]->fPad&&time==fDPt[fNUsed]->fTime) return 0;
 138
 139   if(pad<fDPt[fNUsed]->fPad) return -1;
 140   if(pad==fDPt[fNUsed]->fPad&&time<fDPt[fNUsed]->fTime)  return -1;
 141
 142   return 1;
 143 }
 144
 145 inline Int_t AliL3MemHandler::CompareDigits(AliL3RandomDigitData *a,AliL3RandomDigitData *b){
 146   if(a->fPad==b->fPad && a->fTime == b->fTime) return 0;
 147
 148   if(a->fPad<b->fPad) return -1;
 149   if(a->fPad==b->fPad && a->fTime<b->fTime) return -1;
 150
 151   return 1;
 152 }
 153
 154 #endif