Fill_MBSpill.cxx

Go to the documentation of this file.

#include "Riostream.h"
#include "TFile.h"
#include "TTree.h"
#include "TSystem.h"
#include "TMath.h"
#include <map>
#include <algorithm>
#include <vector>
#include <iostream>
#include "Validity/VldTimeStamp.h"
#include "DatabaseInterface/DbiWriter.h"
#include "MiniBooNEAna/MBSpill.h"
#include "Conventions/Munits.h"


using namespace std;

void Fill_MBSpill(){
  //Dbi::SetLogLevel(2);
  //Char_t dirname[]="/afs/fnal.gov/files/data/minos/d77/shanahan/SpillTimes/";
  Char_t dirname[]="/afs/fnal.gov/files/data/minos/d69/tjyang/test/timing/new/";
  void *dirhandle = gSystem->OpenDirectory(gSystem->ExpandPathName(dirname));
  
  Int_t nfiles = 0;
  Char_t *file;
  Char_t path_file[100];


  map<VldTimeStamp, MBSpill> spills;
  map<VldTimeStamp, MBSpill>::iterator p;

  Double_t fTableDuration = 8.0*Munits::hour;


 //Loop over the filenames inside the directory
  while ( ( file = const_cast<char*>(gSystem->GetDirEntry(dirhandle)) ) ) {
    ifstream in;
    if (strstr(file,"spill_info_") != 0) {
      strcpy(path_file,dirname);
      strcat(path_file,file);
      in.open(path_file);
      cout<<"open file:"<<path_file<<endl;
      nfiles++;
      int sec,nsec;
      float pot,hI;
      while (1) {
        in >> sec >> nsec >> pot >> hI;
        if (!in.good()) break;
        MBSpill mbspill(sec,nsec,pot,hI);
        VldTimeStamp ts((time_t)sec,nsec);
        p=spills.find(ts);
        if (p != spills.end()){
          if (pot>p->second.GetPOT()){//replace the existing one
            spills.erase(p);
            spills.insert(pair<VldTimeStamp,MBSpill>(ts,mbspill));
          }
        }
        else if (TMath::Abs(hI)>10) {//remove bad spills
          spills.insert(pair<VldTimeStamp,MBSpill>(ts,mbspill));
        }
      }
    }
    in.close();
  }
  
  
  if (spills.size()==0) {
    cout<<"no data found, check directory"<<endl;
    exit(0);
  }
  
  cout<<nfiles<<" files found."<<endl;
  
  p = spills.begin();
  VldTimeStamp fStart = p->first;
  VldTimeStamp fEnd = fStart;
  cout<<"Start "<<" "<<fStart<<" End "<<fEnd<<endl;
  vector<MBSpill> spills_chunk;
  while (p!=spills.end()){
    spills_chunk.push_back(p->second);
    fEnd = p->first;
    if((double)(fEnd-fStart) > fTableDuration && spills_chunk.size()){
      VldTimeStamp vldCreate = fStart;
      VldRange range( (int)Detector::kNear,
                      (int)SimFlag::kData,
                      fStart,
                      fEnd,
                      "MiniBooNE spill information"
                      );
      cout<<"Starting write: " << range.AsString() << endl;
      DbiWriter<MBSpill> writer (range,
                                 -1, //Aggregate
                                 0,  // task
                                 vldCreate, // Creation date
                                 0,   // db no
                                 "MiniBooNE spill information"
                                 );
      //writer.SetRequireGlobalSeqno(1);
      for (UInt_t i = 0; i<spills_chunk.size(); i++){
        writer << spills_chunk[i];
      }
      writer.Close();
      spills_chunk.clear();
      fStart = fEnd;
      fEnd = fStart;      
    }
    p++;
  }

  if(spills_chunk.size()){
    VldTimeStamp vldCreate = fStart;
    VldRange range( (int)Detector::kNear,
                    (int)SimFlag::kData,
                    fStart,
                    fEnd,
                    "MiniBooNE spill information"
                    );
    cout<<"Starting write: " << range.AsString() << endl;
    DbiWriter<MBSpill> writer (range,
                               -1, //Aggregate
                               0,  // task
                               vldCreate, // Creation date
                               0,   // db no
                               "MiniBooNE spill information"
                               );
    for (UInt_t i = 0; i<spills_chunk.size(); i++){
      writer << spills_chunk[i];
    }
    writer.Close();
    spills_chunk.clear();
    
  }

}

---