Implementation of 2.6.3/2.6.4 improvements

idl/SMESH_Mesh.idl

@@ -313,6 +313,10 @@ module SMESH
      * happen if mesh data is not yet fully loaded from the file of study.
      */
     boolean IsMeshInfoCorrect();
+    /*!
+     * Returns mesh unstructed grid information.
+     */
+    SALOMEDS::TMPFile GetVtkUgStream();
   };
 
   interface SMESH_Group;

src/SMESH_I/SMESH_Filter_i.hxx

@@ -952,7 +952,7 @@ namespace SMESH
     virtual SMESH::array_of_ElementType* GetTypes();
     virtual SMESH::SMESH_Mesh_ptr        GetMesh();
     virtual bool                         IsMeshInfoCorrect() { return true; }
-
+    virtual SALOMEDS::TMPFile* GetVtkUgStream(){ return NULL; }
     /*!
      * \brief Object notified on change of predicate
      */

src/SMESH_I/SMESH_Group_i.hxx

@@ -95,6 +95,10 @@ class SMESH_I_EXPORT SMESH_GroupBase_i:
    * happen if mesh data is not yet fully loaded from the file of study.
    */
   virtual bool IsMeshInfoCorrect();
+  /*!
+   * Returns mesh unstructed grid information.
+   */
+  virtual SALOMEDS::TMPFile* GetVtkUgStream(){ return NULL; }
 
   // Internal C++ interface
   int GetLocalID() const { return myLocalID; }

src/SMESH_I/SMESH_MeshEditor_i.cxx

@@ -779,6 +779,7 @@ struct SMESH_MeshEditor_i::_IDSource : public virtual POA_SMESH::SMESH_IDSource,
     }
     return types._retn();
   }
+  SALOMEDS::TMPFile* GetVtkUgStream(){ return NULL; }
 };
 
 SMESH::SMESH_IDSource_ptr SMESH_MeshEditor_i::MakeIDSource(const SMESH::long_array& ids,

src/SMESH_I/SMESH_Mesh_i.cxx

@@ -79,6 +79,8 @@
 #include <iostream>
 #include <sstream>
 
+#include <vtkUnstructuredGridWriter.h>
+
 // to pass CORBA exception through SMESH_TRY
 #define SMY_OWN_CATCH catch( SALOME::SALOME_Exception& se ) { throw se; }
 
@@ -5048,6 +5050,43 @@ void SMESH_Mesh_i::CollectMeshInfo(const SMDS_ElemIteratorPtr theItr,
   while (theItr->more())
     theInfo[ theItr->next()->GetEntityType() ]++;
 }
+//=============================================================================
+/*
+ * Returns mesh unstructed grid information.
+ */
+//=============================================================================
+
+SALOMEDS::TMPFile* SMESH_Mesh_i::GetVtkUgStream()
+{
+	if ( SMESHDS_Mesh* aMeshDS = _impl->GetMeshDS() )
+	{
+		vtkUnstructuredGrid* aCopy;
+		SMDS_UnstructuredGrid* aGrid = aMeshDS->getGrid();
+		if(aGrid)
+		{
+			aCopy = aGrid->NewInstance();
+			aCopy->ShallowCopy(aGrid);
+			vtkUnstructuredGridWriter* aWriter = vtkUnstructuredGridWriter::New();
+			aWriter->WriteToOutputStringOn();
+			aWriter->SetInputData(aCopy);
+			aWriter->SetFileTypeToBinary();
+			aWriter->Write();
+			char* str = aWriter->GetOutputString();
+			int size = aWriter->GetOutputStringLength();
+
+			//Allocate octect buffer of required size
+			CORBA::Octet* OctetBuf = SALOMEDS::TMPFile::allocbuf(size);
+			//Copy ostrstream content to the octect buffer
+			memcpy(OctetBuf, str, size);
+			//Create and return TMPFile
+			SALOMEDS::TMPFile_var SeqFile = new SALOMEDS::TMPFile(size, size, OctetBuf, 1);
+
+			aWriter->Delete();
+			return SeqFile._retn();
+		}
+	}
+ return NULL;
+}
 
 //=============================================================================
 namespace /* Iterators used in SMESH_Mesh_i::GetElements(SMESH::SMESH_IDSource_var obj,

src/SMESH_I/SMESH_Mesh_i.hxx

@@ -650,6 +650,10 @@ public:
    * happen if mesh data is not yet fully loaded from the file of study.
    */
   bool IsMeshInfoCorrect();
+   /*!
+    * Returns mesh unstructed grid information.
+    */
+   virtual SALOMEDS::TMPFile* GetVtkUgStream();
 
   std::map<int, SMESH_subMesh_i*> _mapSubMesh_i; //NRI
   std::map<int, ::SMESH_subMesh*> _mapSubMesh;   //NRI

src/SMESH_I/SMESH_subMesh_i.hxx

@@ -110,6 +110,10 @@ public:
    * happen if mesh data is not yet fully loaded from the file of study.
    */
   virtual bool IsMeshInfoCorrect();
+  /*!
+   * Returns mesh unstructed grid information.
+   */
+  virtual SALOMEDS::TMPFile* GetVtkUgStream(){ return NULL; }
 
 
 protected: