//____________________________________________________________________
AliAnalysisTaskMCParticleFilter::AliAnalysisTaskMCParticleFilter():
-AliAnalysisTaskSE(),
+ AliAnalysisTaskSE(),
fTrackFilterMother(0x0),
+ fAODMcHeader(0x0),
+ fAODMcParticles(0x0),
fHistList(0x0)
{
// Default constructor
}
TTree *xtree = (TTree*)fxsec->Get("Xsection");
if(!xtree){
- Printf("%s:%d tree not found in the pyxsec.root",(char*)__FILE__,__LINE__);
+ AliWarning(Form("%s:%d tree not found in the pyxsec.root",(char*)__FILE__,__LINE__));
return kTRUE;
}
xtree->SetBranchAddress("xsection",&xsection);
AliAnalysisTaskMCParticleFilter::AliAnalysisTaskMCParticleFilter(const char* name):
AliAnalysisTaskSE(name),
fTrackFilterMother(0x0),
+ fAODMcHeader(0x0),
+ fAODMcParticles(0x0),
fHistList(0x0)
{
// Default constructor
- DefineOutput(1,TList::Class());
+ DefineOutput(1, TList::Class());
}
/*
//____________________________________________________________________
AliAnalysisTaskMCParticleFilter::~AliAnalysisTaskMCParticleFilter()
{
- // if( fTrackFilterMother ) delete fTrackFilterMother;
+
+ if(fAODMcHeader){
+ delete fAODMcHeader;
+ }
+ if(fAODMcParticles){
+ fAODMcParticles->Delete();
+ delete fAODMcParticles;
+ }
}
/*
void AliAnalysisTaskMCParticleFilter::UserCreateOutputObjects()
{
// Create the output container
+ PostData(1,fHistList);
+
if (OutputTree()&&fTrackFilterMother)
OutputTree()->GetUserInfo()->Add(fTrackFilterMother);
// mcparticles
- TClonesArray *tca = new TClonesArray("AliAODMCParticle", 0);
- tca->SetName(AliAODMCParticle::StdBranchName());
- AddAODBranch("TClonesArray",&tca);
+ fAODMcParticles = new TClonesArray("AliAODMCParticle", 0);
+ fAODMcParticles->SetName(AliAODMCParticle::StdBranchName());
+ AddAODBranch("TClonesArray",&fAODMcParticles);
// MC header...
- AliAODMCHeader *mcHeader = new AliAODMCHeader();
- mcHeader->SetName(AliAODMCHeader::StdBranchName());
- AddAODBranch("AliAODMCHeader",&mcHeader);
-
-
+ fAODMcHeader = new AliAODMCHeader();
+ fAODMcHeader->SetName(AliAODMCHeader::StdBranchName());
+ AddAODBranch("AliAODMCHeader",&fAODMcHeader);
AliMCEventHandler *mcH = (AliMCEventHandler*) ((AliAnalysisManager::GetAnalysisManager())->GetMCtruthEventHandler());
AliAODHandler *aodH = dynamic_cast<AliAODHandler*> ((AliAnalysisManager::GetAnalysisManager())->GetOutputEventHandler());
if(!aodH){
- Printf("%s:&d Could not get AODHandler",(char*)__FILE__,__LINE__);
+ AliWarning("Could not get AODHandler");
return;
}
aodH->SetMCEventHandler(mcH);
AliWarning("No MC handler Found");
return;
}
+
+ AliAODHandler *aodH = dynamic_cast<AliAODHandler*> ((AliAnalysisManager::GetAnalysisManager())->GetOutputEventHandler());
+ if(!aodH){
+ AliWarning("Could not get AODHandler");
+ return;
+ }
+
// fetch the output
Int_t nprim = mcE->GetNumberOfPrimaries();
// TODO ADD MC VERTEX
- AliAODMCHeader *aodMCHo = (AliAODMCHeader *) aod->FindListObject("mcHeader");
// Get the proper MC Collision Geometry
AliGenEventHeader* mcEH = mcE->GenEventHeader();
- AliGenPythiaEventHeader *pyH = dynamic_cast<AliGenPythiaEventHeader*>(mcEH);
- AliGenHijingEventHeader *hiH = 0;
- AliCollisionGeometry *colG = 0;
+ AliGenPythiaEventHeader *pyH
= dynamic_cast<AliGenPythiaEventHeader*>(mcEH);
+ AliGenHijingEventHeader *hiH = 0;
+ AliCollisionGeometry *colG = 0;
AliGenDPMjetEventHeader *dpmH = 0;
// it can be only one save some casts
// assuming PYTHIA and HIJING are the most likely ones...
if(!pyH){
- hiH = dynamic_cast<AliGenHijingEventHeader*>(mcEH);
- if(!hiH){
- colG = dynamic_cast<AliCollisionGeometry *>(mcEH);
- if(!colG)dpmH = dynamic_cast<AliGenDPMjetEventHeader*>(mcEH);
- }
+ hiH = dynamic_cast<AliGenHijingEventHeader*>(mcEH);
+ if(!hiH){
+ dpmH = dynamic_cast<AliGenDPMjetEventHeader*>(mcEH);
+ }
}
-
+
+ if (hiH || dpmH) colG = dynamic_cast<AliCollisionGeometry*>(mcEH);
// fetch the trials on a event by event basis, not from pyxsec.root otherwise
// we will get a problem when running on proof since Notify may be called
if (colG) {
- aodMCHo->SetReactionPlaneAngle(colG->ReactionPlaneAngle());
- printf("Found Collision Geometry. Got Reaction Plane %lf\n", colG->ReactionPlaneAngle());
+ fAODMcHeader->SetReactionPlaneAngle(colG->ReactionPlaneAngle());
+ AliInfo(Form("Found Collision Geometry. Got Reaction Plane %lf\n", colG->ReactionPlaneAngle()));
}
if (ip < nprim) {
// Select the primary event
write = kTRUE;
- } else if (part->GetUniqueID() == 4) {
+ } else if (part->GetUniqueID() == kPDecay) {
// Particles from decay
// Check that the decay chain ends at a primary particle
AliMCParticle* mother = mcpart;
Int_t imo = mcpart->GetMother();
- while((imo >= nprim) && (mother->GetUniqueID() == 4)) {
+ while((imo >= nprim) && (mother->GetUniqueID() == kPDecay)) {
mother = (AliMCParticle*) mcE->GetTrack(imo);
imo = mother->GetMother();
}
// Select according to pseudorapidity and production point of primary ancestor
if (imo < nprim && Select(((AliMCParticle*) mcE->GetTrack(imo))->Particle(), rv, zv))write = kTRUE;
- } else if (part->GetUniqueID() == 5) {
+ } else if (part->GetUniqueID() == kPPair) {
// Now look for pair production
Int_t imo = mcpart->GetMother();
if (imo < nprim) {
// Check if the gamma comes from the decay chain of a primary particle
AliMCParticle* mother = (AliMCParticle*) mcE->GetTrack(imo);
imo = mother->GetMother();
- while((imo >= nprim) && (mother->GetUniqueID() == 4)) {
+ while((imo >= nprim) && (mother->GetUniqueID() == kPDecay)) {
mother = (AliMCParticle*) mcE->GetTrack(imo);
imo = mother->GetMother();
}
AliInfo(Form("Taken daughters %d/%d of %d charm",iTaken,iAll,iCharm));
+ aodH->StoreMCParticles();
PostData(1,fHistList);
return;
//
- TTree *tree = (TTree*)GetOutputData(0);
fHistList = (TList*)GetOutputData(1);
if(!fHistList){
Printf("%s:%d Output list not found",(char*)__FILE__,__LINE__);
return;
}
- if(!tree){
- Printf("%s:%d Output tree not found",(char*)__FILE__,__LINE__);
- return;
- }
-
TProfile *hXsec = ((TProfile*)(fHistList->FindObject("h1Xsec")));
TH1F* hTrials = ((TH1F*)(fHistList->FindObject("h1Trials")));
if(!hXsec)return;
Float_t xsec = hXsec->GetBinContent(1);
Float_t trials = hTrials->Integral();
AliInfo(Form("Average -section %.4E and total number of trials %E",xsec,trials));
-
- // This only works if the tree is not a special output
- // in this case it is already written
- /*
- TParameter<float> *x = new TParameter<float>("xsection",xsec);
- tree->GetUserInfo()->Add(x);
- TParameter<float> *t = new TParameter<float>("trials",trials);
- tree->GetUserInfo()->Add(t);
- */
-
}
git://git.uio.no / u / mrichter / AliRoot.git / blobdiff