120ca87015
git-svn-id: https://free-cad.svn.sourceforge.net/svnroot/free-cad/trunk@5000 e8eeb9e2-ec13-0410-a4a9-efa5cf37419d
238 lines
7.7 KiB
C++
238 lines
7.7 KiB
C++
/***************************************************************************
|
|
* Copyright (c) Juergen Riegel <juergen.riegel@web.de> *
|
|
* *
|
|
* This file is part of the FreeCAD CAx development system. *
|
|
* *
|
|
* This library is free software; you can redistribute it and/or *
|
|
* modify it under the terms of the GNU Library General Public *
|
|
* License as published by the Free Software Foundation; either *
|
|
* version 2 of the License, or (at your option) any later version. *
|
|
* *
|
|
* 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 Library General Public License for more details. *
|
|
* *
|
|
* You should have received a copy of the GNU Library General Public *
|
|
* License along with this library; see the file COPYING.LIB. If not, *
|
|
* write to the Free Software Foundation, Inc., 59 Temple Place, *
|
|
* Suite 330, Boston, MA 02111-1307, USA *
|
|
* *
|
|
***************************************************************************/
|
|
|
|
#include "PreCompiled.h"
|
|
#ifndef _PreComp_
|
|
#endif
|
|
|
|
|
|
#include <Python.h>
|
|
|
|
#include <Base/Console.h>
|
|
#include <Base/FileInfo.h>
|
|
|
|
#include <App/Application.h>
|
|
#include <App/Document.h>
|
|
|
|
#include <Mod/Mesh/App/Core/MeshKernel.h>
|
|
#include <Mod/Mesh/App/Core/Elements.h>
|
|
#include <Mod/Mesh/App/MeshPy.h>
|
|
#include <Mod/Mesh/App/MeshPy.h>
|
|
#include <Mod/Mesh/App/MeshFeature.h>
|
|
|
|
#include "JtReader.h"
|
|
|
|
using std::vector;
|
|
using namespace MeshCore;
|
|
|
|
|
|
//using namespace JtReader;
|
|
|
|
/* module functions */
|
|
static PyObject * read(PyObject *self, PyObject *args)
|
|
{
|
|
const char* Name;
|
|
if (! PyArg_ParseTuple(args, "s",&Name))
|
|
return NULL;
|
|
|
|
PY_TRY {
|
|
|
|
std::auto_ptr<MeshCore::MeshKernel> apcKernel(new MeshCore::MeshKernel());
|
|
|
|
readFile(Name,0);
|
|
|
|
vector<MeshGeomFacet> facets;
|
|
facets.resize(iterSize());
|
|
|
|
const SimpleMeshFacet* It=iterStart();
|
|
int i=0;
|
|
while(It=iterGetNext())
|
|
{
|
|
facets[i]._aclPoints[0].x = It->p1[0];
|
|
facets[i]._aclPoints[0].y = It->p1[1];
|
|
facets[i]._aclPoints[0].z = It->p1[2];
|
|
facets[i]._aclPoints[1].x = It->p2[0];
|
|
facets[i]._aclPoints[1].y = It->p2[1];
|
|
facets[i]._aclPoints[1].z = It->p2[2];
|
|
facets[i]._aclPoints[2].x = It->p3[0];
|
|
facets[i]._aclPoints[2].y = It->p3[1];
|
|
facets[i]._aclPoints[2].z = It->p3[2];
|
|
}
|
|
clearData();
|
|
(*apcKernel) = facets;
|
|
|
|
return new Mesh::MeshPy(new Mesh::MeshObject(*(apcKernel.release())));
|
|
|
|
} PY_CATCH;
|
|
|
|
Py_Return;
|
|
}
|
|
|
|
static PyObject *
|
|
open(PyObject *self, PyObject *args)
|
|
{
|
|
const char* Name;
|
|
if (! PyArg_ParseTuple(args, "s",&Name))
|
|
return NULL;
|
|
|
|
PY_TRY {
|
|
|
|
//Base::Console().Log("Open in Mesh with %s",Name);
|
|
Base::FileInfo file(Name);
|
|
|
|
// extract ending
|
|
if(file.extension() == "")
|
|
Py_Error(PyExc_Exception,"no file ending");
|
|
|
|
if(file.hasExtension("jt"))
|
|
{
|
|
// create new document and add Import feature
|
|
App::Document *pcDoc = App::GetApplication().newDocument("Unnamed");
|
|
Mesh::Feature *pcFeature = (Mesh::Feature*)pcDoc->addObject("Mesh::Feature",file.fileNamePure().c_str());
|
|
|
|
std::auto_ptr<MeshCore::MeshKernel> apcKernel(new MeshCore::MeshKernel());
|
|
|
|
readFile(Name,0);
|
|
|
|
vector<MeshGeomFacet> facets;
|
|
facets.resize(iterSize());
|
|
|
|
const SimpleMeshFacet* It=iterStart();
|
|
int i=0;
|
|
while(It=iterGetNext())
|
|
{
|
|
facets[i]._aclPoints[0].x = It->p1[0];
|
|
facets[i]._aclPoints[0].y = It->p1[1];
|
|
facets[i]._aclPoints[0].z = It->p1[2];
|
|
facets[i]._aclPoints[1].x = It->p2[0];
|
|
facets[i]._aclPoints[1].y = It->p2[1];
|
|
facets[i]._aclPoints[1].z = It->p2[2];
|
|
facets[i]._aclPoints[2].x = It->p3[0];
|
|
facets[i]._aclPoints[2].y = It->p3[1];
|
|
facets[i]._aclPoints[2].z = It->p3[2];
|
|
i++;
|
|
}
|
|
clearData();
|
|
(*apcKernel) = facets;
|
|
pcFeature->Mesh.setValue(*(apcKernel.get()));
|
|
|
|
//pcFeature->FileName.setValue( Name );
|
|
pcDoc->recompute();
|
|
}
|
|
else
|
|
{
|
|
Py_Error(PyExc_Exception,"unknown file ending");
|
|
}
|
|
|
|
|
|
} PY_CATCH;
|
|
|
|
Py_Return;
|
|
}
|
|
|
|
|
|
/* module functions */
|
|
static PyObject *
|
|
insert(PyObject *self, PyObject *args)
|
|
{
|
|
const char* Name;
|
|
const char* DocName;
|
|
if (! PyArg_ParseTuple(args, "ss",&Name,&DocName))
|
|
return NULL;
|
|
|
|
PY_TRY {
|
|
|
|
Base::FileInfo file(Name);
|
|
|
|
// extract ending
|
|
if(file.extension() == "")
|
|
Py_Error(PyExc_Exception,"no file ending");
|
|
|
|
if(file.hasExtension("jt") )
|
|
{
|
|
// add Import feature
|
|
App::Document *pcDoc = App::GetApplication().getDocument(DocName);
|
|
if (!pcDoc)
|
|
{
|
|
char szBuf[200];
|
|
snprintf(szBuf, 200, "Import called to the non-existing document '%s'", DocName);
|
|
Py_Error(PyExc_Exception,szBuf);
|
|
}
|
|
|
|
readFile(Name,0);
|
|
|
|
vector<MeshGeomFacet> facets;
|
|
|
|
if(iterSize()>0){
|
|
facets.resize(iterSize());
|
|
|
|
const SimpleMeshFacet* It=iterStart();
|
|
int i=0;
|
|
while(It=iterGetNext())
|
|
{
|
|
facets[i]._aclPoints[0].x = It->p1[0];
|
|
facets[i]._aclPoints[0].y = It->p1[1];
|
|
facets[i]._aclPoints[0].z = It->p1[2];
|
|
facets[i]._aclPoints[1].x = It->p2[0];
|
|
facets[i]._aclPoints[1].y = It->p2[1];
|
|
facets[i]._aclPoints[1].z = It->p2[2];
|
|
facets[i]._aclPoints[2].x = It->p3[0];
|
|
facets[i]._aclPoints[2].y = It->p3[1];
|
|
facets[i]._aclPoints[2].z = It->p3[2];
|
|
i++;
|
|
}
|
|
clearData();
|
|
Mesh::Feature *pcFeature = (Mesh::Feature*)pcDoc->addObject("Mesh::Feature",file.fileNamePure().c_str());
|
|
|
|
std::auto_ptr<MeshCore::MeshKernel> apcKernel(new MeshCore::MeshKernel());
|
|
(*apcKernel) = facets;
|
|
pcFeature->Mesh.setValue(*(apcKernel.get()));
|
|
|
|
//pcDoc->recompute();
|
|
|
|
}else{
|
|
clearData();
|
|
//Py_Error(PyExc_Exception,"No Mesh in file");
|
|
Base::Console().Warning("No Mesh in file: %s\n",Name);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
Py_Error(PyExc_Exception,"unknown file ending");
|
|
}
|
|
|
|
} PY_CATCH;
|
|
|
|
Py_Return;
|
|
}
|
|
|
|
|
|
/* registration table */
|
|
struct PyMethodDef JtReader_methods[] = {
|
|
{"open" ,open , Py_NEWARGS, "open a jt file in a new Document"},
|
|
{"insert" ,insert, Py_NEWARGS, "isert a jt file in a existing document"},
|
|
{"read" ,read, Py_NEWARGS, "Read a Mesh from a jt file and returns a Mesh object."},
|
|
{NULL, NULL}
|
|
};
|
|
|
|
|