[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 "AliLog.h"

#include "AliTRDrecoParam.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("DEFAULT")
  ,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));

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