X-Git-Url: http://git.uio.no/git/?a=blobdiff_plain;f=ANALYSIS%2FAliAnalysisTaskTagCreator.cxx;h=48c4b961e4abb1a2d5f2ff1faab457014118a279;hb=39d9c03cb41cbc47026f9f1d646aa8f72359596e;hp=82bc7ff408f53f1eaa0d637dfd21c23911946d20;hpb=7e3c2e0415867f22626cb23bb1779cfcddf716af;p=u%2Fmrichter%2FAliRoot.git

diff --git a/ANALYSIS/AliAnalysisTaskTagCreator.cxx b/ANALYSIS/AliAnalysisTaskTagCreator.cxx
index 82bc7ff408f..48c4b961e4a 100644
--- a/ANALYSIS/AliAnalysisTaskTagCreator.cxx
+++ b/ANALYSIS/AliAnalysisTaskTagCreator.cxx
@@ -14,11 +14,14 @@
  **************************************************************************/
 
 /* $Id$ */
- 
+
+#include <Riostream.h>
+
 #include <TChain.h>
 #include <TTree.h>
 #include <TString.h>
 #include <TFile.h>
+#include <TSystem.h>
 
 #include "AliAnalysisTaskTagCreator.h"
 #include "AliAnalysisManager.h"
@@ -31,6 +34,10 @@
 #include "AliAODTagCreator.h"
 #include "AliLog.h"
 
+
+using std::cout;
+using std::endl;
+using std::ofstream;
 ClassImp(AliAnalysisTaskTagCreator)
 
 ////////////////////////////////////////////////////////////////////////
@@ -41,7 +48,9 @@ AliAnalysisTaskTagCreator::AliAnalysisTaskTagCreator():
     fFirstFile(kTRUE),
     fRunTag(0), 
     fTreeT(0),
-    fTagCreator(0)
+    fTagCreator(0),
+    fAODFileName(""),
+    fGUID(0)
 {
   // Default constructor
 }
@@ -52,7 +61,9 @@ AliAnalysisTaskTagCreator::AliAnalysisTaskTagCreator(const char* name):
     fFirstFile(kTRUE),
     fRunTag(0), 
     fTreeT(0),
-    fTagCreator(0)
+    fTagCreator(0),
+    fAODFileName(""),
+    fGUID(0)
 {
   // Constructor
     DefineOutput(1, TTree::Class()); 	
@@ -67,43 +78,62 @@ void AliAnalysisTaskTagCreator::UserCreateOutputObjects()
     TBranch * btag = fTreeT->Branch("AliTAG", "AliRunTag", &fRunTag);
     btag->SetCompressionLevel(9);
     fTagCreator = new AliAODTagCreator();
+    PostData(1, fTreeT);
 }
 
 void AliAnalysisTaskTagCreator::Init()
 {
-    // Initialization
-    if (fDebug > 1) AliInfo("Init() \n");
-    // Call configuration file
+
 }
 
+void AliAnalysisTaskTagCreator::ConnectInputData(Option_t * /*option*/)
+{
+    // Initialization
+    const char* turl = gSystem->Getenv("ALIEN_JDL_OUTPUTDIR");
+    TString sturl = turl;
+    
+    if (sturl.Length() != 0) {
+      fAODFileName = "alien://";
+      fAODFileName += turl;
+      fAODFileName += "/AliAOD.root";
+    }  
+}
 
 void AliAnalysisTaskTagCreator::UserExec(Option_t */*option*/)
 {
+
     // Create Tags for the current event
     AliEventTag* evtTag = new AliEventTag();
     fTagCreator->FillEventTag(AODEvent(), evtTag);
     // Reference to the input file
-    TString fturl, fturltemp, fguid;
+    TString fturl, fturltemp, guid;
     
     TString opt(fInputHandler->GetAnalysisType());
     opt.ToLower();
     
     TFile *file = OutputTree()->GetCurrentFile();
     const TUrl *url = file->GetEndpointUrl();
-    fguid = file->GetUUID().AsString();
-    if (opt.Contains("grid")) {
-	fturltemp = "alien://"; fturltemp += url->GetFile();
-	fturl = fturltemp(0,fturltemp.Index(".root",5,0,TString::kExact)+5);
+    guid = file->GetUUID().AsString();
+    if (fAODFileName.Length() != 0) {
+	fturl = fAODFileName;
+	guid  = fGUID;
     } else {
 	fturl = url->GetFile();
     }
-    evtTag->SetGUID(fguid);
-    if(opt.Contains("grid")) {
-	evtTag->SetMD5(0);
-	evtTag->SetTURL(fturl);
-	evtTag->SetSize(0);
+
+    if (fRunTag->GetFileId(guid) == -1) {
+      AliFileTag *eftag = new AliFileTag();
+
+      eftag->SetGUID(guid);
+      if(fAODFileName.Length() != 0) {
+	eftag->SetMD5("");
+	eftag->SetTURL(fturl);
+	eftag->SetSize(0);
+      }
+      else eftag->SetPath(fturl);
+
+      fRunTag->AddFileTag(eftag);
     }
-    else evtTag->SetPath(fturl);
     //
     // Add the event tag
     fRunTag->AddEventTag(*evtTag);
@@ -122,17 +152,31 @@ void AliAnalysisTaskTagCreator::FinishTaskOutput()
 Bool_t AliAnalysisTaskTagCreator::Notify()
 {
     // Notify file change
-    if (!fFirstFile) {
-	if (fInputHandler->GetRunTag()) fRunTag->CopyStandardContent(fInputHandler->GetRunTag());	    
-	fTreeT->Fill();
-	fRunTag->Clear();
-    } else {
-	fFirstFile = kFALSE;
-    }
+    fInputHandler = (AliInputEventHandler*) 
+      ((AliAnalysisManager::GetAnalysisManager())->GetInputEventHandler());
     return kTRUE;
 }
 
 
+void AliAnalysisTaskTagCreator::GetGUID(TString &guid) {
+    // Get the guid of the AliAOD.root file
+    ofstream myfile ("guid.txt");
+    if (myfile.is_open()) {
+	TFile *f = TFile::Open("AliAOD.root","read");
+	if(f->IsOpen()) {
+	    guid = f->GetUUID().AsString();
+	    myfile << "AliAOD.root \t"<<f->GetUUID().AsString();
+	    cout<<guid.Data()<<endl;
+	    myfile.close();
+	}
+	else cout<<"Input file not found"<<endl;
+	f->Close();
+    }
+    else cout<<"Output file can't be created..."<<endl;
+}
+
+
+
 void AliAnalysisTaskTagCreator::Terminate(Option_t */*option*/)
 {
 // Terminate analysis
@@ -140,3 +184,4 @@ void AliAnalysisTaskTagCreator::Terminate(Option_t */*option*/)
     if (fDebug > 1) printf("AnalysisTagCreator: Terminate() \n");
 }
 
+