SymbolicFunction.hpp

/*
 * ------------------------------------------------------------------------------------------------------------
 * SPDX-License-Identifier: LGPL-2.1-only
 *
 * Copyright (c) 2016-2024 Lawrence Livermore National Security LLC
 * Copyright (c) 2018-2024 Total, S.A
 * Copyright (c) 2018-2024 The Board of Trustees of the Leland Stanford Junior University
 * Copyright (c) 2023-2024 Chevron
 * Copyright (c) 2019-     GEOS/GEOSX Contributors
 * All rights reserved
 *
 * See top level LICENSE, COPYRIGHT, CONTRIBUTORS, NOTICE, and ACKNOWLEDGEMENTS files for details.
 * ------------------------------------------------------------------------------------------------------------
 */
 
#ifndef GEOS_FUNCTIONS_SYMBOLICFUNCTION_HPP_
#define GEOS_FUNCTIONS_SYMBOLICFUNCTION_HPP_
 
#include "FunctionBase.hpp"
 
#include <mathpresso/mathpresso.h>
 
namespace geos
{
 
class SymbolicFunction : public FunctionBase
{
public:
  SymbolicFunction( const string & name,
                    dataRepository::Group * const parent );
 
  virtual ~SymbolicFunction() override;
 
  static string catalogName() { return "SymbolicFunction"; }
 
  virtual void initializeFunction() override;
 
  inline void evaluate( dataRepository::Group const & group,
                        real64 const time,
                        SortedArrayView< localIndex const > const & set,
                        arrayView1d< real64 > const & result ) const override final
  {
    FunctionBase::evaluateT< SymbolicFunction >( group, time, set, result );
  }
 
  inline real64 evaluate( real64 const * const input ) const override final
  {
    return parserExpression.evaluate( reinterpret_cast< void * >( const_cast< real64 * >(input) ) );
  }
 
 
  void setSymbolicVariableNames( string_array variableNames ) { m_variableNames = std::move( variableNames ); }
 
  void setSymbolicExpression( string expression ) { m_expression = std::move( expression ); }
 
 
 
private:
  // Symbolic math driver objects
  mathpresso::Context parserContext;
  mathpresso::Expression parserExpression;
 
  string_array m_variableNames;
 
  string m_expression;
};
 
 
} /* namespace geos */
 
#endif /* GEOS_FUNCTIONS_SYMBOLICFUNCTION_HPP_ */