[u/mrichter/AliRoot.git] / FMD / AliFMDReconstructor.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$ */
/** @file    AliFMDReconstructor.cxx
    @author  Christian Holm Christensen <cholm@nbi.dk>
    @date    Mon Mar 27 12:47:09 2006
    @brief   FMD reconstruction 
*/
//____________________________________________________________________
//
// This is a class that constructs AliFMDRecPoint objects from of Digits
//
// This class reads either digits from a TClonesArray or raw data from
// a DDL file (or similar), and stores the read ADC counts in an
// internal cache (fAdcs). 
//
//-- Authors: Evgeny Karpechev(INR) and Alla Maevsksia
//  Latest changes by Christian Holm Christensen <cholm@nbi.dk>
//
//
//____________________________________________________________________

#include <AliLog.h>                        // ALILOG_H
#include <AliRun.h>                        // ALIRUN_H
#include <AliRunLoader.h>                  // ALIRUNLOADER_H
#include <AliHeader.h>                     // ALIHEADER_H
#include <AliGenEventHeader.h>             // ALIGENEVENTHEADER_H
#include "AliFMDGeometry.h"                // ALIFMDGEOMETRY_H
#include "AliFMDParameters.h"              // ALIFMDPARAMETERS_H
#include "AliFMDDigit.h"                   // ALIFMDDIGIT_H
#include "AliFMDReconstructor.h"           // ALIFMDRECONSTRUCTOR_H
#include "AliFMDRawReader.h"               // ALIFMDRAWREADER_H
#include "AliFMDRecPoint.h"	   	   // ALIFMDMULTNAIIVE_H
#include "AliESD.h"			   // ALIESD_H
#include <AliESDFMD.h>			   // ALIESDFMD_H
class AliRawReader;

//____________________________________________________________________
ClassImp(AliFMDReconstructor)
#if 0
  ; // This is here to keep Emacs for indenting the next line
#endif

//____________________________________________________________________
AliFMDReconstructor::AliFMDReconstructor() 
  : AliReconstructor(),
    fMult(0x0), 
    fNMult(0),
    fTreeR(0x0),
    fCurrentVertex(0),
    fESDObj(0x0),
    fESD(0x0)
{
  // Make a new FMD reconstructor object - default CTOR.  
}
  

//____________________________________________________________________
AliFMDReconstructor::AliFMDReconstructor(const AliFMDReconstructor& other) 
  : AliReconstructor(), 
    fMult(other.fMult),
    fNMult(other.fNMult),
    fTreeR(other.fTreeR),
    fCurrentVertex(other.fCurrentVertex),
    fESDObj(other.fESDObj),
    fESD(other.fESD)
{
  // Copy constructor 
}
  

//____________________________________________________________________
AliFMDReconstructor&
AliFMDReconstructor::operator=(const AliFMDReconstructor& other) 
{
  // Assignment operator
  fMult   = other.fMult;
  fNMult = other.fNMult;
  fTreeR = other.fTreeR;
  fCurrentVertex = other.fCurrentVertex;
  fESDObj = other.fESDObj;
  fESD = other.fESD;
  return *this;
}

//____________________________________________________________________
AliFMDReconstructor::~AliFMDReconstructor() 
{
  // Destructor 
  if (fMult)   fMult->Delete();
  if (fMult)   delete fMult;
  if (fESDObj) delete fESDObj;
}

//____________________________________________________________________
void 
AliFMDReconstructor::Init(AliRunLoader* runLoader) 
{
  // Initialize the reconstructor 
  AliDebug(1, Form("Init called with runloader 0x%x", runLoader));
  // Initialize the geometry 
  AliFMDGeometry* geom = AliFMDGeometry::Instance();
  geom->Init();
  geom->InitTransformations();

  // Initialize the parameters
  AliFMDParameters* param = AliFMDParameters::Instance();
  param->Init();
  
  // Current vertex position
  fCurrentVertex = 0;
  // Create array of reconstructed strip multiplicities 
  fMult = new TClonesArray("AliFMDRecPoint", 51200);
  // Create ESD output object 
  fESDObj = new AliESDFMD;
  
  // Check that we have a run loader
  if (!runLoader) { 
    Warning("Init", "No run loader");
    return;
  }

  // Check if we can get the header tree 
  AliHeader* header = runLoader->GetHeader();
  if (!header) {
    Warning("Init", "No header");
    return;
  }

  // Check if we can get a simulation header 
  AliGenEventHeader* eventHeader = header->GenEventHeader();
  if (eventHeader) {
    TArrayF vtx;
    eventHeader->PrimaryVertex(vtx);
    fCurrentVertex = vtx[2];
    AliDebug(1, Form("Primary vertex Z coordinate for event # %d/%d is %f", 
		     header->GetRun(), header->GetEvent(), fCurrentVertex));
    Warning("Init", "no generator event header");
  }
  else {
    Warning("Init", "No generator event header - "
	    "perhaps we get the vertex from ESD?");
  }
}

//____________________________________________________________________
void 
AliFMDReconstructor::ConvertDigits(AliRawReader* reader, 
				   TTree* digitsTree) const
{
  // Convert Raw digits to AliFMDDigit's in a tree 
  AliDebug(1, "Reading raw data into digits tree");
  AliFMDRawReader rawRead(reader, digitsTree);
  // rawRead.SetSampleRate(fFMD->GetSampleRate());
  rawRead.Exec();
}

//____________________________________________________________________
void 
AliFMDReconstructor::Reconstruct(TTree* digitsTree, 
				 TTree* clusterTree) const 
{
  // Reconstruct event from digits in tree 
  // Get the FMD branch holding the digits. 
  // FIXME: The vertex may not be known yet, so we may have to move
  // some of this to FillESD. 
  AliDebug(1, "Reconstructing from digits in a tree");
#if 1
  if (fESD) {
    const AliESDVertex* vertex = fESD->GetVertex();
    if (vertex) {
      AliDebug(1, Form("Got vertex from ESD: %f", vertex->GetZv()));
      fCurrentVertex = vertex->GetZv();
    }
  }
#endif  
  TBranch *digitBranch = digitsTree->GetBranch("FMD");
  if (!digitBranch) {
    Error("Exec", "No digit branch for the FMD found");
    return;
  }
  TClonesArray* digits = new TClonesArray("AliFMDDigit");
  digitBranch->SetAddress(&digits);

  if (fMult)   fMult->Clear();
  if (fESDObj) fESDObj->Clear();
  
  fNMult = 0;
  fTreeR = clusterTree;
  fTreeR->Branch("FMD", &fMult);
  
  AliDebug(5, "Getting entry 0 from digit branch");
  digitBranch->GetEntry(0);
  
  AliDebug(5, "Processing digits");
  ProcessDigits(digits);

  Int_t written = clusterTree->Fill();
  AliDebug(10, Form("Filled %d bytes into cluster tree", written));
  digits->Delete();
  delete digits;
}
 

//____________________________________________________________________
void
AliFMDReconstructor::ProcessDigits(TClonesArray* digits) const
{
  // For each digit, find the pseudo rapdity, azimuthal angle, and
  // number of corrected ADC counts, and pass it on to the algorithms
  // used. 
  Int_t nDigits = digits->GetEntries();
  AliDebug(1, Form("Got %d digits", nDigits));
  for (Int_t i = 0; i < nDigits; i++) {
    AliFMDDigit* digit = static_cast<AliFMDDigit*>(digits->At(i));
    AliFMDParameters* param  = AliFMDParameters::Instance();
    // Check that the strip is not marked as dead 
    if (param->IsDead(digit->Detector(), digit->Ring(), 
		      digit->Sector(), digit->Strip())) continue;

    // digit->Print();
    // Get eta and phi 
    Float_t eta, phi;
    PhysicalCoordinates(digit, eta, phi);
    
    // Substract pedestal. 
    UShort_t counts   = SubtractPedestal(digit);
    
    // Gain match digits. 
    Double_t edep     = Adc2Energy(digit, eta, counts);
    
    // Make rough multiplicity 
    Double_t mult     = Energy2Multiplicity(digit, edep);
    
    AliDebug(10, Form("FMD%d%c[%2d,%3d]: "
		      "ADC: %d, Counts: %d, Energy: %f, Mult: %f", 
		      digit->Detector(), digit->Ring(), digit->Sector(),
		      digit->Strip(), digit->Counts(), counts, edep, mult));
    
    // Create a `RecPoint' on the output branch. 
    AliFMDRecPoint* m = 
      new ((*fMult)[fNMult]) AliFMDRecPoint(digit->Detector(), 
					    digit->Ring(), 
					    digit->Sector(),
					    digit->Strip(),
					    eta, phi, 
					    edep, mult);
    (void)m; // Suppress warnings about unused variables. 
    fNMult++;

    fESDObj->SetMultiplicity(digit->Detector(), digit->Ring(), 
			     digit->Sector(),  digit->Strip(), mult);
    fESDObj->SetEta(digit->Detector(), digit->Ring(), 
		    digit->Sector(),  digit->Strip(), eta);
  }
}

//____________________________________________________________________
UShort_t
AliFMDReconstructor::SubtractPedestal(AliFMDDigit* digit) const
{
  // Member function to subtract the pedestal from a digit
  // This implementation does nothing, but a derived class could over
  // load this to subtract a pedestal that was given in a database or
  // something like that. 

  Int_t             counts = 0;
  AliFMDParameters* param  = AliFMDParameters::Instance();
  Float_t           pedM   = param->GetPedestal(digit->Detector(), 
						digit->Ring(), 
						digit->Sector(), 
						digit->Strip());
  AliDebug(10, Form("Subtracting pedestal %f from signal %d", 
		   pedM, digit->Counts()));
  if (digit->Count3() > 0)      counts = digit->Count3();
  else if (digit->Count2() > 0) counts = digit->Count2();
  else                          counts = digit->Count1();
  counts = TMath::Max(Int_t(counts - pedM), 0);
  if (counts > 0) AliDebug(10, "Got a hit strip");
  
  return  UShort_t(counts);
}

//____________________________________________________________________
Float_t
AliFMDReconstructor::Adc2Energy(AliFMDDigit* digit, 
				Float_t      /* eta */, 
				UShort_t     count) const
{
  // Converts number of ADC counts to energy deposited. 
  // Note, that this member function can be overloaded by derived
  // classes to do strip-specific look-ups in databases or the like,
  // to find the proper gain for a strip. 
  // 
  // In this simple version, we calculate the energy deposited as 
  // 
  //    EnergyDeposited = cos(theta) * gain * count
  // 
  // where 
  // 
  //           Pre_amp_MIP_Range
  //    gain = ----------------- * Energy_deposited_per_MIP
  //           ADC_channel_size    
  // 
  // is constant and the same for all strips.

  // Double_t theta = 2 * TMath::ATan(TMath::Exp(-eta));
  // Double_t edep  = TMath::Abs(TMath::Cos(theta)) * fGain * count;
  AliFMDParameters* param = AliFMDParameters::Instance();
  Float_t           gain  = param->GetPulseGain(digit->Detector(), 
						digit->Ring(), 
						digit->Sector(), 
						digit->Strip());
  Double_t          edep  = count * gain;
  AliDebug(10, Form("Converting counts %d to energy via factor %f", 
		    count, gain));
  return edep;
}

//____________________________________________________________________
Float_t
AliFMDReconstructor::Energy2Multiplicity(AliFMDDigit* /* digit */, 
					 Float_t      edep) const
{
  // Converts an energy signal to number of particles. 
  // Note, that this member function can be overloaded by derived
  // classes to do strip-specific look-ups in databases or the like,
  // to find the proper gain for a strip. 
  // 
  // In this simple version, we calculate the multiplicity as 
  // 
  //   multiplicity = Energy_deposited / Energy_deposited_per_MIP
  // 
  // where 
  //
  //   Energy_deposited_per_MIP = 1.664 * SI_density * SI_thickness 
  // 
  // is constant and the same for all strips 
  AliFMDParameters* param   = AliFMDParameters::Instance();
  Double_t          edepMIP = param->GetEdepMip();
  Float_t           mult    = edep / edepMIP;
  if (edep > 0) 
    AliDebug(10, Form("Translating energy %f to multiplicity via "
		     "divider %f->%f", edep, edepMIP, mult));
  return mult;
}

//____________________________________________________________________
void
AliFMDReconstructor::PhysicalCoordinates(AliFMDDigit* digit, 
					 Float_t& eta, 
					 Float_t& phi) const
{
  // Get the eta and phi of a digit 
  // 
  // Get geometry. 
  AliFMDGeometry* geom = AliFMDGeometry::Instance();
  Double_t x, y, z, r, theta;
  geom->Detector2XYZ(digit->Detector(), digit->Ring(), digit->Sector(), 
		    digit->Strip(), x, y, z);
  // Correct for vertex offset. 
  z     += fCurrentVertex;
  phi   =  TMath::ATan2(y, x);
  r     =  TMath::Sqrt(y * y + x * x);
  theta =  TMath::ATan2(r, z);
  eta   = -TMath::Log(TMath::Tan(theta / 2));
}

      
//____________________________________________________________________
void 
AliFMDReconstructor::FillESD(TTree*  /* digitsTree */, 
			     TTree*  /* clusterTree */,
			     AliESD* esd) const
{
  // nothing to be done
  // FIXME: The vertex may not be known when Reconstruct is executed,
  // so we may have to move some of that member function here. 
  AliDebug(1, Form("Calling FillESD with two trees and one ESD"));
  // fESDObj->Print();

  if (esd) { 
    AliDebug(1, Form("Writing FMD data to ESD tree"));
    esd->SetFMDData(fESDObj);
  }
}


//____________________________________________________________________
void 
AliFMDReconstructor::Reconstruct(AliRawReader*,TTree*) const 
{
  // Cannot be used.  See member function with same name but with 2
  // TTree arguments.   Make sure you do local reconstrucion 
  AliDebug(1, Form("Calling FillESD with loader and tree"));
  AliError("MayNotUse");
}
//____________________________________________________________________
void 
AliFMDReconstructor::Reconstruct(AliRunLoader*) const 
{
  // Cannot be used.  See member function with same name but with 2
  // TTree arguments.   Make sure you do local reconstrucion 
  AliDebug(1, Form("Calling FillESD with loader"));
  AliError("MayNotUse");
}
//____________________________________________________________________
void 
AliFMDReconstructor::Reconstruct(AliRunLoader*, AliRawReader*) const 
{
  // Cannot be used.  See member function with same name but with 2
  // TTree arguments.   Make sure you do local reconstrucion 
  AliDebug(1, Form("Calling FillESD with loader and raw reader"));
  AliError("MayNotUse");
}
//____________________________________________________________________
void 
AliFMDReconstructor::FillESD(AliRawReader*,TTree*,AliESD*) const 
{
  // Cannot be used.  See member function with same name but with 2
  // TTree arguments.   Make sure you do local reconstrucion 
  AliDebug(1, Form("Calling FillESD with raw reader, tree, and ESD"));
  AliError("MayNotUse");
}
//____________________________________________________________________
void 
AliFMDReconstructor::FillESD(AliRunLoader*,AliESD*) const
{
  // Cannot be used.  See member function with same name but with 2
  // TTree arguments.   Make sure you do local reconstrucion 
  AliDebug(1, Form("Calling FillESD with loader and ESD"));
  AliError("MayNotUse");
}
//____________________________________________________________________
void 
AliFMDReconstructor::FillESD(AliRunLoader*,AliRawReader*,AliESD*) const 
{
  // Cannot be used.  See member function with same name but with 2
  // TTree arguments.   Make sure you do local reconstrucion 
  AliDebug(1, Form("Calling FillESD with loader, raw reader, and ESD"));
  AliError("MayNotUse");
}

//____________________________________________________________________
//
// EOF
//
Commit	Line	Data
37c55dc0	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	**************************************************************************/
0d0e6995	15	/* $Id$ */
c2fc1258	16	/** @file AliFMDReconstructor.cxx
	17	@author Christian Holm Christensen <cholm@nbi.dk>
	18	@date Mon Mar 27 12:47:09 2006
	19	@brief FMD reconstruction
	20	*/
4347b38f	21	//____________________________________________________________________
42403906	22	//
02a27b50	23	// This is a class that constructs AliFMDRecPoint objects from of Digits
4347b38f	24	//
	25	// This class reads either digits from a TClonesArray or raw data from
	26	// a DDL file (or similar), and stores the read ADC counts in an
	27	// internal cache (fAdcs).
	28	//
37c55dc0	29	//-- Authors: Evgeny Karpechev(INR) and Alla Maevsksia
4347b38f	30	// Latest changes by Christian Holm Christensen <cholm@nbi.dk>
	31	//
	32	//
	33	//____________________________________________________________________
37c55dc0	34
56b1929b	35	#include <AliLog.h> // ALILOG_H
	36	#include <AliRun.h> // ALIRUN_H
	37	#include <AliRunLoader.h> // ALIRUNLOADER_H
56b1929b	38	#include <AliHeader.h> // ALIHEADER_H
56b1929b	39	#include <AliGenEventHeader.h> // ALIGENEVENTHEADER_H
1a1fdef7	40	#include "AliFMDGeometry.h" // ALIFMDGEOMETRY_H
1a1fdef7	41	#include "AliFMDParameters.h" // ALIFMDPARAMETERS_H
56b1929b	42	#include "AliFMDDigit.h" // ALIFMDDIGIT_H
56b1929b	43	#include "AliFMDReconstructor.h" // ALIFMDRECONSTRUCTOR_H
56b1929b	44	#include "AliFMDRawReader.h" // ALIFMDRAWREADER_H
bf000c32	45	#include "AliFMDRecPoint.h" // ALIFMDMULTNAIIVE_H
a3537838	46	#include "AliESD.h" // ALIESD_H
8f6ee336	47	#include <AliESDFMD.h> // ALIESDFMD_H
02a27b50	48	class AliRawReader;
4347b38f	49
4347b38f	50	//____________________________________________________________________
925e6570	51	ClassImp(AliFMDReconstructor)
1a1fdef7	52	#if 0
	53	; // This is here to keep Emacs for indenting the next line
	54	#endif
37c55dc0	55
4347b38f	56	//____________________________________________________________________
	57	AliFMDReconstructor::AliFMDReconstructor()
	58	: AliReconstructor(),
e9fd1e20	59	fMult(0x0),
	60	fNMult(0),
	61	fTreeR(0x0),
	62	fCurrentVertex(0),
	63	fESDObj(0x0),
	64	fESD(0x0)
4347b38f	65	{
8f6ee336	66	// Make a new FMD reconstructor object - default CTOR.
4347b38f	67	}
4347b38f	68
42403906	69
	70	//____________________________________________________________________
	71	AliFMDReconstructor::AliFMDReconstructor(const AliFMDReconstructor& other)
8f6ee336	72	: AliReconstructor(),
8f6ee336	73	fMult(other.fMult),
e9fd1e20	74	fNMult(other.fNMult),
	75	fTreeR(other.fTreeR),
	76	fCurrentVertex(other.fCurrentVertex),
	77	fESDObj(other.fESDObj),
	78	fESD(other.fESD)
42403906	79	{
56b1929b	80	// Copy constructor
42403906	81	}
	82
	83
	84	//____________________________________________________________________
	85	AliFMDReconstructor&
	86	AliFMDReconstructor::operator=(const AliFMDReconstructor& other)
	87	{
56b1929b	88	// Assignment operator
8f6ee336	89	fMult = other.fMult;
e9fd1e20	90	fNMult = other.fNMult;
	91	fTreeR = other.fTreeR;
	92	fCurrentVertex = other.fCurrentVertex;
8f6ee336	93	fESDObj = other.fESDObj;
e9fd1e20	94	fESD = other.fESD;
42403906	95	return *this;
42403906	96	}
56b1929b	97
	98	//____________________________________________________________________
	99	AliFMDReconstructor::~AliFMDReconstructor()
	100	{
	101	// Destructor
8f6ee336	102	if (fMult) fMult->Delete();
	103	if (fMult) delete fMult;
	104	if (fESDObj) delete fESDObj;
56b1929b	105	}
4347b38f	106
	107	//____________________________________________________________________
	108	void
1a1fdef7	109	AliFMDReconstructor::Init(AliRunLoader* runLoader)
	110	{
	111	// Initialize the reconstructor
	112	AliDebug(1, Form("Init called with runloader 0x%x", runLoader));
bf000c32	113	// Initialize the geometry
9b48326f	114	AliFMDGeometry* geom = AliFMDGeometry::Instance();
	115	geom->Init();
	116	geom->InitTransformations();
	117
	118	// Initialize the parameters
	119	AliFMDParameters* param = AliFMDParameters::Instance();
	120	param->Init();
bf000c32	121
8f6ee336	122	// Current vertex position
1a1fdef7	123	fCurrentVertex = 0;
8f6ee336	124	// Create array of reconstructed strip multiplicities
bf000c32	125	fMult = new TClonesArray("AliFMDRecPoint", 51200);
8f6ee336	126	// Create ESD output object
	127	fESDObj = new AliESDFMD;
	128
	129	// Check that we have a run loader
1a1fdef7	130	if (!runLoader) {
1a1fdef7	131	Warning("Init", "No run loader");
4347b38f	132	return;
4347b38f	133	}
8f6ee336	134
8f6ee336	135	// Check if we can get the header tree
1a1fdef7	136	AliHeader* header = runLoader->GetHeader();
	137	if (!header) {
	138	Warning("Init", "No header");
4347b38f	139	return;
4347b38f	140	}
8f6ee336	141
8f6ee336	142	// Check if we can get a simulation header
1a1fdef7	143	AliGenEventHeader* eventHeader = header->GenEventHeader();
a3537838	144	if (eventHeader) {
	145	TArrayF vtx;
	146	eventHeader->PrimaryVertex(vtx);
	147	fCurrentVertex = vtx[2];
	148	AliDebug(1, Form("Primary vertex Z coordinate for event # %d/%d is %f",
	149	header->GetRun(), header->GetEvent(), fCurrentVertex));
	150	Warning("Init", "no generator event header");
	151	}
	152	else {
	153	Warning("Init", "No generator event header - "
	154	"perhaps we get the vertex from ESD?");
4347b38f	155	}
4347b38f	156	}
dc8af42e	157
4347b38f	158	//____________________________________________________________________
4347b38f	159	void
1a1fdef7	160	AliFMDReconstructor::ConvertDigits(AliRawReader* reader,
	161	TTree* digitsTree) const
	162	{
	163	// Convert Raw digits to AliFMDDigit's in a tree
8f6ee336	164	AliDebug(1, "Reading raw data into digits tree");
1a1fdef7	165	AliFMDRawReader rawRead(reader, digitsTree);
	166	// rawRead.SetSampleRate(fFMD->GetSampleRate());
	167	rawRead.Exec();
4347b38f	168	}
4347b38f	169
4347b38f	170	//____________________________________________________________________
4347b38f	171	void
1a1fdef7	172	AliFMDReconstructor::Reconstruct(TTree* digitsTree,
1a1fdef7	173	TTree* clusterTree) const
4347b38f	174	{
1a1fdef7	175	// Reconstruct event from digits in tree
4347b38f	176	// Get the FMD branch holding the digits.
69b696b9	177	// FIXME: The vertex may not be known yet, so we may have to move
69b696b9	178	// some of this to FillESD.
1a1fdef7	179	AliDebug(1, "Reconstructing from digits in a tree");
1e8f773e	180	#if 1
a3537838	181	if (fESD) {
8f6ee336	182	const AliESDVertex* vertex = fESD->GetVertex();
	183	if (vertex) {
	184	AliDebug(1, Form("Got vertex from ESD: %f", vertex->GetZv()));
	185	fCurrentVertex = vertex->GetZv();
	186	}
a3537838	187	}
54e415a8	188	#endif
1a1fdef7	189	TBranch *digitBranch = digitsTree->GetBranch("FMD");
4347b38f	190	if (!digitBranch) {
1a1fdef7	191	Error("Exec", "No digit branch for the FMD found");
1a1fdef7	192	return;
4347b38f	193	}
0abe7182	194	TClonesArray* digits = new TClonesArray("AliFMDDigit");
1a1fdef7	195	digitBranch->SetAddress(&digits);
4347b38f	196
8f6ee336	197	if (fMult) fMult->Clear();
	198	if (fESDObj) fESDObj->Clear();
	199
	200	fNMult = 0;
	201	fTreeR = clusterTree;
	202	fTreeR->Branch("FMD", &fMult);
	203
	204	AliDebug(5, "Getting entry 0 from digit branch");
1a1fdef7	205	digitBranch->GetEntry(0);
1a1fdef7	206
8f6ee336	207	AliDebug(5, "Processing digits");
e802be3e	208	ProcessDigits(digits);
e802be3e	209
8f6ee336	210	Int_t written = clusterTree->Fill();
8f6ee336	211	AliDebug(10, Form("Filled %d bytes into cluster tree", written));
0abe7182	212	digits->Delete();
0abe7182	213	delete digits;
4347b38f	214	}
1a1fdef7	215
8f6ee336	216
4347b38f	217	//____________________________________________________________________
e802be3e	218	void
e802be3e	219	AliFMDReconstructor::ProcessDigits(TClonesArray* digits) const
4347b38f	220	{
69b696b9	221	// For each digit, find the pseudo rapdity, azimuthal angle, and
	222	// number of corrected ADC counts, and pass it on to the algorithms
	223	// used.
e802be3e	224	Int_t nDigits = digits->GetEntries();
1a1fdef7	225	AliDebug(1, Form("Got %d digits", nDigits));
e802be3e	226	for (Int_t i = 0; i < nDigits; i++) {
e802be3e	227	AliFMDDigit* digit = static_cast<AliFMDDigit*>(digits->At(i));
8f6ee336	228	AliFMDParameters* param = AliFMDParameters::Instance();
	229	// Check that the strip is not marked as dead
	230	if (param->IsDead(digit->Detector(), digit->Ring(),
	231	digit->Sector(), digit->Strip())) continue;
	232
	233	// digit->Print();
	234	// Get eta and phi
	235	Float_t eta, phi;
	236	PhysicalCoordinates(digit, eta, phi);
4347b38f	237
8f6ee336	238	// Substract pedestal.
e802be3e	239	UShort_t counts = SubtractPedestal(digit);
4347b38f	240
8f6ee336	241	// Gain match digits.
	242	Double_t edep = Adc2Energy(digit, eta, counts);
	243
	244	// Make rough multiplicity
	245	Double_t mult = Energy2Multiplicity(digit, edep);
	246
	247	AliDebug(10, Form("FMD%d%c[%2d,%3d]: "
	248	"ADC: %d, Counts: %d, Energy: %f, Mult: %f",
	249	digit->Detector(), digit->Ring(), digit->Sector(),
	250	digit->Strip(), digit->Counts(), counts, edep, mult));
	251
	252	// Create a `RecPoint' on the output branch.
bf000c32	253	AliFMDRecPoint* m =
	254	new ((*fMult)[fNMult]) AliFMDRecPoint(digit->Detector(),
	255	digit->Ring(),
	256	digit->Sector(),
	257	digit->Strip(),
	258	eta, phi,
	259	edep, mult);
8f6ee336	260	(void)m; // Suppress warnings about unused variables.
	261	fNMult++;
	262
	263	fESDObj->SetMultiplicity(digit->Detector(), digit->Ring(),
	264	digit->Sector(), digit->Strip(), mult);
	265	fESDObj->SetEta(digit->Detector(), digit->Ring(),
	266	digit->Sector(), digit->Strip(), eta);
4347b38f	267	}
4347b38f	268	}
8f6ee336	269
1a1fdef7	270	//____________________________________________________________________
	271	UShort_t
	272	AliFMDReconstructor::SubtractPedestal(AliFMDDigit* digit) const
	273	{
	274	// Member function to subtract the pedestal from a digit
	275	// This implementation does nothing, but a derived class could over
	276	// load this to subtract a pedestal that was given in a database or
	277	// something like that.
	278
8f6ee336	279	Int_t counts = 0;
	280	AliFMDParameters* param = AliFMDParameters::Instance();
	281	Float_t pedM = param->GetPedestal(digit->Detector(),
	282	digit->Ring(),
	283	digit->Sector(),
	284	digit->Strip());
	285	AliDebug(10, Form("Subtracting pedestal %f from signal %d",
	286	pedM, digit->Counts()));
1a1fdef7	287	if (digit->Count3() > 0) counts = digit->Count3();
	288	else if (digit->Count2() > 0) counts = digit->Count2();
	289	else counts = digit->Count1();
8f6ee336	290	counts = TMath::Max(Int_t(counts - pedM), 0);
	291	if (counts > 0) AliDebug(10, "Got a hit strip");
	292
1a1fdef7	293	return UShort_t(counts);
	294	}
	295
8f6ee336	296	//____________________________________________________________________
	297	Float_t
	298	AliFMDReconstructor::Adc2Energy(AliFMDDigit* digit,
	299	Float_t /* eta */,
	300	UShort_t count) const
	301	{
	302	// Converts number of ADC counts to energy deposited.
	303	// Note, that this member function can be overloaded by derived
	304	// classes to do strip-specific look-ups in databases or the like,
	305	// to find the proper gain for a strip.
	306	//
	307	// In this simple version, we calculate the energy deposited as
	308	//
	309	// EnergyDeposited = cos(theta) * gain * count
	310	//
	311	// where
	312	//
	313	// Pre_amp_MIP_Range
	314	// gain = ----------------- * Energy_deposited_per_MIP
	315	// ADC_channel_size
	316	//
	317	// is constant and the same for all strips.
	318
	319	// Double_t theta = 2 * TMath::ATan(TMath::Exp(-eta));
	320	// Double_t edep = TMath::Abs(TMath::Cos(theta)) * fGain * count;
	321	AliFMDParameters* param = AliFMDParameters::Instance();
	322	Float_t gain = param->GetPulseGain(digit->Detector(),
	323	digit->Ring(),
	324	digit->Sector(),
	325	digit->Strip());
	326	Double_t edep = count * gain;
	327	AliDebug(10, Form("Converting counts %d to energy via factor %f",
	328	count, gain));
	329	return edep;
	330	}
	331
	332	//____________________________________________________________________
	333	Float_t
	334	AliFMDReconstructor::Energy2Multiplicity(AliFMDDigit* /* digit */,
	335	Float_t edep) const
	336	{
	337	// Converts an energy signal to number of particles.
	338	// Note, that this member function can be overloaded by derived
	339	// classes to do strip-specific look-ups in databases or the like,
	340	// to find the proper gain for a strip.
	341	//
	342	// In this simple version, we calculate the multiplicity as
	343	//
	344	// multiplicity = Energy_deposited / Energy_deposited_per_MIP
	345	//
	346	// where
	347	//
	348	// Energy_deposited_per_MIP = 1.664 * SI_density * SI_thickness
	349	//
	350	// is constant and the same for all strips
	351	AliFMDParameters* param = AliFMDParameters::Instance();
	352	Double_t edepMIP = param->GetEdepMip();
	353	Float_t mult = edep / edepMIP;
	354	if (edep > 0)
	355	AliDebug(10, Form("Translating energy %f to multiplicity via "
	356	"divider %f->%f", edep, edepMIP, mult));
	357	return mult;
	358	}
	359
360	//____________________________________________________________________
361	void
362	AliFMDReconstructor::PhysicalCoordinates(AliFMDDigit* digit,
363	Float_t& eta,
364	Float_t& phi) const
365	{
366	// Get the eta and phi of a digit
367	//
368	// Get geometry.
9b48326f	369	AliFMDGeometry* geom = AliFMDGeometry::Instance();
bf000c32	370	Double_t x, y, z, r, theta;
9b48326f	371	geom->Detector2XYZ(digit->Detector(), digit->Ring(), digit->Sector(),
bf000c32	372	digit->Strip(), x, y, z);
	373	// Correct for vertex offset.
	374	z += fCurrentVertex;
	375	phi = TMath::ATan2(y, x);
	376	r = TMath::Sqrt(y * y + x * x);
	377	theta = TMath::ATan2(r, z);
	378	eta = -TMath::Log(TMath::Tan(theta / 2));
8f6ee336	379	}
	380
	381
	382
4347b38f	383	//____________________________________________________________________
4347b38f	384	void
1a1fdef7	385	AliFMDReconstructor::FillESD(TTree* /* digitsTree */,
1a1fdef7	386	TTree* /* clusterTree */,
8f6ee336	387	AliESD* esd) const
121a60bd	388	{
42403906	389	// nothing to be done
69b696b9	390	// FIXME: The vertex may not be known when Reconstruct is executed,
69b696b9	391	// so we may have to move some of that member function here.
8f6ee336	392	AliDebug(1, Form("Calling FillESD with two trees and one ESD"));
	393	// fESDObj->Print();
	394
	395	if (esd) {
	396	AliDebug(1, Form("Writing FMD data to ESD tree"));
	397	esd->SetFMDData(fESDObj);
a3537838	398	}
121a60bd	399	}
121a60bd	400
54e415a8	401
	402	//____________________________________________________________________
	403	void
	404	AliFMDReconstructor::Reconstruct(AliRawReader,TTree) const
	405	{
	406	// Cannot be used. See member function with same name but with 2
	407	// TTree arguments. Make sure you do local reconstrucion
8f6ee336	408	AliDebug(1, Form("Calling FillESD with loader and tree"));
54e415a8	409	AliError("MayNotUse");
	410	}
	411	//____________________________________________________________________
	412	void
	413	AliFMDReconstructor::Reconstruct(AliRunLoader*) const
	414	{
	415	// Cannot be used. See member function with same name but with 2
	416	// TTree arguments. Make sure you do local reconstrucion
8f6ee336	417	AliDebug(1, Form("Calling FillESD with loader"));
54e415a8	418	AliError("MayNotUse");
	419	}
	420	//____________________________________________________________________
	421	void
	422	AliFMDReconstructor::Reconstruct(AliRunLoader, AliRawReader) const
	423	{
	424	// Cannot be used. See member function with same name but with 2
	425	// TTree arguments. Make sure you do local reconstrucion
8f6ee336	426	AliDebug(1, Form("Calling FillESD with loader and raw reader"));
54e415a8	427	AliError("MayNotUse");
	428	}
	429	//____________________________________________________________________
	430	void
	431	AliFMDReconstructor::FillESD(AliRawReader,TTree,AliESD*) const
	432	{
	433	// Cannot be used. See member function with same name but with 2
	434	// TTree arguments. Make sure you do local reconstrucion
8f6ee336	435	AliDebug(1, Form("Calling FillESD with raw reader, tree, and ESD"));
54e415a8	436	AliError("MayNotUse");
	437	}
	438	//____________________________________________________________________
	439	void
	440	AliFMDReconstructor::FillESD(AliRunLoader,AliESD) const
	441	{
	442	// Cannot be used. See member function with same name but with 2
	443	// TTree arguments. Make sure you do local reconstrucion
8f6ee336	444	AliDebug(1, Form("Calling FillESD with loader and ESD"));
54e415a8	445	AliError("MayNotUse");
	446	}
	447	//____________________________________________________________________
	448	void
	449	AliFMDReconstructor::FillESD(AliRunLoader,AliRawReader,AliESD*) const
	450	{
	451	// Cannot be used. See member function with same name but with 2
	452	// TTree arguments. Make sure you do local reconstrucion
8f6ee336	453	AliDebug(1, Form("Calling FillESD with loader, raw reader, and ESD"));
54e415a8	454	AliError("MayNotUse");
	455	}
	456
42403906	457	//____________________________________________________________________
	458	//
	459	// EOF
	460	//