Files
opm-common/python/cxx/parser.cpp
2020-02-19 12:35:07 +01:00

64 lines
2.0 KiB
C++

#include <string>
#include <exception>
#include <opm/json/JsonObject.hpp>
#include <opm/parser/eclipse/Parser/Parser.hpp>
#include <opm/parser/eclipse/Parser/ParserKeyword.hpp>
#include <opm/parser/eclipse/Deck/Deck.hpp>
#include <pybind11/stl.h>
#include <opm/parser/eclipse/Parser/ErrorGuard.hpp>
#include "export.hpp"
namespace {
Deck create_deck( const std::string& deckStr,
const ParseContext& pc,
const Parser& parser) {
Opm::ErrorGuard guard;
auto p = parser.parseFile( deckStr, pc, guard );
guard.clear();
return p;
}
Deck create_deck_string( const std::string& deckStr,
const ParseContext& pc,
const Parser& parser) {
Opm::ErrorGuard guard;
auto p = parser.parseString( deckStr, pc, guard );
guard.clear();
return p;
}
void add_keyword(Parser* parser, const std::string& json_string) {
const Json::JsonObject keyword(json_string);
parser->addParserKeyword(keyword);
}
}
void python::common::export_Parser(py::module& module) {
module.def( "create_deck", &create_deck );
module.def( "create_deck_string", &create_deck_string);
py::class_<ParserKeyword>(module, "ParserKeyword")
.def_property_readonly("name", &ParserKeyword::getName);
py::class_<Parser>(module, "Parser")
.def(py::init<>())
.def("parse", py::overload_cast<const std::string&>(&Parser::parseFile, py::const_))
.def("parse" , py::overload_cast<const std::string&, const ParseContext&>(&Parser::parseFile, py::const_))
.def("parse_string", py::overload_cast<const std::string&>(&Parser::parseString, py::const_))
.def("parse_string", py::overload_cast<const std::string&, const ParseContext&>(&Parser::parseString, py::const_))
.def("add_keyword", add_keyword)
.def("__getitem__", &Parser::getKeyword, ref_internal);
}