[u/mrichter/AliRoot.git] / FMD / AliFMDDigitizer.cxx

//************************************************************************
// Copyright(c) 2004, 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    AliFMDDigitizer.cxx
 * 
 * @author  Christian Holm Christensen <cholm@nbi.dk>
 * @date    Mon Mar 27 12:38:26 2006
 * @brief   FMD Digitizers implementation
 *
 * @ingroup FMD_sim
 */
//////////////////////////////////////////////////////////////////////////////
//
//  This class contains the procedures simulation ADC  signal for the
//  Forward Multiplicity detector  : SDigits->Digits
// 
//  Digits consists of
//   - Detector #
//   - Ring ID                                             
//   - Sector #     
//   - Strip #
//   - ADC count in this channel                                  
//
//  Digits consists of
//   - Detector #
//   - Ring ID                                             
//   - Sector #     
//   - Strip #
//   - Total energy deposited in the strip
//   - ADC count in this channel                                  
//
// As the Digits and SDigits have so much in common, the classes
// AliFMDDigitizer and AliFMDSDigitizer are implemented via a base
// class AliFMDBaseDigitizer.
//
//                 +---------------------+
//                 | AliFMDBaseDigitizer |
//                 +---------------------+
//                           ^
//                           |
//                +----------+---------+
//                |                    |
//      +-----------------+     +------------------+
//      | AliFMDDigitizer |	| AliFMDSDigitizer |
//      +-----------------+	+------------------+
//                |
//     +-------------------+
//     | AliFMDSSDigitizer |
//     +-------------------+
//
// These classes has several paramters: 
//
//     fPedestal
//     fPedestalWidth
//         (Only AliFMDDigitizer)
//         Mean and width of the pedestal.  The pedestal is simulated
//         by a Guassian, but derived classes my override MakePedestal
//         to simulate it differently (or pick it up from a database).
//
//     fVA1MipRange
//         The dymamic MIP range of the VA1_ALICE pre-amplifier chip 
//
//     fAltroChannelSize
//         The largest number plus one that can be stored in one
//         channel in one time step in the ALTRO ADC chip. 
//
//     fSampleRate
//         How many times the ALTRO ADC chip samples the VA1_ALICE
//         pre-amplifier signal.   The VA1_ALICE chip is read-out at
//         10MHz, while it's possible to drive the ALTRO chip at
//         25MHz.  That means, that the ALTRO chip can have time to
//         sample each VA1_ALICE signal up to 2 times.  Although it's
//         not certain this feature will be used in the production,
//         we'd like have the option, and so it should be reflected in
//         the code.
//
//
// The shaping function of the VA1_ALICE is generally given by 
//
//      f(x) = A(1 - exp(-Bx))
//
// where A is the total charge collected in the pre-amp., and B is a
// paramter that depends on the shaping time of the VA1_ALICE circut.
// 
// When simulating the shaping function of the VA1_ALICe
// pre-amp. chip, we have to take into account, that the shaping
// function depends on the previous value of read from the pre-amp. 
//
// That results in the following algorithm:
//
//    last = 0;
//    FOR charge IN pre-amp. charge train DO 
//      IF last < charge THEN 
//        f(t) = (charge - last) * (1 - exp(-B * t)) + last
//      ELSE
//        f(t) = (last - charge) * exp(-B * t) + charge)
//      ENDIF
//      FOR i IN # samples DO 
//        adc_i = f(i / (# samples))
//      DONE
//      last = charge
//   DONE
//
// Here, 
//
//   pre-amp. charge train 
//       is a series of 128 charges read from the VA1_ALICE chip
//
//   # samples
//       is the number of times the ALTRO ADC samples each of the 128
//       charges from the pre-amp. 
//
// Where Q is the total charge collected by the VA1_ALICE
// pre-amplifier.   Q is then given by 
//
//           E S 
//      Q =  - -
//           e R
//
// where E is the total energy deposited in a silicon strip, R is the
// dynamic range of the VA1_ALICE pre-amp (fVA1MipRange), e is the
// energy deposited by a single MIP, and S ALTRO channel size in each
// time step (fAltroChannelSize).  
//
// The energy deposited per MIP is given by 
//
//      e = M * rho * w 
//
// where M is the universal number 1.664, rho is the density of
// silicon, and w is the depth of the silicon sensor. 
//
// The final ADC count is given by 
//
//      C' = C + P
//
// where P is the (randomized) pedestal (see MakePedestal)
//
// This class uses the class template AliFMDMap<Type> to make an
// internal cache of the energy deposted of the hits.  The class
// template is instantasized as 
//
//  typedef AliFMDMap<std::pair<Float_t, UShort_t> > AliFMDEdepMap;
//
// The first member of the values is the summed energy deposition in a
// given strip, while the second member of the values is the number of
// hits in a given strip.  Using the second member, it's possible to
// do some checks on just how many times a strip got hit, and what
// kind of error we get in our reconstructed hits.  Note, that this
// information is currently not written to the digits tree.  I think a
// QA (Quality Assurance) digit tree is better suited for that task.
// However, the information is there to be used in the future. 
//
//
// Latest changes by Christian Holm Christensen
//
//////////////////////////////////////////////////////////////////////////////

//      /1
//      |           A(-1 + B + exp(-B))
//      | f(x) dx = ------------------- = 1
//      |                    B
//      / 0
//
// and B is the a parameter defined by the shaping time (fShapingTime).  
//
// Solving the above equation, for A gives
//
//                 B
//      A = ----------------
//          -1 + B + exp(-B)
//
// So, if we define the function g: [0,1] -> [0:1] by 
//
//               / v
//               |              Bu + exp(-Bu) - Bv - exp(-Bv) 
//      g(u,v) = | f(x) dx = -A -----------------------------
//               |                            B
//               / u
//
// we can evaluate the ALTRO sample of the VA1_ALICE pre-amp between
// any two times (u, v), by 
//       
//
//                                B	    Bu + exp(-Bu) - Bv - exp(-Bv)
//      C = Q g(u,v) = - Q ---------------- -----------------------------
//		           -1 + B + exp(-B)              B	            
//
//               Bu + exp(-Bu) - Bv - exp(-Bv) 
//        = -  Q -----------------------------
//                    -1 + B + exp(-B)
//

#include <TTree.h>		// ROOT_TTree
#include <TFile.h>
#include "AliFMDDebug.h" 	// Better debug macros
#include "AliFMDDigitizer.h"	// ALIFMDSSDIGITIZER_H
#include "AliFMD.h"		// ALIFMD_H
#include "AliFMDSDigit.h"	// ALIFMDDIGIT_H
#include "AliFMDDigit.h"	// ALIFMDDIGIT_H
#include "AliFMDParameters.h"   // ALIFMDPARAMETERS_H
#include <AliDigitizationInput.h>	// ALIRUNDIGITIZER_H
#include <AliRun.h>		// ALIRUN_H
#include <AliLoader.h>		// ALILOADER_H
#include <AliRunLoader.h>	// ALIRUNLOADER_H
    
//====================================================================
ClassImp(AliFMDDigitizer)
#if 0
;
#endif

//____________________________________________________________________
Bool_t
AliFMDDigitizer::Init()
{
  // 
  // Initialisation
  // 
  if (!AliFMDBaseDigitizer::Init()) return kFALSE;
  
#if 0
  // Get the AliRun object 
  AliRun* run = fRunLoader->GetAliRun();
  if (!run) { 
    AliWarning("Loading gAlice");
    fRunLoader->LoadgAlice();
    if (!run) { 
      AliError("Can not get Run from Run Loader");
      return kFALSE;
    }
  }
  
  // Get the AliFMD object 
  fFMD = static_cast<AliFMD*>(run->GetDetector("FMD"));
  if (!fFMD) {
    AliError("Can not get FMD from gAlice");
    return kFALSE;
  }  
#endif
  return kTRUE;
}


//____________________________________________________________________
void
AliFMDDigitizer::Digitize(Option_t*)
{
  // 
  // Execute this digitizer.  
  // This member function will be called once per event by the passed
  // AliDigitizationInput* digInput object. 
  // 
  // Parameters:
  //    options Not used 
  //
  if (!fDigInput) { 
    AliError("No digitisation input defined");
    return;
  }

  // Clear array of deposited energies 
  fEdep.Reset();

  AliRunLoader* runLoader = 0;
  if (!gAlice) { 
    TString folderName(fDigInput->GetInputFolderName(0));
    runLoader = AliRunLoader::GetRunLoader(folderName.Data());
    if (!runLoader) { 
      AliError(Form("Failed at getting run loader from %s",
		    folderName.Data()));
      return;
    }
    if (!runLoader->GetAliRun()) runLoader->LoadgAlice();
    runLoader->GetAliRun();
  }
  if (!gAlice) { 
    AliError("Can not get Run from Run Loader");
    return;
  }
  
  // Get the AliFMD object 
  fFMD = static_cast<AliFMD*>(gAlice->GetDetector("FMD"));
  if (!fFMD) {
    AliError("Can not get FMD from gAlice");
    return;
  }  


  // Loop over input files
  Int_t nFiles= fDigInput->GetNinputs();
  AliFMDDebug(1, (" Digitizing event number %d, got %d inputs",
		  fDigInput->GetOutputEventNr(), nFiles));
  for (Int_t inputFile = 0; inputFile < nFiles; inputFile++) {
    AliFMDDebug(5, ("Now reading input # %d", inputFile));
    // Get the current loader 
    AliRunLoader* currentLoader = 
      AliRunLoader::GetRunLoader(fDigInput->GetInputFolderName(inputFile));
    if (!currentLoader) { 
      Error("Exec", "no run loader for input file # %d", inputFile);
      continue;
    }

    // Cache contriutions 
    AliFMDDebug(5, ("Now summing the contributions from input # %d",inputFile));

    // Get the FMD loader 
    AliLoader* inFMD = currentLoader->GetLoader("FMDLoader");
    // And load the summable digits
    inFMD->LoadSDigits("READ");
  
    // Get the tree of summable digits
    TTree* sdigitsTree = inFMD->TreeS();
    if (!sdigitsTree)  {
      AliError("No sdigit tree from input");
      continue;
    }
    if (AliLog::GetDebugLevel("FMD","") >= 10) {
      TFile* file = sdigitsTree->GetCurrentFile();
      if (!file) {
	AliWarning("Input tree has no file!");
      }
      else { 
	AliFMDDebug(10, ("Input tree file %s content:", file->GetName()));
	file->ls();
      }
      // AliFMDDebug(5, ("Input tree %s file structure:", 
      //                 sdigitsTree->GetName()));
      // sdigitsTree->Print();
    }

    // Get the FMD branch 
    TBranch* sdigitsBranch = sdigitsTree->GetBranch("FMD");
    if (!sdigitsBranch) {
      AliError("Failed to get sdigit branch");
      return;
    }

    // Set the branch addresses 
    fFMD->SetTreeAddress();

    // Sum contributions from the sdigits
    AliFMDDebug(3, ("Will now sum contributions from SDigits"));
    SumContributions(sdigitsBranch);

    // Unload the sdigits
    inFMD->UnloadSDigits();
  }  

  TString       outFolder(fDigInput->GetOutputFolderName());
  AliRunLoader* out    = AliRunLoader::GetRunLoader(outFolder.Data());
  AliLoader*    outFMD = out->GetLoader("FMDLoader");
  if (!outFMD) { 
    AliError("Cannot get the FMDLoader output folder");
    return;
  }
  TTree* outTree = MakeOutputTree(outFMD);
  if (!outTree) { 
    AliError("Failed to get output tree");
    return;
  }
  // Set the branch address 
  fFMD->SetTreeAddress();
  
  // And digitize the cached data 
  DigitizeHits();
  
  // Fill the tree
  Int_t write = outTree->Fill();
  AliFMDDebug(5, ("Wrote %d bytes to digit tree", write));
  
  // Store the digits
  StoreDigits(outFMD);
}

//____________________________________________________________________
void
AliFMDDigitizer::SumContributions(TBranch* sdigitsBranch) 
{
  // 
  // Sum contributions from SDigits 
  // 
  // Parameters:
  //    sdigitsBranch Branch of SDigit data 
  //
  AliFMDDebug(3, ("Runnin our version of SumContributions"));

  // Get a list of hits from the FMD manager 
  TClonesArray *fmdSDigits = fFMD->SDigits();
  
  // Get number of entries in the tree 
  Int_t nevents  = Int_t(sdigitsBranch->GetEntries());
  
  Int_t read = 0;
  // Loop over the events in the 
  for (Int_t event = 0; event < nevents; event++)  {
    // Read in entry number `event' 
    read += sdigitsBranch->GetEntry(event);
    
    // Get the number of sdigits 
    Int_t nsdigits = fmdSDigits->GetEntries ();
    AliFMDDebug(3, ("Got %5d SDigits", nsdigits));
    for (Int_t sdigit = 0; sdigit < nsdigits; sdigit++) {
      // Get the sdigit number `sdigit'
      AliFMDSDigit* fmdSDigit = 
	static_cast<AliFMDSDigit*>(fmdSDigits->UncheckedAt(sdigit));

      AliFMDDebug(5, ("Adding contribution of %d tracks", 
		      fmdSDigit->GetNTrack()));
      AliFMDDebug(15, ("Contrib from FMD%d%c[%2d,%3d] (%s) from track %d", 
		       fmdSDigit->Detector(),
		       fmdSDigit->Ring(),
		       fmdSDigit->Sector(),
		       fmdSDigit->Strip(),
		       fmdSDigit->GetName(), 
		       fmdSDigit->GetTrack(0)));
      
      // Extract parameters 
      AddContribution(fmdSDigit->Detector(),
		      fmdSDigit->Ring(),
		      fmdSDigit->Sector(),
		      fmdSDigit->Strip(),
		      fmdSDigit->Edep(), 
		      kTRUE,
		      fmdSDigit->GetNTrack(),
		      fmdSDigit->GetTracks());
    }  // sdigit loop
  } // event loop


  AliFMDDebug(3, ("Size of cache: %d bytes, read %d bytes", 
		  int(sizeof(fEdep)), read));
}

//____________________________________________________________________
//
// EOF
//
Commit	Line	Data
09b6c804	1	//************************************************************************
	2	// Copyright(c) 2004, 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	// $Id$ */
	16	/**
	17	* @file AliFMDDigitizer.cxx
	18	*
	19	* @author Christian Holm Christensen <cholm@nbi.dk>
	20	* @date Mon Mar 27 12:38:26 2006
	21	* @brief FMD Digitizers implementation
	22	*
	23	* @ingroup FMD_sim
	24	*/
4347b38f	25	//////////////////////////////////////////////////////////////////////////////
	26	//
	27	// This class contains the procedures simulation ADC signal for the
ef8e8623	28	// Forward Multiplicity detector : SDigits->Digits
4347b38f	29	//
	30	// Digits consists of
	31	// - Detector #
	32	// - Ring ID
	33	// - Sector #
	34	// - Strip #
	35	// - ADC count in this channel
	36	//
ef8e8623	37	// Digits consists of
	38	// - Detector #
	39	// - Ring ID
	40	// - Sector #
	41	// - Strip #
	42	// - Total energy deposited in the strip
	43	// - ADC count in this channel
	44	//
4347b38f	45	// As the Digits and SDigits have so much in common, the classes
	46	// AliFMDDigitizer and AliFMDSDigitizer are implemented via a base
	47	// class AliFMDBaseDigitizer.
	48	//
	49	// +---------------------+
	50	// \| AliFMDBaseDigitizer \|
	51	// +---------------------+
	52	// ^
	53	// \|
	54	// +----------+---------+
	55	// \| \|
	56	// +-----------------+ +------------------+
	57	// \| AliFMDDigitizer \| \| AliFMDSDigitizer \|
	58	// +-----------------+ +------------------+
ef8e8623	59	// \|
	60	// +-------------------+
	61	// \| AliFMDSSDigitizer \|
	62	// +-------------------+
4347b38f	63	//
	64	// These classes has several paramters:
	65	//
	66	// fPedestal
	67	// fPedestalWidth
	68	// (Only AliFMDDigitizer)
	69	// Mean and width of the pedestal. The pedestal is simulated
	70	// by a Guassian, but derived classes my override MakePedestal
	71	// to simulate it differently (or pick it up from a database).
	72	//
	73	// fVA1MipRange
	74	// The dymamic MIP range of the VA1_ALICE pre-amplifier chip
	75	//
	76	// fAltroChannelSize
	77	// The largest number plus one that can be stored in one
	78	// channel in one time step in the ALTRO ADC chip.
	79	//
	80	// fSampleRate
	81	// How many times the ALTRO ADC chip samples the VA1_ALICE
	82	// pre-amplifier signal. The VA1_ALICE chip is read-out at
	83	// 10MHz, while it's possible to drive the ALTRO chip at
	84	// 25MHz. That means, that the ALTRO chip can have time to
	85	// sample each VA1_ALICE signal up to 2 times. Although it's
	86	// not certain this feature will be used in the production,
	87	// we'd like have the option, and so it should be reflected in
	88	// the code.
	89	//
4347b38f	90	//
e802be3e	91	// The shaping function of the VA1_ALICE is generally given by
4347b38f	92	//
e802be3e	93	// f(x) = A(1 - exp(-Bx))
4347b38f	94	//
e802be3e	95	// where A is the total charge collected in the pre-amp., and B is a
	96	// paramter that depends on the shaping time of the VA1_ALICE circut.
	97	//
	98	// When simulating the shaping function of the VA1_ALICe
	99	// pre-amp. chip, we have to take into account, that the shaping
	100	// function depends on the previous value of read from the pre-amp.
	101	//
	102	// That results in the following algorithm:
	103	//
	104	// last = 0;
	105	// FOR charge IN pre-amp. charge train DO
	106	// IF last < charge THEN
	107	// f(t) = (charge - last) * (1 - exp(-B * t)) + last
	108	// ELSE
	109	// f(t) = (last - charge) * exp(-B * t) + charge)
	110	// ENDIF
	111	// FOR i IN # samples DO
	112	// adc_i = f(i / (# samples))
	113	// DONE
	114	// last = charge
	115	// DONE
	116	//
	117	// Here,
	118	//
	119	// pre-amp. charge train
	120	// is a series of 128 charges read from the VA1_ALICE chip
	121	//
	122	// # samples
	123	// is the number of times the ALTRO ADC samples each of the 128
	124	// charges from the pre-amp.
4347b38f	125	//
	126	// Where Q is the total charge collected by the VA1_ALICE
	127	// pre-amplifier. Q is then given by
	128	//
	129	// E S
	130	// Q = - -
	131	// e R
	132	//
	133	// where E is the total energy deposited in a silicon strip, R is the
	134	// dynamic range of the VA1_ALICE pre-amp (fVA1MipRange), e is the
	135	// energy deposited by a single MIP, and S ALTRO channel size in each
	136	// time step (fAltroChannelSize).
	137	//
	138	// The energy deposited per MIP is given by
	139	//
	140	// e = M * rho * w
	141	//
	142	// where M is the universal number 1.664, rho is the density of
	143	// silicon, and w is the depth of the silicon sensor.
	144	//
	145	// The final ADC count is given by
	146	//
	147	// C' = C + P
	148	//
	149	// where P is the (randomized) pedestal (see MakePedestal)
	150	//
	151	// This class uses the class template AliFMDMap<Type> to make an
	152	// internal cache of the energy deposted of the hits. The class
	153	// template is instantasized as
	154	//
	155	// typedef AliFMDMap<std::pair<Float_t, UShort_t> > AliFMDEdepMap;
	156	//
	157	// The first member of the values is the summed energy deposition in a
	158	// given strip, while the second member of the values is the number of
	159	// hits in a given strip. Using the second member, it's possible to
	160	// do some checks on just how many times a strip got hit, and what
	161	// kind of error we get in our reconstructed hits. Note, that this
	162	// information is currently not written to the digits tree. I think a
	163	// QA (Quality Assurance) digit tree is better suited for that task.
	164	// However, the information is there to be used in the future.
	165	//
	166	//
	167	// Latest changes by Christian Holm Christensen
	168	//
	169	//////////////////////////////////////////////////////////////////////////////
	170
e802be3e	171	// /1
	172	// \| A(-1 + B + exp(-B))
	173	// \| f(x) dx = ------------------- = 1
	174	// \| B
	175	// / 0
	176	//
	177	// and B is the a parameter defined by the shaping time (fShapingTime).
	178	//
	179	// Solving the above equation, for A gives
	180	//
	181	// B
	182	// A = ----------------
	183	// -1 + B + exp(-B)
	184	//
	185	// So, if we define the function g: [0,1] -> [0:1] by
	186	//
	187	// / v
	188	// \| Bu + exp(-Bu) - Bv - exp(-Bv)
	189	// g(u,v) = \| f(x) dx = -A -----------------------------
	190	// \| B
	191	// / u
	192	//
	193	// we can evaluate the ALTRO sample of the VA1_ALICE pre-amp between
	194	// any two times (u, v), by
	195	//
	196	//
	197	// B Bu + exp(-Bu) - Bv - exp(-Bv)
	198	// C = Q g(u,v) = - Q ---------------- -----------------------------
	199	// -1 + B + exp(-B) B
	200	//
	201	// Bu + exp(-Bu) - Bv - exp(-Bv)
	202	// = - Q -----------------------------
	203	// -1 + B + exp(-B)
	204	//
	205
56b1929b	206	#include <TTree.h> // ROOT_TTree
ef8e8623	207	#include <TFile.h>
	208	#include "AliFMDDebug.h" // Better debug macros
	209	#include "AliFMDDigitizer.h" // ALIFMDSSDIGITIZER_H
e802be3e	210	#include "AliFMD.h" // ALIFMD_H
ef8e8623	211	#include "AliFMDSDigit.h" // ALIFMDDIGIT_H
e802be3e	212	#include "AliFMDDigit.h" // ALIFMDDIGIT_H
8f6ee336	213	#include "AliFMDParameters.h" // ALIFMDPARAMETERS_H
f21fc003	214	#include <AliDigitizationInput.h> // ALIRUNDIGITIZER_H
56b1929b	215	#include <AliRun.h> // ALIRUN_H
	216	#include <AliLoader.h> // ALILOADER_H
	217	#include <AliRunLoader.h> // ALIRUNLOADER_H
4347b38f	218
4347b38f	219	//====================================================================
925e6570	220	ClassImp(AliFMDDigitizer)
ef8e8623	221	#if 0
	222	;
	223	#endif
66d2ede1	224
4347b38f	225	//____________________________________________________________________
ef8e8623	226	Bool_t
ef8e8623	227	AliFMDDigitizer::Init()
4347b38f	228	{
09b6c804	229	//
ef8e8623	230	// Initialisation
09b6c804	231	//
ef8e8623	232	if (!AliFMDBaseDigitizer::Init()) return kFALSE;
42f1b2f5	233
ef8e8623	234	#if 0
	235	// Get the AliRun object
	236	AliRun* run = fRunLoader->GetAliRun();
	237	if (!run) {
	238	AliWarning("Loading gAlice");
	239	fRunLoader->LoadgAlice();
	240	if (!run) {
	241	AliError("Can not get Run from Run Loader");
	242	return kFALSE;
	243	}
42f1b2f5	244	}
66d2ede1	245
ef8e8623	246	// Get the AliFMD object
	247	fFMD = static_cast<AliFMD*>(run->GetDetector("FMD"));
	248	if (!fFMD) {
	249	AliError("Can not get FMD from gAlice");
	250	return kFALSE;
	251	}
	252	#endif
	253	return kTRUE;
4347b38f	254	}
3d44ce66	255
88cb7938	256
4347b38f	257	//____________________________________________________________________
4347b38f	258	void
f21fc003	259	AliFMDDigitizer::Digitize(Option_t*)
4347b38f	260	{
09b6c804	261	//
	262	// Execute this digitizer.
	263	// This member function will be called once per event by the passed
f21fc003	264	// AliDigitizationInput* digInput object.
09b6c804	265	//
	266	// Parameters:
	267	// options Not used
	268	//
f21fc003	269	if (!fDigInput) {
f21fc003	270	AliError("No digitisation input defined");
ef8e8623	271	return;
	272	}
	273
	274	// Clear array of deposited energies
	275	fEdep.Reset();
	276
	277	AliRunLoader* runLoader = 0;
	278	if (!gAlice) {
f21fc003	279	TString folderName(fDigInput->GetInputFolderName(0));
ef8e8623	280	runLoader = AliRunLoader::GetRunLoader(folderName.Data());
	281	if (!runLoader) {
	282	AliError(Form("Failed at getting run loader from %s",
	283	folderName.Data()));
	284	return;
	285	}
faf80567	286	if (!runLoader->GetAliRun()) runLoader->LoadgAlice();
ef8e8623	287	runLoader->GetAliRun();
	288	}
	289	if (!gAlice) {
	290	AliError("Can not get Run from Run Loader");
	291	return;
	292	}
	293
	294	// Get the AliFMD object
	295	fFMD = static_cast<AliFMD*>(gAlice->GetDetector("FMD"));
	296	if (!fFMD) {
	297	AliError("Can not get FMD from gAlice");
	298	return;
	299	}
	300
	301
	302	// Loop over input files
f21fc003	303	Int_t nFiles= fDigInput->GetNinputs();
ef8e8623	304	AliFMDDebug(1, (" Digitizing event number %d, got %d inputs",
f21fc003	305	fDigInput->GetOutputEventNr(), nFiles));
ef8e8623	306	for (Int_t inputFile = 0; inputFile < nFiles; inputFile++) {
	307	AliFMDDebug(5, ("Now reading input # %d", inputFile));
	308	// Get the current loader
	309	AliRunLoader* currentLoader =
f21fc003	310	AliRunLoader::GetRunLoader(fDigInput->GetInputFolderName(inputFile));
ef8e8623	311	if (!currentLoader) {
09b6c804	312	Error("Exec", "no run loader for input file # %d", inputFile);
ef8e8623	313	continue;
	314	}
	315
	316	// Cache contriutions
	317	AliFMDDebug(5, ("Now summing the contributions from input # %d",inputFile));
	318
	319	// Get the FMD loader
	320	AliLoader* inFMD = currentLoader->GetLoader("FMDLoader");
	321	// And load the summable digits
	322	inFMD->LoadSDigits("READ");
	323
	324	// Get the tree of summable digits
	325	TTree* sdigitsTree = inFMD->TreeS();
	326	if (!sdigitsTree) {
f21fc003	327	AliError("No sdigit tree from input");
ef8e8623	328	continue;
	329	}
	330	if (AliLog::GetDebugLevel("FMD","") >= 10) {
	331	TFile* file = sdigitsTree->GetCurrentFile();
	332	if (!file) {
	333	AliWarning("Input tree has no file!");
	334	}
	335	else {
	336	AliFMDDebug(10, ("Input tree file %s content:", file->GetName()));
	337	file->ls();
	338	}
	339	// AliFMDDebug(5, ("Input tree %s file structure:",
	340	// sdigitsTree->GetName()));
	341	// sdigitsTree->Print();
	342	}
	343
	344	// Get the FMD branch
	345	TBranch* sdigitsBranch = sdigitsTree->GetBranch("FMD");
	346	if (!sdigitsBranch) {
	347	AliError("Failed to get sdigit branch");
	348	return;
	349	}
	350
	351	// Set the branch addresses
	352	fFMD->SetTreeAddress();
	353
	354	// Sum contributions from the sdigits
faf80567	355	AliFMDDebug(3, ("Will now sum contributions from SDigits"));
ef8e8623	356	SumContributions(sdigitsBranch);
	357
	358	// Unload the sdigits
	359	inFMD->UnloadSDigits();
	360	}
	361
f21fc003	362	TString outFolder(fDigInput->GetOutputFolderName());
ef8e8623	363	AliRunLoader* out = AliRunLoader::GetRunLoader(outFolder.Data());
	364	AliLoader* outFMD = out->GetLoader("FMDLoader");
	365	if (!outFMD) {
	366	AliError("Cannot get the FMDLoader output folder");
	367	return;
	368	}
	369	TTree* outTree = MakeOutputTree(outFMD);
	370	if (!outTree) {
	371	AliError("Failed to get output tree");
	372	return;
	373	}
	374	// Set the branch address
	375	fFMD->SetTreeAddress();
	376
	377	// And digitize the cached data
	378	DigitizeHits();
	379
	380	// Fill the tree
	381	Int_t write = outTree->Fill();
83ad576a	382	AliFMDDebug(5, ("Wrote %d bytes to digit tree", write));
ef8e8623	383
	384	// Store the digits
	385	StoreDigits(outFMD);
4347b38f	386	}
3d44ce66	387
4347b38f	388	//____________________________________________________________________
4347b38f	389	void
ef8e8623	390	AliFMDDigitizer::SumContributions(TBranch* sdigitsBranch)
4347b38f	391	{
09b6c804	392	//
	393	// Sum contributions from SDigits
	394	//
	395	// Parameters:
	396	// sdigitsBranch Branch of SDigit data
	397	//
83ad576a	398	AliFMDDebug(3, ("Runnin our version of SumContributions"));
ef8e8623	399
	400	// Get a list of hits from the FMD manager
	401	TClonesArray *fmdSDigits = fFMD->SDigits();
	402
	403	// Get number of entries in the tree
	404	Int_t nevents = Int_t(sdigitsBranch->GetEntries());
4347b38f	405
ef8e8623	406	Int_t read = 0;
	407	// Loop over the events in the
	408	for (Int_t event = 0; event < nevents; event++) {
	409	// Read in entry number `event'
	410	read += sdigitsBranch->GetEntry(event);
	411
	412	// Get the number of sdigits
	413	Int_t nsdigits = fmdSDigits->GetEntries ();
83ad576a	414	AliFMDDebug(3, ("Got %5d SDigits", nsdigits));
ef8e8623	415	for (Int_t sdigit = 0; sdigit < nsdigits; sdigit++) {
	416	// Get the sdigit number `sdigit'
	417	AliFMDSDigit* fmdSDigit =
	418	static_cast<AliFMDSDigit*>(fmdSDigits->UncheckedAt(sdigit));
faf80567	419
	420	AliFMDDebug(5, ("Adding contribution of %d tracks",
	421	fmdSDigit->GetNTrack()));
	422	AliFMDDebug(15, ("Contrib from FMD%d%c[%2d,%3d] (%s) from track %d",
	423	fmdSDigit->Detector(),
	424	fmdSDigit->Ring(),
	425	fmdSDigit->Sector(),
	426	fmdSDigit->Strip(),
	427	fmdSDigit->GetName(),
	428	fmdSDigit->GetTrack(0)));
ef8e8623	429
	430	// Extract parameters
	431	AddContribution(fmdSDigit->Detector(),
	432	fmdSDigit->Ring(),
	433	fmdSDigit->Sector(),
	434	fmdSDigit->Strip(),
83ad576a	435	fmdSDigit->Edep(),
b2e6f0b0	436	kTRUE,
faf80567	437	fmdSDigit->GetNTrack(),
faf80567	438	fmdSDigit->GetTracks());
ef8e8623	439	} // sdigit loop
	440	} // event loop
	441
	442
	443	AliFMDDebug(3, ("Size of cache: %d bytes, read %d bytes",
a828379a	444	int(sizeof(fEdep)), read));
4347b38f	445	}
66d2ede1	446
4347b38f	447	//____________________________________________________________________
	448	//
	449	// EOF
	450	//
88cb7938	451
	452
	453
66d2ede1	454