#include <TGeoManager.h>
#include <TList.h>
#include <TMath.h>
#include <TRandom.h>
#include <TTree.h>
#include "AliRun.h"
#include "AliMC.h"
#include "AliRunLoader.h"
#include "AliLoader.h"
#include "AliConfig.h"
#include "AliDigitizationInput.h"
#include "AliRunLoader.h"
#include "AliLoader.h"
#include "AliLog.h"
#include "AliTRD.h"
#include "AliTRDhit.h"
#include "AliTRDdigitizer.h"
#include "AliTRDarrayDictionary.h"
#include "AliTRDarrayADC.h"
#include "AliTRDarraySignal.h"
#include "AliTRDdigitsManager.h"
#include "AliTRDgeometry.h"
#include "AliTRDpadPlane.h"
#include "AliTRDcalibDB.h"
#include "AliTRDSimParam.h"
#include "AliTRDCommonParam.h"
#include "AliTRDfeeParam.h"
#include "AliTRDmcmSim.h"
#include "AliTRDdigitsParam.h"
#include "Cal/AliTRDCalROC.h"
#include "Cal/AliTRDCalDet.h"
#include "Cal/AliTRDCalOnlineGainTableROC.h"
ClassImp(AliTRDdigitizer)
AliTRDdigitizer::AliTRDdigitizer()
:AliDigitizer()
,fRunLoader(0)
,fDigitsManager(0)
,fSDigitsManager(0)
,fSDigitsManagerList(0)
,fTRD(0)
,fGeo(0)
,fMcmSim(new AliTRDmcmSim)
,fEvent(0)
,fMasks(0)
,fCompress(kTRUE)
,fSDigits(kFALSE)
,fMergeSignalOnly(kFALSE)
{
}
AliTRDdigitizer::AliTRDdigitizer(const Text_t *name, const Text_t *title)
:AliDigitizer(name,title)
,fRunLoader(0)
,fDigitsManager(0)
,fSDigitsManager(0)
,fSDigitsManagerList(0)
,fTRD(0)
,fGeo(0)
,fMcmSim(new AliTRDmcmSim)
,fEvent(0)
,fMasks(0)
,fCompress(kTRUE)
,fSDigits(kFALSE)
,fMergeSignalOnly(kFALSE)
{
}
AliTRDdigitizer::AliTRDdigitizer(AliDigitizationInput* digInput
, const Text_t *name, const Text_t *title)
:AliDigitizer(digInput,name,title)
,fRunLoader(0)
,fDigitsManager(0)
,fSDigitsManager(0)
,fSDigitsManagerList(0)
,fTRD(0)
,fGeo(0)
,fMcmSim(new AliTRDmcmSim)
,fEvent(0)
,fMasks(0)
,fCompress(kTRUE)
,fSDigits(kFALSE)
,fMergeSignalOnly(kFALSE)
{
}
AliTRDdigitizer::AliTRDdigitizer(AliDigitizationInput* digInput)
:AliDigitizer(digInput,"AliTRDdigitizer","TRD digitizer")
,fRunLoader(0)
,fDigitsManager(0)
,fSDigitsManager(0)
,fSDigitsManagerList(0)
,fTRD(0)
,fGeo(0)
,fMcmSim(new AliTRDmcmSim)
,fEvent(0)
,fMasks(0)
,fCompress(kTRUE)
,fSDigits(kFALSE)
,fMergeSignalOnly(kFALSE)
{
}
AliTRDdigitizer::AliTRDdigitizer(const AliTRDdigitizer &d)
:AliDigitizer(d)
,fRunLoader(0)
,fDigitsManager(0)
,fSDigitsManager(0)
,fSDigitsManagerList(0)
,fTRD(0)
,fGeo(0)
,fMcmSim(new AliTRDmcmSim)
,fEvent(0)
,fMasks(0)
,fCompress(d.fCompress)
,fSDigits(d.fSDigits)
,fMergeSignalOnly(d.fMergeSignalOnly)
{
}
AliTRDdigitizer::~AliTRDdigitizer()
{
delete fDigitsManager;
fDigitsManager = 0;
fSDigitsManager = 0;
if (fSDigitsManagerList) {
fSDigitsManagerList->Delete();
delete fSDigitsManagerList;
}
fSDigitsManagerList = 0;
delete [] fMasks;
fMasks = 0;
delete fMcmSim;
fMcmSim = 0;
delete fGeo;
fGeo = 0;
}
AliTRDdigitizer &AliTRDdigitizer::operator=(const AliTRDdigitizer &d)
{
if (this != &d) {
((AliTRDdigitizer &) d).Copy(*this);
}
return *this;
}
void AliTRDdigitizer::Copy(TObject &d) const
{
((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).fEvent = 0;
((AliTRDdigitizer &) d).fMasks = 0;
((AliTRDdigitizer &) d).fCompress = fCompress;
((AliTRDdigitizer &) d).fSDigits = fSDigits;
((AliTRDdigitizer &) d).fMergeSignalOnly = fMergeSignalOnly;
}
void AliTRDdigitizer::Digitize(const Option_t* option)
{
Int_t iInput;
AliTRDdigitsManager *sdigitsManager;
TString optionString = option;
if (optionString.Contains("deb")) {
AliLog::SetClassDebugLevel("AliTRDdigitizer",1);
AliInfo("Called with debug option");
}
AliRunLoader *inrl = 0x0;
if (gAlice) {
AliDebug(1,"AliRun object found on file.");
}
else {
inrl = AliRunLoader::GetRunLoader(fDigInput->GetInputFolderName(0));
inrl->LoadgAlice();
gAlice = inrl->GetAliRun();
if (!gAlice) {
AliError("Could not find AliRun object.");
return;
}
}
Int_t nInput = fDigInput->GetNinputs();
fMasks = new Int_t[nInput];
for (iInput = 0; iInput < nInput; iInput++) {
fMasks[iInput] = fDigInput->GetMask(iInput);
}
AliRunLoader *orl = AliRunLoader::GetRunLoader(fDigInput->GetOutputFolderName());
if (InitDetector()) {
AliLoader *ogime = orl->GetLoader("TRDLoader");
TTree *tree = 0;
if (fSDigits) {
tree = ogime->TreeS();
if (!tree) {
ogime->MakeTree("S");
tree = ogime->TreeS();
}
}
else {
tree = ogime->TreeD();
if (!tree) {
ogime->MakeTree("D");
tree = ogime->TreeD();
}
}
MakeBranch(tree);
}
for (iInput = 0; iInput < nInput; iInput++) {
AliDebug(1,Form("Add input stream %d",iInput));
inrl = AliRunLoader::GetRunLoader(fDigInput->GetInputFolderName(iInput));
AliLoader *gime = inrl->GetLoader("TRDLoader");
TTree *treees = gime->TreeS();
if (treees == 0x0) {
if (gime->LoadSDigits()) {
AliError(Form("Error Occured while loading S. Digits for input %d.",iInput));
return;
}
treees = gime->TreeS();
}
if (treees == 0x0) {
AliError(Form("Input stream %d does not exist",iInput));
return;
}
sdigitsManager = new AliTRDdigitsManager();
sdigitsManager->SetSDigits(kTRUE);
AliRunLoader *rl = AliRunLoader::GetRunLoader(fDigInput->GetInputFolderName(iInput));
AliLoader *gimme = rl->GetLoader("TRDLoader");
if (!gimme->TreeS())
{
gimme->LoadSDigits();
}
sdigitsManager->ReadDigits(gimme->TreeS());
AddSDigitsManager(sdigitsManager);
}
AliDebug(1,"Do the conversion");
SDigits2Digits();
AliDebug(1,"Write the digits");
fDigitsManager->WriteDigits();
orl->CdGAFile();
DeleteSDigitsManager();
AliDebug(1,"Done");
}
Bool_t AliTRDdigitizer::Open(const Char_t *file, Int_t nEvent)
{
TString evfoldname = AliConfig::GetDefaultEventFolderName();
fRunLoader = AliRunLoader::GetRunLoader(evfoldname);
if (!fRunLoader) {
fRunLoader = AliRunLoader::Open(file,evfoldname,"UPDATE");
}
if (!fRunLoader) {
AliError(Form("Can not open session for file %s.",file));
return kFALSE;
}
if (!fRunLoader->GetAliRun()) {
fRunLoader->LoadgAlice();
}
gAlice = fRunLoader->GetAliRun();
if (gAlice) {
AliDebug(1,"AliRun object found on file.");
}
else {
AliError("Could not find AliRun object.");
return kFALSE;
}
fEvent = nEvent;
AliLoader *loader = fRunLoader->GetLoader("TRDLoader");
if (!loader) {
AliError("Can not get TRD loader from Run Loader");
return kFALSE;
}
if (InitDetector()) {
TTree *tree = 0;
if (fSDigits) {
tree = loader->TreeS();
if (!tree) {
loader->MakeTree("S");
tree = loader->TreeS();
}
}
else {
tree = loader->TreeD();
if (!tree) {
loader->MakeTree("D");
tree = loader->TreeD();
}
}
return MakeBranch(tree);
}
else {
return kFALSE;
}
}
Bool_t AliTRDdigitizer::Open(AliRunLoader * const runLoader, Int_t nEvent)
{
fRunLoader = runLoader;
if (!fRunLoader) {
AliError("RunLoader does not exist");
return kFALSE;
}
if (!fRunLoader->GetAliRun()) {
fRunLoader->LoadgAlice();
}
gAlice = fRunLoader->GetAliRun();
if (gAlice) {
AliDebug(1,"AliRun object found on file.");
}
else {
AliError("Could not find AliRun object.");
return kFALSE;
}
fEvent = nEvent;
AliLoader *loader = fRunLoader->GetLoader("TRDLoader");
if (!loader) {
AliError("Can not get TRD loader from Run Loader");
return kFALSE;
}
if (InitDetector()) {
TTree *tree = 0;
if (fSDigits) {
tree = loader->TreeS();
if (!tree) {
loader->MakeTree("S");
tree = loader->TreeS();
}
}
else {
tree = loader->TreeD();
if (!tree) {
loader->MakeTree("D");
tree = loader->TreeD();
}
}
return MakeBranch(tree);
}
else {
return kFALSE;
}
}
Bool_t AliTRDdigitizer::InitDetector()
{
fTRD = (AliTRD *) gAlice->GetDetector("TRD");
if (!fTRD) {
AliFatal("No TRD module found");
exit(1);
}
if (fTRD->IsVersion() != 1) {
AliFatal("TRD must be version 1 (slow simulator)");
exit(1);
}
fGeo = new AliTRDgeometry();
if (fDigitsManager) {
delete fDigitsManager;
}
fDigitsManager = new AliTRDdigitsManager();
fDigitsManager->SetSDigits(fSDigits);
fDigitsManager->CreateArrays();
fDigitsManager->SetEvent(fEvent);
if (fSDigitsManagerList) {
fSDigitsManagerList->Delete();
}
else {
fSDigitsManagerList = new TList();
}
return kTRUE;
}
Bool_t AliTRDdigitizer::MakeBranch(TTree *tree) const
{
return fDigitsManager->MakeBranch(tree);
}
void AliTRDdigitizer::AddSDigitsManager(AliTRDdigitsManager *man)
{
fSDigitsManagerList->Add(man);
}
void AliTRDdigitizer::DeleteSDigitsManager()
{
fSDigitsManagerList->Delete();
}
Bool_t AliTRDdigitizer::MakeDigits()
{
AliDebug(1,"Start creating digits");
if (!fGeo) {
AliError("No geometry defined");
return kFALSE;
}
AliTRDcalibDB *calibration = AliTRDcalibDB::Instance();
if (!calibration) {
AliFatal("Could not get calibration object");
return kFALSE;
}
const Int_t kNdet = AliTRDgeometry::Ndet();
Float_t **hits = new Float_t*[kNdet];
Int_t *nhit = new Int_t[kNdet];
memset(nhit,0,kNdet*sizeof(Int_t));
AliTRDarraySignal *signals = 0x0;
if (AliTRDSimParam::Instance()->GetNTBoverwriteOCDB()) {
if (calibration->GetNumberOfTimeBinsDCS() != AliTRDSimParam::Instance()->GetNTimeBins()) {
AliWarning(Form("Number of time bins is different to OCDB value [SIM=%d, OCDB=%d]"
,AliTRDSimParam::Instance()->GetNTimeBins()
,calibration->GetNumberOfTimeBinsDCS()));
}
fDigitsManager->GetDigitsParam()->SetNTimeBinsAll(AliTRDSimParam::Instance()->GetNTimeBins());
}
else {
Int_t nTB = calibration->GetNumberOfTimeBinsDCS();
if (nTB < 0) {
AliError("No useful DCS information available for this run! Using standard values.");
nTB = AliTRDSimParam::Instance()->GetNTimeBins();
}
fDigitsManager->GetDigitsParam()->SetNTimeBinsAll(nTB);
}
fDigitsManager->GetDigitsParam()->SetADCbaselineAll(AliTRDSimParam::Instance()->GetADCbaseline());
if (!SortHits(hits,nhit)) {
AliError("Sorting hits failed");
delete [] hits;
delete [] nhit;
return kFALSE;
}
for (Int_t det = 0; det < kNdet; det++) {
if ((!calibration->IsChamberNoData(det)) &&
( fGeo->ChamberInGeometry(det)) &&
(nhit[det] > 0)) {
signals = new AliTRDarraySignal();
if (!ConvertHits(det,hits[det],nhit[det],signals)) {
AliError(Form("Conversion of hits failed for detector=%d",det));
delete [] hits;
delete [] nhit;
delete signals;
signals = 0x0;
return kFALSE;
}
if (!ConvertSignals(det,signals)) {
AliError(Form("Conversion of signals failed for detector=%d",det));
delete [] hits;
delete [] nhit;
delete signals;
signals = 0x0;
return kFALSE;
}
delete signals;
signals = 0x0;
}
delete [] hits[det];
}
if (!fSDigits) {
if (AliDataLoader *trklLoader
= AliRunLoader::Instance()->GetLoader("TRDLoader")->GetDataLoader("tracklets")) {
if (trklLoader->Tree())
trklLoader->WriteData("OVERWRITE");
}
}
delete [] hits;
delete [] nhit;
return kTRUE;
}
Bool_t AliTRDdigitizer::SortHits(Float_t **hits, Int_t *nhit)
{
AliDebug(1,"Start sorting hits");
const Int_t kNdet = AliTRDgeometry::Ndet();
const Int_t kNhit = 6;
Float_t *xyz = 0;
Int_t nhitTrk = 0;
Int_t *lhit = new Int_t[kNdet];
memset(lhit,0,kNdet*sizeof(Int_t));
for (Int_t det = 0; det < kNdet; det++) {
hits[det] = 0x0;
}
AliLoader *gimme = fRunLoader->GetLoader("TRDLoader");
if (!gimme->TreeH()) {
gimme->LoadHits();
}
TTree *hitTree = gimme->TreeH();
if (hitTree == 0x0) {
AliError("Can not get TreeH");
delete [] lhit;
return kFALSE;
}
fTRD->SetTreeAddress();
Int_t nTrk = (Int_t) hitTree->GetEntries();
AliDebug(1,Form("Found %d tracks",nTrk));
for (Int_t iTrk = 0; iTrk < nTrk; iTrk++) {
gAlice->GetMCApp()->ResetHits();
hitTree->GetEvent(iTrk);
if (!fTRD->Hits()) {
AliError(Form("No hits array for track = %d",iTrk));
continue;
}
nhitTrk = fTRD->Hits()->GetEntriesFast();
Int_t hitCnt = 0;
AliTRDhit *hit = (AliTRDhit *) fTRD->FirstHit(-1);
while (hit) {
hitCnt++;
if (((Int_t) hit->GetCharge()) != 0) {
Int_t trk = hit->Track();
Int_t det = hit->GetDetector();
Int_t q = hit->GetCharge();
Float_t x = hit->X();
Float_t y = hit->Y();
Float_t z = hit->Z();
Float_t time = hit->GetTime();
if (nhit[det] == lhit[det]) {
xyz = new Float_t[kNhit*(nhitTrk+lhit[det])];
if (hits[det]) {
memcpy(xyz,hits[det],sizeof(Float_t)*kNhit*lhit[det]);
delete [] hits[det];
}
lhit[det] += nhitTrk;
hits[det] = xyz;
}
else {
xyz = hits[det];
}
xyz[nhit[det]*kNhit+0] = x;
xyz[nhit[det]*kNhit+1] = y;
xyz[nhit[det]*kNhit+2] = z;
xyz[nhit[det]*kNhit+3] = q;
xyz[nhit[det]*kNhit+4] = trk;
xyz[nhit[det]*kNhit+5] = time;
nhit[det]++;
}
hit = (AliTRDhit *) fTRD->NextHit();
}
}
delete [] lhit;
return kTRUE;
}
Bool_t AliTRDdigitizer::ConvertHits(Int_t det
, const Float_t * const hits
, Int_t nhit
, AliTRDarraySignal *signals)
{
AliDebug(1,Form("Start converting hits for detector=%d (nhits=%d)",det,nhit));
const Int_t kNpad = 3;
const Int_t kNdict = AliTRDdigitsManager::kNDict;
const Int_t kNhit = 6;
const Float_t kAmWidth = AliTRDgeometry::AmThick();
const Float_t kDrWidth = AliTRDgeometry::DrThick();
const Float_t kDrMin = - 0.5 * kAmWidth;
const Float_t kDrMax = kDrWidth + 0.5 * kAmWidth;
Int_t iPad = 0;
Int_t dict = 0;
Int_t timeBinTRFend = 1;
Double_t pos[3];
Double_t loc[3];
Double_t padSignal[kNpad];
Double_t signalOld[kNpad];
AliTRDarrayDictionary *dictionary[kNdict];
AliTRDSimParam *simParam = AliTRDSimParam::Instance();
AliTRDCommonParam *commonParam = AliTRDCommonParam::Instance();
AliTRDcalibDB *calibration = AliTRDcalibDB::Instance();
if (!commonParam) {
AliFatal("Could not get common parameterss");
return kFALSE;
}
if (!simParam) {
AliFatal("Could not get simulation parameters");
return kFALSE;
}
if (!calibration) {
AliFatal("Could not get calibration object");
return kFALSE;
}
AliTRDCalROC *calVdriftROC = 0;
Float_t calVdriftDetValue = 0.0;
const AliTRDCalDet *calVdriftDet = calibration->GetVdriftDet();
AliTRDCalROC *calT0ROC = 0;
Float_t calT0DetValue = 0.0;
const AliTRDCalDet *calT0Det = calibration->GetT0Det();
Double_t calExBDetValue = 0.0;
const AliTRDCalDet *calExBDet = calibration->GetExBDet();
if (simParam->TRFOn()) {
timeBinTRFend = ((Int_t) (simParam->GetTRFhi()
* commonParam->GetSamplingFrequency())) - 1;
}
Int_t nTimeTotal = fDigitsManager->GetDigitsParam()->GetNTimeBins(det);
Float_t samplingRate = commonParam->GetSamplingFrequency();
Float_t elAttachProp = simParam->GetElAttachProp() / 100.0;
AliTRDpadPlane *padPlane = fGeo->GetPadPlane(det);
Int_t layer = fGeo->GetLayer(det);
Float_t row0 = padPlane->GetRow0ROC();
Int_t nRowMax = padPlane->GetNrows();
Int_t nColMax = padPlane->GetNcols();
signals->Allocate(nRowMax,nColMax,nTimeTotal);
for (dict = 0; dict < kNdict; dict++) {
dictionary[dict] = (AliTRDarrayDictionary *) fDigitsManager->GetDictionary(det,dict);
dictionary[dict]->Allocate(nRowMax,nColMax,nTimeTotal);
}
for (Int_t hit = 0; hit < nhit; hit++) {
pos[0] = hits[hit*kNhit+0];
pos[1] = hits[hit*kNhit+1];
pos[2] = hits[hit*kNhit+2];
Float_t q = hits[hit*kNhit+3];
Float_t hittime = hits[hit*kNhit+5];
Int_t track = ((Int_t) hits[hit*kNhit+4]);
Int_t inDrift = 1;
gGeoManager->SetCurrentPoint(pos);
gGeoManager->FindNode();
if (strstr(gGeoManager->GetPath(),"/UK")) {
inDrift = 0;
}
calVdriftROC = calibration->GetVdriftROC(det);
calVdriftDetValue = calVdriftDet->GetValue(det);
calT0ROC = calibration->GetT0ROC(det);
calT0DetValue = calT0Det->GetValue(det);
calExBDetValue = calExBDet->GetValue(det);
gGeoManager->MasterToLocal(pos,loc);
if (inDrift) {
loc[2] = loc[2] - kDrWidth/2.0 - kAmWidth/2.0;
}
Double_t driftlength = -1.0 * loc[2];
if (q < 0.0) {
if ((loc[1] < padPlane->GetRowEndROC()) ||
(loc[1] > padPlane->GetRow0ROC())) {
continue;
}
if ((driftlength < kDrMin) ||
(driftlength > kDrMax)) {
continue;
}
}
Int_t rowE = padPlane->GetPadRowNumberROC(loc[1]);
if (rowE < 0) {
continue;
}
Double_t rowOffset = padPlane->GetPadRowOffsetROC(rowE,loc[1]);
Double_t offsetTilt = padPlane->GetTiltOffset(rowOffset);
Int_t colE = padPlane->GetPadColNumber(loc[0]+offsetTilt);
if (colE < 0) {
continue;
}
Double_t colOffset = 0.0;
Float_t driftvelocity = calVdriftDetValue * calVdriftROC->GetValue(colE,rowE);
Double_t absdriftlength = TMath::Abs(driftlength);
if (commonParam->ExBOn()) {
absdriftlength /= TMath::Sqrt(1.0 / (1.0 + calExBDetValue*calExBDetValue));
}
Int_t nEl = ((Int_t) TMath::Abs(q));
for (Int_t iEl = 0; iEl < nEl; iEl++) {
Double_t locC = loc[0];
Double_t locR = loc[1];
Double_t locT = loc[2];
if (simParam->ElAttachOn()) {
if (gRandom->Rndm() < (absdriftlength * elAttachProp)) {
continue;
}
}
if (simParam->DiffusionOn()) {
if (!(Diffusion(driftvelocity,absdriftlength,calExBDetValue,locR,locC,locT))) {
continue;
}
}
if (commonParam->ExBOn()) {
locC = locC + calExBDetValue * driftlength;
}
rowE = padPlane->GetPadRowNumberROC(locR);
if (rowE < 0) continue;
rowOffset = padPlane->GetPadRowOffsetROC(rowE,locR);
offsetTilt = padPlane->GetTiltOffset(rowOffset);
colE = padPlane->GetPadColNumber(locC+offsetTilt);
if (colE < 0) continue;
colOffset = padPlane->GetPadColOffset(colE,locC+offsetTilt);
driftvelocity = calVdriftDetValue * calVdriftROC->GetValue(colE,rowE);
Float_t t0 = calT0DetValue + calT0ROC->GetValue(colE,rowE);
Double_t drifttime;
if (simParam->TimeStructOn()) {
Double_t zz = row0 - locR + padPlane->GetAnodeWireOffset();
zz -= ((Int_t)(2 * zz)) / 2.0;
if (zz > 0.25) {
zz = 0.5 - zz;
}
drifttime = commonParam->TimeStruct(driftvelocity,0.5*kAmWidth-1.0*locT,zz)
+ hittime;
}
else {
drifttime = TMath::Abs(locT) / driftvelocity
+ hittime;
}
Double_t ggRndm = 0.0;
do {
ggRndm = gRandom->Rndm();
} while (ggRndm <= 0);
Double_t signal = -(simParam->GetGasGain()) * TMath::Log(ggRndm);
if (simParam->PRFOn()) {
Double_t dist = (colOffset - 0.5*padPlane->GetColSize(colE))
/ padPlane->GetColSize(colE);
if (!(calibration->PadResponse(signal,dist,layer,padSignal))) continue;
}
else {
padSignal[0] = 0.0;
padSignal[1] = signal;
padSignal[2] = 0.0;
}
Double_t timeBinIdeal = drifttime * samplingRate + t0;
if (TMath::Abs(timeBinIdeal) > 2*nTimeTotal) {
timeBinIdeal = 2 * nTimeTotal;
}
Int_t timeBinTruncated = ((Int_t) timeBinIdeal);
Double_t timeOffset = ((Float_t) timeBinTruncated
+ 0.5 - timeBinIdeal) / samplingRate;
for (Int_t iTimeBin = TMath::Max(timeBinTruncated,0)
;iTimeBin < TMath::Min(timeBinTruncated+timeBinTRFend,nTimeTotal)
;iTimeBin++) {
Double_t timeResponse = 1.0;
Double_t crossTalk = 0.0;
Double_t time = (iTimeBin - timeBinTruncated) / samplingRate + timeOffset;
if (simParam->TRFOn()) {
timeResponse = simParam->TimeResponse(time);
}
if (simParam->CTOn()) {
crossTalk = simParam->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;
signalOld[iPad] = signals->GetData(rowE,colPos,iTimeBin);
if (colPos != colE) {
signalOld[iPad] += padSignal[iPad]
* (timeResponse + crossTalk);
}
else {
signalOld[iPad] += padSignal[iPad]
* timeResponse;
}
signals->SetData(rowE,colPos,iTimeBin,signalOld[iPad]);
if (signalOld[iPad] > 0.0) {
for (dict = 0; dict < kNdict; dict++) {
Int_t oldTrack = dictionary[dict]->GetData(rowE,colPos,iTimeBin);
if (oldTrack == track) break;
if (oldTrack == -1 ) {
dictionary[dict]->SetData(rowE,colPos,iTimeBin,track);
break;
}
}
}
}
}
}
}
AliDebug(2,Form("Finished analyzing %d hits",nhit));
return kTRUE;
}
Bool_t AliTRDdigitizer::ConvertSignals(Int_t det, AliTRDarraySignal *signals)
{
AliDebug(1,Form("Start converting the signals for detector %d",det));
if (fSDigits) {
if (!Signal2SDigits(det,signals)) {
return kFALSE;
}
}
else {
if (!Signal2ADC(det,signals)) {
return kFALSE;
}
RunDigitalProcessing(det);
}
CompressOutputArrays(det);
return kTRUE;
}
Bool_t AliTRDdigitizer::Signal2ADC(Int_t det, AliTRDarraySignal *signals)
{
AliDebug(1,Form("Start converting signals to ADC values for detector=%d",det));
AliTRDcalibDB *calibration = AliTRDcalibDB::Instance();
if (!calibration) {
AliFatal("Could not get calibration object");
return kFALSE;
}
AliTRDSimParam *simParam = AliTRDSimParam::Instance();
if (!simParam) {
AliFatal("Could not get simulation parameters");
return kFALSE;
}
const Double_t kEl2fC = 1.602e-19 * 1.0e15;
Double_t coupling = simParam->GetPadCoupling()
* simParam->GetTimeCoupling();
Double_t convert = kEl2fC
* simParam->GetChipGain();
Double_t adcConvert = simParam->GetADCoutRange()
/ simParam->GetADCinRange();
Double_t baseline = simParam->GetADCbaseline()
/ adcConvert;
Double_t baselineEl = baseline
/ convert;
Int_t row = 0;
Int_t col = 0;
Int_t time = 0;
Int_t nRowMax = fGeo->GetPadPlane(det)->GetNrows();
Int_t nColMax = fGeo->GetPadPlane(det)->GetNcols();
Int_t nTimeTotal = fDigitsManager->GetDigitsParam()->GetNTimeBins(det);
if (fSDigitsManager->GetDigitsParam()->GetNTimeBins(det)) {
nTimeTotal = fSDigitsManager->GetDigitsParam()->GetNTimeBins(det);
}
else {
AliFatal("Could not get number of time bins");
return kFALSE;
}
const AliTRDCalDet *calGainFactorDet = calibration->GetGainFactorDet();
AliTRDCalROC *calGainFactorROC = 0x0;
Float_t calGainFactorDetValue = 0.0;
AliTRDarrayADC *digits = 0x0;
if (!signals) {
AliError(Form("Signals array for detector %d does not exist\n",det));
return kFALSE;
}
if (signals->HasData()) {
signals->Expand();
}
else {
signals->Allocate(nRowMax,nColMax,nTimeTotal);
}
if (fDigitsManager->HasSDigits()) {
AliError("Digits manager has s-digits");
return kFALSE;
}
digits = (AliTRDarrayADC *) fDigitsManager->GetDigits(det);
if (!digits->HasData()) {
digits->Allocate(nRowMax,nColMax,nTimeTotal);
}
calGainFactorROC = calibration->GetGainFactorROC(det);
calGainFactorDetValue = calGainFactorDet->GetValue(det);
for (row = 0; row < nRowMax; row++ ) {
for (col = 0; col < nColMax; col++ ) {
Int_t iMcm = (Int_t)(col/18);
Int_t halfchamberside = (iMcm>3 ? 1 : 0);
if (calibration->IsHalfChamberNoData(det, halfchamberside))
continue;
if (calibration->IsPadMasked(det,col,row) ||
calibration->IsPadNotConnected(det,col,row)) {
continue;
}
Float_t padgain = calGainFactorDetValue
* calGainFactorROC->GetValue(col,row);
if (padgain <= 0) {
AliError(Form("Not a valid gain %f, %d %d %d",padgain,det,col,row));
}
for (time = 0; time < nTimeTotal; time++) {
Float_t signalAmp = signals->GetData(row,col,time);
signalAmp *= coupling;
signalAmp *= padgain;
signalAmp = TMath::Max((Double_t) gRandom->Gaus(signalAmp,simParam->GetNoise())
,-baselineEl);
signalAmp *= convert;
signalAmp += baseline;
Short_t adc = 0;
if (signalAmp >= simParam->GetADCinRange()) {
adc = ((Short_t) simParam->GetADCoutRange());
}
else {
adc = TMath::Nint(signalAmp * adcConvert);
}
digits->SetData(row,col,time,adc);
}
}
}
return kTRUE;
}
Bool_t AliTRDdigitizer::Signal2SDigits(Int_t det, AliTRDarraySignal *signals)
{
AliDebug(1,Form("Start converting signals to s-digits for detector=%d",det));
AliTRDcalibDB *calibration = AliTRDcalibDB::Instance();
if (!calibration) {
AliFatal("Could not get calibration object");
return kFALSE;
}
Int_t row = 0;
Int_t col = 0;
Int_t time = 0;
Int_t nRowMax = fGeo->GetPadPlane(det)->GetNrows();
Int_t nColMax = fGeo->GetPadPlane(det)->GetNcols();
Int_t nTimeTotal = fDigitsManager->GetDigitsParam()->GetNTimeBins(det);
if (!fDigitsManager->HasSDigits()) {
AliError("Digits manager has no s-digits");
return kFALSE;
}
AliTRDarraySignal *digits = (AliTRDarraySignal *) fDigitsManager->GetSDigits(det);
if (!digits->HasData()) {
digits->Allocate(nRowMax,nColMax,nTimeTotal);
}
for (row = 0; row < nRowMax; row++ ) {
for (col = 0; col < nColMax; col++ ) {
Int_t iMcm = (Int_t)(col/18);
Int_t halfchamberside = (iMcm>3 ? 1 : 0);
if (calibration->IsHalfChamberNoData(det, halfchamberside))
continue;
for (time = 0; time < nTimeTotal; time++) {
digits->SetData(row,col,time,signals->GetData(row,col,time));
}
}
}
return kTRUE;
}
Bool_t AliTRDdigitizer::Digits2SDigits(AliTRDdigitsManager * const manDig
, AliTRDdigitsManager * const manSDig)
{
AliDebug(1,"Start converting digits to s-digits");
if (!fGeo) {
fGeo = new AliTRDgeometry();
}
AliTRDcalibDB *calibration = AliTRDcalibDB::Instance();
if (!calibration) {
AliFatal("Could not get calibration object");
return kFALSE;
}
AliTRDSimParam *simParam = AliTRDSimParam::Instance();
if (!simParam) {
AliFatal("Could not get simulation parameters");
return kFALSE;
}
const Double_t kEl2fC = 1.602e-19 * 1.0e15;
Double_t coupling = simParam->GetPadCoupling()
* simParam->GetTimeCoupling();
Double_t convert = kEl2fC
* simParam->GetChipGain();
Double_t adcConvert = simParam->GetADCoutRange()
/ simParam->GetADCinRange();
Double_t baseline = simParam->GetADCbaseline()
/ adcConvert;
Int_t row = 0;
Int_t col = 0;
Int_t time = 0;
for (Int_t det = 0; det < AliTRDgeometry::Ndet(); det++) {
Int_t nRowMax = fGeo->GetPadPlane(det)->GetNrows();
Int_t nColMax = fGeo->GetPadPlane(det)->GetNcols();
Int_t nTimeTotal = manDig->GetDigitsParam()->GetNTimeBins(det);
AliTRDarrayADC *digits = (AliTRDarrayADC *) manDig->GetDigits(det);
if (!manSDig->HasSDigits()) {
AliError("SDigits manager has no s-digits");
return kFALSE;
}
AliTRDarraySignal *sdigits = (AliTRDarraySignal *) manSDig->GetSDigits(det);
AliTRDarrayDictionary *tracks0 = (AliTRDarrayDictionary *) manSDig->GetDictionary(det,0);
AliTRDarrayDictionary *tracks1 = (AliTRDarrayDictionary *) manSDig->GetDictionary(det,1);
AliTRDarrayDictionary *tracks2 = (AliTRDarrayDictionary *) manSDig->GetDictionary(det,2);
sdigits->Allocate(nRowMax,nColMax,nTimeTotal);
tracks0->Allocate(nRowMax,nColMax,nTimeTotal);
tracks1->Allocate(nRowMax,nColMax,nTimeTotal);
tracks2->Allocate(nRowMax,nColMax,nTimeTotal);
manSDig->GetDigitsParam()->SetNTimeBinsAll(manDig->GetDigitsParam()->GetNTimeBins(0));
manSDig->GetDigitsParam()->SetADCbaselineAll(manDig->GetDigitsParam()->GetADCbaseline(0));
if (digits->HasData()) {
digits->Expand();
for (row = 0; row < nRowMax; row++ ) {
for (col = 0; col < nColMax; col++ ) {
for (time = 0; time < nTimeTotal; time++) {
Short_t adcVal = digits->GetData(row,col,time);
Double_t signal = (Double_t) adcVal;
signal /= adcConvert;
signal -= baseline;
signal /= convert;
signal /= coupling;
sdigits->SetData(row,col,time,signal);
tracks0->SetData(row,col,time,0);
tracks1->SetData(row,col,time,0);
tracks2->SetData(row,col,time,0);
}
}
}
}
sdigits->Compress(0);
tracks0->Compress();
tracks1->Compress();
tracks2->Compress();
manDig->RemoveDigits(det);
manDig->RemoveDictionaries(det);
}
return kTRUE;
}
Bool_t AliTRDdigitizer::SDigits2Digits()
{
if (!MergeSDigits()) {
return kFALSE;
}
return ConvertSDigits();
}
Bool_t AliTRDdigitizer::MergeSDigits()
{
const Int_t kNDict = AliTRDdigitsManager::kNDict;
AliTRDSimParam *simParam = AliTRDSimParam::Instance();
if (!simParam) {
AliFatal("Could not get simulation parameters");
return kFALSE;
}
AliTRDcalibDB *calibration = AliTRDcalibDB::Instance();
if (!calibration) {
AliFatal("Could not get calibration object");
return kFALSE;
}
Int_t iDict = 0;
Int_t jDict = 0;
AliTRDarraySignal *digitsA;
AliTRDarraySignal *digitsB;
AliTRDarrayDictionary *dictionaryA[kNDict];
AliTRDarrayDictionary *dictionaryB[kNDict];
AliTRDdigitsManager *mergeSDigitsManager = 0x0;
fSDigitsManager = (AliTRDdigitsManager *) fSDigitsManagerList->First();
if (!fSDigitsManager) {
AliError("No SDigits manager");
return kFALSE;
}
mergeSDigitsManager = (AliTRDdigitsManager *) fSDigitsManagerList->After(fSDigitsManager);
if (mergeSDigitsManager) {
AliDebug(1,Form("Merge %d input files.",fSDigitsManagerList->GetSize()));
}
else {
AliDebug(1,"Only one input file.");
}
Int_t iMerge = 0;
while (mergeSDigitsManager) {
iMerge++;
for (Int_t iDet = 0; iDet < AliTRDgeometry::Ndet(); iDet++) {
Int_t nTimeTotal = fSDigitsManager->GetDigitsParam()->GetNTimeBins(iDet);
if (mergeSDigitsManager->GetDigitsParam()->GetNTimeBins(iDet) != nTimeTotal) {
AliError(Form("Mismatch in the number of time bins [%d,%d] in detector %d"
,nTimeTotal
,mergeSDigitsManager->GetDigitsParam()->GetNTimeBins(iDet)
,iDet));
return kFALSE;
}
Int_t nRowMax = fGeo->GetPadPlane(iDet)->GetNrows();
Int_t nColMax = fGeo->GetPadPlane(iDet)->GetNcols();
digitsA = (AliTRDarraySignal *) fSDigitsManager->GetSDigits(iDet);
digitsB = (AliTRDarraySignal *) mergeSDigitsManager->GetSDigits(iDet);
digitsA->Expand();
if (!digitsA->HasData()) continue;
digitsB->Expand();
if (!digitsB->HasData()) continue;
for (iDict = 0; iDict < kNDict; iDict++) {
dictionaryA[iDict] = (AliTRDarrayDictionary *) fSDigitsManager->GetDictionary(iDet,iDict);
dictionaryB[iDict] = (AliTRDarrayDictionary *) mergeSDigitsManager->GetDictionary(iDet,iDict);
dictionaryA[iDict]->Expand();
dictionaryB[iDict]->Expand();
}
Bool_t doMerge = kTRUE;
if (fMergeSignalOnly) {
if (digitsA->GetOverThreshold(0) == 0) {
doMerge = kFALSE;
}
}
if (doMerge) {
AliDebug(1,Form("Merge detector %d of input no.%d",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++) {
Float_t ampA = digitsA->GetData(iRow,iCol,iTime);
Float_t ampB = digitsB->GetData(iRow,iCol,iTime);
ampA += ampB;
digitsA->SetData(iRow,iCol,iTime,ampA);
for (iDict = 0; iDict < kNDict; iDict++) {
Int_t trackB = dictionaryB[iDict]->GetData(iRow,iCol,iTime);
if ((fMasks) && (trackB > 0)) {
for (jDict = 0; jDict < kNDict; jDict++) {
Int_t trackA = dictionaryA[iDict]->GetData(iRow,iCol,iTime);
if (trackA == 0) {
trackA = trackB + fMasks[iMerge];
dictionaryA[iDict]->SetData(iRow,iCol,iTime,trackA);
}
}
}
}
}
}
}
}
mergeSDigitsManager->RemoveDigits(iDet);
mergeSDigitsManager->RemoveDictionaries(iDet);
if (fCompress) {
digitsA->Compress(0);
for (iDict = 0; iDict < kNDict; iDict++) {
dictionaryA[iDict]->Compress();
}
}
}
mergeSDigitsManager = (AliTRDdigitsManager *) fSDigitsManagerList->After(mergeSDigitsManager);
}
return kTRUE;
}
Bool_t AliTRDdigitizer::ConvertSDigits()
{
AliTRDarraySignal *digitsIn = 0x0;
if (!fSDigitsManager->HasSDigits()) {
AliError("No s-digits in digits manager");
return kFALSE;
}
for (Int_t det = 0; det < AliTRDgeometry::Ndet(); det++) {
digitsIn = (AliTRDarraySignal *) fSDigitsManager->GetSDigits(det);
if (!digitsIn->HasData()) {
AliDebug(2,Form("No digits for det=%d",det));
continue;
}
if (!Signal2ADC(det,digitsIn)) {
continue;
}
if (!CopyDictionary(det)) {
continue;
}
fSDigitsManager->RemoveDigits(det);
fSDigitsManager->RemoveDictionaries(det);
RunDigitalProcessing(det);
CompressOutputArrays(det);
}
if (AliDataLoader *trklLoader = AliRunLoader::Instance()->GetLoader("TRDLoader")->GetDataLoader("tracklets")) {
if (trklLoader->Tree())
trklLoader->WriteData("OVERWRITE");
}
if (AliTRDSimParam::Instance()->GetNTBoverwriteOCDB()) {
fDigitsManager->GetDigitsParam()->SetNTimeBinsAll(AliTRDSimParam::Instance()->GetNTimeBins());
}
else {
fDigitsManager->GetDigitsParam()->SetNTimeBinsAll(AliTRDcalibDB::Instance()->GetNumberOfTimeBinsDCS());
}
fDigitsManager->GetDigitsParam()->SetADCbaselineAll(AliTRDSimParam::Instance()->GetADCbaseline());
return kTRUE;
}
Bool_t AliTRDdigitizer::CopyDictionary(Int_t det)
{
AliTRDcalibDB *calibration = AliTRDcalibDB::Instance();
if (!calibration) {
AliFatal("Could not get calibration object");
return kFALSE;
}
AliDebug(1,Form("Start copying dictionaries for detector=%d",det));
const Int_t kNDict = AliTRDdigitsManager::kNDict;
AliTRDarrayDictionary *dictionaryIn[kNDict];
AliTRDarrayDictionary *dictionaryOut[kNDict];
Int_t nRowMax = fGeo->GetPadPlane(det)->GetNrows();
Int_t nColMax = fGeo->GetPadPlane(det)->GetNcols();
Int_t nTimeTotal = fSDigitsManager->GetDigitsParam()->GetNTimeBins(det);
Int_t row = 0;
Int_t col = 0;
Int_t time = 0;
Int_t dict = 0;
for (dict = 0; dict < kNDict; dict++) {
dictionaryIn[dict] = (AliTRDarrayDictionary *) fSDigitsManager->GetDictionary(det,dict);
dictionaryIn[dict]->Expand();
dictionaryOut[dict] = (AliTRDarrayDictionary *) fDigitsManager->GetDictionary(det,dict);
dictionaryOut[dict]->Allocate(nRowMax,nColMax,nTimeTotal);
for (row = 0; row < nRowMax; row++) {
for (col = 0; col < nColMax; col++) {
for (time = 0; time < nTimeTotal; time++) {
Int_t track = dictionaryIn[dict]->GetData(row,col,time);
dictionaryOut[dict]->SetData(row,col,time,track);
}
}
}
}
return kTRUE;
}
void AliTRDdigitizer::CompressOutputArrays(Int_t det)
{
const Int_t kNDict = AliTRDdigitsManager::kNDict;
AliTRDarrayDictionary *dictionary = 0x0;
if (fCompress) {
if (!fSDigits) {
AliTRDarrayADC *digits = 0x0;
digits = (AliTRDarrayADC *) fDigitsManager->GetDigits(det);
digits->Compress();
}
if (fSDigits) {
AliTRDarraySignal *digits = 0x0;
digits = (AliTRDarraySignal *) fDigitsManager->GetSDigits(det);
digits->Compress(0);
}
for (Int_t dict = 0; dict < kNDict; dict++) {
dictionary = (AliTRDarrayDictionary *) fDigitsManager->GetDictionary(det,dict);
dictionary->Compress();
}
}
}
Bool_t AliTRDdigitizer::WriteDigits() const
{
fRunLoader->CdGAFile();
return fDigitsManager->WriteDigits();
}
void AliTRDdigitizer::InitOutput(Int_t iEvent)
{
fEvent = iEvent;
if (!fRunLoader) {
AliError("Run Loader is NULL");
return;
}
AliLoader *loader = fRunLoader->GetLoader("TRDLoader");
if (!loader) {
AliError("Can not get TRD loader from Run Loader");
return;
}
TTree *tree = 0;
if (fSDigits) {
tree = loader->TreeS();
if (!tree) {
loader->MakeTree("S");
tree = loader->TreeS();
}
}
else {
tree = loader->TreeD();
if (!tree) {
loader->MakeTree("D");
tree = loader->TreeD();
}
}
fDigitsManager->SetEvent(iEvent);
fDigitsManager->MakeBranch(tree);
}
Int_t AliTRDdigitizer::Diffusion(Float_t vdrift, Double_t absdriftlength
, Double_t exbvalue
, Double_t &lRow, Double_t &lCol, Double_t &lTime)
{
Float_t diffL = 0.0;
Float_t diffT = 0.0;
if (AliTRDCommonParam::Instance()->GetDiffCoeff(diffL,diffT,vdrift)) {
Float_t driftSqrt = TMath::Sqrt(absdriftlength);
Float_t sigmaT = driftSqrt * diffT;
Float_t sigmaL = driftSqrt * diffL;
lRow = gRandom->Gaus(lRow ,sigmaT);
if (AliTRDCommonParam::Instance()->ExBOn()) {
lCol = gRandom->Gaus(lCol ,sigmaT * 1.0 / (1.0 + exbvalue*exbvalue));
lTime = gRandom->Gaus(lTime,sigmaL * 1.0 / (1.0 + exbvalue*exbvalue));
}
else {
lCol = gRandom->Gaus(lCol ,sigmaT);
lTime = gRandom->Gaus(lTime,sigmaL);
}
return 1;
}
else {
return 0;
}
}
void AliTRDdigitizer::RunDigitalProcessing(Int_t det)
{
AliTRDfeeParam *feeParam = AliTRDfeeParam::Instance();
AliTRDarrayADC *digits = fDigitsManager->GetDigits(det);
if (!digits)
return;
for (Int_t side = 0; side <= 1; side++) {
for(Int_t rob = side; rob < digits->GetNrow() / 2; rob += 2) {
for(Int_t mcm = 0; mcm < 16; mcm++) {
fMcmSim->Init(det, rob, mcm);
fMcmSim->SetDataByPad(digits, fDigitsManager);
fMcmSim->Filter();
if (feeParam->GetTracklet()) {
fMcmSim->Tracklet();
fMcmSim->StoreTracklets();
}
fMcmSim->ZSMapping();
fMcmSim->WriteData(digits);
}
}
}
}