StdMeshers_Prism_3D.hxx

Go to the documentation of this file.

//  Copyright (C) 2007-2008  CEA/DEN, EDF R&D, OPEN CASCADE
//
//  Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
//  CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
//
//  This library is free software; you can redistribute it and/or
//  modify it under the terms of the GNU Lesser General Public
//  License as published by the Free Software Foundation; either
//  version 2.1 of the License.
//
//  This library 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 this library; if not, write to the Free Software
//  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
//
//  See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
//
//  SMESH SMESH : implementaion of SMESH idl descriptions
//  File   : StdMeshers_Prism_3D.hxx
//  Module : SMESH
//
#ifndef _SMESH_Prism_3D_HXX_
#define _SMESH_Prism_3D_HXX_
 
#include "SMESH_StdMeshers.hxx"
 
#include "SMESH_3D_Algo.hxx"
#include "SMDS_TypeOfPosition.hxx"
#include "SMDS_MeshNode.hxx"
#include "SMESH_Block.hxx"
#include "SMESH_Mesh.hxx"
#include "SMESHDS_Mesh.hxx"
#include "SMESH_subMesh.hxx"
#include "SMESH_MesherHelper.hxx"
#include "SMESH_Comment.hxx"
 
#include <vector>
 
#include <Adaptor3d_Curve.hxx>
#include <Adaptor3d_Surface.hxx>
#include <Adaptor2d_Curve2d.hxx>
#include <BRepAdaptor_Surface.hxx>
#include <TopTools_IndexedMapOfOrientedShape.hxx>
#include <gp_XYZ.hxx>
 
 
class SMESHDS_SubMesh;
class TopoDS_Edge;
class TopoDS_Faces;
struct TNode;
 
//typedef std::map<const SMDS_MeshNode*, const SMDS_MeshNode*> TNodeNodeMap;
typedef std::vector<const SMDS_MeshNode* > TNodeColumn;
 
// map of bottom nodes to the column of nodes above them
// (the column includes the bottom nodes)
typedef std::map< TNode, TNodeColumn > TNode2ColumnMap;
typedef std::map< double, TNodeColumn > TParam2ColumnMap;
typedef std::map< double, TNodeColumn >::const_iterator TParam2ColumnIt;
 
typedef TopTools_IndexedMapOfOrientedShape TBlockShapes;
 
// ===============================================
// ===============================================
 
struct TNode
{
  const SMDS_MeshNode* myNode;
  gp_XYZ               myParams;
 
  gp_XYZ GetCoords() const { return gp_XYZ( myNode->X(), myNode->Y(), myNode->Z() ); }
  gp_XYZ GetParams() const { return myParams; }
  gp_XYZ& ChangeParams() { return myParams; }
  bool HasParams() const { return myParams.X() >= 0.0; }
  SMDS_TypeOfPosition GetPositionType() const
  { return myNode ? myNode->GetPosition()->GetTypeOfPosition() : SMDS_TOP_UNSPEC; }
  bool IsNeighbor( const TNode& other ) const;
 
  TNode(const SMDS_MeshNode* node = 0): myNode(node), myParams(-1,-1,-1) {}
  bool operator < (const TNode& other) const { return myNode->GetID() < other.myNode->GetID(); }
};
 
// ===============================================================
// ===============================================================
 
class STDMESHERS_EXPORT StdMeshers_PrismAsBlock: public SMESH_Block
{
public:
  StdMeshers_PrismAsBlock();
 
  ~StdMeshers_PrismAsBlock();
 
  bool Init(SMESH_MesherHelper* helper, const TopoDS_Shape& shape3D);
 
  SMESH_ComputeErrorPtr GetError() const { return myError; }
 
  int VerticalSize() const { return myParam2ColumnMaps[0].begin()->second.size(); }
 
  bool HasNotQuadElemOnTop() const { return myNotQuadOnTop; }
 
  const TNodeColumn* GetNodeColumn(const SMDS_MeshNode* node) const;
 
  const TParam2ColumnMap& GetParam2ColumnMap(const int baseEdgeID,
                                             bool &    isReverse)
  {
    std::pair< TParam2ColumnMap*, bool > & col_frw =
      myShapeIndex2ColumnMap[ baseEdgeID ];
    isReverse = !col_frw.second;
    return * col_frw.first;
  }
  
  SMESH_Mesh* Mesh() const { return myHelper->GetMesh(); }
 
  SMESHDS_Mesh* MeshDS() const { return Mesh()->GetMeshDS(); }
 
  SMESH_subMesh* SubMesh(const int shapeID) const
  { return Mesh()->GetSubMesh( Shape( shapeID )); }
 
  SMESHDS_SubMesh* SubMeshDS(const int shapeID) const
  { return SubMesh(shapeID)->GetSubMeshDS(); }
 
  const TopoDS_Shape& Shape(const int shapeID) const
  { return myShapeIDMap( shapeID ); }
 
  int ShapeID(const TopoDS_Shape& shape) const
  { return myShapeIDMap.FindIndex( shape ); }
 
  static bool IsForwardEdge(SMESHDS_Mesh*           meshDS,
                            const TParam2ColumnMap& columnsMap,
                            const TopoDS_Edge &     bottomEdge,
                            const int               sideFaceID);
  static bool GetWallFaces( SMESH_Mesh*                     mesh,
                            const TopoDS_Shape &            mainShape,
                            const TopoDS_Shape &            bottomFace,
                            const std::list< TopoDS_Edge >& bottomEdges,
                            std::list< TopoDS_Face >&       wallFaces);
 
private:
 
  // --------------------------------------------------------------------
  // --------------------------------------------------------------------
  class TSideFace: public Adaptor3d_Surface
  {
    int                             myID; 
    // map used to find out real UV by it's normalized UV
    TParam2ColumnMap*               myParamToColumnMap;
    BRepAdaptor_Surface             mySurface;
    TopoDS_Edge                     myBaseEdge;
    // first and last normalized params and orientaion for each component or it-self
    std::vector< std::pair< double, double> > myParams;
    bool                            myIsForward;
    std::vector< TSideFace* >       myComponents;
    SMESH_MesherHelper *            myHelper;
  public:
    TSideFace( SMESH_MesherHelper* helper,
               const int           faceID,
               const TopoDS_Face&  face,
               const TopoDS_Edge&  baseEdge,
               TParam2ColumnMap*   columnsMap,
               const double        first = 0.0,
               const double        last = 1.0);
    TSideFace( const std::vector< TSideFace* >&             components,
               const std::vector< std::pair< double, double> > & params);
    TSideFace( const TSideFace& other );
    ~TSideFace();
    bool IsComplex() const
    { return ( NbComponents() > 0 || myParams[0].first != 0. || myParams[0].second != 1. ); }
    int FaceID() const { return myID; }
    TParam2ColumnMap* GetColumns() const { return myParamToColumnMap; }
    gp_XY GetNodeUV(const TopoDS_Face& F, const SMDS_MeshNode* n) const
    { return myHelper->GetNodeUV( F, n ); }
    const TopoDS_Edge & BaseEdge() const { return myBaseEdge; }
    int ColumnHeight() const {
      if ( NbComponents() ) return GetComponent(0)->GetColumns()->begin()->second.size();
      else                  return GetColumns()->begin()->second.size(); }
    double GetColumns(const double U, TParam2ColumnIt & col1, TParam2ColumnIt& col2 ) const;
    int NbComponents() const { return myComponents.size(); }
    TSideFace* GetComponent(const int i) const { return myComponents.at( i ); }
    void SetComponent(const int i, TSideFace* c)
    { if ( myComponents[i] ) delete myComponents[i]; myComponents[i]=c; }
    TSideFace* GetComponent(const double U, double& localU) const;
    bool IsForward() const { return myIsForward; }
    // boundary geometry for a face
    Adaptor3d_Surface* Surface() const { return new TSideFace( *this ); }
    bool GetPCurves(Adaptor2d_Curve2d* pcurv[4]) const;
    Adaptor2d_Curve2d* HorizPCurve(const bool isTop, const TopoDS_Face& horFace) const;
    Adaptor3d_Curve* HorizCurve(const bool isTop) const;
    Adaptor3d_Curve* VertiCurve(const bool isMax) const;
    TopoDS_Edge GetEdge( const int edge ) const;
    int InsertSubShapes( TBlockShapes& shapeMap ) const;
    // redefine Adaptor methods
    gp_Pnt Value(const Standard_Real U,const Standard_Real V) const;
  };
 
  // --------------------------------------------------------------------
  // --------------------------------------------------------------------
  class STDMESHERS_EXPORT TVerticalEdgeAdaptor: public Adaptor3d_Curve
  {
    const TNodeColumn* myNodeColumn;
  public:
    TVerticalEdgeAdaptor( const TParam2ColumnMap* columnsMap, const double parameter );
    gp_Pnt Value(const Standard_Real U) const;
    Standard_Real FirstParameter() const { return 0; }
    Standard_Real LastParameter() const { return 1; }
  };
 
  // --------------------------------------------------------------------
  // --------------------------------------------------------------------
  class STDMESHERS_EXPORT THorizontalEdgeAdaptor: public Adaptor3d_Curve
  {
    const TSideFace* mySide;
    double           myV;
  public:
    THorizontalEdgeAdaptor( const TSideFace* sideFace, const bool isTop)
      :mySide(sideFace), myV( isTop ? 1.0 : 0.0 ) {}
    gp_Pnt Value(const Standard_Real U) const;
    Standard_Real FirstParameter() const { return 0; }
    Standard_Real LastParameter() const { return 1; }
  };
 
  // --------------------------------------------------------------------
  // --------------------------------------------------------------------
  class STDMESHERS_EXPORT TPCurveOnHorFaceAdaptor: public Adaptor2d_Curve2d
  {
    const TSideFace*  mySide;
    int               myZ;
    TopoDS_Face       myFace;
  public:
    TPCurveOnHorFaceAdaptor( const TSideFace*   sideFace,
                             const bool         isTop,
                             const TopoDS_Face& horFace)
      : mySide(sideFace), myFace(horFace), myZ(isTop ? mySide->ColumnHeight() - 1 : 0 ) {}
    gp_Pnt2d Value(const Standard_Real U) const;
    Standard_Real FirstParameter() const { return 0; }
    Standard_Real LastParameter() const { return 1; }
  };
  // --------------------------------------------------------------------
 
  bool myNotQuadOnTop;
  SMESH_MesherHelper* myHelper;
  TBlockShapes myShapeIDMap;
 
  // container of 4 side faces
  TSideFace*                 mySide; 
  // node columns for each base edge
  std::vector< TParam2ColumnMap > myParam2ColumnMaps;
  // to find a column for a node by edge SMESHDS Index
  std::map< int, std::pair< TParam2ColumnMap*, bool > > myShapeIndex2ColumnMap;
 
  SMESH_ComputeErrorPtr myError;
  bool error(int error, const SMESH_Comment& comment = "") {
    myError = SMESH_ComputeError::New(error,comment);
    return myError->IsOK();
  }
  //std::vector< SMESH_subMesh* >           mySubMeshesVec; // submesh by in-block id
};
 
// =============================================
// =============================================
 
class STDMESHERS_EXPORT StdMeshers_Prism_3D: public SMESH_3D_Algo
{
public:
  StdMeshers_Prism_3D(int hypId, int studyId, SMESH_Gen* gen);
  virtual ~StdMeshers_Prism_3D();
 
  virtual bool CheckHypothesis(SMESH_Mesh&                          aMesh,
                               const TopoDS_Shape&                  aShape,
                               SMESH_Hypothesis::Hypothesis_Status& aStatus);
 
  virtual bool Compute(SMESH_Mesh& aMesh, const TopoDS_Shape& aShape);
 
  virtual bool Evaluate(SMESH_Mesh & aMesh, const TopoDS_Shape & aShape,
                        MapShapeNbElems& aResMap);
 
  void ProjectTriangles() { myProjectTriangles = true; }
 
  static void AddPrisms( std::vector<const TNodeColumn*> & nodeColumns,
                         SMESH_MesherHelper*          helper);
 
private:
 
  bool assocOrProjBottom2Top();
 
  bool projectBottomToTop();
 
  bool setFaceAndEdgesXYZ( const int faceID, const gp_XYZ& params, int z );
 
private:
 
  bool myProjectTriangles;
 
  StdMeshers_PrismAsBlock myBlock;
  SMESH_MesherHelper*     myHelper;
 
  std::vector<gp_XYZ>     myShapeXYZ; // point on each sub-shape
 
  // map of bottom nodes to the column of nodes above them
  // (the column includes the bottom node)
  TNode2ColumnMap         myBotToColumnMap;
};
 
#endif