NCPOTCounter Class Reference

Calculates POT contained in a set of uDSTs. More...

#include <NCPOTCounter.h>

List of all members.

Public Member Functions
	NCPOTCounter ()
void	Config (const Registry &r, std::vector< BeamType::BeamType_t > beamIndex, std::vector< NC::RunUtil::ERunType > runsToUse)
void	SetPOTValues (std::map< NCBeam::Info, double > &pot, Detector::Detector_t det, NCType::EFileType fileType, NCType::EDataMC dataMC)
double	GetPOTValue (Detector::Detector_t det, NCType::EFileType fileType, NCType::EDataMC dataMC, BeamType::BeamType_t beamType, NC::RunUtil::ERunType run)
std::vector< TString >	GetListOfFiles (NCBeam::Info info, NCType::EFileType fileType, Detector::Detector_t det, NCType::EDataMC dataMC) const
	List of stripped uDSTs under fDataMCPath for a particular run period.
void	SetMCExposureForBeam (NCBeam::Info beam, double pot)
Private Types
typedef std::map< NCBeam::Info, double >	POTMap
typedef std::map< NCBeam::Info, double >	ScaleMap
typedef std::vector< int >	POTKey
typedef std::map< POTKey, POTMap >	Cache
Private Member Functions
int	GetRunsInFile (TString fileName)
	Get the number of runs contained in fileName.
double	GetPOTForSingleFile (TString fileName, NCType::EFileType fileType, Detector::Detector_t det, NCType::EDataMC dataMC, int bt, int nRunsOffset) const
	Get the POT for a file fileName and additional details.
TString	FileDesignator (NCType::EFileType fileType, Detector::Detector_t det, NCType::EDataMC dataMC) const
	Can be "data", "blind", "tau", "electron", "mc".
TString	HistName (NCType::EFileType fileType, NCType::EDataMC dataMC) const
	Can be "dataPOT" or "mcPOT".
double	POTPerRun (int bt, Detector::Detector_t det, NCType::EFileType fileType) const
	POT per subrun for near detector, POT per run for far.
int	MockFileSubRun (TString fileName) const
Private Attributes
TString	fDataMCPath
TString	fMockDataSet
bool	fMockData
bool	fMCAsData
int	fRunLimits [3][2]
	[detector][dataMC]
std::vector< BeamType::BeamType_t >	fBeamIndex
std::vector< NC::RunUtil::ERunType >	fRunsToUse
std::map< NCBeam::Info, double >	fFarMCExposures
	Allows different MC configurations to be scaled to different POT's.
int	fMockDataSubRun
Static Private Attributes
Cache	fgCache

---

Detailed Description

Calculates POT contained in a set of uDSTs.

Definition at line 67 of file NCPOTCounter.h.

---

Member Typedef Documentation

typedef std::map<POTKey, POTMap> NCPOTCounter::Cache [private]

Definition at line 141 of file NCPOTCounter.h.

typedef std::vector<int> NCPOTCounter::POTKey [private]

Definition at line 139 of file NCPOTCounter.h. Referenced by SetPOTValues().

typedef std::map<NCBeam::Info, double> NCPOTCounter::POTMap [private]

Definition at line 101 of file NCPOTCounter.h.

typedef std::map<NCBeam::Info, double> NCPOTCounter::ScaleMap [private]

Definition at line 102 of file NCPOTCounter.h.

---

Constructor & Destructor Documentation

NCPOTCounter::NCPOTCounter (   )

Definition at line 26 of file NCPOTCounter.cxx. { }

---

Member Function Documentation

void NCPOTCounter::Config (  const Registry &  r, std::vector< BeamType::BeamType_t >  beamIndex, std::vector< NC::RunUtil::ERunType >  runsToUse )

Definition at line 31 of file NCPOTCounter.cxx. References fBeamIndex, fDataMCPath, fMCAsData, fMockData, fMockDataSet, fMockDataSubRun, fRunLimits, fRunsToUse, Registry::Get(), and gSystem(). Referenced by NCExtrapolationModule::Config(), and NCDataQualityModule::SetPOTValues(). { const char* tmps; int tmpb, tmpi; // TODO - this code is copy-pasted from NCExtrapolationModule if(r.Get("DataMCPath", tmps)){ //for some reason the TSystemDirectory constructor can't resolve //environmental variables, so just take care of it here TString temp = tmps; // Assume the form $ENVVAR/path if(temp.Contains("$")){ int pos = temp.Index("/"); // find the first / temp.Resize(pos); // and truncate at it temp.Remove(TString::kLeading, '$'); // drop the $ off the front fDataMCPath = tmps; // Copy the original string // and then replace the part that we think is the environment variable fDataMCPath.Replace(0, pos, gSystem->Getenv(temp)); } else fDataMCPath = temp; // Add a trailing slash so that concatenating the path later will work if(!fDataMCPath.EndsWith("/")) fDataMCPath += "/"; } if(r.Get("MockDataSet", tmps)) fMockDataSet = tmps; if(r.Get("UseMCAsData", tmpb)) fMCAsData = tmpb; if(r.Get("UseMockData", tmpb)) fMockData = tmpb; if(r.Get("RunLimitNearData", tmpi)) fRunLimits[Detector::kNear][NCType::kData] = tmpi; if(r.Get("RunLimitFarData", tmpi)) fRunLimits[Detector::kFar][NCType::kData] = tmpi; if(r.Get("RunLimitNearMC", tmpi)) fRunLimits[Detector::kNear][NCType::kMC] = tmpi; if(r.Get("RunLimitFarMC", tmpi)) fRunLimits[Detector::kFar][NCType::kMC] = tmpi; if(r.Get("MockDataSubRun", tmpi)) fMockDataSubRun = tmpi; fBeamIndex = beamIndex; fRunsToUse = runsToUse; }

TString NCPOTCounter::FileDesignator (  NCType::EFileType  fileType, Detector::Detector_t  det, NCType::EDataMC  dataMC )  const [private]

Can be "data", "blind", "tau", "electron", "mc". Definition at line 351 of file NCPOTCounter.cxx. References det, and fMCAsData. Referenced by GetListOfFiles(), GetPOTForSingleFile(), and SetPOTValues(). { if(fileType == NCType::kMockFile) return fMockDataSet; if(dataMC == NCType::kData && !(fMCAsData || fMockData)){ if(det == Detector::kNear) return "data"; else if(det == Detector::kFar) return "blind"; } if(fileType == NCType::kTauFile) return "tau"; if(fileType == NCType::kElectronFile) return "electron"; return "mc"; }

vector< TString > NCPOTCounter::GetListOfFiles (  NCBeam::Info  info, NCType::EFileType  fileType, Detector::Detector_t  det, NCType::EDataMC  dataMC )  const

List of stripped uDSTs under fDataMCPath for a particular run period. with correct dataMC, beaminfo info and detector det, Definition at line 221 of file NCPOTCounter.cxx. References BeamType::AsString(), det, fDataMCPath, FileDesignator(), files, NC::RunUtil::FindRunType(), fMockDataSubRun, Form(), NCBeam::Info::GetBeamType(), NCBeam::Info::GetRunType(), MockFileSubRun(), and runPeriod. Referenced by NCExtrapolationModule::AddFilesToChain(), and SetPOTValues(). { // MSG("NCPOTCounter", Msg::kInfo) << "GetListOfFiles(" << bt << ", " << fileType // << ", " << det << ", " << dataMC << endl; TDirectory* saveDir = gDirectory; vector<TString> ret; const TString fileDesignator = FileDesignator(fileType, det, dataMC); const TString beamType = BeamType::AsString(info.GetBeamType()); const TString runPeriod= Form("run%d", (int)info.GetRunType()); const TString detName = (det == Detector::kFar) ? "far" : "near"; //have a path to the relevant uDST's. each uDST has the //BeamType::AsString() for the beam in its name, also has near //or far in the name. //make a TSystemDirectory object to get a list of files in the //path for the data and mc //the files for different beam configurations are in uDST/LXXXzYYYi //directories. that is directories named for the beam type TSystemDirectory dataDir(beamType, fDataMCPath+beamType+"/"); TList* files = dataDir.GetListOfFiles(); assert(files); for(int de = 0; de < files->GetEntries(); ++de){ TString fileName = files->At(de)->GetName(); //Do nothing for "." and ".." entries if(fileName == "." || fileName == "..") continue; //check that this file is the right kind if(fileName.Contains(beamType) && fileName.Contains("strip") && fileName.Contains(detName) && fileName.Contains(fileDesignator)){ // Check it's the right run type - this should work for both data and mc const NC::RunUtil::ERunType fileRunPeriod = NC::RunUtil::FindRunType(fileName); if(fileRunPeriod != info.GetRunType()) continue; if(fileType == NCType::kMockFile && fMockDataSubRun != -1){ // For mock data, only add files with subrun requested int subRun = MockFileSubRun(fileName); // If it's not the one we want, don't add it to the chain. // // subRun==-1 if the file is the one-uDST-per-run type. In // that case, the fileDesignator check above is enough to tell // us that we want to use this file, so don't skip it if(subRun != -1 && subRun != fMockDataSubRun) continue; } ret.push_back(fileName); }//end if the right detector and beam and designator }//end loop over data files delete files; saveDir->cd(); return ret; }

double NCPOTCounter::GetPOTForSingleFile (  TString  fileName, NCType::EFileType  fileType, Detector::Detector_t  det, NCType::EDataMC  dataMC, int  bt, int  nRunsOffset )  const [private]

Get the POT for a file fileName and additional details. nRunsOffset is the number of runs that we've previously read from other files, which should be subtracted from the file limit to get the number of runs we should read from *this* file. This is needed to deal with the fact that we have multiple MC files in the near detector Definition at line 303 of file NCPOTCounter.cxx. References det, FileDesignator(), fRunLimits, HistName(), MockFileSubRun(), MSG, and POTPerRun(). Referenced by SetPOTValues(). { // TDirectory* saveDir = gDirectory; TFile tf(fileName); assert(!tf.IsZombie()); const TString histName = HistName(fileType, dataMC); const TString fileDesignator = FileDesignator(fileType, det, dataMC); TH1F* potHist = (TH1F*)(tf.Get(histName)); if(fileType==NCType::kMockFile) { // The one-subrun-per-uDST files have the correct POT stored in // the dataPOT histogram. The one-run-per-uDST files have to use // the hardcoded value from POTPerRun if (MockFileSubRun(fileName) != -1) return potHist->Integral(0, potHist->GetNbinsX()+1)*1e12; else return POTPerRun(bt, det, fileType); } MSG("NCPOTCounter", Msg::kInfo) << " Running on " << fileName << " with " << potHist->GetEntries() << " runs" << endl; //use all pot's for data if(fileDesignator == "data" || fileDesignator == "blind"){ return potHist->Integral(0, potHist->GetNbinsX()+1)*1e12; } const double potPerRun = POTPerRun(bt, det, fileType); const int runLimit = fRunLimits[det][dataMC]-nRunsOffset; if(runLimit <= potHist->GetEntries()){ return potPerRun*runLimit; } else{ return potHist->Integral(0, potHist->GetNbinsX()+1)*1e12; } }

double NCPOTCounter::GetPOTValue (  Detector::Detector_t  det, NCType::EFileType  fileType, NCType::EDataMC  dataMC, BeamType::BeamType_t  beamType, NC::RunUtil::ERunType  run )

Definition at line 205 of file NCPOTCounter.cxx. References det, run(), and SetPOTValues(). { map<NCBeam::Info, double> pot; SetPOTValues(pot, det, fileType, dataMC); const NCBeam::Info beamInfo(beamType, run); return pot[beamInfo]; }

int NCPOTCounter::GetRunsInFile (  TString  fileName  )  [private]

Get the number of runs contained in fileName. as given by the pot histogram Definition at line 285 of file NCPOTCounter.cxx. Referenced by SetPOTValues(). { TFile tf(fileName); assert(!tf.IsZombie()); TH1F* potHist = (TH1F*)(tf.Get("mcPOT")); if(!potHist) potHist=(TH1F*)(tf.Get("dataPOT")); assert(potHist); // Hopefully this cast does what I want. Why does GetEntries() not return an int? Gah return int(potHist->GetEntries()); }

TString NCPOTCounter::HistName (  NCType::EFileType  fileType, NCType::EDataMC  dataMC )  const [private]

Can be "dataPOT" or "mcPOT". Definition at line 370 of file NCPOTCounter.cxx. References fMCAsData. Referenced by GetPOTForSingleFile(). { if(fileType == NCType::kMockFile || (dataMC == NCType::kData && !(fMCAsData || fMockData))) return "dataPOT"; return "mcPOT"; }

int NCPOTCounter::MockFileSubRun (  TString  fileName  )  const [private]

The subrun number of a mock data file if it is of the one-subrun-per-uDST type, or -1 otherwise Definition at line 402 of file NCPOTCounter.cxx. Referenced by GetListOfFiles(), and GetPOTForSingleFile(). { // Filenames look like: // "far_mock_L010z185i_F21910001_0000.uDST_strip.root" // // ROOT is useless and doesn't support character classes, or // specifying a number of times to match, so we get this RE // to find the subrun from the filename. TRegexp re("_[0-9][0-9][0-9][0-9]\\.uDST"); int stupidROOT; int index = re.Index(fileName, &stupidROOT); if(index != kNPOS) { // We found a subrun string in the filename, so use that. TString subRunStr(fileName(index+1, 4)); return subRunStr.Atoi(); } else { // This will happen for one-uDST-per-run files return -1; } }

double NCPOTCounter::POTPerRun (  int  bt, Detector::Detector_t  det, NCType::EFileType  fileType )  const [private]

POT per subrun for near detector, POT per run for far. Definition at line 380 of file NCPOTCounter.cxx. References det, fBeamIndex, and kPOTNear. Referenced by GetPOTForSingleFile(). { //really need to do something better than the next few lines //to account for changes in MC version if(det == Detector::kFar){ if(fileType == NCType::kMockFile) { const double kPotFarMock = 2.5e20; return kPotFarMock; } // TODO: Is this a good name for it? If so should go in NCType const double kPotFarMC = 6.5e20; return kPotFarMC; } const double ret = kPOTNear[fBeamIndex[bt]]; assert(ret >= 0); return ret; }

void NCPOTCounter::SetMCExposureForBeam (  NCBeam::Info  beam, double  pot )

Definition at line 297 of file NCPOTCounter.cxx. References fFarMCExposures. Referenced by NCExtrapolationModule::SetMCExposureForBeam(). { fFarMCExposures[beam] = pot; }

void NCPOTCounter::SetPOTValues (  std::map< NCBeam::Info, double > &  pot, Detector::Detector_t  det, NCType::EFileType  fileType, NCType::EDataMC  dataMC )

pot returns a map of pot's as would otherwise be returned by GetPOTValue Definition at line 77 of file NCPOTCounter.cxx. References det, fBeamIndex, fDataMCPath, fFarMCExposures, fgCache, FileDesignator(), fRunLimits, fRunsToUse, GetListOfFiles(), GetPOTForSingleFile(), GetRunsInFile(), MSG, and POTKey. Referenced by GetPOTValue(), NCExtrapolationModule::Run(), and NCDataQualityModule::SetPOTValues(). { POTKey key(3); key[0] = det; key[1] = fileType; key[2] = dataMC; Cache::iterator it = fgCache.find(key); if(it != fgCache.end()){ pot = it->second; return; } pot.clear(); MSG("NCPOTCounter", Msg::kInfo) << "NCPOTCounter::SetPOTValues:" << " fDataMCPath = " << fDataMCPath << endl; // Make POT map with an entry for each run/beam type // combination. If we actually have nonzero POTs for a particular // combination, it'll get added to the 'pot' map at the end. map<NCBeam::Info, double> POTtmp; for(int r = NC::RunUtil::kRunI; r <= NC::RunUtil::kMaxRun; ++r){ for(unsigned int j = 0; j < fBeamIndex.size(); ++j){ NCBeam::Info info(fBeamIndex[j], NC::RunUtil::ERunType(r)); POTtmp[info] = 0.; } } //loop over the beams/runperiods we're using and fill the pot values for(unsigned int bt = 0; bt < fBeamIndex.size(); ++bt){ for(unsigned int runIdx = 0; runIdx<fRunsToUse.size(); ++runIdx){ NCBeam::Info info(fBeamIndex[bt], fRunsToUse[runIdx]); // The total number of runs we've read from files, // so we can apply the file limits int nRunsRead = 0; const vector<TString> fileList = GetListOfFiles(info, fileType, det, dataMC); for(unsigned int nf = 0; nf < fileList.size(); ++nf){ const TString fileName = fileList[nf]; const double potToAdd = GetPOTForSingleFile(fileName, fileType, det, dataMC, bt, nRunsRead); assert(potToAdd >= 0); //const int runType = NC::RunUtil::FindRunType(fileName); assert(POTtmp.find(info)!=POTtmp.end()); POTtmp[info] += potToAdd; MSG("NCPOTCounter", Msg::kInfo) << " Adding " << potToAdd << " POT to " << FileDesignator(fileType, det, dataMC) << endl; nRunsRead += GetRunsInFile(fileName); // If we've already hit the file limit, stop if(nRunsRead > fRunLimits[det][dataMC]) break; } // end for nf /* const TString beamType = BeamType::AsString(fBeamIndex[bt]); MSG("NCPOTCounter", Msg::kInfo) << "POT INFORMATION(" << (det == Detector::kFar ? "far" : "near") << "): " << endl << FileDesignator(fileType, det, dataMC) << " run I POT[" << beamType << "] = " << POTtmp[0].at(bt) << " run II POT[" << beamType << "] = " << POTtmp[1].at(bt) << endl; */ if(POTtmp[info] > 0){ if(dataMC == NCType::kData && fMCAsData){ assert(fFarMCExposures.find(info) != fFarMCExposures.end() && "Did you call set_mc_exposure_for_beam for this beam?"); pot[info] = fFarMCExposures[info]; } else{ pot[info] = POTtmp[info]; } }//end if pot > 0 for this combination of run type and beam }//end loop over run types // Now we've looped over all the run periods, add their total to // the RunAll NCBeam for this beamtype double totalPOT=0; for(unsigned int n = 0; n < fRunsToUse.size(); ++n){ NCBeam::Info info(fBeamIndex[bt], fRunsToUse[n]); totalPOT+=pot[info]; } const NCBeam::Info infoAll(fBeamIndex[bt], NC::RunUtil::kRunAll); pot[infoAll] = totalPOT; }//end loop over beam types for(map<NCBeam::Info, double>::const_iterator it = pot.begin(); it != pot.end(); ++it){ MSG("NCPOTCounter", Msg::kInfo) << "POT INFO: " << (det == Detector::kFar ? "Far" : "Near") << " " << (dataMC == NCType::kMC ? "MC" : "data") << " from \"" << FileDesignator(fileType, det, dataMC) << "\" files " << it->first << " has " << it->second << " POT" << endl; } fgCache[key] = pot; }

---

Member Data Documentation

std::vector<BeamType::BeamType_t> NCPOTCounter::fBeamIndex [private]

Definition at line 153 of file NCPOTCounter.h. Referenced by Config(), POTPerRun(), and SetPOTValues().

TString NCPOTCounter::fDataMCPath [private]

Definition at line 146 of file NCPOTCounter.h. Referenced by Config(), GetListOfFiles(), and SetPOTValues().

std::map<NCBeam::Info, double> NCPOTCounter::fFarMCExposures [private]

Allows different MC configurations to be scaled to different POT's. Definition at line 156 of file NCPOTCounter.h. Referenced by SetMCExposureForBeam(), and SetPOTValues().

NCPOTCounter::Cache NCPOTCounter::fgCache [static, private]

Definition at line 24 of file NCPOTCounter.cxx. Referenced by SetPOTValues().

bool NCPOTCounter::fMCAsData [private]

Definition at line 149 of file NCPOTCounter.h. Referenced by Config(), FileDesignator(), and HistName().

bool NCPOTCounter::fMockData [private]

Definition at line 148 of file NCPOTCounter.h. Referenced by Config().

TString NCPOTCounter::fMockDataSet [private]

Definition at line 147 of file NCPOTCounter.h. Referenced by Config().

int NCPOTCounter::fMockDataSubRun [private]

Definition at line 157 of file NCPOTCounter.h. Referenced by Config(), and GetListOfFiles().

int NCPOTCounter::fRunLimits[3][2] [private]

[detector][dataMC] Definition at line 152 of file NCPOTCounter.h. Referenced by Config(), GetPOTForSingleFile(), and SetPOTValues().

std::vector<NC::RunUtil::ERunType> NCPOTCounter::fRunsToUse [private]

Definition at line 154 of file NCPOTCounter.h. Referenced by Config(), and SetPOTValues().

---

The documentation for this class was generated from the following files:

• NCPOTCounter.h
• NCPOTCounter.cxx

---