Updated detailed geometry needed by FMD people for some studies

[u/mrichter/AliRoot.git] / ITS / AliITSFDigitizer.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.                  *
 **************************************************************************/
 
/*
$Log$
*/

#include <stdlib.h>
#include <iostream.h>
#include <TObjArray.h>
#include <TClonesArray.h>
#include <TTree.h>
#include <TBranch.h>
#include <TFile.h>

#include <AliRun.h>
#include <AliRunDigitizer.h>

#include "AliITSFDigitizer.h"
// #include "AliITSpList.h"
#include "AliITSmodule.h"
#include "AliITSgeom.h"
#include "AliITSsimulationFastPoints.h"

ClassImp(AliITSFDigitizer)

//______________________________________________________________________
AliITSFDigitizer::AliITSFDigitizer() : AliDigitizer(){
//
// Default constructor.
//
    fITS      = 0;
    fInit     = kFALSE;
}
//______________________________________________________________________
AliITSFDigitizer::AliITSFDigitizer(AliRunDigitizer *mngr) : AliDigitizer(mngr){
//
// Standard constructor.
//
    fITS      = 0;
    fInit     = kFALSE;
}
//______________________________________________________________________
AliITSFDigitizer::~AliITSFDigitizer(){
//
// Default destructor. 
//
    fITS = 0; // don't delete fITS. Done else where.
}
//______________________________________________________________________
Bool_t AliITSFDigitizer::Init(){
//
// Initialization. 
// loads ITS and ITSgeom.
// Inputs:
//      none.
// Outputs:
//      none.

  
  fInit = kFALSE;
  if(!gAlice) {
    fITS      = 0;
    Warning("Init","gAlice not found");
    return fInit;
  }
  fITS = (AliITS *)(gAlice->GetDetector("ITS"));
  if(!fITS){
    Warning("Init","ITS not found");
    return fInit;
  } 
  if(!fITS->GetITSgeom()){
    Warning("Init","ITS geometry not found");
    return fInit;
  }
  return fInit = kTRUE;
}
////////////////////////////////////////////////////////////////////////
void AliITSFDigitizer::Exec(Option_t* opt){
//
// Main digitization function. 
// Inputs:
//      Option_t * opt  "deb" ... more verbose output 
//

  AliITSsimulationFastPoints *sim = new AliITSsimulationFastPoints();

  TTree *outputTreeR = fManager->GetTreeR();
  TClonesArray *recPoints = fITS->RecPoints();
//  TBranch *branch =
      fITS->MakeBranchInTree(outputTreeR,"ITSRecPointsF",
					   &recPoints,4000,0);
  
  Int_t nModules;
  fITS->InitModules(-1,nModules);

// load hits into modules

  for (Int_t iFile = 0; iFile < fManager->GetNinputs(); iFile++) {
    fITS->FillModules(fManager->GetInputTreeH(iFile),
		      fManager->GetMask(iFile));
  }
  
// transform hits to fast rec points

  AliITSgeom *geom = fITS->GetITSgeom();
  for(Int_t moduleIndex = 0; moduleIndex < geom->GetIndexMax(); moduleIndex++){
    sim->CreateFastRecPoints(moduleIndex);
//    branch->Fill();
    outputTreeR->Fill();
    fITS->ResetRecPoints();
  }
  outputTreeR->AutoSave();

}
////////////////////////////////////////////////////////////////////////