[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;
  rec->fRecEveryNTB = 2;
  rec->fkNMeanClusters = 10;
  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;
	203	rec->fRecEveryNTB = 2;
82ddb093	204	rec->fkNMeanClusters = 10;
8c499dbf	205	return rec;
	206
	207	}
	208
9716329b	209	//______________________________________________________________
	210	AliTRDrecoParam *AliTRDrecoParam::GetCosmicTestParam()
	211	{
	212	//
	213	// Parameters for the cosmics data
	214	//
	215
9716329b	216	AliTRDrecoParam *par = new AliTRDrecoParam();
6e39bde4	217	par->fSysCovMatrix[0] = 2.; // y direction (1 cm)
6e39bde4	218	par->fSysCovMatrix[1] = 2.; // z direction (1 cm)
b72f4eaf	219	par->fkChi2YSlope = 0.11853;
	220	par->fkChi2ZSlope = 0.04527;
	221	par->fkChi2YCut = 1.;
6e39bde4	222	par->fkPhiSlope = 10.; //3.17954;
8ae98148	223	par->fkMaxTheta = 2.1445;
8ae98148	224	par->fkMaxPhi = 2.7475;
5a2e200c	225	par->fkNMeanClusters = 12.89;
5a2e200c	226	par->fkNSigmaClusters = 2.095;
566bf887	227	par->fkRoadzMultiplicator = 3.;
a2fbb6ec	228	par->fStreamLevel[kTracker] = 1;
828c6f80	229	par->SetCheckTimeConsistency();
9716329b	230	return par;
	231
	232	}
8ae98148	233
	234
	235	//______________________________________________________________
	236	Float_t AliTRDrecoParam::GetNClusters() const
	237	{
	238	// Estimate the number of clusters in the TRD detector
	239
	240	Float_t nclusters = (fkNMeanClusters + 2fkNSigmaClusters)fkNMeanTracklets*fkdNchdy;
	241	nclusters *= 1.+fkNClusterNoise;
	242	return nclusters;
	243	}
11d80e40	244
	245	//______________________________________________________________
	246	void AliTRDrecoParam::SetPIDLQslices(Int_t s)
	247	{
	248	// Setting number of slices used by the PID LQ method s={1, 2}
	249	// If PID NN is set this function will change to PID LQ.
	250
	251	if(IsPIDNeuralNetwork()){
	252	AliWarning("PID set to NN. Changing to LQ.");
	253	SetPIDNeuralNetwork(kFALSE);
	254	}
	255
	256	switch(s){
	257	case 1:
	258	if(TESTBIT(fFlags, kLQ2D)) CLRBIT(fFlags, kLQ2D);
	259	break;
	260	case 2:
	261	SETBIT(fFlags, kLQ2D);
	262	break;
	263	default:
	264	AliWarning(Form("N[%d] PID LQ slices not implemented. Using default 2.", s));
	265	SETBIT(fFlags, kLQ2D);
	266	break;
	267	}
	268	}
	269