PFunction.h

Go to the documentation of this file.

/***************************************************************************
 
  PFunction.h
 
  Author: Andreas Suter
  e-mail: andreas.suter@psi.ch
 
***************************************************************************/
 
/***************************************************************************
 *   Copyright (C) 2007-2026 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 _PFUNCTION_H_
#define _PFUNCTION_H_
 
#include <vector>
 
#include <boost/version.hpp>
#include <boost/variant/static_visitor.hpp>
#include <boost/variant/apply_visitor.hpp>
 
#include <TString.h>
 
#include "PMusr.h"
#include "PFunctionAst.h"
#include "PFunctionGrammar.h"
 
//----------------------------------------------------------------------------
class PFunction {
  public:
    //------------------------------------------------------------------------
    PFunction(const musrfit::ast::assignment& assignment);
 
    //------------------------------------------------------------------------
    virtual ~PFunction();
 
    //------------------------------------------------------------------------
    virtual Bool_t IsValid() { return fValid; }
 
    //------------------------------------------------------------------------
    virtual Int_t GetFuncNo() { return fFuncNo; }
 
    //------------------------------------------------------------------------
    virtual Bool_t CheckMapAndParamRange(UInt_t mapSize, UInt_t paramSize);
 
    //------------------------------------------------------------------------
    virtual Double_t Eval(std::vector<Double_t> param, PMetaData metaData);
 
    //------------------------------------------------------------------------
    virtual void SetMap(std::vector<Int_t> map) { fMap = map; }
 
    //------------------------------------------------------------------------
    virtual TString* GetFuncString() { return &fFuncString; }
 
  protected:
    //------------------------------------------------------------------------
    virtual Bool_t FindAndCheckMapAndParamRange(const musrfit::ast::operand& operand,
                                                  UInt_t mapSize, UInt_t paramSize);
 
    //------------------------------------------------------------------------
    virtual TString GenerateString(const musrfit::ast::expression& expr);
 
    //------------------------------------------------------------------------
    virtual TString GenerateStringOperand(const musrfit::ast::operand& operand);
 
  private:
    class EvalVisitor : public boost::static_visitor<Double_t>
    {
    public:
      EvalVisitor(const std::vector<Int_t>& map,
                  const std::vector<Double_t>& param,
                  const PMetaData& metaData)
        : fMap(map), fParam(param), fMetaData(metaData) {}
 
      Double_t operator()(const musrfit::ast::leer&) const;
      Double_t operator()(double val) const;
      Double_t operator()(const musrfit::ast::constant& c) const;
      Double_t operator()(const musrfit::ast::parameter& p) const;
      Double_t operator()(const musrfit::ast::map_ref& m) const;
      Double_t operator()(const musrfit::ast::function_call& f) const;
      Double_t operator()(const musrfit::ast::power_call& p) const;
      Double_t operator()(const musrfit::ast::expression& expr) const;
 
    private:
      const std::vector<Int_t>& fMap;
      const std::vector<Double_t>& fParam;
      const PMetaData& fMetaData;
    };
 
    musrfit::ast::expression fAst; 
    std::vector<Double_t> fParam;  
    std::vector<Int_t> fMap;       
 
    Bool_t fValid;        
    Int_t fFuncNo;        
    TString fFuncString;  
    PMetaData fMetaData;  
};
 
#endif // _PFUNCTION_H_