BeamMonMap.cxx

Go to the documentation of this file.

//
// BeamMonMap
//
// accesses beam monitoring... a bit of a temporary solution (hopefully!)
//
// Created:  P. Vahle -- March, 2005
//
// $Author: gmieg $ 
//
// $Revision: 1.3 $
// 
// $Name:  $
//
// $Id: BeamMonMap.cxx,v 1.3 2009/02/28 21:46:13 gmieg Exp $
//

#include <dirent.h>
#include <vector>
#include <string>
#include <iostream>
#include <cmath>
#include <cstdlib>

#include "TFile.h"
#include "TTree.h"
#include "Mad/BeamMonMap.h"

BeamMonTV::BeamMonTV():
   bI(0.),
   hbw(0.),
   vbw(0.),
   hpos1(0.),
   vpos1(0.),
   hpos2(0.),
   vpos2(0.),
   htan(0.),
   vtan(0.),
   hornI(0.),
   nuTarZ(0.),
   timens(0.),
   closestspill(-99999.),
   time(0)
{}

BeamMonTV::~BeamMonTV()
{}

BeamMonTV BeamMonMap::FillBeamMonTV(struct bmon1 bmn)
{
   BeamMonTV b;
//   std::cout<<"Filling beammontv from type 1"<<std::endl;
   b.bI=bmn.beamIntensity;
   b.hbw=bmn.hBeamWidth;
   b.vbw=bmn.vBeamWidth;
   b.hpos1=bmn.hPosatTargetPM;
   b.vpos1=bmn.vPosatTargetPM;
   b.hpos2=bmn.hPosatTargetBPM;
   b.vpos2=bmn.vPosatTargetBPM;
   b.htan=bmn.tanHoriz;
   b.vtan=bmn.tanVert;
   b.hornI=bmn.hornPeakCurrent;
   b.nuTarZ=bmn.nuTarZ;
   b.timens=bmn.timeStampD;
   b.time=bmn.timeStampSec;
//   std::cout<<"done"<<std::endl;
   return b;
}
 

BeamMonTV BeamMonMap::FillBeamMonTV(struct bmon2 bmn)
{
   BeamMonTV b;
//   std::cout<<"Filling beammontv from type 2"<<std::endl;
   b.bI=bmn.beamIntensity;
   b.hbw=bmn.hBeamWidth;
   b.vbw=bmn.vBeamWidth;
   b.hpos1=bmn.hPosatTargetPM;
   b.vpos1=bmn.vPosatTargetPM;
   b.hpos2=bmn.hPosatTargetBPM;
   b.vpos2=bmn.vPosatTargetBPM;
   b.htan=bmn.tanHoriz;
   b.vtan=bmn.tanVert;
   b.hornI=bmn.hornPeakCurrent;
   b.nuTarZ=bmn.nuTarZ;
   b.timens=bmn.timeStampD;
   b.time=bmn.timeStampSec;
//   std::cout<<"done"<<std::endl;
   return b;
}

 
std::map<VldTimeStamp, BeamMonTV> BeamMonMap::MakeBeamMonMap(const char* path)
{

   std::map<VldTimeStamp, BeamMonTV> m;
   if(path==NULL){
      std::cout<<"Didn't specify a path for beam mon files, wont get em in your tree"<<std::endl;
      return m;
   }
   std::cout<<"opening dir "<<path<<std::endl;

   DIR *dfd;
   dirent *dp;
   if(!(dfd =  opendir(path))){
      std::cout<<" path "<<path<<" "<<dfd<<std::endl;
      return m;
   }

   std::vector<std::string> names;
   while((dp=readdir(dfd))!=NULL){
      if(strstr(dp->d_name,"beamsummary_")!=NULL){
         names.push_back((std::string)(dp->d_name));
      }
   }

   std::cout<<"Found "<<names.size()<<" names "<<std::endl;
   for(unsigned int i=0;i<names.size();i++){
      //figure out what version of bmon ntuple we've got
      std::string n=names[i].substr(names[i].find_last_of("_")+1,
                               names[i].find_last_of(".")-names[i].find_last_of("_")-1);
//      std::cout<<"n is "<<n<<std::endl;

      //open tree file
      std::string fname = path+names[i];
      TFile f(fname.c_str());
      TTree *b = (TTree *)(f.Get("btree"));
      if(atoi(n.c_str())<=7){
         struct bmon1 bs;
         b->SetBranchAddress("Beaminfo",&bs.beamIntensity);
//       std::cout<<"In first type"<<std::endl;
         //loop over tree
         for(int z=0;z<b->GetEntries();z++){
            b->GetEntry(z);
            BeamMonTV bmtv = FillBeamMonTV(bs);
            VldTimeStamp vts(bs.timeStampSec,0);
            m[vts]=bmtv;
         }
      }
      else{
//       std::cout<<"In second type"<<std::endl;
         struct bmon2 bs;
         b->SetBranchAddress("Beaminfo",&bs.beamIntensity);
         //loop over tree
         for(int z=0;z<b->GetEntries();z++){
            b->GetEntry(z);
            BeamMonTV bmtv = FillBeamMonTV(bs);
            VldTimeStamp vts(bs.timeStampSec,0);
            m[vts]=bmtv;
         }
      }
      f.Close();
   }
   return m;
}

BeamMonTV BeamMonMap::FindClosestSpill(const std::map<VldTimeStamp, BeamMonTV>& m, VldTimeStamp vts)
{
   if(m.size()==0){
      BeamMonTV b;
      return b;
   }

   std::map<VldTimeStamp, BeamMonTV>::const_iterator lb = m.lower_bound(vts);
   if(lb==m.begin()){ return lb->second; }
   std::map<VldTimeStamp, BeamMonTV>::const_iterator prev=lb;
   prev--;
   if(lb==m.end()){ return prev->second; }

   double early = fabs(vts-prev->first);
   double late = fabs(lb->first-vts);

//   std::cout<<"early "<<early<<" late "<<late<<std::endl;

   if(early<late){
     BeamMonTV bmtv = prev->second;
     bmtv.closestspill=early;
     return bmtv;
   }
   BeamMonTV bmtv2 = lb->second;
   bmtv2.closestspill=late;
   return bmtv2;
}

bool BeamMonMap::IsGoodSpill(BeamMonTV b)
{

   if(b.bI>0.1&&
      b.hbw<2&&b.vbw<2&&
      b.hpos2<-1.5&&b.hpos2>-2.0&&
      b.vpos2>.2&&b.hpos2<1.8&&b.closestspill<2){
      return true;
   }
   return false;
}

---