ScaleGeometry.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 "ScaleGeometry.hpp"
 
namespace mimmo{
 
 
ScaleGeometry::ScaleGeometry(darray3E scaling){
    m_scaling = scaling;
    m_origin.fill(0.0);
    m_meanP = false;
    m_name = "mimmo.ScaleGeometry";
};
 
ScaleGeometry::ScaleGeometry(const bitpit::Config::Section & rootXML){
 
    m_scaling.fill(1.0);
    m_origin.fill(0.0);
    m_meanP = false;
    m_name = "mimmo.ScaleGeometry";
 
    std::string fallback_name = "ClassNONE";
    std::string input = rootXML.get("ClassName", fallback_name);
    input = bitpit::utils::string::trim(input);
    if(input == "mimmo.ScaleGeometry"){
        absorbSectionXML(rootXML);
    }else{
        warningXML(m_log, m_name);
    };
}
 
ScaleGeometry::~ScaleGeometry(){};
 
ScaleGeometry::ScaleGeometry(const ScaleGeometry & other):BaseManipulation(other){
    m_scaling = other.m_scaling;
    m_origin = other.m_origin;
    m_meanP = other.m_meanP;
    m_filter = other.m_filter;
};
 
ScaleGeometry & ScaleGeometry::operator=(ScaleGeometry other){
    swap(other);
    return *this;
}
 
void ScaleGeometry::swap(ScaleGeometry & x) noexcept
{
    std::swap(m_scaling, x.m_scaling);
    std::swap(m_origin, x.m_origin);
    std::swap(m_meanP, x.m_meanP);
    m_filter.swap(x.m_filter);
    m_displ.swap(x.m_displ);
    BaseManipulation::swap(x);
}
 
 
void
ScaleGeometry::buildPorts(){
    bool built = true;
    built = (built && createPortIn<darray3E, ScaleGeometry>(&m_origin, M_POINT));
    built = (built && createPortIn<darray3E, ScaleGeometry>(&m_scaling, M_SPAN));
    built = (built && createPortIn<dmpvector1D*, ScaleGeometry>(this, &mimmo::ScaleGeometry::setFilter, M_FILTER));
    built = (built && createPortIn<MimmoSharedPointer<MimmoObject>, ScaleGeometry>(&m_geometry, M_GEOM, true));
    built = (built && createPortOut<dmpvecarr3E*, ScaleGeometry>(this, &mimmo::ScaleGeometry::getDisplacements, M_GDISPLS));
    built = (built && createPortOut<MimmoSharedPointer<MimmoObject>, ScaleGeometry>(this, &BaseManipulation::getGeometry, M_GEOM));
    m_arePortsBuilt = built;
};
 
void
ScaleGeometry::setOrigin(darray3E origin){
    m_origin = origin;
}
 
void
ScaleGeometry::setMeanPoint(bool meanP){
    m_meanP = meanP;
}
 
void
ScaleGeometry::setScaling(darray3E scaling){
    m_scaling = scaling;
}
 
void
ScaleGeometry::setFilter(dmpvector1D *filter){
    if(!filter) return;
    m_filter = *filter;
}
 
dmpvecarr3E*
ScaleGeometry::getDisplacements(){
    return &m_displ;
};
 
void
ScaleGeometry::execute(){
 
    if(getGeometry() == nullptr){
        (*m_log)<<m_name + " : nullptr pointer to linked geometry found"<<std::endl;
        throw std::runtime_error(m_name + "nullptr pointer to linked geometry found");
    }
 
    checkFilter();
    m_displ.clear();
    m_displ.setDataLocation(mimmo::MPVLocation::POINT);
    m_displ.reserve(getGeometry()->getNVertices());
    m_displ.setGeometry(getGeometry());
 
 
    long ID;
    darray3E value;
    //computing centroid
    int nV = m_geometry->getNVertices();
    darray3E center = m_origin;
    if (m_meanP){
        center.fill(0.0);
        for (const auto & vertex : m_geometry->getVertices()){
            center += vertex.getCoords();
        }
        center /=double(nV);
    }
    for (const auto & vertex : m_geometry->getVertices()){
        ID = vertex.getId();
 
        darray3E coords = vertex.getCoords();
        value = (( m_scaling*(coords - center) + center ) - coords) * m_filter[ID] ;
        m_displ.insert(ID, value);
    }
};
 
void
ScaleGeometry::apply(){
    _apply(m_displ);
}
 
void
ScaleGeometry::checkFilter(){
    bool check = m_filter.getDataLocation() == mimmo::MPVLocation::POINT;
    check = check && m_filter.completeMissingData(0.0);
    check = check && m_filter.getGeometry() == getGeometry();
 
    if (!check){
        m_log->setPriority(bitpit::log::Verbosity::DEBUG);
        (*m_log)<<"Not valid filter found in "<<m_name<<". Proceeding with default unitary field"<<std::endl;
        m_log->setPriority(bitpit::log::Verbosity::NORMAL);
 
        m_filter.clear();
        m_filter.setGeometry(m_geometry);
        m_filter.setDataLocation(mimmo::MPVLocation::POINT);
        m_filter.reserve(getGeometry()->getNVertices());
        for (const auto & vertex : getGeometry()->getVertices()){
            m_filter.insert(vertex.getId(), 1.0);
        }
    }
}
 
void
ScaleGeometry::absorbSectionXML(const bitpit::Config::Section & slotXML, std::string name){
 
    BITPIT_UNUSED(name);
 
    BaseManipulation::absorbSectionXML(slotXML, name);
 
    if(slotXML.hasOption("MeanPoint")){
        std::string input = slotXML.get("MeanPoint");
        bool value = false;
        if(!input.empty()){
            std::stringstream ss(bitpit::utils::string::trim(input));
            ss >> value;
        }
        setMeanPoint(value);
    };
 
 
    if(slotXML.hasOption("Origin")){
        std::string input = slotXML.get("Origin");
        input = bitpit::utils::string::trim(input);
        darray3E temp = {{0.0,0.0,0.0}};
        if(!input.empty()){
            std::stringstream ss(input);
            for(auto &val : temp) ss>>val;
        }
        setOrigin(temp);
    }
 
    if(slotXML.hasOption("Scaling")){
        std::string input = slotXML.get("Scaling");
        input = bitpit::utils::string::trim(input);
        darray3E temp = {{0.0,0.0,0.0}};
        if(!input.empty()){
            std::stringstream ss(input);
            for(auto &val : temp) ss>>val;
        }
        setScaling(temp);
    }
 
};
 
void
ScaleGeometry::flushSectionXML(bitpit::Config::Section & slotXML, std::string name){
 
    BITPIT_UNUSED(name);
 
    BaseManipulation::flushSectionXML(slotXML, name);
    {
        std::stringstream ss;
        ss<<std::scientific<<m_origin[0]<<'\t'<<m_origin[1]<<'\t'<<m_origin[2];
        slotXML.set("Origin", ss.str());
    }
 
    {
        bool value = m_meanP;
        std::string towrite = std::to_string(value);
        slotXML.set("MeanPoint", towrite);
    }
 
    {
        std::stringstream ss;
        ss<<std::scientific<<m_scaling[0]<<'\t'<<m_scaling[1]<<'\t'<<m_scaling[2];
        slotXML.set("Scaling", ss.str());
    }
 
 
};
 
}