Mesh.h

Go to the documentation of this file.

/*=============================================================================
TexGen: Geometric textile modeller.
Copyright (C) 2006 Martin Sherburn
 
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 2
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, write to the Free Software
Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
=============================================================================*/
 
#pragma once
//#include "../tinyxml/tinyxml.h"
#include "MeshData.h"
 
//class TiXmlElement;
namespace TexGen
{
    using namespace std;
 
 
    class CLASS_DECLSPEC CMesh
    {
    public:
        CMesh(void);
        CMesh(TiXmlElement &Element);
        ~CMesh(void);
 
        enum ELEMENT_TYPE
        {
            TRI,
            QUAD,
            TET,
            PYRAMID,
            WEDGE,
            HEX,
            LINE,
            POLYLINE,
            QUADRATIC_TET,
            POLYGON,
            NUM_ELEMENT_TYPES,      // Note this should always be directly after the last valid element
        };
 
        static inline int GetNumNodes(ELEMENT_TYPE Type)
        {
            switch (Type)
            {
                case TRI:
                    return 3;
                case QUAD:
                    return 4;
                case TET:
                    return 4;
                case PYRAMID:
                    return 5;
                case WEDGE:
                    return 6;
                case HEX:
                    return 8;
                case LINE:
                    return 2;
                case POLYLINE:
                    return 2;
                case QUADRATIC_TET:
                    return 10;
                case POLYGON:
                    return -1;  // Closed polygon: start node = end node
            }
            return 0;
        }
 
        void PopulateTiXmlElement(TiXmlElement &Element, OUTPUT_TYPE OutputType) const;
 
        // Methods to convert between element types //
 
        list<int>::iterator ConvertQuadtoTriangles(list<int>::iterator itQuad);
 
        void ConvertTrianglestoSegments();
 
        void ConvertQuadstoTriangles(bool bQuality = true);
 
        void ConvertHextoQuad();
 
        void ConvertWedgeto2D();
 
        void ConvertTettoTriangle();
 
        void ConvertHextoWedge(bool bQuality = true);
 
        void ConvertWedgetoTetandPyramid(bool bQuality = true);
 
        void ConvertPyramidtoTet(bool bQuality = true);
 
 
        void ConvertWedgetoTet(bool bQuality = true);
 
 
        void ConvertHextoTet(bool bQuality = true);
 
        void ConvertPyramidto2D();
 
        void Convert3Dto2D();
 
        void ConvertToSurfaceMesh();
 
        void ConvertToTetMesh();
 
        void ConvertToTriangleMesh();
 
        void ConvertToSegmentMesh();
 
 
        void ConvertTriToQuad( double Tolerance = 1e-6 );
 
        // Various utility methods //
 
        pair<XYZ, XYZ> GetAABB(double dGrowDistance = 0) const;
 
 
        int MergeNodes(const double Tolerance = 1e-8);
 
        int RemoveUnreferencedNodes();
 
        void RemoveAllElementsExcept(ELEMENT_TYPE Type);
 
        void RemoveElementType( ELEMENT_TYPE Type );
 
        int DeleteNodes(const set<int> &Nodes);
 
        void ChangeNodeIndices(int iChangeTo, int iChangeFrom);
 
 
        void ChangeNodeIndices(int iChangeTo, int iChangeFrom, vector<vector<int*> > &References);
 
        void RemoveOpposingTriangles();
 
        void RemoveOpposingQuads();
 
        void RemoveDegenerateTriangles();
 
        // Remove triangles which have the same indices
        void RemoveDuplicateTriangles();
 
        void RemoveDuplicateSegments();
 
 
        void RemoveDuplicateElements(CMesh::ELEMENT_TYPE ElementType);
 
 
        void GetNodeElementReferences(vector<vector<int*> > &References);
 
 
        int InsertNodes(const CMesh &Mesh, XYZ Offset = XYZ(0, 0, 0));
 
 
        void InsertMesh(const CMesh &Mesh, XYZ Offset = XYZ(0, 0, 0));
 
        void Rotate(WXYZ Rotation, XYZ Origin = XYZ(0,0,0));
 
        void Translate(XYZ Vector);
 
        void FlipNormals();
 
 
        void MeshClosedLoop(const XYZ &Normal, const vector<int> &ClosedLoopVector, bool bQuality = false);
 
 
        void MeshConvexHull();
 
        void Clear();
 
        void CopySelfToRange(XYZ Vector, int iLowerLimit, int iUpperLimit);
 
 
        int OutputNodes(ostream &Output, int iStartIndex = 1, string Seperator = ", ", bool bSCIRun = false) const;
 
 
        int OutputElements(ostream &Output, ELEMENT_TYPE ElementType, int iStartIndex = 1, int iIndexOffset = 1, string Seperator = ", ", bool bSCIRun = false) const;
 
 
        double CalculateVolume() const;
 
        vector<XYZ> GetElementCenters() const;
 
        vector<XYZ> GetElementCenters( ELEMENT_TYPE type );
 
 
        int CountInvertedElements() const;
 
 
        int CountInvertedElements(ELEMENT_TYPE ElementType) const;
 
 
        int IntersectLine(const XYZ &P1, const XYZ &P2, vector< pair<double, XYZ> > &IntersectionPoints, pair<bool, bool> TrimResults = make_pair(false, false), bool bForceFind = false/*, const CElementsOctree *pOctree = NULL*/) const;
 
 
        void BuildGrid(XYZ Min, XYZ Max, int iNumX, int iNumY, int iNumZ);
 
 
        void BuildGrid(XYZ Min, XYZ Max, double dPointsPerUnit);
 
 
        vector<pair<int, int> > GetNodePairs(XYZ Vector, const double Tolerance = 1e-6) const;
        void GetNodePairs(XYZ Vector, vector<pair<int, int> > &NodePairs, const double Tolerance = 1e-6) const;
 
 
        int GetClosestNode(XYZ Position) const;
 
 
        int GetClosestNodeDistance(XYZ Position, double dTol ) const;
 
 
        void ConvertElementListToVector( ELEMENT_TYPE ElementType, vector<int> &Indices );
 
        // Methods to output to various file formats //
 
 
        bool SaveToSTL(string Filename, bool bBinary = true) const;
 
 
        bool SaveToVTK(string Filename, const vector<CMeshDataBase*> *pMeshData = NULL) const;
 
 
        bool SaveToABAQUS(string Filename, const vector<POINT_INFO> *pElementInfo = NULL, bool bCreateStep = true, bool bCreateMaterial = true, int iElementType = 0);
 
        bool SaveToSMESH(string Filename) const;
 
 
        bool SaveToSCIRun(string Filename);
 
//      bool ReadFromTetGen(string NodesFile, string ElementsFile, string FacesFile = "");
 
        // Bunch of accessor methods to prevent direct access to underlying vector //
 
        bool AddElement(ELEMENT_TYPE Type, const vector<int> &Indices);
 
        int GetNumElements(ELEMENT_TYPE Type) const;
 
        int GetNumElements() const;
 
        vector<XYZ>::const_iterator NodesBegin() const;
        vector<XYZ>::const_iterator NodesEnd() const;
        vector<XYZ>::iterator NodesBegin();
        vector<XYZ>::iterator NodesEnd();
 
        const int AddNode(XYZ Node);
        void SetNode(int iIndex, XYZ Node);
        const XYZ& GetNode(int iIndex)const;
        vector<XYZ>::iterator DeleteNode(vector<XYZ>::iterator it);
        int GetNumNodes() const;
        const bool NodesEmpty() const;
        const vector<XYZ>& GetNodes() const;
        vector<XYZ>& GetNodes();
        void SetNumNodes(int NumNodes);
 
        const list<int>& GetIndices(ELEMENT_TYPE ElemType) const;
        list<int>& GetIndices(ELEMENT_TYPE ElemType);
 
 
    protected:
 
        static void AddOrCancel(list<pair<int, int> > &EdgeStack, pair<int, int> Edge);
 
        static void WriteBinaryXYZ(ostream &Output, XYZ Vector);
 
        int FillVTKPointData(TiXmlElement &Points) const;
        int FillVTKCellData(TiXmlElement &Cells) const;
 
        vector<XYZ> m_Nodes;
        list<int> m_Indices[NUM_ELEMENT_TYPES];
 
    };
 
};  // namespace TexGen