concentration_model.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    concentration_model.hh
 * @brief   Discontinuous Galerkin method for equation of transport with dispersion.
 * @author  Jan Stebel
 */

#ifndef CONC_TRANS_MODEL_HH_
#define CONC_TRANS_MODEL_HH_

#include "advection_diffusion_model.hh"

#include "fields/bc_field.hh"
#include "fields/bc_multi_field.hh"
#include "fields/field.hh"
#include "fields/multi_field.hh"




class ConcentrationTransportModel : public AdvectionDiffusionModel, public ConcentrationTransportBase {
public:

    class ModelEqData : public TransportEqData {
    public:

        enum Concentration_bc_types {
            bc_inflow,
            bc_dirichlet,
            bc_total_flux,
            bc_diffusive_flux
        };

        /// Type of boundary condition (see also BC_Type)
        BCMultiField<3, FieldValue<3>::Enum > bc_type;
        /// Prescribed concentration for Dirichlet/reference concentration for flux b.c.
        BCMultiField<3, FieldValue<3>::Scalar> bc_dirichlet_value;
        /// Flux value in total/diffusive flux b.c.
        BCMultiField<3, FieldValue<3>::Scalar > bc_flux;
        /// Transition coefficient in total/diffusive flux b.c.
        BCMultiField<3, FieldValue<3>::Scalar > bc_robin_sigma;
        /// Initial concentrations.
        MultiField<3, FieldValue<3>::Scalar> init_conc;
        /// Longitudal dispersivity (for each substance).
        MultiField<3, FieldValue<3>::Scalar> disp_l;
        /// Transversal dispersivity (for each substance).
        MultiField<3, FieldValue<3>::Scalar> disp_t;
        /// Molecular diffusivity (for each substance).
        MultiField<3, FieldValue<3>::Scalar> diff_m;

        Field<3, FieldValue<3>::Scalar > rock_density;      ///< Rock matrix density.
        MultiField<3, FieldValue<3>::Scalar > sorption_coefficient;     ///< Coefficient of linear sorption.


        MultiField<3, FieldValue<3>::Scalar> output_field;



        ModelEqData();

        static constexpr const char * name() { return "Solute_AdvectionDiffusion"; }

        static string default_output_field() { return "\"conc\""; }

        static const Input::Type::Selection & get_bc_type_selection();

        static IT::Selection get_output_selection();

    };


    typedef ConcentrationTransportBase FactoryBaseType;


    ConcentrationTransportModel(Mesh &mesh, const Input::Record &in_rec);

        void init_from_input(const Input::Record &in_rec) override;

    void compute_mass_matrix_coefficient(const std::vector<arma::vec3 > &point_list,
            const ElementAccessor<3> &ele_acc,
            std::vector<double> &mm_coef) override;

    void compute_retardation_coefficient(const std::vector<arma::vec3 > &point_list,
            const ElementAccessor<3> &ele_acc,
            std::vector<std::vector<double> > &ret_coef) override;



    void compute_advection_diffusion_coefficients(const std::vector<arma::vec3 > &point_list,
            const std::vector<arma::vec3> &velocity,
            const ElementAccessor<3> &ele_acc,
            std::vector<std::vector<arma::vec3> > &ad_coef,
            std::vector<std::vector<arma::mat33> > &dif_coef) override;

    void compute_init_cond(const std::vector<arma::vec3> &point_list,
            const ElementAccessor<3> &ele_acc,
            std::vector< arma::vec > &init_values) override;

    void get_bc_type(const ElementAccessor<3> &ele_acc,
                arma::uvec &bc_types) override;

    void get_flux_bc_data(unsigned int index,
            const std::vector<arma::vec3> &point_list,
            const ElementAccessor<3> &ele_acc,
            std::vector< double > &bc_flux,
            std::vector< double > &bc_sigma,
            std::vector< double > &bc_ref_value) override;

    void get_flux_bc_sigma(unsigned int index,
            const std::vector<arma::vec3> &point_list,
            const ElementAccessor<3> &ele_acc,
            std::vector< double > &bc_sigma) override;

    void compute_source_coefficients(const std::vector<arma::vec3> &point_list,
                const ElementAccessor<3> &ele_acc,
                std::vector<arma::vec> &sources_conc,
                std::vector<arma::vec> &sources_density,
                std::vector<arma::vec> &sources_sigma) override;

    void compute_sources_sigma(const std::vector<arma::vec3> &point_list,
                const ElementAccessor<3> &ele_acc,
                std::vector<arma::vec> &sources_sigma) override;

    ~ConcentrationTransportModel() override;


    /**
     * @brief Updates the velocity field which determines some coefficients of the transport equation.
     *
         * @param dh mixed hybrid dof handler
         *
     * (So far it does not work since the flow module returns a vector of zeros.)
     * @param velocity_vector Input array of velocity values.
     */
    inline void set_velocity_field(const MH_DofHandler &dh) override
    {
        mh_dh = &dh;
        flux_changed = true;
    }

    /// Returns number of transported substances.
    inline unsigned int n_substances() override
    { return substances_.size(); }

    /// Returns reference to the vector of substance names.
    inline SubstanceList &substances() override
    { return substances_; }


    // Methods inherited from ConcentrationTransportBase:

    void set_target_time(double target_time) override {};

    void set_balance_object(std::shared_ptr<Balance> balance) override;

    const vector<unsigned int> &get_subst_idx() override
    { return subst_idx; }

    void set_output_stream(std::shared_ptr<OutputTime> stream)
    { output_stream_ = stream; }

    std::shared_ptr<OutputTime> output_stream() override
    { return output_stream_; }



protected:

    /// Derived class should implement getter for ModelEqData instance.
    virtual ModelEqData &data() = 0;

    /**
     * Create input type that can be passed to the derived class.
     * @param implementation String characterizing the numerical method, e.g. DG, FEM, FVM.
     * @param description    Comment used to describe the record key.
     * @return
     */
    static IT::Record get_input_type(const string &implementation, const string &description);

    /**
     * Formula to calculate the dispersivity tensor.
     * @param velocity  Fluid velocity.
     * @param Dm        Molecular diffusivity.
     * @param alphaL    Longitudal dispersivity.
     * @param alphaT    Transversal dispersivity.
     * @param porosity  Porosity.
     * @param cross_cut Cross-section.
     * @param K         Dispersivity tensor (output).
     */
    void calculate_dispersivity_tensor(const arma::vec3 &velocity,
            double Dm,
            double alphaL,
            double alphaT,
            double water_content,
            double porosity,
            double cross_cut,
            arma::mat33 &K);

    /// Indicator of change in advection vector field.
    bool flux_changed;

    /// Transported substances.
    SubstanceList substances_;

    /// List of indices used to call balance methods for a set of quantities.
    vector<unsigned int> subst_idx;

    /// Density of liquid (a global constant).
    double solvent_density_;

    /**
     * Temporary solution how to pass velocity field form the flow model.
     * TODO: introduce FieldDiscrete -containing true DOFHandler and data vector and pass such object together with other
     * data. Possibly make more general set_data method, allowing setting data given by name. needs support from EqDataBase.
     */
    const MH_DofHandler *mh_dh;

    std::shared_ptr<OutputTime> output_stream_;



};








#endif /* CONC_TRANS_MODEL_HH_ */