added track covariance matrix (yz plane) to the tracklet

[u/mrichter/AliRoot.git] / TRD / AliTRDarrayDictionary.cxx

/************************************************************************* 
* Copyright(c) 1998-2008, ALICE Experiment at CERN, All rights reserved. * 
*                                                                        * 
* Author: The ALICE Off-line Project.                                    * 
* Contributors are mentioned in the code where appropriate.              * 
*                                                                        * 
* Permission to use, copy, modify and distribute this software and its   * 
* documentation strictly for non-commercial purposes is hereby granted   * 
* without fee, provided that the above copyright notice appears in all   * 
* copies and that both the copyright notice and this permission notice   * 
* appear in the supporting documentation. The authors make no claims     * 
* about the suitability of this software for any purpose. It is          * 
* provided "as is" without express or implied warranty.                  *
**************************************************************************/

/* $Id: AliTRDarrayDictionary.cxx 25392 2008-04-23 19:40:29Z cblume $ */

/////////////////////////////////////////////////////////
//                                                     //
// Container Class for Dictionary Info                 //
//                                                     //
// Author:                                             //
//   Hermes Leon Vargas (hleon@ikf.uni-frankfurt.de)   //
//                                                     //
/////////////////////////////////////////////////////////

#include "AliTRDarrayDictionary.h"

ClassImp(AliTRDarrayDictionary)

//________________________________________________________________________________
AliTRDarrayDictionary::AliTRDarrayDictionary()
                      :TObject()
                      ,fNdet(0)
                      ,fNrow(0)
                      ,fNcol(0)
                      ,fNtime(0)
                      ,fNDdim(0)
                      ,fDictionary(0)
{
  //
  // AliTRDarrayDictionary default contructor
  //

}

//________________________________________________________________________________
AliTRDarrayDictionary::AliTRDarrayDictionary(Int_t nrow, Int_t ncol, Int_t ntime)
                      :TObject()
		      ,fNdet(0)
                      ,fNrow(0)
                      ,fNcol(0)
                      ,fNtime(0)
		      ,fNDdim(0)
		      ,fDictionary(0)

{
  //
  // AliTRDarrayDictionary contructor
  //

  Allocate(nrow,ncol,ntime);

}

//________________________________________________________________________________
AliTRDarrayDictionary::AliTRDarrayDictionary(const AliTRDarrayDictionary &a)
                      :TObject()
		      ,fNdet(a.fNdet)
                      ,fNrow(a.fNrow)
                      ,fNcol(a.fNcol)
                      ,fNtime(a.fNtime)
		      ,fNDdim(a.fNDdim)
		      ,fDictionary(0)
{
  //
  // AliTRDarrayDictionary copy constructor
  //

  fDictionary = new Int_t[fNDdim];
  for(Int_t i=0; i<fNDdim; i++)
    {
      fDictionary[i]=a.fDictionary[i];
    }

}

//________________________________________________________________________________
AliTRDarrayDictionary::~AliTRDarrayDictionary()
{
  //
  //   AliTRDarrayDictionary destructor
  //

  if(fDictionary)
    {
      delete [] fDictionary;
      fDictionary=0;
    }

}

//________________________________________________________________________________
AliTRDarrayDictionary &AliTRDarrayDictionary::operator=(const AliTRDarrayDictionary &a)
{
  //
  // Assignment operator
  //

  if(this==&a)
    {
      return *this;
    }

  if(fDictionary)
    {
      delete [] fDictionary;
    }
  fNdet=a.fNdet;
  fNDdim=a.fNDdim;
  fNrow=a.fNrow;
  fNcol=a.fNcol;
  fNtime=a.fNtime;
  fDictionary = new Int_t[fNDdim];
  for(Int_t i=0; i<fNDdim; i++)
    {
      fDictionary[i]=a.fDictionary[i];
    }
  return *this;

}

//________________________________________________________________________________
void AliTRDarrayDictionary::Allocate(Int_t nrow, Int_t ncol, Int_t ntime)
{
  //
  // Allocates memory for the dictionary array with dimensions
  // Row*Col*Time
  // Object initialized to -1
  //

  fNrow=nrow;
  fNcol=ncol;
  fNtime=ntime;
  fNDdim=nrow*ncol*ntime;
  if(fDictionary)
    {
      delete [] fDictionary;
      fDictionary=0;
    }
  fDictionary = new Int_t[fNDdim];
  for(Int_t i=0; i<fNDdim; i++)
    {
      fDictionary[i] = -1; 
    }

}

//________________________________________________________________________________
void AliTRDarrayDictionary::Compress()
{
  //
  // Compress the array
  //


  //  AliDebug(1,"Compressing");
  Int_t counter=0;
  Int_t newDim=0;
  Int_t j;                 
  Int_t r=0;
  Int_t *longArr;            
  longArr = new Int_t[fNDdim];  
  Int_t k=0;
  for(Int_t i=0; i<fNDdim;i++)
    {
      longArr[i]=0;  
    }
  for(Int_t i=0;i<fNDdim; i++)
    {
      j=0;
      if(fDictionary[i]==-1)
	{
	  for(k=i;k<fNDdim;k++)
	    {
	      if(fDictionary[k]==-1)
		{
		  j=j+1;
		  longArr[r]=j;
		}
	      else
		{
		  break;
		}
	    } 
	  r=r+1;    
	}
      i=i+j;
    }
  //Calculate the size of the compressed array
  for(Int_t i=0; i<fNDdim;i++)
    {
      if(longArr[i]!=0)  
	{
	  counter=counter+longArr[i]-1;
	}
    }
  newDim=fNDdim-counter;   //Size of the compressed array
  //Fill the buffer of the compressed array
  Int_t* buffer;
  buffer = new Int_t[newDim];
  Int_t counterTwo=0;
  Int_t g=0;
  for(Int_t i=0; i<newDim; i++)
    {
      if(counterTwo<fNDdim)
	{
	  if(fDictionary[counterTwo]!=-1)
	    {
	      buffer[i]=fDictionary[counterTwo];
	    }
	  if(fDictionary[counterTwo]==-1)
	    {
	      buffer[i]=-(longArr[g]);
	      counterTwo=counterTwo+longArr[g]-1;
	      g++;
	    }  
	  counterTwo++;
	}
    }

  //Copy the buffer
  if(fDictionary)
    {
      delete [] fDictionary;
      fDictionary=0;
    }
  fDictionary = new Int_t[newDim];
  fNDdim = newDim;
  for(Int_t i=0; i<newDim; i++)
    {
      fDictionary[i] = buffer[i]; 
    }
  if(buffer)
    {
      delete [] buffer;
      buffer=0;
    }
  if(longArr) 
    {
      delete [] longArr;
      longArr=0;
    }

}

//________________________________________________________________________________
void AliTRDarrayDictionary::Expand()
{
  //  
  //  Expand the array
  //  

  Int_t *longArr;
  longArr = new Int_t[fNDdim];
  Int_t dimexp=0;
  for(Int_t i=0; i<fNDdim;i++)
    {
      longArr[i]=0;
    }
  Int_t r2=0;
  for(Int_t i=0; i<fNDdim;i++)
    {
      if((fDictionary[i]<0)&&(fDictionary[i]!=-1))  
	{
	  longArr[r2]=-fDictionary[i]; 
	  r2++;
	}
    }

  //Calculate new dimensions
  for(Int_t i=0; i<fNDdim;i++)
    {
      if(longArr[i]!=0)      
	{
	  dimexp=dimexp+longArr[i]-1;
	}
    }
  dimexp=dimexp+fNDdim;  

  //Write in the buffer the new array
  Int_t* bufferE;
  bufferE = new Int_t[dimexp];
  Int_t contaexp =0;    
  Int_t h=0;
  for(Int_t i=0; i<dimexp; i++)
    {
      if(fDictionary[contaexp]>=-1)  
	{
	  bufferE[i]=fDictionary[contaexp];
	}
      if(fDictionary[contaexp]<-1)  
	{
	  for(Int_t j=0; j<longArr[h];j++)
	    {
	      bufferE[i+j]=-1;
	    }
	  i=i+longArr[h]-1;
	  h++;
	}
      contaexp++;
    }

  //Copy the buffer
  if(fDictionary)
    {
      delete [] fDictionary;
      fDictionary=0;
    }

  fDictionary = new Int_t[dimexp];
  fNDdim = dimexp;
  for(Int_t i=0; i<dimexp; i++)
    {
      fDictionary[i] = bufferE[i]; 
    }
  if(bufferE) delete [] bufferE;
  if(longArr) delete [] longArr;

}