CompositeFunction.hpp

Go to the documentation of this file.

/*
 * ------------------------------------------------------------------------------------------------------------
 * 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_COMPOSITEFUNCTION_HPP_
#define GEOSX_MANAGERS_FUNCTIONS_COMPOSITEFUNCTION_HPP_

#include "FunctionBase.hpp"

#include <mathpresso/mathpresso.h>

namespace geosx
{

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

  virtual ~CompositeFunction() override;

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

  virtual void InitializeFunction() override;

  virtual void Evaluate( dataRepository::Group const * const group,
                         real64 const time,
                         SortedArrayView< localIndex const > const & set,
                         real64_array & result ) const override final;

  virtual real64 Evaluate( real64 const * const input ) const override final;

private:

  string_array m_functionNames;
  string_array m_variableNames;
  string m_expression;

  mathpresso::Context parserContext;
  mathpresso::Expression parserExpression;

  localIndex m_numSubFunctions;
  static constexpr localIndex m_maxNumSubFunctions = 10;
  std::vector< FunctionBase * > m_subFunctions;

};

} /* namespace geosx */

#endif /* GEOSX_MANAGERS_FUNCTIONS_COMPOSITEFUNCTION_HPP_ */