STEER/AliLHCDipValT.h

   1 #ifndef ALILHCDIPVALT_H
   2 #define ALILHCDIPVALT_H
   3
   4 #include <typeinfo>
   5 #include <TString.h>
   6 #include <TObjString.h>
   7 #include <TObject.h>
   8 #include "AliLog.h"
   9
  10
  11 /////////////////////////////////////////////////////////////////////////////////
  12 //                                                                             //
  13 //  AliLHCDipValT: templated class to store the array from processed           //
  14 //  LHC DIP data. Associated with the time stamp either of the value           //
  15 //  acquisition or the last sample timestamp if many samples were processed    //
  16 //                                                                             //
  17 //  Author: ruben.shahoyan@cern.ch                                             //
  18 //                                                                             //
  19 /////////////////////////////////////////////////////////////////////////////////
  20
  21 //_______________________________________________________________________________________
  22 template<class Element> class AliLHCDipValT : public TObject
  23 {
  24  public:
  25   //
  26   enum {
  27     kLastSpecial=BIT(14),         // last element of array is special (e.g. assigned error etc)
  28     kProcessed1 =BIT(15),         // last element of array is special (e.g. assigned error etc)
  29     kChar=BIT(22),
  30     kFloat=BIT(23)
  31   };
  32   //
  33   AliLHCDipValT(Int_t size=0, Double_t t=0);
  34   AliLHCDipValT(const AliLHCDipValT<Element> &src);
  35   virtual ~AliLHCDipValT() {delete[] fArray;}
  36   AliLHCDipValT& operator=(const AliLHCDipValT<Element> &src);
  37   Element&       operator[](Int_t i);
  38   Element        operator[](Int_t i)                          const;
  39   //
  40   void                SetSize(Int_t size);
  41   void                SetValue(Int_t i,Element v);
  42   void                SetValues(const Element *v, Int_t n);
  43   void                SetTimeStamp(Double_t v)                      {fTimeStamp = v;}
  44   //
  45   Int_t               GetSizeTotal()                          const {return GetUniqueID();}
  46   Element             GetValue(Int_t i=0)                     const;
  47   Element*            GetValues()                             const {return (Element*)fArray;}
  48   Double_t            GetTimeStamp()                          const {return fTimeStamp;}
  49   Char_t*             GetTimeAsString(Bool_t utc=kTRUE)       const {return TimeAsString(fTimeStamp,utc);}
  50   //
  51   void                SetProcessed1(Bool_t v=kTRUE)                 {SetBit(kProcessed1,v);}
  52   void                SetLastSpecial(Bool_t v=kTRUE)                {SetBit(kLastSpecial,v);}
  53   Bool_t              IsProcessed1()                          const {return TestBit(kProcessed1);}
  54   Bool_t              IsLastSpecial()                         const {return TestBit(kLastSpecial);}
  55   Int_t               GetSize()                               const {return IsLastSpecial() ? GetSizeTotal()-1:GetSizeTotal();}
  56   Bool_t              IsTypeC()                               const {return TestBit(kChar);}
  57   Bool_t              IsTypeF()                               const {return TestBit(kFloat);}
  58   Bool_t              IsTypeI()                               const {return !TestBit(kFloat|kChar);}
  59   //
  60   virtual void Clear(const Option_t *opt="");
  61   virtual void Print(const Option_t *opt="")                  const;
  62   //
  63   static Char_t*      TimeAsString(double t,Bool_t utc=kTRUE);
  64   //
  65  protected:
  66   //
  67   Double_t            fTimeStamp;        // timestamp of the entry
  68   Element*            fArray;            //[fUniqueID] array of entries
  69   //
  70   ClassDef(AliLHCDipValT,1)
  71 };
  72
  73
  74 //__________________________________________________________________________
  75 template<class Element>
  76 AliLHCDipValT<Element>::AliLHCDipValT(Int_t size,Double_t t)
  77 : fTimeStamp(t),fArray(0)
  78 {
  79   //def. constructor
  80   SetSize(size);
  81   if      (!strcmp(typeid(fArray).name(),typeid(Char_t*).name())) SetBit(kChar);
  82   else if (!strcmp(typeid(fArray).name(),typeid(Double_t*).name()) ||
  83            !strcmp(typeid(fArray).name(),typeid(Float_t*).name() )) SetBit(kFloat);
  84   //
  85 }
  86
  87 //__________________________________________________________________________
  88 template<class Element>
  89 AliLHCDipValT<Element>::AliLHCDipValT(const AliLHCDipValT<Element> &src)
  90 : TObject(src),fTimeStamp(src.fTimeStamp),fArray(0)
  91 {
  92   //copy constructor
  93   SetUniqueID(0);
  94   SetSize(src.GetSizeTotal());
  95   memcpy(fArray,src.fArray,GetSizeTotal()*sizeof(Element));
  96 }
  97
  98 //__________________________________________________________________________
  99 template<class Element>
 100 AliLHCDipValT<Element>& AliLHCDipValT<Element>::operator=(const AliLHCDipValT<Element> &src)
 101 {
 102   //assingment
 103   if (this != &src) {
 104     ((TObject*)this)->operator=(src);
 105     SetUniqueID(0);
 106     if (GetSizeTotal()!=src.GetSizeTotal()) SetSize(src.GetSizeTotal());
 107     SetTimeStamp(src.GetTimeStamp());
 108     memcpy(fArray,src.fArray,GetSizeTotal()*sizeof(Element));
 109   }
 110   return *this;
 111 }
 112
 113 //__________________________________________________________________________
 114 template<class Element>
 115 Char_t* AliLHCDipValT<Element>::TimeAsString(double t, Bool_t utc)
 116 {
 117   // time as string in UTC or local format
 118   time_t tt = (time_t) t;
 119   char* st = utc ? asctime(gmtime(&tt)) : ctime(&tt);
 120   *(strchr(st,'\n')) = 0;
 121   return st;
 122 }
 123
 124 //__________________________________________________________________________
 125 template<class Element>
 126 void AliLHCDipValT<Element>::SetValue(Int_t i,Element v)
 127 {
 128   //assign value
 129   if (i>=GetSizeTotal() || i<0) {
 130     AliError(Form("Index %d is out of range 0:%d",i,GetSizeTotal()-1));
 131     return;
 132   }
 133   fArray[i] = v;
 134 }
 135
 136 //__________________________________________________________________________
 137 template<class Element>
 138 void AliLHCDipValT<Element>::SetValues(const Element* v, Int_t n)
 139 {
 140   //assign value
 141   if (n!=GetSizeTotal()) SetSize(n);
 142   memcpy(fArray,v,n*sizeof(Element));
 143 }
 144
 145 //__________________________________________________________________________
 146 template<class Element>
 147 Element& AliLHCDipValT<Element>::operator[](Int_t i)
 148 {
 149   //obtain value refeterence
 150   if (i>=GetSizeTotal() || i<0) {
 151     AliError(Form("Index %d is out of range 0:%d",i,GetSizeTotal()-1));
 152     return fArray[0];
 153   }
 154   return fArray[i];
 155 }
 156
 157 //__________________________________________________________________________
 158 template<class Element>
 159 Element AliLHCDipValT<Element>::operator[](Int_t i) const
 160 {
 161   //obtain value
 162   if (i>=GetSizeTotal() || i<0) {
 163     AliError(Form("Index %d is out of range 0:%d",i,GetSizeTotal()-1));
 164     return 0;
 165   }
 166   return fArray[i];
 167 }
 168
 169 //__________________________________________________________________________
 170 template<class Element>
 171 Element AliLHCDipValT<Element>::GetValue(Int_t i) const
 172 {
 173   //obtain value
 174   if (i>=GetSizeTotal() || i<0) {
 175     AliError(Form("Index %d is out of range 0:%d",i,GetSizeTotal()-1));
 176     return 0;
 177   }
 178   return fArray[i];
 179 }
 180
 181 //__________________________________________________________________________
 182 template<class Element>
 183 void AliLHCDipValT<Element>::SetSize(Int_t sz)
 184 {
 185   //resize
 186   Element* arr = 0;
 187   if (sz>0) {
 188     arr = new Element[sz];
 189     int nc = GetSizeTotal() > sz ? sz:GetSizeTotal(); // n elems to copy
 190     if (nc) memcpy(arr, fArray, nc*sizeof(Element));
 191     if (nc<sz) memset(arr+nc, 0, (sz-nc)*sizeof(Element));
 192     if (GetSizeTotal()) delete[] fArray;
 193     fArray = arr;
 194     SetUniqueID(sz);
 195   }
 196   else {
 197     delete[] fArray;
 198     SetUniqueID(0);
 199   }
 200 }
 201
 202 //__________________________________________________________________________
 203 template<class Element>
 204 void AliLHCDipValT<Element>::Print(const Option_t *opt) const
 205 {
 206   // print time and value
 207   TString str = opt;
 208   str.ToLower();
 209   if (str.Contains("raw")) printf("%.1f ",GetTimeStamp());
 210   else printf("[%s] ",GetTimeAsString(!str.Contains("loc")));
 211   //
 212   TString tp = typeid(fArray).name();
 213   if ( tp==typeid(Char_t*).name() ) printf(": %s\n",(Char_t*)fArray);
 214   else {
 215     int sz = GetSize();
 216     if (sz>1) printf("\n");
 217     Bool_t eolOK = kFALSE;
 218     for (int i=0;i<sz;i++) {
 219       if      (tp == typeid(Int_t*).name()    || tp == typeid(UInt_t*).name() ) {
 220         if (!str.Contains("bit")) printf(" %6d |" ,(Int_t)fArray[i]);
 221         else {
 222           printf(" ");
 223           int val = (int)fArray[i];
 224           for (int j=sizeof(int)*8;j--;) printf("%d",(val>>j)&0x1);
 225           printf(" |");
 226         }
 227
 228       }
 229       else if (tp == typeid(Double_t*).name() || tp == typeid(Float_t*).name()) printf(" %+.3e |",(Float_t)fArray[i]);
 230       else printf(" ");
 231       eolOK = kFALSE;
 232       if ( (i+1)%5 == 0) {printf("\n"); eolOK = kTRUE;}
 233     }
 234     if (IsLastSpecial()) {
 235       if (sz>1 && !eolOK) {printf("\n"); eolOK = kTRUE;}
 236       if (tp == typeid(Double_t*).name() || tp == typeid(Float_t*).name()) {
 237         printf(" Error: %+e\n",(Float_t)fArray[sz]);
 238         eolOK = kTRUE;
 239       }
 240     }
 241     if (!eolOK) printf("\n");
 242   }
 243   //
 244 }
 245
 246 //__________________________________________________________________________
 247 template<class Element>
 248 void AliLHCDipValT<Element>::Clear(const Option_t *)
 249 {
 250   // reset to 0 everything
 251   SetTimeStamp(0);
 252   memset(fArray,0,GetSizeTotal()*sizeof(Element));
 253 }
 254
 255
 256 typedef AliLHCDipValT<Double_t> AliLHCDipValD;
 257 typedef AliLHCDipValT<Float_t>  AliLHCDipValF;
 258 typedef AliLHCDipValT<Int_t>    AliLHCDipValI;
 259 typedef AliLHCDipValT<Char_t>   AliLHCDipValC;
 260
 261
 262
 263 #endif