[u/mrichter/AliRoot.git] / MUON / AliMUONSegmentationV01.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$ */

/////////////////////////////////////////////////////
//  Segmentation and Response classes version 01   //
/////////////////////////////////////////////////////

#include <TBox.h> 
#include <TTUBE.h>
#include <TBRIK.h>
#include <TNode.h>  
#include <TGeometry.h>  
#include <TF1.h> 
#include <TObjArray.h>
#include <Riostream.h>

#include "AliMUONSegmentationV01.h"
#include "AliMUON.h"
#include "AliMUONChamber.h"
#include "AliRun.h"


//___________________________________________
ClassImp(AliMUONSegmentationV01)

AliMUONSegmentationV01::AliMUONSegmentationV01(const AliMUONSegmentationV01& segmentation)
{
// Dummy copy constructor
}

AliMUONSegmentationV01::AliMUONSegmentationV01() 
{
// Default constructor
    fRSec = 0;
    fNDiv = 0;      
    fDpxD = 0;
    fCorrA = 0;
}

AliMUONSegmentationV01::AliMUONSegmentationV01(Int_t nsec) 
{
//  Non default constructor

    fNsec = nsec;
    fRSec = new TArrayF(fNsec);
    fNDiv = new TArrayI(fNsec);      
    fDpxD = new TArrayF(fNsec);      


    (*fRSec)[0]=(*fRSec)[1]=(*fRSec)[2]=(*fRSec)[3]=0;     
    (*fNDiv)[0]=(*fNDiv)[1]=(*fNDiv)[2]=(*fNDiv)[3]=0;     
    (*fDpxD)[0]=(*fDpxD)[1]=(*fDpxD)[2]=(*fDpxD)[3]=0;     
    fCorrA = new TObjArray(3);
    fCorrA->AddAt(0,0);
    fCorrA->AddAt(0,1);
    fCorrA->AddAt(0,2);
    fOffsetY=0;
} 

AliMUONSegmentationV01::~AliMUONSegmentationV01() 
{
// Destructor
    if (fRSec) delete fRSec;
    if (fNDiv) delete fNDiv;
    if (fDpxD) delete fDpxD;
    if (fCorrA) {
	fCorrA->Delete();
	delete fCorrA;
    }
} 


Float_t AliMUONSegmentationV01::Dpx(Int_t isec) const
{
//
// Returns x-pad size for given sector isec
   Float_t dpx = (*fDpxD)[isec];
   return dpx;
}

Float_t AliMUONSegmentationV01::Dpy(Int_t isec) const
{
//
// Returns y-pad size for given sector isec
   return fDpy;
}
    
void   AliMUONSegmentationV01::SetSegRadii(Float_t  r[4])
{
//
// Set the radii of the segmentation zones 
    for (Int_t i=0; i<4; i++) {
	(*fRSec)[i]=r[i];
    }
}


void AliMUONSegmentationV01::SetPadDivision(Int_t ndiv[4])
{
//
// Defines the pad size perp. to the anode wire (y) for different sectors. 
// Pad sizes are defined as integral fractions ndiv of a basis pad size
// fDpx
// 
    for (Int_t i=0; i<4; i++) {
	(*fNDiv)[i]=ndiv[i];
    }
    ndiv[0]=ndiv[1];
}


void AliMUONSegmentationV01::Init(Int_t chamber)
{
//
//  Fill the arrays fCx (x-contour) and fNpxS (ix-contour) for each sector
//  These arrays help in converting from real to pad co-ordinates and
//  vice versa.
//  This version approximates concentric segmentation zones
//
    Int_t isec;
    //printf("\n Initialise Segmentation V01\n");


    fNpy=Int_t((*fRSec)[fNsec-1]/fDpy)+1;

    (*fDpxD)[fNsec-1]=fDpx;
    if (fNsec > 1) {
	for (Int_t i=fNsec-2; i>=0; i--){
	    (*fDpxD)[i]=(*fDpxD)[fNsec-1]/(*fNDiv)[i];
	}
    }
//
// fill the arrays defining the pad segmentation boundaries
    Float_t ry;
    Int_t   dnx;
    Int_t   add;
//
//  loop over sections
    for(isec=0; isec<fNsec; isec++) {
//  
//  loop over pads along the aode wires
	for (Int_t iy=1; iy<=fNpy; iy++) {
//
	    Float_t x=iy*fDpy-fDpy/2;
	    if (x > (*fRSec)[isec]) {
		fNpxS[isec][iy]=0;
		fCx[isec][iy]=0;
	    } else {
		ry=TMath::Sqrt((*fRSec)[isec]*(*fRSec)[isec]-x*x);
		if (isec > 1) {
		    dnx= Int_t((ry-fCx[isec-1][iy])/(*fDpxD)[isec]);
		    if (isec < fNsec-1) {
			if (TMath::Odd((Long_t)dnx)) dnx++;    		
		    }
                    fNpxS[isec][iy]=fNpxS[isec-1][iy]+dnx;
          	    fCx[isec][iy]=fCx[isec-1][iy]+dnx*(*fDpxD)[isec];
		} else if (isec == 1) {
		    dnx= Int_t((ry-fCx[isec-1][iy])/(*fDpxD)[isec]);
		    fNpxS[isec][iy]=fNpxS[isec-1][iy]+dnx;
                    add=4 - (fNpxS[isec][iy])%4;
                    if (add < 4) fNpxS[isec][iy]+=add; 
		    dnx=fNpxS[isec][iy]-fNpxS[isec-1][iy];
		    fCx[isec][iy]=fCx[isec-1][iy]+dnx*(*fDpxD)[isec];
		} else {
		    dnx=Int_t(ry/(*fDpxD)[isec]);
                    fNpxS[isec][iy]=dnx;
		    fCx[isec][iy]=dnx*(*fDpxD)[isec];
		}
	    }
	} // y-pad loop
    } // sector loop
// reference to chamber
    AliMUON *pMUON  = (AliMUON *) gAlice->GetModule("MUON");
    fChamber=&(pMUON->Chamber(chamber));
    fZ = fChamber->Z();
    fId=chamber;
}

Int_t AliMUONSegmentationV01::Sector(Int_t ix, Int_t iy)
{
// Returns sector number for given pad position
//
    Int_t absix=TMath::Abs(ix);
    Int_t absiy=TMath::Abs(iy);
    Int_t isec=0;
    for (Int_t i=0; i<fNsec; i++) {
	if (absix<=fNpxS[i][absiy]){
	    isec=i;
	    break;
	}
    }
    return isec;
}

void AliMUONSegmentationV01::
GetPadI(Float_t x, Float_t y, Int_t &ix, Int_t &iy)
{
//  Returns pad coordinates (ix,iy) for given real coordinates (x,y)
//
    iy = (y-fOffsetY >0)? 
      Int_t((y-fOffsetY)/fDpy)+1 
      : 
      Int_t((y-fOffsetY)/fDpy)-1;
  
    if (iy >  fNpy) iy= fNpy;
    if (iy < -fNpy) iy=-fNpy;
//
//  Find sector isec
    Int_t isec=-1;
    Float_t absx=TMath::Abs(x);
    Int_t absiy=TMath::Abs(iy);
    for (Int_t i=0; i < fNsec; i++) {
	if (absx <= fCx[i][absiy]) {
	    isec=i;
	    break;
	}
    }
    if (isec>0) {
	ix= Int_t((absx-fCx[isec-1][absiy])/(*fDpxD)[isec])
	    +fNpxS[isec-1][absiy]+1;
    } else if (isec == 0) {
	ix= Int_t(absx/(*fDpxD)[isec])+1;
    } else {
	ix=fNpxS[fNsec-1][absiy]+1;	
    }
    ix = (x>0) ? ix:-ix;
}

void AliMUONSegmentationV01::
GetPadC(Int_t ix, Int_t iy, Float_t &x, Float_t &y)
{
//  Returns real coordinates (x,y) for given pad coordinates (ix,iy)
//
    y = (iy>0) ?
      Float_t(iy*fDpy)-fDpy/2.+fOffsetY
      :
      Float_t(iy*fDpy)+fDpy/2.+fOffsetY; 

//
//  Find sector isec
    Int_t isec=AliMUONSegmentationV01::Sector(ix,iy);
//
    Int_t absix=TMath::Abs(ix);
    Int_t absiy=TMath::Abs(iy);
    if (isec) {
	x=fCx[isec-1][absiy]+(absix-fNpxS[isec-1][absiy])*(*fDpxD)[isec];
	x=(ix>0) ?  x-(*fDpxD)[isec]/2 : -x+(*fDpxD)[isec]/2;
    } else {
	x=y=0;
    }
}

void AliMUONSegmentationV01::
SetPad(Int_t ix, Int_t iy)
{
    //
    // Sets virtual pad coordinates, needed for evaluating pad response 
    // outside the tracking program 
    GetPadC(ix,iy,fX,fY);
    fSector=Sector(ix,iy);
}


void AliMUONSegmentationV01::
FirstPad(Float_t xhit, Float_t yhit, Float_t dx, Float_t dy)
{
// Initialises iteration over pads for charge distribution algorithm
//
    //
    // Find the wire position (center of charge distribution)
    Float_t x0a=GetAnod(xhit);
    fXhit=x0a;
    fYhit=yhit;
    
    //
    // and take fNsigma*sigma around this center
    Float_t x01=x0a  - dx;
    Float_t x02=x0a  + dx;
    Float_t y01=yhit - dy;
    Float_t y02=yhit + dy;
    //
    // find the pads over which the charge distributes

    GetPadI(x01,y01,fIxmin,fIymin);
    GetPadI(x02,y02,fIxmax,fIymax);
    fXmin=x01;
    fXmax=x02;
    fYmin=y01;
    fYmax=y02;
    
    // 
    // Set current pad to lower left corner
    if (fIxmax < fIxmin) fIxmax=fIxmin;
    if (fIymax < fIymin) fIymax=fIymin;    
    fIx=fIxmin;
    fIy=fIymin;
    GetPadC(fIx,fIy,fX,fY);
}


void AliMUONSegmentationV01::NextPad()
{
// Stepper for the iteration over pads
//
// Step to next pad in the integration region
  // 
  // Step to next pad in integration region
    Float_t xc,yc;
    Int_t   iyc;
    
//  step from left to right    

    if (fX < fXmax && fX != 0) {
	if (fIx==-1) fIx++;
	fIx++;
//  step up 
    } else if (fIy != fIymax) {
	if (fIy==-1) fIy++;
	fIy++;
//      get y-position of next row (yc), xc not used here 	
	GetPadC(fIx,fIy,xc,yc);
//      get x-pad coordiante for first pad in row (fIx)
	GetPadI(fXmin,yc,fIx,iyc);
    } else {
	fIx=-1;
	fIy=-1;
    }
    GetPadC(fIx,fIy,fX,fY);
    fSector=Sector(fIx,fIy);
    if (MorePads() && 
	(fSector ==-1 || fSector==0)) 
	NextPad();
}

Int_t AliMUONSegmentationV01::MorePads()

{
// Stopping condition for the iterator over pads
//
// Are there more pads in the integration region
    return  (fIx != -1  || fIy != -1);
/*
    if ((fX >= fXmax  && fIy >= fIymax) || fY==0) {
	return 0;
    } else {
	return 1;
    }
*/
}

void AliMUONSegmentationV01::
IntegrationLimits(Float_t& x1,Float_t& x2,Float_t& y1, Float_t& y2)
{
//  Returns integration limits for current pad
//
    x1=fXhit-fX-Dpx(fSector)/2.;
    x2=x1+Dpx(fSector);
    y1=fYhit-fY-Dpy(fSector)/2.;
    y2=y1+Dpy(fSector);    
}

void AliMUONSegmentationV01::
Neighbours(Int_t iX, Int_t iY, Int_t* Nlist, Int_t Xlist[10], Int_t Ylist[10])
{
// Returns list of next neighbours for given Pad (iX, iY)
//
    const Float_t kEpsilon=fDpy/1000;
    
    Float_t x,y;
    Int_t   ixx, iyy, isec1;
//
    Int_t   isec0=AliMUONSegmentationV01::Sector(iX,iY);
    Int_t i=0;
//    
//  step right
    Xlist[i]=iX+1;
    if (Xlist[i]==0) Xlist[i]++;
    Ylist[i++]=iY;
//
//  step left    
    Xlist[i]=iX-1;
    if (Xlist[i]==0) Xlist[i]--;
    Ylist[i++]=iY;
//
//  step up 
    AliMUONSegmentationV01::GetPadC(iX,iY,x,y);
    AliMUONSegmentationV01::GetPadI(x+kEpsilon,y+fDpy,ixx,iyy);
    Xlist[i]=ixx;
    Ylist[i++]=iyy;
    isec1=AliMUONSegmentationV01::Sector(ixx,iyy);
    if (isec1==isec0) {
//
//  no sector boundary crossing
//	Xlist[i]=ixx+1;
//	Ylist[i++]=iY+1;
	
//	Xlist[i]=ixx-1;
//	Ylist[i++]=iY+1;
    } else if (isec1 < isec0) {
// finer segmentation
//	Xlist[i]=ixx+1;
//	Ylist[i++]=iY+1;
	
	Xlist[i]=ixx-1;
	Ylist[i++]=iyy;
	
//	Xlist[i]=ixx-2;
//	Ylist[i++]=iY+1;
    } else {
// coarser segmenation	
/*
	if (TMath::Odd(iX-fNpxS[isec1-1][iY+1])) {
	    Xlist[i]=ixx-1;
	    Ylist[i++]=iY+1;
	} else {
	    Xlist[i]=ixx+1;
	    Ylist[i++]=iY+1;
	}
*/
    }

//
// step down 
    AliMUONSegmentationV01::GetPadC(iX,iY,x,y);
    AliMUONSegmentationV01::GetPadI(x+kEpsilon,y-fDpy,ixx,iyy);
    Xlist[i]=ixx;
    Ylist[i++]=iyy;
    isec1=AliMUONSegmentationV01::Sector(ixx,iyy);
    if (isec1==isec0) {
//
//  no sector boundary crossing
/*
    Xlist[i]=ixx+1;
    Ylist[i++]=iY-1;
	
    Xlist[i]=ixx-1;
    Ylist[i++]=iY-1;
*/
    } else if (isec1 < isec0) {
// finer segmentation
//	Xlist[i]=ixx+1;
//	Ylist[i++]=iY-1;
	
	Xlist[i]=ixx-1;
	Ylist[i++]=iyy;
	
//	Xlist[i]=ixx-2;
//	Ylist[i++]=iY-1;
    } else {
// coarser segmentation	
/*
	if (TMath::Odd(iX-fNpxS[isec1-1][iY-1])) {
	    Xlist[i]=ixx-1;
	    Ylist[i++]=iY-1;
	} else {
	    Xlist[i]=ixx+1;
	    Ylist[i++]=iY-1;
	}
*/
    }
    *Nlist=i;
}

void AliMUONSegmentationV01::GiveTestPoints(Int_t &n, Float_t *x, Float_t *y) const
{
// Returns test point on the pad plane.
// Used during determination of the segmoid correction of the COG-method

    n=3;
    x[0]=((*fRSec)[0]+(*fRSec)[1])/2/TMath::Sqrt(2.);
    y[0]=x[0];
    x[1]=((*fRSec)[1]+(*fRSec)[2])/2/TMath::Sqrt(2.);
    y[1]=x[1];
    x[2]=((*fRSec)[2]+(*fRSec)[3])/2/TMath::Sqrt(2.);
    y[2]=x[2];
}

void AliMUONSegmentationV01::Draw(const char* opt) const
{
 
// Draws the segmentation zones
//
  if (!strcmp(opt,"eventdisplay")) { 
    const int kColorMUON  = kBlue;

    TRotMatrix* rot000 = new TRotMatrix("Rot000"," ", 90,  0, 90, 90, 0, 0);
    TRotMatrix* rot090 = new TRotMatrix("Rot090"," ", 90, 90, 90,180, 0, 0);
    TRotMatrix* rot180 = new TRotMatrix("Rot180"," ", 90,180, 90,270, 0, 0);
    TRotMatrix* rot270 = new TRotMatrix("Rot270"," ", 90,270, 90,  0, 0, 0);

    char nameChamber[9], nameSense[9], nameFrame[9], nameNode[9];
    char nameSense1[9], nameSense2[9];    
    TNode *node, *nodeF;
 
    sprintf(nameChamber,"C_MUON%d",fId+1);
    sprintf(nameSense,"S_MUON%d",fId+1);
    sprintf(nameSense1,"S1_MUON%d",fId+1);
    sprintf(nameSense2,"S2_MUON%d",fId+1);
    sprintf(nameFrame,"F_MUON%d",fId+1);	

    TNode* top=gAlice->GetGeometry()->GetNode("alice");

    Float_t rmin = (*fRSec)[0]-3;
    Float_t rmax = (*fRSec)[3]+3;
    new TTUBE(nameChamber,"Mother","void",rmin,rmax,0.25,1.);
    rmin = (*fRSec)[0];
    rmax = (*fRSec)[3];
    new TTUBE(nameSense,"Sens. region","void",rmin,rmax,0.25, 1.);
    Float_t dx=(rmax-rmin)/2;
    Float_t dy=3.;
    Float_t dz=0.25;
    TBRIK* frMUON = new TBRIK(nameFrame,"Frame","void",dx,dy,dz);
    top->cd();
    sprintf(nameNode,"MUON%d",100+fId+1);
    node = new TNode(nameNode,"ChamberNode",nameChamber,0,0,fChamber->Z(),"");
    node->SetLineColor(kColorMUON);
    AliMUON *pMUON  = (AliMUON *) gAlice->GetModule("MUON");
    (pMUON->Nodes())->Add(node);
    node->cd();
    sprintf(nameNode,"MUON%d",200+fId+1);
    node = new TNode(nameNode,"Sens. Region Node",nameSense,0,0,0,"");
    node->SetLineColor(kColorMUON);
    node->cd();
    Float_t dr=dx+rmin;
    sprintf(nameNode,"MUON%d",300+fId+1);
    nodeF = new TNode(nameNode,"Frame0",frMUON,dr, 0, 0,rot000,"");
    nodeF->SetLineColor(kColorMUON);
    node->cd();
    sprintf(nameNode,"MUON%d",400+fId+1);
    nodeF = new TNode(nameNode,"Frame1",frMUON,0 ,dr,0,rot090,"");
    nodeF->SetLineColor(kColorMUON);
    node->cd();
    sprintf(nameNode,"MUON%d",500+fId+1);
    nodeF = new TNode(nameNode,"Frame2",frMUON,-dr,0,0,rot180,"");
    nodeF->SetLineColor(kColorMUON);
    node  ->cd();
    sprintf(nameNode,"MUON%d",600+fId+1);   
    nodeF = new TNode(nameNode,"Frame3",frMUON,0,-dr,0,rot270,"");
    nodeF->SetLineColor(kColorMUON);   
  } else {
    TBox *box;
    
    Float_t dx=0.95/fCx[3][1]/2;
    Float_t dy=0.95/(Float_t(Npy()))/2;
    Float_t x0,y0,x1,y1;
    Float_t xc=0.5;
    Float_t yc=0.5;
    
    for (Int_t iy=1; iy<Npy(); iy++) {
      for (Int_t isec=0; isec<4; isec++) {
	if (isec==0) {
	  x0=0;
	  x1=fCx[isec][iy]*dx;
	} else {
	  x0=fCx[isec-1][iy]*dx;
	  x1=fCx[isec][iy]*dx;
	}
	y0=Float_t(iy-1)*dy;
	y1=y0+dy;
	box=new TBox(x0+xc,y0+yc,x1+xc,y1+yc);
	box->SetFillColor(isec+1);
	box->Draw();
	
	box=new TBox(-x1+xc,y0+yc,-x0+xc,y1+yc);
	box->SetFillColor(isec+1);
	box->Draw();
	
	box=new TBox(x0+xc,-y1+yc,x1+xc,-y0+yc);
	box->SetFillColor(isec+1);
	box->Draw();
	
	box=new TBox(-x1+xc,-y1+yc,-x0+xc,-y0+yc);
	box->SetFillColor(isec+1);
	box->Draw();
      }
    }
  }
}
void AliMUONSegmentationV01::SetCorrFunc(Int_t isec, TF1* func)
{
// Set the correction function
    fCorrA->AddAt(func,isec);
}

TF1* AliMUONSegmentationV01::CorrFunc(Int_t isec) const
{ 
// Get correction function
  //PH    return (TF1*) (*fCorrA)[isec];
    return (TF1*) fCorrA->At(isec);
}

AliMUONSegmentationV01& AliMUONSegmentationV01::operator 
=(const AliMUONSegmentationV01 & rhs)
{
// Dummy assignment operator
    return *this;
}
Commit	Line	Data
a9e2aefa	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
88cb7938	16	/* $Id$ */
a9e2aefa	17
	18	/////////////////////////////////////////////////////
	19	// Segmentation and Response classes version 01 //
	20	/////////////////////////////////////////////////////
	21
	22	#include <TBox.h>
aaf4addd	23	#include <TTUBE.h>
	24	#include <TBRIK.h>
	25	#include <TNode.h>
	26	#include <TGeometry.h>
a9e2aefa	27	#include <TF1.h>
a9e2aefa	28	#include <TObjArray.h>
70479d0e	29	#include <Riostream.h>
a9e2aefa	30
	31	#include "AliMUONSegmentationV01.h"
	32	#include "AliMUON.h"
3e1872ed	33	#include "AliMUONChamber.h"
3e1872ed	34	#include "AliRun.h"
a9e2aefa	35
	36
	37
	38	//___________________________________________
	39	ClassImp(AliMUONSegmentationV01)
	40
	41	AliMUONSegmentationV01::AliMUONSegmentationV01(const AliMUONSegmentationV01& segmentation)
	42	{
	43	// Dummy copy constructor
	44	}
e9e4cdf2	45
a9e2aefa	46	AliMUONSegmentationV01::AliMUONSegmentationV01()
	47	{
	48	// Default constructor
e9e4cdf2	49	fRSec = 0;
e9e4cdf2	50	fNDiv = 0;
a8a6107b	51	fDpxD = 0;
a8a6107b	52	fCorrA = 0;
e9e4cdf2	53	}
	54
	55	AliMUONSegmentationV01::AliMUONSegmentationV01(Int_t nsec)
	56	{
	57	// Non default constructor
	58
	59	fNsec = nsec;
c3eff6ad	60	fRSec = new TArrayF(fNsec);
	61	fNDiv = new TArrayI(fNsec);
	62	fDpxD = new TArrayF(fNsec);
e9e4cdf2	63
e9e4cdf2	64
c3eff6ad	65	(fRSec)[0]=(fRSec)[1]=(fRSec)[2]=(fRSec)[3]=0;
	66	(fNDiv)[0]=(fNDiv)[1]=(fNDiv)[2]=(fNDiv)[3]=0;
	67	(fDpxD)[0]=(fDpxD)[1]=(fDpxD)[2]=(fDpxD)[3]=0;
bac96a1c	68	fCorrA = new TObjArray(3);
cd4df77b	69	fCorrA->AddAt(0,0);
	70	fCorrA->AddAt(0,1);
	71	fCorrA->AddAt(0,2);
e80ad807	72	fOffsetY=0;
a9e2aefa	73	}
a9e2aefa	74
e9e4cdf2	75	AliMUONSegmentationV01::~AliMUONSegmentationV01()
	76	{
	77	// Destructor
	78	if (fRSec) delete fRSec;
	79	if (fNDiv) delete fNDiv;
	80	if (fDpxD) delete fDpxD;
	81	if (fCorrA) {
	82	fCorrA->Delete();
	83	delete fCorrA;
	84	}
	85	}
	86
	87
94de3818	88	Float_t AliMUONSegmentationV01::Dpx(Int_t isec) const
a9e2aefa	89	{
	90	//
	91	// Returns x-pad size for given sector isec
c3eff6ad	92	Float_t dpx = (*fDpxD)[isec];
c3eff6ad	93	return dpx;
a9e2aefa	94	}
a9e2aefa	95
94de3818	96	Float_t AliMUONSegmentationV01::Dpy(Int_t isec) const
a9e2aefa	97	{
	98	//
	99	// Returns y-pad size for given sector isec
	100	return fDpy;
	101	}
	102
	103	void AliMUONSegmentationV01::SetSegRadii(Float_t r[4])
	104	{
	105	//
	106	// Set the radii of the segmentation zones
	107	for (Int_t i=0; i<4; i++) {
c3eff6ad	108	(*fRSec)[i]=r[i];
a9e2aefa	109	}
	110	}
	111
	112
	113	void AliMUONSegmentationV01::SetPadDivision(Int_t ndiv[4])
	114	{
	115	//
	116	// Defines the pad size perp. to the anode wire (y) for different sectors.
	117	// Pad sizes are defined as integral fractions ndiv of a basis pad size
	118	// fDpx
	119	//
	120	for (Int_t i=0; i<4; i++) {
c3eff6ad	121	(*fNDiv)[i]=ndiv[i];
a9e2aefa	122	}
	123	ndiv[0]=ndiv[1];
	124	}
	125
	126
d81db581	127	void AliMUONSegmentationV01::Init(Int_t chamber)
a9e2aefa	128	{
	129	//
	130	// Fill the arrays fCx (x-contour) and fNpxS (ix-contour) for each sector
	131	// These arrays help in converting from real to pad co-ordinates and
	132	// vice versa.
	133	// This version approximates concentric segmentation zones
	134	//
	135	Int_t isec;
9e1a0ddb	136	//printf("\n Initialise Segmentation V01\n");
e9e4cdf2	137
e9e4cdf2	138
c3eff6ad	139	fNpy=Int_t((*fRSec)[fNsec-1]/fDpy)+1;
a9e2aefa	140
c3eff6ad	141	(*fDpxD)[fNsec-1]=fDpx;
a9e2aefa	142	if (fNsec > 1) {
a9e2aefa	143	for (Int_t i=fNsec-2; i>=0; i--){
c3eff6ad	144	(fDpxD)[i]=(fDpxD)[fNsec-1]/(*fNDiv)[i];
a9e2aefa	145	}
	146	}
	147	//
	148	// fill the arrays defining the pad segmentation boundaries
	149	Float_t ry;
	150	Int_t dnx;
	151	Int_t add;
	152	//
	153	// loop over sections
	154	for(isec=0; isec<fNsec; isec++) {
	155	//
	156	// loop over pads along the aode wires
	157	for (Int_t iy=1; iy<=fNpy; iy++) {
	158	//
	159	Float_t x=iy*fDpy-fDpy/2;
c3eff6ad	160	if (x > (*fRSec)[isec]) {
a9e2aefa	161	fNpxS[isec][iy]=0;
	162	fCx[isec][iy]=0;
	163	} else {
c3eff6ad	164	ry=TMath::Sqrt((fRSec)[isec](fRSec)[isec]-xx);
a9e2aefa	165	if (isec > 1) {
c3eff6ad	166	dnx= Int_t((ry-fCx[isec-1][iy])/(*fDpxD)[isec]);
a9e2aefa	167	if (isec < fNsec-1) {
	168	if (TMath::Odd((Long_t)dnx)) dnx++;
	169	}
	170	fNpxS[isec][iy]=fNpxS[isec-1][iy]+dnx;
c3eff6ad	171	fCx[isec][iy]=fCx[isec-1][iy]+dnx(fDpxD)[isec];
a9e2aefa	172	} else if (isec == 1) {
c3eff6ad	173	dnx= Int_t((ry-fCx[isec-1][iy])/(*fDpxD)[isec]);
a9e2aefa	174	fNpxS[isec][iy]=fNpxS[isec-1][iy]+dnx;
	175	add=4 - (fNpxS[isec][iy])%4;
	176	if (add < 4) fNpxS[isec][iy]+=add;
	177	dnx=fNpxS[isec][iy]-fNpxS[isec-1][iy];
c3eff6ad	178	fCx[isec][iy]=fCx[isec-1][iy]+dnx(fDpxD)[isec];
a9e2aefa	179	} else {
c3eff6ad	180	dnx=Int_t(ry/(*fDpxD)[isec]);
a9e2aefa	181	fNpxS[isec][iy]=dnx;
c3eff6ad	182	fCx[isec][iy]=dnx(fDpxD)[isec];
a9e2aefa	183	}
	184	}
	185	} // y-pad loop
	186	} // sector loop
3e1872ed	187	// reference to chamber
	188	AliMUON pMUON = (AliMUON ) gAlice->GetModule("MUON");
	189	fChamber=&(pMUON->Chamber(chamber));
	190	fZ = fChamber->Z();
aaf4addd	191	fId=chamber;
a9e2aefa	192	}
	193
	194	Int_t AliMUONSegmentationV01::Sector(Int_t ix, Int_t iy)
	195	{
	196	// Returns sector number for given pad position
	197	//
	198	Int_t absix=TMath::Abs(ix);
	199	Int_t absiy=TMath::Abs(iy);
	200	Int_t isec=0;
	201	for (Int_t i=0; i<fNsec; i++) {
	202	if (absix<=fNpxS[i][absiy]){
	203	isec=i;
	204	break;
	205	}
	206	}
	207	return isec;
	208	}
	209
	210	void AliMUONSegmentationV01::
a30a000f	211	GetPadI(Float_t x, Float_t y, Int_t &ix, Int_t &iy)
a9e2aefa	212	{
	213	// Returns pad coordinates (ix,iy) for given real coordinates (x,y)
	214	//
e80ad807	215	iy = (y-fOffsetY >0)?
	216	Int_t((y-fOffsetY)/fDpy)+1
	217	:
	218	Int_t((y-fOffsetY)/fDpy)-1;
	219
a9e2aefa	220	if (iy > fNpy) iy= fNpy;
	221	if (iy < -fNpy) iy=-fNpy;
	222	//
	223	// Find sector isec
	224	Int_t isec=-1;
	225	Float_t absx=TMath::Abs(x);
	226	Int_t absiy=TMath::Abs(iy);
	227	for (Int_t i=0; i < fNsec; i++) {
	228	if (absx <= fCx[i][absiy]) {
	229	isec=i;
	230	break;
	231	}
	232	}
	233	if (isec>0) {
c3eff6ad	234	ix= Int_t((absx-fCx[isec-1][absiy])/(*fDpxD)[isec])
a9e2aefa	235	+fNpxS[isec-1][absiy]+1;
a9e2aefa	236	} else if (isec == 0) {
c3eff6ad	237	ix= Int_t(absx/(*fDpxD)[isec])+1;
a9e2aefa	238	} else {
	239	ix=fNpxS[fNsec-1][absiy]+1;
	240	}
	241	ix = (x>0) ? ix:-ix;
	242	}
	243
	244	void AliMUONSegmentationV01::
a30a000f	245	GetPadC(Int_t ix, Int_t iy, Float_t &x, Float_t &y)
a9e2aefa	246	{
	247	// Returns real coordinates (x,y) for given pad coordinates (ix,iy)
	248	//
bdf39337	249	y = (iy>0) ?
	250	Float_t(iy*fDpy)-fDpy/2.+fOffsetY
	251	:
	252	Float_t(iy*fDpy)+fDpy/2.+fOffsetY;
	253
a9e2aefa	254	//
	255	// Find sector isec
	256	Int_t isec=AliMUONSegmentationV01::Sector(ix,iy);
	257	//
	258	Int_t absix=TMath::Abs(ix);
	259	Int_t absiy=TMath::Abs(iy);
	260	if (isec) {
c3eff6ad	261	x=fCx[isec-1][absiy]+(absix-fNpxS[isec-1][absiy])(fDpxD)[isec];
c3eff6ad	262	x=(ix>0) ? x-(fDpxD)[isec]/2 : -x+(fDpxD)[isec]/2;
a9e2aefa	263	} else {
	264	x=y=0;
	265	}
	266	}
	267
	268	void AliMUONSegmentationV01::
	269	SetPad(Int_t ix, Int_t iy)
	270	{
	271	//
	272	// Sets virtual pad coordinates, needed for evaluating pad response
	273	// outside the tracking program
ecfa008b	274	GetPadC(ix,iy,fX,fY);
a9e2aefa	275	fSector=Sector(ix,iy);
	276	}
	277
	278
	279	void AliMUONSegmentationV01::
	280	FirstPad(Float_t xhit, Float_t yhit, Float_t dx, Float_t dy)
	281	{
	282	// Initialises iteration over pads for charge distribution algorithm
	283	//
	284	//
	285	// Find the wire position (center of charge distribution)
	286	Float_t x0a=GetAnod(xhit);
ecfa008b	287	fXhit=x0a;
ecfa008b	288	fYhit=yhit;
a9e2aefa	289
	290	//
	291	// and take fNsigma*sigma around this center
	292	Float_t x01=x0a - dx;
	293	Float_t x02=x0a + dx;
	294	Float_t y01=yhit - dy;
	295	Float_t y02=yhit + dy;
	296	//
	297	// find the pads over which the charge distributes
c3eff6ad	298
ecfa008b	299	GetPadI(x01,y01,fIxmin,fIymin);
	300	GetPadI(x02,y02,fIxmax,fIymax);
	301	fXmin=x01;
	302	fXmax=x02;
	303	fYmin=y01;
	304	fYmax=y02;
a9e2aefa	305
	306	//
	307	// Set current pad to lower left corner
ecfa008b	308	if (fIxmax < fIxmin) fIxmax=fIxmin;
	309	if (fIymax < fIymin) fIymax=fIymin;
	310	fIx=fIxmin;
	311	fIy=fIymin;
	312	GetPadC(fIx,fIy,fX,fY);
a9e2aefa	313	}
	314
	315
	316	void AliMUONSegmentationV01::NextPad()
	317	{
	318	// Stepper for the iteration over pads
	319	//
	320	// Step to next pad in the integration region
	321	//
	322	// Step to next pad in integration region
	323	Float_t xc,yc;
	324	Int_t iyc;
	325
	326	// step from left to right
c3eff6ad	327
ecfa008b	328	if (fX < fXmax && fX != 0) {
	329	if (fIx==-1) fIx++;
	330	fIx++;
a9e2aefa	331	// step up
ecfa008b	332	} else if (fIy != fIymax) {
	333	if (fIy==-1) fIy++;
	334	fIy++;
a9e2aefa	335	// get y-position of next row (yc), xc not used here
ecfa008b	336	GetPadC(fIx,fIy,xc,yc);
	337	// get x-pad coordiante for first pad in row (fIx)
	338	GetPadI(fXmin,yc,fIx,iyc);
a9e2aefa	339	} else {
d7d176c9	340	fIx=-1;
d7d176c9	341	fIy=-1;
a9e2aefa	342	}
ecfa008b	343	GetPadC(fIx,fIy,fX,fY);
ecfa008b	344	fSector=Sector(fIx,fIy);
a9e2aefa	345	if (MorePads() &&
	346	(fSector ==-1 \|\| fSector==0))
	347	NextPad();
	348	}
	349
	350	Int_t AliMUONSegmentationV01::MorePads()
de05461e	351
de05461e	352	{
a9e2aefa	353	// Stopping condition for the iterator over pads
	354	//
	355	// Are there more pads in the integration region
d7d176c9	356	return (fIx != -1 \|\| fIy != -1);
d7d176c9	357	/*
ecfa008b	358	if ((fX >= fXmax && fIy >= fIymax) \|\| fY==0) {
a9e2aefa	359	return 0;
	360	} else {
	361	return 1;
	362	}
d7d176c9	363	*/
a9e2aefa	364	}
	365
	366	void AliMUONSegmentationV01::
	367	IntegrationLimits(Float_t& x1,Float_t& x2,Float_t& y1, Float_t& y2)
	368	{
	369	// Returns integration limits for current pad
	370	//
ecfa008b	371	x1=fXhit-fX-Dpx(fSector)/2.;
a9e2aefa	372	x2=x1+Dpx(fSector);
ecfa008b	373	y1=fYhit-fY-Dpy(fSector)/2.;
a9e2aefa	374	y2=y1+Dpy(fSector);
	375	}
	376
	377	void AliMUONSegmentationV01::
	378	Neighbours(Int_t iX, Int_t iY, Int_t* Nlist, Int_t Xlist[10], Int_t Ylist[10])
	379	{
	380	// Returns list of next neighbours for given Pad (iX, iY)
	381	//
	382	const Float_t kEpsilon=fDpy/1000;
	383
	384	Float_t x,y;
	385	Int_t ixx, iyy, isec1;
	386	//
	387	Int_t isec0=AliMUONSegmentationV01::Sector(iX,iY);
	388	Int_t i=0;
	389	//
	390	// step right
	391	Xlist[i]=iX+1;
	392	if (Xlist[i]==0) Xlist[i]++;
	393	Ylist[i++]=iY;
	394	//
	395	// step left
	396	Xlist[i]=iX-1;
	397	if (Xlist[i]==0) Xlist[i]--;
	398	Ylist[i++]=iY;
	399	//
	400	// step up
a30a000f	401	AliMUONSegmentationV01::GetPadC(iX,iY,x,y);
a30a000f	402	AliMUONSegmentationV01::GetPadI(x+kEpsilon,y+fDpy,ixx,iyy);
a9e2aefa	403	Xlist[i]=ixx;
	404	Ylist[i++]=iyy;
	405	isec1=AliMUONSegmentationV01::Sector(ixx,iyy);
	406	if (isec1==isec0) {
	407	//
	408	// no sector boundary crossing
	409	// Xlist[i]=ixx+1;
	410	// Ylist[i++]=iY+1;
	411
	412	// Xlist[i]=ixx-1;
	413	// Ylist[i++]=iY+1;
	414	} else if (isec1 < isec0) {
	415	// finer segmentation
	416	// Xlist[i]=ixx+1;
	417	// Ylist[i++]=iY+1;
	418
	419	Xlist[i]=ixx-1;
	420	Ylist[i++]=iyy;
	421
	422	// Xlist[i]=ixx-2;
	423	// Ylist[i++]=iY+1;
	424	} else {
	425	// coarser segmenation
	426	/*
	427	if (TMath::Odd(iX-fNpxS[isec1-1][iY+1])) {
	428	Xlist[i]=ixx-1;
	429	Ylist[i++]=iY+1;
	430	} else {
	431	Xlist[i]=ixx+1;
	432	Ylist[i++]=iY+1;
	433	}
	434	*/
	435	}
	436
	437	//
	438	// step down
a30a000f	439	AliMUONSegmentationV01::GetPadC(iX,iY,x,y);
a30a000f	440	AliMUONSegmentationV01::GetPadI(x+kEpsilon,y-fDpy,ixx,iyy);
a9e2aefa	441	Xlist[i]=ixx;
	442	Ylist[i++]=iyy;
	443	isec1=AliMUONSegmentationV01::Sector(ixx,iyy);
	444	if (isec1==isec0) {
	445	//
	446	// no sector boundary crossing
	447	/*
	448	Xlist[i]=ixx+1;
	449	Ylist[i++]=iY-1;
	450
	451	Xlist[i]=ixx-1;
	452	Ylist[i++]=iY-1;
	453	*/
	454	} else if (isec1 < isec0) {
	455	// finer segmentation
	456	// Xlist[i]=ixx+1;
	457	// Ylist[i++]=iY-1;
	458
	459	Xlist[i]=ixx-1;
	460	Ylist[i++]=iyy;
	461
	462	// Xlist[i]=ixx-2;
	463	// Ylist[i++]=iY-1;
	464	} else {
	465	// coarser segmentation
	466	/*
	467	if (TMath::Odd(iX-fNpxS[isec1-1][iY-1])) {
	468	Xlist[i]=ixx-1;
	469	Ylist[i++]=iY-1;
	470	} else {
	471	Xlist[i]=ixx+1;
	472	Ylist[i++]=iY-1;
	473	}
	474	*/
	475	}
	476	*Nlist=i;
	477	}
	478
c3eff6ad	479	void AliMUONSegmentationV01::GiveTestPoints(Int_t &n, Float_t x, Float_t y) const
a9e2aefa	480	{
	481	// Returns test point on the pad plane.
	482	// Used during determination of the segmoid correction of the COG-method
	483
	484	n=3;
c3eff6ad	485	x[0]=((fRSec)[0]+(fRSec)[1])/2/TMath::Sqrt(2.);
a9e2aefa	486	y[0]=x[0];
c3eff6ad	487	x[1]=((fRSec)[1]+(fRSec)[2])/2/TMath::Sqrt(2.);
a9e2aefa	488	y[1]=x[1];
c3eff6ad	489	x[2]=((fRSec)[2]+(fRSec)[3])/2/TMath::Sqrt(2.);
a9e2aefa	490	y[2]=x[2];
	491	}
	492
aaf4addd	493	void AliMUONSegmentationV01::Draw(const char* opt) const
a9e2aefa	494	{
aaf4addd	495
a9e2aefa	496	// Draws the segmentation zones
a9e2aefa	497	//
aaf4addd	498	if (!strcmp(opt,"eventdisplay")) {
	499	const int kColorMUON = kBlue;
	500
	501	TRotMatrix* rot000 = new TRotMatrix("Rot000"," ", 90, 0, 90, 90, 0, 0);
	502	TRotMatrix* rot090 = new TRotMatrix("Rot090"," ", 90, 90, 90,180, 0, 0);
	503	TRotMatrix* rot180 = new TRotMatrix("Rot180"," ", 90,180, 90,270, 0, 0);
	504	TRotMatrix* rot270 = new TRotMatrix("Rot270"," ", 90,270, 90, 0, 0, 0);
	505
	506	char nameChamber[9], nameSense[9], nameFrame[9], nameNode[9];
	507	char nameSense1[9], nameSense2[9];
	508	TNode node, nodeF;
	509
	510	sprintf(nameChamber,"C_MUON%d",fId+1);
	511	sprintf(nameSense,"S_MUON%d",fId+1);
	512	sprintf(nameSense1,"S1_MUON%d",fId+1);
	513	sprintf(nameSense2,"S2_MUON%d",fId+1);
	514	sprintf(nameFrame,"F_MUON%d",fId+1);
	515
	516	TNode* top=gAlice->GetGeometry()->GetNode("alice");
	517
	518	Float_t rmin = (*fRSec)[0]-3;
	519	Float_t rmax = (*fRSec)[3]+3;
	520	new TTUBE(nameChamber,"Mother","void",rmin,rmax,0.25,1.);
	521	rmin = (*fRSec)[0];
	522	rmax = (*fRSec)[3];
	523	new TTUBE(nameSense,"Sens. region","void",rmin,rmax,0.25, 1.);
	524	Float_t dx=(rmax-rmin)/2;
	525	Float_t dy=3.;
	526	Float_t dz=0.25;
	527	TBRIK* frMUON = new TBRIK(nameFrame,"Frame","void",dx,dy,dz);
	528	top->cd();
	529	sprintf(nameNode,"MUON%d",100+fId+1);
	530	node = new TNode(nameNode,"ChamberNode",nameChamber,0,0,fChamber->Z(),"");
	531	node->SetLineColor(kColorMUON);
	532	AliMUON pMUON = (AliMUON ) gAlice->GetModule("MUON");
	533	(pMUON->Nodes())->Add(node);
	534	node->cd();
	535	sprintf(nameNode,"MUON%d",200+fId+1);
	536	node = new TNode(nameNode,"Sens. Region Node",nameSense,0,0,0,"");
	537	node->SetLineColor(kColorMUON);
	538	node->cd();
	539	Float_t dr=dx+rmin;
	540	sprintf(nameNode,"MUON%d",300+fId+1);
	541	nodeF = new TNode(nameNode,"Frame0",frMUON,dr, 0, 0,rot000,"");
	542	nodeF->SetLineColor(kColorMUON);
	543	node->cd();
	544	sprintf(nameNode,"MUON%d",400+fId+1);
	545	nodeF = new TNode(nameNode,"Frame1",frMUON,0 ,dr,0,rot090,"");
	546	nodeF->SetLineColor(kColorMUON);
	547	node->cd();
	548	sprintf(nameNode,"MUON%d",500+fId+1);
	549	nodeF = new TNode(nameNode,"Frame2",frMUON,-dr,0,0,rot180,"");
	550	nodeF->SetLineColor(kColorMUON);
	551	node ->cd();
	552	sprintf(nameNode,"MUON%d",600+fId+1);
	553	nodeF = new TNode(nameNode,"Frame3",frMUON,0,-dr,0,rot270,"");
	554	nodeF->SetLineColor(kColorMUON);
	555	} else {
a9e2aefa	556	TBox *box;
	557
	558	Float_t dx=0.95/fCx[3][1]/2;
	559	Float_t dy=0.95/(Float_t(Npy()))/2;
	560	Float_t x0,y0,x1,y1;
	561	Float_t xc=0.5;
	562	Float_t yc=0.5;
	563
aaf4addd	564	for (Int_t iy=1; iy<Npy(); iy++) {
	565	for (Int_t isec=0; isec<4; isec++) {
	566	if (isec==0) {
	567	x0=0;
	568	x1=fCx[isec][iy]*dx;
	569	} else {
	570	x0=fCx[isec-1][iy]*dx;
	571	x1=fCx[isec][iy]*dx;
a9e2aefa	572	}
aaf4addd	573	y0=Float_t(iy-1)*dy;
	574	y1=y0+dy;
	575	box=new TBox(x0+xc,y0+yc,x1+xc,y1+yc);
	576	box->SetFillColor(isec+1);
	577	box->Draw();
	578
	579	box=new TBox(-x1+xc,y0+yc,-x0+xc,y1+yc);
	580	box->SetFillColor(isec+1);
	581	box->Draw();
	582
	583	box=new TBox(x0+xc,-y1+yc,x1+xc,-y0+yc);
	584	box->SetFillColor(isec+1);
	585	box->Draw();
	586
	587	box=new TBox(-x1+xc,-y1+yc,-x0+xc,-y0+yc);
	588	box->SetFillColor(isec+1);
	589	box->Draw();
	590	}
a9e2aefa	591	}
aaf4addd	592	}
a9e2aefa	593	}
	594	void AliMUONSegmentationV01::SetCorrFunc(Int_t isec, TF1* func)
	595	{
de05461e	596	// Set the correction function
cd4df77b	597	fCorrA->AddAt(func,isec);
a9e2aefa	598	}
a9e2aefa	599
c3eff6ad	600	TF1* AliMUONSegmentationV01::CorrFunc(Int_t isec) const
a9e2aefa	601	{
de05461e	602	// Get correction function
2682e810	603	//PH return (TF1) (fCorrA)[isec];
2682e810	604	return (TF1*) fCorrA->At(isec);
a9e2aefa	605	}
	606
	607	AliMUONSegmentationV01& AliMUONSegmentationV01::operator
	608	=(const AliMUONSegmentationV01 & rhs)
	609	{
	610	// Dummy assignment operator
	611	return *this;
	612	}
de05461e	613