/*
  Copyright 2012 SINTEF ICT, Applied Mathematics.

  This file is part of the Open Porous Media project (OPM).

  OPM is free software: you can redistribute it and/or modify
  it under the terms of the GNU General Public License as published by
  the Free Software Foundation, either version 3 of the License, or
  (at your option) any later version.

  OPM 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.

  You should have received a copy of the GNU General Public License
  along with OPM.  If not, see <http://www.gnu.org/licenses/>.
*/


#include "config.h"
#include <opm/core/props/pvt/PvtDead.hpp>
#include <algorithm>

// Extra includes for debug dumping of tables.
// #include <boost/lexical_cast.hpp>
// #include <string>
// #include <fstream>

namespace Opm
{

    //------------------------------------------------------------------------
    // Member functions
    //-------------------------------------------------------------------------
    /// Constructor
    void PvtDead::initFromOil(Opm::DeckKeywordConstPtr pvdoKeyword)
    {
        int numRegions = Opm::PvdoTable::numTables(pvdoKeyword);

        // resize the attributes of the object
        b_.resize(numRegions);
        viscosity_.resize(numRegions);
        inverseBV_.resize(numRegions);

        for (int regionIdx = 0; regionIdx < numRegions; ++regionIdx) {
            Opm::PvdoTable pvdoTable(pvdoKeyword, regionIdx);
            // Copy data
            const std::vector<double>& press = pvdoTable.getPressureColumn();
            const std::vector<double>& b = pvdoTable.getFormationFactorColumn();
            const std::vector<double>& visc = pvdoTable.getViscosityColumn();

            const int sz = b.size();
            std::vector<double> bInv(sz);
            for (int i = 0; i < sz; ++i) {
                bInv[i] = 1.0 / b[i];
            }

            // TODO: should we change the name of b so that we know it is the 
            // inverse more explicitly?
            // or use the captial B in a exlicit way to show the difference.
            std::vector<double> bvInv(sz);
            for (int i = 0; i < sz; ++i) {
                bvInv[i] = 1.0 / (b[i] * visc[i]);
            }

            b_[regionIdx] = NonuniformTableLinear<double>(press, bInv);
            viscosity_[regionIdx] = NonuniformTableLinear<double>(press, visc);
            inverseBV_[regionIdx] = NonuniformTableLinear<double>(press, bvInv);
        }
    }


    void PvtDead::initFromGas(Opm::DeckKeywordConstPtr pvdgKeyword)
    {
        int numRegions = Opm::PvdgTable::numTables(pvdgKeyword);

        // resize the attributes of the object
        b_.resize(numRegions);
        viscosity_.resize(numRegions);
        inverseBV_.resize(numRegions);

        for (int regionIdx = 0; regionIdx < numRegions; ++regionIdx) {
            Opm::PvdgTable pvdgTable(pvdgKeyword, regionIdx);

            // Copy data
            const std::vector<double>& press = pvdgTable.getPressureColumn();
            const std::vector<double>& b = pvdgTable.getFormationFactorColumn();
            const std::vector<double>& visc = pvdgTable.getViscosityColumn();

            const int sz = b.size();
            std::vector<double> bInv(sz);
            for (int i = 0; i < sz; ++i) {
                bInv[i] = 1.0 / b[i];
            }

            std::vector<double> bvInv(sz);
            for (int i = 0; i < sz; ++i) {
                bvInv[i] = 1.0 / (b[i] * visc[i]);
            }

            b_[regionIdx] = NonuniformTableLinear<double>(press, bInv);
            viscosity_[regionIdx] = NonuniformTableLinear<double>(press, visc);
            inverseBV_[regionIdx] = NonuniformTableLinear<double>(press, bvInv);
        }
    }

    // Destructor
    PvtDead::~PvtDead()
    {
    }



    void PvtDead::mu(const int n,
                     const int* pvtTableIdx,
                           const double* p,
                           const double* /*z*/,
                           double* output_mu) const
    {
// #pragma omp parallel for
        for (int i = 0; i < n; ++i) {
            int regionIdx = getTableIndex_(pvtTableIdx, i);
            // output_mu[i] = viscosity_[regionIdx](p[i]);
            double tempInvB = b_[regionIdx](p[i]);
            double tempInvBV = inverseBV_[regionIdx](p[i]);
            output_mu[i] = tempInvB / tempInvBV;
        }
    }

    void PvtDead::mu(const int n,
                     const int* pvtTableIdx,
                               const double* p,
                               const double* /*r*/,
                               double* output_mu,
                               double* output_dmudp,
                               double* output_dmudr) const
        {
    // #pragma omp parallel for
            for (int i = 0; i < n; ++i) {
                int regionIdx = getTableIndex_(pvtTableIdx, i);
                // output_mu[i] = viscosity_[regionIdx](p[i]);
                // output_dmudp[i] = viscosity_[regionIdx].derivative(p[i]);
                double tempInvB = b_[regionIdx](p[i]);
                double tempInvBV = inverseBV_[regionIdx](p[i]);
                output_mu[i] = tempInvB / tempInvBV;
                output_dmudp[i] = (tempInvBV * b_[regionIdx].derivative(p[i])
                                 - tempInvB * inverseBV_[regionIdx].derivative(p[i])) / (tempInvBV * tempInvBV);
            }
            std::fill(output_dmudr, output_dmudr + n, 0.0);

        }

    void PvtDead::mu(const int n,
                     const int* pvtTableIdx,
                               const double* p,
                               const double* /*r*/,
                               const PhasePresence* /*cond*/,
                               double* output_mu,
                               double* output_dmudp,
                               double* output_dmudr) const
        {
    // #pragma omp parallel for
            for (int i = 0; i < n; ++i) {
                int regionIdx = getTableIndex_(pvtTableIdx, i);
                double tempInvB = b_[regionIdx](p[i]);
                double tempInvBV = inverseBV_[regionIdx](p[i]);
                // output_mu[i] = viscosity_[regionIdx](p[i]);
                output_mu[i] = tempInvB / tempInvBV;
                // output_dmudp[i] = viscosity_[regionIdx].derivative(p[i]);
                // output_dmudp[i] = tempInvB / (tempInvBV * tempInvBV) * inverseBV_[regionIdx].derivative(p[i]);
                output_dmudp[i] = (tempInvBV * b_[regionIdx].derivative(p[i])
                                 - tempInvB * inverseBV_[regionIdx].derivative(p[i])) / (tempInvBV * tempInvBV);
            }
            std::fill(output_dmudr, output_dmudr + n, 0.0);

        }

    void PvtDead::B(const int n,
                    const int* pvtTableIdx,
                          const double* p,
                          const double* /*z*/,
                          double* output_B) const
    {
// #pragma omp parallel for
        // B = 1/b
        for (int i = 0; i < n; ++i) {
            int regionIdx = getTableIndex_(pvtTableIdx, i);
            output_B[i] = 1.0/b_[regionIdx](p[i]);
        }
    }

    void PvtDead::dBdp(const int n,
                       const int* pvtTableIdx,
                             const double* p,
                             const double* /*z*/,
                             double* output_B,
                             double* output_dBdp) const
    {
        B(n, pvtTableIdx, p, 0, output_B);
// #pragma omp parallel for
        for (int i = 0; i < n; ++i) {
            int regionIdx = getTableIndex_(pvtTableIdx, i);
            double Bg = output_B[i];
            output_dBdp[i] = -Bg*Bg*b_[regionIdx].derivative(p[i]);
        }
    }

    void PvtDead::b(const int n,
                    const int* pvtTableIdx,
                              const double* p,
                              const double* /*r*/,
                              double* output_b,
                              double* output_dbdp,
                              double* output_dbdr) const

        {
    // #pragma omp parallel for
            for (int i = 0; i < n; ++i) {
                int regionIdx = getTableIndex_(pvtTableIdx, i);

                output_b[i] = b_[regionIdx](p[i]);
                output_dbdp[i] = b_[regionIdx].derivative(p[i]);

            }
            std::fill(output_dbdr, output_dbdr + n, 0.0);

        }

    void PvtDead::b(const int n,
                    const int* pvtTableIdx,
                              const double* p,
                              const double* /*r*/,
                              const PhasePresence* /*cond*/,
                              double* output_b,
                              double* output_dbdp,
                              double* output_dbdr) const

        {
    // #pragma omp parallel for
            for (int i = 0; i < n; ++i) {
                int regionIdx = getTableIndex_(pvtTableIdx, i);

                output_b[i] = b_[regionIdx](p[i]);
                output_dbdp[i] = b_[regionIdx].derivative(p[i]);

            }
            std::fill(output_dbdr, output_dbdr + n, 0.0);

        }

    void PvtDead::rsSat(const int n,
                        const int* /*pvtTableIdx*/,
                             const double* /*p*/,
                             double* output_rsSat,
                             double* output_drsSatdp) const
    {
        std::fill(output_rsSat, output_rsSat + n, 0.0);
        std::fill(output_drsSatdp, output_drsSatdp + n, 0.0);
    }

    void PvtDead::rvSat(const int n,
                        const int* /*pvtTableIdx*/,
                             const double* /*p*/,
                             double* output_rvSat,
                             double* output_drvSatdp) const
    {
        std::fill(output_rvSat, output_rvSat + n, 0.0);
        std::fill(output_drvSatdp, output_drvSatdp + n, 0.0);
    }

    void PvtDead::R(const int n,
                    const int* /*pvtTableIdx*/,
                          const double* /*p*/,
                          const double* /*z*/,
                          double* output_R) const
    {
        std::fill(output_R, output_R + n, 0.0);
    }

    void PvtDead::dRdp(const int n,
                       const int* /*pvtTableIdx*/,
                             const double* /*p*/,
                             const double* /*z*/,
                             double* output_R,
                             double* output_dRdp) const
    {
        std::fill(output_R, output_R + n, 0.0);
        std::fill(output_dRdp, output_dRdp + n, 0.0);
    }

}