Fixing index

[u/mrichter/AliRoot.git] / TPC / FindKrClusters.C

\r
/****************************************************************************\r
 *           Origin: A.Matyja amatyja@cern.ch                               *\r
 ****************************************************************************/\r
\r
/*\r
\r
  macro to create array of clusters from TPC digits\r
  input files - galice.root \r
                digits.root - file with digits - usualy use link to galice.root\r
		            - in splitted mode - neccesary to create link to proper file\r
			    \r
   output file - TPC.RecPoints.root\r
\r
\r
//  Warning - if cluster file AliTPCclusters.root already exist - macro exit and don't produce anything\r
	       \r
 \r
*/\r
\r
\r
#ifndef __CINT__\r
#include <iostream.h>\r
#include "AliRun.h"\r
#include "AliTPCv4.h"\r
#include "AliTPCParam.h"\r
#include "AliTPCclusterKr.h"\r
#include "AliTPCclustererKr.h"\r
#include "TFile.h"\r
#include "TStopwatch.h"\r
#include "TTree.h"\r
#endif\r
\r
Int_t FindKrClusters(){\r
\r
  //\r
  //Load DataBase\r
  //\r
  //char *ocdbpath ="local:///afs/cern.ch/alice/tpctest/OCDB";\r
  //char *ocdbpath ="local:///home/matyja/baza/OCDB";\r
  char *ocdbpath ="local:///data/baza/OCDB";\r
  if (ocdbpath==0){\r
    ocdbpath="alien://folder=/alice/data/2007/LHC07w/OCDB/";\r
  }\r
  printf("OCDB PATH = %s\n",ocdbpath); \r
  AliCDBManager * man = AliCDBManager::Instance();\r
  man->SetDefaultStorage(ocdbpath);\r
  man->SetRun(0);\r
\r
  AliRunLoader* rl = AliRunLoader::Open("galice.root");\r
  if (rl == 0x0) {\r
    cerr<<"Can not open session"<<endl;\r
    return 1;\r
  }\r
  \r
  AliTPCLoader *tpcl = (AliTPCLoader*)rl->GetLoader("TPCLoader");\r
  if (tpcl == 0x0) {\r
    cerr<<"Can not get TPC Loader"<<endl;\r
    return 1;\r
  }\r
\r
  if (tpcl->LoadDigits()) {\r
    cerr<<"Error occured while loading digits"<<endl;\r
    return 1;\r
  }\r
\r
  if (rl->LoadgAlice()) {\r
    cerr<<"Error occured while LoadgAlice"<<endl;\r
    return 1;\r
  }\r
  \r
  gAlice=rl->GetAliRun();\r
  if (!gAlice) {\r
    cerr<<"Can't get gAlice !\n";\r
    return 1;\r
  }\r
\r
  TDirectory *cwd = gDirectory;\r
\r
  AliTPCv4 *tpc = (AliTPCv4*)gAlice->GetDetector("TPC");\r
  Int_t ver = tpc->IsVersion(); \r
  cerr<<"TPC version "<<ver<<" has been found !\n";\r
\r
  rl->CdGAFile();\r
  \r
  AliTPCParam *param=(AliTPCParamSR *)gDirectory->Get("75x40_100x60_150x60");\r
  if (!param) {cerr<<"TPC parameters have not been found !\n"; return 4;}\r
  \r
  AliTPCDigitsArray *digarr=new AliTPCDigitsArray;\r
  digarr->Setup(param);\r
\r
  cerr<<"It has begun"<<endl;  \r
  TStopwatch timer;\r
  timer.Start();\r
  cwd->cd();\r
\r
  TTree *output_tree;\r
  \r
  AliTPCclustererKr *clusters = new AliTPCclustererKr();\r
  clusters->SetParam(param);\r
  clusters->SetOutput(output_tree);\r
\r
  clusters->SetMinAdc(3);//signal threshold (everything below is treated as 0)\r
  clusters->SetMinTimeBins(2);//number of neighbouring timebins\r
  clusters->SetMaxPadRangeCm(5.);//distance of the cluster center to the center of a pad (in cm)\r
  clusters->SetMaxRowRangeCm(5.);//distance of the cluster center to the center of a padrow (in cm)\r
  clusters->SetMaxTimeRange(7.);//distance of the cluster center to the max time bin on a pad (in tackts)\r
  //ie. fabs(centerT - time)<7\r
\r
  clusters->SetIsolCut(3);//set isolation cut threshold\r
  clusters->SetValueToSize(3.1);//cut reduce peak at 0\r
\r
  Int_t nevmax=rl->GetNumberOfEvents();//number of events in run\r
  for(Int_t nev=0;nev<nevmax /*&& nev<1*/ ;nev++){\r
    rl->GetEvent(nev);\r
    \r
    TTree* input_tree= tpcl->TreeD();//tree with digits\r
    if (input_tree == 0x0){\r
      cerr << "Can not get TreeD for event " <<nev<<endl;\r
      continue;\r
    }\r
    digarr->ConnectTree(input_tree);\r
    clusters->SetInput(input_tree);\r
    clusters->SetDigArr(digarr);\r
    cout<<"Processing event "<<nev<<endl;\r
    clusters->FinderIO();\r
\r
  }\r
  delete clusters;\r
  timer.Stop(); timer.Print();\r
  \r
  delete rl;//cleans everything\r
  \r
  return 0;\r
}\r