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

/**************************************************************************
 * Copyright(c) 1998-1999, 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$ */

///////////////////////////////////////////////////////////////////////////////
//                                                                           //
// TRD cluster finder for the slow simulator. 
//                                                                           //
///////////////////////////////////////////////////////////////////////////////

#include <TF1.h>
#include <TTree.h>
#include <TH1.h>
#include <TFile.h>

#include "AliRun.h"
#include "AliRunLoader.h"
#include "AliLoader.h"
#include "AliRawReader.h"

#include "AliTRDclusterizerV1.h"
#include "AliTRDmatrix.h"
#include "AliTRDgeometry.h"
#include "AliTRDdataArrayF.h"
#include "AliTRDdataArrayI.h"
#include "AliTRDdigitsManager.h"
#include "AliTRDparameter.h"
#include "AliTRDpadPlane.h"
#include "AliTRDrawData.h"
#include "AliTRDcalibDB.h"
#include "AliTRDRecParam.h"
#include "AliTRDCommonParam.h"
#include "AliTRDcluster.h"

ClassImp(AliTRDclusterizerV1)

//_____________________________________________________________________________
AliTRDclusterizerV1::AliTRDclusterizerV1():AliTRDclusterizer()
{
  //
  // AliTRDclusterizerV1 default constructor
  //

  fDigitsManager = 0;

}

//_____________________________________________________________________________
AliTRDclusterizerV1::AliTRDclusterizerV1(const Text_t* name, const Text_t* title)
                    :AliTRDclusterizer(name,title)
{
  //
  // AliTRDclusterizerV1 default constructor
  //

  fDigitsManager = new AliTRDdigitsManager();
  fDigitsManager->CreateArrays();

}

//_____________________________________________________________________________
AliTRDclusterizerV1::AliTRDclusterizerV1(const AliTRDclusterizerV1 &c)
:AliTRDclusterizer(c)
{
  //
  // AliTRDclusterizerV1 copy constructor
  //

  ((AliTRDclusterizerV1 &) c).Copy(*this);

}

//_____________________________________________________________________________
AliTRDclusterizerV1::~AliTRDclusterizerV1()
{
  //
  // AliTRDclusterizerV1 destructor
  //

  if (fDigitsManager) {
    delete fDigitsManager;
    fDigitsManager = NULL;
  }

}

//_____________________________________________________________________________
AliTRDclusterizerV1 &AliTRDclusterizerV1::operator=(const AliTRDclusterizerV1 &c)
{
  //
  // Assignment operator
  //

  if (this != &c) ((AliTRDclusterizerV1 &) c).Copy(*this);
  return *this;

}

//_____________________________________________________________________________
void AliTRDclusterizerV1::Copy(TObject &c) const
{
  //
  // Copy function
  //

  ((AliTRDclusterizerV1 &) c).fDigitsManager = 0;

  AliTRDclusterizer::Copy(c);

}

//_____________________________________________________________________________
Bool_t AliTRDclusterizerV1::ReadDigits()
{
  //
  // Reads the digits arrays from the input aliroot file
  //

  if (!fRunLoader) {
    printf("<AliTRDclusterizerV1::ReadDigits> ");
    printf("No input file open\n");
    return kFALSE;
  }
  AliLoader* loader = fRunLoader->GetLoader("TRDLoader");
  if (!loader->TreeD()) loader->LoadDigits();

  // Read in the digit arrays
  return (fDigitsManager->ReadDigits(loader->TreeD()));

}

//_____________________________________________________________________________
Bool_t AliTRDclusterizerV1::ReadDigits(AliRawReader* rawReader)
{
  //
  // Reads the digits arrays from the ddl file
  //

  AliTRDrawData *raw = new AliTRDrawData();
  raw->SetDebug(1);

  fDigitsManager = raw->Raw2Digits(rawReader);

  return kTRUE;

}

//_____________________________________________________________________________
Bool_t AliTRDclusterizerV1::MakeClusters()
{
  //
  // Generates the cluster.
  //

  Int_t row, col, time;

  /*
  if (fTRD->IsVersion() != 1) {
    printf("<AliTRDclusterizerV1::MakeCluster> ");
    printf("TRD must be version 1 (slow simulator).\n");
    return kFALSE; 
  }
  */

  // Get the geometry
  AliTRDgeometry *geo = AliTRDgeometry::GetGeometry(fRunLoader);

  // Create a default parameter class if none is defined
  if (!fPar) {
    fPar = new AliTRDparameter("TRDparameter","Standard TRD parameter");
    printf("<AliTRDclusterizerV1::MakeCluster> ");
    printf("Create the default parameter object.\n");
  }
  fPar->Init();
  
  AliTRDcalibDB* calibration = AliTRDcalibDB::Instance();
  if (!calibration)
  {
    printf("<AliTRDclusterizerMI::MakeCluster> ");
    printf("ERROR getting instance of AliTRDcalibDB");
    return kFALSE;  
  }
  
  AliTRDRecParam* recParam = AliTRDRecParam::Instance();
  if (!recParam)
  {
    printf("<AliTRDclusterizerMI::MakeCluster> ");
    printf("ERROR getting instance of AliTRDRecParam");
    return kFALSE;  
  }
  
  AliTRDCommonParam* commonParam = AliTRDCommonParam::Instance();
  if (!commonParam)
  {
    printf("<AliTRDdigitizer::MakeDigits> ");
    printf("Could not get common params\n");
    return kFALSE;
  }
    
  //Float_t timeBinSize = fPar->GetDriftVelocity()
  //                    / fPar->GetSamplingFrequency();
  // Half of ampl.region
  //  const Float_t kAmWidth = AliTRDgeometry::AmThick()/2.; 

  //Float_t omegaTau = fPar->GetOmegaTau();
  if (fVerbose > 0) {
    //printf("<AliTRDclusterizerV1::MakeCluster> ");
    //printf("OmegaTau = %f \n",omegaTau);
    printf("<AliTRDclusterizerV1::MakeCluster> ");
    printf("Start creating clusters.\n");
  } 

  AliTRDdataArrayI *digits;
  AliTRDdataArrayI *track0;
  AliTRDdataArrayI *track1;
  AliTRDdataArrayI *track2; 

  // Threshold value for the maximum
  Int_t maxThresh = recParam->GetClusMaxThresh();   
  // Threshold value for the digit signal
  Int_t sigThresh = recParam->GetClusSigThresh();   
  // Iteration limit for unfolding procedure
  const Float_t kEpsilon = 0.01;             

  const Int_t   kNclus   = 3;  
  const Int_t   kNsig    = 5;
  const Int_t   kNtrack  = 3 * kNclus;

  Int_t    iType         = 0;
  Int_t    iUnfold       = 0;  
  Double_t ratioLeft     = 1.0;
  Double_t ratioRight    = 1.0;

  //
  Double_t padSignal[kNsig];   
  Double_t clusterSignal[kNclus];
  Double_t clusterPads[kNclus];   
  Int_t    clusterDigit[kNclus];
  Int_t    clusterTracks[kNtrack];   

  Int_t    chamBeg = 0;
  Int_t    chamEnd = AliTRDgeometry::Ncham();
  Int_t    planBeg = 0;
  Int_t    planEnd = AliTRDgeometry::Nplan();
  Int_t    sectBeg = 0;
  Int_t    sectEnd = AliTRDgeometry::Nsect();

  // Start clustering in every chamber
  for (Int_t icham = chamBeg; icham < chamEnd; icham++) {
    for (Int_t iplan = planBeg; iplan < planEnd; iplan++) {
      for (Int_t isect = sectBeg; isect < sectEnd; isect++) {

        Int_t idet = geo->GetDetector(iplan,icham,isect);

        Int_t nClusters      = 0;
        Int_t nClusters2pad  = 0;
        Int_t nClusters3pad  = 0;
        Int_t nClusters4pad  = 0;
        Int_t nClusters5pad  = 0;
        Int_t nClustersLarge = 0;

        if (fVerbose > 0) {
          printf("<AliTRDclusterizerV1::MakeCluster> ");
          printf("Analyzing chamber %d, plane %d, sector %d.\n"
                ,icham,iplan,isect);
	}

        Int_t    nRowMax     = commonParam->GetRowMax(iplan,icham,isect);
        Int_t    nColMax     = commonParam->GetColMax(iplan);
        Int_t    nTimeTotal  = calibration->GetNumberOfTimeBins();

        AliTRDpadPlane *padPlane = commonParam->GetPadPlane(iplan,icham);

        // Get the digits
        digits = fDigitsManager->GetDigits(idet);
        digits->Expand();
        track0 = fDigitsManager->GetDictionary(idet,0);
        track0->Expand();
        track1 = fDigitsManager->GetDictionary(idet,1);
        track1->Expand();
        track2 = fDigitsManager->GetDictionary(idet,2); 
        track2->Expand();

        // Loop through the chamber and find the maxima 
        for ( row = 0;  row <  nRowMax;    row++) {
	  for ( col = 2;  col <  nColMax;    col++) {
            //for ( col = 4;  col <  nColMax-2;    col++) {
            for (time = 0; time < nTimeTotal; time++) {

              Int_t signalL = TMath::Abs(digits->GetDataUnchecked(row,col  ,time));
              Int_t signalM = TMath::Abs(digits->GetDataUnchecked(row,col-1,time));
              Int_t signalR = TMath::Abs(digits->GetDataUnchecked(row,col-2,time));
 
// 	      // Look for the maximum
//               if (signalM >= maxThresh) {
//                 if (((signalL >= sigThresh) &&
//                      (signalL <  signalM))  ||
//                     ((signalR >= sigThresh) &&
//                      (signalR <  signalM))) {
//                   // Maximum found, mark the position by a negative signal
//                   digits->SetDataUnchecked(row,col-1,time,-signalM);
// 		}
// 	      }
	      // Look for the maximum
              if (signalM >= maxThresh) {
                if ( (TMath::Abs(signalL)<=signalM) && (TMath::Abs(signalR)<=signalM) && 
		     (TMath::Abs(signalL)+TMath::Abs(signalR))>sigThresh ) {
                  // Maximum found, mark the position by a negative signal
                  digits->SetDataUnchecked(row,col-1,time,-signalM);
		}
	      }

            }  
          }    
        }      

        // Now check the maxima and calculate the cluster position
        for ( row = 0;  row <  nRowMax  ;  row++) {
          for (time = 0; time < nTimeTotal; time++) {
            for ( col = 1;  col <  nColMax-1;  col++) {

              // Maximum found ?             
              if (digits->GetDataUnchecked(row,col,time) < 0) {

                Int_t iPad;
                for (iPad = 0; iPad < kNclus; iPad++) {
                  Int_t iPadCol = col - 1 + iPad;
                  clusterSignal[iPad]     = TMath::Abs(digits->GetDataUnchecked(row
                                                                               ,iPadCol
                                                                               ,time));
                  clusterDigit[iPad]      = digits->GetIndexUnchecked(row,iPadCol,time);
                  clusterTracks[3*iPad  ] = track0->GetDataUnchecked(row,iPadCol,time) - 1;
		  clusterTracks[3*iPad+1] = track1->GetDataUnchecked(row,iPadCol,time) - 1;
		  clusterTracks[3*iPad+2] = track2->GetDataUnchecked(row,iPadCol,time) - 1;
                }

		// Count the number of pads in the cluster
                Int_t nPadCount = 0;
                Int_t ii        = 0;
                while (TMath::Abs(digits->GetDataUnchecked(row,col-ii  ,time))
                                                                  >= sigThresh) {
                  nPadCount++;
                  ii++;
                  if (col-ii   <        0) break;
		}
                ii = 0;
                while (TMath::Abs(digits->GetDataUnchecked(row,col+ii+1,time))
                                                                  >= sigThresh) {
                  nPadCount++;
                  ii++;
                  if (col+ii+1 >= nColMax) break;
		}

                nClusters++;
                switch (nPadCount) {
                case 2:
                  iType = 0;
                  nClusters2pad++;
                  break;
                case 3:
                  iType = 1;
                  nClusters3pad++;
                  break;
                case 4:
                  iType = 2;
                  nClusters4pad++;
                  break;
                case 5:
                  iType = 3;
                  nClusters5pad++;
                  break;
                default:
                  iType = 4;
                  nClustersLarge++;
                  break;
		};

		 // Look for 5 pad cluster with minimum in the middle
                Bool_t fivePadCluster = kFALSE;
                if (col < nColMax-3) {
                  if (digits->GetDataUnchecked(row,col+2,time) < 0) {
                    fivePadCluster = kTRUE;
	  	  }
                  if ((fivePadCluster) && (col < nColMax-5)) {
                    if (digits->GetDataUnchecked(row,col+4,time) >= sigThresh) {
                      fivePadCluster = kFALSE;
		    }
		  }
                  if ((fivePadCluster) && (col >         1)) {
                    if (digits->GetDataUnchecked(row,col-2,time) >= sigThresh) {
                      fivePadCluster = kFALSE;
		    }
		  }
		}

		// 5 pad cluster
                // Modify the signal of the overlapping pad for the left part 
		// of the cluster which remains from a previous unfolding
                if (iUnfold) {
                  clusterSignal[0] *= ratioLeft;
                  iType   = 5;
                  iUnfold = 0;
		}

		// Unfold the 5 pad cluster
                if (fivePadCluster) {
                  for (iPad = 0; iPad < kNsig; iPad++) {
                    padSignal[iPad] = TMath::Abs(digits->GetDataUnchecked(row
                                                                         ,col-1+iPad
                                                                         ,time));
                  }
                  // Unfold the two maxima and set the signal on 
                  // the overlapping pad to the ratio
                  ratioRight        = Unfold(kEpsilon,iplan,padSignal);
                  ratioLeft         = 1.0 - ratioRight; 
                  clusterSignal[2] *= ratioRight;
                  iType   = 5;
                  iUnfold = 1;
                }


                Double_t clusterCharge = clusterSignal[0]
                                       + clusterSignal[1]
                                       + clusterSignal[2];
                
		// The position of the cluster
                clusterPads[0] = row + 0.5;
		// Take the shift of the additional time bins into account
                clusterPads[2] = time + 0.5;

                // correct for t0
                clusterPads[2] -= calibration->GetT0(idet, col, row);
                
                if (recParam->LUTOn()) {
  		  // Calculate the position of the cluster by using the
		  // lookup table method
                  clusterPads[1] =
                      recParam->LUTposition(iplan,clusterSignal[0]
                                                          ,clusterSignal[1]
					                  ,clusterSignal[2]);
		}
		else {
  		  // Calculate the position of the cluster by using the
		  // center of gravity method
		  for (Int_t i=0;i<5;i++) padSignal[i]=0;
		  padSignal[2] = TMath::Abs(digits->GetDataUnchecked(row,col,time));   // central  pad
		  padSignal[1] = TMath::Abs(digits->GetDataUnchecked(row,col-1,time)); // left     pad
		  padSignal[3] = TMath::Abs(digits->GetDataUnchecked(row,col+1,time)); // right    pad
		  if (col>2 &&TMath::Abs(digits->GetDataUnchecked(row,col-2,time)<padSignal[1])){
		    padSignal[0] = TMath::Abs(digits->GetDataUnchecked(row,col-2,time));
		  }
		  if (col<nColMax-3 &&TMath::Abs(digits->GetDataUnchecked(row,col+2,time)<padSignal[3])){
		    padSignal[4] = TMath::Abs(digits->GetDataUnchecked(row,col+2,time));
		  }		  
		  clusterPads[1] =  GetCOG(padSignal);
		  //Double_t check = fPar->LUTposition(iplan,clusterSignal[0]
                  //                                        ,clusterSignal[1]
		  //	  		                    ,clusterSignal[2]);
		  //		  Float_t diff = clusterPads[1] -  check;

		}

                Double_t q0 = clusterSignal[0];
		Double_t q1 = clusterSignal[1];
                Double_t q2 = clusterSignal[2];
                Double_t clusterSigmaY2 = (q1*(q0+q2)+4*q0*q2) /
                                          (clusterCharge*clusterCharge);

                Float_t vdrift = calibration->GetVdrift(idet, col, row);  
		
                // Calculate the position and the error
                Double_t colSize = padPlane->GetColSize(col);
                Double_t rowSize = padPlane->GetRowSize(row);
                Double_t clusterPos[3];
		clusterPos[0] = padPlane->GetColPos(col) + (clusterPads[1]-0.5)*colSize;  // MI change
		clusterPos[1] = padPlane->GetRowPos(row) -0.5*rowSize; //MI change
                clusterPos[2] = CalcXposFromTimebin(clusterPads[2], vdrift);
                Double_t clusterSig[2];
                clusterSig[0] = (clusterSigmaY2 + 1./12.) * colSize*colSize;
                clusterSig[1] = rowSize * rowSize / 12.;                                       
                
                
                // Add the cluster to the output array 
                AliTRDcluster * cluster = AddCluster(clusterPos
                          ,(Int_t) clusterPads[2]
                          ,idet
			  ,clusterCharge
			  ,clusterTracks
			  ,clusterSig
			   ,iType,clusterPads[1]);
		//
		//
		Short_t signals[7]={0,0,0,0,0,0,0};
		for (Int_t jPad = col-3;jPad<=col+3;jPad++){
		  if (jPad<0 ||jPad>=nColMax-1) continue;
		  signals[jPad-col+3] =  TMath::Abs(digits->GetDataUnchecked(row,jPad,time));
		}
		cluster->SetSignals(signals);
              }
            } 
          }   
        }     

	// Compress the arrays
        digits->Compress(1,0);
        track0->Compress(1,0);
	track1->Compress(1,0);
        track2->Compress(1,0);

        // Write the cluster and reset the array
	WriteClusters(idet);
	ResetRecPoints();
      }    
    }      
  }        

  if (fVerbose > 0) {
    printf("<AliTRDclusterizerV1::MakeCluster> ");
    printf("Done.\n");
  }

  return kTRUE;

}

Double_t AliTRDclusterizerV1::GetCOG(Double_t signal[5])
{
  //
  // get COG position
  // used for clusters with more than 3 pads - where LUT not applicable
  Double_t sum = signal[0]+signal[1]+signal[2]+signal[3]+signal[4];
  Double_t res = (0.0*(-signal[0]+signal[4])+(-signal[1]+signal[3]))/sum;
  return res;		  
}


//_____________________________________________________________________________
Double_t AliTRDclusterizerV1::Unfold(Double_t eps, Int_t plane, Double_t* padSignal)
{
  //
  // Method to unfold neighbouring maxima.
  // The charge ratio on the overlapping pad is calculated
  // until there is no more change within the range given by eps.
  // The resulting ratio is then returned to the calling method.
  //

  AliTRDcalibDB* calibration = AliTRDcalibDB::Instance();
  if (!calibration)
  {
    printf("<AliTRDclusterizerMI::Unfold> ");
    printf("ERROR getting instance of AliTRDcalibDB");
    return kFALSE;  
  }
  
  Int_t   irc                = 0;
  Int_t   itStep             = 0;      // Count iteration steps

  Double_t ratio             = 0.5;    // Start value for ratio
  Double_t prevRatio         = 0;      // Store previous ratio

  Double_t newLeftSignal[3]  = {0};    // Array to store left cluster signal
  Double_t newRightSignal[3] = {0};    // Array to store right cluster signal
  Double_t newSignal[3]      = {0};

  // Start the iteration
  while ((TMath::Abs(prevRatio - ratio) > eps) && (itStep < 10)) {

    itStep++;
    prevRatio = ratio;

    // Cluster position according to charge ratio
    Double_t maxLeft  = (ratio*padSignal[2] - padSignal[0]) 
                      / (padSignal[0] + padSignal[1] + ratio*padSignal[2]);
    Double_t maxRight = (padSignal[4] - (1-ratio)*padSignal[2]) 
                      / ((1-ratio)*padSignal[2] + padSignal[3] + padSignal[4]);

    // Set cluster charge ratio
    irc = calibration->PadResponse(1.0,maxLeft ,plane,newSignal);
    Double_t ampLeft  = padSignal[1] / newSignal[1];
    irc = calibration->PadResponse(1.0,maxRight,plane,newSignal);
    Double_t ampRight = padSignal[3] / newSignal[1];

    // Apply pad response to parameters
    irc = calibration->PadResponse(ampLeft ,maxLeft ,plane,newLeftSignal );
    irc = calibration->PadResponse(ampRight,maxRight,plane,newRightSignal);

    // Calculate new overlapping ratio
    ratio = TMath::Min((Double_t)1.0,newLeftSignal[2] / 
                          (newLeftSignal[2] + newRightSignal[0]));

  }

  return ratio;

}
Commit	Line	Data
f7336fa3	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
88cb7938	16	/* $Id$ */
f7336fa3	17
	18	///////////////////////////////////////////////////////////////////////////////
	19	// //
	20	// TRD cluster finder for the slow simulator.
	21	// //
	22	///////////////////////////////////////////////////////////////////////////////
	23
	24	#include <TF1.h>
94de3818	25	#include <TTree.h>
793ff80c	26	#include <TH1.h>
a819a5f7	27	#include <TFile.h>
f7336fa3	28
793ff80c	29	#include "AliRun.h"
88cb7938	30	#include "AliRunLoader.h"
88cb7938	31	#include "AliLoader.h"
928e9fae	32	#include "AliRawReader.h"
793ff80c	33
f7336fa3	34	#include "AliTRDclusterizerV1.h"
	35	#include "AliTRDmatrix.h"
	36	#include "AliTRDgeometry.h"
6f1e466d	37	#include "AliTRDdataArrayF.h"
793ff80c	38	#include "AliTRDdataArrayI.h"
793ff80c	39	#include "AliTRDdigitsManager.h"
17b26de4	40	#include "AliTRDparameter.h"
a5cadd36	41	#include "AliTRDpadPlane.h"
928e9fae	42	#include "AliTRDrawData.h"
3551db50	43	#include "AliTRDcalibDB.h"
	44	#include "AliTRDRecParam.h"
	45	#include "AliTRDCommonParam.h"
c85a4951	46	#include "AliTRDcluster.h"
f7336fa3	47
	48	ClassImp(AliTRDclusterizerV1)
	49
	50	//_____________________________________________________________________________
	51	AliTRDclusterizerV1::AliTRDclusterizerV1():AliTRDclusterizer()
	52	{
	53	//
	54	// AliTRDclusterizerV1 default constructor
	55	//
	56
17b26de4	57	fDigitsManager = 0;
db30bf0f	58
f7336fa3	59	}
	60
	61	//_____________________________________________________________________________
	62	AliTRDclusterizerV1::AliTRDclusterizerV1(const Text_t* name, const Text_t* title)
	63	:AliTRDclusterizer(name,title)
	64	{
	65	//
	66	// AliTRDclusterizerV1 default constructor
	67	//
	68
6f1e466d	69	fDigitsManager = new AliTRDdigitsManager();
17b26de4	70	fDigitsManager->CreateArrays();
f7336fa3	71
	72	}
	73
8230f242	74	//_____________________________________________________________________________
dd9a6ee3	75	AliTRDclusterizerV1::AliTRDclusterizerV1(const AliTRDclusterizerV1 &c)
73ae7b59	76	:AliTRDclusterizer(c)
8230f242	77	{
	78	//
	79	// AliTRDclusterizerV1 copy constructor
	80	//
	81
dd9a6ee3	82	((AliTRDclusterizerV1 &) c).Copy(*this);
8230f242	83
	84	}
	85
f7336fa3	86	//_____________________________________________________________________________
	87	AliTRDclusterizerV1::~AliTRDclusterizerV1()
	88	{
8230f242	89	//
	90	// AliTRDclusterizerV1 destructor
	91	//
f7336fa3	92
6f1e466d	93	if (fDigitsManager) {
6f1e466d	94	delete fDigitsManager;
abaf1f1d	95	fDigitsManager = NULL;
f7336fa3	96	}
	97
	98	}
	99
dd9a6ee3	100	//_____________________________________________________________________________
	101	AliTRDclusterizerV1 &AliTRDclusterizerV1::operator=(const AliTRDclusterizerV1 &c)
	102	{
	103	//
	104	// Assignment operator
	105	//
	106
	107	if (this != &c) ((AliTRDclusterizerV1 &) c).Copy(*this);
	108	return *this;
	109
	110	}
	111
8230f242	112	//_____________________________________________________________________________
e0d47c25	113	void AliTRDclusterizerV1::Copy(TObject &c) const
8230f242	114	{
	115	//
	116	// Copy function
	117	//
	118
17b26de4	119	((AliTRDclusterizerV1 &) c).fDigitsManager = 0;
8230f242	120
	121	AliTRDclusterizer::Copy(c);
	122
	123	}
	124
f7336fa3	125	//_____________________________________________________________________________
	126	Bool_t AliTRDclusterizerV1::ReadDigits()
	127	{
	128	//
	129	// Reads the digits arrays from the input aliroot file
	130	//
	131
88cb7938	132	if (!fRunLoader) {
17b26de4	133	printf("<AliTRDclusterizerV1::ReadDigits> ");
f7336fa3	134	printf("No input file open\n");
	135	return kFALSE;
	136	}
88cb7938	137	AliLoader* loader = fRunLoader->GetLoader("TRDLoader");
88cb7938	138	if (!loader->TreeD()) loader->LoadDigits();
abaf1f1d	139
f7336fa3	140	// Read in the digit arrays
88cb7938	141	return (fDigitsManager->ReadDigits(loader->TreeD()));
f7336fa3	142
	143	}
	144
928e9fae	145	//_____________________________________________________________________________
	146	Bool_t AliTRDclusterizerV1::ReadDigits(AliRawReader* rawReader)
	147	{
	148	//
	149	// Reads the digits arrays from the ddl file
	150	//
	151
	152	AliTRDrawData *raw = new AliTRDrawData();
	153	raw->SetDebug(1);
	154
	155	fDigitsManager = raw->Raw2Digits(rawReader);
	156
	157	return kTRUE;
	158
	159	}
	160
f7336fa3	161	//_____________________________________________________________________________
793ff80c	162	Bool_t AliTRDclusterizerV1::MakeClusters()
f7336fa3	163	{
	164	//
	165	// Generates the cluster.
	166	//
	167
	168	Int_t row, col, time;
	169
bdbb05bb	170	/*
3e1a3ad8	171	if (fTRD->IsVersion() != 1) {
17b26de4	172	printf("<AliTRDclusterizerV1::MakeCluster> ");
f7336fa3	173	printf("TRD must be version 1 (slow simulator).\n");
	174	return kFALSE;
	175	}
bdbb05bb	176	*/
f7336fa3	177
f7336fa3	178	// Get the geometry
bdbb05bb	179	AliTRDgeometry *geo = AliTRDgeometry::GetGeometry(fRunLoader);
f7336fa3	180
17b26de4	181	// Create a default parameter class if none is defined
	182	if (!fPar) {
	183	fPar = new AliTRDparameter("TRDparameter","Standard TRD parameter");
5443e65e	184	printf("<AliTRDclusterizerV1::MakeCluster> ");
5443e65e	185	printf("Create the default parameter object.\n");
17b26de4	186	}
598156ef	187	fPar->Init();
3551db50	188
	189	AliTRDcalibDB* calibration = AliTRDcalibDB::Instance();
	190	if (!calibration)
	191	{
	192	printf("<AliTRDclusterizerMI::MakeCluster> ");
	193	printf("ERROR getting instance of AliTRDcalibDB");
	194	return kFALSE;
	195	}
	196
	197	AliTRDRecParam* recParam = AliTRDRecParam::Instance();
	198	if (!recParam)
	199	{
	200	printf("<AliTRDclusterizerMI::MakeCluster> ");
	201	printf("ERROR getting instance of AliTRDRecParam");
	202	return kFALSE;
	203	}
	204
	205	AliTRDCommonParam* commonParam = AliTRDCommonParam::Instance();
	206	if (!commonParam)
	207	{
	208	printf("<AliTRDdigitizer::MakeDigits> ");
	209	printf("Could not get common params\n");
	210	return kFALSE;
	211	}
	212
7ad19338	213	//Float_t timeBinSize = fPar->GetDriftVelocity()
7ad19338	214	// / fPar->GetSamplingFrequency();
a819a5f7	215	// Half of ampl.region
7ad19338	216	// const Float_t kAmWidth = AliTRDgeometry::AmThick()/2.;
a819a5f7	217
3551db50	218	//Float_t omegaTau = fPar->GetOmegaTau();
47517f42	219	if (fVerbose > 0) {
3551db50	220	//printf("<AliTRDclusterizerV1::MakeCluster> ");
3551db50	221	//printf("OmegaTau = %f \n",omegaTau);
17b26de4	222	printf("<AliTRDclusterizerV1::MakeCluster> ");
47517f42	223	printf("Start creating clusters.\n");
47517f42	224	}
f7336fa3	225
8230f242	226	AliTRDdataArrayI *digits;
793ff80c	227	AliTRDdataArrayI *track0;
	228	AliTRDdataArrayI *track1;
	229	AliTRDdataArrayI *track2;
f7336fa3	230
3e1a3ad8	231	// Threshold value for the maximum
3551db50	232	Int_t maxThresh = recParam->GetClusMaxThresh();
3e1a3ad8	233	// Threshold value for the digit signal
3551db50	234	Int_t sigThresh = recParam->GetClusSigThresh();
f7336fa3	235	// Iteration limit for unfolding procedure
8230f242	236	const Float_t kEpsilon = 0.01;
f7336fa3	237
8230f242	238	const Int_t kNclus = 3;
8230f242	239	const Int_t kNsig = 5;
3e1a3ad8	240	const Int_t kNtrack = 3 * kNclus;
3e1a3ad8	241
a5cadd36	242	Int_t iType = 0;
7ad19338	243	Int_t iUnfold = 0;
a5cadd36	244	Double_t ratioLeft = 1.0;
a5cadd36	245	Double_t ratioRight = 1.0;
db30bf0f	246
7ad19338	247	//
a5cadd36	248	Double_t padSignal[kNsig];
	249	Double_t clusterSignal[kNclus];
	250	Double_t clusterPads[kNclus];
	251	Int_t clusterDigit[kNclus];
	252	Int_t clusterTracks[kNtrack];
f7336fa3	253
a5cadd36	254	Int_t chamBeg = 0;
	255	Int_t chamEnd = AliTRDgeometry::Ncham();
	256	Int_t planBeg = 0;
	257	Int_t planEnd = AliTRDgeometry::Nplan();
	258	Int_t sectBeg = 0;
	259	Int_t sectEnd = AliTRDgeometry::Nsect();
f7336fa3	260
3e1a3ad8	261	// Start clustering in every chamber
f7336fa3	262	for (Int_t icham = chamBeg; icham < chamEnd; icham++) {
	263	for (Int_t iplan = planBeg; iplan < planEnd; iplan++) {
	264	for (Int_t isect = sectBeg; isect < sectEnd; isect++) {
	265
8230f242	266	Int_t idet = geo->GetDetector(iplan,icham,isect);
f7336fa3	267
db30bf0f	268	Int_t nClusters = 0;
	269	Int_t nClusters2pad = 0;
	270	Int_t nClusters3pad = 0;
	271	Int_t nClusters4pad = 0;
	272	Int_t nClusters5pad = 0;
	273	Int_t nClustersLarge = 0;
3e1a3ad8	274
47517f42	275	if (fVerbose > 0) {
17b26de4	276	printf("<AliTRDclusterizerV1::MakeCluster> ");
47517f42	277	printf("Analyzing chamber %d, plane %d, sector %d.\n"
	278	,icham,iplan,isect);
	279	}
f7336fa3	280
3551db50	281	Int_t nRowMax = commonParam->GetRowMax(iplan,icham,isect);
	282	Int_t nColMax = commonParam->GetColMax(iplan);
	283	Int_t nTimeTotal = calibration->GetNumberOfTimeBins();
5443e65e	284
3551db50	285	AliTRDpadPlane *padPlane = commonParam->GetPadPlane(iplan,icham);
f7336fa3	286
3e1a3ad8	287	// Get the digits
8230f242	288	digits = fDigitsManager->GetDigits(idet);
3e1a3ad8	289	digits->Expand();
793ff80c	290	track0 = fDigitsManager->GetDictionary(idet,0);
3e1a3ad8	291	track0->Expand();
793ff80c	292	track1 = fDigitsManager->GetDictionary(idet,1);
3e1a3ad8	293	track1->Expand();
793ff80c	294	track2 = fDigitsManager->GetDictionary(idet,2);
3e1a3ad8	295	track2->Expand();
	296
	297	// Loop through the chamber and find the maxima
	298	for ( row = 0; row < nRowMax; row++) {
de4b10e5	299	for ( col = 2; col < nColMax; col++) {
3551db50	300	//for ( col = 4; col < nColMax-2; col++) {
3e1a3ad8	301	for (time = 0; time < nTimeTotal; time++) {
3e1a3ad8	302
a819a5f7	303	Int_t signalL = TMath::Abs(digits->GetDataUnchecked(row,col ,time));
	304	Int_t signalM = TMath::Abs(digits->GetDataUnchecked(row,col-1,time));
	305	Int_t signalR = TMath::Abs(digits->GetDataUnchecked(row,col-2,time));
3e1a3ad8	306
7ad19338	307	// // Look for the maximum
	308	// if (signalM >= maxThresh) {
	309	// if (((signalL >= sigThresh) &&
	310	// (signalL < signalM)) \|\|
	311	// ((signalR >= sigThresh) &&
	312	// (signalR < signalM))) {
	313	// // Maximum found, mark the position by a negative signal
	314	// digits->SetDataUnchecked(row,col-1,time,-signalM);
	315	// }
	316	// }
3e1a3ad8	317	// Look for the maximum
db30bf0f	318	if (signalM >= maxThresh) {
de4b10e5	319	if ( (TMath::Abs(signalL)<=signalM) && (TMath::Abs(signalR)<=signalM) &&
de4b10e5	320	(TMath::Abs(signalL)+TMath::Abs(signalR))>sigThresh ) {
3e1a3ad8	321	// Maximum found, mark the position by a negative signal
	322	digits->SetDataUnchecked(row,col-1,time,-signalM);
	323	}
	324	}
	325
	326	}
	327	}
	328	}
	329
	330	// Now check the maxima and calculate the cluster position
	331	for ( row = 0; row < nRowMax ; row++) {
db30bf0f	332	for (time = 0; time < nTimeTotal; time++) {
db30bf0f	333	for ( col = 1; col < nColMax-1; col++) {
3e1a3ad8	334
	335	// Maximum found ?
	336	if (digits->GetDataUnchecked(row,col,time) < 0) {
f7336fa3	337
9d0b222b	338	Int_t iPad;
8230f242	339	for (iPad = 0; iPad < kNclus; iPad++) {
3e1a3ad8	340	Int_t iPadCol = col - 1 + iPad;
	341	clusterSignal[iPad] = TMath::Abs(digits->GetDataUnchecked(row
	342	,iPadCol
	343	,time));
	344	clusterDigit[iPad] = digits->GetIndexUnchecked(row,iPadCol,time);
	345	clusterTracks[3*iPad ] = track0->GetDataUnchecked(row,iPadCol,time) - 1;
	346	clusterTracks[3*iPad+1] = track1->GetDataUnchecked(row,iPadCol,time) - 1;
	347	clusterTracks[3*iPad+2] = track2->GetDataUnchecked(row,iPadCol,time) - 1;
f7336fa3	348	}
f7336fa3	349
db30bf0f	350	// Count the number of pads in the cluster
	351	Int_t nPadCount = 0;
	352	Int_t ii = 0;
	353	while (TMath::Abs(digits->GetDataUnchecked(row,col-ii ,time))
	354	>= sigThresh) {
	355	nPadCount++;
	356	ii++;
	357	if (col-ii < 0) break;
	358	}
	359	ii = 0;
	360	while (TMath::Abs(digits->GetDataUnchecked(row,col+ii+1,time))
	361	>= sigThresh) {
	362	nPadCount++;
	363	ii++;
	364	if (col+ii+1 >= nColMax) break;
	365	}
	366
	367	nClusters++;
	368	switch (nPadCount) {
	369	case 2:
	370	iType = 0;
	371	nClusters2pad++;
	372	break;
	373	case 3:
	374	iType = 1;
	375	nClusters3pad++;
	376	break;
	377	case 4:
	378	iType = 2;
	379	nClusters4pad++;
	380	break;
	381	case 5:
	382	iType = 3;
	383	nClusters5pad++;
	384	break;
	385	default:
	386	iType = 4;
	387	nClustersLarge++;
	388	break;
	389	};
	390
7ad19338	391	// Look for 5 pad cluster with minimum in the middle
db30bf0f	392	Bool_t fivePadCluster = kFALSE;
3e1a3ad8	393	if (col < nColMax-3) {
3e1a3ad8	394	if (digits->GetDataUnchecked(row,col+2,time) < 0) {
db30bf0f	395	fivePadCluster = kTRUE;
	396	}
	397	if ((fivePadCluster) && (col < nColMax-5)) {
	398	if (digits->GetDataUnchecked(row,col+4,time) >= sigThresh) {
	399	fivePadCluster = kFALSE;
	400	}
	401	}
	402	if ((fivePadCluster) && (col > 1)) {
	403	if (digits->GetDataUnchecked(row,col-2,time) >= sigThresh) {
	404	fivePadCluster = kFALSE;
	405	}
	406	}
	407	}
	408
	409	// 5 pad cluster
	410	// Modify the signal of the overlapping pad for the left part
	411	// of the cluster which remains from a previous unfolding
	412	if (iUnfold) {
	413	clusterSignal[0] *= ratioLeft;
7ad19338	414	iType = 5;
db30bf0f	415	iUnfold = 0;
	416	}
	417
	418	// Unfold the 5 pad cluster
	419	if (fivePadCluster) {
	420	for (iPad = 0; iPad < kNsig; iPad++) {
	421	padSignal[iPad] = TMath::Abs(digits->GetDataUnchecked(row
	422	,col-1+iPad
	423	,time));
f7336fa3	424	}
db30bf0f	425	// Unfold the two maxima and set the signal on
db30bf0f	426	// the overlapping pad to the ratio
17b26de4	427	ratioRight = Unfold(kEpsilon,iplan,padSignal);
db30bf0f	428	ratioLeft = 1.0 - ratioRight;
db30bf0f	429	clusterSignal[2] *= ratioRight;
7ad19338	430	iType = 5;
db30bf0f	431	iUnfold = 1;
f7336fa3	432	}
f7336fa3	433
7ad19338	434
a5cadd36	435	Double_t clusterCharge = clusterSignal[0]
	436	+ clusterSignal[1]
	437	+ clusterSignal[2];
3e1a3ad8	438
db30bf0f	439	// The position of the cluster
3e1a3ad8	440	clusterPads[0] = row + 0.5;
3e1a3ad8	441	// Take the shift of the additional time bins into account
dde59437	442	clusterPads[2] = time + 0.5;
3e1a3ad8	443
3551db50	444	// correct for t0
	445	clusterPads[2] -= calibration->GetT0(idet, col, row);
	446
	447	if (recParam->LUTOn()) {
db30bf0f	448	// Calculate the position of the cluster by using the
db30bf0f	449	// lookup table method
7ad19338	450	clusterPads[1] =
3551db50	451	recParam->LUTposition(iplan,clusterSignal[0]
17b26de4	452	,clusterSignal[1]
17b26de4	453	,clusterSignal[2]);
db30bf0f	454	}
db30bf0f	455	else {
db30bf0f	456	// Calculate the position of the cluster by using the
db30bf0f	457	// center of gravity method
7ad19338	458	for (Int_t i=0;i<5;i++) padSignal[i]=0;
	459	padSignal[2] = TMath::Abs(digits->GetDataUnchecked(row,col,time)); // central pad
	460	padSignal[1] = TMath::Abs(digits->GetDataUnchecked(row,col-1,time)); // left pad
	461	padSignal[3] = TMath::Abs(digits->GetDataUnchecked(row,col+1,time)); // right pad
	462	if (col>2 &&TMath::Abs(digits->GetDataUnchecked(row,col-2,time)<padSignal[1])){
	463	padSignal[0] = TMath::Abs(digits->GetDataUnchecked(row,col-2,time));
	464	}
	465	if (col<nColMax-3 &&TMath::Abs(digits->GetDataUnchecked(row,col+2,time)<padSignal[3])){
	466	padSignal[4] = TMath::Abs(digits->GetDataUnchecked(row,col+2,time));
	467	}
	468	clusterPads[1] = GetCOG(padSignal);
ed25c022	469	//Double_t check = fPar->LUTposition(iplan,clusterSignal[0]
	470	// ,clusterSignal[1]
	471	// ,clusterSignal[2]);
de4b10e5	472	// Float_t diff = clusterPads[1] - check;
db30bf0f	473
	474	}
	475
a5cadd36	476	Double_t q0 = clusterSignal[0];
	477	Double_t q1 = clusterSignal[1];
	478	Double_t q2 = clusterSignal[2];
	479	Double_t clusterSigmaY2 = (q1(q0+q2)+4q0*q2) /
	480	(clusterCharge*clusterCharge);
a819a5f7	481
3551db50	482	Float_t vdrift = calibration->GetVdrift(idet, col, row);
	483
	484	// Calculate the position and the error
7ad19338	485	Double_t colSize = padPlane->GetColSize(col);
7ad19338	486	Double_t rowSize = padPlane->GetRowSize(row);
a5cadd36	487	Double_t clusterPos[3];
7ad19338	488	clusterPos[0] = padPlane->GetColPos(col) + (clusterPads[1]-0.5)*colSize; // MI change
7ad19338	489	clusterPos[1] = padPlane->GetRowPos(row) -0.5*rowSize; //MI change
3551db50	490	clusterPos[2] = CalcXposFromTimebin(clusterPads[2], vdrift);
a5cadd36	491	Double_t clusterSig[2];
5443e65e	492	clusterSig[0] = (clusterSigmaY2 + 1./12.) * colSize*colSize;
7ad19338	493	clusterSig[1] = rowSize * rowSize / 12.;
3551db50	494
3551db50	495
f7336fa3	496	// Add the cluster to the output array
c85a4951	497	AliTRDcluster * cluster = AddCluster(clusterPos
3551db50	498	,(Int_t) clusterPads[2]
3551db50	499	,idet
a5cadd36	500	,clusterCharge
	501	,clusterTracks
	502	,clusterSig
de4b10e5	503	,iType,clusterPads[1]);
c85a4951	504	//
	505	//
	506	Short_t signals[7]={0,0,0,0,0,0,0};
	507	for (Int_t jPad = col-3;jPad<=col+3;jPad++){
	508	if (jPad<0 \|\|jPad>=nColMax-1) continue;
	509	signals[jPad-col+3] = TMath::Abs(digits->GetDataUnchecked(row,jPad,time));
	510	}
	511	cluster->SetSignals(signals);
f7336fa3	512	}
3e1a3ad8	513	}
	514	}
	515	}
f7336fa3	516
3e1a3ad8	517	// Compress the arrays
	518	digits->Compress(1,0);
	519	track0->Compress(1,0);
7ad19338	520	track1->Compress(1,0);
3e1a3ad8	521	track2->Compress(1,0);
f7336fa3	522
3e1a3ad8	523	// Write the cluster and reset the array
793ff80c	524	WriteClusters(idet);
bdbb05bb	525	ResetRecPoints();
3e1a3ad8	526	}
	527	}
	528	}
f7336fa3	529
47517f42	530	if (fVerbose > 0) {
17b26de4	531	printf("<AliTRDclusterizerV1::MakeCluster> ");
47517f42	532	printf("Done.\n");
47517f42	533	}
f7336fa3	534
	535	return kTRUE;
	536
	537	}
	538
7ad19338	539	Double_t AliTRDclusterizerV1::GetCOG(Double_t signal[5])
	540	{
	541	//
	542	// get COG position
	543	// used for clusters with more than 3 pads - where LUT not applicable
	544	Double_t sum = signal[0]+signal[1]+signal[2]+signal[3]+signal[4];
	545	Double_t res = (0.0*(-signal[0]+signal[4])+(-signal[1]+signal[3]))/sum;
	546	return res;
	547	}
	548
	549
	550
f7336fa3	551	//_____________________________________________________________________________
a5cadd36	552	Double_t AliTRDclusterizerV1::Unfold(Double_t eps, Int_t plane, Double_t* padSignal)
f7336fa3	553	{
	554	//
	555	// Method to unfold neighbouring maxima.
	556	// The charge ratio on the overlapping pad is calculated
	557	// until there is no more change within the range given by eps.
	558	// The resulting ratio is then returned to the calling method.
	559	//
	560
6a739e92	561	AliTRDcalibDB* calibration = AliTRDcalibDB::Instance();
6a739e92	562	if (!calibration)
3551db50	563	{
6a739e92	564	printf("<AliTRDclusterizerMI::Unfold> ");
	565	printf("ERROR getting instance of AliTRDcalibDB");
	566	return kFALSE;
3551db50	567	}
6a739e92	568
a5cadd36	569	Int_t irc = 0;
a5cadd36	570	Int_t itStep = 0; // Count iteration steps
f7336fa3	571
a5cadd36	572	Double_t ratio = 0.5; // Start value for ratio
a5cadd36	573	Double_t prevRatio = 0; // Store previous ratio
f7336fa3	574
a5cadd36	575	Double_t newLeftSignal[3] = {0}; // Array to store left cluster signal
	576	Double_t newRightSignal[3] = {0}; // Array to store right cluster signal
	577	Double_t newSignal[3] = {0};
f7336fa3	578
3e1a3ad8	579	// Start the iteration
f7336fa3	580	while ((TMath::Abs(prevRatio - ratio) > eps) && (itStep < 10)) {
	581
	582	itStep++;
	583	prevRatio = ratio;
	584
3e1a3ad8	585	// Cluster position according to charge ratio
a5cadd36	586	Double_t maxLeft = (ratio*padSignal[2] - padSignal[0])
	587	/ (padSignal[0] + padSignal[1] + ratio*padSignal[2]);
	588	Double_t maxRight = (padSignal[4] - (1-ratio)*padSignal[2])
	589	/ ((1-ratio)*padSignal[2] + padSignal[3] + padSignal[4]);
f7336fa3	590
3e1a3ad8	591	// Set cluster charge ratio
6a739e92	592	irc = calibration->PadResponse(1.0,maxLeft ,plane,newSignal);
a5cadd36	593	Double_t ampLeft = padSignal[1] / newSignal[1];
6a739e92	594	irc = calibration->PadResponse(1.0,maxRight,plane,newSignal);
a5cadd36	595	Double_t ampRight = padSignal[3] / newSignal[1];
f7336fa3	596
3e1a3ad8	597	// Apply pad response to parameters
6a739e92	598	irc = calibration->PadResponse(ampLeft ,maxLeft ,plane,newLeftSignal );
6a739e92	599	irc = calibration->PadResponse(ampRight,maxRight,plane,newRightSignal);
f7336fa3	600
3e1a3ad8	601	// Calculate new overlapping ratio
a5cadd36	602	ratio = TMath::Min((Double_t)1.0,newLeftSignal[2] /
db30bf0f	603	(newLeftSignal[2] + newRightSignal[0]));
f7336fa3	604
	605	}
	606
	607	return ratio;
	608
	609	}
	610