233 lines
7.3 KiB
C++
233 lines
7.3 KiB
C++
/***************************************************************************
|
|
|
|
PFunctionHandler.cpp
|
|
|
|
Author: Andreas Suter
|
|
e-mail: andreas.suter@psi.ch
|
|
|
|
***************************************************************************/
|
|
|
|
/***************************************************************************
|
|
* Copyright (C) 2007-2016 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 <string>
|
|
#include <cassert>
|
|
|
|
#include "PFunctionHandler.h"
|
|
|
|
//-------------------------------------------------------------
|
|
// Constructor
|
|
//-------------------------------------------------------------
|
|
/**
|
|
* <p>Constructor
|
|
*
|
|
* \param lines msr-file FUNCTIONS block in clear text.
|
|
*/
|
|
PFunctionHandler::PFunctionHandler(PMsrLines lines) : fLines(lines)
|
|
{
|
|
fValid = true;
|
|
}
|
|
|
|
//-------------------------------------------------------------
|
|
// Destructor
|
|
//-------------------------------------------------------------
|
|
/**
|
|
* <p>Destructor
|
|
*/
|
|
PFunctionHandler::~PFunctionHandler()
|
|
{
|
|
fLines.clear();
|
|
fFuncs.clear();
|
|
fFuncComment.clear();
|
|
}
|
|
|
|
//-------------------------------------------------------------
|
|
// DoParse (public)
|
|
//-------------------------------------------------------------
|
|
/**
|
|
* <p>Calls the function parser.
|
|
*/
|
|
Bool_t PFunctionHandler::DoParse()
|
|
{
|
|
Bool_t success = true;
|
|
PFunctionGrammar function;
|
|
TString line;
|
|
|
|
// feed the function block into the parser. Start with i=1, since i=0 is FUNCTIONS
|
|
for (UInt_t i=1; i<fLines.size(); i++) {
|
|
|
|
// function line to upper case after cutting out prepended comment
|
|
line = fLines[i].fLine;
|
|
Ssiz_t pos = line.First('#'); // find prepended comment
|
|
TString Comment("");
|
|
if (pos != kNPOS) { // comment present
|
|
for (Int_t i=pos; i<line.Length(); i++) {
|
|
Comment += line[i];
|
|
}
|
|
}
|
|
fFuncComment.push_back(Comment);
|
|
if (pos != kNPOS) { // comment present, hence remove it from the string to be parsed
|
|
line.Remove(pos);
|
|
line.Remove(TString::kTrailing, ' ');
|
|
}
|
|
line.ToUpper();
|
|
|
|
// do parsing
|
|
tree_parse_info<> info = ast_parse(line.Data(), function, space_p);
|
|
|
|
if (info.full) { // parsing successful
|
|
PFunction func(info); // generate an evaluation function object based on the AST tree
|
|
fFuncs.push_back(func); // feeds it to the functions vector
|
|
} else {
|
|
cerr << endl << "**ERROR**: FUNCTIONS parse failed in line " << fLines[i].fLineNo << endl;
|
|
success = false;
|
|
break;
|
|
}
|
|
}
|
|
|
|
// check that the function numbers are unique
|
|
if (success) {
|
|
for (UInt_t i=0; i<fFuncs.size(); i++) {
|
|
for (UInt_t j=i+1; j<fFuncs.size(); j++) {
|
|
if (fFuncs[i].GetFuncNo() == fFuncs[j].GetFuncNo()) {
|
|
cerr << endl << "**ERROR**: function number " << fFuncs[i].GetFuncNo();
|
|
cerr << " is at least twice present! Fix this first.";
|
|
cerr << endl;
|
|
success = false;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
return success;
|
|
}
|
|
|
|
//-------------------------------------------------------------
|
|
// CheckMapAndParamRange (public)
|
|
//-------------------------------------------------------------
|
|
/**
|
|
* <p>Check all functions if the map and fit parameters are within valid ranges, i.e. map < mapSize, param < paramSize.
|
|
*
|
|
* <b>return:</b> true if map and fit parameters are withing valid ranges, otherwise false.
|
|
*
|
|
* \param mapSize size of the map vector
|
|
* \param paramSize size of the fit parameter vector
|
|
*/
|
|
Bool_t PFunctionHandler::CheckMapAndParamRange(UInt_t mapSize, UInt_t paramSize)
|
|
{
|
|
Bool_t success = true;
|
|
|
|
for (UInt_t i=0; i<fFuncs.size(); i++) {
|
|
success = fFuncs[i].CheckMapAndParamRange(mapSize, paramSize);
|
|
if (!success)
|
|
break;
|
|
}
|
|
|
|
return success;
|
|
}
|
|
|
|
//-------------------------------------------------------------
|
|
// Eval (public)
|
|
//-------------------------------------------------------------
|
|
/**
|
|
* <p>Evaluate function number funNo for given map and param.
|
|
*
|
|
* <b>return:</b> value of the function for given map and param.
|
|
*
|
|
* \param funNo function number
|
|
* \param map map vector
|
|
* \param param fit parameter vector
|
|
*/
|
|
Double_t PFunctionHandler::Eval(Int_t funNo, vector<Int_t> map, vector<double> param)
|
|
{
|
|
if (GetFuncIndex(funNo) == -1) {
|
|
cerr << endl << "**ERROR**: Couldn't find FUN" << funNo << " for evaluation";
|
|
cerr << endl;
|
|
return 0.0;
|
|
}
|
|
|
|
// set correct map
|
|
fFuncs[GetFuncIndex(funNo)].SetMap(map);
|
|
|
|
// return evaluated function
|
|
return fFuncs[GetFuncIndex(funNo)].Eval(param);
|
|
}
|
|
|
|
//-------------------------------------------------------------
|
|
// GetFuncNo (public)
|
|
//-------------------------------------------------------------
|
|
/**
|
|
* <p>returns the function number
|
|
*
|
|
* \param idx index of the function
|
|
*/
|
|
Int_t PFunctionHandler::GetFuncNo(UInt_t idx)
|
|
{
|
|
if (idx > fFuncs.size())
|
|
return -1;
|
|
|
|
return fFuncs[idx].GetFuncNo();
|
|
}
|
|
|
|
//-------------------------------------------------------------
|
|
// GetFuncIndex (public)
|
|
//-------------------------------------------------------------
|
|
/**
|
|
* <p>return function index for a given function number
|
|
*
|
|
* \param funcNo function number
|
|
*/
|
|
Int_t PFunctionHandler::GetFuncIndex(Int_t funcNo)
|
|
{
|
|
Int_t index = -1;
|
|
|
|
for (UInt_t i=0; i<fFuncs.size(); i++) {
|
|
if (fFuncs[i].GetFuncNo() == funcNo) {
|
|
index = i;
|
|
break;
|
|
}
|
|
}
|
|
|
|
return index;
|
|
}
|
|
|
|
//-------------------------------------------------------------
|
|
// GetFuncString (public)
|
|
//-------------------------------------------------------------
|
|
/**
|
|
* <p>return the (clean and tidy) function string at index idx
|
|
*
|
|
* \param idx index of the function
|
|
*/
|
|
TString PFunctionHandler::GetFuncString(UInt_t idx)
|
|
{
|
|
TString funStr("");
|
|
|
|
if ((idx > fFuncs.size()) || (idx > fFuncComment.size()))
|
|
return funStr;
|
|
|
|
if (fFuncComment[idx].Length() > 0)
|
|
funStr = *fFuncs[idx].GetFuncString() + " " + fFuncComment[idx];
|
|
else
|
|
funStr = *fFuncs[idx].GetFuncString();
|
|
|
|
return funStr;
|
|
}
|