#include "TFile.h"
#include "TTree.h"
#include "TBranch.h"
#include "TAuthenticate.h"
#include "stdio.h"
#include <vector>
#include "Stntuple/obj/TStnHeaderBlock.hh"
#include "Stntuple/mod/addStntupleDataBlocks.hh"

class RsInfo {
 public:
  RsInfo():run(-1),rsmin(65535),rsmax(0) {}
  ~RsInfo(){};
  int run;
  int rsmin;
  int rsmax;
};

main(int argc, char* argv[]) {

  if(argc != 2) {
    printf("USAGE: getStnFileInfo full_file_name\n");
    return 1;
  }
  

  addStntupleDataBlocks();

  TAuthenticate::SetGlobalUser("anonymous");
  TFile* file = TFile::Open(argv[1]);
  if( !file) {
    printf("Error opening file %s\n ");
    return 1;
  }

  TTree* tree =(TTree*) file->Get("STNTUPLE");

  if (!tree) { 
    printf("Could not find STNTUPLE tree\n ");
    return 1;
  }
  TBranch* b = tree->GetBranch("HeaderBlock");
  TStnHeaderBlock* h = new TStnHeaderBlock();
  b->SetAddress(&h);

  int nevt = int(b->GetEntries());
  
  int runLow = 1000000000;
  int evtLow = 1000000000;
  int runHgh = 0;
  int evtHgh = 0;

  std::vector<RsInfo> rsinfo;
  for (int i=0; i<nevt; i++) {
    b->GetEntry(i);
    int run = h->RunNumber();
    int evt = h->EventNumber();
    int sec = h->SectionNumber();
    
    std::vector<RsInfo>::iterator runIt = rsinfo.end();
    std::vector<RsInfo>::iterator it;

    if(run<runLow) {
      runLow = run;
      evtLow = evt;
    } else if (run==runLow) {
      if(evt<evtLow) evtLow = evt;
    }
    if(run>runHgh) {
      runHgh = run;
      evtHgh = evt;
    } else if (run==runHgh) {
      if(evt>evtHgh) evtHgh = evt;
    }

    bool found = false;
    for(it=rsinfo.begin(); it!=rsinfo.end(); it++) {
      if(it->run == run) {
	found = true;
	runIt = it;
	if(sec<it->rsmin) it->rsmin = sec;
	if(sec>it->rsmax) it->rsmax = sec;
      }
    }
    if(!found) {
      RsInfo rs;
      rs.run = run;
      rs.rsmin = sec;
      rs.rsmax = sec;
      rsinfo.push_back(rs);
   }
  }

  for(int i=0; i<int(rsinfo.size()-1); i++) {
    for(int j=i+1; j<int(rsinfo.size()); j++) {
      if(rsinfo[j].run<rsinfo[i].run) {
	RsInfo k = rsinfo[i];
	rsinfo[i] = rsinfo[j];
	rsinfo[j] = k;
      }
    }
  }

  printf("File size (bytes): %d \n",file->GetSize());
  printf("First run/event: %d/%d\n",runLow,evtLow);
  printf("Last run/event: %d/%d\n",runHgh,evtHgh);
  printf("Physics events: %d\n",nevt);
  printf("Contributing runsections: ");

  std::vector<RsInfo>::iterator it,jt;
  for(it=rsinfo.begin(); it!=rsinfo.end(); it++) {
    printf("%d/%d:%d ",it->run,it->rsmin,it->rsmax);
  }
  printf("\n");

  if(int(rsinfo.size())>0) {
    printf("First run and runsection in hex: %06x %04x\n",
	   rsinfo[0].run,rsinfo[0].rsmin);
  } else {
    printf("First run and runsection in hex: %06x %04x\n",
	   0,0);
  }

}