BasisFunction.cpp

 
/* SPDX-License-Identifier: CC0-1.0 */
#include <iostream>
#include "BasisFunction.h"
#include "IException.h"
#include "IString.h"
 
namespace Isis {
 
  BasisFunction::BasisFunction(const QString &name, int numVars, int numCoefs) {
    
    p_name = name;
    p_numVars = numVars;
    p_numCoefs = numCoefs;
  }
 
  
  void BasisFunction::SetCoefficients(const std::vector<double> &coefs) {
    
    if ( (int)coefs.size() != p_numCoefs ) {
      QString msg = "Unable to set coefficients vector. The size of the given vector [" 
                    + toString((int)coefs.size()) + "] does not match number of coefficients "
                    "in the basis equation [" + toString(p_numCoefs) + "]";
      throw IException(IException::Programmer, msg, _FILEINFO_);
    }
    
    p_coefs = coefs;
  }
 
  
  double BasisFunction::Evaluate(const std::vector<double> &vars) {
    
    if ( (int)vars.size() != p_numVars ) {
      QString msg = "Unable to evaluate function for the given vector of values. "
                    "The size of the given vector [" 
                    + toString((int)vars.size()) + "] does not match number of variables "
                    "in the basis equation [" + toString(p_numVars) + "]";
      throw IException(IException::Programmer, msg, _FILEINFO_);
    }
 
    Expand(vars);
    
    if ( (int)p_terms.size() != p_numCoefs ) {
      QString msg = "Unable to evaluate function for the given vector of values. "
                    "The number of terms in the expansion [" 
                    + toString( (int)p_terms.size() ) + "] does not match number of coefficients "
                    "in the basis equation [" + toString(p_numCoefs) + "]";
      throw IException(IException::Programmer, msg, _FILEINFO_);
    }
 
    double result = 0.0;
    
    for (int i = 0; i < p_numCoefs; i++) {
      result += p_coefs[i] * p_terms[i];
    }
    
    return result;
  }
  
  
  double BasisFunction::Evaluate(const double &var) {
    
    std::vector<double> vars;
    vars.push_back(var);
    return BasisFunction::Evaluate(vars);
  }
 
  
  void BasisFunction::Expand(const std::vector<double> &vars) {
    p_terms = vars;
  }
}