X-Git-Url: http://git.uio.no/git/?a=blobdiff_plain;f=TRD%2FAliTRDdigitizer.cxx;h=715ae6ed2ca2a2bdce1fb2f3542d1475e5f3778c;hb=54d12e2fadf406ad2333e2ad2edae50d9f890ec4;hp=c0c022359a98d7d5df4d23b80c1f32395512185d;hpb=c3a4830fdcc6d2f15ab8691eb4b229d814cf05a3;p=u%2Fmrichter%2FAliRoot.git diff --git a/TRD/AliTRDdigitizer.cxx b/TRD/AliTRDdigitizer.cxx index c0c022359a9..715ae6ed2ca 100644 --- a/TRD/AliTRDdigitizer.cxx +++ b/TRD/AliTRDdigitizer.cxx @@ -13,97 +13,19 @@ * provided "as is" without express or implied warranty. * **************************************************************************/ -/* -$Log$ -Revision 1.19 2001/02/14 18:22:26 cblume -Change in the geometry of the padplane - -Revision 1.18 2001/01/26 19:56:57 hristov -Major upgrade of AliRoot code - -Revision 1.17 2000/12/08 12:53:27 cblume -Change in Copy() function for HP-compiler - -Revision 1.16 2000/12/07 12:20:46 cblume -Go back to array compression. Use sampled PRF to speed up digitization - -Revision 1.15 2000/11/23 14:34:08 cblume -Fixed bug in expansion routine of arrays (initialize buffers properly) - -Revision 1.14 2000/11/20 08:54:44 cblume -Switch off compression as default - -Revision 1.13 2000/11/10 14:57:52 cblume -Changes in the geometry constants for the DEC compiler - -Revision 1.12 2000/11/01 14:53:20 cblume -Merge with TRD-develop - -Revision 1.1.4.9 2000/10/26 17:00:22 cblume -Fixed bug in CheckDetector() - -Revision 1.1.4.8 2000/10/23 13:41:35 cblume -Added protection against Log(0) in the gas gain calulation - -Revision 1.1.4.7 2000/10/17 02:27:34 cblume -Get rid of global constants - -Revision 1.1.4.6 2000/10/16 01:16:53 cblume -Changed timebin 0 to be the one closest to the readout - -Revision 1.1.4.5 2000/10/15 23:34:29 cblume -Faster version of the digitizer - -Revision 1.1.4.4 2000/10/06 16:49:46 cblume -Made Getters const - -Revision 1.1.4.3 2000/10/04 16:34:58 cblume -Replace include files by forward declarations - -Revision 1.1.4.2 2000/09/22 14:41:10 cblume -Bug fix in PRF. Included time response. New structure - -Revision 1.10 2000/10/05 07:27:53 cblume -Changes in the header-files by FCA - -Revision 1.9 2000/10/02 21:28:19 fca -Removal of useless dependecies via forward declarations - -Revision 1.8 2000/06/09 11:10:07 cblume -Compiler warnings and coding conventions, next round - -Revision 1.7 2000/06/08 18:32:58 cblume -Make code compliant to coding conventions - -Revision 1.6 2000/06/07 16:27:32 cblume -Try to remove compiler warnings on Sun and HP - -Revision 1.5 2000/05/09 16:38:57 cblume -Removed PadResponse(). Merge problem - -Revision 1.4 2000/05/08 15:53:45 cblume -Resolved merge conflict - -Revision 1.3 2000/04/28 14:49:27 cblume -Only one declaration of iDict in MakeDigits() - -Revision 1.1.4.1 2000/05/08 14:42:04 cblume -Introduced AliTRDdigitsManager - -Revision 1.1 2000/02/28 19:00:13 cblume -Add new TRD classes - -*/ +/* $Id$ */ /////////////////////////////////////////////////////////////////////////////// // // // Creates and handles digits from TRD hits // +// Author: C. Blume (C.Blume@gsi.de) // // // // The following effects are included: // // - Diffusion // // - ExB effects // // - Gas gain including fluctuations // // - Pad-response (simple Gaussian approximation) // +// - Time-response // // - Electronics noise // // - Electronics gain // // - Digitization // @@ -111,8 +33,11 @@ Add new TRD classes // The corresponding parameter can be adjusted via the various // // Set-functions. If these parameters are not explicitly set, default // // values are used (see Init-function). // -// To produce digits from a root-file with TRD-hits use the // -// slowDigitsCreate.C macro. // +// As an example on how to use this class to produce digits from hits // +// have a look at the macro hits2digits.C // +// The production of summable digits is demonstrated in hits2sdigits.C // +// and the subsequent conversion of the s-digits into normal digits is // +// explained in sdigits2digits.C. // // // /////////////////////////////////////////////////////////////////////////////// @@ -125,8 +50,17 @@ Add new TRD classes #include #include #include +#include +#include #include "AliRun.h" +#include "AliRunLoader.h" +#include "AliLoader.h" +#include "AliConfig.h" +#include "AliMagF.h" +#include "AliRunDigitizer.h" +#include "AliRunLoader.h" +#include "AliLoader.h" #include "AliTRD.h" #include "AliTRDhit.h" @@ -136,390 +70,385 @@ Add new TRD classes #include "AliTRDsegmentArray.h" #include "AliTRDdigitsManager.h" #include "AliTRDgeometry.h" +#include "AliTRDparameter.h" +#include "AliTRDpadPlane.h" ClassImp(AliTRDdigitizer) //_____________________________________________________________________________ -AliTRDdigitizer::AliTRDdigitizer():TNamed() +AliTRDdigitizer::AliTRDdigitizer() { // // AliTRDdigitizer default constructor // - fInputFile = NULL; - fDigits = NULL; - fTRD = NULL; - fGeo = NULL; - fPRF = NULL; - fPRFsmp = NULL; - fTRF = NULL; - fTRFint = NULL; - - fEvent = 0; - fGasGain = 0.0; - fNoise = 0.0; - fChipGain = 0.0; - fADCoutRange = 0.0; - fADCinRange = 0.0; - fADCthreshold = 0; - fDiffusionOn = 0; - fDiffusionT = 0.0; - fDiffusionL = 0.0; - fElAttachOn = 0; - fElAttachProp = 0.0; - fExBOn = 0; - fOmegaTau = 0.0; - fPRFOn = 0; - fTRFOn = 0; - fDriftVelocity = 0.0; - - fPRFbin = 0; - fPRFlo = 0.0; - fPRFhi = 0.0; - fPRFwid = 0.0; - fPRFpad = 0; - fTRFbin = 0; - fTRFlo = 0.0; - fTRFhi = 0.0; - fTRFwid = 0.0; - - fCompress = kTRUE; - fVerbose = 1; - + fRunLoader = 0; + fDigitsManager = 0; + fSDigitsManager = 0; + fSDigitsManagerList = 0; + fTRD = 0; + fGeo = 0; + fPar = 0; + fEvent = 0; + fMasks = 0; + fCompress = kTRUE; + fDebug = 0; + fSDigits = kFALSE; + fSDigitsScale = 0.0; + fMergeSignalOnly = kFALSE; + fSimpleSim = kFALSE; + fSimpleDet = 0; + } //_____________________________________________________________________________ AliTRDdigitizer::AliTRDdigitizer(const Text_t *name, const Text_t *title) - :TNamed(name,title) + :AliDigitizer(name,title) { // - // AliTRDdigitizer default constructor + // AliTRDdigitizer constructor // - fInputFile = NULL; - fDigits = NULL; - fTRD = NULL; - fGeo = NULL; - fPRF = NULL; - fPRFsmp = NULL; - fTRF = NULL; - fTRFint = NULL; - - fEvent = 0; - - fCompress = kTRUE; - fVerbose = 1; - - Init(); + fRunLoader = 0; + + //NewIO: These data members probably are not needed anymore + fDigitsManager = 0; + fSDigitsManager = 0; + fSDigitsManagerList = 0; + fTRD = 0; + fGeo = 0; + fPar = 0; + //End NewIO comment + fEvent = 0; + fMasks = 0; + fCompress = kTRUE; + fDebug = 0; + fSDigits = kFALSE; + fSDigitsScale = 100.; // For the summable digits + fMergeSignalOnly = kFALSE; + fSimpleSim = kFALSE; + fSimpleDet = 0; + } //_____________________________________________________________________________ -AliTRDdigitizer::AliTRDdigitizer(const AliTRDdigitizer &d) +AliTRDdigitizer::AliTRDdigitizer(AliRunDigitizer *manager + , const Text_t *name, const Text_t *title) + :AliDigitizer(manager,name,title) { // - // AliTRDdigitizer copy constructor + // AliTRDdigitizer constructor // - ((AliTRDdigitizer &) d).Copy(*this); + fRunLoader = 0; + fDigitsManager = 0; + fSDigitsManager = 0; + fSDigitsManagerList = 0; + fTRD = 0; + fGeo = 0; + fPar = 0; + fEvent = 0; + fMasks = 0; + fCompress = kTRUE; + fDebug = 0; + fSDigits = kFALSE; + fSDigitsScale = 100.; // For the summable digits + fMergeSignalOnly = kFALSE; + fSimpleSim = kFALSE; + fSimpleDet = 0; + } //_____________________________________________________________________________ -AliTRDdigitizer::~AliTRDdigitizer() +AliTRDdigitizer::AliTRDdigitizer(AliRunDigitizer *manager) + :AliDigitizer(manager,"AliTRDdigitizer","TRD digitizer") { // - // AliTRDdigitizer destructor + // AliTRDdigitizer constructor // - if (fInputFile) { - fInputFile->Close(); - delete fInputFile; - } - if (fDigits) { - delete fDigits; - } + fRunLoader = 0; + fDigitsManager = 0; + fSDigitsManager = 0; + fSDigitsManagerList = 0; + fTRD = 0; + fGeo = 0; + fPar = 0; + fEvent = 0; + fMasks = 0; + fCompress = kTRUE; + fDebug = 0; + fSDigits = kFALSE; + fSDigitsScale = 100.; // For the summable digits + fMergeSignalOnly = kFALSE; + fSimpleSim = kFALSE; + fSimpleDet = 0; - if (fPRF) delete fPRF; - if (fTRF) delete fTRF; } //_____________________________________________________________________________ -AliTRDdigitizer &AliTRDdigitizer::operator=(const AliTRDdigitizer &d) +AliTRDdigitizer::AliTRDdigitizer(const AliTRDdigitizer &d):AliDigitizer(d) { // - // Assignment operator + // AliTRDdigitizer copy constructor // - if (this != &d) ((AliTRDdigitizer &) d).Copy(*this); - return *this; + ((AliTRDdigitizer &) d).Copy(*this); } //_____________________________________________________________________________ -void AliTRDdigitizer::Copy(TObject &d) +AliTRDdigitizer::~AliTRDdigitizer() { // - // Copy function + // AliTRDdigitizer destructor // - Int_t iBin; - - ((AliTRDdigitizer &) d).fInputFile = NULL; - ((AliTRDdigitizer &) d).fDigits = NULL; - ((AliTRDdigitizer &) d).fTRD = NULL; - ((AliTRDdigitizer &) d).fGeo = NULL; - - ((AliTRDdigitizer &) d).fEvent = 0; - - ((AliTRDdigitizer &) d).fGasGain = fGasGain; - ((AliTRDdigitizer &) d).fNoise = fNoise; - ((AliTRDdigitizer &) d).fChipGain = fChipGain; - ((AliTRDdigitizer &) d).fADCoutRange = fADCoutRange; - ((AliTRDdigitizer &) d).fADCinRange = fADCinRange; - ((AliTRDdigitizer &) d).fADCthreshold = fADCthreshold; - ((AliTRDdigitizer &) d).fDiffusionOn = fDiffusionOn; - ((AliTRDdigitizer &) d).fDiffusionT = fDiffusionT; - ((AliTRDdigitizer &) d).fDiffusionL = fDiffusionL; - ((AliTRDdigitizer &) d).fElAttachOn = fElAttachOn; - ((AliTRDdigitizer &) d).fElAttachProp = fElAttachProp; - ((AliTRDdigitizer &) d).fExBOn = fExBOn; - ((AliTRDdigitizer &) d).fOmegaTau = fOmegaTau; - ((AliTRDdigitizer &) d).fLorentzFactor = fLorentzFactor; - ((AliTRDdigitizer &) d).fPRFOn = fPRFOn; - ((AliTRDdigitizer &) d).fTRFOn = fTRFOn; - - ((AliTRDdigitizer &) d).fCompress = fCompress; - ((AliTRDdigitizer &) d).fVerbose = fVerbose; - - fPRF->Copy(*((AliTRDdigitizer &) d).fPRF); - fTRF->Copy(*((AliTRDdigitizer &) d).fTRF); - - ((AliTRDdigitizer &) d).fPRFbin = fPRFbin; - ((AliTRDdigitizer &) d).fPRFlo = fPRFlo; - ((AliTRDdigitizer &) d).fPRFhi = fPRFhi; - ((AliTRDdigitizer &) d).fPRFwid = fPRFwid; - ((AliTRDdigitizer &) d).fPRFpad = fPRFpad; - if (((AliTRDdigitizer &) d).fPRFsmp) delete ((AliTRDdigitizer &) d).fPRFsmp; - ((AliTRDdigitizer &) d).fPRFsmp = new Float_t[fPRFbin]; - for (iBin = 0; iBin < fPRFbin; iBin++) { - ((AliTRDdigitizer &) d).fPRFsmp[iBin] = fPRFsmp[iBin]; - } - ((AliTRDdigitizer &) d).fTRFbin = fTRFbin; - ((AliTRDdigitizer &) d).fTRFlo = fTRFlo; - ((AliTRDdigitizer &) d).fTRFhi = fTRFhi; - ((AliTRDdigitizer &) d).fTRFwid = fTRFwid; - if (((AliTRDdigitizer &) d).fTRFint) delete ((AliTRDdigitizer &) d).fTRFint; - ((AliTRDdigitizer &) d).fTRFint = new Float_t[fTRFbin]; - for (iBin = 0; iBin < fTRFbin; iBin++) { - ((AliTRDdigitizer &) d).fTRFint[iBin] = fTRFint[iBin]; - } -} + if (fDigitsManager) { + delete fDigitsManager; + fDigitsManager = 0; + } -//_____________________________________________________________________________ -Int_t AliTRDdigitizer::Diffusion(Float_t driftlength, Float_t *xyz) -{ - // - // Applies the diffusion smearing to the position of a single electron - // + fSDigitsManager = 0; - Float_t driftSqrt = TMath::Sqrt(driftlength); - Float_t sigmaT = driftSqrt * fDiffusionT; - Float_t sigmaL = driftSqrt * fDiffusionL; - xyz[0] = gRandom->Gaus(xyz[0], sigmaL * fLorentzFactor); - xyz[1] = gRandom->Gaus(xyz[1], sigmaT * fLorentzFactor); - xyz[2] = gRandom->Gaus(xyz[2], sigmaT); + if (fSDigitsManagerList) { + fSDigitsManagerList->Delete(); + delete fSDigitsManagerList; + fSDigitsManagerList = 0; + } - return 1; + if (fMasks) { + delete [] fMasks; + fMasks = 0; + } } //_____________________________________________________________________________ -Int_t AliTRDdigitizer::ExB(Float_t driftlength, Float_t *xyz) +AliTRDdigitizer &AliTRDdigitizer::operator=(const AliTRDdigitizer &d) { // - // Applies E x B effects to the position of a single electron + // Assignment operator // - xyz[0] = xyz[0]; - xyz[1] = xyz[1] + fOmegaTau * driftlength; - xyz[2] = xyz[2]; - - return 1; + if (this != &d) ((AliTRDdigitizer &) d).Copy(*this); + return *this; } //_____________________________________________________________________________ -Int_t AliTRDdigitizer::PadResponse(Float_t signal, Float_t dist, Float_t *pad) +void AliTRDdigitizer::Copy(TObject &d) const { // - // Applies the pad response + // Copy function // - Int_t iBin = ((Int_t) (( - dist - fPRFlo) / fPRFwid)); - - Int_t iBin0 = iBin - fPRFpad; - Int_t iBin1 = iBin; - Int_t iBin2 = iBin + fPRFpad; - - if ((iBin0 >= 0) && (iBin2 < fPRFbin)) { - - pad[0] = signal * fPRFsmp[iBin0]; - pad[1] = signal * fPRFsmp[iBin1]; - pad[2] = signal * fPRFsmp[iBin2]; - - return 1; - - } - else { - - return 0; - - } - + ((AliTRDdigitizer &) d).fRunLoader = 0; + ((AliTRDdigitizer &) d).fDigitsManager = 0; + ((AliTRDdigitizer &) d).fSDigitsManager = 0; + ((AliTRDdigitizer &) d).fSDigitsManagerList = 0; + ((AliTRDdigitizer &) d).fTRD = 0; + ((AliTRDdigitizer &) d).fGeo = 0; + ((AliTRDdigitizer &) d).fPar = 0; + ((AliTRDdigitizer &) d).fEvent = 0; + ((AliTRDdigitizer &) d).fMasks = 0; + ((AliTRDdigitizer &) d).fCompress = fCompress; + ((AliTRDdigitizer &) d).fDebug = fDebug ; + ((AliTRDdigitizer &) d).fSDigits = fSDigits; + ((AliTRDdigitizer &) d).fSDigitsScale = fSDigitsScale; + ((AliTRDdigitizer &) d).fMergeSignalOnly = fMergeSignalOnly; + ((AliTRDdigitizer &) d).fSimpleSim = fSimpleSim; + ((AliTRDdigitizer &) d).fSimpleDet = fSimpleDet; + } //_____________________________________________________________________________ -Float_t AliTRDdigitizer::TimeResponse(Float_t time) +void AliTRDdigitizer::Exec(Option_t* option) { // - // Applies the preamp shaper time response + // Executes the merging // - Int_t iBin = ((Int_t) ((time - fTRFlo) / fTRFwid)); - if ((iBin >= 0) && (iBin < fTRFbin)) { - return fTRFint[iBin]; + Int_t iInput; + + AliTRDdigitsManager *sdigitsManager; + + TString optionString = option; + if (optionString.Contains("deb")) { + fDebug = 1; + if (optionString.Contains("2")) { + fDebug = 2; + } + printf(" "); + printf("Called with debug option %d\n",fDebug); } - else { - return 0.0; - } -} + // The AliRoot file is already connected by the manager + AliRunLoader* inrl; + + if (gAlice) + { + if (fDebug > 0) { + printf(" "); + printf("AliRun object found on file.\n"); + } + } + else { + inrl = AliRunLoader::GetRunLoader(fManager->GetInputFolderName(0)); + inrl->LoadgAlice(); + gAlice = inrl->GetAliRun(); + if (!gAlice) + { + printf(" "); + printf("Could not find AliRun object.\n"); + return; + } + } + + Int_t nInput = fManager->GetNinputs(); + fMasks = new Int_t[nInput]; + for (iInput = 0; iInput < nInput; iInput++) { + fMasks[iInput] = fManager->GetMask(iInput); + } -//_____________________________________________________________________________ -void AliTRDdigitizer::Init() -{ - // - // Initializes the digitization procedure with standard values - // + // Initialization + + AliRunLoader* orl = AliRunLoader::GetRunLoader(fManager->GetOutputFolderName()); + if (InitDetector()) { + AliLoader* ogime = orl->GetLoader("TRDLoader"); + + TTree* tree = 0; + if (fSDigits) + { + //if we produce SDigits + tree = ogime->TreeS(); + if (!tree) + { + ogime->MakeTree("S"); + tree = ogime->TreeS(); + } + } + else + {//if we produce Digits + tree = ogime->TreeD(); + if (!tree) + { + ogime->MakeTree("D"); + tree = ogime->TreeD(); + } + } + MakeBranch(tree); + } + + for (iInput = 0; iInput < nInput; iInput++) { - // The default parameter for the digitization - fGasGain = 8.0E3; - fNoise = 3000.; - fChipGain = 10.; - fADCoutRange = 255.; - fADCinRange = 2000.; - fADCthreshold = 1; - - // Transverse and longitudinal diffusion coefficients (Xe/Isobutane) - fDiffusionOn = 1; - fDiffusionT = 0.060; - fDiffusionL = 0.017; - - // Propability for electron attachment - fElAttachOn = 0; - fElAttachProp = 0.0; - - // E x B effects - fExBOn = 0; - // omega * tau. (tau ~ 12 * 10^-12, B = 0.2T) - fOmegaTau = 17.6 * 12.0 * 0.2 * 0.01; - - // The pad response function - fPRFOn = 1; - fPRFlo = -3.0; - fPRFhi = 3.0; - fPRFbin = 1200; - fPRFwid = (fPRFhi - fPRFlo) / ((Float_t) fPRFbin); - fPRFpad = ((Int_t) (1.0 / fPRFwid)); - fPRF = new TF1("PRF","[0]*([1]+exp(-x*x/(2.0*[2])))",fPRFlo,fPRFhi); - fPRF->SetParameter(0, 0.8872); - fPRF->SetParameter(1,-0.00573); - fPRF->SetParameter(2, 0.454 * 0.454); - - // The drift velocity (cm / mus) - fDriftVelocity = 1.0; - - // The time response function - fTRFOn = 1; - Float_t loTRF = -200.0; - Float_t hiTRF = 1000.0; - fTRF = new TF1("TRF",TRFlandau,loTRF,hiTRF,3); - fTRF->SetParameter(0, 1.0 / 24.24249); - fTRF->SetParameter(1, 0.0); - fTRF->SetParameter(2, 25.0); - fTRFbin = 1200; - fTRFlo = loTRF * fDriftVelocity / 1000.0; - fTRFhi = hiTRF * fDriftVelocity / 1000.0; - fTRFwid = (fTRFhi - fTRFlo) / ((Float_t) fTRFbin); + if (fDebug > 0) { + printf(" "); + printf("Add input stream %d\n",iInput); + } -} + // check if the input tree exists + inrl = AliRunLoader::GetRunLoader(fManager->GetInputFolderName(iInput)); + AliLoader* gime = inrl->GetLoader("TRDLoader"); + + TTree * treees = gime->TreeS(); + if (treees == 0x0) + { + if (gime->LoadSDigits()) + { + Error("Exec","Error Occured while loading S. Digits for input %d.",iInput); + return; + } + treees = gime->TreeS(); + } + + if (treees == 0x0) { + printf(" "); + printf("Input stream %d does not exist\n",iInput); + return; + } + + // Read the s-digits via digits manager + sdigitsManager = new AliTRDdigitsManager(); + sdigitsManager->SetDebug(fDebug); + sdigitsManager->SetSDigits(kTRUE); + + AliRunLoader* rl = AliRunLoader::GetRunLoader(fManager->GetInputFolderName(iInput)); + AliLoader* gimme = rl->GetLoader("TRDLoader"); + if (!gimme->TreeS()) gimme->LoadSDigits(); + sdigitsManager->ReadDigits(gimme->TreeS()); + + // Add the s-digits to the input list + AddSDigitsManager(sdigitsManager); -//_____________________________________________________________________________ -void AliTRDdigitizer::IntegrateTRF() -{ - // - // Integrates the time response function over the time bin size - // + } - if (fTRFint) delete fTRFint; - fTRFint = new Float_t[fTRFbin]; - Float_t hiTRF = fTRFhi / fDriftVelocity * 1000.0; - Float_t loTRF = fTRFlo / fDriftVelocity * 1000.0; - Float_t timeBin = fGeo->GetTimeBinSize() / fDriftVelocity * 1000.0; - Float_t binWidth = (hiTRF - loTRF) / ((Float_t) fTRFbin); - for (Int_t iBin = 0; iBin < fTRFbin; iBin++) { - Float_t bin = iBin * binWidth + loTRF - 0.5 * timeBin; - fTRFint[iBin] = fTRF->Integral(bin,bin + timeBin); + // Convert the s-digits to normal digits + if (fDebug > 0) { + printf(" "); + printf("Do the conversion\n"); } + SDigits2Digits(); -} + // Store the digits + if (fDebug > 0) { + printf(" "); + printf("Write the digits\n"); + } + + fDigitsManager->WriteDigits(); -//_____________________________________________________________________________ -void AliTRDdigitizer::SamplePRF() -{ - // - // Samples the pad response function - // + //Write parameters + orl->CdGAFile(); + if (!gFile->Get("TRDparameter")) GetParameter()->Write(); - if (fPRFsmp) delete fPRFsmp; - fPRFsmp = new Float_t[fPRFbin]; - for (Int_t iBin = 0; iBin < fPRFbin; iBin++) { - Float_t bin = (((Float_t ) iBin) + 0.5) * fPRFwid + fPRFlo; - fPRFsmp[iBin] = TMath::Max(fPRF->Eval(bin),0.0); + if (fDebug > 0) { + printf(" "); + printf("Done\n"); } + DeleteSDigitsManager(); + } //_____________________________________________________________________________ -Bool_t AliTRDdigitizer::Open(const Char_t *name, Int_t nEvent) +Bool_t AliTRDdigitizer::Open(const Char_t *file, Int_t nEvent) { // // Opens a ROOT-file with TRD-hits and reads in the hit-tree // // Connect the AliRoot file containing Geometry, Kine, and Hits - fInputFile = (TFile*) gROOT->GetListOfFiles()->FindObject(name); - if (!fInputFile) { - printf("AliTRDdigitizer::Open -- "); - printf("Open the ALIROOT-file %s.\n",name); - fInputFile = new TFile(name,"UPDATE"); - } - else { - printf("AliTRDdigitizer::Open -- "); - printf("%s is already open.\n",name); - } - - gAlice = (AliRun*) fInputFile->Get("gAlice"); + + + TString evfoldname = AliConfig::GetDefaultEventFolderName(); + fRunLoader = AliRunLoader::GetRunLoader(evfoldname); + if (!fRunLoader) + fRunLoader = AliRunLoader::Open(file,AliConfig::GetDefaultEventFolderName(), + "UPDATE"); + + if (!fRunLoader) + { + Error("Open","Can not open session for file %s.",file); + return kFALSE; + } + + if (!fRunLoader->GetAliRun()) fRunLoader->LoadgAlice(); + gAlice = fRunLoader->GetAliRun(); + if (gAlice) { - printf("AliTRDdigitizer::Open -- "); - printf("AliRun object found on file.\n"); + if (fDebug > 0) { + printf(" "); + printf("AliRun object found on file.\n"); + } } else { - printf("AliTRDdigitizer::Open -- "); + printf(" "); printf("Could not find AliRun object.\n"); return kFALSE; } @@ -527,15 +456,41 @@ Bool_t AliTRDdigitizer::Open(const Char_t *name, Int_t nEvent) fEvent = nEvent; // Import the Trees for the event nEvent in the file - Int_t nparticles = gAlice->GetEvent(fEvent); - if (nparticles <= 0) { - printf("AliTRDdigitizer::Open -- "); - printf("No entries in the trees for event %d.\n",fEvent); + fRunLoader->GetEvent(fEvent); + + AliLoader* loader = fRunLoader->GetLoader("TRDLoader"); + if (!loader) + { + Error("Open","Can not get TRD loader from Run Loader"); + return kFALSE; + } + + if (InitDetector()) { + TTree* tree = 0; + if (fSDigits) + { + //if we produce SDigits + tree = loader->TreeS(); + if (!tree) + { + loader->MakeTree("S"); + tree = loader->TreeS(); + } + } + else + {//if we produce Digits + if (!tree) + { + loader->MakeTree("D"); + tree = loader->TreeD(); + } + } + return MakeBranch(tree); + } + else { return kFALSE; } - return InitDetector(); - } //_____________________________________________________________________________ @@ -546,27 +501,60 @@ Bool_t AliTRDdigitizer::InitDetector() // // Get the pointer to the detector class and check for version 1 - fTRD = (AliTRD*) gAlice->GetDetector("TRD"); + fTRD = (AliTRD *) gAlice->GetDetector("TRD"); + if (!fTRD) { + printf(" "); + printf("No TRD module found\n"); + exit(1); + } if (fTRD->IsVersion() != 1) { - printf("AliTRDdigitizer::Open -- "); + printf(" "); printf("TRD must be version 1 (slow simulator).\n"); exit(1); } // Get the geometry fGeo = fTRD->GetGeometry(); - printf("AliTRDdigitizer::Open -- "); - printf("Geometry version %d\n",fGeo->IsVersion()); + if (fDebug > 0) { + printf(" "); + printf("Geometry version %d\n",fGeo->IsVersion()); + } + + // Create a digits manager + delete fDigitsManager; + fDigitsManager = new AliTRDdigitsManager(); + fDigitsManager->SetSDigits(fSDigits); + fDigitsManager->CreateArrays(); + fDigitsManager->SetEvent(fEvent); + fDigitsManager->SetDebug(fDebug); + + // The list for the input s-digits manager to be merged + if (fSDigitsManagerList) { + fSDigitsManagerList->Delete(); + } else { + fSDigitsManagerList = new TList(); + } return kTRUE; } +//_____________________________________________________________________________ +Bool_t AliTRDdigitizer::MakeBranch(TTree* tree) const +{ + // + // Create the branches for the digits array + // + + return fDigitsManager->MakeBranch(tree); + +} + //_____________________________________________________________________________ Bool_t AliTRDdigitizer::MakeDigits() { // - // Loops through the TRD-hits and creates the digits. + // Creates digits. // /////////////////////////////////////////////////////////////// @@ -574,117 +562,157 @@ Bool_t AliTRDdigitizer::MakeDigits() /////////////////////////////////////////////////////////////// // Converts number of electrons to fC - const Float_t kEl2fC = 1.602E-19 * 1.0E15; + const Double_t kEl2fC = 1.602E-19 * 1.0E15; /////////////////////////////////////////////////////////////// // Number of pads included in the pad response - const Int_t kNpad = 3; + const Int_t kNpad = 3; // Number of track dictionary arrays - const Int_t kNDict = AliTRDdigitsManager::kNDict; + const Int_t kNDict = AliTRDdigitsManager::kNDict; + + // Half the width of the amplification region + const Float_t kAmWidth = AliTRDgeometry::AmThick() / 2.; + // Width of the drit region + const Float_t kDrWidth = AliTRDgeometry::DrThick(); - Int_t iRow, iCol, iTime, iPad; - Int_t iDict = 0; - Int_t nBytes = 0; + Int_t iRow, iCol, iTime, iPad; + Int_t iDict = 0; + Int_t nBytes = 0; - Int_t totalSizeDigits = 0; - Int_t totalSizeDict0 = 0; - Int_t totalSizeDict1 = 0; - Int_t totalSizeDict2 = 0; + Int_t totalSizeDigits = 0; + Int_t totalSizeDict0 = 0; + Int_t totalSizeDict1 = 0; + Int_t totalSizeDict2 = 0; - AliTRDdataArrayF *signals = 0; - AliTRDdataArrayI *digits = 0; + Int_t timeBinTRFend = 1; + + Double_t pos[3]; + Double_t rot[3]; + Double_t xyz[3]; + Double_t padSignal[kNpad]; + Double_t signalOld[kNpad]; + + AliTRDdataArrayF *signals = 0; + AliTRDdataArrayI *digits = 0; AliTRDdataArrayI *dictionary[kNDict]; - // Create a digits manager - fDigits = new AliTRDdigitsManager(); + AliTRDpadPlane *padPlane = 0; + + // Create a default parameter class if none is defined + if (!fPar) { + fPar = new AliTRDparameter("TRDparameter","Standard TRD parameter"); + if (fDebug > 0) { + printf(" "); + printf("Create the default parameter object\n"); + } + } // Create a container for the amplitudes AliTRDsegmentArray *signalsArray - = new AliTRDsegmentArray("AliTRDdataArrayF",AliTRDgeometry::Ndet()); + = new AliTRDsegmentArray("AliTRDdataArrayF" + ,AliTRDgeometry::Ndet()); + + if (fPar->TRFOn()) { + timeBinTRFend = ((Int_t) ( fPar->GetTRFhi() * fPar->GetSamplingFrequency())) - 1; + if (fDebug > 0) { + printf(" "); + printf("Sample the TRF up to bin %d\n",timeBinTRFend); + } + } + + Float_t elAttachProp = fPar->GetElAttachProp() / 100.; if (!fGeo) { - printf("AliTRDdigitizer::MakeDigits -- "); + printf(" "); printf("No geometry defined\n"); return kFALSE; } - printf("AliTRDdigitizer::MakeDigits -- "); - printf("Start creating digits.\n"); - if (fVerbose > 0) this->Dump(); - - // The Lorentz factor - if (fExBOn) { - fLorentzFactor = 1.0 / (1.0 + fOmegaTau*fOmegaTau); - } - else { - fLorentzFactor = 1.0; + if (fDebug > 0) { + printf(" "); + printf("Start creating digits.\n"); } - // Create the sampled PRF - SamplePRF(); - - // Create the integrated TRF - IntegrateTRF(); - - // Get the pointer to the hit tree - TTree *HitTree = gAlice->TreeH(); - + AliLoader* gimme = fRunLoader->GetLoader("TRDLoader"); + if (!gimme->TreeH()) gimme->LoadHits(); + TTree* hitTree = gimme->TreeH(); + if (hitTree == 0x0) + { + Error("MakeDigits","Can not get TreeH"); + return kFALSE; + } + fTRD->SetTreeAddress(); + // Get the number of entries in the hit tree // (Number of primary particles creating a hit somewhere) - Int_t nTrack = (Int_t) HitTree->GetEntries(); - if (fVerbose > 0) { - printf("AliTRDdigitizer::MakeDigits -- "); - printf("Found %d primary particles\n",nTrack); - } + Int_t nTrack = 1; + if (!fSimpleSim) { + nTrack = (Int_t) hitTree->GetEntries(); + if (fDebug > 0) { + printf(" "); + printf("Found %d primary particles\n",nTrack); + } + } Int_t detectorOld = -1; Int_t countHits = 0; + if (fDebug > 0) { + fPar->PrintDriftVelocity(); + printf(" Sampling = %.0fMHz\n", fPar->GetSamplingFrequency()); + printf(" Gain = %d\n",(Int_t)fPar->GetGasGain()); + printf(" Noise = %d\n",(Int_t)fPar->GetNoise()); + if (fPar->TimeStructOn()) { + printf(" "); + printf("Time Structure of drift cells implemented.\n"); + } else { + printf(" "); + printf("Constant drift velocity in drift cells.\n"); + } + } + // Loop through all entries in the tree for (Int_t iTrack = 0; iTrack < nTrack; iTrack++) { - gAlice->ResetHits(); - nBytes += HitTree->GetEvent(iTrack); - - // Get the number of hits in the TRD created by this particle - Int_t nHit = fTRD->Hits()->GetEntriesFast(); - if (fVerbose > 0) { - printf("AliTRDdigitizer::MakeDigits -- "); - printf("Found %d hits for primary particle %d\n",nHit,iTrack); + if (!fSimpleSim) { + gAlice->ResetHits(); + nBytes += hitTree->GetEvent(iTrack); } - // Loop through the TRD hits - for (Int_t iHit = 0; iHit < nHit; iHit++) { - + // Loop through the TRD hits + Int_t iHit = 0; + AliTRDhit *hit = (AliTRDhit *) fTRD->FirstHit(-1); + while (hit) { + countHits++; - - AliTRDhit *hit = (AliTRDhit *) fTRD->Hits()->UncheckedAt(iHit); - Float_t pos[3]; - pos[0] = hit->X(); - pos[1] = hit->Y(); - pos[2] = hit->Z(); - Float_t q = hit->GetCharge(); - Int_t track = hit->Track(); - Int_t detector = hit->GetDetector(); - Int_t plane = fGeo->GetPlane(detector); - Int_t sector = fGeo->GetSector(detector); - Int_t chamber = fGeo->GetChamber(detector); - - if (!(CheckDetector(plane,chamber,sector))) continue; - - Int_t nRowMax = fGeo->GetRowMax(plane,chamber,sector); - Int_t nColMax = fGeo->GetColMax(plane); - Int_t nTimeMax = fGeo->GetTimeMax(); - Float_t row0 = fGeo->GetRow0(plane,chamber,sector); - Float_t col0 = fGeo->GetCol0(plane); - Float_t time0 = fGeo->GetTime0(plane); - Float_t rowPadSize = fGeo->GetRowPadSize(plane,chamber,sector); - Float_t colPadSize = fGeo->GetColPadSize(plane); - Float_t timeBinSize = fGeo->GetTimeBinSize(); - - if (fVerbose > 1) { + iHit++; + + pos[0] = hit->X(); + pos[1] = hit->Y(); + pos[2] = hit->Z(); + Float_t q = hit->GetCharge(); + Int_t track = hit->Track(); + Int_t detector = hit->GetDetector(); + Int_t plane = fGeo->GetPlane(detector); + Int_t sector = fGeo->GetSector(detector); + Int_t chamber = fGeo->GetChamber(detector); + Int_t nTimeMax = fPar->GetTimeMax(); + Int_t nTimeBefore = fPar->GetTimeBefore(); + Int_t nTimeAfter = fPar->GetTimeAfter(); + Int_t nTimeTotal = fPar->GetTimeTotal(); + Float_t time0 = fPar->GetTime0(plane); + Float_t driftvelocity = fPar->GetDriftVelocity(); + Float_t samplingRate = fPar->GetSamplingFrequency(); + + padPlane = fPar->GetPadPlane(plane,chamber); + Float_t row0 = padPlane->GetRow0(); + Float_t col0 = padPlane->GetCol0(); + Int_t nRowMax = padPlane->GetNrows(); + Int_t nColMax = padPlane->GetNcols(); + + if (fDebug > 1) { printf("Analyze hit no. %d ",iHit); printf("-----------------------------------------------------------\n"); hit->Dump(); @@ -692,355 +720,781 @@ Bool_t AliTRDdigitizer::MakeDigits() ,plane,sector,chamber); printf("nRowMax = %d, nColMax = %d, nTimeMax = %d\n" ,nRowMax,nColMax,nTimeMax); + printf("nTimeBefore = %d, nTimeAfter = %d, nTimeTotal = %d\n" + ,nTimeBefore,nTimeAfter,nTimeTotal); printf("row0 = %f, col0 = %f, time0 = %f\n" ,row0,col0,time0); - printf("rowPadSize = %f, colPadSize = %f, timeBinSize = %f\n" - ,rowPadSize,colPadSize,timeBinSize); + printf("samplingRate = %f\n" + ,samplingRate); } - // Don't analyze test hits with amplitude 0. - if (((Int_t) q) == 0) continue; + // Don't analyze test hits and switched off detectors + if ((CheckDetector(plane,chamber,sector)) && + (((Int_t) q) != 0)) { - if (detector != detectorOld) { + if (detector != detectorOld) { - if (fVerbose > 1) { - printf("AliTRDdigitizer::MakeDigits -- "); - printf("Get new container. New det = %d, Old det = %d\n" - ,detector,detectorOld); - } - // Compress the old one if enabled - if ((fCompress) && (detectorOld > -1)) { - if (fVerbose > 1) { - printf("AliTRDdigitizer::MakeDigits -- "); - printf("Compress the old container ..."); - } - signals->Compress(1,0); - for (iDict = 0; iDict < kNDict; iDict++) { - dictionary[iDict]->Compress(1,0); + if (fDebug > 1) { + printf(" "); + printf("Get new container. New det = %d, Old det = %d\n" + ,detector,detectorOld); } - if (fVerbose > 1) printf("done\n"); - } - // Get the new container - signals = (AliTRDdataArrayF *) signalsArray->At(detector); - if (signals->GetNtime() == 0) { - // Allocate a new one if not yet existing - if (fVerbose > 1) { - printf("AliTRDdigitizer::MakeDigits -- "); - printf("Allocate a new container ... "); - } - signals->Allocate(nRowMax,nColMax,nTimeMax); - } - else { - // Expand an existing one - if (fCompress) { - if (fVerbose > 1) { - printf("AliTRDdigitizer::MakeDigits -- "); - printf("Expand an existing container ... "); + // Compress the old one if enabled + if ((fCompress) && (detectorOld > -1)) { + if (fDebug > 1) { + printf(" "); + printf("Compress the old container ..."); + } + signals->Compress(1,0); + for (iDict = 0; iDict < kNDict; iDict++) { + dictionary[iDict]->Compress(1,0); } - signals->Expand(); + if (fDebug > 1) printf("done\n"); } - } - // The same for the dictionary - for (iDict = 0; iDict < kNDict; iDict++) { - dictionary[iDict] = fDigits->GetDictionary(detector,iDict); - if (dictionary[iDict]->GetNtime() == 0) { - dictionary[iDict]->Allocate(nRowMax,nColMax,nTimeMax); + // Get the new container + signals = (AliTRDdataArrayF *) signalsArray->At(detector); + if (signals->GetNtime() == 0) { + // Allocate a new one if not yet existing + if (fDebug > 1) { + printf(" "); + printf("Allocate a new container ... "); + } + signals->Allocate(nRowMax,nColMax,nTimeTotal); } + else if (fSimpleSim) { + // Clear an old one for the simple simulation + if (fDebug > 1) { + printf(" "); + printf("Clear a old container ... "); + } + signals->Clear(); + } else { - if (fCompress) dictionary[iDict]->Expand(); + // Expand an existing one + if (fCompress) { + if (fDebug > 1) { + printf(" "); + printf("Expand an existing container ... "); + } + signals->Expand(); + } } - } - if (fVerbose > 1) printf("done\n"); - detectorOld = detector; - } - - // Rotate the sectors on top of each other - Float_t rot[3]; - fGeo->Rotate(detector,pos,rot); - - // The driftlength - Float_t driftlength = time0 - rot[0]; - if ((driftlength < 0) || - (driftlength > AliTRDgeometry::DrThick())) continue; - Float_t driftlengthL = driftlength; - if (fExBOn) driftlengthL /= TMath::Sqrt(fLorentzFactor); - - // The hit position in pad coordinates (center pad) - // The pad row (z-direction) - Int_t rowH = (Int_t) ((rot[2] - row0) / rowPadSize); - // The pad column (rphi-direction) - Int_t colH = (Int_t) ((rot[1] - col0) / colPadSize); - // The time bucket - Int_t timeH = (Int_t) (driftlength / timeBinSize); - if (fVerbose > 1) { - printf("rowH = %d, colH = %d, timeH = %d\n" - ,rowH,colH,timeH); - } - - // Loop over all electrons of this hit - // TR photons produce hits with negative charge - Int_t nEl = ((Int_t) TMath::Abs(q)); - for (Int_t iEl = 0; iEl < nEl; iEl++) { - - Float_t xyz[3]; - xyz[0] = rot[0]; - xyz[1] = rot[1]; - xyz[2] = rot[2]; - - // Electron attachment - if (fElAttachOn) { - if (gRandom->Rndm() < (driftlengthL * fElAttachProp / 100.)) - continue; + // The same for the dictionary + if (!fSimpleSim) { + for (iDict = 0; iDict < kNDict; iDict++) { + dictionary[iDict] = fDigitsManager->GetDictionary(detector,iDict); + if (dictionary[iDict]->GetNtime() == 0) { + dictionary[iDict]->Allocate(nRowMax,nColMax,nTimeTotal); + } + else { + if (fCompress) dictionary[iDict]->Expand(); + } + } + } + if (fDebug > 1) printf("done\n"); + detectorOld = detector; } - // Apply the diffusion smearing - if (fDiffusionOn) { - if (!(Diffusion(driftlengthL,xyz))) continue; + // Rotate the sectors on top of each other + if (fSimpleSim) { + rot[0] = pos[0]; + rot[1] = pos[1]; + rot[2] = pos[2]; + } + else { + fGeo->Rotate(detector,pos,rot); } - // Apply E x B effects - if (fExBOn) { - if (!(ExB(driftlength,xyz))) continue; - } + // The driftlength. It is negative if the hit is between pad plane and anode wires. + Double_t driftlength = time0 - rot[0]; + + // Normalised drift length + Double_t absdriftlength = TMath::Abs(driftlength); + if (fPar->ExBOn()) absdriftlength /= TMath::Sqrt(fPar->GetLorentzFactor()); + + // Loop over all electrons of this hit + // TR photons produce hits with negative charge + Int_t nEl = ((Int_t) TMath::Abs(q)); + for (Int_t iEl = 0; iEl < nEl; iEl++) { + + xyz[0] = rot[0]; + xyz[1] = rot[1]; + xyz[2] = rot[2]; + + // Stupid patch to take care of TR photons that are absorbed + // outside the chamber volume. A real fix would actually need + // a more clever implementation of the TR hit generation + if (q < 0.0) { + if ((xyz[2] < padPlane->GetRowEnd()) || + (xyz[2] > padPlane->GetRow0())) { + if (iEl == 0) { + printf(" "); + printf("Hit outside of sensitive volume, row (z=%f, row0=%f, rowE=%f)\n" + ,xyz[2],padPlane->GetRow0(),padPlane->GetRowEnd()); + } + continue; + } + Float_t tt = driftlength + kAmWidth; + if (tt < 0.0 || tt > kDrWidth + 2.*kAmWidth) { + if (iEl == 0) { + printf(" "); + printf("Hit outside of sensitive volume, time (Q = %d)\n",((Int_t) q)); + } + continue; + } + } - // The electron position - // The pad row (z-direction) - Int_t rowE = (Int_t) ((xyz[2] - row0) / rowPadSize); - // The pad column (rphi-direction) - Int_t colE = (Int_t) ((xyz[1] - col0) / colPadSize); - // The time bucket - Int_t timeE = (Int_t) ((time0 - xyz[0]) / timeBinSize); - - if (( rowE < 0) || ( rowE >= nRowMax)) continue; - if (( colE < 0) || ( colE >= nColMax)) continue; - if ((timeE < 0) || (timeE >= nTimeMax)) continue; - - // Apply the gas gain including fluctuations - Float_t ggRndm = 0.0; - do { - ggRndm = gRandom->Rndm(); - } while (ggRndm <= 0); - Int_t signal = (Int_t) (-fGasGain * TMath::Log(ggRndm)); - - if (fVerbose > 2) { - printf(" electron no. %d, signal = %d\n",iEl,signal); - printf(" rowE = %d, colE = %d, timeE = %d\n" - ,rowE,colE,timeE); - } + // Electron attachment + if (fPar->ElAttachOn()) { + if (gRandom->Rndm() < (absdriftlength * elAttachProp)) continue; + } - // Apply the pad response - Float_t padSignal[kNpad]; - if (fPRFOn) { - // The distance of the electron to the center of the pad - // in units of pad width - Float_t dist = (xyz[1] - col0 - (colE + 0.5) * colPadSize) - / colPadSize; - if (!(PadResponse(signal,dist,padSignal))) continue; - } - else { - padSignal[0] = 0.0; - padSignal[1] = signal; - padSignal[2] = 0.0; - } + // Apply the diffusion smearing + if (fPar->DiffusionOn()) { + if (!(fPar->Diffusion(absdriftlength,xyz))) continue; + } - // The distance of the position to the beginning of the timebin - Float_t timeOffset = (time0 - timeE * timeBinSize) - xyz[0]; - Int_t timeTRDbeg = 0; - Int_t timeTRDend = 1; - if (fTRFOn) { - timeTRDbeg = 2; - timeTRDend = 11; - } - for (Int_t iTimeBin = TMath::Max(timeE - timeTRDbeg, 0) - ; iTimeBin < TMath::Min(timeE + timeTRDend,nTimeMax) - ; iTimeBin++) { - - // Apply the time response - Float_t timeResponse = 1.0; - if (fTRFOn) { - Float_t time = (iTimeBin - timeE) * timeBinSize + timeOffset; - timeResponse = TimeResponse(time); + // Apply E x B effects (depends on drift direction) + if (fPar->ExBOn()) { + if (!(fPar->ExB(driftlength,xyz))) continue; + } + + // The electron position after diffusion and ExB in pad coordinates. + // The pad row (z-direction) + Int_t rowE = padPlane->GetPadRowNumber(xyz[2]); + if (rowE < 0) continue; + Double_t rowOffset = padPlane->GetPadRowOffset(rowE,xyz[2]); + + // The pad column (rphi-direction) + Int_t colE = padPlane->GetPadColNumber(xyz[1],rowOffset); + if (colE < 0) continue; + Double_t colOffset = padPlane->GetPadColOffset(colE,xyz[1]); + + // Convert the position to drift time, using either constant drift velocity or + // time structure of drift cells (non-isochronity, GARFIELD calculation). + Double_t drifttime; + if (fPar->TimeStructOn()) { + // Get z-position with respect to anode wire: + //Double_t Z = xyz[2] - row0 + fPar->GetAnodeWireOffset(); + Double_t Z = row0 - xyz[2] + fPar->GetAnodeWireOffset(); + Z -= ((Int_t)(2*Z))/2.; + if (Z>0.25) Z = 0.5-Z; + // use drift time map (GARFIELD) + drifttime = fPar->TimeStruct(time0 - xyz[0] + kAmWidth, Z); + } + else { + // use constant drift velocity + drifttime = TMath::Abs(time0 - xyz[0]) / driftvelocity; } - // Add the signals - Float_t signalOld[kNpad] = { 0.0, 0.0, 0.0 }; - for (iPad = 0; iPad < kNpad; iPad++) { - Int_t colPos = colE + iPad - 1; - if (colPos < 0) continue; - if (colPos >= nColMax) break; - signalOld[iPad] = signals->GetData(rowE,colPos,iTimeBin); - signalOld[iPad] += padSignal[iPad] * timeResponse; - signals->SetData(rowE,colPos,iTimeBin,signalOld[iPad]); + // The time bin (always positive) + Int_t timeE = ((Int_t) (drifttime * samplingRate)); + // The distance of the position to the middle of the timebin + Double_t timeOffset = ((((Float_t) timeE) + 0.5) / samplingRate) - drifttime; + + // Apply the gas gain including fluctuations + Double_t ggRndm = 0.0; + do { + ggRndm = gRandom->Rndm(); + } while (ggRndm <= 0); + Int_t signal = (Int_t) (-fPar->GetGasGain() * TMath::Log(ggRndm)); + + // Apply the pad response + if (fPar->PRFOn()) { + // The distance of the electron to the center of the pad + // in units of pad width + //Double_t dist = - colOffset / padPlane->GetColSize(colE); + Double_t dist = (0.5 * padPlane->GetColSize(colE) - colOffset) + / padPlane->GetColSize(colE); + if (!(fPar->PadResponse(signal,dist,plane,padSignal))) continue; } - if (fVerbose > 3) { - printf(" iTimeBin = %d, timeResponse = %f\n" - ,iTimeBin,timeResponse); - printf(" pad-signal = %f, %f, %f\n" - ,signalOld[0],signalOld[1],signalOld[2]); + else { + padSignal[0] = 0.0; + padSignal[1] = signal; + padSignal[2] = 0.0; } - // Store the track index in the dictionary - // Note: We store index+1 in order to allow the array to be compressed - //for (iDict = 0; iDict < kNDict; iDict++) { - // Int_t oldTrack = dictionary[iDict]->GetData(rowE,colE,timeE); - // if (oldTrack == track+1) break; - // //if (oldTrack == -1) break; - // if (oldTrack == 0) { - // dictionary[iDict]->SetData(rowE,colE,timeE,track+1); - // if (fVerbose > 3) { - // printf(" track index = %d\n",track); - // } - // break; - // } - //} - for (iPad = 0; iPad < kNpad; iPad++) { - Int_t colPos = colE + iPad - 1; - if (colPos < 0) continue; - if (colPos >= nColMax) break; - if (signals->GetData(rowE,colPos,iTimeBin) > 0) { - for (iDict = 0; iDict < kNDict; iDict++) { - Int_t oldTrack = dictionary[iDict]->GetData(rowE,colPos,iTimeBin); - if (oldTrack == track+1) break; - //if (oldTrack == -1) break; - if (oldTrack == 0) { - dictionary[iDict]->SetData(rowE,colPos,iTimeBin,track+1); - if (fVerbose > 3) { - printf(" track index = %d\n",track); + // Sample the time response inside the drift region + // + additional time bins before and after. + // The sampling is done always in the middle of the time bin + for (Int_t iTimeBin = TMath::Max(timeE,-nTimeBefore) ; + iTimeBin < TMath::Min(timeE+timeBinTRFend,nTimeMax+nTimeAfter ) ; + iTimeBin++ ) { + + // Apply the time response + Double_t timeResponse = 1.0; + Double_t crossTalk = 0.0; + Double_t time = (iTimeBin - timeE) / samplingRate + timeOffset; + if (fPar->TRFOn()) { + timeResponse = fPar->TimeResponse(time); + } + if (fPar->CTOn()) { + crossTalk = fPar->CrossTalk(time); + } + + signalOld[0] = 0.0; + signalOld[1] = 0.0; + signalOld[2] = 0.0; + + for (iPad = 0; iPad < kNpad; iPad++) { + + Int_t colPos = colE + iPad - 1; + if (colPos < 0) continue; + if (colPos >= nColMax) break; + + // Add the signals + // Note: The time bin number is shifted by nTimeBefore to avoid negative + // time bins. This has to be subtracted later. + Int_t iCurrentTimeBin = iTimeBin + nTimeBefore; + signalOld[iPad] = signals->GetDataUnchecked(rowE,colPos,iCurrentTimeBin); + if( colPos != colE ) { + signalOld[iPad] += padSignal[iPad] * (timeResponse + crossTalk); + } + else { + signalOld[iPad] += padSignal[iPad] * timeResponse; + } + signals->SetDataUnchecked(rowE,colPos,iCurrentTimeBin,signalOld[iPad]); + + // Store the track index in the dictionary + // Note: We store index+1 in order to allow the array to be compressed + if ((signalOld[iPad] > 0) && (!fSimpleSim)) { + for (iDict = 0; iDict < kNDict; iDict++) { + Int_t oldTrack = dictionary[iDict]->GetDataUnchecked(rowE + ,colPos + ,iCurrentTimeBin); + if (oldTrack == track+1) break; + if (oldTrack == 0) { + dictionary[iDict]->SetDataUnchecked(rowE,colPos,iCurrentTimeBin,track+1); + break; } - break; } } - } - } - if ((fVerbose > 1) && (iDict == kNDict)) { - printf("AliTRDdigitizer::MakeDigits -- "); - printf("More than three tracks for one digit!\n"); - } - } + } // Loop: pads - } + } // Loop: time bins - } + } // Loop: electrons of a single hit + + } // If: detector and test hit + + hit = (AliTRDhit *) fTRD->NextHit(); - } // All hits finished + } // Loop: hits of one primary track - printf("AliTRDdigitizer::MakeDigits -- "); - printf("Finished analyzing %d hits\n",countHits); + } // Loop: primary tracks + + if (fDebug > 0) { + printf(" "); + printf("Finished analyzing %d hits\n",countHits); + } + + // The coupling factor + Double_t coupling = fPar->GetPadCoupling() + * fPar->GetTimeCoupling(); + + // The conversion factor + Double_t convert = kEl2fC + * fPar->GetChipGain(); // Loop through all chambers to finalize the digits - for (Int_t iDet = 0; iDet < AliTRDgeometry::Ndet(); iDet++) { + Int_t iDetBeg = 0; + Int_t iDetEnd = AliTRDgeometry::Ndet(); + if (fSimpleSim) { + iDetBeg = fSimpleDet; + iDetEnd = iDetBeg + 1; + } + for (Int_t iDet = iDetBeg; iDet < iDetEnd; iDet++) { - Int_t plane = fGeo->GetPlane(iDet); - Int_t sector = fGeo->GetSector(iDet); - Int_t chamber = fGeo->GetChamber(iDet); - Int_t nRowMax = fGeo->GetRowMax(plane,chamber,sector); - Int_t nColMax = fGeo->GetColMax(plane); - Int_t nTimeMax = fGeo->GetTimeMax(); + Int_t plane = fGeo->GetPlane(iDet); + Int_t sector = fGeo->GetSector(iDet); + Int_t chamber = fGeo->GetChamber(iDet); + Int_t nRowMax = fPar->GetRowMax(plane,chamber,sector); + Int_t nColMax = fPar->GetColMax(plane); + Int_t nTimeMax = fPar->GetTimeMax(); + Int_t nTimeTotal = fPar->GetTimeTotal(); - //if (!(CheckDetector(plane,chamber,sector))) continue; - if (fVerbose > 0) { - printf("AliTRDdigitizer::MakeDigits -- "); + Double_t *inADC = new Double_t[nTimeTotal]; + Double_t *outADC = new Double_t[nTimeTotal]; + + if (fDebug > 0) { + printf(" "); printf("Digitization for chamber %d\n",iDet); } // Add a container for the digits of this detector - digits = fDigits->GetDigits(iDet); + digits = fDigitsManager->GetDigits(iDet); // Allocate memory space for the digits buffer - digits->Allocate(nRowMax,nColMax,nTimeMax); - + if (digits->GetNtime() == 0) { + digits->Allocate(nRowMax,nColMax,nTimeTotal); + } + else if (fSimpleSim) { + digits->Clear(); + } + // Get the signal container signals = (AliTRDdataArrayF *) signalsArray->At(iDet); if (signals->GetNtime() == 0) { // Create missing containers - signals->Allocate(nRowMax,nColMax,nTimeMax); + signals->Allocate(nRowMax,nColMax,nTimeTotal); } else { // Expand the container if neccessary if (fCompress) signals->Expand(); } // Create the missing dictionary containers - for (iDict = 0; iDict < kNDict; iDict++) { - dictionary[iDict] = fDigits->GetDictionary(iDet,iDict); - if (dictionary[iDict]->GetNtime() == 0) { - dictionary[iDict]->Allocate(nRowMax,nColMax,nTimeMax); - } + if (!fSimpleSim) { + for (iDict = 0; iDict < kNDict; iDict++) { + dictionary[iDict] = fDigitsManager->GetDictionary(iDet,iDict); + if (dictionary[iDict]->GetNtime() == 0) { + dictionary[iDict]->Allocate(nRowMax,nColMax,nTimeTotal); + } + } } Int_t nDigits = 0; - // Create the digits for this chamber - for (iRow = 0; iRow < nRowMax; iRow++ ) { - for (iCol = 0; iCol < nColMax; iCol++ ) { - for (iTime = 0; iTime < nTimeMax; iTime++) { + // Don't create noise in detectors that are switched off + if (CheckDetector(plane,chamber,sector)) { + + // Create the digits for this chamber + for (iRow = 0; iRow < nRowMax; iRow++ ) { + for (iCol = 0; iCol < nColMax; iCol++ ) { + + // Create summable digits + if (fSDigits) { + + for (iTime = 0; iTime < nTimeTotal; iTime++) { + Float_t signalAmp = signals->GetDataUnchecked(iRow,iCol,iTime); + signalAmp *= fSDigitsScale; + signalAmp = TMath::Min(signalAmp,(Float_t) 1.0e9); + Int_t adc = (Int_t) signalAmp; + if (adc > 0) nDigits++; + digits->SetDataUnchecked(iRow,iCol,iTime,adc); + } + + } + // Create normal digits + else { + + for (iTime = 0; iTime < nTimeTotal; iTime++) { + Float_t signalAmp = signals->GetDataUnchecked(iRow,iCol,iTime); + // Pad and time coupling + signalAmp *= coupling; + // Add the noise, starting from minus ADC baseline in electrons + Double_t baselineEl = fPar->GetADCbaseline() * (fPar->GetADCinRange() + / fPar->GetADCoutRange()) + / convert; + signalAmp = TMath::Max((Double_t) gRandom->Gaus(signalAmp,fPar->GetNoise()) + ,-baselineEl); + // Convert to mV + signalAmp *= convert; + // Add ADC baseline in mV + signalAmp += fPar->GetADCbaseline() * (fPar->GetADCinRange() + / fPar->GetADCoutRange()); + // Convert to ADC counts. Set the overflow-bit fADCoutRange if the + // signal is larger than fADCinRange + Int_t adc = 0; + if (signalAmp >= fPar->GetADCinRange()) { + adc = ((Int_t) fPar->GetADCoutRange()); + } + else { + adc = ((Int_t) (signalAmp * (fPar->GetADCoutRange() + / fPar->GetADCinRange()))); + } + inADC[iTime] = adc; + outADC[iTime] = adc; + } + + // Apply the tail cancelation via the digital filter + if (fPar->TCOn()) { + DeConvExp(inADC,outADC,nTimeTotal,fPar->GetTCnexp()); + } + + for (iTime = 0; iTime < nTimeTotal; iTime++) { + // Store the amplitude of the digit if above threshold + if (outADC[iTime] > fPar->GetADCthreshold()) { + if (fDebug > 2) { + printf(" iRow = %d, iCol = %d, iTime = %d, adc = %f\n" + ,iRow,iCol,iTime,outADC[iTime]); + } + nDigits++; + digits->SetDataUnchecked(iRow,iCol,iTime,((Int_t) outADC[iTime])); + } + } + + } + + } + } + + } + + // Compress the arrays + if (!fSimpleSim) { + digits->Compress(1,0); + for (iDict = 0; iDict < kNDict; iDict++) { + dictionary[iDict]->Compress(1,0); + } + + totalSizeDigits += digits->GetSize(); + totalSizeDict0 += dictionary[0]->GetSize(); + totalSizeDict1 += dictionary[1]->GetSize(); + totalSizeDict2 += dictionary[2]->GetSize(); + + Float_t nPixel = nRowMax * nColMax * nTimeMax; + if (fDebug > 0) { + printf(" "); + printf("Found %d digits in detector %d (%3.0f).\n" + ,nDigits,iDet + ,100.0 * ((Float_t) nDigits) / nPixel); + } + + if (fCompress) signals->Compress(1,0); + + } - Float_t signalAmp = signals->GetData(iRow,iCol,iTime); + delete [] inADC; + delete [] outADC; - // Add the noise - signalAmp = TMath::Max((Double_t) gRandom->Gaus(signalAmp,fNoise),0.0); - // Convert to fC - signalAmp *= kEl2fC; + } + + if (signalsArray) { + delete signalsArray; + signalsArray = 0; + } + + if (fDebug > 0) { + printf(" "); + printf("Total number of analyzed hits = %d\n",countHits); + if (!fSimpleSim) { + printf(" "); + printf("Total digits data size = %d, %d, %d, %d\n",totalSizeDigits + ,totalSizeDict0 + ,totalSizeDict1 + ,totalSizeDict2); + } + } + + return kTRUE; + +} + +//_____________________________________________________________________________ +void AliTRDdigitizer::AddSDigitsManager(AliTRDdigitsManager *man) +{ + // + // Add a digits manager for s-digits to the input list. + // + + fSDigitsManagerList->Add(man); + +} + +//_____________________________________________________________________________ +void AliTRDdigitizer::DeleteSDigitsManager() +{ + // + // Removes digits manager from the input list. + // + + fSDigitsManagerList->Delete(); + +} + +//_____________________________________________________________________________ +Bool_t AliTRDdigitizer::ConvertSDigits() +{ + // + // Converts s-digits to normal digits + // + + // Number of track dictionary arrays + const Int_t kNDict = AliTRDdigitsManager::kNDict; + + // Converts number of electrons to fC + const Double_t kEl2fC = 1.602E-19 * 1.0E15; + + Int_t iDict = 0; + Int_t iRow; + Int_t iCol; + Int_t iTime; + + if (!fPar) { + fPar = new AliTRDparameter("TRDparameter","Standard parameter"); + if (fDebug > 0) { + printf(" "); + printf("Create the default parameter object\n"); + } + } + + Double_t sDigitsScale = 1.0 / GetSDigitsScale(); + Double_t noise = fPar->GetNoise(); + Double_t padCoupling = fPar->GetPadCoupling(); + Double_t timeCoupling = fPar->GetTimeCoupling(); + Double_t chipGain = fPar->GetChipGain(); + Double_t coupling = padCoupling * timeCoupling; + Double_t convert = kEl2fC * chipGain; + Double_t adcInRange = fPar->GetADCinRange(); + Double_t adcOutRange = fPar->GetADCoutRange(); + Int_t adcThreshold = fPar->GetADCthreshold(); + Int_t adcBaseline = fPar->GetADCbaseline(); + + AliTRDdataArrayI *digitsIn; + AliTRDdataArrayI *digitsOut; + AliTRDdataArrayI *dictionaryIn[kNDict]; + AliTRDdataArrayI *dictionaryOut[kNDict]; + + // Loop through the detectors + for (Int_t iDet = 0; iDet < AliTRDgeometry::Ndet(); iDet++) { + + if (fDebug > 0) { + printf(" "); + printf("Convert detector %d to digits.\n",iDet); + } + + Int_t plane = fGeo->GetPlane(iDet); + Int_t sector = fGeo->GetSector(iDet); + Int_t chamber = fGeo->GetChamber(iDet); + Int_t nRowMax = fPar->GetRowMax(plane,chamber,sector); + Int_t nColMax = fPar->GetColMax(plane); + Int_t nTimeTotal = fPar->GetTimeTotal(); + + Double_t *inADC = new Double_t[nTimeTotal]; + Double_t *outADC = new Double_t[nTimeTotal]; + + digitsIn = fSDigitsManager->GetDigits(iDet); + digitsIn->Expand(); + digitsOut = fDigitsManager->GetDigits(iDet); + digitsOut->Allocate(nRowMax,nColMax,nTimeTotal); + for (iDict = 0; iDict < kNDict; iDict++) { + dictionaryIn[iDict] = fSDigitsManager->GetDictionary(iDet,iDict); + dictionaryIn[iDict]->Expand(); + dictionaryOut[iDict] = fDigitsManager->GetDictionary(iDet,iDict); + dictionaryOut[iDict]->Allocate(nRowMax,nColMax,nTimeTotal); + } + + for (iRow = 0; iRow < nRowMax; iRow++ ) { + for (iCol = 0; iCol < nColMax; iCol++ ) { + + for (iTime = 0; iTime < nTimeTotal; iTime++) { + Double_t signal = (Double_t) digitsIn->GetDataUnchecked(iRow,iCol,iTime); + signal *= sDigitsScale; + // Pad and time coupling + signal *= coupling; + // Add the noise, starting from minus ADC baseline in electrons + Double_t baselineEl = adcBaseline * (adcInRange / adcOutRange) / convert; + signal = TMath::Max((Double_t) gRandom->Gaus(signal,noise),-baselineEl); // Convert to mV - signalAmp *= fChipGain; - // Convert to ADC counts. Set the overflow-bit fADCoutRange if the - // signal is larger than fADCinRange + signal *= convert; + // add ADC baseline in mV + signal += adcBaseline * (adcInRange / adcOutRange); + // Convert to ADC counts. Set the overflow-bit adcOutRange if the + // signal is larger than adcInRange Int_t adc = 0; - if (signalAmp >= fADCinRange) { - adc = ((Int_t) fADCoutRange); + if (signal >= adcInRange) { + adc = ((Int_t) adcOutRange); } else { - adc = ((Int_t) (signalAmp * (fADCoutRange / fADCinRange))); + adc = ((Int_t) (signal * (adcOutRange / adcInRange))); } + inADC[iTime] = adc; + outADC[iTime] = adc; + } + + // Apply the tail cancelation via the digital filter + if (fPar->TCOn()) { + DeConvExp(inADC,outADC,nTimeTotal,fPar->GetTCnexp()); + } + for (iTime = 0; iTime < nTimeTotal; iTime++) { // Store the amplitude of the digit if above threshold - if (adc > fADCthreshold) { - if (fVerbose > 2) { - printf(" iRow = %d, iCol = %d, iTime = %d\n" - ,iRow,iCol,iTime); - printf(" signal = %f, adc = %d\n",signalAmp,adc); + if (outADC[iTime] > adcThreshold) { + digitsOut->SetDataUnchecked(iRow,iCol,iTime,((Int_t) outADC[iTime])); + // Copy the dictionary + for (iDict = 0; iDict < kNDict; iDict++) { + Int_t track = dictionaryIn[iDict]->GetDataUnchecked(iRow,iCol,iTime); + dictionaryOut[iDict]->SetDataUnchecked(iRow,iCol,iTime,track); } - nDigits++; - digits->SetData(iRow,iCol,iTime,adc); } - } + } } - // Compress the arrays - digits->Compress(1,0); - for (iDict = 0; iDict < kNDict; iDict++) { - dictionary[iDict]->Compress(1,0); + if (fCompress) { + digitsIn->Compress(1,0); + digitsOut->Compress(1,0); + for (iDict = 0; iDict < kNDict; iDict++) { + dictionaryIn[iDict]->Compress(1,0); + dictionaryOut[iDict]->Compress(1,0); + } } - totalSizeDigits += digits->GetSize(); - totalSizeDict0 += dictionary[0]->GetSize(); - totalSizeDict1 += dictionary[1]->GetSize(); - totalSizeDict2 += dictionary[2]->GetSize(); + delete [] inADC; + delete [] outADC; - Float_t nPixel = nRowMax * nColMax * nTimeMax; - printf("AliTRDdigitizer::MakeDigits -- "); - printf("Found %d digits in detector %d (%3.0f).\n" - ,nDigits,iDet - ,100.0 * ((Float_t) nDigits) / nPixel); - - if (fCompress) signals->Compress(1,0); + } + + return kTRUE; +} + +//_____________________________________________________________________________ +Bool_t AliTRDdigitizer::MergeSDigits() +{ + // + // Merges the input s-digits: + // - The amplitude of the different inputs are summed up. + // - Of the track IDs from the input dictionaries only one is + // kept for each input. This works for maximal 3 different merged inputs. + // + + // Number of track dictionary arrays + const Int_t kNDict = AliTRDdigitsManager::kNDict; + + if (!fPar) { + fPar = new AliTRDparameter("TRDparameter","Standard parameter"); + if (fDebug > 0) { + printf(" "); + printf("Create the default parameter object\n"); + } } - printf("AliTRDdigitizer::MakeDigits -- "); - printf("Total number of analyzed hits = %d\n",countHits); + Int_t iDict = 0; + Int_t jDict = 0; + + AliTRDdataArrayI *digitsA; + AliTRDdataArrayI *digitsB; + AliTRDdataArrayI *dictionaryA[kNDict]; + AliTRDdataArrayI *dictionaryB[kNDict]; - printf("AliTRDdigitizer::MakeDigits -- "); - printf("Total digits data size = %d, %d, %d, %d\n",totalSizeDigits - ,totalSizeDict0 - ,totalSizeDict1 - ,totalSizeDict2); + // Get the first s-digits + fSDigitsManager = (AliTRDdigitsManager *) fSDigitsManagerList->First(); + if (!fSDigitsManager) return kFALSE; + + // Loop through the other sets of s-digits + AliTRDdigitsManager *mergeSDigitsManager; + mergeSDigitsManager = (AliTRDdigitsManager *) + fSDigitsManagerList->After(fSDigitsManager); + + if (fDebug > 0) { + if (mergeSDigitsManager) { + printf(" "); + printf("Merge %d input files.\n",fSDigitsManagerList->GetSize()); + } + else { + printf(" "); + printf("Only one input file.\n"); + } + } + + Int_t iMerge = 0; + while (mergeSDigitsManager) { + + iMerge++; + + // Loop through the detectors + for (Int_t iDet = 0; iDet < AliTRDgeometry::Ndet(); iDet++) { + + Int_t plane = fGeo->GetPlane(iDet); + Int_t sector = fGeo->GetSector(iDet); + Int_t chamber = fGeo->GetChamber(iDet); + Int_t nRowMax = fPar->GetRowMax(plane,chamber,sector); + Int_t nColMax = fPar->GetColMax(plane); + Int_t nTimeTotal = fPar->GetTimeTotal(); + + // Loop through the pixels of one detector and add the signals + digitsA = fSDigitsManager->GetDigits(iDet); + digitsB = mergeSDigitsManager->GetDigits(iDet); + digitsA->Expand(); + digitsB->Expand(); + for (iDict = 0; iDict < kNDict; iDict++) { + dictionaryA[iDict] = fSDigitsManager->GetDictionary(iDet,iDict); + dictionaryB[iDict] = mergeSDigitsManager->GetDictionary(iDet,iDict); + dictionaryA[iDict]->Expand(); + dictionaryB[iDict]->Expand(); + } + + // Merge only detectors that contain a signal + Bool_t doMerge = kTRUE; + if (fMergeSignalOnly) { + if (digitsA->GetOverThreshold(0) == 0) { + doMerge = kFALSE; + } + } + + if (doMerge) { + + if (fDebug > 0) { + printf(" "); + printf("Merge detector %d of input no.%d\n",iDet,iMerge+1); + } + + for (Int_t iRow = 0; iRow < nRowMax; iRow++ ) { + for (Int_t iCol = 0; iCol < nColMax; iCol++ ) { + for (Int_t iTime = 0; iTime < nTimeTotal; iTime++) { + + // Add the amplitudes of the summable digits + Int_t ampA = digitsA->GetDataUnchecked(iRow,iCol,iTime); + Int_t ampB = digitsB->GetDataUnchecked(iRow,iCol,iTime); + ampA += ampB; + digitsA->SetDataUnchecked(iRow,iCol,iTime,ampA); + + // Add the mask to the track id if defined. + for (iDict = 0; iDict < kNDict; iDict++) { + Int_t trackB = dictionaryB[iDict]->GetDataUnchecked(iRow,iCol,iTime); + if ((fMasks) && (trackB > 0)) { + for (jDict = 0; jDict < kNDict; jDict++) { + Int_t trackA = dictionaryA[iDict]->GetDataUnchecked(iRow,iCol,iTime); + if (trackA == 0) { + trackA = trackB + fMasks[iMerge]; + dictionaryA[iDict]->SetDataUnchecked(iRow,iCol,iTime,trackA); + } + } + } + } + + } + } + } + + } + + if (fCompress) { + digitsA->Compress(1,0); + digitsB->Compress(1,0); + for (iDict = 0; iDict < kNDict; iDict++) { + dictionaryA[iDict]->Compress(1,0); + dictionaryB[iDict]->Compress(1,0); + } + } + + } + + // The next set of s-digits + mergeSDigitsManager = (AliTRDdigitsManager *) + fSDigitsManagerList->After(mergeSDigitsManager); + + } return kTRUE; } +//_____________________________________________________________________________ +Bool_t AliTRDdigitizer::SDigits2Digits() +{ + // + // Merges the input s-digits and converts them to normal digits + // + + if (!MergeSDigits()) return kFALSE; + + return ConvertSDigits(); + +} + //_____________________________________________________________________________ Bool_t AliTRDdigitizer::CheckDetector(Int_t plane, Int_t chamber, Int_t sector) { @@ -1048,6 +1502,8 @@ Bool_t AliTRDdigitizer::CheckDetector(Int_t plane, Int_t chamber, Int_t sector) // Checks whether a detector is enabled // + if (fSimpleSim) return kTRUE; + if ((fTRD->GetSensChamber() >= 0) && (fTRD->GetSensChamber() != chamber)) return kFALSE; if ((fTRD->GetSensPlane() >= 0) && @@ -1070,63 +1526,125 @@ Bool_t AliTRDdigitizer::CheckDetector(Int_t plane, Int_t chamber, Int_t sector) } //_____________________________________________________________________________ -Bool_t AliTRDdigitizer::WriteDigits() +Bool_t AliTRDdigitizer::WriteDigits() const { // // Writes out the TRD-digits and the dictionaries // - // Create the branches - if (!(gAlice->TreeD()->GetBranch("TRDdigits"))) { - if (!fDigits->MakeBranch()) return kFALSE; - } + //Write parameters + fRunLoader->CdGAFile(); + if (!gFile->Get("TRDparameter")) GetParameter()->Write(); // Store the digits and the dictionary in the tree - fDigits->WriteDigits(); - - // Write the new tree into the input file (use overwrite option) - Char_t treeName[15]; - sprintf(treeName,"TreeD%d",fEvent); - printf("AliTRDdigitizer::WriteDigits -- "); - printf("Write the digits tree %s for event %d.\n" - ,treeName,fEvent); - gAlice->TreeD()->Write(treeName,TObject::kOverwrite); - - return kTRUE; + return fDigitsManager->WriteDigits(); } //_____________________________________________________________________________ -void AliTRDdigitizer::SetPRF(TF1 *prf) +void AliTRDdigitizer::DeConvExp(Double_t *source, Double_t *target + , Int_t n, Int_t nexp) { // - // Defines a new pad response function + // Does the deconvolution by the digital filter. + // + // Author: Marcus Gutfleisch, KIP Heidelberg + // Optimized for: New TRF from Venelin Angelov, simulated with CADENCE + // Pad-ground capacitance = 25 pF + // Pad-pad cross talk capacitance = 6 pF + // For 10 MHz digitization, corresponding to 20 time bins + // in the drift region // - if (fPRF) delete fPRF; - fPRF = prf; + Double_t rates[2]; + Double_t coefficients[2]; + Double_t Dt = 0.100; // time bin width [mus] for 10 MHz sampling frequence -} + /* initialize (coefficient = alpha, rates = lambda) */ + + rates[0] = 0.7680; + rates[1] = 0.0995; -//_____________________________________________________________________________ -void AliTRDdigitizer::SetTRF(TF1 *trf) -{ - // - // Defines a new time response function - // + rates[0] = TMath::Exp(-Dt/(rates[0])); + rates[1] = TMath::Exp(-Dt/(rates[1])); - if (fTRF) delete fTRF; - fTRF = trf; + // dummy initialization + coefficients[0] = 0.0000; + coefficients[1] = 0.0000; + if( nexp == 1 ) { + coefficients[0] = 0.0844; + coefficients[1] = 0.0000; + } + if( nexp == 2 ) { + coefficients[0] = 0.1445; + coefficients[1] = 0.7524; + } + + Int_t i, k; + Double_t reminder[2]; + Double_t correction, result; + + /* attention: computation order is important */ + correction=0.0; + for ( k = 0; k < nexp; k++ ) reminder[k]=0.0; + + for ( i = 0; i < n; i++ ) { + result = ( source[i] - correction ); /* no rescaling */ + target[i] = result; + + for ( k = 0; k < nexp; k++ ) reminder[k] = rates[k] + * ( reminder[k] + coefficients[k] * result); + + correction=0.0; + for ( k = 0; k < nexp; k++ ) correction += reminder[k]; + } + } //_____________________________________________________________________________ -Double_t TRFlandau(Double_t *x, Double_t *par) +void AliTRDdigitizer::InitOutput(Int_t iEvent) { + // + // Initializes the output branches + // - Double_t xx = x[0]; - Double_t landau = par[0] * TMath::Landau(xx,par[1],par[2]); - - return landau; + fEvent = iEvent; + + if (!fRunLoader) + { + Error("InitOutput","Run Loader is NULL"); + return; + } + AliLoader* loader = fRunLoader->GetLoader("TRDLoader"); + if (!loader) + { + Error("Open","Can not get TRD loader from Run Loader"); + return; + } + + TTree* tree = 0; + + if (fSDigits) + { + //if we produce SDigits + tree = loader->TreeS(); + if (!tree) + { + loader->MakeTree("S"); + tree = loader->TreeS(); + } + } + else + {//if we produce Digits + tree = loader->TreeD(); + if (!tree) + { + loader->MakeTree("D"); + tree = loader->TreeD(); + } + } + fDigitsManager->SetEvent(iEvent); + fDigitsManager->MakeBranch(tree); }