PabloUniform.hpp

/*---------------------------------------------------------------------------*\
 *
 *  bitpit
 *
 *  Copyright (C) 2015-2021 OPTIMAD engineering Srl
 *
 *  -------------------------------------------------------------------------
 *  License
 *  This file is part of bitpit.
 *
 *  bitpit 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.
 *
 *  bitpit 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 bitpit. If not, see <http://www.gnu.org/licenses/>.
 *
 \*---------------------------------------------------------------------------*/
 
#ifndef __BITPIT_PABLO_UNIFORM_HPP__
#define __BITPIT_PABLO_UNIFORM_HPP__
 
// =================================================================================== //
// INCLUDES                                                                            //
// =================================================================================== //
#include "ParaTree.hpp"
 
namespace bitpit {
 
    // =================================================================================== //
    // TYPEDEFS                                                                            //
    // =================================================================================== //
    typedef std::vector<bool>               bvector;
    typedef std::bitset<72>                 octantID;
    typedef std::vector<Octant*>            ptroctvector;
    typedef ptroctvector::iterator          octantIterator;
    typedef std::vector<darray3>            darray3vector;
 
 
    // =================================================================================== //
    // CLASS DEFINITION                                                                    //
    // =================================================================================== //
    class PabloUniform : public ParaTree
    {
        // =================================================================================== //
        // MEMBERS                                                                             //
        // =================================================================================== //
    private:
        darray3     m_origin;               
        double      m_L;                    
        double      m_area;                 
        double      m_volume;               
 
        // =================================================================================== //
        // CONSTRUCTORS AND OPERATORS
        // =================================================================================== //
 
        // =================================================================================== //
        // METHODS
        // =================================================================================== //
        void    __reset();
    public:
#if BITPIT_ENABLE_MPI==1
        PabloUniform(const std::string &logfile = DEFAULT_LOG_FILE, MPI_Comm comm = MPI_COMM_WORLD);
        PabloUniform(uint8_t dim, const std::string &logfile = DEFAULT_LOG_FILE, MPI_Comm comm = MPI_COMM_WORLD);
        PabloUniform(double X, double Y, double Z, double L, uint8_t dim = 2, const std::string &logfile = DEFAULT_LOG_FILE, MPI_Comm comm = MPI_COMM_WORLD);
#else
        PabloUniform(const std::string &logfile = DEFAULT_LOG_FILE);
        PabloUniform(uint8_t dim, const std::string &logfile = DEFAULT_LOG_FILE);
        PabloUniform(double X, double Y, double Z, double L, uint8_t dim = 2, const std::string &logfile = DEFAULT_LOG_FILE);
#endif
 
        // =================================================================================== //
        // METHODS
        // =================================================================================== //
        void    reset() override;
 
        int     getDumpVersion() const override;
        void    dump(std::ostream &stream, bool full = true) override;
        void    restore(std::istream &stream) override;
 
        // =================================================================================== //
        // BASIC GET/SET METHODS                                                               //
        // =================================================================================== //
        darray3     getOrigin() const;
        double      getX0() const;
        double      getY0() const;
        double      getZ0() const;
        double      getL() const;
        void        setL(double L);
        void        setOrigin(const darray3 &origin);
        double      levelToSize( uint8_t level) const;
 
        // =================================================================================== //
        // INDEX BASED METHODS                                                                 //
        // =================================================================================== //
        darray3     getCoordinates(uint32_t idx) const;
        double      getX(uint32_t idx) const;
        double      getY(uint32_t idx) const;
        double      getZ(uint32_t idx) const;
        double      getSize(uint32_t idx) const;
        double      getArea(uint32_t idx) const;
        double      getVolume(uint32_t idx) const;
        void        getCenter(uint32_t idx, darray3& center) const;
        darray3     getCenter(uint32_t idx) const;
        darray3     getFaceCenter(uint32_t idx, uint8_t iface) const;
        void        getFaceCenter(uint32_t idx, uint8_t iface, darray3& center) const;
        darray3     getNode(uint32_t idx, uint8_t inode) const;
        void        getNode(uint32_t idx, uint8_t inode, darray3& node) const;
        void        getNodes(uint32_t idx, darr3vector & nodes) const;
        darr3vector getNodes(uint32_t idx) const;
        void        getNormal(uint32_t idx, uint8_t iface, darray3 & normal) const;
        darray3     getNormal(uint32_t idx, uint8_t iface) const;
 
        // =================================================================================== //
        // POINTER BASED METHODS                                                               //
        // =================================================================================== //
        darray3     getCoordinates(const Octant* oct) const;
        double      getX(const Octant* oct) const;
        double      getY(const Octant* oct) const;
        double      getZ(const Octant* oct) const;
        double      getSize(const Octant* oct) const;
        double      getArea(const Octant* oct) const;
        double      getVolume(const Octant* oct) const;
        void        getCenter(const Octant* oct, darray3& center) const;
        darray3     getCenter(const Octant* oct) const;
        darray3     getFaceCenter(const Octant* oct, uint8_t iface) const;
        void        getFaceCenter(const Octant* oct, uint8_t iface, darray3& center) const;
        darray3     getNode(const Octant* oct, uint8_t inode) const;
        void        getNode(const Octant* oct, uint8_t inode, darray3& node) const;
        void        getNodes(const Octant* oct, darr3vector & nodes) const;
        darr3vector getNodes(const Octant* oct) const;
        void        getNormal(const Octant* oct, uint8_t iface, darray3 & normal) const;
        darray3     getNormal(const Octant* oct, uint8_t iface) const;
 
        // =================================================================================== //
        // LOCAL TREE GET/SET METHODS                                                          //
        // =================================================================================== //
        double      getLocalMaxSize() const;
        double      getLocalMinSize() const;
        void        getBoundingBox(darray3 & P0, darray3 & P1) const;
 
        // =================================================================================== //
        // INTERSECTION GET/SET METHODS                                                        //
        // =================================================================================== //
        double      getSize(const Intersection* inter) const;
        double      getArea(const Intersection* inter) const;
        darray3     getCenter(const Intersection* inter) const;
        darr3vector getNodes(const Intersection* inter) const;
        darray3     getNormal(const Intersection* inter) const;
 
        // =================================================================================== //
        // OTHER OCTANT BASED METHODS                                                          //
        // =================================================================================== //
        Octant* getPointOwner(darray3 point);
        uint32_t getPointOwnerIdx(darray3 point) const;
        Octant* getPointOwner(darray3 point, bool & isghost);
        uint32_t getPointOwnerIdx(darray3 point, bool & isghost) const;
        int getPointOwnerRank(darray3 point);
 
        // =================================================================================== //
        // OTHER PARATREE BASED METHODS                                                        //
        // =================================================================================== //
        darray3     getNodeCoordinates(uint32_t inode) const;
 
        // =================================================================================== //
        // TESTING OUTPUT METHODS                                                              //
        // =================================================================================== //
        void        write(const std::string &filename);
        void        writeTest(const std::string &filename, dvector data);
    };
 
}
 
#endif /* __BITPIT_PABLO_UNIFORM_HPP__ */