Request by Martin: added flag for big output

[u/mrichter/AliRoot.git] / PWG2 / SPECTRA / AliAnalysisTaskProtons.cxx

diff --git a/PWG2/SPECTRA/AliAnalysisTaskProtons.cxx b/PWG2/SPECTRA/AliAnalysisTaskProtons.cxx
index 297bbffab88bcc56c7d3accdd458ae254d9c2883..d70e974a82b079372bb646aaf43a0f33a19cbca6 100644 (file)
--- a/PWG2/SPECTRA/AliAnalysisTaskProtons.cxx
+++ b/PWG2/SPECTRA/AliAnalysisTaskProtons.cxx
@@ -37,7 +37,7 @@ ClassImp(AliAnalysisTaskProtons)
 AliAnalysisTaskProtons::AliAnalysisTaskProtons()
   : AliAnalysisTask(), fESD(0), fAOD(0), fMC(0),
     fListAnalysis(0), fListQA(0), fHistEventStats(0), 
-    fProtonAnalysis(0), fCutCanvas(0) {
+  fProtonAnalysis(0) {//, fCutCanvas(0) {
   //Dummy constructor
   
 }
@@ -46,7 +46,7 @@ AliAnalysisTaskProtons::AliAnalysisTaskProtons()
 AliAnalysisTaskProtons::AliAnalysisTaskProtons(const char *name) 
   : AliAnalysisTask(name, ""), fESD(0), fAOD(0), fMC(0),
     fListAnalysis(0), fListQA(0), fHistEventStats(0), 
-    fProtonAnalysis(0), fCutCanvas(0) {
+  fProtonAnalysis(0) {//, fCutCanvas(0) {
   // Constructor
   
   // Define input and output slots here
@@ -101,13 +101,13 @@ void AliAnalysisTaskProtons::ConnectInputData(Option_t *) {
 void AliAnalysisTaskProtons::CreateOutputObjects() {
   // Create output objects
   // Called once
-  char *gCutName[5] = {"Total","Triggered","Offline trigger",
-                      "Vertex","Analyzed"};
+  TString gCutName[5] = {"Total","Triggered","Offline trigger",
+                        "Vertex","Analyzed"};
   fHistEventStats = new TH1F("fHistEventStats",
                             "Event statistics;;N_{events}",
                             5,0.5,5.5);
   for(Int_t i = 1; i <= 5; i++) 
-    fHistEventStats->GetXaxis()->SetBinLabel(i,gCutName[i-1]);
+    fHistEventStats->GetXaxis()->SetBinLabel(i,gCutName[i-1].Data());
 
   fListAnalysis = new TList();
   fListAnalysis->Add(fProtonAnalysis->GetProtonYPtHistogram());
@@ -117,12 +117,14 @@ void AliAnalysisTaskProtons::CreateOutputObjects() {
   fListAnalysis->Add(fProtonAnalysis->GetAntiProtonContainer());
   fListAnalysis->Add(fHistEventStats);
 
-  fListQA = new TList();
-  fListQA->SetName("fListQA");
-  fListQA->Add(fProtonAnalysis->GetQAList());
-  fListQA->Add(dynamic_cast<AliProtonAnalysisBase*>(fProtonAnalysis->GetProtonAnalysisBaseObject())->GetVertexQAList());
+  if(dynamic_cast<AliProtonAnalysisBase*>(fProtonAnalysis->GetProtonAnalysisBaseObject())->IsQARun()) {
+    fListQA = new TList();
+    fListQA->SetName("fListQA");
+    fListQA->Add(fProtonAnalysis->GetQAList());
+    fListQA->Add(dynamic_cast<AliProtonAnalysisBase*>(fProtonAnalysis->GetProtonAnalysisBaseObject())->GetVertexQAList());
+  }
 
-  fCutCanvas = dynamic_cast<AliProtonAnalysisBase*>(fProtonAnalysis->GetProtonAnalysisBaseObject())->GetListOfCuts();
+  //fCutCanvas = dynamic_cast<AliProtonAnalysisBase*>(fProtonAnalysis->GetProtonAnalysisBaseObject())->GetListOfCuts();
 }
 
 //________________________________________________________________________
@@ -237,8 +239,9 @@ void AliAnalysisTaskProtons::Exec(Option_t *) {
 
   // Post output data.
   PostData(0, fListAnalysis);
-  PostData(1, fListQA);
-  PostData(2, fCutCanvas);
+  if(dynamic_cast<AliProtonAnalysisBase*>(fProtonAnalysis->GetProtonAnalysisBaseObject())->IsQARun()) 
+    PostData(1, fListQA);
+  //PostData(2, fCutCanvas);
 }      
 
 //________________________________________________________________________