manipulators_example_00004.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 "mimmo_manipulators.hpp"
#include "mimmo_iogeneric.hpp"
#include <bitpit_common.hpp>
#include <random>
 
// =================================================================================== //
void test00004() {
 
    /*
        Read a sphere from STL file. Convert mode is to save the just read geometry in
        another file with name manipulators_output_00004.0000.stl
     */
    mimmo::MimmoGeometry * mimmo0 = new mimmo::MimmoGeometry(mimmo::MimmoGeometry::IOMode::CONVERT);
    mimmo0->setReadDir("geodata");
    mimmo0->setReadFileType(FileType::STL);
    mimmo0->setReadFilename("sphere2");
    mimmo0->setWriteDir(".");
    mimmo0->setWriteFileType(FileType::STL);
    mimmo0->setWriteFilename("manipulators_output_00004.0000");
 
    /*
        write the deformed geometry to file
    */
    mimmo::MimmoGeometry * mimmo1 = new mimmo::MimmoGeometry(mimmo::MimmoGeometry::IOMode::WRITE);
    mimmo1->setWriteDir(".");
    mimmo1->setWriteFileType(FileType::STL);
    mimmo1->setWriteFilename("manipulators_output_00004.0001");
 
    /*
        Instantiation of a FFDobject with spherical shape.
        Setup of sphere dimensions and number of lattice nodes/nurbs degrees
        for each spherical coordinate.
     */
    mimmo::FFDLattice* lattice = new mimmo::FFDLattice();
    darray3E origin = {0.0, 0.0,0.0};
    darray3E span;
    span[0]= 3.01;
    span[1]= 2*BITPIT_PI;
    span[2]= BITPIT_PI;
 
    /* Set number of nodes of the mesh (dim) and degree of nurbs functions (deg).
     */
    iarray3E dim, deg;
    dim[0] = 30;
    dim[1] = 30;
    dim[2] = 30;
    deg[0] = 2;
    deg[1] = 2;
    deg[2] = 2;
 
    lattice->setLattice(origin,span,mimmo::ShapeType::SPHERE,dim, deg);
 
    /*
        Change reference system to work in local spherical coordinates.
        Set coordinates as CLAMPED (continuity in origins of angles).
     */
    lattice->setRefSystem(2, darray3E{0,1,0});
    lattice->setCoordType(mimmo::CoordType::CLAMPED, 2);
    lattice->setPlotInExecution(true);
 
    /*
        Build mesh of lattice outside the execution chain
        to use it during setup the displacements.
     */
    lattice->build();
 
    /* Creation of displacements for control nodes of the lattice.
     * Use random values to set the displacements of the control nodes at a longitude
     * angle smaller than PI and expansion on radius direction for nodes with longitude
     * angle greater than PI.
     * As exercise GenericInput/Output blocks are used to write them down to file
     */
    int ndeg = lattice->getNNodes();
    dvecarr3E displ(ndeg, darray3E{0,0,0});
    std::minstd_rand rgen;
    rgen.seed(16);
    double distRand = (rgen.max()-rgen.min());
    for (int i=0; i<ndeg; i++){
        int l1,l2,l3;
        int index = lattice->accessGridFromDOF(i);
        lattice->accessPointIndex(index,l1,l2,l3);
        if(l1 > 0 && lattice->getLocalPoint(l1,l2,l3)[1] < BITPIT_PI){
            displ[i][0] = 1.0*( double( rgen() - rgen.min() ) / distRand );
        }
        if( (l1 > 0 && lattice->getLocalPoint(l1,l2,l3)[1] >= BITPIT_PI)
                || lattice->getLocalPoint(l1,l2,l3)[1] == 0){
            displ[i][0] = 1.25;
        }
 
    }
 
    /*
       Set Generic input block with the
       displacements defined above.
     */
    mimmo::GenericInput* input = new mimmo::GenericInput();
    input->setInput(displ);
 
    /*
        Set Generic output block to write the
        displacements defined above.
     */
    mimmo::GenericOutput * output = new mimmo::GenericOutput();
    output->setFilename("manipulators_output_00004.csv");
    output->setCSV(true);
 
    /*
        Create applier block.
        It applies the deformation displacements to the original input geometry.
     */
    mimmo::Apply* applier = new mimmo::Apply();
 
    /*
        Setup pin connections.
     */
    mimmo::pin::addPin(mimmo0, lattice, M_GEOM, M_GEOM);
    mimmo::pin::addPin(mimmo0, applier, M_GEOM, M_GEOM);
    mimmo::pin::addPin(input, lattice, M_DISPLS, M_DISPLS);
    mimmo::pin::addPin(input, output, M_DISPLS, M_DISPLS);
    mimmo::pin::addPin(lattice, applier, M_GDISPLS, M_GDISPLS);
    mimmo::pin::addPin(applier, mimmo1, M_GEOM, M_GEOM);
 
    /*
        Setup execution chain.
     */
    mimmo::Chain ch0;
    ch0.addObject(input);
    ch0.addObject(output);
    ch0.addObject(applier);
    ch0.addObject(lattice);
    ch0.addObject(mimmo1);
    ch0.addObject(mimmo0);
 
    /* Execute the chain.
     * Use debug flag false to avoid printing intermediate results of the execution steps.
     */
    ch0.exec(false);
 
    /*
        Clean up & exit;
     */
    delete lattice;
    delete applier;
    delete input;
    delete output;
    delete mimmo0;
    delete mimmo1;
 
    return;
}
 
int main( int argc, char *argv[] ) {
 
    BITPIT_UNUSED(argc);
    BITPIT_UNUSED(argv);
 
#if MIMMO_ENABLE_MPI
    MPI_Init(&argc, &argv);
#endif
        try{
            test00004() ;
        }
        catch(std::exception & e){
            std::cout<<"manipulators_example_00004 exited with an error of type : "<<e.what()<<std::endl;
            return 1;
        }
#if MIMMO_ENABLE_MPI
    MPI_Finalize();
#endif
 
    return  0;
}