field_add_potential.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_add_potential.impl.hh
 * @brief   
 */

#ifndef FIELD_ADD_POTENTIAL_IMPL_HH_
#define FIELD_ADD_POTENTIAL_IMPL_HH_

#include "fields/field_add_potential.hh"


template <int spacedim, class Value>
FieldAddPotential<spacedim, Value>::FieldAddPotential(const arma::vec::fixed<spacedim+1> &potential, const Input::AbstractRecord &rec, unsigned int n_comp)
: FieldAlgorithmBase<spacedim, Value>(n_comp),
  inner_field_( FieldAlgorithmBase<spacedim, Value>::function_factory(rec, FieldAlgoBaseInitData(rec.type().type_name(), this->value_.n_rows(), UnitSI::dimensionless())) )
{
    grad_=potential.subvec(0,spacedim-1);
    zero_level_=potential[spacedim];
}



template <int spacedim, class Value>
FieldAddPotential<spacedim, Value>::FieldAddPotential(const arma::vec::fixed<spacedim+1> &potential,
        std::shared_ptr< FieldAlgorithmBase<spacedim, Value> > inner_field, unsigned int n_comp)
: FieldAlgorithmBase<spacedim, Value>(n_comp),
  inner_field_( inner_field )
{
    grad_=potential.subvec(0,spacedim-1);
    zero_level_=potential[spacedim];
}



/**
 * Returns one value in one given point. ResultType can be used to avoid some costly calculation if the result is trivial.
 */
template <int spacedim, class Value>
typename Value::return_type const & FieldAddPotential<spacedim, Value>::value(const Point &p, const ElementAccessor<spacedim> &elm)
{
    this->r_value_ = inner_field_->value(p,elm);

    double potential=arma::dot(grad_ , p) + zero_level_;
    for(unsigned int row=0; row < this->value_.n_rows(); row++)
        for(unsigned int col=0; col < this->value_.n_cols(); col++)
            this->value_(row,col) += potential;

    return this->r_value_;
}



/**
 * Returns std::vector of scalar values in several points at once.
 */
template <int spacedim, class Value>
void FieldAddPotential<spacedim, Value>::value_list (const Armor::array &point_list, const ElementAccessor<spacedim> &elm,
                   std::vector<typename Value::return_type>  &value_list)
{
    ASSERT_EQ( point_list.size(), value_list.size() ).error();
    ASSERT_DBG( point_list.n_rows() == spacedim && point_list.n_cols() == 1 ).error("Invalid point size.\n");
    inner_field_->value_list(point_list, elm, value_list);
    for(unsigned int i=0; i< point_list.size(); i++) {
        double potential= arma::dot(grad_ , point_list.vec<spacedim>(i)) + zero_level_;
        Value envelope(value_list[i]);

        for(unsigned int row=0; row < this->value_.n_rows(); row++)
            for(unsigned int col=0; col < this->value_.n_cols(); col++)
                envelope(row,col) += potential;
    }
}

template <int spacedim, class Value>
bool FieldAddPotential<spacedim,Value>::set_time (const TimeStep &time)
{
    ASSERT_PTR(inner_field_).error("Null data pointer.\n");
    return inner_field_->set_time(time);
}


template <int spacedim, class Value>
void FieldAddPotential<spacedim, Value>::set_mesh(const Mesh *mesh, bool boundary_domain)
{
    ASSERT_PTR(inner_field_).error("Null data pointer.\n");
    return inner_field_->set_mesh(mesh, boundary_domain);
}


template <int spacedim, class Value>
FieldAddPotential<spacedim, Value>::~FieldAddPotential() {
}


#endif /* FIELD_ADD_POTENTIAL_IMPL_HH_ */