[u/mrichter/AliRoot.git] / ITS / AliITSreconstruction.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$ */
/////////////////////////////////////////////////////////////////////////
//                                                                     //
// Class for ITS RecPoint reconstruction                               //
//                                                                     //
////////////////////////////////////////////////////////////////////////

#include <TString.h>
#include "AliRun.h"
#include "AliRunLoader.h"
#include "AliITSDetTypeRec.h"
#include "AliITSLoader.h"
#include "AliITSreconstruction.h"
#include "AliITSgeom.h"


ClassImp(AliITSreconstruction)

//______________________________________________________________________
AliITSreconstruction::AliITSreconstruction():
 fInit(kFALSE),
 fEnt(0),
 fEnt0(0),
 fDetTypeRec(0x0),
 fDfArp(kFALSE),
 fITSgeom(0x0),
 fLoader(0x0),
 fRunLoader(0x0)
{
    // Default constructor.
    // Inputs:
    //  none.
    // Outputs:
    //   none.
    // Return:
    //    A zero-ed constructed AliITSreconstruction class.
    fDet[0] = fDet[1] = fDet[2] = kTRUE;
}
//______________________________________________________________________

AliITSreconstruction::AliITSreconstruction(AliRunLoader *rl):
 fInit(kFALSE),
 fEnt(0),
 fEnt0(0),
 fDetTypeRec(0x0),
 fDfArp(kFALSE),
 fITSgeom(0x0),
 fLoader(0x0),
 fRunLoader(rl)
{
  fDet[0] = fDet[1] = fDet[2] = kTRUE;
}
//______________________________________________________________________
AliITSreconstruction::AliITSreconstruction(const char* filename):
 fInit(kFALSE),
 fEnt(0),
 fEnt0(0),
 fDetTypeRec(0x0),
 fDfArp(kFALSE),
 fITSgeom(0x0),
 fLoader(0x0),
 fRunLoader(0x0)
{
    // Standard constructor.
    // Inputs:
    //  const char* filename    filename containing the digits to be
    //                          reconstructed. If filename = 0 (nil)
    //                          then no file is opened but a file is
    //                          assumed to already be opened. This 
    //                          already opened file will be used.
    // Outputs:
    //   none.
    // Return:
    //    A standardly constructed AliITSreconstruction class.

    fDet[0] = fDet[1] = fDet[2] = kTRUE;

    fRunLoader = AliRunLoader::Open(filename);
    if (fRunLoader == 0x0)
     {
       Error("AliITSreconstruction","Can not load the session",filename);
       return;
     }

}

//______________________________________________________________________
AliITSreconstruction::AliITSreconstruction(const AliITSreconstruction &rec):TTask(rec),
fInit(rec.fInit),
fEnt(rec.fEnt),
fEnt0(rec.fEnt0),
fDetTypeRec(rec.fDetTypeRec),
fDfArp(rec.fDfArp),
fITSgeom(rec.fITSgeom),
fLoader(rec.fLoader),
fRunLoader(rec.fRunLoader)
{
    // Copy constructor. 

  
}

//______________________________________________________________________
AliITSreconstruction& AliITSreconstruction::operator=(const AliITSreconstruction& source){
    // Assignment operator. 
  this->~AliITSreconstruction();
  new(this) AliITSreconstruction(source);
  return *this;

}

//______________________________________________________________________
AliITSreconstruction::~AliITSreconstruction(){
    //    A destroyed AliITSreconstruction class.
    
    //fITS      = 0;
    delete fRunLoader;
    
}
//______________________________________________________________________
Bool_t AliITSreconstruction::Init(){
    // Class Initilizer.
    // Inputs:
    //  none.
    // Outputs:
    //   none.
    // Return:
    //    kTRUE if no errors initilizing this class occurse else kFALSE
    Info("Init","");
    if (fRunLoader == 0x0)
     {
       Error("Init","Run Loader is NULL");
       return kFALSE;
     }
    //  fRunLoader->LoadgAlice();
    //   fRunLoader->LoadHeader();  

    fLoader = (AliITSLoader*) fRunLoader->GetLoader("ITSLoader");
    if(!fLoader) {
      Error("Init","ITS loader not found");
      fInit = kFALSE;
    }

    // Now ready to init.
 
    //fRunLoader->CdGAFile();
    fITSgeom = fLoader->GetITSgeom();

    fDetTypeRec = new AliITSDetTypeRec();
    fDetTypeRec->SetITSgeom(fITSgeom);
    fDetTypeRec->SetDefaults();
    fDet[0] = fDet[1] = fDet[2] = kTRUE;
    fEnt0 = 0;

    //fEnt  = gAlice->GetEventsPerRun();
    fEnt = Int_t(fRunLoader->GetNumberOfEvents());

    fLoader->LoadDigits("read");
    fLoader->LoadRecPoints("recreate");
    if (fLoader->TreeR() == 0x0) fLoader->MakeTree("R");
 
    fDetTypeRec->SetTreeAddressD(fLoader->TreeD());
    fDetTypeRec->MakeBranchR(fLoader->TreeR());
    fDetTypeRec->SetTreeAddressR(fLoader->TreeR());

    fInit = InitRec();

    Info("Init","  Done\n\n\n");

    return fInit;
}
//______________________________________________________________________
Bool_t AliITSreconstruction::InitRec(){
    // Sets up Reconstruction part of AliITSDetType..
    // Inputs:
    //      none.
    // Outputs:
    //      none.
    // Return:
    //      none.
  /*
  //AliITSDetType *idt;
  fDetTypeRec->SetLoader(fLoader);
    // SPD
  if(fDet[kSPD]){
    Info("InitRec","SPD");
    //idt = fITS->DetType(kSPD);
    AliITSsegmentationSPD *segSPD = (AliITSsegmentationSPD*)fDetTypeRec->GetSegmentationModel(0);
      TClonesArray *digSPD = fDetTypeRec->DigitsAddress(kSPD);
      TClonesArray *recpSPD = fDetTypeRec->ClustersAddress(kSPD);
      Info("InitRec","idt = %#x; digSPD = %#x; recpSPD = %#x",fDetTypeRec,digSPD,recpSPD);
      AliITSClusterFinderSPD *recSPD = new AliITSClusterFinderSPD(segSPD,digSPD,recpSPD);
      fDetTypeRec->SetReconstructionModel(kSPD,recSPD);
    } // end if fDet[kSPD].
  // SDD
  if(fDet[kSDD]){
    Info("InitRec","SDD");
    //    idt = fITS->DetType(kSDD);
    AliITSsegmentationSDD *segSDD = (AliITSsegmentationSDD*)
      fDetTypeRec->GetSegmentationModel(1);
    AliITSresponseSDD *resSDD = (AliITSresponseSDD*)
      fDetTypeRec->GetCalibrationModel(fDetTypeRec->GetITSgeom()->GetStartSDD()); 
    TClonesArray *digSDD = fDetTypeRec->DigitsAddress(kSDD);
    TClonesArray *recpSDD = fDetTypeRec->ClustersAddress(kSDD);
    AliITSClusterFinderSDD *recSDD =new AliITSClusterFinderSDD(segSDD,
							       resSDD,
							       digSDD,recpSDD);
    fDetTypeRec->SetReconstructionModel(kSDD,recSDD);
  } // end if fDet[kSDD]
    // SSD
  if(fDet[kSSD]){
    Info("InitRec","SSD");
    //idt = fITS->DetType(kSSD);
    AliITSsegmentationSSD *segSSD = (AliITSsegmentationSSD*)
                                       fDetTypeRec->GetSegmentationModel(2);
      TClonesArray *digSSD = fDetTypeRec->DigitsAddress(kSSD);
      AliITSClusterFinderSSD *recSSD =new AliITSClusterFinderSSD(segSSD,
								 digSSD);
      recSSD->SetITSgeom(fDetTypeRec->GetITSgeom());
      fDetTypeRec->SetReconstructionModel(kSSD,recSSD);
    } // end if fDet[kSSD]
  */
  fDetTypeRec->SetDefaultClusterFinders();
    Info("InitRec","    Done\n");
    return kTRUE;
}
//______________________________________________________________________ 
void AliITSreconstruction::Exec(const Option_t *opt){
    // Main reconstruction function.
    // Inputs:
    //      Option_t * opt   list of subdetector to digitize. =0 all.
    // Outputs:
    //      none.
    // Return:
    //      none.
    Option_t *lopt;
    Int_t evnt;

    if(strstr(opt,"All")||strstr(opt,"ALL")||strstr(opt,"ITS")||opt==0){
      fDet[0] = fDet[1] = fDet[2] = kTRUE;
      lopt = "All";
    }else{
      fDet[0] = fDet[1] = fDet[2] = kFALSE;
      if(strstr(opt,"SPD")) fDet[kSPD] = kTRUE;
      if(strstr(opt,"SDD")) fDet[kSDD] = kTRUE;
      if(strstr(opt,"SSD")) fDet[kSSD] = kTRUE;
      if(fDet[kSPD] && fDet[kSDD] && fDet[kSSD]) lopt = "All";
      else lopt = opt;
    } // end if strstr(opt,...)

    if(!fInit){
      cout << "Initilization Failed, Can't run Exec." << endl;
      return;
    } // end if !fInit
    for(evnt=0;evnt<fEnt;evnt++)
     {
      Info("Exec","");
      Info("Exec","Processing Event %d",evnt);
      Info("Exec","");

      fRunLoader->GetEvent(evnt);
      if (fLoader->TreeR() == 0x0) fLoader->MakeTree("R");
      fDetTypeRec->MakeBranchR(0);
      fDetTypeRec->SetTreeAddressR(fLoader->TreeR());
      fDetTypeRec->SetTreeAddressD(fLoader->TreeD());
      fDetTypeRec->DigitsToRecPoints(fLoader->TreeD(),fLoader->TreeR(),0,lopt);
    } // end for evnt
}
//______________________________________________________________________ 
void AliITSreconstruction::SetOutputFile(TString filename){
  // Set a new file name for recpoints. 
  // It must be called before Init()
  if(!fLoader)fLoader = (AliITSLoader*) fRunLoader->GetLoader("ITSLoader");
  if(fLoader){
    Info("SetOutputFile","name for rec points is %s",filename.Data());
    fLoader->SetRecPointsFileName(filename);
  }
  else {
    Error("SetOutputFile",
    "ITS loader not available. Not possible to set name: %s",filename.Data());
  }
}
Commit	Line	Data
	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
	16	/* $Id$ */
	17	/////////////////////////////////////////////////////////////////////////
	18	// //
	19	// Class for ITS RecPoint reconstruction //
	20	// //
	21	////////////////////////////////////////////////////////////////////////
	22
	23	#include <TString.h>
	24	#include "AliRun.h"
	25	#include "AliRunLoader.h"
	26	#include "AliITSDetTypeRec.h"
	27	#include "AliITSLoader.h"
	28	#include "AliITSreconstruction.h"
	29	#include "AliITSgeom.h"
	30
	31
	32	ClassImp(AliITSreconstruction)
	33
	34	//______________________________________________________________________
	35	AliITSreconstruction::AliITSreconstruction():
	36	fInit(kFALSE),
	37	fEnt(0),
	38	fEnt0(0),
	39	fDetTypeRec(0x0),
	40	fDfArp(kFALSE),
	41	fITSgeom(0x0),
	42	fLoader(0x0),
	43	fRunLoader(0x0)
	44	{
	45	// Default constructor.
	46	// Inputs:
	47	// none.
	48	// Outputs:
	49	// none.
	50	// Return:
	51	// A zero-ed constructed AliITSreconstruction class.
	52	fDet[0] = fDet[1] = fDet[2] = kTRUE;
	53	}
	54	//______________________________________________________________________
	55
	56	AliITSreconstruction::AliITSreconstruction(AliRunLoader *rl):
	57	fInit(kFALSE),
	58	fEnt(0),
	59	fEnt0(0),
	60	fDetTypeRec(0x0),
	61	fDfArp(kFALSE),
	62	fITSgeom(0x0),
	63	fLoader(0x0),
	64	fRunLoader(rl)
	65	{
	66	fDet[0] = fDet[1] = fDet[2] = kTRUE;
	67	}
	68	//______________________________________________________________________
	69	AliITSreconstruction::AliITSreconstruction(const char* filename):
	70	fInit(kFALSE),
	71	fEnt(0),
	72	fEnt0(0),
	73	fDetTypeRec(0x0),
	74	fDfArp(kFALSE),
	75	fITSgeom(0x0),
	76	fLoader(0x0),
	77	fRunLoader(0x0)
	78	{
	79	// Standard constructor.
	80	// Inputs:
	81	// const char* filename filename containing the digits to be
	82	// reconstructed. If filename = 0 (nil)
	83	// then no file is opened but a file is
	84	// assumed to already be opened. This
	85	// already opened file will be used.
	86	// Outputs:
	87	// none.
	88	// Return:
	89	// A standardly constructed AliITSreconstruction class.
	90
	91	fDet[0] = fDet[1] = fDet[2] = kTRUE;
	92
	93	fRunLoader = AliRunLoader::Open(filename);
	94	if (fRunLoader == 0x0)
	95	{
	96	Error("AliITSreconstruction","Can not load the session",filename);
	97	return;
	98	}
	99
	100	}
	101
	102	//______________________________________________________________________
	103	AliITSreconstruction::AliITSreconstruction(const AliITSreconstruction &rec):TTask(rec),
	104	fInit(rec.fInit),
	105	fEnt(rec.fEnt),
	106	fEnt0(rec.fEnt0),
	107	fDetTypeRec(rec.fDetTypeRec),
	108	fDfArp(rec.fDfArp),
	109	fITSgeom(rec.fITSgeom),
	110	fLoader(rec.fLoader),
	111	fRunLoader(rec.fRunLoader)
	112	{
	113	// Copy constructor.
	114
	115
	116	}
	117
	118	//______________________________________________________________________
	119	AliITSreconstruction& AliITSreconstruction::operator=(const AliITSreconstruction& source){
	120	// Assignment operator.
	121	this->~AliITSreconstruction();
	122	new(this) AliITSreconstruction(source);
	123	return *this;
	124
	125	}
	126
	127	//______________________________________________________________________
	128	AliITSreconstruction::~AliITSreconstruction(){
	129	// A destroyed AliITSreconstruction class.
	130
	131	//fITS = 0;
	132	delete fRunLoader;
	133
	134	}
	135	//______________________________________________________________________
	136	Bool_t AliITSreconstruction::Init(){
	137	// Class Initilizer.
	138	// Inputs:
	139	// none.
	140	// Outputs:
	141	// none.
	142	// Return:
	143	// kTRUE if no errors initilizing this class occurse else kFALSE
	144	Info("Init","");
	145	if (fRunLoader == 0x0)
	146	{
	147	Error("Init","Run Loader is NULL");
	148	return kFALSE;
	149	}
	150	// fRunLoader->LoadgAlice();
	151	// fRunLoader->LoadHeader();
	152
	153	fLoader = (AliITSLoader*) fRunLoader->GetLoader("ITSLoader");
	154	if(!fLoader) {
	155	Error("Init","ITS loader not found");
	156	fInit = kFALSE;
	157	}
	158
	159	// Now ready to init.
	160
	161	//fRunLoader->CdGAFile();
	162	fITSgeom = fLoader->GetITSgeom();
	163
	164	fDetTypeRec = new AliITSDetTypeRec();
	165	fDetTypeRec->SetITSgeom(fITSgeom);
	166	fDetTypeRec->SetDefaults();
	167	fDet[0] = fDet[1] = fDet[2] = kTRUE;
	168	fEnt0 = 0;
	169
	170	//fEnt = gAlice->GetEventsPerRun();
	171	fEnt = Int_t(fRunLoader->GetNumberOfEvents());
	172
	173	fLoader->LoadDigits("read");
	174	fLoader->LoadRecPoints("recreate");
	175	if (fLoader->TreeR() == 0x0) fLoader->MakeTree("R");
	176
	177	fDetTypeRec->SetTreeAddressD(fLoader->TreeD());
	178	fDetTypeRec->MakeBranchR(fLoader->TreeR());
	179	fDetTypeRec->SetTreeAddressR(fLoader->TreeR());
	180
	181	fInit = InitRec();
	182
	183	Info("Init"," Done\n\n\n");
	184
	185	return fInit;
	186	}
	187	//______________________________________________________________________
	188	Bool_t AliITSreconstruction::InitRec(){
	189	// Sets up Reconstruction part of AliITSDetType..
	190	// Inputs:
	191	// none.
	192	// Outputs:
	193	// none.
	194	// Return:
	195	// none.
	196	/*
	197	//AliITSDetType *idt;
	198	fDetTypeRec->SetLoader(fLoader);
	199	// SPD
	200	if(fDet[kSPD]){
	201	Info("InitRec","SPD");
	202	//idt = fITS->DetType(kSPD);
	203	AliITSsegmentationSPD segSPD = (AliITSsegmentationSPD)fDetTypeRec->GetSegmentationModel(0);
	204	TClonesArray *digSPD = fDetTypeRec->DigitsAddress(kSPD);
	205	TClonesArray *recpSPD = fDetTypeRec->ClustersAddress(kSPD);
	206	Info("InitRec","idt = %#x; digSPD = %#x; recpSPD = %#x",fDetTypeRec,digSPD,recpSPD);
	207	AliITSClusterFinderSPD *recSPD = new AliITSClusterFinderSPD(segSPD,digSPD,recpSPD);
	208	fDetTypeRec->SetReconstructionModel(kSPD,recSPD);
	209	} // end if fDet[kSPD].
	210	// SDD
	211	if(fDet[kSDD]){
	212	Info("InitRec","SDD");
	213	// idt = fITS->DetType(kSDD);
	214	AliITSsegmentationSDD segSDD = (AliITSsegmentationSDD)
	215	fDetTypeRec->GetSegmentationModel(1);
	216	AliITSresponseSDD resSDD = (AliITSresponseSDD)
	217	fDetTypeRec->GetCalibrationModel(fDetTypeRec->GetITSgeom()->GetStartSDD());
	218	TClonesArray *digSDD = fDetTypeRec->DigitsAddress(kSDD);
	219	TClonesArray *recpSDD = fDetTypeRec->ClustersAddress(kSDD);
	220	AliITSClusterFinderSDD *recSDD =new AliITSClusterFinderSDD(segSDD,
	221	resSDD,
	222	digSDD,recpSDD);
	223	fDetTypeRec->SetReconstructionModel(kSDD,recSDD);
	224	} // end if fDet[kSDD]
	225	// SSD
	226	if(fDet[kSSD]){
	227	Info("InitRec","SSD");
	228	//idt = fITS->DetType(kSSD);
	229	AliITSsegmentationSSD segSSD = (AliITSsegmentationSSD)
	230	fDetTypeRec->GetSegmentationModel(2);
	231	TClonesArray *digSSD = fDetTypeRec->DigitsAddress(kSSD);
	232	AliITSClusterFinderSSD *recSSD =new AliITSClusterFinderSSD(segSSD,
	233	digSSD);
	234	recSSD->SetITSgeom(fDetTypeRec->GetITSgeom());
	235	fDetTypeRec->SetReconstructionModel(kSSD,recSSD);
	236	} // end if fDet[kSSD]
	237	*/
	238	fDetTypeRec->SetDefaultClusterFinders();
	239	Info("InitRec"," Done\n");
	240	return kTRUE;
	241	}
	242	//______________________________________________________________________
	243	void AliITSreconstruction::Exec(const Option_t *opt){
	244	// Main reconstruction function.
	245	// Inputs:
	246	// Option_t * opt list of subdetector to digitize. =0 all.
	247	// Outputs:
	248	// none.
	249	// Return:
	250	// none.
	251	Option_t *lopt;
	252	Int_t evnt;
	253
	254	if(strstr(opt,"All")\|\|strstr(opt,"ALL")\|\|strstr(opt,"ITS")\|\|opt==0){
	255	fDet[0] = fDet[1] = fDet[2] = kTRUE;
	256	lopt = "All";
	257	}else{
	258	fDet[0] = fDet[1] = fDet[2] = kFALSE;
	259	if(strstr(opt,"SPD")) fDet[kSPD] = kTRUE;
	260	if(strstr(opt,"SDD")) fDet[kSDD] = kTRUE;
	261	if(strstr(opt,"SSD")) fDet[kSSD] = kTRUE;
	262	if(fDet[kSPD] && fDet[kSDD] && fDet[kSSD]) lopt = "All";
	263	else lopt = opt;
	264	} // end if strstr(opt,...)
	265
	266	if(!fInit){
	267	cout << "Initilization Failed, Can't run Exec." << endl;
	268	return;
	269	} // end if !fInit
	270	for(evnt=0;evnt<fEnt;evnt++)
	271	{
	272	Info("Exec","");
	273	Info("Exec","Processing Event %d",evnt);
	274	Info("Exec","");
	275
	276	fRunLoader->GetEvent(evnt);
	277	if (fLoader->TreeR() == 0x0) fLoader->MakeTree("R");
	278	fDetTypeRec->MakeBranchR(0);
	279	fDetTypeRec->SetTreeAddressR(fLoader->TreeR());
	280	fDetTypeRec->SetTreeAddressD(fLoader->TreeD());
	281	fDetTypeRec->DigitsToRecPoints(fLoader->TreeD(),fLoader->TreeR(),0,lopt);
	282	} // end for evnt
	283	}
	284	//______________________________________________________________________
	285	void AliITSreconstruction::SetOutputFile(TString filename){
	286	// Set a new file name for recpoints.
	287	// It must be called before Init()
	288	if(!fLoader)fLoader = (AliITSLoader*) fRunLoader->GetLoader("ITSLoader");
	289	if(fLoader){
	290	Info("SetOutputFile","name for rec points is %s",filename.Data());
	291	fLoader->SetRecPointsFileName(filename);
	292	}
	293	else {
	294	Error("SetOutputFile",
	295	"ITS loader not available. Not possible to set name: %s",filename.Data());
	296	}
	297	}