/////////////////////////////////////////////////////////////////////////////////
//
//  Copyright (C) 2017-     Statoil ASA
// 
//  ResInsight 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.
// 
//  ResInsight 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 at <http://www.gnu.org/licenses/gpl.html> 
//  for more details.
//
/////////////////////////////////////////////////////////////////////////////////

#include "RigTransmissibilityEquations.h"

#include "cvfBase.h"
#include "cvfMath.h"

#include <cmath>


//--------------------------------------------------------------------------------------------------
/// 
//--------------------------------------------------------------------------------------------------
double RigTransmissibilityEquations::peacemanRadius(double permeabilityNormalDirection1, 
                                                    double permeabilityNormalDirection2, 
                                                    double cellSizeNormalDirection1, 
                                                    double cellSizeNormalDirection2)
{
    double numerator = cvf::Math::sqrt(
                       pow(cellSizeNormalDirection2, 2) * pow(permeabilityNormalDirection1 / permeabilityNormalDirection2, 1 / 2)
                     + pow(cellSizeNormalDirection1, 2) * pow(permeabilityNormalDirection2 / permeabilityNormalDirection1, 1 / 2));

    double denominator = pow((permeabilityNormalDirection1 / permeabilityNormalDirection2), 1 / 4)
                       + pow((permeabilityNormalDirection2 / permeabilityNormalDirection1), 1 / 4);

    double r0 = 0.28 * numerator / denominator;

    return r0;
}

//--------------------------------------------------------------------------------------------------
/// 
//--------------------------------------------------------------------------------------------------
double RigTransmissibilityEquations::wellBoreTransmissibilityComponent(double cellPerforationVectorComponent, 
                                                                       double permeabilityNormalDirection1, 
                                                                       double permeabilityNormalDirection2, 
                                                                       double cellSizeNormalDirection1, 
                                                                       double cellSizeNormalDirection2, 
                                                                       double wellRadius,
                                                                       double skinFactor, 
                                                                       double cDarcyForRelevantUnit)
{
    double K = cvf::Math::sqrt(permeabilityNormalDirection1 * permeabilityNormalDirection2);

    double nominator = cDarcyForRelevantUnit * 2 * cvf::PI_D * K * cellPerforationVectorComponent;
    
    double peaceManRad = peacemanRadius(permeabilityNormalDirection1, 
                                        permeabilityNormalDirection2, 
                                        cellSizeNormalDirection1, 
                                        cellSizeNormalDirection2);

    double denominator = log(peaceManRad / wellRadius) + skinFactor;

    double trans = nominator / denominator;
    return trans;
}

//--------------------------------------------------------------------------------------------------
/// 
//--------------------------------------------------------------------------------------------------
double RigTransmissibilityEquations::totalConnectionFactor(double transX, double transY, double transZ)
{
    return  cvf::Math::sqrt(
        pow(transX, 2.0) + pow(transY, 2.0) + pow(transZ, 2));
}