#include <iostream>
#include <iomanip>
#include "TCanvas.h"
#include "TH1F.h"
#include "TH2F.h"
#include "TGraph.h"
#include "TLine.h"
#include "TRandom.h"
#include "TClonesArray.h"
#include "TMath.h"
#include <TTree.h>
#include "AliLog.h"
#include "AliRunLoader.h"
#include "AliLoader.h"
#include "AliTRDfeeParam.h"
#include "AliTRDcalibDB.h"
#include "AliTRDtrapConfig.h"
#include "AliTRDdigitsManager.h"
#include "AliTRDarrayADC.h"
#include "AliTRDarrayDictionary.h"
#include "AliTRDtrackletMCM.h"
#include "AliTRDmcmSim.h"
#include "TTreeStream.h"
ClassImp(AliTRDmcmSim)
Bool_t AliTRDmcmSim::fgApplyCut = kTRUE;
Int_t AliTRDmcmSim::fgAddBaseline = 0;
Bool_t AliTRDmcmSim::fgStoreClusters = kFALSE;
const Int_t AliTRDmcmSim::fgkFormatIndex = std::ios_base::xalloc();
const UShort_t AliTRDmcmSim::fgkFPshifts[4] = {11, 14, 17, 21};
AliTRDmcmSim::AliTRDmcmSim() :
TObject(),
fInitialized(kFALSE),
fDetector(-1),
fRobPos(-1),
fMcmPos(-1),
fRow (-1),
fNTimeBin(-1),
fADCR(NULL),
fADCF(NULL),
fMCMT(NULL),
fTrackletArray(NULL),
fZSMap(NULL),
fTrklBranchName("mcmtrklbranch"),
fFeeParam(NULL),
fTrapConfig(NULL),
fDigitsManager(NULL),
fPedAcc(NULL),
fGainCounterA(NULL),
fGainCounterB(NULL),
fTailAmplLong(NULL),
fTailAmplShort(NULL),
fNHits(0),
fFitReg(NULL),
fDebugStream(0x0)
{
for (Int_t iDict = 0; iDict < 3; iDict++)
fDict[iDict] = 0x0;
fFitPtr[0] = 0;
fFitPtr[1] = 0;
fFitPtr[2] = 0;
fFitPtr[3] = 0;
}
AliTRDmcmSim::~AliTRDmcmSim()
{
if(fInitialized) {
for( Int_t iAdc = 0 ; iAdc < AliTRDfeeParam::GetNadcMcm(); iAdc++ ) {
delete [] fADCR[iAdc];
delete [] fADCF[iAdc];
}
delete [] fADCR;
delete [] fADCF;
delete [] fZSMap;
delete [] fMCMT;
delete [] fPedAcc;
delete [] fGainCounterA;
delete [] fGainCounterB;
delete [] fTailAmplLong;
delete [] fTailAmplShort;
delete [] fFitReg;
fTrackletArray->Delete();
delete fTrackletArray;
}
}
void AliTRDmcmSim::Init( Int_t det, Int_t robPos, Int_t mcmPos, Bool_t )
{
if (!fInitialized) {
fFeeParam = AliTRDfeeParam::Instance();
fTrapConfig = AliTRDcalibDB::Instance()->GetTrapConfig();
}
fDetector = det;
fRobPos = robPos;
fMcmPos = mcmPos;
fRow = fFeeParam->GetPadRowFromMCM( fRobPos, fMcmPos );
if (!fInitialized) {
fADCR = new Int_t *[AliTRDfeeParam::GetNadcMcm()];
fADCF = new Int_t *[AliTRDfeeParam::GetNadcMcm()];
fZSMap = new Int_t [AliTRDfeeParam::GetNadcMcm()];
fGainCounterA = new UInt_t[AliTRDfeeParam::GetNadcMcm()];
fGainCounterB = new UInt_t[AliTRDfeeParam::GetNadcMcm()];
fNTimeBin = fTrapConfig->GetTrapReg(AliTRDtrapConfig::kC13CPUA, fDetector, fRobPos, fMcmPos);
for( Int_t iAdc = 0 ; iAdc < AliTRDfeeParam::GetNadcMcm(); iAdc++ ) {
fADCR[iAdc] = new Int_t[fNTimeBin];
fADCF[iAdc] = new Int_t[fNTimeBin];
}
fPedAcc = new UInt_t[AliTRDfeeParam::GetNadcMcm()];
fTailAmplLong = new UShort_t[AliTRDfeeParam::GetNadcMcm()];
fTailAmplShort = new UShort_t[AliTRDfeeParam::GetNadcMcm()];
fFitReg = new FitReg_t[AliTRDfeeParam::GetNadcMcm()];
fTrackletArray = new TClonesArray("AliTRDtrackletMCM", fgkMaxTracklets);
fMCMT = new UInt_t[fgkMaxTracklets];
}
fInitialized = kTRUE;
Reset();
}
void AliTRDmcmSim::Reset()
{
if( !CheckInitialized() )
return;
for( Int_t iAdc = 0 ; iAdc < AliTRDfeeParam::GetNadcMcm(); iAdc++ ) {
for( Int_t it = 0 ; it < fNTimeBin ; it++ ) {
fADCR[iAdc][it] = 0;
fADCF[iAdc][it] = 0;
}
fZSMap[iAdc] = -1;
fGainCounterA[iAdc] = 0;
fGainCounterB[iAdc] = 0;
}
for(Int_t i = 0; i < fgkMaxTracklets; i++) {
fMCMT[i] = 0;
}
for (Int_t iDict = 0; iDict < 3; iDict++)
fDict[iDict] = 0x0;
FilterPedestalInit();
FilterGainInit();
FilterTailInit();
}
void AliTRDmcmSim::SetNTimebins(Int_t ntimebins)
{
if( !CheckInitialized() )
return;
fNTimeBin = ntimebins;
for( Int_t iAdc = 0 ; iAdc < AliTRDfeeParam::GetNadcMcm(); iAdc++ ) {
delete [] fADCR[iAdc];
delete [] fADCF[iAdc];
fADCR[iAdc] = new Int_t[fNTimeBin];
fADCF[iAdc] = new Int_t[fNTimeBin];
}
}
Bool_t AliTRDmcmSim::LoadMCM(AliRunLoader* const runloader, Int_t det, Int_t rob, Int_t mcm)
{
Init(det, rob, mcm);
if (!runloader) {
AliError("No Runloader given");
return kFALSE;
}
AliLoader *trdLoader = runloader->GetLoader("TRDLoader");
if (!trdLoader) {
AliError("Could not get TRDLoader");
return kFALSE;
}
Bool_t retval = kTRUE;
trdLoader->LoadDigits();
fDigitsManager = 0x0;
AliTRDdigitsManager *digMgr = new AliTRDdigitsManager();
digMgr->SetSDigits(0);
digMgr->CreateArrays();
digMgr->ReadDigits(trdLoader->TreeD());
AliTRDarrayADC *digits = (AliTRDarrayADC*) digMgr->GetDigits(det);
if (digits->HasData()) {
digits->Expand();
if (fNTimeBin != digits->GetNtime()) {
AliWarning(Form("Changing no. of timebins from %i to %i", fNTimeBin, digits->GetNtime()));
SetNTimebins(digits->GetNtime());
}
SetData(digits);
}
else
retval = kFALSE;
delete digMgr;
return retval;
}
void AliTRDmcmSim::NoiseTest(Int_t nsamples, Int_t mean, Int_t sigma, Int_t inputGain, Int_t inputTail)
{
if( !CheckInitialized() )
return;
TString nameInputGain;
TString nameInputTail;
switch (inputGain) {
case 0:
nameInputGain = "Noise";
break;
case 1:
nameInputGain = "Pedestal";
break;
default:
AliError("Undefined input to tail cancellation filter");
return;
}
switch (inputTail) {
case 0:
nameInputTail = "Noise";
break;
case 1:
nameInputTail = "Pedestal";
break;
case 2:
nameInputTail = "Gain";
break;
default:
AliError("Undefined input to tail cancellation filter");
return;
}
TH1F *h = new TH1F("noise", "Gaussian Noise;sample;ADC count",
nsamples, 0, nsamples);
TH1F *hfp = new TH1F("ped", "Noise #rightarrow Pedestal filter;sample;ADC count", nsamples, 0, nsamples);
TH1F *hfg = new TH1F("gain",
(nameInputGain + "#rightarrow Gain;sample;ADC count").Data(),
nsamples, 0, nsamples);
TH1F *hft = new TH1F("tail",
(nameInputTail + "#rightarrow Tail;sample;ADC count").Data(),
nsamples, 0, nsamples);
h->SetStats(kFALSE);
hfp->SetStats(kFALSE);
hfg->SetStats(kFALSE);
hft->SetStats(kFALSE);
Int_t value;
Int_t valuep;
Int_t valueg;
Int_t valuet;
for (Int_t i = 0; i < nsamples; i++) {
value = (Int_t) gRandom->Gaus(mean, sigma);
h->SetBinContent(i, value);
valuep = FilterPedestalNextSample(1, 0, ((Int_t) value) << 2);
if (inputGain == 0)
valueg = FilterGainNextSample(1, ((Int_t) value) << 2);
else
valueg = FilterGainNextSample(1, valuep);
if (inputTail == 0)
valuet = FilterTailNextSample(1, ((Int_t) value) << 2);
else if (inputTail == 1)
valuet = FilterTailNextSample(1, valuep);
else
valuet = FilterTailNextSample(1, valueg);
hfp->SetBinContent(i, valuep >> 2);
hfg->SetBinContent(i, valueg >> 2);
hft->SetBinContent(i, valuet >> 2);
}
TCanvas *c = new TCanvas;
c->Divide(2,2);
c->cd(1);
h->Draw();
c->cd(2);
hfp->Draw();
c->cd(3);
hfg->Draw();
c->cd(4);
hft->Draw();
}
Bool_t AliTRDmcmSim::CheckInitialized() const
{
if( ! fInitialized )
AliError(Form ("AliTRDmcmSim is not initialized but function other than Init() is called."));
return fInitialized;
}
void AliTRDmcmSim::Print(Option_t* const option) const
{
if ( !CheckInitialized() )
return;
printf("MCM %i on ROB %i in detector %i\n", fMcmPos, fRobPos, fDetector);
TString opt = option;
if (opt.Contains("R") || opt.Contains("F")) {
std::cout << *this;
}
if (opt.Contains("H")) {
printf("Found %i hits:\n", fNHits);
for (Int_t iHit = 0; iHit < fNHits; iHit++) {
printf("Hit %3i in timebin %2i, ADC %2i has charge %3i and position %3i\n",
iHit, fHits[iHit].fTimebin, fHits[iHit].fChannel, fHits[iHit].fQtot, fHits[iHit].fYpos);
}
}
if (opt.Contains("T")) {
printf("Tracklets:\n");
for (Int_t iTrkl = 0; iTrkl < fTrackletArray->GetEntriesFast(); iTrkl++) {
printf("tracklet %i: 0x%08x\n", iTrkl, ((AliTRDtrackletMCM*) (*fTrackletArray)[iTrkl])->GetTrackletWord());
}
}
}
void AliTRDmcmSim::Draw(Option_t* const option)
{
if( !CheckInitialized() )
return;
TString opt = option;
TH2F *hist = new TH2F("mcmdata", Form("Data of MCM %i on ROB %i in detector %i", \
fMcmPos, fRobPos, fDetector), \
AliTRDfeeParam::GetNadcMcm(), -0.5, AliTRDfeeParam::GetNadcMcm()-.5, fNTimeBin, -.5, fNTimeBin-.5);
hist->GetXaxis()->SetTitle("ADC Channel");
hist->GetYaxis()->SetTitle("Timebin");
hist->SetStats(kFALSE);
if (opt.Contains("R")) {
for (Int_t iTimeBin = 0; iTimeBin < fNTimeBin; iTimeBin++) {
for (Int_t iAdc = 0; iAdc < AliTRDfeeParam::GetNadcMcm(); iAdc++) {
hist->SetBinContent(iAdc+1, iTimeBin+1, fADCR[iAdc][iTimeBin] >> fgkAddDigits);
}
}
}
else {
for (Int_t iTimeBin = 0; iTimeBin < fNTimeBin; iTimeBin++) {
for (Int_t iAdc = 0; iAdc < AliTRDfeeParam::GetNadcMcm(); iAdc++) {
hist->SetBinContent(iAdc+1, iTimeBin+1, fADCF[iAdc][iTimeBin] >> fgkAddDigits);
}
}
}
hist->Draw("colz");
if (opt.Contains("H")) {
TGraph *grHits = new TGraph();
for (Int_t iHit = 0; iHit < fNHits; iHit++) {
grHits->SetPoint(iHit,
fHits[iHit].fChannel + 1 + fHits[iHit].fYpos/256.,
fHits[iHit].fTimebin);
}
grHits->Draw("*");
}
if (opt.Contains("T")) {
TLine *trklLines = new TLine[4];
for (Int_t iTrkl = 0; iTrkl < fTrackletArray->GetEntries(); iTrkl++) {
AliTRDtrackletMCM *trkl = (AliTRDtrackletMCM*) (*fTrackletArray)[iTrkl];
Float_t padWidth = 0.635 + 0.03 * (fDetector % 6);
Float_t offset = padWidth/256. * ((((((fRobPos & 0x1) << 2) + (fMcmPos & 0x3)) * 18) << 8) - ((18*4*2 - 18*2 - 3) << 7));
Int_t ndrift = fTrapConfig->GetDmemUnsigned(fgkDmemAddrNdrift, fDetector, fRobPos, fMcmPos) >> 5;
Float_t slope = 0;
if (ndrift)
slope = trkl->GetdY() * 140e-4 / ndrift;
Int_t t0 = fTrapConfig->GetTrapReg(AliTRDtrapConfig::kTPFS, fDetector, fRobPos, fMcmPos);
Int_t t1 = fTrapConfig->GetTrapReg(AliTRDtrapConfig::kTPFE, fDetector, fRobPos, fMcmPos);
trklLines[iTrkl].SetX1((offset - (trkl->GetY() - slope * t0)) / padWidth);
trklLines[iTrkl].SetY1(t0);
trklLines[iTrkl].SetX2((offset - (trkl->GetY() - slope * t1)) / padWidth);
trklLines[iTrkl].SetY2(t1);
trklLines[iTrkl].SetLineColor(2);
trklLines[iTrkl].SetLineWidth(2);
printf("Tracklet %i: y = %f, dy = %f, offset = %f\n", iTrkl, trkl->GetY(), (trkl->GetdY() * 140e-4), offset);
trklLines[iTrkl].Draw();
}
}
}
void AliTRDmcmSim::SetData( Int_t adc, const Int_t* const data )
{
if( !CheckInitialized() ) return;
if( adc < 0 || adc >= AliTRDfeeParam::GetNadcMcm() ) {
AliError(Form ("Error: ADC %i is out of range (0 .. %d).", adc, AliTRDfeeParam::GetNadcMcm()-1));
return;
}
for( Int_t it = 0 ; it < fNTimeBin ; it++ ) {
fADCR[adc][it] = (Int_t) (data[it]) << fgkAddDigits;
fADCF[adc][it] = (Int_t) (data[it]) << fgkAddDigits;
}
}
void AliTRDmcmSim::SetData( Int_t adc, Int_t it, Int_t data )
{
if( !CheckInitialized() ) return;
if( adc < 0 || adc >= AliTRDfeeParam::GetNadcMcm() ) {
AliError(Form ("Error: ADC %i is out of range (0 .. %d).", adc, AliTRDfeeParam::GetNadcMcm()-1));
return;
}
fADCR[adc][it] = data << fgkAddDigits;
fADCF[adc][it] = data << fgkAddDigits;
}
void AliTRDmcmSim::SetData(AliTRDarrayADC* const adcArray, AliTRDdigitsManager * const digitsManager)
{
if( !CheckInitialized() )
return;
fDigitsManager = digitsManager;
if (fDigitsManager) {
for (Int_t iDict = 0; iDict < 3; iDict++) {
AliTRDarrayDictionary *newDict = (AliTRDarrayDictionary*) fDigitsManager->GetDictionary(fDetector, iDict);
if (fDict[iDict] != 0x0 && newDict != 0x0) {
if (fDict[iDict] == newDict)
continue;
fDict[iDict] = newDict;
if(fDict[iDict]->GetDim() != 0)
fDict[iDict]->Expand();
}
else {
fDict[iDict] = newDict;
if (fDict[iDict] && (fDict[iDict]->GetDim() != 0) )
fDict[iDict]->Expand();
}
if (fDict[iDict]->GetDim() == 0) {
fDict[iDict] = 0x0;
}
}
}
if (fNTimeBin != adcArray->GetNtime())
SetNTimebins(adcArray->GetNtime());
Int_t offset = (fMcmPos % 4 + 1) * 21 + (fRobPos % 2) * 84 - 1;
for (Int_t iTimeBin = 0; iTimeBin < fNTimeBin; iTimeBin++) {
for (Int_t iAdc = 0; iAdc < AliTRDfeeParam::GetNadcMcm(); iAdc++) {
Int_t value = adcArray->GetDataByAdcCol(GetRow(), offset - iAdc, iTimeBin);
if (value <= 0 || (offset - iAdc < 1) || (offset - iAdc > 165)) {
fADCR[iAdc][iTimeBin] = fTrapConfig->GetTrapReg(AliTRDtrapConfig::kFPNP, fDetector, fRobPos, fMcmPos) + (fgAddBaseline << fgkAddDigits);
fADCF[iAdc][iTimeBin] = fTrapConfig->GetTrapReg(AliTRDtrapConfig::kTPFP, fDetector, fRobPos, fMcmPos) + (fgAddBaseline << fgkAddDigits);
}
else {
fZSMap[iAdc] = 0;
fADCR[iAdc][iTimeBin] = (value << fgkAddDigits) + (fgAddBaseline << fgkAddDigits);
fADCF[iAdc][iTimeBin] = (value << fgkAddDigits) + (fgAddBaseline << fgkAddDigits);
}
}
}
}
void AliTRDmcmSim::SetDataByPad(const AliTRDarrayADC* const adcArray, AliTRDdigitsManager * const digitsManager)
{
if( !CheckInitialized() )
return;
fDigitsManager = digitsManager;
if (fDigitsManager) {
for (Int_t iDict = 0; iDict < 3; iDict++) {
AliTRDarrayDictionary *newDict = (AliTRDarrayDictionary*) fDigitsManager->GetDictionary(fDetector, iDict);
if (fDict[iDict] != 0x0 && newDict != 0x0) {
if (fDict[iDict] == newDict)
continue;
fDict[iDict] = newDict;
fDict[iDict]->Expand();
}
else {
fDict[iDict] = newDict;
if (fDict[iDict])
fDict[iDict]->Expand();
}
if (fDict[iDict]->GetDim() == 0) {
AliError(Form("Dictionary %i of det. %i has dim. 0", iDict, fDetector));
fDict[iDict] = 0x0;
}
}
}
if (fNTimeBin != adcArray->GetNtime())
SetNTimebins(adcArray->GetNtime());
Int_t offset = (fMcmPos % 4 + 1) * 18 + (fRobPos % 2) * 72 + 1;
for (Int_t iTimeBin = 0; iTimeBin < fNTimeBin; iTimeBin++) {
for (Int_t iAdc = 0; iAdc < AliTRDfeeParam::GetNadcMcm(); iAdc++) {
Int_t value = -1;
Int_t pad = offset - iAdc;
if (pad > -1 && pad < 144)
value = adcArray->GetData(GetRow(), offset - iAdc, iTimeBin);
if (value < 0 || (offset - iAdc < 1) || (offset - iAdc > 165)) {
fADCR[iAdc][iTimeBin] = fTrapConfig->GetTrapReg(AliTRDtrapConfig::kFPNP, fDetector, fRobPos, fMcmPos) + (fgAddBaseline << fgkAddDigits);
fADCF[iAdc][iTimeBin] = fTrapConfig->GetTrapReg(AliTRDtrapConfig::kTPFP, fDetector, fRobPos, fMcmPos) + (fgAddBaseline << fgkAddDigits);
}
else {
fZSMap[iAdc] = 0;
fADCR[iAdc][iTimeBin] = (value << fgkAddDigits) + (fgAddBaseline << fgkAddDigits);
fADCF[iAdc][iTimeBin] = (value << fgkAddDigits) + (fgAddBaseline << fgkAddDigits);
}
}
}
}
void AliTRDmcmSim::SetDataPedestal( Int_t adc )
{
if( !CheckInitialized() )
return;
if( adc < 0 || adc >= AliTRDfeeParam::GetNadcMcm() ) {
return;
}
for( Int_t it = 0 ; it < fNTimeBin ; it++ ) {
fADCR[adc][it] = fTrapConfig->GetTrapReg(AliTRDtrapConfig::kFPNP, fDetector, fRobPos, fMcmPos) + (fgAddBaseline << fgkAddDigits);
fADCF[adc][it] = fTrapConfig->GetTrapReg(AliTRDtrapConfig::kTPFP, fDetector, fRobPos, fMcmPos) + (fgAddBaseline << fgkAddDigits);
}
}
Bool_t AliTRDmcmSim::GetHit(Int_t index, Int_t &channel, Int_t &timebin, Int_t &qtot, Int_t &ypos, Float_t &y, Int_t &label) const
{
if (index < 0 || index >= fNHits)
return kFALSE;
channel = fHits[index].fChannel;
timebin = fHits[index].fTimebin;
qtot = fHits[index].fQtot;
ypos = fHits[index].fYpos;
y = (Float_t) ((((((fRobPos & 0x1) << 2) + (fMcmPos & 0x3)) * 18) << 8) - ((18*4*2 - 18*2 - 1) << 7) -
(channel << 8) - ypos)
* (0.635 + 0.03 * (fDetector % 6))
/ 256.0;
label = fHits[index].fLabel[0];
return kTRUE;
}
Int_t AliTRDmcmSim::GetCol( Int_t adc )
{
if( !CheckInitialized() )
return -1;
Int_t col = fFeeParam->GetPadColFromADC(fRobPos, fMcmPos, adc);
if (col < 0 || col >= fFeeParam->GetNcol())
return -1;
else
return col;
}
Int_t AliTRDmcmSim::ProduceRawStream( UInt_t *buf, Int_t bufSize, UInt_t iEv) const
{
if( !CheckInitialized() )
return 0;
UInt_t x;
UInt_t mcmHeader = 0;
UInt_t adcMask = 0;
Int_t nw = 0;
Int_t of = 0;
Int_t rawVer = fFeeParam->GetRAWversion();
Int_t **adc;
Int_t nActiveADC = 0;
if( !CheckInitialized() )
return 0;
if (fTrapConfig->GetTrapReg(AliTRDtrapConfig::kEBSF, fDetector, fRobPos, fMcmPos) != 0)
adc = fADCR;
else
adc = fADCF;
if( rawVer >= 3 &&
(fTrapConfig->GetTrapReg(AliTRDtrapConfig::kC15CPUA, fDetector, fRobPos, fMcmPos) & (1 << 13))) {
for( Int_t iAdc = 0 ; iAdc < AliTRDfeeParam::GetNadcMcm() ; iAdc++ ) {
if( ~fZSMap[iAdc] != 0 ) {
adcMask |= (1 << (iAdc+4) );
nActiveADC++;
}
}
if ((nActiveADC == 0) &&
(fTrapConfig->GetTrapReg(AliTRDtrapConfig::kC15CPUA, fDetector, fRobPos, fMcmPos) & (1 << 8)))
return 0;
adcMask |= (1 << 30) | ( ( 0x3FFFFFFC ) & (~(nActiveADC) << 25) ) | 0xC;
}
mcmHeader = (1<<31) | (fRobPos << 28) | (fMcmPos << 24) | ((iEv % 0x100000) << 4) | 0xC;
if (nw < bufSize)
buf[nw++] = mcmHeader;
else
of++;
if( adcMask != 0 ) {
if (nw < bufSize)
buf[nw++] = adcMask;
else
of++;
}
UInt_t aa=0, a1=0, a2=0, a3=0;
for (Int_t iAdc = 0; iAdc < 21; iAdc++ ) {
if( rawVer>= 3 && ~fZSMap[iAdc] == 0 ) continue;
aa = !(iAdc & 1) + 2;
for (Int_t iT = 0; iT < fNTimeBin; iT+=3 ) {
a1 = ((iT ) < fNTimeBin ) ? adc[iAdc][iT ] >> fgkAddDigits : 0;
a2 = ((iT + 1) < fNTimeBin ) ? adc[iAdc][iT+1] >> fgkAddDigits : 0;
a3 = ((iT + 2) < fNTimeBin ) ? adc[iAdc][iT+2] >> fgkAddDigits : 0;
x = (a3 << 22) | (a2 << 12) | (a1 << 2) | aa;
if (nw < bufSize) {
buf[nw++] = x;
}
else {
of++;
}
}
}
if( of != 0 ) return -of; else return nw;
}
Int_t AliTRDmcmSim::ProduceTrackletStream( UInt_t *buf, Int_t bufSize )
{
if( !CheckInitialized() )
return 0;
Int_t nw = 0;
Int_t of = 0;
for (Int_t iTracklet = 0; iTracklet < fTrackletArray->GetEntriesFast(); iTracklet++) {
if (nw < bufSize)
buf[nw++] = ((AliTRDtrackletMCM*) (*fTrackletArray)[iTracklet])->GetTrackletWord();
else
of++;
}
if( of != 0 ) return -of; else return nw;
}
void AliTRDmcmSim::Filter()
{
if( !CheckInitialized() )
return;
FilterPedestal();
FilterGain();
FilterTail();
}
void AliTRDmcmSim::FilterPedestalInit(Int_t baseline)
{
UShort_t fptc = fTrapConfig->GetTrapReg(AliTRDtrapConfig::kFPTC, fDetector, fRobPos, fMcmPos);
for (Int_t iAdc = 0; iAdc < AliTRDfeeParam::GetNadcMcm(); iAdc++)
fPedAcc[iAdc] = (baseline << 2) * (1 << fgkFPshifts[fptc]);
}
UShort_t AliTRDmcmSim::FilterPedestalNextSample(Int_t adc, Int_t timebin, UShort_t value)
{
UShort_t fpnp = fTrapConfig->GetTrapReg(AliTRDtrapConfig::kFPNP, fDetector, fRobPos, fMcmPos);
UShort_t fptc = fTrapConfig->GetTrapReg(AliTRDtrapConfig::kFPTC, fDetector, fRobPos, fMcmPos);
UShort_t fpby = fTrapConfig->GetTrapReg(AliTRDtrapConfig::kFPBY, fDetector, fRobPos, fMcmPos);
UShort_t accumulatorShifted;
Int_t correction;
UShort_t inpAdd;
inpAdd = value + fpnp;
accumulatorShifted = (fPedAcc[adc] >> fgkFPshifts[fptc]) & 0x3FF;
if (timebin == 0)
{
correction = (value & 0x3FF) - accumulatorShifted;
fPedAcc[adc] = (fPedAcc[adc] + correction) & 0x7FFFFFFF;
}
if (fpby == 0)
return value;
if (inpAdd <= accumulatorShifted)
return 0;
else
{
inpAdd = inpAdd - accumulatorShifted;
if (inpAdd > 0xFFF)
return 0xFFF;
else
return inpAdd;
}
}
void AliTRDmcmSim::FilterPedestal()
{
for (Int_t iTimeBin = 0; iTimeBin < fNTimeBin; iTimeBin++) {
for (Int_t iAdc = 0; iAdc < AliTRDfeeParam::GetNadcMcm(); iAdc++) {
fADCF[iAdc][iTimeBin] = FilterPedestalNextSample(iAdc, iTimeBin, fADCR[iAdc][iTimeBin]);
}
}
}
void AliTRDmcmSim::FilterGainInit()
{
for (Int_t iAdc = 0; iAdc < AliTRDfeeParam::GetNadcMcm(); iAdc++) {
fGainCounterA[iAdc] = 0;
fGainCounterB[iAdc] = 0;
}
}
UShort_t AliTRDmcmSim::FilterGainNextSample(Int_t adc, UShort_t value)
{
UShort_t fgby = fTrapConfig->GetTrapReg(AliTRDtrapConfig::kFGBY, fDetector, fRobPos, fMcmPos);
UShort_t fgf = fTrapConfig->GetTrapReg(AliTRDtrapConfig::TrapReg_t(AliTRDtrapConfig::kFGF0 + adc), fDetector, fRobPos, fMcmPos);
UShort_t fga = fTrapConfig->GetTrapReg(AliTRDtrapConfig::TrapReg_t(AliTRDtrapConfig::kFGA0 + adc), fDetector, fRobPos, fMcmPos);
UShort_t fgta = fTrapConfig->GetTrapReg(AliTRDtrapConfig::kFGTA, fDetector, fRobPos, fMcmPos);
UShort_t fgtb = fTrapConfig->GetTrapReg(AliTRDtrapConfig::kFGTB, fDetector, fRobPos, fMcmPos);
UInt_t fgfExtended = 0x700 + fgf;
UInt_t corr;
value &= 0xFFF;
corr = (value * fgfExtended) >> 11;
corr = corr > 0xfff ? 0xfff : corr;
corr = AddUintClipping(corr, fga, 12);
if (!((fGainCounterA[adc] == 0x3FFFFFF) || (fGainCounterB[adc] == 0x3FFFFFF)))
{
if (corr >= fgtb)
fGainCounterB[adc]++;
else if (corr >= fgta)
fGainCounterA[adc]++;
}
if (fgby == 1)
return corr;
else
return value;
}
void AliTRDmcmSim::FilterGain()
{
for (Int_t iAdc = 0; iAdc < AliTRDfeeParam::GetNadcMcm(); iAdc++) {
for (Int_t iTimeBin = 0; iTimeBin < fNTimeBin; iTimeBin++) {
fADCF[iAdc][iTimeBin] = FilterGainNextSample(iAdc, fADCF[iAdc][iTimeBin]);
}
}
}
void AliTRDmcmSim::FilterTailInit(Int_t baseline)
{
UShort_t alphaLong = 0x3ff & fTrapConfig->GetTrapReg(AliTRDtrapConfig::kFTAL, fDetector, fRobPos, fMcmPos);
UShort_t lambdaLong = (1 << 10) | (1 << 9) | (fTrapConfig->GetTrapReg(AliTRDtrapConfig::kFTLL, fDetector, fRobPos, fMcmPos) & 0x1FF);
UShort_t lambdaShort = (0 << 10) | (1 << 9) | (fTrapConfig->GetTrapReg(AliTRDtrapConfig::kFTLS, fDetector, fRobPos, fMcmPos) & 0x1FF);
Float_t lambdaL = lambdaLong * 1.0 / (1 << 11);
Float_t lambdaS = lambdaShort * 1.0 / (1 << 11);
Float_t alphaL = alphaLong * 1.0 / (1 << 11);
Float_t qup, qdn;
qup = (1 - lambdaL) * (1 - lambdaS);
qdn = 1 - lambdaS * alphaL - lambdaL * (1 - alphaL);
Float_t kdc = qup/qdn;
Float_t kt, ql, qs;
UShort_t aout;
if (baseline < 0)
baseline = fTrapConfig->GetTrapReg(AliTRDtrapConfig::kFPNP, fDetector, fRobPos, fMcmPos);
ql = lambdaL * (1 - lambdaS) * alphaL;
qs = lambdaS * (1 - lambdaL) * (1 - alphaL);
for (Int_t iAdc = 0; iAdc < AliTRDfeeParam::GetNadcMcm(); iAdc++) {
Int_t value = baseline & 0xFFF;
Int_t corr = (value * fTrapConfig->GetTrapReg(AliTRDtrapConfig::TrapReg_t(AliTRDtrapConfig::kFGF0 + iAdc), fDetector, fRobPos, fMcmPos)) >> 11;
corr = corr > 0xfff ? 0xfff : corr;
corr = AddUintClipping(corr, fTrapConfig->GetTrapReg(AliTRDtrapConfig::TrapReg_t(AliTRDtrapConfig::kFGA0 + iAdc), fDetector, fRobPos, fMcmPos), 12);
kt = kdc * baseline;
aout = baseline - (UShort_t) kt;
fTailAmplLong[iAdc] = (UShort_t) (aout * ql / (ql + qs));
fTailAmplShort[iAdc] = (UShort_t) (aout * qs / (ql + qs));
}
}
UShort_t AliTRDmcmSim::FilterTailNextSample(Int_t adc, UShort_t value)
{
UShort_t alphaLong = 0x3ff & fTrapConfig->GetTrapReg(AliTRDtrapConfig::kFTAL, fDetector, fRobPos, fMcmPos);
UShort_t lambdaLong = (1 << 10) | (1 << 9) | (fTrapConfig->GetTrapReg(AliTRDtrapConfig::kFTLL, fDetector, fRobPos, fMcmPos) & 0x1FF);
UShort_t lambdaShort = (0 << 10) | (1 << 9) | (fTrapConfig->GetTrapReg(AliTRDtrapConfig::kFTLS, fDetector, fRobPos, fMcmPos) & 0x1FF);
UInt_t aDiff;
UInt_t alInpv;
UShort_t aQ;
UInt_t tmp;
UShort_t inpVolt = value & 0xFFF;
aQ = AddUintClipping(fTailAmplLong[adc], fTailAmplShort[adc], 12);
if (inpVolt > aQ)
aDiff = inpVolt - aQ;
else
aDiff = 0;
alInpv = (aDiff * alphaLong) >> 11;
tmp = AddUintClipping(fTailAmplLong[adc], alInpv, 12);
tmp = (tmp * lambdaLong) >> 11;
fTailAmplLong[adc] = tmp & 0xFFF;
tmp = AddUintClipping(fTailAmplShort[adc], aDiff - alInpv, 12);
tmp = (tmp * lambdaShort) >> 11;
fTailAmplShort[adc] = tmp & 0xFFF;
if (fTrapConfig->GetTrapReg(AliTRDtrapConfig::kFTBY, fDetector, fRobPos, fMcmPos) == 0)
return value;
else
return aDiff;
}
void AliTRDmcmSim::FilterTail()
{
for (Int_t iTimeBin = 0; iTimeBin < fNTimeBin; iTimeBin++) {
for (Int_t iAdc = 0; iAdc < AliTRDfeeParam::GetNadcMcm(); iAdc++) {
fADCF[iAdc][iTimeBin] = FilterTailNextSample(iAdc, fADCF[iAdc][iTimeBin]);
}
}
}
void AliTRDmcmSim::ZSMapping()
{
if( !CheckInitialized() )
return;
Int_t eBIS = fTrapConfig->GetTrapReg(AliTRDtrapConfig::kEBIS, fDetector, fRobPos, fMcmPos);
Int_t eBIT = fTrapConfig->GetTrapReg(AliTRDtrapConfig::kEBIT, fDetector, fRobPos, fMcmPos);
Int_t eBIL = fTrapConfig->GetTrapReg(AliTRDtrapConfig::kEBIL, fDetector, fRobPos, fMcmPos);
Int_t eBIN = fTrapConfig->GetTrapReg(AliTRDtrapConfig::kEBIN, fDetector, fRobPos, fMcmPos);
Int_t **adc = fADCF;
for (Int_t iAdc = 0; iAdc < AliTRDfeeParam::GetNadcMcm(); iAdc++)
fZSMap[iAdc] = -1;
for( Int_t it = 0 ; it < fNTimeBin ; it++ ) {
Int_t iAdc;
Int_t ap;
Int_t ac;
Int_t an;
Int_t mask;
Int_t supp;
iAdc = 0;
ap = 0 >> fgkAddDigits;
ac = adc[iAdc ][it] >> fgkAddDigits;
an = adc[iAdc+1][it] >> fgkAddDigits;
mask = ( ac >= ap && ac >= an ) ? 0 : 0x1;
mask += ( ap + ac + an > eBIT ) ? 0 : 0x2;
mask += ( ac > eBIS ) ? 0 : 0x4;
supp = (eBIL >> mask) & 1;
fZSMap[iAdc] &= ~((1-supp) << it);
if( eBIN == 0 ) {
fZSMap[iAdc+1] &= ~((1-supp) << it);
}
iAdc = AliTRDfeeParam::GetNadcMcm() - 1;
ap = adc[iAdc-1][it] >> fgkAddDigits;
ac = adc[iAdc ][it] >> fgkAddDigits;
an = 0 >> fgkAddDigits;
mask = ( ac >= ap && ac >= an ) ? 0 : 0x1;
mask += ( ap + ac + an > eBIT ) ? 0 : 0x2;
mask += ( ac > eBIS ) ? 0 : 0x4;
supp = (eBIL >> mask) & 1;
fZSMap[iAdc] &= ~((1-supp) << it);
if( eBIN == 0 ) {
fZSMap[iAdc-1] &= ~((1-supp) << it);
}
for( iAdc = 1 ; iAdc < AliTRDfeeParam::GetNadcMcm()-1; iAdc++ ) {
ap = adc[iAdc-1][it] >> fgkAddDigits;
ac = adc[iAdc ][it] >> fgkAddDigits;
an = adc[iAdc+1][it] >> fgkAddDigits;
mask = ( ac >= ap && ac >= an ) ? 0 : 0x1;
mask += ( ap + ac + an > eBIT ) ? 0 : 0x2;
mask += ( ac > eBIS ) ? 0 : 0x4;
supp = (eBIL >> mask) & 1;
fZSMap[iAdc] &= ~((1-supp) << it);
if( eBIN == 0 ) {
fZSMap[iAdc-1] &= ~((1-supp) << it);
fZSMap[iAdc+1] &= ~((1-supp) << it);
}
}
}
}
void AliTRDmcmSim::AddHitToFitreg(Int_t adc, UShort_t timebin, UShort_t qtot, Short_t ypos, Int_t label[])
{
if ((timebin >= fTrapConfig->GetTrapReg(AliTRDtrapConfig::kTPQS0, fDetector, fRobPos, fMcmPos)) &&
(timebin < fTrapConfig->GetTrapReg(AliTRDtrapConfig::kTPQE0, fDetector, fRobPos, fMcmPos)))
fFitReg[adc].fQ0 += qtot;
if ((timebin >= fTrapConfig->GetTrapReg(AliTRDtrapConfig::kTPQS1, fDetector, fRobPos, fMcmPos)) &&
(timebin < fTrapConfig->GetTrapReg(AliTRDtrapConfig::kTPQE1, fDetector, fRobPos, fMcmPos)))
fFitReg[adc].fQ1 += qtot;
if ((timebin >= fTrapConfig->GetTrapReg(AliTRDtrapConfig::kTPFS, fDetector, fRobPos, fMcmPos) ) &&
(timebin < fTrapConfig->GetTrapReg(AliTRDtrapConfig::kTPFE, fDetector, fRobPos, fMcmPos)))
{
fFitReg[adc].fSumX += timebin;
fFitReg[adc].fSumX2 += timebin*timebin;
fFitReg[adc].fNhits++;
fFitReg[adc].fSumY += ypos;
fFitReg[adc].fSumY2 += ypos*ypos;
fFitReg[adc].fSumXY += timebin*ypos;
AliDebug(10, Form("fitreg[%2i] in timebin %2i: X=%i, X2=%i, N=%i, Y=%i, Y2=%i, XY=%i, Q0=%i, Q1=%i",
adc, timebin, fFitReg[adc].fSumX, fFitReg[adc].fSumX2, fFitReg[adc].fNhits,
fFitReg[adc].fSumY, fFitReg[adc].fSumY2, fFitReg[adc].fSumXY, fFitReg[adc].fQ0, fFitReg[adc].fQ1));
}
fHits[fNHits].fChannel = adc;
fHits[fNHits].fQtot = qtot;
fHits[fNHits].fYpos = ypos;
fHits[fNHits].fTimebin = timebin;
fHits[fNHits].fLabel[0] = label[0];
fHits[fNHits].fLabel[1] = label[1];
fHits[fNHits].fLabel[2] = label[2];
fNHits++;
}
void AliTRDmcmSim::CalcFitreg()
{
UInt_t adcMask = 0xffffffff;
Bool_t hitQual;
Int_t adcLeft, adcCentral, adcRight;
UShort_t timebin, adcch, timebin1, timebin2, qtotTemp;
Short_t ypos, fromLeft, fromRight, found;
UShort_t qTotal[19+1];
UShort_t marked[6], qMarked[6], worse1, worse2;
if (fgStoreClusters) {
timebin1 = 0;
timebin2 = fNTimeBin;
}
else {
timebin1 = fTrapConfig->GetTrapReg(AliTRDtrapConfig::kTPFS, fDetector, fRobPos, fMcmPos);
if (fTrapConfig->GetTrapReg(AliTRDtrapConfig::kTPQS0, fDetector, fRobPos, fMcmPos)
< timebin1)
timebin1 = fTrapConfig->GetTrapReg(AliTRDtrapConfig::kTPQS0, fDetector, fRobPos, fMcmPos);
if (fTrapConfig->GetTrapReg(AliTRDtrapConfig::kTPQS1, fDetector, fRobPos, fMcmPos)
< timebin1)
timebin1 = fTrapConfig->GetTrapReg(AliTRDtrapConfig::kTPQS1, fDetector, fRobPos, fMcmPos);
timebin2 = fTrapConfig->GetTrapReg(AliTRDtrapConfig::kTPFE, fDetector, fRobPos, fMcmPos);
if (fTrapConfig->GetTrapReg(AliTRDtrapConfig::kTPQE0, fDetector, fRobPos, fMcmPos)
> timebin2)
timebin2 = fTrapConfig->GetTrapReg(AliTRDtrapConfig::kTPQE0, fDetector, fRobPos, fMcmPos);
if (fTrapConfig->GetTrapReg(AliTRDtrapConfig::kTPQE1, fDetector, fRobPos, fMcmPos)
> timebin2)
timebin2 = fTrapConfig->GetTrapReg(AliTRDtrapConfig::kTPQE1, fDetector, fRobPos, fMcmPos);
}
fNHits = 0;
for (adcch = 0; adcch < AliTRDfeeParam::GetNadcMcm()-2; adcch++)
{
fFitReg[adcch].fNhits = 0;
fFitReg[adcch].fQ0 = 0;
fFitReg[adcch].fQ1 = 0;
fFitReg[adcch].fSumX = 0;
fFitReg[adcch].fSumY = 0;
fFitReg[adcch].fSumX2 = 0;
fFitReg[adcch].fSumY2 = 0;
fFitReg[adcch].fSumXY = 0;
}
for (timebin = timebin1; timebin < timebin2; timebin++)
{
for (adcch = 0; adcch < AliTRDfeeParam::GetNadcMcm()-2; adcch++) {
if ( ( (adcMask >> adcch) & 7) == 7)
{
adcLeft = fADCF[adcch ][timebin];
adcCentral = fADCF[adcch+1][timebin];
adcRight = fADCF[adcch+2][timebin];
if (fTrapConfig->GetTrapReg(AliTRDtrapConfig::kTPVBY, fDetector, fRobPos, fMcmPos) == 0) {
hitQual = kTRUE;
}
else {
hitQual = ( (adcLeft * adcRight) <
((fTrapConfig->GetTrapReg(AliTRDtrapConfig::kTPVT, fDetector, fRobPos, fMcmPos) * adcCentral*adcCentral) >> 10) );
if (hitQual)
AliDebug(5, Form("cluster quality cut passed with %3i, %3i, %3i - threshold %3i -> %i",
adcLeft, adcCentral, adcRight,
fTrapConfig->GetTrapReg(AliTRDtrapConfig::kTPVT, fDetector, fRobPos, fMcmPos),
fTrapConfig->GetTrapReg(AliTRDtrapConfig::kTPVT, fDetector, fRobPos, fMcmPos) * adcCentral*adcCentral));
}
qtotTemp = adcLeft + adcCentral + adcRight;
if ((fDebugStream) && (qtotTemp > 130)) {
(*fDebugStream) << "testtree"
<< "qtot=" << qtotTemp
<< "qleft=" << adcLeft
<< "qcent=" << adcCentral
<< "qright=" << adcRight
<< "\n";
}
if ( (hitQual) &&
(qtotTemp >= fTrapConfig->GetTrapReg(AliTRDtrapConfig::kTPHT, fDetector, fRobPos, fMcmPos)) &&
(adcLeft <= adcCentral) &&
(adcCentral > adcRight) )
qTotal[adcch] = qtotTemp;
else
qTotal[adcch] = 0;
}
else
qTotal[adcch] = 0;
if (qTotal[adcch] != 0)
AliDebug(10,Form("ch %2d qTotal %5d",adcch, qTotal[adcch]));
}
fromLeft = -1;
adcch = 0;
found = 0;
marked[4] = 19;
marked[5] = 19;
qTotal[19] = 0;
while ((adcch < 16) && (found < 3))
{
if (qTotal[adcch] > 0)
{
fromLeft = adcch;
marked[2*found+1]=adcch;
found++;
}
adcch++;
}
fromRight = -1;
adcch = 18;
found = 0;
while ((adcch > 2) && (found < 3))
{
if (qTotal[adcch] > 0)
{
marked[2*found]=adcch;
found++;
fromRight = adcch;
}
adcch--;
}
AliDebug(10,Form("Fromleft=%d, Fromright=%d",fromLeft, fromRight));
if ((fromLeft >= 0) && (fromRight >= 0) && (fromLeft < fromRight))
for (adcch = fromLeft+1; adcch < fromRight; adcch++)
qTotal[adcch] = 0;
found = 0;
for (adcch = 0; adcch < 19; adcch++)
if (qTotal[adcch] > 0) found++;
if (found > 4)
{
if (marked[4] == marked[5]) marked[5] = 19;
for (found=0; found<6; found++)
{
qMarked[found] = qTotal[marked[found]] >> 4;
AliDebug(10,Form("ch_%d qTotal %d qTotals %d",marked[found],qTotal[marked[found]],qMarked[found]));
}
Sort6To2Worst(marked[0], marked[3], marked[4], marked[1], marked[2], marked[5],
qMarked[0],
qMarked[3],
qMarked[4],
qMarked[1],
qMarked[2],
qMarked[5],
&worse1, &worse2);
if (worse1 < 19)
{
qTotal[worse1] = 0;
AliDebug(10,Form("Kill ch %d\n",worse1));
}
if (worse2 < 19)
{
qTotal[worse2] = 0;
AliDebug(10,Form("Kill ch %d\n",worse2));
}
}
for (adcch = 0; adcch < 19; adcch++) {
if (qTotal[adcch] > 0)
{
adcLeft = fADCF[adcch ][timebin];
adcCentral = fADCF[adcch+1][timebin];
adcRight = fADCF[adcch+2][timebin];
Int_t regTPFP = fTrapConfig->GetTrapReg(AliTRDtrapConfig::kTPFP, fDetector, fRobPos, fMcmPos);
AliDebug(10, Form("Hit found, time=%d, adcch=%d/%d/%d, adc values=%d/%d/%d, regTPFP=%d, TPHT=%d\n",
timebin, adcch, adcch+1, adcch+2, adcLeft, adcCentral, adcRight, regTPFP,
fTrapConfig->GetTrapReg(AliTRDtrapConfig::kTPHT, fDetector, fRobPos, fMcmPos)));
if (adcLeft < regTPFP) adcLeft = 0; else adcLeft -= regTPFP;
if (adcCentral < regTPFP) adcCentral = 0; else adcCentral -= regTPFP;
if (adcRight < regTPFP) adcRight = 0; else adcRight -= regTPFP;
if (adcCentral == 0)
continue;
ypos = 128*(adcRight - adcLeft) / adcCentral;
if (ypos < 0) ypos = -ypos;
ypos = ypos + fTrapConfig->GetTrapReg((AliTRDtrapConfig::TrapReg_t) (AliTRDtrapConfig::kTPL00 + (ypos & 0x7F)),
fDetector, fRobPos, fMcmPos);
if (adcLeft > adcRight) ypos = -ypos;
Int_t mcLabel[] = {-1, -1, -1};
if (fDigitsManager) {
const Int_t maxLabels = 9;
Int_t label[maxLabels] = { 0 };
Int_t count[maxLabels] = { 0 };
Int_t nLabels = 0;
Int_t padcol[3];
padcol[0] = fFeeParam->GetPadColFromADC(fRobPos, fMcmPos, adcch);
padcol[1] = fFeeParam->GetPadColFromADC(fRobPos, fMcmPos, adcch+1);
padcol[2] = fFeeParam->GetPadColFromADC(fRobPos, fMcmPos, adcch+2);
Int_t padrow = fFeeParam->GetPadRowFromMCM(fRobPos, fMcmPos);
for (Int_t iDict = 0; iDict < 3; iDict++) {
if (!fDict[iDict])
continue;
for (Int_t iPad = 0; iPad < 3; iPad++) {
if (padcol[iPad] < 0)
continue;
Int_t currLabel = fDict[iDict]->GetData(padrow, padcol[iPad], timebin);
AliDebug(10, Form("Read label: %4i for det: %3i, row: %i, col: %i, tb: %i\n", currLabel, fDetector, padrow, padcol[iPad], timebin));
for (Int_t iLabel = 0; iLabel < nLabels; iLabel++) {
if (currLabel == label[iLabel]) {
count[iLabel]++;
currLabel = -1;
break;
}
}
if (currLabel >= 0) {
label[nLabels] = currLabel;
count[nLabels] = 1;
nLabels++;
}
}
}
Int_t index[2*maxLabels];
TMath::Sort(maxLabels, count, index);
for (Int_t i = 0; i < 3; i++) {
if (count[index[i]] <= 0)
break;
mcLabel[i] = label[index[i]];
}
}
AddHitToFitreg(adcch, timebin, qTotal[adcch] >> fgkAddDigits, ypos, mcLabel);
}
}
}
for (Int_t iAdc = 0; iAdc < AliTRDfeeParam::GetNadcMcm(); iAdc++) {
if (fFitReg[iAdc].fNhits != 0) {
AliDebug(2, Form("fitreg[%i]: nHits = %i, sumX = %i, sumY = %i, sumX2 = %i, sumY2 = %i, sumXY = %i", iAdc,
fFitReg[iAdc].fNhits,
fFitReg[iAdc].fSumX,
fFitReg[iAdc].fSumY,
fFitReg[iAdc].fSumX2,
fFitReg[iAdc].fSumY2,
fFitReg[iAdc].fSumXY
));
}
}
}
void AliTRDmcmSim::TrackletSelection()
{
UShort_t adcIdx, i, j, ntracks, tmp;
UShort_t trackletCand[18][2];
ntracks = 0;
for (adcIdx = 0; adcIdx < 18; adcIdx++)
if ( (fFitReg[adcIdx].fNhits
>= fTrapConfig->GetTrapReg(AliTRDtrapConfig::kTPCL, fDetector, fRobPos, fMcmPos)) &&
(fFitReg[adcIdx].fNhits+fFitReg[adcIdx+1].fNhits
>= fTrapConfig->GetTrapReg(AliTRDtrapConfig::kTPCT, fDetector, fRobPos, fMcmPos)))
{
trackletCand[ntracks][0] = adcIdx;
trackletCand[ntracks][1] = fFitReg[adcIdx].fNhits+fFitReg[adcIdx+1].fNhits;
AliDebug(10,Form("%d %2d %4d\n", ntracks, trackletCand[ntracks][0], trackletCand[ntracks][1]));
ntracks++;
};
for (i=0; i<ntracks;i++)
AliDebug(10,Form("%d %d %d\n",i,trackletCand[i][0], trackletCand[i][1]));
if (ntracks > 4)
{
for (j = 0; j < (ntracks-1); j++)
{
for (i = j+1; i < ntracks; i++)
{
if ( (trackletCand[j][1] < trackletCand[i][1]) ||
( (trackletCand[j][1] == trackletCand[i][1]) && (trackletCand[j][0] < trackletCand[i][0]) ) )
{
tmp = trackletCand[j][1];
trackletCand[j][1] = trackletCand[i][1];
trackletCand[i][1] = tmp;
tmp = trackletCand[j][0];
trackletCand[j][0] = trackletCand[i][0];
trackletCand[i][0] = tmp;
}
}
}
ntracks = 4;
}
for (j = 0; j < (ntracks-1); j++)
{
for (i = j+1; i < ntracks; i++)
{
if (trackletCand[j][0] < trackletCand[i][0])
{
tmp = trackletCand[j][1];
trackletCand[j][1] = trackletCand[i][1];
trackletCand[i][1] = tmp;
tmp = trackletCand[j][0];
trackletCand[j][0] = trackletCand[i][0];
trackletCand[i][0] = tmp;
}
}
}
for (i = 0; i < ntracks; i++)
fFitPtr[i] = trackletCand[i][0];
for (i = ntracks; i < 4; i++)
fFitPtr[i] = 31;
AliDebug(10,Form("found %i tracklet candidates\n", ntracks));
for (i = 0; i < 4; i++)
AliDebug(10,Form("fitPtr[%i]: %i\n", i, fFitPtr[i]));
}
void AliTRDmcmSim::FitTracklet()
{
Int_t rndAdd = 0;
Int_t decPlaces = 5;
rndAdd = (1 << (decPlaces-1)) + 1;
Int_t ndriftDp = 5;
Long64_t shift = ((Long64_t) 1 << 32);
Int_t padrow = ((fRobPos >> 1) << 2) | (fMcmPos >> 2);
Int_t yoffs = (((((fRobPos & 0x1) << 2) + (fMcmPos & 0x3)) * 18) << 8) -
((18*4*2 - 18*2 - 1) << 7);
yoffs = yoffs << decPlaces;
Int_t layer = fDetector % 6;
UInt_t scaleY = (UInt_t) ((0.635 + 0.03 * layer)/(256.0 * 160.0e-4) * shift);
UInt_t scaleD = (UInt_t) ((0.635 + 0.03 * layer)/(256.0 * 140.0e-4) * shift);
Int_t deflCorr = (Int_t) fTrapConfig->GetDmemUnsigned(fgkDmemAddrDeflCorr, fDetector, fRobPos, fMcmPos);
Int_t ndrift = (Int_t) fTrapConfig->GetDmemUnsigned(fgkDmemAddrNdrift, fDetector, fRobPos, fMcmPos);
Long64_t mult, temp, denom;
UInt_t q0, q1, pid;
UShort_t nHits;
Int_t slope, offset;
Int_t sumX, sumY, sumXY, sumX2;
Int_t sumY2;
Float_t fitError, fitSlope, fitOffset;
FitReg_t *fit0, *fit1;
for (Int_t cpu = 0; cpu < 4; cpu++) {
if (fFitPtr[cpu] == 31)
{
fMCMT[cpu] = 0x10001000;
}
else
{
fit0 = &fFitReg[fFitPtr[cpu] ];
fit1 = &fFitReg[fFitPtr[cpu]+1];
mult = 1;
mult = mult << (32 + decPlaces);
mult = -mult;
nHits = fit0->fNhits + fit1->fNhits;
sumX = fit0->fSumX + fit1->fSumX;
sumX2 = fit0->fSumX2 + fit1->fSumX2;
denom = ((Long64_t) nHits)*((Long64_t) sumX2) - ((Long64_t) sumX)*((Long64_t) sumX);
mult = mult / denom;
q0 = fit0->fQ0 + fit1->fQ0;
q1 = fit0->fQ1 + fit1->fQ1;
sumY = fit0->fSumY + fit1->fSumY + 256*fit1->fNhits;
sumXY = fit0->fSumXY + fit1->fSumXY + 256*fit1->fSumX;
sumY2 = fit0->fSumY2 + fit1->fSumY2 + 512*fit1->fSumY + 256*256*fit1->fNhits;
slope = nHits*sumXY - sumX * sumY;
offset = sumX2*sumY - sumX * sumXY;
temp = mult * slope;
slope = temp >> 32;
slope = -slope;
temp = mult * offset;
offset = temp >> 32;
offset = offset + yoffs;
AliDebug(10, Form("slope = %i, slope * ndrift = %i, deflCorr: %i",
slope, slope * ndrift, deflCorr));
slope = ((slope * ndrift) >> ndriftDp) + deflCorr;
offset = offset - (fFitPtr[cpu] << (8 + decPlaces));
temp = slope;
temp = temp * scaleD;
slope = (temp >> 32);
temp = offset;
temp = temp * scaleY;
offset = (temp >> 32);
slope = (slope + rndAdd) >> decPlaces;
offset = (offset + rndAdd) >> decPlaces;
AliDebug(5, Form("Det: %3i, ROB: %i, MCM: %2i: deflection: %i, min: %i, max: %i",
fDetector, fRobPos, fMcmPos, slope,
(Int_t) fTrapConfig->GetDmemUnsigned(fgkDmemAddrDeflCutStart + 2*fFitPtr[cpu], fDetector, fRobPos, fMcmPos),
(Int_t) fTrapConfig->GetDmemUnsigned(fgkDmemAddrDeflCutStart + 1 + 2*fFitPtr[cpu], fDetector, fRobPos, fMcmPos)));
AliDebug(5, Form("Fit sums: x = %i, X = %i, y = %i, Y = %i, Z = %i, q0 = %i, q1 = %i",
sumX, sumX2, sumY, sumY2, sumXY, q0, q1));
fitSlope = (Float_t) (nHits * sumXY - sumX * sumY) / (nHits * sumX2 - sumX*sumX);
fitOffset = (Float_t) (sumX2 * sumY - sumX * sumXY) / (nHits * sumX2 - sumX*sumX);
Float_t sx = (Float_t) sumX;
Float_t sx2 = (Float_t) sumX2;
Float_t sy = (Float_t) sumY;
Float_t sy2 = (Float_t) sumY2;
Float_t sxy = (Float_t) sumXY;
fitError = sy2 - (sx2 * sy*sy - 2 * sx * sxy * sy + nHits * sxy*sxy) / (nHits * sx2 - sx*sx);
Bool_t rejected = kFALSE;
if ((slope < ((Int_t) fTrapConfig->GetDmemUnsigned(fgkDmemAddrDeflCutStart + 2*fFitPtr[cpu], fDetector, fRobPos, fMcmPos))) ||
(slope > ((Int_t) fTrapConfig->GetDmemUnsigned(fgkDmemAddrDeflCutStart + 1 + 2*fFitPtr[cpu], fDetector, fRobPos, fMcmPos))))
rejected = kTRUE;
if (rejected && GetApplyCut())
{
fMCMT[cpu] = 0x10001000;
}
else
{
if (slope > 63 || slope < -64) {
AliDebug(1,Form("Overflow in slope: %i, tracklet discarded!", slope));
fMCMT[cpu] = 0x10001000;
continue;
}
slope = slope & 0x7F;
if (offset > 0xfff || offset < -0xfff)
AliWarning("Overflow in offset");
offset = offset & 0x1FFF;
pid = GetPID(q0, q1);
if (pid > 0xff)
AliWarning("Overflow in PID");
pid = pid & 0xFF;
fMCMT[cpu] = (pid << 24) | (padrow << 20) | (slope << 13) | offset;
Int_t mcLabel[] = { -1, -1, -1};
Int_t nHits0 = 0;
Int_t nHits1 = 0;
const Int_t maxLabels = 30;
Int_t label[maxLabels] = {0};
Int_t count[maxLabels] = {0};
Int_t nLabels = 0;
for (Int_t iHit = 0; iHit < fNHits; iHit++) {
if ((fHits[iHit].fChannel - fFitPtr[cpu] < 0) ||
(fHits[iHit].fChannel - fFitPtr[cpu] > 1))
continue;
if (fHits[iHit].fTimebin >= fTrapConfig->GetTrapReg(AliTRDtrapConfig::kTPQS0, fDetector, fRobPos, fMcmPos) &&
fHits[iHit].fTimebin < fTrapConfig->GetTrapReg(AliTRDtrapConfig::kTPQE0, fDetector, fRobPos, fMcmPos))
nHits0++;
if (fHits[iHit].fTimebin >= fTrapConfig->GetTrapReg(AliTRDtrapConfig::kTPQS1, fDetector, fRobPos, fMcmPos) &&
fHits[iHit].fTimebin < fTrapConfig->GetTrapReg(AliTRDtrapConfig::kTPQE1, fDetector, fRobPos, fMcmPos))
nHits1++;
if (fDigitsManager) {
for (Int_t i = 0; i < 3; i++) {
Int_t currLabel = fHits[iHit].fLabel[i];
for (Int_t iLabel = 0; iLabel < nLabels; iLabel++) {
if (currLabel == label[iLabel]) {
count[iLabel]++;
currLabel = -1;
break;
}
}
if (currLabel >= 0 && nLabels < maxLabels) {
label[nLabels] = currLabel;
count[nLabels]++;
nLabels++;
}
}
}
if (fDigitsManager) {
Int_t index[2*maxLabels];
TMath::Sort(maxLabels, count, index);
for (Int_t i = 0; i < 3; i++) {
if (count[index[i]] <= 0)
break;
mcLabel[i] = label[index[i]];
}
}
}
new ((*fTrackletArray)[fTrackletArray->GetEntriesFast()]) AliTRDtrackletMCM((UInt_t) fMCMT[cpu], fDetector*2 + fRobPos%2, fRobPos, fMcmPos);
((AliTRDtrackletMCM*) (*fTrackletArray)[fTrackletArray->GetEntriesFast()-1])->SetLabel(mcLabel);
((AliTRDtrackletMCM*) (*fTrackletArray)[fTrackletArray->GetEntriesFast()-1])->SetNHits(fit0->fNhits + fit1->fNhits);
((AliTRDtrackletMCM*) (*fTrackletArray)[fTrackletArray->GetEntriesFast()-1])->SetNHits0(nHits0);
((AliTRDtrackletMCM*) (*fTrackletArray)[fTrackletArray->GetEntriesFast()-1])->SetNHits1(nHits1);
((AliTRDtrackletMCM*) (*fTrackletArray)[fTrackletArray->GetEntriesFast()-1])->SetQ0(q0);
((AliTRDtrackletMCM*) (*fTrackletArray)[fTrackletArray->GetEntriesFast()-1])->SetQ1(q1);
((AliTRDtrackletMCM*) (*fTrackletArray)[fTrackletArray->GetEntriesFast()-1])->SetSlope(fitSlope);
((AliTRDtrackletMCM*) (*fTrackletArray)[fTrackletArray->GetEntriesFast()-1])->SetOffset(fitOffset);
((AliTRDtrackletMCM*) (*fTrackletArray)[fTrackletArray->GetEntriesFast()-1])->SetError(TMath::Sqrt(TMath::Abs(fitError)/nHits));
if (fgStoreClusters) {
Float_t *res = new Float_t[fNTimeBin];
Float_t *qtot = new Float_t[fNTimeBin];
for (Int_t iTimebin = 0; iTimebin < fNTimeBin; ++iTimebin) {
res[iTimebin] = 0;
qtot[iTimebin] = 0;
}
for (Int_t iHit = 0; iHit < fNHits; iHit++) {
Int_t timebin = fHits[iHit].fTimebin;
if (fHits[iHit].fChannel == fFitPtr[cpu]) {
res[timebin] = fHits[iHit].fYpos - (fitSlope * timebin + fitOffset);
qtot[timebin] = fHits[iHit].fQtot;
}
else if (fHits[iHit].fChannel == fFitPtr[cpu] + 1) {
res[timebin] = fHits[iHit].fYpos + 256 - (fitSlope * timebin + fitOffset);
qtot[timebin] = fHits[iHit].fQtot;
}
}
((AliTRDtrackletMCM*) (*fTrackletArray)[fTrackletArray->GetEntriesFast()-1])->SetClusters(res, qtot, fNTimeBin);
delete [] res;
delete [] qtot;
}
if (fitError < 0)
AliError(Form("Strange fit error: %f from Sx: %i, Sy: %i, Sxy: %i, Sx2: %i, Sy2: %i, nHits: %i",
fitError, sumX, sumY, sumXY, sumX2, sumY2, nHits));
AliDebug(3, Form("fit slope: %f, offset: %f, error: %f",
fitSlope, fitOffset, TMath::Sqrt(TMath::Abs(fitError)/nHits)));
}
}
}
}
void AliTRDmcmSim::Tracklet()
{
if (!fInitialized) {
AliError("Called uninitialized! Nothing done!");
return;
}
fTrackletArray->Delete();
CalcFitreg();
if (fNHits == 0)
return;
TrackletSelection();
FitTracklet();
}
Bool_t AliTRDmcmSim::StoreTracklets()
{
if (fTrackletArray->GetEntriesFast() == 0)
return kTRUE;
AliRunLoader *rl = AliRunLoader::Instance();
AliDataLoader *dl = 0x0;
if (rl)
dl = rl->GetLoader("TRDLoader")->GetDataLoader("tracklets");
if (!dl) {
AliError("Could not get the tracklets data loader!");
return kFALSE;
}
TTree *trackletTree = dl->Tree();
if (!trackletTree) {
dl->MakeTree();
trackletTree = dl->Tree();
}
AliTRDtrackletMCM *trkl = 0x0;
TBranch *trkbranch = trackletTree->GetBranch(fTrklBranchName.Data());
if (!trkbranch)
trkbranch = trackletTree->Branch(fTrklBranchName.Data(), "AliTRDtrackletMCM", &trkl, 32000);
for (Int_t iTracklet = 0; iTracklet < fTrackletArray->GetEntriesFast(); iTracklet++) {
trkl = ((AliTRDtrackletMCM*) (*fTrackletArray)[iTracklet]);
trkbranch->SetAddress(&trkl);
trkbranch->Fill();
}
return kTRUE;
}
void AliTRDmcmSim::WriteData(AliTRDarrayADC *digits)
{
if( !CheckInitialized() )
return;
Int_t offset = (fMcmPos % 4 + 1) * 21 + (fRobPos % 2) * 84 - 1;
if (fTrapConfig->GetTrapReg(AliTRDtrapConfig::kEBSF, fDetector, fRobPos, fMcmPos) != 0)
{
for (Int_t iAdc = 0; iAdc < AliTRDfeeParam::GetNadcMcm(); iAdc++) {
if (~fZSMap[iAdc] == 0) {
for (Int_t iTimeBin = 0; iTimeBin < fNTimeBin; iTimeBin++) {
digits->SetDataByAdcCol(GetRow(), offset - iAdc, iTimeBin, -1);
}
}
else if (iAdc < 2 || iAdc == 20) {
for (Int_t iTimeBin = 0; iTimeBin < fNTimeBin; iTimeBin++) {
digits->SetDataByAdcCol(GetRow(), offset - iAdc, iTimeBin, (fADCR[iAdc][iTimeBin] >> fgkAddDigits) - fgAddBaseline);
}
}
}
}
else {
for (Int_t iAdc = 0; iAdc < AliTRDfeeParam::GetNadcMcm(); iAdc++) {
if (~fZSMap[iAdc] != 0) {
for (Int_t iTimeBin = 0; iTimeBin < fNTimeBin; iTimeBin++) {
digits->SetDataByAdcCol(GetRow(), offset - iAdc, iTimeBin, (fADCF[iAdc][iTimeBin] >> fgkAddDigits) - fgAddBaseline);
}
}
else {
for (Int_t iTimeBin = 0; iTimeBin < fNTimeBin; iTimeBin++) {
digits->SetDataByAdcCol(GetRow(), offset - iAdc, iTimeBin, -1);
}
}
}
}
}
Int_t AliTRDmcmSim::GetPID(Int_t q0, Int_t q1)
{
ULong64_t addrQ0;
ULong64_t addr;
UInt_t nBinsQ0 = fTrapConfig->GetDmemUnsigned(fgkDmemAddrLUTnbins, fDetector, fRobPos, fMcmPos);
UInt_t pidTotalSize = fTrapConfig->GetDmemUnsigned(fgkDmemAddrLUTLength, fDetector, fRobPos, fMcmPos);
if(nBinsQ0==0 || pidTotalSize==0)
return 0;
ULong_t corrQ0 = fTrapConfig->GetDmemUnsigned(fgkDmemAddrLUTcor0, fDetector, fRobPos, fMcmPos);
ULong_t corrQ1 = fTrapConfig->GetDmemUnsigned(fgkDmemAddrLUTcor1, fDetector, fRobPos, fMcmPos);
if(corrQ0==0)
return 0;
addrQ0 = corrQ0;
addrQ0 = (((addrQ0*q0)>>16)>>16);
if(addrQ0 >= nBinsQ0) {
AliDebug(5,Form("Overflow in q0: %llu/4 is bigger then %u", addrQ0, nBinsQ0));
addrQ0 = nBinsQ0 -1;
}
addr = corrQ1;
addr = (((addr*q1)>>16)>>16);
addr = addrQ0 + nBinsQ0*addr;
if(addr >= pidTotalSize) {
AliDebug(5,Form("Overflow in q1. Address %llu/4 is bigger then %u", addr, pidTotalSize));
addr = pidTotalSize -1;
}
UInt_t result = fTrapConfig->GetDmemUnsigned(fgkDmemAddrLUTStart+(addr/4), fDetector, fRobPos, fMcmPos);
return (result>>((addr%4)*8)) & 0xFF;
}
UInt_t AliTRDmcmSim::AddUintClipping(UInt_t a, UInt_t b, UInt_t nbits) const
{
UInt_t sum = a + b;
if (nbits < 32)
{
UInt_t maxv = (1 << nbits) - 1;;
if (sum > maxv)
sum = maxv;
}
else
{
if ((sum < a) || (sum < b))
sum = 0xFFFFFFFF;
}
return sum;
}
void AliTRDmcmSim::Sort2(UShort_t idx1i, UShort_t idx2i, \
UShort_t val1i, UShort_t val2i, \
UShort_t * const idx1o, UShort_t * const idx2o, \
UShort_t * const val1o, UShort_t * const val2o) const
{
if (val1i > val2i)
{
*idx1o = idx1i;
*idx2o = idx2i;
*val1o = val1i;
*val2o = val2i;
}
else
{
*idx1o = idx2i;
*idx2o = idx1i;
*val1o = val2i;
*val2o = val1i;
}
}
void AliTRDmcmSim::Sort3(UShort_t idx1i, UShort_t idx2i, UShort_t idx3i, \
UShort_t val1i, UShort_t val2i, UShort_t val3i, \
UShort_t * const idx1o, UShort_t * const idx2o, UShort_t * const idx3o, \
UShort_t * const val1o, UShort_t * const val2o, UShort_t * const val3o)
{
Int_t sel;
if (val1i > val2i) sel=4; else sel=0;
if (val2i > val3i) sel=sel + 2;
if (val3i > val1i) sel=sel + 1;
switch(sel)
{
case 6 :
case 0 :
*idx1o = idx1i;
*idx2o = idx2i;
*idx3o = idx3i;
*val1o = val1i;
*val2o = val2i;
*val3o = val3i;
break;
case 4 :
*idx1o = idx1i;
*idx2o = idx3i;
*idx3o = idx2i;
*val1o = val1i;
*val2o = val3i;
*val3o = val2i;
break;
case 2 :
*idx1o = idx2i;
*idx2o = idx1i;
*idx3o = idx3i;
*val1o = val2i;
*val2o = val1i;
*val3o = val3i;
break;
case 3 :
*idx1o = idx2i;
*idx2o = idx3i;
*idx3o = idx1i;
*val1o = val2i;
*val2o = val3i;
*val3o = val1i;
break;
case 1 :
*idx1o = idx3i;
*idx2o = idx2i;
*idx3o = idx1i;
*val1o = val3i;
*val2o = val2i;
*val3o = val1i;
break;
case 5 :
*idx1o = idx3i;
*idx2o = idx1i;
*idx3o = idx2i;
*val1o = val3i;
*val2o = val1i;
*val3o = val2i;
break;
default:
AliError("ERROR in Sort3!!!\n");
break;
}
}
void AliTRDmcmSim::Sort6To4(UShort_t idx1i, UShort_t idx2i, UShort_t idx3i, UShort_t idx4i, UShort_t idx5i, UShort_t idx6i, \
UShort_t val1i, UShort_t val2i, UShort_t val3i, UShort_t val4i, UShort_t val5i, UShort_t val6i, \
UShort_t * const idx1o, UShort_t * const idx2o, UShort_t * const idx3o, UShort_t * const idx4o, \
UShort_t * const val1o, UShort_t * const val2o, UShort_t * const val3o, UShort_t * const val4o)
{
UShort_t idx21s, idx22s, idx23s, dummy;
UShort_t val21s, val22s, val23s;
UShort_t idx23as, idx23bs;
UShort_t val23as, val23bs;
Sort3(idx1i, idx2i, idx3i, val1i, val2i, val3i,
idx1o, &idx21s, &idx23as,
val1o, &val21s, &val23as);
Sort3(idx4i, idx5i, idx6i, val4i, val5i, val6i,
idx2o, &idx22s, &idx23bs,
val2o, &val22s, &val23bs);
Sort2(idx23as, idx23bs, val23as, val23bs, &idx23s, &dummy, &val23s, &dummy);
Sort3(idx21s, idx22s, idx23s, val21s, val22s, val23s,
idx3o, idx4o, &dummy,
val3o, val4o, &dummy);
}
void AliTRDmcmSim::Sort6To2Worst(UShort_t idx1i, UShort_t idx2i, UShort_t idx3i, UShort_t idx4i, UShort_t idx5i, UShort_t idx6i, \
UShort_t val1i, UShort_t val2i, UShort_t val3i, UShort_t val4i, UShort_t val5i, UShort_t val6i, \
UShort_t * const idx5o, UShort_t * const idx6o)
{
UShort_t idx21s, idx22s, idx23s, dummy1, dummy2, dummy3, dummy4, dummy5;
UShort_t val21s, val22s, val23s;
UShort_t idx23as, idx23bs;
UShort_t val23as, val23bs;
Sort3(idx1i, idx2i, idx3i, val1i, val2i, val3i,
&dummy1, &idx21s, &idx23as,
&dummy2, &val21s, &val23as);
Sort3(idx4i, idx5i, idx6i, val4i, val5i, val6i,
&dummy1, &idx22s, &idx23bs,
&dummy2, &val22s, &val23bs);
Sort2(idx23as, idx23bs, val23as, val23bs, &idx23s, idx5o, &val23s, &dummy1);
Sort3(idx21s, idx22s, idx23s, val21s, val22s, val23s,
&dummy1, &dummy2, idx6o,
&dummy3, &dummy4, &dummy5);
}
ostream& AliTRDmcmSim::Text(ostream& os)
{
os.iword(fgkFormatIndex) = 0;
return os;
}
ostream& AliTRDmcmSim::Cfdat(ostream& os)
{
os.iword(fgkFormatIndex) = 1;
return os;
}
ostream& AliTRDmcmSim::Raw(ostream& os)
{
os.iword(fgkFormatIndex) = 2;
return os;
}
ostream& operator<<(ostream& os, const AliTRDmcmSim& mcm)
{
if (!mcm.CheckInitialized())
return os;
if (os.iword(AliTRDmcmSim::fgkFormatIndex) == 0) {
os << "MCM " << mcm.fMcmPos << " on ROB " << mcm.fRobPos <<
" in detector " << mcm.fDetector << std::endl;
os << "----- Unfiltered ADC data (10 bit) -----" << std::endl;
os << "ch ";
for (Int_t iChannel = 0; iChannel < AliTRDfeeParam::GetNadcMcm(); iChannel++)
os << std::setw(5) << iChannel;
os << std::endl;
for (Int_t iTimeBin = 0; iTimeBin < mcm.fNTimeBin; iTimeBin++) {
os << "tb " << std::setw(2) << iTimeBin << ":";
for (Int_t iChannel = 0; iChannel < AliTRDfeeParam::GetNadcMcm(); iChannel++) {
os << std::setw(5) << (mcm.fADCR[iChannel][iTimeBin] >> mcm.fgkAddDigits);
}
os << std::endl;
}
os << "----- Filtered ADC data (10+2 bit) -----" << std::endl;
os << "ch ";
for (Int_t iChannel = 0; iChannel < AliTRDfeeParam::GetNadcMcm(); iChannel++)
os << std::setw(4) << iChannel
<< ((~mcm.fZSMap[iChannel] != 0) ? "!" : " ");
os << std::endl;
for (Int_t iTimeBin = 0; iTimeBin < mcm.fNTimeBin; iTimeBin++) {
os << "tb " << std::setw(2) << iTimeBin << ":";
for (Int_t iChannel = 0; iChannel < AliTRDfeeParam::GetNadcMcm(); iChannel++) {
os << std::setw(4) << (mcm.fADCF[iChannel][iTimeBin])
<< (((mcm.fZSMap[iChannel] & (1 << iTimeBin)) == 0) ? "!" : " ");
}
os << std::endl;
}
}
else if(os.iword(AliTRDmcmSim::fgkFormatIndex) == 1) {
Int_t dest = 127;
Int_t addrOffset = 0x2000;
Int_t addrStep = 0x80;
for (Int_t iTimeBin = 0; iTimeBin < mcm.fNTimeBin; iTimeBin++) {
for (Int_t iChannel = 0; iChannel < AliTRDfeeParam::GetNadcMcm(); iChannel++) {
os << std::setw(5) << 10
<< std::setw(5) << addrOffset + iChannel * addrStep + iTimeBin
<< std::setw(5) << (mcm.fADCF[iChannel][iTimeBin])
<< std::setw(5) << dest << std::endl;
}
os << std::endl;
}
}
else if (os.iword(AliTRDmcmSim::fgkFormatIndex) == 2) {
Int_t bufSize = 300;
UInt_t *buf = new UInt_t[bufSize];
Int_t bufLength = mcm.ProduceRawStream(&buf[0], bufSize);
for (Int_t i = 0; i < bufLength; i++)
std::cout << "0x" << std::hex << buf[i] << std::dec << std::endl;
delete [] buf;
}
else {
os << "unknown format set" << std::endl;
}
return os;
}
void AliTRDmcmSim::PrintFitRegXml(ostream& os) const
{
bool tracklet=false;
for (Int_t cpu = 0; cpu < 4; cpu++) {
if(fFitPtr[cpu] != 31)
tracklet=true;
}
if(tracklet==true) {
os << "<nginject>" << std::endl;
os << "<ack roc=\""<< fDetector << "\" cmndid=\"0\">" << std::endl;
os << "<dmem-readout>" << std::endl;
os << "<d det=\"" << fDetector << "\">" << std::endl;
os << " <ro-board rob=\"" << fRobPos << "\">" << std::endl;
os << " <m mcm=\"" << fMcmPos << "\">" << std::endl;
for(int cpu=0; cpu<4; cpu++) {
os << " <c cpu=\"" << cpu << "\">" << std::endl;
if(fFitPtr[cpu] != 31) {
for(int adcch=fFitPtr[cpu]; adcch<fFitPtr[cpu]+2; adcch++) {
os << " <ch chnr=\"" << adcch << "\">"<< std::endl;
os << " <hits>" << fFitReg[adcch].fNhits << "</hits>"<< std::endl;
os << " <q0>" << fFitReg[adcch].fQ0 << "</q0>"<< std::endl;
os << " <q1>" << fFitReg[adcch].fQ1 << "</q1>"<< std::endl;
os << " <sumx>" << fFitReg[adcch].fSumX << "</sumx>"<< std::endl;
os << " <sumxsq>" << fFitReg[adcch].fSumX2 << "</sumxsq>"<< std::endl;
os << " <sumy>" << fFitReg[adcch].fSumY << "</sumy>"<< std::endl;
os << " <sumysq>" << fFitReg[adcch].fSumY2 << "</sumysq>"<< std::endl;
os << " <sumxy>" << fFitReg[adcch].fSumXY << "</sumxy>"<< std::endl;
os << " </ch>" << std::endl;
}
}
os << " </c>" << std::endl;
}
os << " </m>" << std::endl;
os << " </ro-board>" << std::endl;
os << "</d>" << std::endl;
os << "</dmem-readout>" << std::endl;
os << "</ack>" << std::endl;
os << "</nginject>" << std::endl;
}
}
void AliTRDmcmSim::PrintTrackletsXml(ostream& os) const
{
os << "<nginject>" << std::endl;
os << "<ack roc=\""<< fDetector << "\" cmndid=\"0\">" << std::endl;
os << "<dmem-readout>" << std::endl;
os << "<d det=\"" << fDetector << "\">" << std::endl;
os << " <ro-board rob=\"" << fRobPos << "\">" << std::endl;
os << " <m mcm=\"" << fMcmPos << "\">" << std::endl;
Int_t pid, padrow, slope, offset;
for(Int_t cpu=0; cpu<4; cpu++) {
if(fMCMT[cpu] == 0x10001000) {
pid=-1;
padrow=-1;
slope=-1;
offset=-1;
}
else {
pid = (fMCMT[cpu] & 0xFF000000) >> 24;
padrow = (fMCMT[cpu] & 0xF00000 ) >> 20;
slope = (fMCMT[cpu] & 0xFE000 ) >> 13;
offset = (fMCMT[cpu] & 0x1FFF ) ;
}
os << " <trk> <pid>" << pid << "</pid>" << " <padrow>" << padrow << "</padrow>"
<< " <slope>" << slope << "</slope>" << " <offset>" << offset << "</offset>" << "</trk>" << std::endl;
}
os << " </m>" << std::endl;
os << " </ro-board>" << std::endl;
os << "</d>" << std::endl;
os << "</dmem-readout>" << std::endl;
os << "</ack>" << std::endl;
os << "</nginject>" << std::endl;
}
void AliTRDmcmSim::PrintAdcDatTxt(ostream& os) const
{
os << "# MCM " << fMcmPos << " on ROB " << fRobPos <<
" in detector " << fDetector << std::endl;
for (Int_t iTimeBin = 0; iTimeBin < fNTimeBin; iTimeBin++) {
for (Int_t iChannel = 0; iChannel < AliTRDfeeParam::GetNadcMcm(); ++iChannel) {
os << std::setw(5) << (fADCR[iChannel][iTimeBin] >> fgkAddDigits);
}
os << std::endl;
}
}
void AliTRDmcmSim::PrintAdcDatHuman(ostream& os) const
{
os << "MCM " << fMcmPos << " on ROB " << fRobPos <<
" in detector " << fDetector << std::endl;
os << "----- Unfiltered ADC data (10 bit) -----" << std::endl;
os << "ch ";
for (Int_t iChannel = 0; iChannel < AliTRDfeeParam::GetNadcMcm(); iChannel++)
os << std::setw(5) << iChannel;
os << std::endl;
for (Int_t iTimeBin = 0; iTimeBin < fNTimeBin; iTimeBin++) {
os << "tb " << std::setw(2) << iTimeBin << ":";
for (Int_t iChannel = 0; iChannel < AliTRDfeeParam::GetNadcMcm(); iChannel++) {
os << std::setw(5) << (fADCR[iChannel][iTimeBin] >> fgkAddDigits);
}
os << std::endl;
}
os << "----- Filtered ADC data (10+2 bit) -----" << std::endl;
os << "ch ";
for (Int_t iChannel = 0; iChannel < AliTRDfeeParam::GetNadcMcm(); iChannel++)
os << std::setw(4) << iChannel
<< ((~fZSMap[iChannel] != 0) ? "!" : " ");
os << std::endl;
for (Int_t iTimeBin = 0; iTimeBin < fNTimeBin; iTimeBin++) {
os << "tb " << std::setw(2) << iTimeBin << ":";
for (Int_t iChannel = 0; iChannel < AliTRDfeeParam::GetNadcMcm(); iChannel++) {
os << std::setw(4) << (fADCF[iChannel][iTimeBin])
<< (((fZSMap[iChannel] & (1 << iTimeBin)) == 0) ? "!" : " ");
}
os << std::endl;
}
}
void AliTRDmcmSim::PrintAdcDatXml(ostream& os) const
{
os << "<nginject>" << std::endl;
os << "<ack roc=\""<< fDetector << "\" cmndid=\"0\">" << std::endl;
os << "<dmem-readout>" << std::endl;
os << "<d det=\"" << fDetector << "\">" << std::endl;
os << " <ro-board rob=\"" << fRobPos << "\">" << std::endl;
os << " <m mcm=\"" << fMcmPos << "\">" << std::endl;
for(Int_t iChannel = 0; iChannel < AliTRDfeeParam::GetNadcMcm(); iChannel++) {
os << " <ch chnr=\"" << iChannel << "\">" << std::endl;
for (Int_t iTimeBin = 0; iTimeBin < fNTimeBin; iTimeBin++) {
os << "<tb>" << fADCF[iChannel][iTimeBin]/4 << "</tb>";
}
os << " </ch>" << std::endl;
}
os << " </m>" << std::endl;
os << " </ro-board>" << std::endl;
os << "</d>" << std::endl;
os << "</dmem-readout>" << std::endl;
os << "</ack>" << std::endl;
os << "</nginject>" << std::endl;
}
void AliTRDmcmSim::PrintAdcDatDatx(ostream& os, Bool_t broadcast, Int_t timeBinOffset) const
{
fTrapConfig->PrintDatx(os, 2602, 1, 0, 127);
os << std::endl;
Int_t addrOffset = 0x2000;
Int_t addrStep = 0x80;
Int_t addrOffsetEBSIA = 0x20;
for (Int_t iTimeBin = 0; iTimeBin < fNTimeBin; iTimeBin++) {
for (Int_t iChannel = 0; iChannel < AliTRDfeeParam::GetNadcMcm(); iChannel++) {
if ((iTimeBin < timeBinOffset) || (iTimeBin >= fNTimeBin+timeBinOffset)) {
if(broadcast==kFALSE)
fTrapConfig->PrintDatx(os, addrOffset+iChannel*addrStep+addrOffsetEBSIA+iTimeBin, 10, GetRobPos(), GetMcmPos());
else
fTrapConfig->PrintDatx(os, addrOffset+iChannel*addrStep+addrOffsetEBSIA+iTimeBin, 10, 0, 127);
}
else {
if(broadcast==kFALSE)
fTrapConfig->PrintDatx(os, addrOffset+iChannel*addrStep+addrOffsetEBSIA+iTimeBin, (fADCF[iChannel][iTimeBin-timeBinOffset]/4), GetRobPos(), GetMcmPos());
else
fTrapConfig->PrintDatx(os, addrOffset+iChannel*addrStep+addrOffsetEBSIA+iTimeBin, (fADCF[iChannel][iTimeBin-timeBinOffset]/4), 0, 127);
}
}
os << std::endl;
}
}
void AliTRDmcmSim::PrintPidLutHuman()
{
UInt_t result;
UInt_t addrEnd = fgkDmemAddrLUTStart + fTrapConfig->GetDmemUnsigned(fgkDmemAddrLUTLength, fDetector, fRobPos, fMcmPos)/4;
UInt_t nBinsQ0 = fTrapConfig->GetDmemUnsigned(fgkDmemAddrLUTnbins, fDetector, fRobPos, fMcmPos);
std::cout << "nBinsQ0: " << nBinsQ0 << std::endl;
std::cout << "LUT table length: " << fTrapConfig->GetDmemUnsigned(fgkDmemAddrLUTLength, fDetector, fRobPos, fMcmPos) << std::endl;
if (nBinsQ0>0) {
for(UInt_t addr=fgkDmemAddrLUTStart; addr< addrEnd; addr++) {
result = fTrapConfig->GetDmemUnsigned(addr, fDetector, fRobPos, fMcmPos);
std::cout << addr << " # x: " << ((addr-fgkDmemAddrLUTStart)%((nBinsQ0)/4))*4 << ", y: " <<(addr-fgkDmemAddrLUTStart)/(nBinsQ0/4)
<< " # " <<((result>>0)&0xFF)
<< " | " << ((result>>8)&0xFF)
<< " | " << ((result>>16)&0xFF)
<< " | " << ((result>>24)&0xFF) << std::endl;
}
}
}
Bool_t AliTRDmcmSim::ReadPackedConfig(AliTRDtrapConfig *cfg, Int_t hc, UInt_t *data, Int_t size)
{
AliDebugClass(1, "Reading packed configuration");
Int_t det = hc/2;
Int_t idx = 0;
Int_t err = 0;
Int_t step, bwidth, nwords, exitFlag, bitcnt;
UShort_t caddr;
UInt_t dat, msk, header, dataHi;
while (idx < size && *data != 0x00000000) {
Int_t rob = (*data >> 28) & 0x7;
Int_t mcm = (*data >> 24) & 0xf;
AliDebugClass(1, Form("Config of det. %3i MCM %i:%02i (0x%08x)", det, rob, mcm, *data));
data++;
while (idx < size && *data != 0x00000000) {
header = *data;
data++;
idx++;
AliDebugClass(5, Form("read: 0x%08x", header));
if (header & 0x01)
{
dat = (header >> 2) & 0xFFFF;
caddr = (header >> 18) & 0x3FFF;
if (caddr != 0x1FFF)
{
if (header & 0x02)
{
dataHi = *data;
AliDebugClass(5, Form("read: 0x%08x", dataHi));
data++;
idx++;
err += ((dataHi ^ (dat | 1)) & 0xFFFF) != 0;
dat = (dataHi & 0xFFFF0000) | dat;
}
AliDebugClass(5, Form("addr=0x%04x (%s) data=0x%08x\n", caddr, cfg->GetRegName(cfg->GetRegByAddress(caddr)), dat));
if ( ! cfg->Poke(caddr, dat, det, rob, mcm) )
AliDebugClass(5, Form("(single-write): non-existing address 0x%04x containing 0x%08x\n", caddr, header));
if (idx > size)
{
AliDebugClass(5, Form("(single-write): no more data, missing end marker\n"));
return -err;
}
}
else
{
AliDebugClass(5, Form("(single-write): address 0x%04x => old endmarker?\n", caddr));
return err;
}
}
else
{
step = (header >> 1) & 0x0003;
bwidth = ((header >> 3) & 0x001F) + 1;
nwords = (header >> 8) & 0x00FF;
caddr = (header >> 16) & 0xFFFF;
exitFlag = (step == 0) || (step == 3) || (nwords == 0);
if (exitFlag)
break;
switch (bwidth)
{
case 15:
case 10:
case 7:
case 6:
case 5:
{
msk = (1 << bwidth) - 1;
bitcnt = 0;
while (nwords > 0)
{
nwords--;
bitcnt -= bwidth;
if (bitcnt < 0)
{
header = *data;
AliDebugClass(5, Form("read 0x%08x", header));
data++;
idx++;
err += (header & 1);
header = header >> 1;
bitcnt = 31 - bwidth;
}
AliDebugClass(5, Form("addr=0x%04x (%s) data=0x%08x\n", caddr, cfg->GetRegName(cfg->GetRegByAddress(caddr)), header & msk));
if ( ! cfg->Poke(caddr, header & msk, det, rob, mcm) )
AliDebugClass(5, Form("(single-write): non-existing address 0x%04x containing 0x%08x\n", caddr, header));
caddr += step;
header = header >> bwidth;
if (idx >= size)
{
AliDebugClass(5, Form("(block-write): no end marker! %d words read\n", idx));
return -err;
}
}
break;
}
case 31:
{
while (nwords > 0)
{
header = *data;
AliDebugClass(5, Form("read 0x%08x", header));
data++;
idx++;
nwords--;
err += (header & 1);
AliDebugClass(5, Form("addr=0x%04x (%s) data=0x%08x", caddr, cfg->GetRegName(cfg->GetRegByAddress(caddr)), header >> 1));
if ( ! cfg->Poke(caddr, header >> 1, det, rob, mcm) )
AliDebugClass(5, Form("(single-write): non-existing address 0x%04x containing 0x%08x\n", caddr, header));
caddr += step;
if (idx >= size)
{
AliDebugClass(5, Form("no end marker! %d words read", idx));
return -err;
}
}
break;
}
default: return err;
}
}
}
}
AliDebugClass(5, Form("no end marker! %d words read", idx));
return -err;
}