SymbolicFunction.hpp

/*
 * ------------------------------------------------------------------------------------------------------------
 * SPDX-License-Identifier: LGPL-2.1-only
 *
 * Copyright (c) 2018-2020 Lawrence Livermore National Security LLC
 * Copyright (c) 2018-2020 The Board of Trustees of the Leland Stanford Junior University
 * Copyright (c) 2018-2020 Total, S.A
 * Copyright (c) 2019-     GEOSX Contributors
 * All rights reserved
 *
 * See top level LICENSE, COPYRIGHT, CONTRIBUTORS, NOTICE, and ACKNOWLEDGEMENTS files for details.
 * ------------------------------------------------------------------------------------------------------------
 */
#ifndef GEOSX_MANAGERS_FUNCTIONS_SYMBOLICFUNCTION_HPP_
#define GEOSX_MANAGERS_FUNCTIONS_SYMBOLICFUNCTION_HPP_

#include "FunctionBase.hpp"

#include <mathpresso/mathpresso.h>

namespace geosx
{

class SymbolicFunction : public FunctionBase
{
public:
  SymbolicFunction( const std::string & name,
                    dataRepository::Group * const parent );

  virtual ~SymbolicFunction() override;

  static string CatalogName() { return "SymbolicFunction"; }

  virtual void InitializeFunction() override;

  inline void Evaluate( dataRepository::Group const * const group,
                        real64 const time,
                        SortedArrayView< localIndex const > const & set,
                        real64_array & 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 = variableNames; }

  void setSymbolicExpression( string expression ) { m_expression = expression; }



private:
  // Symbolic math driver objects
  mathpresso::Context parserContext;
  mathpresso::Expression parserExpression;

  string_array m_variableNames;

  string m_expression;
};


} /* namespace geosx */

#endif /* GEOSX_MANAGERS_FUNCTIONS_SYMBOLICFUNCTION_HPP_ */