smesh/src/SMESH/SMESH_ProxyMesh.hxx

// Copyright (C) 2007-2012  CEA/DEN, EDF R&D, OPEN CASCADE
//
// 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
//

// File      : SMESH_ProxyMesh.hxx
// Created   : Thu Dec  2 10:05:35 2010
// Author    : Edward AGAPOV (eap)

#ifndef __SMESH_ProxyMesh_HXX__
#define __SMESH_ProxyMesh_HXX__

#include "SMESH_SMESH.hxx"

#include "SMDS_MeshElement.hxx"
#include "SMESHDS_SubMesh.hxx"

#include <TopoDS_Shape.hxx>

#include <map>
#include <vector>
#include <boost/shared_ptr.hpp>

class SMDS_MeshNode;
class SMESHDS_Mesh;
class SMESH_Mesh;

/*!
 * \brief Container of mesh faces substituting other faces in the input mesh of 3D algorithm
 */
class SMESH_EXPORT SMESH_ProxyMesh
{
public:

  typedef boost::shared_ptr<SMESH_ProxyMesh> Ptr;

  typedef std::map<const SMDS_MeshNode*, const SMDS_MeshNode*, TIDCompare > TN2NMap;

  //--------------------------------------------------------------------------------
  /*!
   * \brief Proxy sub-mesh
   */
  class SubMesh : public SMESHDS_SubMesh
  {
  public:

    const TN2NMap* GetNodeNodeMap() const { return _n2n; }
    const SMDS_MeshNode* GetProxyNode( const SMDS_MeshNode* n ) const;
    virtual void AddElement(const SMDS_MeshElement * e);
    virtual int NbElements() const;
    virtual SMDS_ElemIteratorPtr GetElements() const;
    virtual void Clear();
    virtual bool Contains(const SMDS_MeshElement * ME) const;

    template< class ITERATOR >
    void ChangeElements( ITERATOR it, ITERATOR end )
    {
      // change SubMesh contents without deleting tmp faces
      // for which the caller is responsible
      _elements.clear();
      while ( it != end ) _elements.push_back( *it++ );
    }
    SubMesh(int index=0):SMESHDS_SubMesh(0,index),_n2n(0) {}
    ~SubMesh() { Clear(); }

  private:
    std::vector<const SMDS_MeshElement *> _elements;
    TN2NMap*                              _n2n;
    friend class SMESH_ProxyMesh;
  };
  //--------------------------------------------------------------------------------
  // Public interface

  SMESH_ProxyMesh();
  SMESH_ProxyMesh(std::vector<SMESH_ProxyMesh::Ptr>& components);
  SMESH_ProxyMesh(const SMESH_Mesh& mesh) { _mesh = &mesh; }
  virtual ~SMESH_ProxyMesh();

  // Returns the submesh of a face; it can be a proxy sub-mesh
  const SMESHDS_SubMesh* GetSubMesh(const TopoDS_Shape& face) const;

  // Returns the proxy sub-mesh of a face; it can be NULL
  const SubMesh* GetProxySubMesh(const TopoDS_Shape& face) const;

  // Returns the proxy node of a node; the input node is returned if no proxy exists
  const SMDS_MeshNode* GetProxyNode( const SMDS_MeshNode* node ) const;

  // Returns iterator on all faces of the mesh taking into account substitutions
  // To be used in case of mesh without shape
  SMDS_ElemIteratorPtr GetFaces() const;

  // Returns iterator on all faces on the face taking into account substitutions
  SMDS_ElemIteratorPtr GetFaces(const TopoDS_Shape& face) const;

  // Return total nb of faces taking into account substitutions
  int NbFaces() const;

  bool IsTemporary(const SMDS_MeshElement* elem ) const;



  const SMESH_Mesh* GetMesh() const { return _mesh; }

  SMESHDS_Mesh* GetMeshDS() const;

  //--------------------------------------------------------------------------------
  // Interface for descendants
 protected:

  void setMesh(const SMESH_Mesh& mesh) { _mesh = &mesh; }

  int shapeIndex(const TopoDS_Shape& shape) const;

  // returns a proxy sub-mesh; zero index is for the case of mesh w/o shape
  SubMesh* findProxySubMesh(int shapeIndex=0) const;

  // returns a proxy sub-mesh; it is created if not yet exists
  SubMesh* getProxySubMesh(int shapeIndex);

  // returns a proxy sub-mesh; it is created if not yet exists
  SubMesh* getProxySubMesh(const TopoDS_Shape& shape=TopoDS_Shape());

  // move proxy sub-mesh from other proxy mesh to this, returns true if sub-mesh found
  bool takeProxySubMesh( const TopoDS_Shape& shape, SMESH_ProxyMesh* proxyMesh );

  // move tmp elements residing the _mesh from other proxy mesh to this
  void takeTmpElemsInMesh( SMESH_ProxyMesh* proxyMesh );

  // removes tmp faces from the _mesh
  void removeTmpElement( const SMDS_MeshElement* face );

  // stores tmp element residing the _mesh
  void storeTmpElement( const SMDS_MeshElement* face );

  // store node-node correspondence
  void setNode2Node(const SMDS_MeshNode* srcNode,
                    const SMDS_MeshNode* proxyNode,
                    const SubMesh*       subMesh);

  // types of elements needed to implement NbFaces() and GetFaces();
  // if _allowedTypes is empty, only elements from _subMeshes are returned,
  // else elements of _mesh filtered using allowedTypes are additionally returned
  std::vector< SMDSAbs_EntityType> _allowedTypes;

 private:

  const SMESH_Mesh*       _mesh;

  // proxy sub-meshes; index in vector == shapeIndex(shape)
  std::vector< SubMesh* > _subMeshes;

  // tmp elements residing the _mesh, to be deleted at destruction
  std::set< const SMDS_MeshElement* > _elemsInMesh;

  // Complex submesh used to iterate over elements in other sub-meshes
  mutable SubMesh _subContainer;
};

#endif