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

//====================================================================================================================================================
//
//      Segmentation class for the planes of the ALICE Muon Forward Tracker
//
//      Contact author: antonio.uras@cern.ch
//
//====================================================================================================================================================

#include "TFile.h"
#include "TNtuple.h"
#include "TClonesArray.h"
#include "TMath.h"
#include "AliMFTPlane.h"
#include "AliMFTSegmentation.h"

ClassImp(AliMFTSegmentation)

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

AliMFTSegmentation::AliMFTSegmentation(): 
  TObject(),
  fMFTPlanes(0)
{ 

  // default constructor

}

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

AliMFTSegmentation::AliMFTSegmentation(const Char_t *nameGeomFile): 
  TObject(),
  fMFTPlanes(new TClonesArray("AliMFTPlane", fNMaxPlanes))
{ 

  // constructor

  Float_t zCenter, rMin, rMax, pixelSizeX, pixelSizeY, thicknessActive, thicknessSupport, thicknessReadout;
  Float_t equivalentSilicon, equivalentSiliconBeforeFront, equivalentSiliconBeforeBack, hasPixelRectangularPatternAlongY;

  TFile *geomFile = new TFile(nameGeomFile);
  TNtuple *geomNtuple = (TNtuple*) geomFile->Get("AliMFTGeometry");

  geomNtuple -> SetBranchAddress("zCenter", &zCenter);
  geomNtuple -> SetBranchAddress("rMin",    &rMin);
  geomNtuple -> SetBranchAddress("rMax",    &rMax);
  geomNtuple -> SetBranchAddress("pixelSizeX", &pixelSizeX);
  geomNtuple -> SetBranchAddress("pixelSizeY", &pixelSizeY);
  geomNtuple -> SetBranchAddress("thicknessActive",  &thicknessActive);
  geomNtuple -> SetBranchAddress("thicknessSupport", &thicknessSupport);
  geomNtuple -> SetBranchAddress("thicknessReadout", &thicknessReadout);
  geomNtuple -> SetBranchAddress("equivalentSilicon",            &equivalentSilicon);
  geomNtuple -> SetBranchAddress("equivalentSiliconBeforeFront", &equivalentSiliconBeforeFront);
  geomNtuple -> SetBranchAddress("equivalentSiliconBeforeBack",  &equivalentSiliconBeforeBack);
  if (geomNtuple -> GetBranch("hasPixelRectangularPatternAlongY")) {
    geomNtuple -> SetBranchAddress("hasPixelRectangularPatternAlongY", &hasPixelRectangularPatternAlongY);
  }
  else hasPixelRectangularPatternAlongY = 0.;
  
  Int_t nPlanes = geomNtuple->GetEntries();

  for (Int_t iPlane=0; iPlane<nPlanes; iPlane++) {

    // Create new plane

    printf("Setting segmentation for MFT plane #%02d\n", iPlane);

    geomNtuple -> GetEntry(iPlane);
    zCenter = TMath::Abs(zCenter);

    AliMFTPlane *plane = new AliMFTPlane(Form("MFTPlane_%02d", iPlane), Form("MFTPlane_%02d", iPlane));

    plane -> Init(iPlane, 
		  zCenter, 
		  rMin, 
		  rMax, 
		  pixelSizeX, 
		  pixelSizeY, 
		  thicknessActive, 
		  thicknessSupport, 
		  thicknessReadout, 
		  (hasPixelRectangularPatternAlongY>0.5));

    plane -> SetEquivalentSilicon(equivalentSilicon);
    plane -> SetEquivalentSiliconBeforeFront(equivalentSiliconBeforeFront);
    plane -> SetEquivalentSiliconBeforeBack(equivalentSiliconBeforeBack);
    plane -> CreateStructure();
    
    new ((*fMFTPlanes)[fMFTPlanes->GetEntries()]) AliMFTPlane(*plane);

  }
  
  delete geomFile;

  printf("MFT segmentation set!\n");

}

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

THnSparseC* AliMFTSegmentation::GetDetElem(Int_t detElemID) const {
      
  // Find det elem

  Int_t planeNb = detElemID/fNMaxDetElemPerPlane;
  Int_t detElemNb = detElemID - planeNb*fNMaxDetElemPerPlane;
  
  THnSparseC *detElem = GetPlane(planeNb)->GetActiveElement(detElemNb);

  return detElem;

}

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

Bool_t AliMFTSegmentation::Hit2PixelID(Double_t xHit, Double_t yHit, Int_t detElemID, Int_t &xPixel, Int_t &yPixel) {

  // xPixel and yPixel start from 0

  THnSparseC *detElem = GetDetElem(detElemID);

  if ( xHit<detElem->GetAxis(0)->GetXmin() ||
       xHit>detElem->GetAxis(0)->GetXmax() ||
       yHit<detElem->GetAxis(1)->GetXmin() ||
       yHit>detElem->GetAxis(1)->GetXmax() ) return kFALSE;

  xPixel = detElem->GetAxis(0)->FindBin(xHit) - 1;
  yPixel = detElem->GetAxis(1)->FindBin(yHit) - 1;

  return kTRUE;

}

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

Bool_t AliMFTSegmentation::DoesPixelExist(Int_t detElemID, Int_t xPixel, Int_t yPixel) {

  THnSparseC *detElem = GetDetElem(detElemID);

  if (xPixel>=0 && xPixel<detElem->GetAxis(0)->GetNbins() && yPixel>=0 && yPixel<detElem->GetAxis(1)->GetNbins()) return kTRUE;
  else return kFALSE;

}

//====================================================================================================================================================
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	// Segmentation class for the planes of the ALICE Muon Forward Tracker
	19	//
	20	// Contact author: antonio.uras@cern.ch
	21	//
	22	//====================================================================================================================================================
	23
	24	#include "TFile.h"
	25	#include "TNtuple.h"
	26	#include "TClonesArray.h"
	27	#include "TMath.h"
	28	#include "AliMFTPlane.h"
	29	#include "AliMFTSegmentation.h"
	30
	31	ClassImp(AliMFTSegmentation)
	32
	33	//====================================================================================================================================================
	34
	35	AliMFTSegmentation::AliMFTSegmentation():
	36	TObject(),
	37	fMFTPlanes(0)
	38	{
	39
820b4d9e	40	// default constructor
820b4d9e	41
820b4d9e	42	}
	43
	44	//====================================================================================================================================================
	45
	46	AliMFTSegmentation::AliMFTSegmentation(const Char_t *nameGeomFile):
	47	TObject(),
d4643a10	48	fMFTPlanes(new TClonesArray("AliMFTPlane", fNMaxPlanes))
820b4d9e	49	{
820b4d9e	50
d4643a10	51	// constructor
820b4d9e	52
820b4d9e	53	Float_t zCenter, rMin, rMax, pixelSizeX, pixelSizeY, thicknessActive, thicknessSupport, thicknessReadout;
bcaf50eb	54	Float_t equivalentSilicon, equivalentSiliconBeforeFront, equivalentSiliconBeforeBack, hasPixelRectangularPatternAlongY;
820b4d9e	55
	56	TFile *geomFile = new TFile(nameGeomFile);
	57	TNtuple geomNtuple = (TNtuple) geomFile->Get("AliMFTGeometry");
	58
	59	geomNtuple -> SetBranchAddress("zCenter", &zCenter);
	60	geomNtuple -> SetBranchAddress("rMin", &rMin);
	61	geomNtuple -> SetBranchAddress("rMax", &rMax);
	62	geomNtuple -> SetBranchAddress("pixelSizeX", &pixelSizeX);
	63	geomNtuple -> SetBranchAddress("pixelSizeY", &pixelSizeY);
	64	geomNtuple -> SetBranchAddress("thicknessActive", &thicknessActive);
	65	geomNtuple -> SetBranchAddress("thicknessSupport", &thicknessSupport);
	66	geomNtuple -> SetBranchAddress("thicknessReadout", &thicknessReadout);
	67	geomNtuple -> SetBranchAddress("equivalentSilicon", &equivalentSilicon);
	68	geomNtuple -> SetBranchAddress("equivalentSiliconBeforeFront", &equivalentSiliconBeforeFront);
	69	geomNtuple -> SetBranchAddress("equivalentSiliconBeforeBack", &equivalentSiliconBeforeBack);
bcaf50eb	70	if (geomNtuple -> GetBranch("hasPixelRectangularPatternAlongY")) {
	71	geomNtuple -> SetBranchAddress("hasPixelRectangularPatternAlongY", &hasPixelRectangularPatternAlongY);
	72	}
	73	else hasPixelRectangularPatternAlongY = 0.;
820b4d9e	74
	75	Int_t nPlanes = geomNtuple->GetEntries();
	76
	77	for (Int_t iPlane=0; iPlane<nPlanes; iPlane++) {
	78
	79	// Create new plane
	80
	81	printf("Setting segmentation for MFT plane #%02d\n", iPlane);
	82
	83	geomNtuple -> GetEntry(iPlane);
	84	zCenter = TMath::Abs(zCenter);
	85
	86	AliMFTPlane *plane = new AliMFTPlane(Form("MFTPlane_%02d", iPlane), Form("MFTPlane_%02d", iPlane));
bcaf50eb	87
	88	plane -> Init(iPlane,
	89	zCenter,
	90	rMin,
	91	rMax,
	92	pixelSizeX,
	93	pixelSizeY,
	94	thicknessActive,
	95	thicknessSupport,
	96	thicknessReadout,
	97	(hasPixelRectangularPatternAlongY>0.5));
	98
820b4d9e	99	plane -> SetEquivalentSilicon(equivalentSilicon);
	100	plane -> SetEquivalentSiliconBeforeFront(equivalentSiliconBeforeFront);
	101	plane -> SetEquivalentSiliconBeforeBack(equivalentSiliconBeforeBack);
	102	plane -> CreateStructure();
	103
	104	new ((fMFTPlanes)[fMFTPlanes->GetEntries()]) AliMFTPlane(plane);
	105
	106	}
	107
	108	delete geomFile;
	109
	110	printf("MFT segmentation set!\n");
	111
	112	}
	113
	114	//====================================================================================================================================================
	115
d4643a10	116	THnSparseC* AliMFTSegmentation::GetDetElem(Int_t detElemID) const {
820b4d9e	117
	118	// Find det elem
	119
	120	Int_t planeNb = detElemID/fNMaxDetElemPerPlane;
	121	Int_t detElemNb = detElemID - planeNb*fNMaxDetElemPerPlane;
	122
	123	THnSparseC *detElem = GetPlane(planeNb)->GetActiveElement(detElemNb);
	124
	125	return detElem;
	126
	127	}
	128
	129	//====================================================================================================================================================
	130
	131	Bool_t AliMFTSegmentation::Hit2PixelID(Double_t xHit, Double_t yHit, Int_t detElemID, Int_t &xPixel, Int_t &yPixel) {
	132
bcaf50eb	133	// xPixel and yPixel start from 0
bcaf50eb	134
820b4d9e	135	THnSparseC *detElem = GetDetElem(detElemID);
	136
	137	if ( xHit<detElem->GetAxis(0)->GetXmin() \|\|
	138	xHit>detElem->GetAxis(0)->GetXmax() \|\|
	139	yHit<detElem->GetAxis(1)->GetXmin() \|\|
	140	yHit>detElem->GetAxis(1)->GetXmax() ) return kFALSE;
	141
	142	xPixel = detElem->GetAxis(0)->FindBin(xHit) - 1;
	143	yPixel = detElem->GetAxis(1)->FindBin(yHit) - 1;
	144
	145	return kTRUE;
	146
	147	}
	148
	149	//====================================================================================================================================================
	150
bcaf50eb	151	Bool_t AliMFTSegmentation::DoesPixelExist(Int_t detElemID, Int_t xPixel, Int_t yPixel) {
	152
	153	THnSparseC *detElem = GetDetElem(detElemID);
	154
	155	if (xPixel>=0 && xPixel<detElem->GetAxis(0)->GetNbins() && yPixel>=0 && yPixel<detElem->GetAxis(1)->GetNbins()) return kTRUE;
	156	else return kFALSE;
	157
	158	}
	159
	160	//====================================================================================================================================================
	161