SelectStringField.cpp

/*---------------------------------------------------------------------------*\
 *
 *  mimmo
 *
 *  Copyright (C) 2015-2021 OPTIMAD engineering Srl
 *
 *  -------------------------------------------------------------------------
 *  License
 *  This file is part of mimmo.
 *
 *  mimmo is free software: you can redistribute it and/or modify it
 *  under the terms of the GNU Lesser General Public License v3 (LGPL)
 *  as published by the Free Software Foundation.
 *
 *  mimmo 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 Lesser General Public
 *  License for more details.
 *
 *  You should have received a copy of the GNU Lesser General Public License
 *  along with mimmo. If not, see <http://www.gnu.org/licenses/>.
 *
\*---------------------------------------------------------------------------*/
 
#include "SelectField.hpp"
#include "SkdTreeUtils.hpp"
#include "ExtractFields.hpp"
#include <unordered_map>
 
namespace mimmo{
 
SelectStringField::SelectStringField(MPVLocation loc):SelectField(){
    m_name = "mimmo.SelectStringField";
    m_loc = loc;
    if(m_loc == MPVLocation::UNDEFINED) m_loc= MPVLocation::POINT;
}
 
SelectStringField::SelectStringField(const bitpit::Config::Section & rootXML){
 
    std::string fallback_name = "ClassNONE";
    std::string fallback_loc  = "-1";
 
    std::string input_name = rootXML.get("ClassName", fallback_name);
    input_name = bitpit::utils::string::trim(input_name);
 
    std::string input_loc = rootXML.get("DataLocation", fallback_loc);
    input_loc = bitpit::utils::string::trim(input_loc);
 
    int loc = std::stoi(input_loc);
    if(loc > 0 && loc < 4){
        m_loc  =static_cast<MPVLocation>(loc);
    }else{
        m_loc = MPVLocation::POINT;
    }
 
    m_name = "mimmo.SelectStringField";
 
    if(input_name == "mimmo.SelectStringField"){
        absorbSectionXML(rootXML);
    }else{
        warningXML(m_log, m_name);
    };
}
 
SelectStringField::~SelectStringField(){}
 
SelectStringField::SelectStringField(const SelectStringField & other):SelectField(other){
    m_loc = other.m_loc;
    m_fields = other.m_fields;
    m_result = other.m_result;
}
 
void SelectStringField::swap(SelectStringField & x) noexcept
{
    std::swap(m_loc, x.m_loc);
    std::swap(m_fields, x.m_fields);
    //std::swap(m_result, x.m_result);
    m_result.swap(x.m_result);
    SelectField::swap(x);
}
 
void
SelectStringField::buildPorts(){
    bool built = true;
    built = (built && createPortIn<std::vector<MimmoPiercedVector<std::string>*>, SelectStringField>(this, &mimmo::SelectStringField::setFields, M_VECSTRINGFIELDS, true, 1));
    built = (built && createPortIn<MimmoPiercedVector<std::string>*, SelectStringField>(this, &mimmo::SelectStringField::addField, M_STRINGFIELD, true, 1));
    built = (built && createPortOut<MimmoPiercedVector<std::string>*, SelectStringField>(this, &mimmo::SelectStringField::getSelectedField, M_STRINGFIELD));
 
    SelectField::buildPorts();
    m_arePortsBuilt = built;
}
 
MimmoPiercedVector<std::string> *
SelectStringField::getSelectedField(){
    return  &m_result;
}
 
void
SelectStringField::setFields(std::vector<MimmoPiercedVector<std::string>*> fields){
    m_fields.clear();
    m_fields.reserve(fields.size());
    for(MimmoPiercedVector<std::string> * ff : fields){
        if(!ff) continue;
        if(ff->getDataLocation() == m_loc && ff->getGeometry()!= nullptr){
            m_fields.push_back(*ff);
        }
    }
    m_fields.shrink_to_fit();
}
 
void
SelectStringField::addField(MimmoPiercedVector<std::string> *field){
    if(!field) return;
    if(field->getDataLocation() == m_loc && field->getGeometry() != nullptr){
        m_fields.push_back(*field);
    }
 
}
 
void
SelectStringField::clear(){
    m_fields.clear();
    m_result.clear();
    SelectField::clear();
}
 
void
SelectStringField::plotOptionalResults(){
 
    // No data to write. String are not supported by bitpit vtk.
 
}
 
bool
SelectStringField::mSelect(){
 
    m_result.clear();
 
    switch(m_mode) {
 
    case SelectType::GEOMETRY:
    {
        //Check if geometry is linked
        if (getGeometry() == nullptr) return false;
 
        //Extract by link to geometry
        for (const auto & field : m_fields){
            if (field.getGeometry() == getGeometry()){
                m_result = field;
                //geometry, location and name are copied from field.
                return true;
            }
        }
        break;
    }
 
    case SelectType::NAME:
    {
        //Extract by link to geometry
        for (const auto & field : m_fields){
            if (field.getName() == getFieldName()){
                m_result = field;
                //geometry, location and name are copied from field.
                return true;
            }
        }
        break;
    }
 
    case SelectType::MAPPING:
    {
        //Check if geometry is linked
        if (getGeometry() == nullptr) return false;
 
        //Extract by geometric mapping if active and if no positive match is found.
        if (getGeometry()->getType() != 3){
 
            m_result.setGeometry(getGeometry());
            m_result.setDataLocation(m_loc);
 
            ExtractStringField * ef = new ExtractStringField();
            ef->setGeometry(getGeometry());
            ef->setMode(ExtractMode::MAPPING);
            ef->setTolerance(m_tol);
 
            //create map for overlapping ids purpose;
            std::unordered_map<long, int> idRepetition;
 
            for (MimmoPiercedVector<std::string> & field : m_fields){
                ef->setField(&field);
                bool check = ef->extract();
                if(!check) continue;
 
                MimmoPiercedVector<std::string> * temp = ef->getExtractedField();
                MimmoPiercedVector<std::string>::iterator itB;
                auto itE = temp->end();
                long id;
                for(itB = temp->begin(); itB != itE; ++itB){
                    id = itB.getId();
                    // For string values insert the first encountered
                    if(!m_result.exists(id)){
                        m_result.insert(id, *itB);
                    }
                }
            }
 
            //not resolve overlapping because I've chosen the first encountered value
 
            delete ef;
            return true;
        }
        break;
    }
 
    } // end switch mode
 
    //if you are here something was wrong.
    return false;
}
 
void SelectStringField::absorbSectionXML(const bitpit::Config::Section & slotXML, std::string name){
 
    BITPIT_UNUSED(name);
 
    if(slotXML.hasOption("DataLocation")){
        std::string input = slotXML.get("DataLocation");
        input = bitpit::utils::string::trim(input);
        int temp = -1;
        if(!input.empty()){
            std::stringstream ss(input);
            ss>>temp;
        }
        if(int(m_loc) != temp){
            (*m_log)<<"Error absorbing DataLocation in "<<m_name<<". Class and read locations mismatch"<<std::endl;
            throw std::runtime_error (m_name + " : xml absorbing failed");
        }
    }
 
    SelectField::absorbSectionXML(slotXML, name);
 
 
};
 
void SelectStringField::flushSectionXML(bitpit::Config::Section & slotXML, std::string name){
 
    BITPIT_UNUSED(name);
    slotXML.set("DataLocation", std::to_string(int(m_loc)));
    SelectField::flushSectionXML(slotXML, name);
};
 
 
}