[u/mrichter/AliRoot.git] / STEER / AliTagCreator.cxx

/**************************************************************************
 * Author: Panos Christakoglou.                                           *
 * Contributors are mentioned in the code where appropriate.              *
 *                                                                        *
 * Permission to use, copy, modify and distribute this software and its   *
 * documentation strictly for non-commercial purposes is hereby granted   *
 * without fee, provided that the above copyright notice appears in all   *
 * copies and that both the copyright notice and this permission notice   *
 * appear in the supporting documentation. The authors make no claims     *
 * about the suitability of this software for any purpose. It is          *
 * provided "as is" without express or implied warranty.                  *
 **************************************************************************/

/* $Id$ */

//-----------------------------------------------------------------
//           AliTagCreator class
//   This is the class to deal with the tag creation (post process)
//   Origin: Panos Christakoglou, UOA-CERN, Panos.Christakoglou@cern.ch
//-----------------------------------------------------------------

//ROOT
#include <TFile.h>
#include <TString.h>
#include <TTree.h>

//ROOT-AliEn
#include <TGrid.h>
#include <TGridResult.h>

//AliRoot
#include "AliRunTag.h"
#include "AliEventTag.h"
#include "AliESD.h"
#include "AliESDVertex.h"
#include "AliLog.h"


#include "AliTagCreator.h"


ClassImp(AliTagCreator)


//______________________________________________________________________________
AliTagCreator::AliTagCreator() //local mode
{
  //==============Default constructor for a AliTagCreator==================
  fgridpath = "";
  fUser = "";
  fPasswd = "";  
  fSE = "";   
  fHost = "";
  fPort = 0; 
  fStorage = 0; 
}

//______________________________________________________________________________
AliTagCreator::AliTagCreator(const char *host, Int_t port, const char *username)
{
  //==============Default constructor for a AliTagCreator==================
  fStorage = 0; 
  fgridpath = "";
  fHost = host;
  fUser = username;
  fPort = port;

  TString grid = "alien://";
  grid += fHost;
  grid += ":";
  grid += port;
  
  //connect to the grid
  TGrid::Connect(grid.Data(),fUser.Data());
  if(!gGrid)
    {
      AliError("Connection failed!!!");
      return;
    }
}


//______________________________________________________________________________
AliTagCreator::AliTagCreator(const char *host, Int_t port, const char *username, const char *passwd)
{
  //==============Default constructor for a AliTagCreator==================
  fStorage = 0; 
  fgridpath = "";
  fHost = host;
  fUser = username;
  fPasswd = passwd;
  fPort = port;

  TString grid = "alien://";
  grid += fHost;
  grid += ":";
  grid += port;
  
  //connect to the grid
  TGrid::Connect(grid.Data(),fUser.Data(),fPasswd.Data());
   if(!gGrid)
    {
      AliError("Connection failed!!!");
      return;
    }
}


//______________________________________________________________________________
AliTagCreator::~AliTagCreator()
{
//================Default destructor for a AliTagCreator=======================
}

//______________________________________________________________________________
void AliTagCreator::SetStorage(Int_t storage)
{
  // Sets correctly the storage: 0 for local, 1 for GRID
  fStorage = storage;
  if(fStorage == 0)
    AliInfo(Form("Tags will be stored locally...."));
  if(fStorage == 1)
    AliInfo(Form("Tags will be stored in the grid...."));
  if((fStorage != 0)&&(fStorage != 1))
    {
      AliInfo(Form("Storage was not properly set!!!"));
      abort();
    }  
}

//______________________________________________________________________________
Bool_t AliTagCreator::ConnectToGrid(const char *host, Int_t port, const char *username)
{
  // Connects to a given AliEn API service
  fHost = host;
  fUser = username;
  fPort = port;

  TString grid = "alien://";
  grid += fHost;
  grid += ":";
  grid += port;
  
  //connect to the grid
  TGrid::Connect(grid.Data(),fUser.Data());
  if(!gGrid)
    {
      AliError("Connection failed!!!");
      return kFALSE;
    } //connect to the grid

  return kTRUE;
}

//______________________________________________________________________________
Bool_t AliTagCreator::ReadESDCollection(TGridResult *fresult)
{
  // Reads the entry of the TGridResult and creates the tags
  Int_t nEntries = fresult->GetEntries();

  TString alienUrl;
  const char *guid;
  const char *md5;
  const char *turl;
  Long64_t size = -1;

  Int_t counter = 0;
  for(Int_t i = 0; i < nEntries; i++)
    {
      alienUrl = fresult->GetKey(i,"turl");
      guid = fresult->GetKey(i,"guid");
      if(fresult->GetKey(i,"size"))
	size = atol (fresult->GetKey(i,"size"));
      md5 = fresult->GetKey(i,"md5");
      turl = fresult->GetKey(i,"turl");
      if(md5 && !strlen(guid))
	md5 = 0;
      if(guid && !strlen(guid))
	guid = 0;

      TFile *f = TFile::Open(alienUrl,"READ");
      //CreateTag(f,guid,counter);
      CreateTag(f,guid,md5,turl,size,counter);
      f->Close();
      delete f;	 
      counter += 1;
    }//grid result loop

  return kTRUE;
}


//_____________________________________________________________________________
//void AliTagCreator::CreateTag(TFile* file, const char *guid, Int_t Counter)
void AliTagCreator::CreateTag(TFile* file, const char *guid, const char *md5, const char *turl, Long64_t size, Int_t Counter)
{
  // Creates the tags for all the events in a given ESD file
  Int_t ntrack;
  Int_t nProtons, nKaons, nPions, nMuons, nElectrons;
  Int_t nPos, nNeg, nNeutr;
  Int_t nK0s, nNeutrons, nPi0s, nGamas;
  Int_t nCh1GeV, nCh3GeV, nCh10GeV;
  Int_t nMu1GeV, nMu3GeV, nMu10GeV;
  Int_t nEl1GeV, nEl3GeV, nEl10GeV;
  Float_t maxPt = .0, meanPt = .0, totalP = .0;

  AliRunTag *tag = new AliRunTag();
  AliEventTag *evTag = new AliEventTag();
  TTree ttag("T","A Tree with event tags");
  TBranch * btag = ttag.Branch("AliTAG", "AliRunTag", &tag);
  btag->SetCompressionLevel(9);
  
  AliInfo(Form("Creating the tags......."));	
  
  Int_t firstEvent = 0,lastEvent = 0;
  TTree *t = (TTree*) file->Get("esdTree");
  TBranch * b = t->GetBranch("ESD");
  AliESD *esd = 0;
  b->SetAddress(&esd);
  
  tag->SetRunId(esd->GetRunNumber());
  
  Int_t iNumberOfEvents = b->GetEntries();
  for (Int_t iEventNumber = 0; iEventNumber < iNumberOfEvents; iEventNumber++)
    {
      ntrack = 0;
      nPos = 0;
      nNeg = 0;
      nNeutr =0;
      nK0s = 0;
      nNeutrons = 0;
      nPi0s = 0;
      nGamas = 0;
      nProtons = 0;
      nKaons = 0;
      nPions = 0;
      nMuons = 0;
      nElectrons = 0;	  
      nCh1GeV = 0;
      nCh3GeV = 0;
      nCh10GeV = 0;
      nMu1GeV = 0;
      nMu3GeV = 0;
      nMu10GeV = 0;
      nEl1GeV = 0;
      nEl3GeV = 0;
      nEl10GeV = 0;
      maxPt = .0;
      meanPt = .0;
      totalP = .0;
      
      b->GetEntry(iEventNumber);
      const AliESDVertex * vertexIn = esd->GetVertex();
      
      for (Int_t iTrackNumber = 0; iTrackNumber < esd->GetNumberOfTracks(); iTrackNumber++)
	{
	  AliESDtrack * esdTrack = esd->GetTrack(iTrackNumber);
	  UInt_t status = esdTrack->GetStatus();
	  
	  //select only tracks with ITS refit
	  if ((status&AliESDtrack::kITSrefit)==0) continue;
	  
	  //select only tracks with TPC refit-->remove extremely high Pt tracks
	  if ((status&AliESDtrack::kTPCrefit)==0) continue;
	  
	  //select only tracks with the "combined PID"
	  if ((status&AliESDtrack::kESDpid)==0) continue;
	  Double_t p[3];
	  esdTrack->GetPxPyPz(p);
	  Double_t momentum = sqrt(pow(p[0],2) + pow(p[1],2) + pow(p[2],2));
	  Double_t fPt = sqrt(pow(p[0],2) + pow(p[1],2));
	  totalP += momentum;
	  meanPt += fPt;
	  if(fPt > maxPt)
	    maxPt = fPt;
	  
	  if(esdTrack->GetSign() > 0)
	    {
	      nPos++;
	      if(fPt > 1.0)
		nCh1GeV++;
	      if(fPt > 3.0)
		nCh3GeV++;
	      if(fPt > 10.0)
		nCh10GeV++;
	    }
	  if(esdTrack->GetSign() < 0)
	    {
	      nNeg++;
	      if(fPt > 1.0)
		nCh1GeV++;
	      if(fPt > 3.0)
		nCh3GeV++;
	      if(fPt > 10.0)
		nCh10GeV++;
	    }
	  if(esdTrack->GetSign() == 0)
	    nNeutr++;
	  
	  //PID
	  Double_t prob[10];
	  esdTrack->GetESDpid(prob);
	  
	  //K0s
	  if ((prob[8]>prob[7])&&(prob[8]>prob[6])&&(prob[8]>prob[5])&&(prob[8]>prob[4])&&(prob[8]>prob[3])&&(prob[8]>prob[2])&&(prob[8]>prob[1])&&(prob[8]>prob[0]))
	    nK0s++;
	  //neutrons
	  if ((prob[7]>prob[8])&&(prob[7]>prob[6])&&(prob[7]>prob[5])&&(prob[7]>prob[4])&&(prob[7]>prob[3])&&(prob[7]>prob[2])&&(prob[7]>prob[1])&&(prob[7]>prob[0]))
	    nNeutrons++; 
	  //pi0s
	  if ((prob[6]>prob[8])&&(prob[6]>prob[7])&&(prob[6]>prob[5])&&(prob[6]>prob[4])&&(prob[6]>prob[3])&&(prob[6]>prob[2])&&(prob[6]>prob[1])&&(prob[6]>prob[0]))
	    nPi0s++;
	  //gamas
	  if ((prob[5]>prob[8])&&(prob[5]>prob[7])&&(prob[5]>prob[6])&&(prob[5]>prob[4])&&(prob[5]>prob[3])&&(prob[5]>prob[2])&&(prob[5]>prob[1])&&(prob[5]>prob[0]))
	    nGamas++;
	  //protons
	  if ((prob[4]>prob[8])&&(prob[4]>prob[7])&&(prob[4]>prob[6])&&(prob[4]>prob[5])&&(prob[4]>prob[3])&&(prob[4]>prob[2])&&(prob[4]>prob[1])&&(prob[4]>prob[0]))
	    nProtons++;
	  //kaons
	  if ((prob[3]>prob[8])&&(prob[3]>prob[7])&&(prob[3]>prob[6])&&(prob[3]>prob[5])&&(prob[3]>prob[4])&&(prob[3]>prob[2])&&(prob[3]>prob[1])&&(prob[3]>prob[0]))
	    nKaons++;
	  //pions
	  if ((prob[2]>prob[8])&&(prob[2]>prob[7])&&(prob[2]>prob[6])&&(prob[2]>prob[5])&&(prob[2]>prob[4])&&(prob[2]>prob[3])&&(prob[2]>prob[1])&&(prob[2]>prob[0]))
	    nPions++; 
	  //muons
	  if ((prob[1]>prob[8])&&(prob[1]>prob[7])&&(prob[1]>prob[6])&&(prob[1]>prob[5])&&(prob[1]>prob[4])&&(prob[1]>prob[3])&&(prob[1]>prob[2])&&(prob[1]>prob[0]))
	    {
	      nMuons++;
	      if(fPt > 1.0)
		nMu1GeV++;
	      if(fPt > 3.0)
		nMu3GeV++;
	      if(fPt > 10.0)
		nMu10GeV++;
	    }
	  //electrons
	  if ((prob[0]>prob[8])&&(prob[0]>prob[7])&&(prob[0]>prob[6])&&(prob[0]>prob[5])&&(prob[0]>prob[4])&&(prob[0]>prob[3])&&(prob[0]>prob[2])&&(prob[0]>prob[1]))
	    {
	      nElectrons++;
	      if(fPt > 1.0)
		nEl1GeV++;
	      if(fPt > 3.0)
		nEl3GeV++;
	      if(fPt > 10.0)
		nEl10GeV++;
	    }
	  
	  
	  ntrack++;
	}//track loop
      // Fill the event tags 
      if(ntrack != 0)
	meanPt = meanPt/ntrack;
      
      evTag->SetEventId(iEventNumber+1);
      evTag->SetGUID(guid);
      evTag->SetMD5(md5);
      evTag->SetTURL(turl);
      evTag->SetSize(size);
      evTag->SetVertexX(vertexIn->GetXv());
      evTag->SetVertexY(vertexIn->GetYv());
      evTag->SetVertexZ(vertexIn->GetZv());
      
      evTag->SetT0VertexZ(esd->GetT0zVertex());
      
      evTag->SetTrigger(esd->GetTrigger());
      
      evTag->SetZDCNeutronEnergy(esd->GetZDCNEnergy());
      evTag->SetZDCProtonEnergy(esd->GetZDCPEnergy());
      evTag->SetZDCEMEnergy(esd->GetZDCEMEnergy());
      evTag->SetNumOfParticipants(esd->GetZDCParticipants());
      
      
      evTag->SetNumOfTracks(esd->GetNumberOfTracks());
      evTag->SetNumOfPosTracks(nPos);
      evTag->SetNumOfNegTracks(nNeg);
      evTag->SetNumOfNeutrTracks(nNeutr);
      
      evTag->SetNumOfV0s(esd->GetNumberOfV0s());
      evTag->SetNumOfCascades(esd->GetNumberOfCascades());
      evTag->SetNumOfKinks(esd->GetNumberOfKinks());
      evTag->SetNumOfPMDTracks(esd->GetNumberOfPmdTracks());
      
      evTag->SetNumOfProtons(nProtons);
      evTag->SetNumOfKaons(nKaons);
      evTag->SetNumOfPions(nPions);
      evTag->SetNumOfMuons(nMuons);
      evTag->SetNumOfElectrons(nElectrons);
      evTag->SetNumOfPhotons(nGamas);
      evTag->SetNumOfPi0s(nPi0s);
      evTag->SetNumOfNeutrons(nNeutrons);
      evTag->SetNumOfKaon0s(nK0s);
      
      evTag->SetNumOfChargedAbove1GeV(nCh1GeV);
      evTag->SetNumOfChargedAbove3GeV(nCh3GeV);
      evTag->SetNumOfChargedAbove10GeV(nCh10GeV);
      evTag->SetNumOfMuonsAbove1GeV(nMu1GeV);
      evTag->SetNumOfMuonsAbove3GeV(nMu3GeV);
      evTag->SetNumOfMuonsAbove10GeV(nMu10GeV);
      evTag->SetNumOfElectronsAbove1GeV(nEl1GeV);
      evTag->SetNumOfElectronsAbove3GeV(nEl3GeV);
      evTag->SetNumOfElectronsAbove10GeV(nEl10GeV);
      
      evTag->SetNumOfPHOSTracks(esd->GetNumberOfPHOSParticles());
      evTag->SetNumOfEMCALTracks(esd->GetNumberOfEMCALParticles());
      
      evTag->SetTotalMomentum(totalP);
      evTag->SetMeanPt(meanPt);
      evTag->SetMaxPt(maxPt);
  
      tag->AddEventTag(*evTag);
    }
  lastEvent = iNumberOfEvents;
  
  t->Delete("");
	
  ttag.Fill();
  tag->Clear();

  TString localFileName = "Run"; localFileName += tag->GetRunId(); 
  localFileName += ".Event"; localFileName += firstEvent; localFileName += "_"; localFileName += lastEvent; localFileName += "."; localFileName += Counter;
  localFileName += ".ESD.tag.root";

  TString alienLocation = "/alien";
  alienLocation += gGrid->Pwd();
  alienLocation += fgridpath.Data();
  alienLocation += "/";
  alienLocation +=  localFileName;
  alienLocation += "?se=";
  alienLocation += fSE.Data();

  TString fileName;
  
  if(fStorage == 0)
    {
      fileName = localFileName.Data();      
      AliInfo(Form("Writing tags to local file: %s",fileName.Data()));
    }
  if(fStorage == 1)
    {
      fileName = alienLocation.Data();
      AliInfo(Form("Writing tags to grid file: %s",fileName.Data()));
    }

  TFile* ftag = TFile::Open(fileName, "recreate");
  ftag->cd();
  ttag.Write();
  ftag->Close();

  delete ftag;
  delete esd;

  delete tag;
  delete evTag;
}
Commit	Line	Data
b45e5084	1	/**************************************************************************
	2	* Author: Panos Christakoglou. *
	3	* Contributors are mentioned in the code where appropriate. *
	4	* *
	5	* Permission to use, copy, modify and distribute this software and its *
	6	* documentation strictly for non-commercial purposes is hereby granted *
	7	* without fee, provided that the above copyright notice appears in all *
	8	* copies and that both the copyright notice and this permission notice *
	9	* appear in the supporting documentation. The authors make no claims *
	10	* about the suitability of this software for any purpose. It is *
	11	* provided "as is" without express or implied warranty. *
	12	**************************************************************************/
	13
	14	/* $Id$ */
	15
	16	//-----------------------------------------------------------------
	17	// AliTagCreator class
	18	// This is the class to deal with the tag creation (post process)
	19	// Origin: Panos Christakoglou, UOA-CERN, Panos.Christakoglou@cern.ch
	20	//-----------------------------------------------------------------
	21
	22	//ROOT
b45e5084	23	#include <TFile.h>
b45e5084	24	#include <TString.h>
cb1645b7	25	#include <TTree.h>
b45e5084	26
	27	//ROOT-AliEn
	28	#include <TGrid.h>
b45e5084	29	#include <TGridResult.h>
b45e5084	30
	31	//AliRoot
	32	#include "AliRunTag.h"
	33	#include "AliEventTag.h"
	34	#include "AliESD.h"
	35	#include "AliESDVertex.h"
b45e5084	36	#include "AliLog.h"
	37
	38
	39	#include "AliTagCreator.h"
	40
	41
	42	ClassImp(AliTagCreator)
	43
	44
	45	//______________________________________________________________________________
	46	AliTagCreator::AliTagCreator() //local mode
	47	{
	48	//==============Default constructor for a AliTagCreator==================
28afeb2e	49	fgridpath = "";
b45e5084	50	fUser = "";
	51	fPasswd = "";
	52	fSE = "";
b45e5084	53	fHost = "";
b45e5084	54	fPort = 0;
28afeb2e	55	fStorage = 0;
b45e5084	56	}
	57
	58	//______________________________________________________________________________
	59	AliTagCreator::AliTagCreator(const char host, Int_t port, const char username)
	60	{
	61	//==============Default constructor for a AliTagCreator==================
28afeb2e	62	fStorage = 0;
28afeb2e	63	fgridpath = "";
b45e5084	64	fHost = host;
	65	fUser = username;
	66	fPort = port;
	67
	68	TString grid = "alien://";
	69	grid += fHost;
	70	grid += ":";
	71	grid += port;
	72
	73	//connect to the grid
	74	TGrid::Connect(grid.Data(),fUser.Data());
	75	if(!gGrid)
	76	{
	77	AliError("Connection failed!!!");
	78	return;
	79	}
	80	}
	81
	82
	83	//______________________________________________________________________________
	84	AliTagCreator::AliTagCreator(const char host, Int_t port, const char username, const char *passwd)
	85	{
	86	//==============Default constructor for a AliTagCreator==================
28afeb2e	87	fStorage = 0;
28afeb2e	88	fgridpath = "";
b45e5084	89	fHost = host;
	90	fUser = username;
	91	fPasswd = passwd;
	92	fPort = port;
	93
	94	TString grid = "alien://";
	95	grid += fHost;
	96	grid += ":";
	97	grid += port;
	98
	99	//connect to the grid
	100	TGrid::Connect(grid.Data(),fUser.Data(),fPasswd.Data());
	101	if(!gGrid)
	102	{
	103	AliError("Connection failed!!!");
	104	return;
	105	}
	106	}
	107
	108
	109	//______________________________________________________________________________
	110	AliTagCreator::~AliTagCreator()
	111	{
	112	//================Default destructor for a AliTagCreator=======================
	113	}
	114
	115	//______________________________________________________________________________
28afeb2e	116	void AliTagCreator::SetStorage(Int_t storage)
28afeb2e	117	{
cb1645b7	118	// Sets correctly the storage: 0 for local, 1 for GRID
28afeb2e	119	fStorage = storage;
	120	if(fStorage == 0)
	121	AliInfo(Form("Tags will be stored locally...."));
	122	if(fStorage == 1)
	123	AliInfo(Form("Tags will be stored in the grid...."));
	124	if((fStorage != 0)&&(fStorage != 1))
	125	{
	126	AliInfo(Form("Storage was not properly set!!!"));
	127	abort();
	128	}
	129	}
	130
	131	//______________________________________________________________________________
b45e5084	132	Bool_t AliTagCreator::ConnectToGrid(const char host, Int_t port, const char username)
b45e5084	133	{
cb1645b7	134	// Connects to a given AliEn API service
b45e5084	135	fHost = host;
	136	fUser = username;
	137	fPort = port;
	138
	139	TString grid = "alien://";
	140	grid += fHost;
	141	grid += ":";
	142	grid += port;
	143
	144	//connect to the grid
	145	TGrid::Connect(grid.Data(),fUser.Data());
	146	if(!gGrid)
	147	{
	148	AliError("Connection failed!!!");
	149	return kFALSE;
	150	} //connect to the grid
	151
	152	return kTRUE;
	153	}
	154
	155	//______________________________________________________________________________
cb1645b7	156	Bool_t AliTagCreator::ReadESDCollection(TGridResult *fresult)
b45e5084	157	{
cb1645b7	158	// Reads the entry of the TGridResult and creates the tags
cb1645b7	159	Int_t nEntries = fresult->GetEntries();
b45e5084	160
cb1645b7	161	TString alienUrl;
b45e5084	162	const char *guid;
e16601cf	163	const char *md5;
	164	const char *turl;
	165	Long64_t size = -1;
	166
b45e5084	167	Int_t counter = 0;
cb1645b7	168	for(Int_t i = 0; i < nEntries; i++)
09c52d4a	169	{
cb1645b7	170	alienUrl = fresult->GetKey(i,"turl");
09c52d4a	171	guid = fresult->GetKey(i,"guid");
e16601cf	172	if(fresult->GetKey(i,"size"))
	173	size = atol (fresult->GetKey(i,"size"));
	174	md5 = fresult->GetKey(i,"md5");
	175	turl = fresult->GetKey(i,"turl");
	176	if(md5 && !strlen(guid))
	177	md5 = 0;
	178	if(guid && !strlen(guid))
	179	guid = 0;
	180
cb1645b7	181	TFile *f = TFile::Open(alienUrl,"READ");
e16601cf	182	//CreateTag(f,guid,counter);
e16601cf	183	CreateTag(f,guid,md5,turl,size,counter);
09c52d4a	184	f->Close();
09c52d4a	185	delete f;
b45e5084	186	counter += 1;
09c52d4a	187	}//grid result loop
b45e5084	188
	189	return kTRUE;
	190	}
	191
	192
	193	//_____________________________________________________________________________
e16601cf	194	//void AliTagCreator::CreateTag(TFile* file, const char *guid, Int_t Counter)
e16601cf	195	void AliTagCreator::CreateTag(TFile* file, const char guid, const char md5, const char *turl, Long64_t size, Int_t Counter)
b45e5084	196	{
cb1645b7	197	// Creates the tags for all the events in a given ESD file
b45e5084	198	Int_t ntrack;
cb1645b7	199	Int_t nProtons, nKaons, nPions, nMuons, nElectrons;
	200	Int_t nPos, nNeg, nNeutr;
	201	Int_t nK0s, nNeutrons, nPi0s, nGamas;
	202	Int_t nCh1GeV, nCh3GeV, nCh10GeV;
	203	Int_t nMu1GeV, nMu3GeV, nMu10GeV;
	204	Int_t nEl1GeV, nEl3GeV, nEl10GeV;
	205	Float_t maxPt = .0, meanPt = .0, totalP = .0;
b45e5084	206
b45e5084	207	AliRunTag *tag = new AliRunTag();
b45e5084	208	AliEventTag *evTag = new AliEventTag();
	209	TTree ttag("T","A Tree with event tags");
	210	TBranch * btag = ttag.Branch("AliTAG", "AliRunTag", &tag);
	211	btag->SetCompressionLevel(9);
	212
	213	AliInfo(Form("Creating the tags......."));
	214
	215	Int_t firstEvent = 0,lastEvent = 0;
	216	TTree t = (TTree) file->Get("esdTree");
	217	TBranch * b = t->GetBranch("ESD");
	218	AliESD *esd = 0;
	219	b->SetAddress(&esd);
	220
	221	tag->SetRunId(esd->GetRunNumber());
	222
cb1645b7	223	Int_t iNumberOfEvents = b->GetEntries();
cb1645b7	224	for (Int_t iEventNumber = 0; iEventNumber < iNumberOfEvents; iEventNumber++)
b45e5084	225	{
b45e5084	226	ntrack = 0;
cb1645b7	227	nPos = 0;
	228	nNeg = 0;
	229	nNeutr =0;
	230	nK0s = 0;
	231	nNeutrons = 0;
	232	nPi0s = 0;
	233	nGamas = 0;
	234	nProtons = 0;
	235	nKaons = 0;
	236	nPions = 0;
	237	nMuons = 0;
	238	nElectrons = 0;
	239	nCh1GeV = 0;
	240	nCh3GeV = 0;
	241	nCh10GeV = 0;
	242	nMu1GeV = 0;
	243	nMu3GeV = 0;
	244	nMu10GeV = 0;
	245	nEl1GeV = 0;
	246	nEl3GeV = 0;
	247	nEl10GeV = 0;
	248	maxPt = .0;
	249	meanPt = .0;
	250	totalP = .0;
b45e5084	251
cb1645b7	252	b->GetEntry(iEventNumber);
cb1645b7	253	const AliESDVertex * vertexIn = esd->GetVertex();
b45e5084	254
cb1645b7	255	for (Int_t iTrackNumber = 0; iTrackNumber < esd->GetNumberOfTracks(); iTrackNumber++)
b45e5084	256	{
cb1645b7	257	AliESDtrack * esdTrack = esd->GetTrack(iTrackNumber);
cb1645b7	258	UInt_t status = esdTrack->GetStatus();
b45e5084	259
	260	//select only tracks with ITS refit
	261	if ((status&AliESDtrack::kITSrefit)==0) continue;
	262
	263	//select only tracks with TPC refit-->remove extremely high Pt tracks
	264	if ((status&AliESDtrack::kTPCrefit)==0) continue;
	265
	266	//select only tracks with the "combined PID"
	267	if ((status&AliESDtrack::kESDpid)==0) continue;
	268	Double_t p[3];
cb1645b7	269	esdTrack->GetPxPyPz(p);
cb1645b7	270	Double_t momentum = sqrt(pow(p[0],2) + pow(p[1],2) + pow(p[2],2));
b45e5084	271	Double_t fPt = sqrt(pow(p[0],2) + pow(p[1],2));
cb1645b7	272	totalP += momentum;
	273	meanPt += fPt;
	274	if(fPt > maxPt)
	275	maxPt = fPt;
b45e5084	276
cb1645b7	277	if(esdTrack->GetSign() > 0)
b45e5084	278	{
cb1645b7	279	nPos++;
b45e5084	280	if(fPt > 1.0)
cb1645b7	281	nCh1GeV++;
b45e5084	282	if(fPt > 3.0)
cb1645b7	283	nCh3GeV++;
b45e5084	284	if(fPt > 10.0)
cb1645b7	285	nCh10GeV++;
b45e5084	286	}
cb1645b7	287	if(esdTrack->GetSign() < 0)
b45e5084	288	{
cb1645b7	289	nNeg++;
b45e5084	290	if(fPt > 1.0)
cb1645b7	291	nCh1GeV++;
b45e5084	292	if(fPt > 3.0)
cb1645b7	293	nCh3GeV++;
b45e5084	294	if(fPt > 10.0)
cb1645b7	295	nCh10GeV++;
b45e5084	296	}
cb1645b7	297	if(esdTrack->GetSign() == 0)
cb1645b7	298	nNeutr++;
b45e5084	299
	300	//PID
	301	Double_t prob[10];
cb1645b7	302	esdTrack->GetESDpid(prob);
b45e5084	303
	304	//K0s
	305	if ((prob[8]>prob[7])&&(prob[8]>prob[6])&&(prob[8]>prob[5])&&(prob[8]>prob[4])&&(prob[8]>prob[3])&&(prob[8]>prob[2])&&(prob[8]>prob[1])&&(prob[8]>prob[0]))
cb1645b7	306	nK0s++;
b45e5084	307	//neutrons
b45e5084	308	if ((prob[7]>prob[8])&&(prob[7]>prob[6])&&(prob[7]>prob[5])&&(prob[7]>prob[4])&&(prob[7]>prob[3])&&(prob[7]>prob[2])&&(prob[7]>prob[1])&&(prob[7]>prob[0]))
cb1645b7	309	nNeutrons++;
b45e5084	310	//pi0s
b45e5084	311	if ((prob[6]>prob[8])&&(prob[6]>prob[7])&&(prob[6]>prob[5])&&(prob[6]>prob[4])&&(prob[6]>prob[3])&&(prob[6]>prob[2])&&(prob[6]>prob[1])&&(prob[6]>prob[0]))
cb1645b7	312	nPi0s++;
b45e5084	313	//gamas
b45e5084	314	if ((prob[5]>prob[8])&&(prob[5]>prob[7])&&(prob[5]>prob[6])&&(prob[5]>prob[4])&&(prob[5]>prob[3])&&(prob[5]>prob[2])&&(prob[5]>prob[1])&&(prob[5]>prob[0]))
cb1645b7	315	nGamas++;
b45e5084	316	//protons
b45e5084	317	if ((prob[4]>prob[8])&&(prob[4]>prob[7])&&(prob[4]>prob[6])&&(prob[4]>prob[5])&&(prob[4]>prob[3])&&(prob[4]>prob[2])&&(prob[4]>prob[1])&&(prob[4]>prob[0]))
cb1645b7	318	nProtons++;
b45e5084	319	//kaons
b45e5084	320	if ((prob[3]>prob[8])&&(prob[3]>prob[7])&&(prob[3]>prob[6])&&(prob[3]>prob[5])&&(prob[3]>prob[4])&&(prob[3]>prob[2])&&(prob[3]>prob[1])&&(prob[3]>prob[0]))
cb1645b7	321	nKaons++;
b45e5084	322	//pions
b45e5084	323	if ((prob[2]>prob[8])&&(prob[2]>prob[7])&&(prob[2]>prob[6])&&(prob[2]>prob[5])&&(prob[2]>prob[4])&&(prob[2]>prob[3])&&(prob[2]>prob[1])&&(prob[2]>prob[0]))
cb1645b7	324	nPions++;
b45e5084	325	//muons
	326	if ((prob[1]>prob[8])&&(prob[1]>prob[7])&&(prob[1]>prob[6])&&(prob[1]>prob[5])&&(prob[1]>prob[4])&&(prob[1]>prob[3])&&(prob[1]>prob[2])&&(prob[1]>prob[0]))
	327	{
cb1645b7	328	nMuons++;
b45e5084	329	if(fPt > 1.0)
cb1645b7	330	nMu1GeV++;
b45e5084	331	if(fPt > 3.0)
cb1645b7	332	nMu3GeV++;
b45e5084	333	if(fPt > 10.0)
cb1645b7	334	nMu10GeV++;
b45e5084	335	}
	336	//electrons
	337	if ((prob[0]>prob[8])&&(prob[0]>prob[7])&&(prob[0]>prob[6])&&(prob[0]>prob[5])&&(prob[0]>prob[4])&&(prob[0]>prob[3])&&(prob[0]>prob[2])&&(prob[0]>prob[1]))
	338	{
cb1645b7	339	nElectrons++;
b45e5084	340	if(fPt > 1.0)
cb1645b7	341	nEl1GeV++;
b45e5084	342	if(fPt > 3.0)
cb1645b7	343	nEl3GeV++;
b45e5084	344	if(fPt > 10.0)
cb1645b7	345	nEl10GeV++;
b45e5084	346	}
	347
	348
	349
	350	ntrack++;
	351	}//track loop
	352	// Fill the event tags
	353	if(ntrack != 0)
cb1645b7	354	meanPt = meanPt/ntrack;
b45e5084	355
cb1645b7	356	evTag->SetEventId(iEventNumber+1);
b45e5084	357	evTag->SetGUID(guid);
e16601cf	358	evTag->SetMD5(md5);
	359	evTag->SetTURL(turl);
	360	evTag->SetSize(size);
cb1645b7	361	evTag->SetVertexX(vertexIn->GetXv());
	362	evTag->SetVertexY(vertexIn->GetYv());
	363	evTag->SetVertexZ(vertexIn->GetZv());
b45e5084	364
	365	evTag->SetT0VertexZ(esd->GetT0zVertex());
	366
	367	evTag->SetTrigger(esd->GetTrigger());
	368
	369	evTag->SetZDCNeutronEnergy(esd->GetZDCNEnergy());
	370	evTag->SetZDCProtonEnergy(esd->GetZDCPEnergy());
	371	evTag->SetZDCEMEnergy(esd->GetZDCEMEnergy());
	372	evTag->SetNumOfParticipants(esd->GetZDCParticipants());
	373
	374
	375	evTag->SetNumOfTracks(esd->GetNumberOfTracks());
cb1645b7	376	evTag->SetNumOfPosTracks(nPos);
	377	evTag->SetNumOfNegTracks(nNeg);
	378	evTag->SetNumOfNeutrTracks(nNeutr);
b45e5084	379
	380	evTag->SetNumOfV0s(esd->GetNumberOfV0s());
	381	evTag->SetNumOfCascades(esd->GetNumberOfCascades());
	382	evTag->SetNumOfKinks(esd->GetNumberOfKinks());
	383	evTag->SetNumOfPMDTracks(esd->GetNumberOfPmdTracks());
	384
cb1645b7	385	evTag->SetNumOfProtons(nProtons);
	386	evTag->SetNumOfKaons(nKaons);
	387	evTag->SetNumOfPions(nPions);
	388	evTag->SetNumOfMuons(nMuons);
	389	evTag->SetNumOfElectrons(nElectrons);
	390	evTag->SetNumOfPhotons(nGamas);
	391	evTag->SetNumOfPi0s(nPi0s);
	392	evTag->SetNumOfNeutrons(nNeutrons);
	393	evTag->SetNumOfKaon0s(nK0s);
b45e5084	394
cb1645b7	395	evTag->SetNumOfChargedAbove1GeV(nCh1GeV);
	396	evTag->SetNumOfChargedAbove3GeV(nCh3GeV);
	397	evTag->SetNumOfChargedAbove10GeV(nCh10GeV);
	398	evTag->SetNumOfMuonsAbove1GeV(nMu1GeV);
	399	evTag->SetNumOfMuonsAbove3GeV(nMu3GeV);
	400	evTag->SetNumOfMuonsAbove10GeV(nMu10GeV);
	401	evTag->SetNumOfElectronsAbove1GeV(nEl1GeV);
	402	evTag->SetNumOfElectronsAbove3GeV(nEl3GeV);
	403	evTag->SetNumOfElectronsAbove10GeV(nEl10GeV);
b45e5084	404
	405	evTag->SetNumOfPHOSTracks(esd->GetNumberOfPHOSParticles());
	406	evTag->SetNumOfEMCALTracks(esd->GetNumberOfEMCALParticles());
	407
cb1645b7	408	evTag->SetTotalMomentum(totalP);
	409	evTag->SetMeanPt(meanPt);
	410	evTag->SetMaxPt(maxPt);
b45e5084	411
cb1645b7	412	tag->AddEventTag(*evTag);
b45e5084	413	}
cb1645b7	414	lastEvent = iNumberOfEvents;
b45e5084	415
	416	t->Delete("");
	417
	418	ttag.Fill();
	419	tag->Clear();
	420
cb1645b7	421	TString localFileName = "Run"; localFileName += tag->GetRunId();
	422	localFileName += ".Event"; localFileName += firstEvent; localFileName += "_"; localFileName += lastEvent; localFileName += "."; localFileName += Counter;
	423	localFileName += ".ESD.tag.root";
b45e5084	424
cb1645b7	425	TString alienLocation = "/alien";
	426	alienLocation += gGrid->Pwd();
	427	alienLocation += fgridpath.Data();
	428	alienLocation += "/";
	429	alienLocation += localFileName;
	430	alienLocation += "?se=";
	431	alienLocation += fSE.Data();
28afeb2e	432
cb1645b7	433	TString fileName;
28afeb2e	434
	435	if(fStorage == 0)
	436	{
cb1645b7	437	fileName = localFileName.Data();
cb1645b7	438	AliInfo(Form("Writing tags to local file: %s",fileName.Data()));
28afeb2e	439	}
	440	if(fStorage == 1)
	441	{
cb1645b7	442	fileName = alienLocation.Data();
cb1645b7	443	AliInfo(Form("Writing tags to grid file: %s",fileName.Data()));
28afeb2e	444	}
b45e5084	445
cb1645b7	446	TFile* ftag = TFile::Open(fileName, "recreate");
b45e5084	447	ftag->cd();
	448	ttag.Write();
	449	ftag->Close();
	450
	451	delete ftag;
	452	delete esd;
	453
	454	delete tag;
b45e5084	455	delete evTag;
	456	}
	457