grid.hpp

Go to the documentation of this file.

/*
**    TP CPE Lyon
**    Copyright (C) 2013 Damien Rohmer
**
**    This program is free software: you can redistribute it and/or modify
**    it under the terms of the GNU General Public License as published by
**    the Free Software Foundation, either version 3 of the License, or
**    (at your option) any later version.
**
**   This program 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 General Public License for more details.
**
**    You should have received a copy of the GNU General Public License
**    along with this program.  If not, see <http://www.gnu.org/licenses/>.
*/


#ifndef GRID_HPP_
#define GRID_HPP_

#include <vec3.hpp>
#include <vector>

namespace cpe
{

class patch4;

class grid
{
    public:

        // ********************************************* //
        // ********************************************* //
        //  CONSTRUCTORS
        // ********************************************* //
        // ********************************************* //

        grid();
        grid(const unsigned int& Nu,const unsigned int& Nv);

        // ********************************************* //
        // ********************************************* //
        //  Size
        // ********************************************* //
        // ********************************************* //

        void resize(const unsigned int& Nu,const unsigned int& Nv);

        const unsigned int& size_u() const;
        const unsigned int& size_v() const;


        // ********************************************* //
        // ********************************************* //
        //  Data
        // ********************************************* //
        // ********************************************* //

        const vec3& operator()(const unsigned int& ku,const unsigned int& kv) const;
        vec3& operator()(const unsigned int& ku,const unsigned int& kv);


        // ********************************************* //
        // ********************************************* //
        //  4x4 patch
        // ********************************************* //
        // ********************************************* //

        patch4 get_patch(const unsigned int& k_patch_u,const unsigned int& k_patch_v) const;


        // ********************************************* //
        // ********************************************* //
        //  Special initialization
        // ********************************************* //
        // ********************************************* //

        void build_meshgrid();
        void build_sphere();


        // ********************************************* //
        // ********************************************* //
        //  Grid transformation
        // ********************************************* //
        // ********************************************* //

        grid& operator*=(const double& s);
        friend grid operator*(const double& s,const grid& g);
        friend grid operator*(const grid& g,const double& s);

        grid& operator+=(const vec3& t);
        grid& operator-=(const vec3& t);
        friend grid operator+(const grid& g,const vec3& t);
        friend grid operator+(const vec3& t,const grid& g);
        friend grid operator-(const grid& g,const vec3& t);
        friend grid operator-(const vec3& t,const grid& g);

        grid operator-() const;


        // ********************************************* //
        // ********************************************* //
        //  Topologie transform
        // ********************************************* //
        // ********************************************* //

        void add_back_u();
        void add_front_u();

        void add_back_v();
        void add_front_v();

        void suppress_back_u();
        void suppress_front_u();

        void suppress_back_v();
        void suppress_front_v();

        void duplicate_boundary();


        // ********************************************* //
        // ********************************************* //
        //  INPUT OUTPUT
        // ********************************************* //
        // ********************************************* //

        void write(const std::string& filename) const;
        static grid read(const std::string& filename);

    private:

        void assert_size(const unsigned int& ku,const unsigned int& kv) const;

        unsigned int internal_size_u;
        unsigned int internal_size_v;

        std::vector<vec3> internal_data;
};
}

#endif