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

///////////////////////////////////////////////////////////////////////////////
//                                                                           //
//  Parameter class for the TRD reconstruction                               //
//                                                                           //
//  Authors:                                                                 //
//    Alex Bercuci <A.Bercuci@gsi.de>                                        //
//    Markus Fasel <M.Fasel@gsi.de>                                          //
//                                                                           //
///////////////////////////////////////////////////////////////////////////////

#include "AliTRDrecoParam.h"
#include "AliTRDrawStreamBase.h"

ClassImp(AliTRDrecoParam)


//______________________________________________________________
AliTRDrecoParam::AliTRDrecoParam()
  :AliDetectorRecoParam()
  ,fkdNchdy(12.)
  ,fkMaxTheta(1.0)	
  ,fkMaxPhi(2.0) 
  ,fkRoad0y(6.0)
  ,fkRoad0z(8.5) 
  ,fkRoad1y(2.0)
  ,fkRoad1z(20.0)	
  ,fkRoad2y(3.0)
  ,fkRoad2z(20.0)
  ,fkPtThreshold(2.0) 
  ,fkPlaneQualityThreshold(5.0)// 4.2? under Investigation
  ,fkRoadzMultiplicator(1.5)
  ,fkFindable(.333)
  ,fkChi2Z(30./*14.*//*12.5*/)
  ,fkChi2Y(.25)
  ,fkChi2YSlope(7.73)
  ,fkChi2ZSlope(0.069)
  ,fkChi2YCut(0.5)
  ,fkPhiSlope(10.6)
  ,fkNMeanClusters(20.)
  ,fkNSigmaClusters(2.)
  ,fkNClusterNoise(0.)
  ,fkNMeanTracklets(5.5)
  ,fkTrackLikelihood(-15.)
  ,fNumberOfConfigs(3)
  ,fFlags(0)
  ,fRawStreamVersion("REAL")
  ,fMinMaxCutSigma(4.)
  ,fMinLeftRightCutSigma(8.)
  ,fClusMaxThresh(4.5)
  ,fClusSigThresh(3.5)
  ,fTCnexp(1)
  ,fRecEveryNTB(1)
  ,fNumberOfPresamples(0)
  ,fNumberOfPostsamples(0)
{
  //
  // Default constructor
  //
  fSysCovMatrix[0] = 0.; // y direction (1 cm)
  fSysCovMatrix[1] = 0.; // z direction (1 cm)
  fSysCovMatrix[2] = 0.; // snp
  fSysCovMatrix[3] = 0.; // tgl
  fSysCovMatrix[4] = 0.; // 1/pt

  // Xe tail cancellation parameters
  fTCParams[0] = 1.156; // r1
  fTCParams[1] = 0.130; // r2
  fTCParams[2] = 0.114; // c1
  fTCParams[3] = 0.624; // c2
  // Ar tail cancellation parameters
  fTCParams[4] = 6.;    // r1
  fTCParams[5] = 0.62;  // r2
  fTCParams[6] = 0.0087;// c1
  fTCParams[7] = 0.07;  // c2

  memset(fPIDThreshold, 0, AliTRDCalPID::kNMom*sizeof(Double_t));
  memset(fStreamLevel, 0, kTRDreconstructionTasks * sizeof(Int_t));

  SetPIDNeuralNetwork();
  SetEightSlices();
  SetImproveTracklets();
  SetLUT();
  SetTailCancelation();
}

//______________________________________________________________
AliTRDrecoParam::AliTRDrecoParam(const AliTRDrecoParam &ref)
  :AliDetectorRecoParam(ref)
  ,fkdNchdy(ref.fkdNchdy)
  ,fkMaxTheta(ref.fkMaxTheta)
  ,fkMaxPhi(ref.fkMaxPhi)
  ,fkRoad0y(ref.fkRoad0y)
  ,fkRoad0z(ref.fkRoad0z) 
  ,fkRoad1y(ref.fkRoad1y)
  ,fkRoad1z(ref.fkRoad1z)	
  ,fkRoad2y(ref.fkRoad2y)
  ,fkRoad2z(ref.fkRoad2z)
  ,fkPtThreshold(ref.fkPtThreshold) 
  ,fkPlaneQualityThreshold(ref.fkPlaneQualityThreshold)
  ,fkRoadzMultiplicator(ref.fkRoadzMultiplicator)
  ,fkFindable(ref.fkFindable)
  ,fkChi2Z(ref.fkChi2Z)
  ,fkChi2Y(ref.fkChi2Y)
  ,fkChi2YSlope(ref.fkChi2YSlope)
  ,fkChi2ZSlope(ref.fkChi2ZSlope)
  ,fkChi2YCut(ref.fkChi2YCut)
  ,fkPhiSlope(ref.fkPhiSlope)
  ,fkNMeanClusters(ref.fkNMeanClusters)
  ,fkNSigmaClusters(ref.fkNSigmaClusters)
  ,fkNClusterNoise(ref.fkNClusterNoise)
  ,fkNMeanTracklets(ref.fkNMeanTracklets)
  ,fkTrackLikelihood(ref.fkTrackLikelihood)
  ,fNumberOfConfigs(ref.fNumberOfConfigs)
  ,fFlags(ref.fFlags)
  ,fRawStreamVersion(ref.fRawStreamVersion)
  ,fMinMaxCutSigma(ref.fMinMaxCutSigma)
  ,fMinLeftRightCutSigma(ref.fMinLeftRightCutSigma)
  ,fClusMaxThresh(ref.fClusMaxThresh)
  ,fClusSigThresh(ref.fClusSigThresh)
  ,fTCnexp(ref.fTCnexp)
  ,fRecEveryNTB(ref.fRecEveryNTB)
  ,fNumberOfPresamples(ref.fNumberOfPresamples)
  ,fNumberOfPostsamples(ref.fNumberOfPostsamples)
{
  //
  // Copy constructor
  //
  memcpy(fSysCovMatrix, ref.fSysCovMatrix, 5*sizeof(Double_t));
  memcpy(fTCParams, ref.fTCParams, 8*sizeof(Double_t));
  memcpy(fPIDThreshold, ref.fPIDThreshold, AliTRDCalPID::kNMom*sizeof(Double_t));
  memcpy(fStreamLevel, ref.fStreamLevel, kTRDreconstructionTasks * sizeof(Int_t));
}

//______________________________________________________________
AliTRDrecoParam *AliTRDrecoParam::GetLowFluxParam()
{
  //
  // Parameters for the low flux environment
  //

  AliTRDrecoParam *rec = new AliTRDrecoParam();
  rec->fkdNchdy = 12.; // pp in TRD
  rec->SetVertexConstrained();
  rec->SetCheckTimeConsistency();
  return rec;

}

//______________________________________________________________
AliTRDrecoParam *AliTRDrecoParam::GetLowFluxHLTParam()
{
  //
  // Parameters for the high flux environment in HLT
  //

  AliTRDrecoParam *rec = GetLowFluxParam();
  rec->fNumberOfConfigs = 2;
  return rec;

}

//______________________________________________________________
AliTRDrecoParam *AliTRDrecoParam::GetHighFluxParam()
{
  //
  // Parameters for the high flux environment
  //

  AliTRDrecoParam *rec = new AliTRDrecoParam();
  rec->fkdNchdy = 4000.; // PbPb in TRD
  rec->SetVertexConstrained();
  rec->SetCheckTimeConsistency();
  return rec;

}

//______________________________________________________________
AliTRDrecoParam *AliTRDrecoParam::GetHighFluxHLTParam()
{
  //
  // Parameters for the high flux environment in HLT
  //

  AliTRDrecoParam *rec = GetHighFluxParam();
  rec->fNumberOfConfigs = 1;
  return rec;

}

//______________________________________________________________
AliTRDrecoParam *AliTRDrecoParam::GetCosmicTestParam()
{
  //
  // Parameters for the cosmics data
  //

  AliTRDrecoParam *par = new AliTRDrecoParam();
  par->fSysCovMatrix[0] = 2.; // y direction (1 cm)
  par->fSysCovMatrix[1] = 2.; // z direction (1 cm)
  par->fkChi2YSlope     = 0.11853;
  par->fkChi2ZSlope     = 0.04527;
  par->fkChi2YCut       = 1.;
  par->fkPhiSlope       = 10.; //3.17954;
  par->fkMaxTheta       = 2.1445;
  par->fkMaxPhi         = 2.7475;
  par->fkNMeanClusters  = 12.89;
  par->fkNSigmaClusters = 2.095;
  par->fkRoadzMultiplicator = 3.;
  par->fStreamLevel[kTracker] = 1;
  par->SetCheckTimeConsistency();
  return par;

}


//______________________________________________________________
Float_t AliTRDrecoParam::GetNClusters() const
{
  // Estimate the number of clusters in the TRD detector
  
  Float_t nclusters = (fkNMeanClusters + 2*fkNSigmaClusters)*fkNMeanTracklets*fkdNchdy;
  nclusters *= 1.+fkNClusterNoise;
  return nclusters;
}

//______________________________________________________________
void AliTRDrecoParam::SetPIDLQslices(Int_t s)
{
// Setting number of slices used by the PID LQ method s={1, 2}
// If PID NN is set this function will change to PID LQ.
 
  if(IsPIDNeuralNetwork()){
    AliWarning("PID set to NN. Changing to LQ.");
    SetPIDNeuralNetwork(kFALSE);
  } 

  switch(s){
  case 1: 
    if(TESTBIT(fFlags, kLQ2D)) CLRBIT(fFlags, kLQ2D);
    break;
  case 2:
    SETBIT(fFlags, kLQ2D);
    break;
  default:
    AliWarning(Form("N[%d] PID LQ slices not implemented. Using default 2.", s));
    SETBIT(fFlags, kLQ2D);
    break;
  }
}
Commit	Line	Data
e4f2f73d	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	// //
	20	// Parameter class for the TRD reconstruction //
	21	// //
	22	// Authors: //
	23	// Alex Bercuci <A.Bercuci@gsi.de> //
	24	// Markus Fasel <M.Fasel@gsi.de> //
	25	// //
	26	///////////////////////////////////////////////////////////////////////////////
	27
	28	#include "AliTRDrecoParam.h"
9716329b	29	#include "AliTRDrawStreamBase.h"
e4f2f73d	30
	31	ClassImp(AliTRDrecoParam)
	32
a2fbb6ec	33
e4f2f73d	34	//______________________________________________________________
	35	AliTRDrecoParam::AliTRDrecoParam()
	36	:AliDetectorRecoParam()
8ae98148	37	,fkdNchdy(12.)
91834b8d	38	,fkMaxTheta(1.0)
91834b8d	39	,fkMaxPhi(2.0)
e4f2f73d	40	,fkRoad0y(6.0)
	41	,fkRoad0z(8.5)
	42	,fkRoad1y(2.0)
	43	,fkRoad1z(20.0)
	44	,fkRoad2y(3.0)
	45	,fkRoad2z(20.0)
b06a50a5	46	,fkPtThreshold(2.0)
e4f2f73d	47	,fkPlaneQualityThreshold(5.0)// 4.2? under Investigation
566bf887	48	,fkRoadzMultiplicator(1.5)
e4f2f73d	49	,fkFindable(.333)
eb38ed55	50	,fkChi2Z(30./14.//12.5/)
e4f2f73d	51	,fkChi2Y(.25)
5a2e200c	52	,fkChi2YSlope(7.73)
5a2e200c	53	,fkChi2ZSlope(0.069)
8c499dbf	54	,fkChi2YCut(0.5)
5a2e200c	55	,fkPhiSlope(10.6)
8ae98148	56	,fkNMeanClusters(20.)
	57	,fkNSigmaClusters(2.)
	58	,fkNClusterNoise(0.)
	59	,fkNMeanTracklets(5.5)
e4f2f73d	60	,fkTrackLikelihood(-15.)
8c499dbf	61	,fNumberOfConfigs(3)
a2fbb6ec	62	,fFlags(0)
a2fbb6ec	63	,fRawStreamVersion("REAL")
df83a620	64	,fMinMaxCutSigma(4.)
df83a620	65	,fMinLeftRightCutSigma(8.)
fc546d21	66	,fClusMaxThresh(4.5)
fc546d21	67	,fClusSigThresh(3.5)
56b32fd7	68	,fTCnexp(1)
8c499dbf	69	,fRecEveryNTB(1)
4e459a9d	70	,fNumberOfPresamples(0)
4e459a9d	71	,fNumberOfPostsamples(0)
e4f2f73d	72	{
	73	//
	74	// Default constructor
	75	//
d937ad7a	76	fSysCovMatrix[0] = 0.; // y direction (1 cm)
d937ad7a	77	fSysCovMatrix[1] = 0.; // z direction (1 cm)
3afdab72	78	fSysCovMatrix[2] = 0.; // snp
	79	fSysCovMatrix[3] = 0.; // tgl
	80	fSysCovMatrix[4] = 0.; // 1/pt
22a4ab0c	81
a2fbb6ec	82	// Xe tail cancellation parameters
	83	fTCParams[0] = 1.156; // r1
	84	fTCParams[1] = 0.130; // r2
	85	fTCParams[2] = 0.114; // c1
	86	fTCParams[3] = 0.624; // c2
	87	// Ar tail cancellation parameters
	88	fTCParams[4] = 6.; // r1
	89	fTCParams[5] = 0.62; // r2
	90	fTCParams[6] = 0.0087;// c1
	91	fTCParams[7] = 0.07; // c2
	92
22a4ab0c	93	memset(fPIDThreshold, 0, AliTRDCalPID::kNMom*sizeof(Double_t));
a2fbb6ec	94	memset(fStreamLevel, 0, kTRDreconstructionTasks * sizeof(Int_t));
	95
	96	SetPIDNeuralNetwork();
	97	SetEightSlices();
	98	SetImproveTracklets();
	99	SetLUT();
	100	SetTailCancelation();
e4f2f73d	101	}
e4f2f73d	102
a7ac01d2	103	//______________________________________________________________
	104	AliTRDrecoParam::AliTRDrecoParam(const AliTRDrecoParam &ref)
	105	:AliDetectorRecoParam(ref)
8ae98148	106	,fkdNchdy(ref.fkdNchdy)
a7ac01d2	107	,fkMaxTheta(ref.fkMaxTheta)
	108	,fkMaxPhi(ref.fkMaxPhi)
	109	,fkRoad0y(ref.fkRoad0y)
	110	,fkRoad0z(ref.fkRoad0z)
	111	,fkRoad1y(ref.fkRoad1y)
	112	,fkRoad1z(ref.fkRoad1z)
	113	,fkRoad2y(ref.fkRoad2y)
	114	,fkRoad2z(ref.fkRoad2z)
b06a50a5	115	,fkPtThreshold(ref.fkPtThreshold)
a7ac01d2	116	,fkPlaneQualityThreshold(ref.fkPlaneQualityThreshold)
566bf887	117	,fkRoadzMultiplicator(ref.fkRoadzMultiplicator)
a7ac01d2	118	,fkFindable(ref.fkFindable)
	119	,fkChi2Z(ref.fkChi2Z)
	120	,fkChi2Y(ref.fkChi2Y)
5a2e200c	121	,fkChi2YSlope(ref.fkChi2YSlope)
5a2e200c	122	,fkChi2ZSlope(ref.fkChi2ZSlope)
8c499dbf	123	,fkChi2YCut(ref.fkChi2YCut)
5a2e200c	124	,fkPhiSlope(ref.fkPhiSlope)
8ae98148	125	,fkNMeanClusters(ref.fkNMeanClusters)
	126	,fkNSigmaClusters(ref.fkNSigmaClusters)
	127	,fkNClusterNoise(ref.fkNClusterNoise)
	128	,fkNMeanTracklets(ref.fkNMeanTracklets)
a7ac01d2	129	,fkTrackLikelihood(ref.fkTrackLikelihood)
8c499dbf	130	,fNumberOfConfigs(ref.fNumberOfConfigs)
a2fbb6ec	131	,fFlags(ref.fFlags)
a2fbb6ec	132	,fRawStreamVersion(ref.fRawStreamVersion)
a7ac01d2	133	,fMinMaxCutSigma(ref.fMinMaxCutSigma)
	134	,fMinLeftRightCutSigma(ref.fMinLeftRightCutSigma)
	135	,fClusMaxThresh(ref.fClusMaxThresh)
	136	,fClusSigThresh(ref.fClusSigThresh)
	137	,fTCnexp(ref.fTCnexp)
8c499dbf	138	,fRecEveryNTB(ref.fRecEveryNTB)
4e459a9d	139	,fNumberOfPresamples(ref.fNumberOfPresamples)
4e459a9d	140	,fNumberOfPostsamples(ref.fNumberOfPostsamples)
a7ac01d2	141	{
	142	//
	143	// Copy constructor
	144	//
a7ac01d2	145	memcpy(fSysCovMatrix, ref.fSysCovMatrix, 5*sizeof(Double_t));
a2fbb6ec	146	memcpy(fTCParams, ref.fTCParams, 8*sizeof(Double_t));
22a4ab0c	147	memcpy(fPIDThreshold, ref.fPIDThreshold, AliTRDCalPID::kNMom*sizeof(Double_t));
a2fbb6ec	148	memcpy(fStreamLevel, ref.fStreamLevel, kTRDreconstructionTasks * sizeof(Int_t));
a7ac01d2	149	}
a7ac01d2	150
e4f2f73d	151	//______________________________________________________________
	152	AliTRDrecoParam *AliTRDrecoParam::GetLowFluxParam()
	153	{
	154	//
	155	// Parameters for the low flux environment
	156	//
	157
8ae98148	158	AliTRDrecoParam *rec = new AliTRDrecoParam();
8ae98148	159	rec->fkdNchdy = 12.; // pp in TRD
a2fbb6ec	160	rec->SetVertexConstrained();
828c6f80	161	rec->SetCheckTimeConsistency();
8ae98148	162	return rec;
e4f2f73d	163
	164	}
	165
8c499dbf	166	//______________________________________________________________
	167	AliTRDrecoParam *AliTRDrecoParam::GetLowFluxHLTParam()
	168	{
	169	//
	170	// Parameters for the high flux environment in HLT
	171	//
	172
	173	AliTRDrecoParam *rec = GetLowFluxParam();
82ddb093	174	rec->fNumberOfConfigs = 2;
8c499dbf	175	return rec;
	176
	177	}
	178
e4f2f73d	179	//______________________________________________________________
	180	AliTRDrecoParam *AliTRDrecoParam::GetHighFluxParam()
	181	{
	182	//
	183	// Parameters for the high flux environment
	184	//
	185
fb872574	186	AliTRDrecoParam *rec = new AliTRDrecoParam();
8ae98148	187	rec->fkdNchdy = 4000.; // PbPb in TRD
a2fbb6ec	188	rec->SetVertexConstrained();
828c6f80	189	rec->SetCheckTimeConsistency();
fb872574	190	return rec;
e4f2f73d	191
e4f2f73d	192	}
9716329b	193
8c499dbf	194	//______________________________________________________________
	195	AliTRDrecoParam *AliTRDrecoParam::GetHighFluxHLTParam()
	196	{
	197	//
	198	// Parameters for the high flux environment in HLT
	199	//
	200
	201	AliTRDrecoParam *rec = GetHighFluxParam();
	202	rec->fNumberOfConfigs = 1;
8c499dbf	203	return rec;
	204
	205	}
	206
9716329b	207	//______________________________________________________________
	208	AliTRDrecoParam *AliTRDrecoParam::GetCosmicTestParam()
	209	{
	210	//
	211	// Parameters for the cosmics data
	212	//
	213
9716329b	214	AliTRDrecoParam *par = new AliTRDrecoParam();
6e39bde4	215	par->fSysCovMatrix[0] = 2.; // y direction (1 cm)
6e39bde4	216	par->fSysCovMatrix[1] = 2.; // z direction (1 cm)
b72f4eaf	217	par->fkChi2YSlope = 0.11853;
	218	par->fkChi2ZSlope = 0.04527;
	219	par->fkChi2YCut = 1.;
6e39bde4	220	par->fkPhiSlope = 10.; //3.17954;
8ae98148	221	par->fkMaxTheta = 2.1445;
8ae98148	222	par->fkMaxPhi = 2.7475;
5a2e200c	223	par->fkNMeanClusters = 12.89;
5a2e200c	224	par->fkNSigmaClusters = 2.095;
566bf887	225	par->fkRoadzMultiplicator = 3.;
a2fbb6ec	226	par->fStreamLevel[kTracker] = 1;
828c6f80	227	par->SetCheckTimeConsistency();
9716329b	228	return par;
	229
	230	}
8ae98148	231
	232
	233	//______________________________________________________________
	234	Float_t AliTRDrecoParam::GetNClusters() const
	235	{
	236	// Estimate the number of clusters in the TRD detector
	237
	238	Float_t nclusters = (fkNMeanClusters + 2fkNSigmaClusters)fkNMeanTracklets*fkdNchdy;
	239	nclusters *= 1.+fkNClusterNoise;
	240	return nclusters;
	241	}
11d80e40	242
	243	//______________________________________________________________
	244	void AliTRDrecoParam::SetPIDLQslices(Int_t s)
	245	{
	246	// Setting number of slices used by the PID LQ method s={1, 2}
	247	// If PID NN is set this function will change to PID LQ.
	248
	249	if(IsPIDNeuralNetwork()){
	250	AliWarning("PID set to NN. Changing to LQ.");
	251	SetPIDNeuralNetwork(kFALSE);
	252	}
	253
	254	switch(s){
	255	case 1:
	256	if(TESTBIT(fFlags, kLQ2D)) CLRBIT(fFlags, kLQ2D);
	257	break;
	258	case 2:
	259	SETBIT(fFlags, kLQ2D);
	260	break;
	261	default:
	262	AliWarning(Form("N[%d] PID LQ slices not implemented. Using default 2.", s));
	263	SETBIT(fFlags, kLQ2D);
	264	break;
	265	}
	266	}
	267