/* 
 * File:   WellsGroup.cpp
 * Author: kjetilo
 * 
 * Created on March 27, 2012, 9:27 AM
 */

#include <opm/core/WellsGroup.hpp>

namespace Opm
{

    WellsGroupInterface::WellsGroupInterface(const std::string& name, ProductionSpecification prod_spec,
            InjectionSpecification inje_spec)
    : name_(name), production_specification_(prod_spec),
    injection_specification_(inje_spec)
    {
    }

    WellsGroupInterface::~WellsGroupInterface()
    {
    }

    const std::string& WellsGroupInterface::name()
    {
        return name_;
    }

    WellsGroup::WellsGroup(const std::string& name, ProductionSpecification prod_spec,
            InjectionSpecification inj_spec)
    : WellsGroupInterface(name, prod_spec, inj_spec)
    {
    }

    WellsGroupInterface* WellsGroup::findGroup(std::string name_of_node)
    {
        if (name() == name_of_node) {
            return this;
        } else {
            for (int i = 0; i < children_.size(); i++) {
                WellsGroupInterface* result = children_[i]->findGroup(name_of_node);
                if (result) {
                    return result;
                }
            }

            // Not found in this node.
            return NULL;
        }
    }
    
    void WellsGroup::addChild(std::tr1::shared_ptr<WellsGroupInterface> child) {
        children_.push_back(child);
    }

    WellNode::WellNode(const std::string& name, ProductionSpecification prod_spec,
            InjectionSpecification inj_spec)
    : WellsGroupInterface(name, prod_spec, inj_spec)
    {
    }

    WellsGroupInterface* WellNode::findGroup(std::string name_of_node)
    {
        if (name() == name_of_node) {
            return this;
        } else {
            return NULL;
        }

    }

    surface_component toSurfaceComponent(std::string type)
    {
        if (type == "OIL") {
            return OIL;
        }
        if (type == "WATER") {
            return WATER;
        }
        if (type == "GAS") {
            return GAS;
        }
        THROW("Unknown type " << type << ", could not convert to surface_component");
    }

    InjectionSpecification::ControlMode toInjectionControlMode(std::string type)
    {
        if (type == "NONE") {
            return InjectionSpecification::NONE;
        }

        if (type == "ORAT") {
            return InjectionSpecification::ORAT;
        }
        if (type == "REIN") {
            return InjectionSpecification::REIN;
        }
        if (type == "RESV") {
            return InjectionSpecification::RESV;
        }
        if (type == "VREP") {
            return InjectionSpecification::VREP;
        }
        if (type == "WGRA") {
            return InjectionSpecification::WGRA;
        }
        if (type == "FLD") {
            return InjectionSpecification::FLD;
        }
        if (type == "GRUP") {
            return InjectionSpecification::GRUP;
        }


        THROW("Unknown type " << type << ", could not convert to ControlMode.");
    }

    ProductionSpecification::ControlMode toProductionControlMode(std::string type)
    {
        if (type == "NONE") {
            return ProductionSpecification::NONE_CM;
        }
        if (type == "ORAT") {
                        return ProductionSpecification::ORAT;

        }
        if (type == "REIN") {
            return ProductionSpecification::REIN;
        }
        if (type == "RESV") {
            return ProductionSpecification::RESV;
        }
        if (type == "VREP") {
            return ProductionSpecification::VREP;
        }
        if (type == "WGRA") {
            return ProductionSpecification::WGRA;
        }
        if (type == "FLD") {
            return ProductionSpecification::FLD;
        }
        if (type == "GRUP") {
            return ProductionSpecification::GRUP;
        }

        THROW("Unknown type " << type << ", could not convert to ControlMode.");
    }
    
    
    ProductionSpecification::Procedure toProductionProcedure(std::string type)
    {
        if (type == "NONE") {
            return ProductionSpecification::NONE_P;
        }
        if (type == "RATE") {
            return ProductionSpecification::RATE;
        }
        if (type == "WELL") {
            return ProductionSpecification::WELL;
        }
        

        THROW("Unknown type " << type << ", could not convert to ControlMode.");
    }

    std::tr1::shared_ptr<WellsGroupInterface> createWellsGroup(std::string name, const EclipseGridParser& deck)
    {
        
        std::tr1::shared_ptr<WellsGroupInterface> return_value;
        // First we need to determine whether it's a group or just a well:
        bool isWell = false;
        if (deck.hasField("WELSPECS")) {
            WELSPECS wspecs = deck.getWELSPECS();
            for (int i = 0; i < wspecs.welspecs.size(); i++) {
                if (wspecs.welspecs[i].name_ == name) {
                    isWell = true;
                    break;
                }
            }
        }
        // For now, assume that if it isn't a well, it's a group

        if (isWell) {

            InjectionSpecification injection_specification;
            if (deck.hasField("WCONINJE")) {
                WCONINJE wconinje = deck.getWCONINJE();
                for (int i = 0; i < wconinje.wconinje.size(); i++) {
                    if (wconinje.wconinje[i].well_ == name) {
                        WconinjeLine line = wconinje.wconinje[i];
                        injection_specification.BHP_limit_ = line.BHP_limit_;
                        injection_specification.injector_type_ = toSurfaceComponent(line.injector_type_);
                        injection_specification.control_mode_ = toInjectionControlMode(line.control_mode_);
                        injection_specification.surface_flow_max_rate_ = line.surface_flow_max_rate_;
                    }
                }
            }

            ProductionSpecification production_specification;
            if (deck.hasField("WCONPROD")) {
                WCONPROD wconprod = deck.getWCONPROD();
                for (int i = 0; i < wconprod.wconprod.size(); i++) {
                    if (wconprod.wconprod[i].well_ == name) {
                        WconprodLine line = wconprod.wconprod[i];
                        production_specification.BHP_limit_ = line.BHP_limit_;
                        production_specification.fluid_volume_max_rate_ = line.fluid_volume_max_rate_;
                        production_specification.oil_max_rate_ = line.oil_max_rate_;
                        production_specification.control_mode_ = toProductionControlMode(line.control_mode_);
                        production_specification.water_production_target_ = line.water_max_rate_;
                    }
                }
            }
            
            return_value.reset(new WellNode(name, production_specification, injection_specification));
        } else {
            InjectionSpecification injection_specification;
            if (deck.hasField("GCONINJE")) {
                GCONINJE gconinje = deck.getGCONINJE();
                for (int i = 0; i < gconinje.gconinje.size(); i++) {
                    if (gconinje.gconinje[i].group_ == name) {
                        GconinjeLine line = gconinje.gconinje[i];
                        injection_specification.injector_type_ = toSurfaceComponent(line.injector_type_);
                        injection_specification.control_mode_ = toInjectionControlMode(line.control_mode_);
                        injection_specification.surface_flow_max_rate_ = line.surface_flow_max_rate_;
                    }
                }
            }

            ProductionSpecification production_specification;
            if (deck.hasField("GCONPROD")) {
                GCONPROD gconprod = deck.getGCONPROD();
                for (int i = 0; i < gconprod.gconprod.size(); i++) {
                    if (gconprod.gconprod[i].group_ == name) {
                        GconprodLine line = gconprod.gconprod[i];
                        production_specification.oil_max_rate_ = line.oil_max_rate_;
                        production_specification.control_mode_ = toProductionControlMode(line.control_mode_);
                        production_specification.water_production_target_ = line.water_max_rate_;
                        production_specification.gas_max_rate_ = line.gas_max_rate_;
                        production_specification.liquid_max_rate_ = line.liquid_max_rate_;
                        production_specification.procedure_ = toProductionProcedure(line.procedure_);
                    }
                }
            }
            
            return_value.reset(new WellsGroup(name, production_specification, injection_specification));
        }
        
        return return_value;
        
    }
}