mirror of
https://github.com/sesps/SPS_SABRE_EventBuilder.git
synced 2025-03-14 05:18:51 -04:00
105 lines
2.2 KiB
C++
105 lines
2.2 KiB
C++
#include "CutHandler.h"
|
|
|
|
namespace EventBuilder {
|
|
|
|
CutHandler::CutHandler() :
|
|
validFlag(false)
|
|
{
|
|
InitVariableMap();
|
|
}
|
|
|
|
CutHandler::CutHandler(const std::string& filename) :
|
|
validFlag(false)
|
|
{
|
|
SetCuts(filename);
|
|
InitVariableMap();
|
|
}
|
|
|
|
CutHandler::~CutHandler()
|
|
{
|
|
for(unsigned int i=0; i<file_array.size(); i++)
|
|
if(file_array[i]->IsOpen())
|
|
file_array[i]->Close();
|
|
}
|
|
|
|
void CutHandler::SetCuts(const std::string& filename)
|
|
{
|
|
std::ifstream cutlist(filename);
|
|
|
|
if(!cutlist.is_open())
|
|
{
|
|
validFlag = false;
|
|
}
|
|
|
|
std::string junk, name, fname, varx, vary;
|
|
cutlist>>junk>>junk>>junk>>junk;
|
|
|
|
cut_array.clear();
|
|
file_array.clear();
|
|
|
|
while(cutlist>>name)
|
|
{
|
|
cutlist>>fname>>varx>>vary;
|
|
TFile* file = TFile::Open(fname.c_str(), "READ");
|
|
TCutG* cut = (TCutG*) file->Get("CUTG");
|
|
if(cut)
|
|
{
|
|
cut->SetVarX(varx.c_str());
|
|
cut->SetVarY(vary.c_str());
|
|
cut->SetName(name.c_str());
|
|
cut_array.push_back(cut);
|
|
file_array.push_back(file);
|
|
}
|
|
else
|
|
{
|
|
validFlag = false;
|
|
EVB_WARN("CutHandler::SetCuts has encountered a bad file ({0}). The file either does not exist or doesn't contain a TCutG CUTG. Cuts ignored.", fname);
|
|
return;
|
|
}
|
|
}
|
|
|
|
if(cut_array.size() > 0)
|
|
validFlag = true;
|
|
else
|
|
validFlag = false;
|
|
}
|
|
|
|
/*
|
|
ADD MORE VARIABLES HERE!
|
|
*/
|
|
void CutHandler::InitVariableMap()
|
|
{
|
|
varmap["x1"] = &m_event.x1;
|
|
varmap["x2"] = &m_event.x2;
|
|
varmap["xavg"] = &m_event.xavg;
|
|
varmap["scintLeft"] = &m_event.scintLeft;
|
|
varmap["anodeBack"] = &m_event.anodeBack;
|
|
varmap["cathode"] = &m_event.cathode;
|
|
}
|
|
|
|
bool CutHandler::IsInside(const ProcessedEvent* eaddress)
|
|
{
|
|
m_event = *eaddress;
|
|
std::string x, y;
|
|
for(unsigned int i=0; i<cut_array.size(); i++)
|
|
{
|
|
TCutG* cut = cut_array[i];
|
|
x = cut->GetVarX();
|
|
y = cut->GetVarY();
|
|
auto xentry = varmap.find(x);
|
|
auto yentry = varmap.find(y);
|
|
if(xentry == varmap.end() || yentry == varmap.end())
|
|
{
|
|
EVB_WARN("Unmapped variable names at CutHandler::IsInside() (x:{0}, y:{1})! Cut not applied.", xentry->first, yentry->first);
|
|
return false;
|
|
}
|
|
|
|
if(!cut->IsInside(*(xentry->second), *(yentry->second)))
|
|
return false;
|
|
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
} |