TPC/AliDigits.cxx

   1 /**************************************************************************
   2  * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
   3  *                                                                        *
   4  * Author: The ALICE Off-line Project.                                    *
   5  * Contributors are mentioned in the code where appropriate.              *
   6  *                                                                        *
   7  * Permission to use, copy, modify and distribute this software and its   *
   8  * documentation strictly for non-commercial purposes is hereby granted   *
   9  * without fee, provided that the above copyright notice appears in all   *
  10  * copies and that both the copyright notice and this permission notice   *
  11  * appear in the supporting documentation. The authors make no claims     *
  12  * about the suitability of this software for any purpose. It is          *
  13  * provided "as is" without express or implied warranty.                  *
  14  **************************************************************************/
  15
  16 /* $Id$ */
  17
  18 /*MI change -- for Rule checker
  19           -- added copy constructor and assignmet operator
  20           -- new GetSize return size of object in Bytes
  21           -- added GetDigitSize and GetOverTh function
  22           -- added GetNRows, GetNCols function
  23           -- for Marek -I had it in my code
  24 */
  25
  26 ///////////////////////////////////////////////////////////////////////////////
  27 //                                                                           //
  28 //   Alice  digits array  object  AliDigits                                  //
  29 //                                                                           //
  30 //                                                                           //
  31 //                                                                           //
  32 ///////////////////////////////////////////////////////////////////////////////
  33
  34
  35 #include "TClass.h"
  36 #include <Riostream.h>
  37 #include "TError.h"
  38 #include "AliSegmentID.h"
  39 #include "AliH2F.h"
  40 #include <TArrayI.h>
  41 #include <TArrayS.h>
  42 #include "AliDigits.h"
  43
  44
  45
  46 //_____________________________________________________________________________
  47 //_____________________________________________________________________________
  48 //_____________________________________________________________________________
  49 ClassImp(AliDigits)
  50
  51
  52 AliDigits::AliDigits()
  53 {
  54   //
  55   //default constructor
  56   fIndex = 0;
  57   fElements = 0;
  58   fThreshold =0;
  59   Invalidate();
  60 }
  61
  62 AliDigits::AliDigits(const AliDigits& digits):
  63   AliSegmentID(digits)
  64 {
  65   //
  66   //copy constructor
  67   fNrows = digits.fNrows;
  68   fNcols = digits.fNcols;
  69   fElements = new TArrayS(*(digits.fElements));
  70   fIndex = new TArrayI(*(digits.fIndex));
  71   fBufType = digits.fBufType;
  72   fThreshold = digits.fThreshold;
  73   fNelems    = digits.fNelems;
  74 }
  75
  76 AliDigits & AliDigits::operator =(const AliDigits & digits)
  77 {
  78  //assignment operator
  79   fNrows = digits.fNrows;
  80   fNcols = digits.fNcols;
  81   if (fElements) delete fElements;
  82   fElements = new TArrayS(*(digits.fElements));
  83   if (fIndex) delete fIndex;
  84   fIndex = new TArrayI(*(digits.fIndex));
  85   fBufType = digits.fBufType;
  86   fThreshold = digits.fThreshold;
  87   fNelems    = digits.fNelems;
  88   return (*this);
  89 }
  90
  91 AliDigits::~AliDigits()
  92 {
  93   //
  94   //default destructor
  95   if (fIndex !=0 ) {
  96     delete fIndex;
  97   }
  98   if (fElements != 0) {
  99     delete fElements;
 100   }
 101
 102
 103 }
 104
 105
 106 Bool_t AliDigits::OutOfBoundsError(const char *where, Int_t row, Int_t column)
 107 {
 108    // Generate an out-of-bounds error. Always returns false.
 109    ::Error(where, "row %d  col %d out of bounds (size: %d x %d, this: 0x%08x)",
 110            row, column, fNrows, fNcols, this);
 111    return kFALSE;
 112 }
 113
 114
 115 void AliDigits::Invalidate()
 116 {
 117   //
 118   //set default (invalid parameters)
 119   if (fIndex != 0)  delete  fIndex;
 120   fIndex = new TArrayI;
 121
 122   if (fElements!= 0)     delete  fElements;
 123
 124   fElements = new TArrayS;
 125
 126   fNrows = fNcols =fNelems= -1;
 127   fElements->Set(0);
 128   fIndex->Set(0);
 129   fBufType = -1;
 130 }
 131
 132 void AliDigits::Allocate(Int_t rows, Int_t columns)
 133 {
 134   //
 135   //construct empty buffer fDigits with size rows x columns
 136   Invalidate();
 137   if (rows <= 0) {
 138       Error("Allocate", "no of rows has to be positive");
 139       return;
 140    }
 141    if (columns <= 0) {
 142       Error("Allocate", "no of columns has to be positive");
 143       return;
 144    }
 145   fNrows = rows;
 146   fNcols=columns;
 147   fNelems = fNrows * fNcols;
 148   fElements->Set(fNelems);
 149   fIndex->Set(fNcols);
 150   for (Int_t i =0,k=0; i<fNcols;i++,k+=fNrows)
 151   (*fIndex)[i]=k;
 152   fBufType =0;
 153 }
 154
 155
 156 Int_t AliDigits::GetSize()
 157 {
 158   //
 159   //return size of object
 160   //
 161   Int_t size = sizeof(this);
 162   if (fIndex!=0) size+= sizeof(fIndex)+fIndex->GetSize()*sizeof(Int_t);
 163   if (fElements!=0) size+= sizeof(fElements)+fElements->GetSize()*sizeof(Short_t);
 164   return size;
 165 }
 166
 167 Int_t AliDigits::GetDigitSize() //return total size of pure digit
 168 {
 169   //
 170   //return size of PURE DIGITS
 171   //
 172   if (fElements==0) return 0;
 173   else return sizeof(fElements)+fElements->GetSize()*sizeof(Short_t);
 174 }
 175
 176 Int_t AliDigits::GetOverTh(Float_t threshold,Float_t x1, Float_t x2, Float_t y1, Float_t y2)
 177 {
 178   //
 179   //return number of digits over threshold
 180   //
 181  if ( (fElements==0) || (fElements->GetSize()<=0)) return 0;
 182
 183  if (x1<=x2) {
 184     x1=0;
 185     x2=fNrows;
 186   }
 187   if (y1<=y2) {
 188      y1=0;
 189      y2=fNcols;
 190   }
 191   Int_t over=0;
 192
 193   Bool_t cont=First();
 194   for ( cont=First(); cont==kTRUE;cont=Next()) {
 195     if ( (CurrentRow()<x1) || (CurrentRow()>x2)) continue;
 196     if ( (CurrentColumn()<y1) || (CurrentColumn()>y2)) continue;
 197     if (CurrentDigit()>threshold) over++;
 198   }
 199   return over;
 200 }
 201
 202
 203 Short_t AliDigits::GetDigit(Int_t row, Int_t column)
 204 {
 205   //
 206   // return digit for given row and collumn
 207   if (fBufType ==0) return GetDigitFast(row,column);
 208   if (fBufType ==1) return GetDigit1(row,column);
 209
 210   return 0;
 211 }
 212
 213
 214 void AliDigits::ExpandBuffer()
 215 {
 216   //
 217   //expand buffer to two dimensional array
 218   if (fBufType<0)  {
 219     Error("ExpandBuffer", "buffer doesn't exist");
 220     return;
 221   }
 222   if (fBufType==0)      return;
 223
 224   //expanding of buffer type 1
 225   if (fBufType==1) ExpandBuffer1();
 226
 227   fBufType = 0;
 228 }
 229
 230 void AliDigits::CompresBuffer(Int_t bufferType,Int_t threshold)
 231 {
 232   //
 233   //compres buffer according buffertype algorithm
 234   if (fBufType<0)  {
 235     Error("CompressBuffer", "buffer doesn't exist");
 236     return;
 237   }
 238   if (fBufType == bufferType) return;
 239   //
 240   if (fBufType>0) ExpandBuffer();
 241   if (fBufType !=0)  {
 242     Error("CompressBuffer", "buffer doesn't exist");
 243     return;
 244   }
 245   fThreshold = threshold;
 246   //compress buffer of type 1
 247   if ( bufferType == 1) CompresBuffer1();//end of compresing bufer of type 1
 248 }
 249
 250 Bool_t AliDigits::First()
 251 {
 252   //adjust  first valid current digit
 253   if (fBufType ==0) return First0();
 254   if (fBufType ==1) return First1();
 255   return kFALSE;
 256 }
 257
 258 Bool_t  AliDigits::Next()
 259 {
 260   //addjust next valid current digit
 261   if (fBufType ==0) return Next0();
 262   if (fBufType ==1) return Next1();
 263   return kFALSE;
 264 }
 265
 266 void AliDigits::AcceptHisto(AliH2F * his)
 267 {
 268   //
 269   //make digits buffer with value according histograms values
 270   //for testing purpose
 271   Int_t idim =his->GetNbinsX();
 272   Int_t jdim =his->GetNbinsY();
 273   if ( (idim<1)|| (jdim<1)) {
 274     return;
 275   }
 276   //allocate proper buffer size
 277   Allocate(idim,jdim);
 278   //set digits values
 279   for (Int_t i = 0; i<idim;i++)
 280     for (Int_t j = 0; j<jdim;j++)
 281       {
 282         Int_t index = his->GetBin(i+1,j+1);
 283         SetDigitFast((Short_t)his->GetBinContent(index),i,j);
 284       }
 285 }
 286
 287 AliH2F *  AliDigits::GenerHisto()
 288 {
 289   //
 290   //make digits histo
 291   char ch[30];
 292   sprintf(ch,"Segment_%d ",GetID());
 293   if ( (fNrows<1)|| (fNcols<1)) {
 294     return 0;
 295   }
 296   AliH2F * his  = new AliH2F("Digit histo",ch,fNrows,0,fNrows,fNcols,0,fNcols);
 297   ExpandBuffer();
 298   //set histogram  values
 299   for (Int_t i = 0; i<fNrows;i++)
 300     for (Int_t j = 0; j<fNcols;j++)
 301         his->Fill(i,j,GetDigitFast(i,j));
 302   return his;
 303 }
 304
 305 AliH2F *AliDigits::DrawDigits(const char *option,Float_t x1, Float_t x2, Float_t y1, Float_t y2)
 306 {
 307   //
 308   //draw digits in given array
 309   //
 310   AliH2F *h2f = GenerHisto();
 311   if (x1>=0) {
 312       AliH2F *h2fsub = h2f->GetSubrange2d(x1,x2,y1,y2);
 313       delete h2f;
 314       h2f=h2fsub;
 315   }
 316   if (h2f==0) return 0;
 317   if (option!=0) h2f->Draw(option);
 318   else h2f->Draw();
 319   return h2f;
 320 }
 321
 322 void AliDigits::ExpandBuffer1()
 323 {
 324   //
 325   //expand buffer of type to twodimensional array
 326   Int_t i,k;
 327   fNelems = fNrows*fNcols;
 328   Short_t * buf = new Short_t[fNelems];
 329   memset(buf,0,fNelems*sizeof(Short_t)); //MI change - 4.12.2000
 330   fIndex->Set(fNcols);
 331   for (i =0,k=0 ;i<fNcols;i++,k+=fNrows) (*fIndex)[i]=k;
 332   Int_t col=0;
 333   Int_t row = 0;
 334   Int_t n=fElements->fN;
 335   for (i=0;i<n;i++){
 336     //oposite signa means how many unwrited (under threshold) values
 337     if ((*fElements)[i]<0) row-=fElements->At(i);
 338     else {
 339       buf[(*fIndex)[col]+row]=fElements->At(i);
 340       row++;
 341     }
 342     if (row==fNrows) {
 343       row=0;
 344       col++;
 345     }else
 346       if (row>fNrows){
 347         Invalidate();
 348         return;
 349       }
 350   }
 351   fElements->Adopt(fNelems,buf);
 352 }
 353
 354 void AliDigits::CompresBuffer1()
 355 {
 356   //
 357   //compres buffer according  algorithm 1
 358   //
 359   TArrayS  buf;  //lets have the nearly the "worst case"
 360   buf.Set(fNelems);
 361   TArrayI  index;
 362   index.Set(fNcols);
 363   Int_t icurrent=-1;
 364   Int_t izero;
 365   Short_t * cbuff = fElements->GetArray();  //MI change
 366
 367   for (Int_t col = 0; col<fNcols; col++){
 368     index[col]=icurrent+1;//set collumn pointer
 369     izero = 0;  //reset zer counter at the begining of the column
 370     for (Int_t row = 0; row< fNrows;row++){
 371       //if under threshold
 372       //if (GetDigitFast(row,col)<=fThreshold)  izero++;
 373       if (*cbuff<=fThreshold)  izero++;
 374
 375       else{
 376         if (izero>0) {
 377           //if we have currently izero count under threshold
 378           icurrent++;
 379           if (icurrent>=buf.fN) buf.Set(icurrent*2);
 380           buf[icurrent]= -izero;  //write how many under zero
 381           izero = 0;
 382         } //end of reseting izero
 383         icurrent++;
 384         if (icurrent>=buf.fN) buf.Set(icurrent*2);
 385         //buf[icurrent] = GetDigitFast(row,col);
 386         buf[icurrent] = *cbuff;
 387       }//if signal bigger then threshold
 388        cbuff++;
 389     } //end of loop over rows
 390     if (izero>0) {
 391       icurrent++;
 392       if (icurrent>=buf.fN) buf.Set(icurrent*2);
 393       buf[icurrent]= -izero;  //write how many under zero
 394     }
 395   }//end of lopping over digits
 396   buf.Set(icurrent+1);
 397   (*fElements)=buf;
 398   fNelems = fElements->fN;
 399   fBufType = 1;
 400   (*fIndex) =index;
 401   //end of compresing bufer of type 1
 402 }
 403
 404
 405
 406 Bool_t AliDigits::First0()
 407 {
 408   //
 409   //first for the buffer type 0
 410   fCurrentRow = -1;
 411   fCurrentCol = -1;
 412   fCurrentIndex = -1;
 413   Int_t i;
 414   for (i=0; (( i<fNelems) && (fElements->At(i)<=fThreshold));i++);  //MI1211
 415   if (i == fNelems) return kFALSE;
 416   fCurrentCol =i/fNrows;
 417   fCurrentRow =i%fNrows;
 418   fCurrentIndex = i;
 419   return kTRUE;
 420 }
 421
 422 Bool_t AliDigits::Next0()
 423 {
 424   //
 425   //next for the buffer type 0
 426   //
 427   if (fCurrentIndex<0) return kFALSE;  // if we didn't adjust first
 428   Int_t i;
 429   for (i=fCurrentIndex+1; ( (i<fNelems) && (fElements->At(i)<=fThreshold) ) ;i++);
 430   if (i >= fNelems)  {
 431     fCurrentIndex = -1;
 432     return kFALSE;
 433   }
 434   fCurrentCol =i/fNrows;
 435   fCurrentRow =i%fNrows;
 436   fCurrentIndex = i;
 437   return kTRUE;
 438 }
 439
 440 Bool_t AliDigits::First1()
 441 {
 442   //
 443   //first for the buffer type 1
 444   fCurrentRow = -1;
 445   fCurrentCol = 0;
 446   fCurrentIndex = -1;
 447   Int_t i;
 448   for (i=0; i<fNelems; i++){
 449     if (fElements->At(i) < 0) fCurrentRow-=fElements->At(i);
 450     else
 451       fCurrentRow++;
 452     if (fCurrentRow>=fNrows) {
 453        fCurrentCol++;
 454        fCurrentRow-=fNrows;
 455     }
 456     if (fElements->At(i)>fThreshold) break;
 457   }
 458   fCurrentIndex = i;
 459   if (fCurrentIndex>=0&&i<fNelems) return kTRUE;
 460   fCurrentRow =-1;
 461   fCurrentCol =-1;
 462   return kFALSE;
 463 }
 464
 465 Bool_t AliDigits::Next1()
 466 {
 467   //
 468   //next for the buffer type 1
 469   if (fCurrentIndex<0) return kFALSE;  // if we didn't adjust first
 470   Int_t i;
 471   for (i=fCurrentIndex+1; i<fNelems;i++){
 472     if (fElements->At(i) < 0) fCurrentRow-=fElements->At(i);
 473     else
 474       fCurrentRow++;
 475     if (fCurrentRow>=fNrows) {
 476       fCurrentCol++;
 477       fCurrentRow-=fNrows;
 478     }
 479     if (fElements->At(i)>fThreshold) break;
 480   }
 481   fCurrentIndex = i;
 482   if ( (i>=0) && (i<fNelems) ) return kTRUE;
 483   fCurrentRow =-1;
 484   fCurrentCol =-1;
 485   return kFALSE;
 486 }
 487
 488 Short_t AliDigits::GetDigit1(Int_t row, Int_t column)
 489 {
 490   //
 491   //return digit for given row and column  the buffer type 1
 492   //no control performed
 493
 494   Int_t i,n2;
 495   if ( (column+1)>=fNcols) n2 = fNelems;
 496   else
 497     n2 = fIndex->At(column+1);
 498   Int_t irow = 0; //current row
 499
 500   for (i=fIndex->At(column); ( (i<n2) && (irow<row) );i++){
 501     if (fElements->At(i) < 0) irow-=fElements->At(i);
 502     else
 503       irow++;
 504   }
 505   if ( irow == row ) return fElements->At(i);
 506   return -1;
 507 }
 508