Attempt to monitor what file is read and merged by what job

[u/mrichter/AliRoot.git] / PWGPP / macros / rawmerge.C

//\r
// Macro to create the "raw" data file with selected events\r
// Paramaters:\r
//  eventListFileName - input ascii file with list of chunks and event numbers - two collumns\r
//  osplit            - split the file after given threhshold value\r
//\r
//  TGrid * alien = TGrid::Connect("alien://",0,0,"t");\r
\r
void rawmerge(const char *eventListFileName,\r
              const char *outputDirectoryURI,\r
              Long64_t osplit=-1)\r
{\r
  TGrid * alien = TGrid::Connect("alien://",0,0,"t");\r
  \r
  Int_t eventNumber;\r
  FILE *files=fopen(eventListFileName,"r");\r
  if (!files) {\r
    fprintf(stderr,"error: could not read event list file \"%s\". Exiting.\n",eventListFileName);\r
    return;\r
  }\r
  char iURI[1000];\r
  TString  iURIold;\r
  char oURI[1000];\r
  TFile *ifile=0;\r
  TFile *ofile=0;\r
  TTree *itree=0;\r
  TTree *otree=0;\r
  Long64_t ievent;\r
  Long64_t oevent;\r
  Int_t ofilenumber=0;\r
  Int_t line=0;\r
  Int_t eventold=0;\r
  while (!feof(files)) {\r
    ++line;\r
    if (fscanf(files,"%s %d\n",iURI,&ievent)!=2) {\r
      fprintf(stderr,"warning: corrupted event line (%d) in input file, skipping it...\n",line);\r
      continue;\r
    } \r
    printf("> processing \"%s\" event %d...\n",iURI,ievent);\r
    if (ievent==eventold) { printf("duplicated continue\n"),continue;}\r
    //\r
    if (iURIold.Contains(iURI)==0){ \r
      printf("NF: %s\n",iURI);\r
      delete ifile;ifile=0;\r
      ifile=TFile::Open(iURI);\r
      if (!ifile) {\r
	fprintf(stderr,"warning: could not open file for event \"%s\", skipping it...\n",iURI);\r
	continue;\r
      }\r
      iURIold=iURI;\r
    }else{\r
      printf("OF: %s\n",iURI);\r
      iURIold=iURI; \r
    }\r
    //\r
    TTree *itree=dynamic_cast<TTree*>(ifile->Get("RAW"));\r
    if (!itree) {\r
      fprintf(stderr,"warning: could not find RAW tree for event \"%s\", skipping it...\n",iURI);\r
      continue;\r
    }\r
\r
    // create (new) output file and tree\r
    if (!ofile || (osplit>0 && oevent%osplit==0)) {\r
      delete ofile;\r
      ++ofilenumber;\r
      sprintf(oURI,"%s/merged_%d.root",outputDirectoryURI,ofilenumber);\r
      printf("< creating output file \"%s\"\n",oURI);\r
      ofile=TFile::Open(oURI,"RECREATE");\r
      if (!ofile) {\r
        fprintf(stderr,"error: could not create output file: \"%s\" Exiting.\n",oURI);\r
        break;\r
      }\r
      otree=itree->CloneTree(0);\r
    }\r
    // copy event and write to file\r
    otree->CopyAddresses(itree);\r
    if (ievent==eventold) continue;\r
    itree->GetEntry(ievent);\r
    eventold=ievent;    \r
    otree->Fill();\r
//    otree->CopyEntries(itree,Form("Entry$==%d",ievent),1);\r
    ofile->Write();\r
    ++oevent;\r
\r
    // reset input\r
    itree->ResetBranchAddresses();\r
  }\r
\r
  printf("Merged %d events.\n",oevent);\r
  delete ifile;\r
  delete ofile;\r
}\r
\r