opm-simulators/opm/autodiff/BlackoilAquiferModel_impl.hpp

#include <opm/grid/utility/cartesianToCompressed.hpp>

namespace Opm {


    template<typename TypeTag>
    BlackoilAquiferModel<TypeTag>::
    BlackoilAquiferModel(Simulator& simulator)
        : simulator_(simulator)
    {
        init();
    }

    template<typename TypeTag>
    void
    BlackoilAquiferModel<TypeTag>::initialSolutionApplied()
    {
        for (auto aquifer = aquifers_.begin(); aquifer != aquifers_.end(); ++aquifer) {
            aquifer->initialSolutionApplied();
        }
    }

    template<typename TypeTag>
    void
    BlackoilAquiferModel<TypeTag>::beginEpisode()
    { }

    template<typename TypeTag>
    void
    BlackoilAquiferModel<TypeTag>::beginTimeStep()
    {
        for (auto aquifer = aquifers_.begin(); aquifer != aquifers_.end(); ++aquifer) {
            aquifer->beginTimeStep();
        }
    }

    template<typename TypeTag>
    void
    BlackoilAquiferModel<TypeTag>::beginIteration()
    { }

    template<typename TypeTag>
    template <class Context>
    void
    BlackoilAquiferModel<TypeTag>::addToSource(RateVector& rates,
                                               const Context& context,
                                               unsigned spaceIdx,
                                               unsigned timeIdx) const
    {
        for (auto& aquifer: aquifers_) {
            aquifer.addToSource(rates, context, spaceIdx, timeIdx);
        }
    }

    template<typename TypeTag>
    void
    BlackoilAquiferModel<TypeTag>::endIteration()
    { }

    template<typename TypeTag>
    void
    BlackoilAquiferModel<TypeTag>::endTimeStep()
    {
        for (auto aquifer = aquifers_.begin(); aquifer != aquifers_.end(); ++aquifer) {
            aquifer->endTimeStep();
        }
    }

    template<typename TypeTag>
    void
    BlackoilAquiferModel<TypeTag>::endEpisode()
    { }

    // Initialize the aquifers in the deck
    template<typename TypeTag>
    void
    BlackoilAquiferModel<TypeTag>:: init()
    {
        const auto& deck = this->simulator_.vanguard().deck();

        if ( !deck.hasKeyword("AQUCT") ) {
            return ;
        }

        //updateConnectionIntensiveQuantities();
        const auto& eclState = this->simulator_.vanguard().eclState();

        // Get all the carter tracy aquifer properties data and put it in aquifers vector
        const AquiferCT aquiferct = AquiferCT(eclState,deck);
        const Aquancon aquifer_connect = Aquancon(eclState.getInputGrid(), deck);

        std::vector<AquiferCT::AQUCT_data> aquifersData = aquiferct.getAquifers();
        std::vector<Aquancon::AquanconOutput> aquifer_connection = aquifer_connect.getAquOutput();

        assert( aquifersData.size() == aquifer_connection.size() );

        // generate the mapping from Cartesian coordinates to the index in the actual grid
        const auto& ugrid = simulator_.vanguard().grid();
        const auto& gridView = simulator_.gridView();
        const int number_of_cells = gridView.size(0);

        cartesian_to_compressed_ = cartesianToCompressed(number_of_cells,
                                                         Opm::UgGridHelpers::globalCell(ugrid));

        for (size_t i = 0; i < aquifersData.size(); ++i)
        {
            aquifers_.push_back(
               AquiferCarterTracy<TypeTag> (aquifersData.at(i),
                    aquifer_connection.at(i), cartesian_to_compressed_, this->simulator_)
                               );
        }


    }

    template<typename TypeTag>
    bool
    BlackoilAquiferModel<TypeTag>:: aquiferActive() const
    {
        return !aquifers_.empty();
    }

} // namespace Opm