LMU/musrfit
LMU/musrfit
5
0
Fork 0
Code Issues 6 Pull Requests Actions Packages Projects Releases 3 Activity

add search facility for runs via musrfit_startup.xml template entries.

Browse Source
This commit is contained in:
suter_a 2022-05-28 14:05:55 +02:00
parent 953cb32ded
commit b72ec88783
2 changed files with 205 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

149
src/classes/PFindRun.cpp Normal file
View File

@ -0,0 +1,149 @@
/***************************************************************************
PFindRun.cpp
Author: Andreas Suter
e-mail: andreas.suter@psi.ch
***************************************************************************/
/***************************************************************************
* Copyright (C) 2007-2022 by Andreas Suter *
* andreas.suter@psi.ch *
* *
* This program is free software; you can redistribute it and/or modify *
* it under the terms of the GNU General Public License as published by *
* the Free Software Foundation; either version 2 of the License, or *
* (at your option) any later version. *
* *
* This program is distributed in the hope that it will be useful, *
* but WITHOUT ANY WARRANTY; without even the implied warranty of *
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the *
* GNU General Public License for more details. *
* *
* You should have received a copy of the GNU General Public License *
* along with this program; if not, write to the *
* Free Software Foundation, Inc., *
* 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. *
***************************************************************************/
#include <iostream>
#include <boost/filesystem.hpp>
#include "PFindRun.h"
//----------------------------------------------------------------------------
/**
* @brief PFindRun::PFindRun
* @param runNameTemplateList
*/
PFindRun::PFindRun(const PStringVector path, const PRunNameTemplateList runNameTemplateList) :
fPath(path), fRunNameTemplateList(runNameTemplateList)
{
// nothing to be done here
}
//----------------------------------------------------------------------------
/**
* @brief PFindRun::PFindRun
*
* @param path
* @param runNameTemplateList
* @param instrument
* @param year
* @param run
*/
PFindRun::PFindRun(const PStringVector path, const PRunNameTemplateList runNameTemplateList,
const TString &instrument, const UInt_t year, const UInt_t run) :
fPath(path), fRunNameTemplateList(runNameTemplateList), fInstrument(instrument), fYear(year), fRun(run)
{
// nothing to be done here
}
//----------------------------------------------------------------------------
/**
* @brief PFindRun::DumpTemplateList
*/
void PFindRun::DumpTemplateList()
{
std::cout << "debug> instrument: " << fInstrument << std::endl;
std::cout << "debug> year: " << fYear << std::endl;
std::cout << "debug> run: " << fRun << std::endl;
std::cout << "debug> ++++" << std::endl;
std::cout << "debug> run name template list:" << std::endl;
for (UInt_t i=0; i<fRunNameTemplateList.size(); i++) {
std::cout << i << ": instrument: " << fRunNameTemplateList[i].instrument << ", template: " << fRunNameTemplateList[i].runNameTemplate << std::endl;
}
std::cout << "debug> ++++" << std::endl;
}
//----------------------------------------------------------------------------
/**
* @brief PFindRun::CreatePathName
* @param path
* @param runNameTemplate
* @return
*/
TString PFindRun::CreatePathName(const TString path, const TString runNameTemplate)
{
TString pathName{""};
TString runName = runNameTemplate;
TString yearStr = TString::Format("%d", fYear);
TString yyStr;
if (fYear < 2000)
yyStr = TString::Format("%02d", fYear-1900);
else
yyStr = TString::Format("%02d", fYear-2000);
runName.ReplaceAll("%yyyy%", yearStr);
runName.ReplaceAll("%yy%", yyStr);
// run handling slightly more complicated, since various number of digits possible
Int_t idx=-1;
TString rr{""};
for (Int_t i=2; i<10; i++) {
rr ="%";
for (Int_t j=0; j<i; j++)
rr += "r";
rr += "%";
if (runName.Index(rr) != -1) {
idx = i;
break;
}
}
TString format = TString::Format("%%0%dd", idx);
TString runStr = TString::Format(format, fRun);
if (idx != -1)
runName.ReplaceAll(rr, runStr);
pathName = path;
pathName += "/";
pathName += runName;
return pathName;
}
//----------------------------------------------------------------------------
/**
* @brief PFindRun::FoundPathName
* @return
*/
Bool_t PFindRun::FoundPathName()
{
// find instrument name in path list
TString pathName{""};
for (Int_t i=0; i<fPath.size(); i++) {
if (fPath[i].Index(fInstrument) != -1) {
for (Int_t j=0; j<fRunNameTemplateList.size(); j++) {
if (fRunNameTemplateList[j].instrument == fInstrument) {
pathName = CreatePathName(fPath[i], fRunNameTemplateList[j].runNameTemplate);
if (boost::filesystem::exists(pathName.Data())) {
fPathName = pathName;
return true;
}
}
}
}
}
return false;
}

56
src/include/PFindRun.h Normal file
View File

@ -0,0 +1,56 @@
/***************************************************************************
PFindRun.h
Author: Andreas Suter
e-mail: andreas.suter@psi.ch
***************************************************************************/
/***************************************************************************
* Copyright (C) 2007-2022 by Andreas Suter *
* andreas.suter@psi.ch *
* *
* This program is free software; you can redistribute it and/or modify *
* it under the terms of the GNU General Public License as published by *
* the Free Software Foundation; either version 2 of the License, or *
* (at your option) any later version. *
* *
* This program is distributed in the hope that it will be useful, *
* but WITHOUT ANY WARRANTY; without even the implied warranty of *
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the *
* GNU General Public License for more details. *
* *
* You should have received a copy of the GNU General Public License *
* along with this program; if not, write to the *
* Free Software Foundation, Inc., *
* 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. *
***************************************************************************/
#ifndef _PFINDRUN_H_
#define _PFINDRUN_H_
#include "PMusr.h"
class PFindRun {
public:
PFindRun(const PStringVector path, const PRunNameTemplateList runNameTemplateList);
PFindRun(const PStringVector path, const PRunNameTemplateList runNameTemplateList,
const TString &instrument, const UInt_t year, const UInt_t run);
Bool_t FoundPathName();
TString GetPathName() { return fPathName; }
void DumpTemplateList();
private:
const PStringVector fPath;
const PRunNameTemplateList fRunNameTemplateList;
TString fInstrument{""};
Int_t fYear{-1};
Int_t fRun{-1};
TString fPathName{""};
TString CreatePathName(const TString path, const TString runNameTemplate);
};
#endif // _PFINDRUN_H_