* provided "as is" without express or implied warranty. *
**************************************************************************/
-/*
-$Log$
-*/
+/* $Id$ */
///////////////////////////////////////////////////////////////////////////////
// //
+// Quality Assurance class //
+// Provides standard histograms for every element of the detector //
// //
///////////////////////////////////////////////////////////////////////////////
+
#include <strings.h>
-#include "TObjArray.h"
-#include "TH1.h"
-#include "TList.h"
-#include "TROOT.h"
-#include "TBrowser.h"
-#include "TMath.h"
-#include "TLorentzVector.h"
-#include "TDatabasePDG.h"
-#include "TMath.h"
+#include <TBrowser.h>
+#include <TCanvas.h>
+#include <TDatabasePDG.h>
+#include <TExec.h>
+#include <TH1.h>
+#include <TList.h>
+#include <TLorentzVector.h>
+#include <TMath.h>
+#include <TObjArray.h>
+#include <TPaveLabel.h>
+#include <TROOT.h>
+#include <TVirtualMC.h>
+#include "AliLog.h"
+#include "AliMC.h"
#include "AliMCQA.h"
-#include "AliRun.h"
#include "AliModule.h"
-#include "AliMC.h"
+#include "AliRun.h"
ClassImp(AliMCQA)
+//_______________________________________________________________________
+AliMCQA::AliMCQA():
+ fNdets(0),
+ fNvolumes(0),
+ fQAList(0),
+ fOldId(0),
+ fDetDone(0),
+ fQAHist(0),
+ fVolNames(0),
+ fModNames(0),
+ fMPaveLabel(0),
+ fVPaveLabel(0)
+{
+ //
+ // Default constructor
+ //
+}
-//_____________________________________________________________________________
-AliMCQA::AliMCQA(Int_t ndets): fNdets(ndets)
+//_______________________________________________________________________
+AliMCQA::AliMCQA(const AliMCQA &qa):
+ TObject(qa),
+ fNdets(0),
+ fNvolumes(0),
+ fQAList(0),
+ fOldId(0),
+ fDetDone(0),
+ fQAHist(0),
+ fVolNames(0),
+ fModNames(0),
+ fMPaveLabel(0),
+ fVPaveLabel(0)
+{
+ //
+ // Copy constructor
+ //
+ qa.Copy(*this);
+}
+
+//_______________________________________________________________________
+AliMCQA::AliMCQA(Int_t ndets):
+ fNdets(ndets),
+ fNvolumes(gMC->NofVolumes()),
+ fQAList(new TObjArray(ndets)),
+ fOldId(0),
+ fDetDone(new Int_t[ndets]),
+ fQAHist(new TObjArray(2)),
+ fVolNames(new TObjArray(fNvolumes)),
+ fModNames(new TObjArray(fNdets)),
+ fMPaveLabel(0),
+ fVPaveLabel(0)
{
//
// Constructor, creates the list of lists of histograms
//
TList *list;
-
- fQAList = new TObjArray(ndets);
+ TH1F* h;
+ Int_t i;
+
TObjArray &hist = *fQAList;
char title[100];
//
TObjArray &mods = *(gAlice->Modules());
- AliModule *mod;
TList *dir = gDirectory->GetList();
- for (Int_t i=0; i<ndets; i++) {
+ for (i=0; i<ndets; i++) {
hist[i] = list = new TList();
- mod = (AliModule *) mods[i];
+ AliModule *mod = dynamic_cast<AliModule*>(mods[i]);
// Energy Spectrum
sprintf(title,"Spectrum entering: %s ",mod->GetName());
//
gROOT->GetListOfBrowsables()->Add(this,"AliMCQA");
- fDetDone = new Int_t[fNdets];
+ //
+ // Global QA histograms
+ //
+
+ fQAHist->Add(new TH1F("hMCVcalls","Monte Carlo calls per volume",
+ fNvolumes, 0.5, fNvolumes+0.5));
+ h = dynamic_cast<TH1F*>(dir->FindObject("hMCVcalls"));
+ h->GetListOfFunctions()->Add(new TExec("ex","gAlice->GetMCQA()->AddVolumeName()"));
+ dir->Remove(dir->FindObject("hMCVcalls"));
+ //
+ // Build list of volume names
+ //
+ AliMC * mc = gAlice->GetMCApp();
+ for(i=0;i<fNvolumes;++i) {
+ Int_t mat = gMC->VolId2Mate(i+1);
+ if (mat > 0) {
+ AliModule *mod = dynamic_cast<AliModule*>
+ ((*gAlice->Modules())[mc->DetFromMate(gMC->VolId2Mate(i+1))]);
+ (*fVolNames)[i] = new TNamed(gMC->VolName(i+1),mod->GetName());
+ } else {
+ (*fVolNames)[i] = new TNamed(gMC->VolName(i+1), "Assembly");
+ }
+ }
+
+ fQAHist->Add(new TH1F("hMCMcalls","Monte Carlo calls per module",
+ fNdets, -0.5, fNdets-0.5));
+ h = dynamic_cast<TH1F*>(dir->FindObject("hMCMcalls"));
+ h->GetListOfFunctions()->Add(new TExec("ex","gAlice->GetMCQA()->AddModuleName()"));
+
+ dir->Remove(dir->FindObject("hMCMcalls"));
+ //
+ // Build list of module names
+ //
+ for(i=0;i<fNdets;++i)
+ (*fModNames)[i]=
+ new TNamed((dynamic_cast<AliModule*>((*gAlice->Modules())[i]))->GetName(),"");
}
-//_____________________________________________________________________________
+//_______________________________________________________________________
+void AliMCQA::Copy(TObject &) const
+{
+ AliFatal("Not implemented!");
+}
+
+//_______________________________________________________________________
+AliMCQA::~AliMCQA()
+ {
+ //
+ // Destructor
+ //
+ gROOT->GetListOfBrowsables()->Remove(this);
+ //if program crashes here - it probobly means that
+ //one of added browsables was deleted and not removed previously from that list
+ //skowron
+
+ if (fQAList) {
+ fQAList->Delete();
+ delete fQAList;
+ fQAList = 0;
+ }
+ if (fQAHist) {
+ fQAHist->Delete();
+ delete fQAHist;
+ fQAHist = 0;
+ }
+ if (fVolNames) {
+ fVolNames->Delete();
+ delete fVolNames;
+ fVolNames = 0;
+ }
+ if (fModNames) {
+ fModNames->Delete();
+ delete fModNames;
+ fModNames = 0;
+ }
+ delete [] fDetDone;
+ delete fMPaveLabel;
+ delete fVPaveLabel;
+}
+
+//_______________________________________________________________________
void AliMCQA::Browse(TBrowser *b)
{
//
// of the Root browser.
// It displays the Root Trees and all detectors.
//
-
+ TH1 *hist;
+ //
+ // Global histos first
+ //
+ TIter global(fQAHist);
+ while((hist = dynamic_cast<TH1*>(global())))
+ b->Add(hist,hist->GetTitle());
+ //
+ // Module histograms now
+ //
TIter next(fQAList);
TList *histos;
- TH1 *hist;
- while((histos = (TList*)next())) {
+ while((histos = dynamic_cast<TList*>(next()))) {
TIter next1(histos);
- while((hist = (TH1*)next1())) {
+ while((hist = dynamic_cast<TH1*>(next1())))
b->Add(hist,hist->GetTitle());
- }
}
}
-//_____________________________________________________________________________
+//_______________________________________________________________________
void AliMCQA::PreTrack()
{
+ //
+ // Called before each track
+ //
fOldId=-1;
for(Int_t i=0; i<fNdets; i++) fDetDone[i]=0;
}
-//_____________________________________________________________________________
+//_______________________________________________________________________
void AliMCQA::StepManager(Int_t id)
{
+ //
+ // Called at each step
+ //
+ TH1F *hist;
+ Int_t copy;
+ //
+ // Fill Global histograms first
+ //
+
+
+ static TH1F* mcvcalls = dynamic_cast<TH1F*>(fQAHist->FindObject("hMCVcalls"));
+ mcvcalls->Fill(gMC->CurrentVolID(copy));
+ static TH1F* mcmcalls = dynamic_cast<TH1F*>(fQAHist->FindObject("hMCMcalls"));
+ mcmcalls->Fill(id);
+
+ //
+ // Now the step manager histograms
+ //
if(fOldId != id) {
- TH1F *hist;
- TLorentzVector p, x;
+ static Double_t mpi0=0;
+ static Double_t mpip=0;
+ static Double_t mpim=0;
+ static Double_t mep=0;
+ static Double_t mem=0;
+ Double_t mass = 0;
+ Int_t num = gMC->TrackPid();
+
+ switch (num) {
+ case 111:
+ if (mpi0==0) mpi0=gAlice->PDGDB()->GetParticle(num)->Mass();
+ mass=mpi0;
+ break;
+ case 211:
+ if (mpip==0) mpip=gAlice->PDGDB()->GetParticle(num)->Mass();
+ mass=mpip;
+ break;
+ case -211:
+ if (mpim==0) mpim=gAlice->PDGDB()->GetParticle(num)->Mass();
+ mass=mpim;
+ break;
+ case 11:
+ if (mep==0) mep=gAlice->PDGDB()->GetParticle(num)->Mass();
+ mass=mep;
+ break;
+ case -11:
+ if (mem==0) mem=gAlice->PDGDB()->GetParticle(num)->Mass();
+ mass=mem;
+ break;
+ default:
+ if (gAlice->PDGDB()->GetParticle(num))
+ mass = gAlice->PDGDB()->GetParticle(num)->Mass();
+ break;
+ }
+
+ static TLorentzVector p, x;
gMC->TrackMomentum(p);
gMC->TrackPosition(x);
- Double_t energy = TMath::Max(
- p[3]-gAlice->PDGDB()->GetParticle(gMC->TrackPid())->Mass(),1.e-12);
+ Double_t energy = TMath::Max(p[3]-mass,1.e-12);
if(fOldId > -1) {
if(!fDetDone[fOldId] && !gMC->IsNewTrack()) {
- TList *histold = (TList*) (*fQAList)[fOldId];
- hist = (TH1F*) histold->FindObject("hEnOut");
+ TList *histold = dynamic_cast<TList*>((*fQAList)[fOldId]);
+ hist = dynamic_cast<TH1F*>(histold->FindObject("hEnOut"));
hist->Fill(TMath::Log10(energy));
- hist = (TH1F*) histold->FindObject("hZOut");
+ hist = dynamic_cast<TH1F*>(histold->FindObject("hZOut"));
hist->Fill(x[2]);
fDetDone[fOldId]=1;
}
}
if(!fDetDone[id] && !gMC->IsNewTrack()) {
- TList *histnew = (TList*) (*fQAList)[id];
- hist = (TH1F*) histnew->FindObject("hEnIn");
+ TList *histnew = dynamic_cast<TList*>((*fQAList)[id]);
+ hist = dynamic_cast<TH1F*>(histnew->FindObject("hEnIn"));
hist->Fill(TMath::Log10(energy));
- hist = (TH1F*) histnew->FindObject("hZIn");
+ hist = dynamic_cast<TH1F*>(histnew->FindObject("hZIn"));
hist->Fill(x[2]);
}
fOldId=id;
}
}
+
+//_______________________________________________________________________
+void AliMCQA::AddModuleName()
+{
+ //
+ // Add function DrawModuleName to the module frequency histogram pad
+ //
+ static TVirtualPad* oldpad=0;
+ if(oldpad!=gPad) {
+ gPad->GetCanvas()->FeedbackMode(kTRUE);
+ if(gPad) gPad->AddExec("ex","gAlice->GetMCQA()->DrawModuleName()");
+ DrawPaveLabel(fMPaveLabel);
+ oldpad=gPad;
+ }
+}
+
+//_______________________________________________________________________
+void AliMCQA::AddVolumeName()
+{
+ //
+ // Add function DrawVolumeName to the volume frequency histogram pad
+ //
+ static TVirtualPad* oldpad=0;
+ if(oldpad!=gPad) {
+ gPad->GetCanvas()->FeedbackMode(kTRUE);
+ if(gPad) gPad->AddExec("ex","gAlice->GetMCQA()->DrawVolumeName()");
+ DrawPaveLabel(fVPaveLabel);
+ oldpad=gPad;
+ }
+}
+
+//_______________________________________________________________________
+void AliMCQA::DrawPaveLabel(TPaveLabel *&pv)
+{
+ //
+ // Draws the PaveLabel with the meaning of the bin
+ //
+ float uxmin = gPad->GetUxmin();
+ float uxmax = gPad->GetUxmax();
+ float uymin = gPad->GetUymin();
+ float uymax = gPad->GetUymax();
+ float lx = uxmax-uxmin;
+ float ly = uymax-uymin;
+
+ if(pv) delete pv;
+ pv
+ = new TPaveLabel(uxmin+0.05*lx,uymax-(0.05+0.1)*ly,
+ uxmin+(0.05+0.2)*lx,uymax-0.05*ly,
+ "");
+ pv->Draw();
+}
+
+//_______________________________________________________________________
+Int_t AliMCQA::GetHBin(const char* hname)
+{
+ //
+ // Get the bin where the cursor is
+ //
+ TList *dir = gDirectory->GetList();
+ TH1 *h=dynamic_cast<TH1*>(dir->FindObject(hname));
+
+
+ int px = gPad->GetEventX();
+ Float_t upx = gPad->AbsPixeltoX(px);
+ Float_t x = gPad->PadtoX(upx);
+
+ return h->GetXaxis()->FindBin(x);
+}
+
+//_______________________________________________________________________
+void AliMCQA::DrawModuleName()
+{
+ //
+ // Writes the name of the module of the bin where the cursor is
+ //
+ TObject *select = gPad->GetSelected();
+ if(!select) return;
+
+ Int_t binx = GetHBin("hMCMcalls");
+ if(0<binx && binx<=fNdets) {
+ char lab[15];
+ strcpy(lab,dynamic_cast<TNamed*>((*fModNames)[binx-1])->GetName());
+ fMPaveLabel->SetLabel(lab);
+
+ gPad->Modified();
+ gPad->Update();
+ }
+}
+
+//_______________________________________________________________________
+void AliMCQA::DrawVolumeName()
+{
+ //
+ // Writes the name of the volume:module of the bin where the cursor is
+ //
+ TObject *select = gPad->GetSelected();
+ if(!select) return;
+
+ Int_t binx = GetHBin("hMCVcalls");
+ if(0<binx && binx<=fNvolumes) {
+ char lab[20];
+ sprintf(lab,"%s: %s",dynamic_cast<TNamed*>((*fVolNames)[binx-1])->GetName(),
+ dynamic_cast<TNamed*>((*fVolNames)[binx-1])->GetTitle());
+ fVPaveLabel->SetLabel(lab);
+
+ gPad->Modified();
+ gPad->Update();
+ }
+}
+
+
git://git.uio.no / u / mrichter / AliRoot.git / blobdiff