[u/mrichter/AliRoot.git] / MFT / AliMFTReconstructor.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.                  *
 **************************************************************************/

//====================================================================================================================================================
//
//      Event reconstruction class for the ALICE Muon Forward Tracker
//
//      Contact author: antonio.uras@cern.ch
//
//====================================================================================================================================================

#include "TObjArray.h"
#include "TTree.h"
#include "AliMFTSegmentation.h"
#include "AliMFTClusterFinder.h"
#include "AliReconstructor.h"
#include "AliMFTReconstructor.h"

ClassImp(AliMFTReconstructor)

//====================================================================================================================================================

AliMFTReconstructor::AliMFTReconstructor():
  AliReconstructor(), 
  fDigits(0x0),
  fNPlanes(0)
{

  // default constructor 

}

//====================================================================================================================================================

AliMFTReconstructor::~AliMFTReconstructor() {
  
  // destructor
  
  if (fDigits) {
    for (Int_t iPlane=0; iPlane<fNPlanes; iPlane++) {
      if (fDigits->At(iPlane)) fDigits->At(iPlane)->Delete();
    }
    fDigits->Delete();
    delete fDigits;
    fDigits=0;
  }
  
}

//====================================================================================================================================================

void AliMFTReconstructor::Clear(const Option_t* /*opt*/) {
	
  // Clear arrays
  
  if (fDigits) {
    for (Int_t iPlane=0; iPlane<fNPlanes; iPlane++) {
      if (fDigits->At(iPlane)) ((TClonesArray*)fDigits->At(iPlane))->Delete();
    }
    fDigits->Delete();
    delete fDigits;
    fDigits = NULL;
  }
  
}

//====================================================================================================================================================

void AliMFTReconstructor::Init() {

  AliMFTSegmentation *segmentation = new AliMFTSegmentation("AliMFTGeometry.root");
  fNPlanes = segmentation->GetNPlanes();
  delete segmentation;

  fDigits = new TObjArray(fNPlanes);
  fDigits->SetOwner(kTRUE);
  for (Int_t iPlane=0; iPlane<fNPlanes; iPlane++) {
    fDigits->AddAt(new TClonesArray("AliMFTDigit"),iPlane);
    ((TClonesArray*)fDigits->At(iPlane))->SetOwner(kTRUE);
  }
  
  AliInfo("    ************* Using the MFT reconstructor! ****** ");
  
  return;

}

//====================================================================================================================================================

void AliMFTReconstructor::ResetDigits() {

  // Reset number of digits and the digits array for the MFT detector.

  if (!fDigits) return;
  for (Int_t iPlane=0; iPlane<fNPlanes; iPlane++) {
    ResetDigits(iPlane);
  }

}

//====================================================================================================================================================

void AliMFTReconstructor::ResetDigits(Int_t plane) {

  // Reset number of digits and the digits array for this branch.

  if (fDigits->At(plane)) ((TClonesArray*)fDigits->At(plane))->Clear();

}

//====================================================================================================================================================

void AliMFTReconstructor::Reconstruct(TTree *digitsTree, TTree *clustersTree) const {

  AliInfo("Starting Reconstruction for MFT");

  // Clusterization

  AliDebug(1, Form("nPlanes = %d",fNPlanes));

  for (Int_t iPlane=0; iPlane<fNPlanes; iPlane++) {
    AliDebug(1, Form("Setting Address for Branch Plane_%02d", iPlane)); 
    digitsTree->SetBranchAddress(Form("Plane_%02d",iPlane), &(*fDigits)[iPlane]);
  }
 
  digitsTree->GetEntry(0);

  AliDebug(1, "Creating clusterFinder");
  AliMFTClusterFinder *clusterFinder = new AliMFTClusterFinder();
  clusterFinder->Init("AliMFTGeometry.root");
  AliDebug(1, "clusterFinder->MakeClusterBranch(clustersTree)");
  clusterFinder->MakeClusterBranch(clustersTree);
  AliDebug(1, "clusterFinder->SetClusterTreeAddress(clustersTree)");
  clusterFinder->SetClusterTreeAddress(clustersTree);
  AliDebug(1, "clusterFinder->DigitsToClusters(fDigits)");
  clusterFinder->DigitsToClusters(fDigits);
  AliDebug(1, "clustersTree->Fill()");
  clustersTree->Fill();                         // fill tree for current event
  AliDebug(1, "delete clusterFinder");
  delete clusterFinder;

  for (Int_t iPlane=0; iPlane<fNPlanes; iPlane++) {
    AliDebug(1, Form("fDigits->At(%d)->Clear()",iPlane));
    ((TClonesArray*)fDigits->At(iPlane))->Delete();
  }

}

//====================================================================================================================================================
Commit	Line	Data
820b4d9e	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	//====================================================================================================================================================
	17	//
	18	// Event reconstruction class for the ALICE Muon Forward Tracker
	19	//
	20	// Contact author: antonio.uras@cern.ch
	21	//
	22	//====================================================================================================================================================
	23
	24	#include "TObjArray.h"
	25	#include "TTree.h"
	26	#include "AliMFTSegmentation.h"
	27	#include "AliMFTClusterFinder.h"
	28	#include "AliReconstructor.h"
	29	#include "AliMFTReconstructor.h"
	30
	31	ClassImp(AliMFTReconstructor)
	32
	33	//====================================================================================================================================================
	34
	35	AliMFTReconstructor::AliMFTReconstructor():
	36	AliReconstructor(),
	37	fDigits(0x0),
	38	fNPlanes(0)
	39	{
	40
	41	// default constructor
	42
	43	}
	44
	45	//====================================================================================================================================================
	46
ffc53def	47	AliMFTReconstructor::~AliMFTReconstructor() {
ffc53def	48
820b4d9e	49	// destructor
ffc53def	50
820b4d9e	51	if (fDigits) {
ffc53def	52	for (Int_t iPlane=0; iPlane<fNPlanes; iPlane++) {
	53	if (fDigits->At(iPlane)) fDigits->At(iPlane)->Delete();
	54	}
820b4d9e	55	fDigits->Delete();
	56	delete fDigits;
	57	fDigits=0;
	58	}
ffc53def	59
ffc53def	60	}
820b4d9e	61
ffc53def	62	//====================================================================================================================================================
	63
	64	void AliMFTReconstructor::Clear(const Option_t* /opt/) {
	65
	66	// Clear arrays
	67
	68	if (fDigits) {
	69	for (Int_t iPlane=0; iPlane<fNPlanes; iPlane++) {
	70	if (fDigits->At(iPlane)) ((TClonesArray*)fDigits->At(iPlane))->Delete();
	71	}
	72	fDigits->Delete();
	73	delete fDigits;
	74	fDigits = NULL;
	75	}
	76
820b4d9e	77	}
	78
	79	//====================================================================================================================================================
	80
	81	void AliMFTReconstructor::Init() {
	82
	83	AliMFTSegmentation *segmentation = new AliMFTSegmentation("AliMFTGeometry.root");
	84	fNPlanes = segmentation->GetNPlanes();
	85	delete segmentation;
	86
	87	fDigits = new TObjArray(fNPlanes);
	88	fDigits->SetOwner(kTRUE);
ffc53def	89	for (Int_t iPlane=0; iPlane<fNPlanes; iPlane++) {
	90	fDigits->AddAt(new TClonesArray("AliMFTDigit"),iPlane);
	91	((TClonesArray*)fDigits->At(iPlane))->SetOwner(kTRUE);
	92	}
	93
820b4d9e	94	AliInfo(" *********** Using the MFT reconstructor! **** ");
ffc53def	95
820b4d9e	96	return;
	97
	98	}
	99
	100	//====================================================================================================================================================
	101
	102	void AliMFTReconstructor::ResetDigits() {
	103
	104	// Reset number of digits and the digits array for the MFT detector.
	105
	106	if (!fDigits) return;
	107	for (Int_t iPlane=0; iPlane<fNPlanes; iPlane++) {
	108	ResetDigits(iPlane);
	109	}
	110
	111	}
	112
	113	//====================================================================================================================================================
	114
	115	void AliMFTReconstructor::ResetDigits(Int_t plane) {
	116
	117	// Reset number of digits and the digits array for this branch.
	118
	119	if (fDigits->At(plane)) ((TClonesArray*)fDigits->At(plane))->Clear();
	120
	121	}
	122
	123	//====================================================================================================================================================
	124
	125	void AliMFTReconstructor::Reconstruct(TTree digitsTree, TTree clustersTree) const {
	126
	127	AliInfo("Starting Reconstruction for MFT");
	128
	129	// Clusterization
	130
	131	AliDebug(1, Form("nPlanes = %d",fNPlanes));
	132
	133	for (Int_t iPlane=0; iPlane<fNPlanes; iPlane++) {
	134	AliDebug(1, Form("Setting Address for Branch Plane_%02d", iPlane));
	135	digitsTree->SetBranchAddress(Form("Plane_%02d",iPlane), &(*fDigits)[iPlane]);
	136	}
	137
	138	digitsTree->GetEntry(0);
	139
	140	AliDebug(1, "Creating clusterFinder");
	141	AliMFTClusterFinder *clusterFinder = new AliMFTClusterFinder();
	142	clusterFinder->Init("AliMFTGeometry.root");
	143	AliDebug(1, "clusterFinder->MakeClusterBranch(clustersTree)");
	144	clusterFinder->MakeClusterBranch(clustersTree);
	145	AliDebug(1, "clusterFinder->SetClusterTreeAddress(clustersTree)");
	146	clusterFinder->SetClusterTreeAddress(clustersTree);
	147	AliDebug(1, "clusterFinder->DigitsToClusters(fDigits)");
	148	clusterFinder->DigitsToClusters(fDigits);
	149	AliDebug(1, "clustersTree->Fill()");
	150	clustersTree->Fill(); // fill tree for current event
	151	AliDebug(1, "delete clusterFinder");
	152	delete clusterFinder;
	153
	154	for (Int_t iPlane=0; iPlane<fNPlanes; iPlane++) {
	155	AliDebug(1, Form("fDigits->At(%d)->Clear()",iPlane));
ffc53def	156	((TClonesArray*)fDigits->At(iPlane))->Delete();
820b4d9e	157	}
	158
	159	}
	160
	161	//====================================================================================================================================================