field_python.impl.hh

Go to the documentation of this file.

/*!
 *
 * Copyright (C) 2015 Technical University of Liberec.  All rights reserved.
 * 
 * This program is free software; you can redistribute it and/or modify it under
 * the terms of the GNU General Public License version 3 as published by the
 * Free Software Foundation. (http://www.gnu.org/licenses/gpl-3.0.en.html)
 * 
 * This program is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
 * FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more details.
 *
 * 
 * @file    field_python.impl.hh
 * @brief   
 */
 
#ifndef FIELD_PYTHON_IMPL_HH_
#define FIELD_PYTHON_IMPL_HH_
 
 
#include <type_traits>
#include "fields/field_python.hh"
#include "fields/field_set.hh"
#include "fields/python_field_proxy.hh" // TODO check if include is necessary
#include <pybind11/pybind11.h>
#include <pybind11/eval.h>
#include <pybind11/stl.h>
 
namespace py = pybind11;
 
/// Implementation.
 
namespace it = Input::Type;
 
FLOW123D_FORCE_LINK_IN_CHILD(field_python)
 
 
 
template <int spacedim, class Value>
const Input::Type::Record & FieldPython<spacedim, Value>::get_input_type()
{
    return it::Record("FieldPython", FieldAlgorithmBase<spacedim,Value>::template_name()+" Field given by a Python script.")
        .derive_from(FieldAlgorithmBase<spacedim, Value>::get_input_type())
        .copy_keys(FieldAlgorithmBase<spacedim, Value>::get_field_algo_common_keys())
        .declare_key("source_file", it::String(), it::Default::obligatory(),
                "Python script given as external file in format 'dir'.'file_name' without .py extension")
        .declare_key("class", it::String(), it::Default::obligatory(),
                "Function in the given script that returns tuple containing components of the return type.\n"
                "For NxM tensor values: tensor(row,col) = tuple( M*row + col ).")
        .declare_key("used_fields", it::Array(it::String()), it::Default("[]"),
                "Defines list of fields necessary in evaluation of actual field.")
        //.declare_key("units", FieldAlgorithmBase<spacedim, Value>::get_field_algo_common_keys(), it::Default::optional(),
        //      "Definition of unit.")
        .close();
}
 
 
template <int spacedim, class Value>
const int FieldPython<spacedim, Value>::registrar =
        Input::register_class< FieldPython<spacedim, Value>, unsigned int >("FieldPython") +
        FieldPython<spacedim, Value>::get_input_type().size();
 
 
 
template <int spacedim, class Value>
FieldPython<spacedim, Value>::FieldPython(unsigned int n_comp)
: FieldAlgorithmBase<spacedim, Value>( n_comp)
{
    this->is_constant_in_space_ = false;
}
 
 
 
 
template <int spacedim, class Value>
void FieldPython<spacedim, Value>::init_from_input(const Input::Record &rec, const struct FieldAlgoBaseInitData& init_data) {
    this->init_unit_conversion_coefficient(rec, init_data);
    this->field_name_ = init_data.field_name_;
 
    std::string source_file = rec.val<string>("source_file");
    std::string source_class = rec.val<string>("class");
    try {
        set_python_field_from_class( source_file, source_class );
    } INPUT_CATCH(FilePath::ExcFileOpen, FilePath::EI_Address_String, rec)
 
    in_rec_ = rec;
}
 
 
 
template <int spacedim, class Value>
void FieldPython<spacedim, Value>::set_python_field_from_class(const string &file_name, const string &class_name)
{
    py::module_ flowpy_module = PythonLoader::load_module_by_name("flowpy");
    py::module_ class_module = PythonLoader::load_module_from_file( string(file_name) );
    try {
        user_class_instance_ = flowpy_module.attr("PythonFieldBase").attr("_create")(class_module, class_name.c_str());
    } catch (const py::error_already_set &ex) {
        PythonLoader::throw_error(ex);
    }
}
 
 
 
template <int spacedim, class Value>
std::vector<const FieldCommon * > FieldPython<spacedim, Value>::set_dependency(FieldSet &field_set) {
    required_fields_.clear();
 
    auto used_fields_array = in_rec_.val<Input::Array>("used_fields");
    std::vector<std::string> used_fields_vec;
    used_fields_array.copy_to(used_fields_vec);
    for(auto field_name : used_fields_vec) {
        auto field_ptr = field_set.field(field_name);
        if (field_ptr != nullptr) required_fields_.push_back( field_ptr );
        else THROW( FieldSet::ExcUnknownField() << FieldCommon::EI_Field(field_name) << FieldSet::EI_FieldType("python declaration") << Input::EI_Address( in_rec_.address_string() ) );
    }
 
    // instance of FieldCommon of this field (see cache_reinit method)
    self_field_ptr_ = field_set.field(this->field_name_);
 
    return required_fields_;
}
 
 
 
template <int spacedim, class Value>
void FieldPython<spacedim, Value>::cache_reinit(FMT_UNUSED const ElementCacheMap &cache_map)
{
    std::vector<FieldCacheProxy> field_data;
    for (auto field_ptr : required_fields_) {
        std::string field_name = field_ptr->name();
        double * cache_data = field_ptr->value_cache()->data_;
        field_data.emplace_back(field_name, field_ptr->shape_, cache_data, CacheMapElementNumber::get()*field_ptr->n_shape());
    }
 
    double * cache_data = self_field_ptr_->value_cache()->data_;
    FieldCacheProxy result_data(this->field_name_, self_field_ptr_->shape_, cache_data, (CacheMapElementNumber::get()*self_field_ptr_->n_shape()));
 
    try {
        py::object p_func = user_class_instance_.attr("_cache_reinit");
        p_func(this->time_.end(), field_data, result_data);
    } catch (const py::error_already_set &ex) {
        PythonLoader::throw_error(ex);
    }
}
 
 
 
template <int spacedim, class Value>
void FieldPython<spacedim, Value>::cache_update(FMT_UNUSED FieldValueCache<typename Value::element_type> &data_cache,
        ElementCacheMap &cache_map, unsigned int region_patch_idx)
{
    unsigned int reg_chunk_begin = cache_map.region_chunk_begin(region_patch_idx);
    unsigned int reg_chunk_end = cache_map.region_chunk_end(region_patch_idx);
    try {
        py::object p_func = user_class_instance_.attr("_cache_update");
        p_func(this->field_name_, reg_chunk_begin, reg_chunk_end);
    } catch (const py::error_already_set &ex) {
        PythonLoader::throw_error(ex);
    }
}
 
 
 
template <int spacedim, class Value>
FieldPython<spacedim, Value>::~FieldPython() {}
 
 
#endif /* FIELD_PYTHON_IMPL_HH_ */