end-of-line normalization

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

//
// Macro to create the "raw" data file with selected events
// Paramaters:
//  eventListFileName - input ascii file with list of chunks and event numbers - two collumns
//  osplit            - split the file after given threhshold value
//
//  TGrid * alien = TGrid::Connect("alien://",0,0,"t");

void rawmerge(const char *eventListFileName,
              const char *outputDirectoryURI,
              Long64_t osplit=-1)
{
  TGrid * alien = TGrid::Connect("alien://",0,0,"t");
  
  Int_t eventNumber;
  FILE *files=fopen(eventListFileName,"r");
  if (!files) {
    fprintf(stderr,"error: could not read event list file \"%s\". Exiting.\n",eventListFileName);
    return;
  }
  char iURI[1000];
  TString  iURIold;
  char oURI[1000];
  TFile *ifile=0;
  TFile *ofile=0;
  TTree *itree=0;
  TTree *otree=0;
  Long64_t ievent;
  Long64_t oevent;
  Int_t ofilenumber=0;
  Int_t line=0;
  Int_t eventold=0;
  while (!feof(files)) {
    ++line;
    if (fscanf(files,"%s %d\n",iURI,&ievent)!=2) {
      fprintf(stderr,"warning: corrupted event line (%d) in input file, skipping it...\n",line);
      continue;
    } 
    printf("> processing \"%s\" event %d...\n",iURI,ievent);
    if (ievent==eventold) { printf("duplicated continue\n"),continue;}
    //
    if (iURIold.Contains(iURI)==0){ 
      printf("NF: %s\n",iURI);
      delete ifile;ifile=0;
      ifile=TFile::Open(iURI);
      if (!ifile) {
	fprintf(stderr,"warning: could not open file for event \"%s\", skipping it...\n",iURI);
	continue;
      }
      iURIold=iURI;
    }else{
      printf("OF: %s\n",iURI);
      iURIold=iURI; 
    }
    //
    TTree *itree=dynamic_cast<TTree*>(ifile->Get("RAW"));
    if (!itree) {
      fprintf(stderr,"warning: could not find RAW tree for event \"%s\", skipping it...\n",iURI);
      continue;
    }

    // create (new) output file and tree
    if (!ofile || (osplit>0 && oevent%osplit==0)) {
      delete ofile;
      ++ofilenumber;
      sprintf(oURI,"%s/merged_%d.root",outputDirectoryURI,ofilenumber);
      printf("< creating output file \"%s\"\n",oURI);
      ofile=TFile::Open(oURI,"RECREATE");
      if (!ofile) {
        fprintf(stderr,"error: could not create output file: \"%s\" Exiting.\n",oURI);
        break;
      }
      otree=itree->CloneTree(0);
    }
    // copy event and write to file
    otree->CopyAddresses(itree);
    if (ievent==eventold) continue;
    itree->GetEntry(ievent);
    eventold=ievent;    
    otree->Fill();
//    otree->CopyEntries(itree,Form("Entry$==%d",ievent),1);
    ofile->Write();
    ++oevent;

    // reset input
    itree->ResetBranchAddresses();
  }

  printf("Merged %d events.\n",oevent);
  delete ifile;
  delete ofile;
}