-using namespace std;
-
-#include "GEOMImpl_IInsertOperations.hxx"
+// Copyright (C) 2007-2010 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
+//
+
+#include <Standard_Stream.hxx>
+
+#include <GEOMImpl_IInsertOperations.hxx>
#include "utilities.h"
-#include "OpUtil.hxx"
-#include "Utils_ExceptHandlers.hxx"
+#include <OpUtil.hxx>
+#include <Utils_ExceptHandlers.hxx>
#include <TFunction_DriverTable.hxx>
#include <TFunction_Driver.hxx>
#include <TFunction_Logbook.hxx>
#include <TDF_Tool.hxx>
-#include "GEOM_Function.hxx"
+#include <GEOM_Function.hxx>
+#include <GEOM_PythonDump.hxx>
+
+#include <GEOMImpl_CopyDriver.hxx>
+#include <GEOMImpl_ExportDriver.hxx>
+#include <GEOMImpl_ImportDriver.hxx>
-#include "GEOMImpl_CopyDriver.hxx"
-#include "GEOMImpl_ExportDriver.hxx"
-#include "GEOMImpl_ImportDriver.hxx"
+#include <GEOMImpl_ICopy.hxx>
+#include <GEOMImpl_IImportExport.hxx>
-#include "GEOMImpl_ICopy.hxx"
-#include "GEOMImpl_IImportExport.hxx"
+#include <GEOMImpl_Types.hxx>
-#include "GEOMImpl_Types.hxx"
+#include <TopoDS.hxx>
+#include <TopoDS_Vertex.hxx>
+#include <BRep_Tool.hxx>
+#include <gp_Pnt.hxx>
+#include <TDataStd_HArray1OfByte.hxx>
+#include <Standard_Failure.hxx>
#include <Standard_ErrorHandler.hxx> // CAREFUL ! position of this file is critic : see Lucien PIGNOLONI / OCC
//=============================================================================
//Compute the Copy value
try {
+#if (OCC_VERSION_MAJOR << 16 | OCC_VERSION_MINOR << 8 | OCC_VERSION_MAINTENANCE) > 0x060100
+ OCC_CATCH_SIGNALS;
+#endif
if (!GetSolver()->ComputeFunction(aFunction)) {
SetErrorCode("Copy driver failed");
return NULL;
}
//Make a Python command
- TCollection_AsciiString anEntry, aDescr;
- TDF_Tool::Entry(aCopy->GetEntry(), anEntry);
- aDescr += (anEntry+" = IInsertOperations.MakeCopy(");
- TDF_Tool::Entry(theOriginal->GetEntry(), anEntry);
- aDescr += (anEntry+")");
-
- aFunction->SetDescription(aDescr);
+ GEOM::TPythonDump(aFunction) << aCopy << " = geompy.MakeCopy(" << theOriginal << ")";
SetErrorCode(OK);
return aCopy;
*/
//=============================================================================
void GEOMImpl_IInsertOperations::Export
- (const Handle(GEOM_Object) theOriginal,
- const char* theFileName,
- const char* theFormatName)
+ (const Handle(GEOM_Object) theOriginal,
+ const TCollection_AsciiString& theFileName,
+ const TCollection_AsciiString& theFormatName)
{
SetErrorCode(KO);
//Check if the function is set correctly
if (aFunction->GetDriverGUID() != GEOMImpl_ExportDriver::GetID()) return;
- //Set parameters
- GEOMImpl_IImportExport aCI (aFunction);
- aCI.SetOriginal(aRefFunction);
- char* aFileName = (char*)theFileName;
- aCI.SetFileName(aFileName);
-
- char* aFormatName = (char*)theFormatName;
Handle(TCollection_HAsciiString) aHLibName;
- if (!IsSupported(Standard_False, aFormatName, aHLibName)) {
+ if (!IsSupported(Standard_False, theFormatName, aHLibName)) {
return;
}
TCollection_AsciiString aLibName = aHLibName->String();
+
+ //Set parameters
+ GEOMImpl_IImportExport aCI (aFunction);
+ aCI.SetOriginal(aRefFunction);
+ aCI.SetFileName(theFileName);
+ aCI.SetFormatName(theFormatName);
aCI.SetPluginName(aLibName);
//Perform the Export
try {
+#if (OCC_VERSION_MAJOR << 16 | OCC_VERSION_MINOR << 8 | OCC_VERSION_MAINTENANCE) > 0x060100
+ OCC_CATCH_SIGNALS;
+#endif
if (!GetSolver()->ComputeFunction(aFunction)) {
- SetErrorCode("Export driver failed");
+ SetErrorCode("Not enough space on disk, or you haven't permissions to write this directory");
return;
}
}
}
//Make a Python command
- TCollection_AsciiString anEntry, aDescr;
- aDescr = "IInsertOperations.Export(";
- TDF_Tool::Entry(theOriginal->GetEntry(), anEntry);
- aDescr += (anEntry + ", ");
- aDescr += (TCollection_AsciiString(aFileName) + ", ");
- aDescr += (TCollection_AsciiString(aFormatName) + ")");
-
- aFunction->SetDescription(aDescr);
+ GEOM::TPythonDump(aFunction) << "geompy.Export(" << theOriginal << ", \""
+ << theFileName.ToCString() << "\", \"" << theFormatName.ToCString() << "\")";
SetErrorCode(OK);
}
*/
//=============================================================================
Handle(GEOM_Object) GEOMImpl_IInsertOperations::Import
- (const char* theFileName,
- const char* theFormatName)
+ (const TCollection_AsciiString& theFileName,
+ const TCollection_AsciiString& theFormatName)
{
SetErrorCode(KO);
- if (!theFileName || !theFormatName) return NULL;
+ if (theFileName.IsEmpty() || theFormatName.IsEmpty()) return NULL;
//Add a new result object
Handle(GEOM_Object) result = GetEngine()->AddObject(GetDocID(), GEOM_IMPORT);
//Check if the function is set correctly
if (aFunction->GetDriverGUID() != GEOMImpl_ImportDriver::GetID()) return result;
- //Set parameters
- GEOMImpl_IImportExport aCI (aFunction);
- char* aFileName = (char*)theFileName;
- aCI.SetFileName(aFileName);
-
- char* aFormatName = (char*)theFormatName;
Handle(TCollection_HAsciiString) aHLibName;
- if (!IsSupported(Standard_True, aFormatName, aHLibName)) {
+ if (!IsSupported(Standard_True, theFormatName.SubString(1,4), aHLibName)) {
return result;
}
TCollection_AsciiString aLibName = aHLibName->String();
+
+ //Set parameters
+ GEOMImpl_IImportExport aCI (aFunction);
+ aCI.SetFileName(theFileName);
+ aCI.SetFormatName(theFormatName);
aCI.SetPluginName(aLibName);
+ //cout<<"IIO: theFormatName = "<<theFormatName.ToCString()<<endl;
//Perform the Import
try {
+#if (OCC_VERSION_MAJOR << 16 | OCC_VERSION_MINOR << 8 | OCC_VERSION_MAINTENANCE) > 0x060100
+ OCC_CATCH_SIGNALS;
+#endif
if (!GetSolver()->ComputeFunction(aFunction)) {
SetErrorCode("Import driver failed");
return NULL;
}
//Make a Python command
- TCollection_AsciiString anEntry, aDescr;
- TDF_Tool::Entry(result->GetEntry(), anEntry);
- aDescr += (anEntry + " = IInsertOperations.Import(");
- aDescr += (TCollection_AsciiString(aFileName) + ", ");
- aDescr += (TCollection_AsciiString(aFormatName) + ")");
-
- aFunction->SetDescription(aDescr);
+ if( theFormatName != "IGES_UNIT" ) {
+ GEOM::TPythonDump(aFunction) << result << " = geompy.ImportFile(\""
+ << theFileName.ToCString() << "\", \"" << theFormatName.ToCString() << "\")";
+ }
SetErrorCode(OK);
+
+ if( theFormatName == "IGES_UNIT" ) {
+ TopoDS_Shape S = aFunction->GetValue();
+ TopoDS_Vertex V = TopoDS::Vertex(S);
+ gp_Pnt P = BRep_Tool::Pnt(V);
+ double scale = P.X();
+ TCollection_AsciiString aUnitName = "UNIT_M";
+ if( fabs(scale-0.01) < 1.e-6 )
+ aUnitName = "UNIT_CM";
+ else if( fabs(scale-0.001) < 1.e-6 )
+ aUnitName = "UNIT_MM";
+ //cout<<"IIO: aUnitName = "<<aUnitName.ToCString()<<endl;
+ SetErrorCode(aUnitName);
+ }
+
return result;
}
//=============================================================================
Standard_Boolean GEOMImpl_IInsertOperations::ImportTranslators
(Handle(TColStd_HSequenceOfAsciiString)& theFormats,
- Handle(TColStd_HSequenceOfAsciiString)& thePatterns)
+ Handle(TColStd_HSequenceOfAsciiString)& thePatterns)
{
if (theFormats.IsNull())
theFormats = new TColStd_HSequenceOfAsciiString;
if (!InitResMgr()) return Standard_False;
- // Read Import formats list
+ // Read Import formats list from install directory
if (myResMgr->Find("Import")) {
TCollection_AsciiString aFormats (myResMgr->Value("Import"));
TCollection_AsciiString aToken = aFormats.Token("| \t", 1);
}
}
+ // Read Import formats from user directory
+ if (myResMgrUser->Find("Import")) {
+ TCollection_AsciiString aFormats (myResMgrUser->Value("Import"));
+ TCollection_AsciiString aToken = aFormats.Token("| \t", 1);
+ int i = 1;
+ for (; !aToken.IsEmpty(); aToken = aFormats.Token("| \t", ++i)) {
+ int aLenFormats = theFormats->Length();
+ bool isFound = false;
+ for(int aInd=1;aInd<=aLenFormats;aInd++){
+ if( theFormats->Value(aInd) == aToken){
+ isFound = true;
+ break;
+ }
+ }
+ if(!isFound)
+ theFormats->Append(aToken);
+ }
+ }
+
// Read Patterns for each supported format
int j = 1, len = theFormats->Length();
for (; j <= len; j++) {
- TCollection_AsciiString aPattern;
- TCollection_AsciiString aKey (theFormats->Value(j));
- aKey += ".Pattern";
+ TCollection_AsciiString aKey, aPattern;
+ aKey = theFormats->Value(j) + ".ImportPattern";
if (myResMgr->Find(aKey.ToCString()))
aPattern = myResMgr->Value(aKey.ToCString());
+ else if(myResMgrUser->Find(aKey.ToCString()))
+ aPattern = myResMgrUser->Value(aKey.ToCString());
else {
- aPattern = theFormats->Value(j);
- aPattern += " Files ( *.* )";
+ aKey = theFormats->Value(j) + ".Pattern";
+ if (myResMgr->Find(aKey.ToCString()))
+ aPattern = myResMgr->Value(aKey.ToCString());
+ else if(myResMgrUser->Find(aKey.ToCString()))
+ aPattern = myResMgrUser->Value(aKey.ToCString());
+ else {
+ aPattern = theFormats->Value(j);
+ aPattern += " Files ( *.* )";
+ }
}
thePatterns->Append(aPattern);
}
//=============================================================================
Standard_Boolean GEOMImpl_IInsertOperations::ExportTranslators
(Handle(TColStd_HSequenceOfAsciiString)& theFormats,
- Handle(TColStd_HSequenceOfAsciiString)& thePatterns)
+ Handle(TColStd_HSequenceOfAsciiString)& thePatterns)
{
if (theFormats.IsNull())
theFormats = new TColStd_HSequenceOfAsciiString;
if (!InitResMgr()) return Standard_False;
- // Read Export formats list
+ // Read Export formats list from install directory
if (myResMgr->Find("Export")) {
TCollection_AsciiString aFormats (myResMgr->Value("Export"));
TCollection_AsciiString aToken = aFormats.Token("| \t", 1);
}
}
+ // Read Export formats list from user directory
+ if (myResMgrUser->Find("Export")) {
+ TCollection_AsciiString aFormats (myResMgrUser->Value("Export"));
+ TCollection_AsciiString aToken = aFormats.Token("| \t", 1);
+ int i = 1;
+ for (; !aToken.IsEmpty(); aToken = aFormats.Token("| \t", ++i)) {
+ int aLenFormats = theFormats->Length();
+ bool isFound = false;
+ for(int aInd=1;aInd<=aLenFormats;aInd++){
+ if( theFormats->Value(aInd) == aToken){
+ isFound = true;
+ break;
+ }
+ }
+ if(!isFound)
+ theFormats->Append(aToken);
+ }
+ }
+
// Read Patterns for each supported format
int j = 1, len = theFormats->Length();
for (; j <= len; j++) {
- TCollection_AsciiString aPattern;
- TCollection_AsciiString aKey (theFormats->Value(j));
- aKey += ".Pattern";
+ TCollection_AsciiString aKey, aPattern;
+ aKey = theFormats->Value(j) + ".ExportPattern";
if (myResMgr->Find(aKey.ToCString()))
aPattern = myResMgr->Value(aKey.ToCString());
+ else if (myResMgrUser->Find(aKey.ToCString()))
+ aPattern = myResMgrUser->Value(aKey.ToCString());
else {
- aPattern = theFormats->Value(j);
- aPattern += " Files ( *.* )";
+ aKey = theFormats->Value(j) + ".Pattern";
+ if (myResMgr->Find(aKey.ToCString()))
+ aPattern = myResMgr->Value(aKey.ToCString());
+ else if (myResMgrUser->Find(aKey.ToCString()))
+ aPattern = myResMgrUser->Value(aKey.ToCString());
+ else {
+ aPattern = theFormats->Value(j);
+ aPattern += " Files ( *.* )";
+ }
}
thePatterns->Append(aPattern);
}
//=============================================================================
Standard_Boolean GEOMImpl_IInsertOperations::IsSupported
(const Standard_Boolean isImport,
- const TCollection_AsciiString theFormat,
+ const TCollection_AsciiString& theFormat,
Handle(TCollection_HAsciiString)& theLibName)
{
if (!InitResMgr()) return Standard_False;
// Import/Export mode
- Standard_CString aMode;
+ TCollection_AsciiString aMode;
+ //Standard_CString aMode;
if (isImport) aMode = "Import";
else aMode = "Export";
- // Read supported formats for the certain mode
- if (myResMgr->Find(aMode)) {
- TCollection_AsciiString aFormats (myResMgr->Value(aMode));
+ // Read supported formats for the certain mode from install directory
+ if (myResMgr->Find(aMode.ToCString())) {
+ TCollection_AsciiString aFormats (myResMgr->Value(aMode.ToCString()));
if (aFormats.Search(theFormat) > -1) {
// Read library name for the supported format
TCollection_AsciiString aKey (theFormat);
aKey += ".";
aKey += aMode;
if (myResMgr->Find(aKey.ToCString())) {
- TCollection_AsciiString aLibName (myResMgr->Value(aKey.ToCString()));
+ TCollection_AsciiString aLibName (myResMgr->Value(aKey.ToCString()));
+#ifndef WNT
+ aLibName += ".so";
+#else
+ aLibName += ".dll";
+#endif
theLibName = new TCollection_HAsciiString (aLibName);
return Standard_True;
}
}
}
-
+
+ // Read supported formats for the certain mode from user directory
+ if (myResMgrUser->Find(aMode.ToCString())) {
+ TCollection_AsciiString aFormats (myResMgrUser->Value(aMode.ToCString()));
+ if (aFormats.Search(theFormat) > -1) {
+ // Read library name for the supported format
+ TCollection_AsciiString aKey (theFormat);
+ aKey += ".";
+ aKey += aMode;
+ if (myResMgrUser->Find(aKey.ToCString())) {
+ TCollection_AsciiString aLibName (myResMgrUser->Value(aKey.ToCString()));
+ theLibName = new TCollection_HAsciiString (aLibName);
+ return Standard_True;
+ }
+ }
+ }
+
return Standard_False;
}
//=============================================================================
Standard_Boolean GEOMImpl_IInsertOperations::InitResMgr()
{
+ bool isResourceFound = false;
+ bool isResourceFoundUser = false;
+ TCollection_AsciiString aUserResDir,aResDir;
+
if (myResMgr.IsNull()) {
// Initialize the Resource Manager
- TCollection_AsciiString aResDir (getenv("GEOM_ROOT_DIR"));
+ TCollection_AsciiString aNull;
+ aResDir = TCollection_AsciiString(getenv("GEOM_ROOT_DIR"));
#ifdef WNT
- aResDir += "\\share\\salome\\resources";
+ aResDir += "\\share\\salome\\resources\\geom";
#else
- aResDir += "/share/salome/resources";
+ aResDir += "/share/salome/resources/geom";
#endif
- TCollection_AsciiString aUserResDir (getenv("HOME"));
+
+ myResMgr = new Resource_Manager ("ImportExport", aResDir, aNull, Standard_False);
+
+ isResourceFound = true;
+ if (!myResMgr->Find("Import") && !myResMgr->Find("Export")) {
+ // instead of complains in Resource_Manager
+ isResourceFound = false;
+ INFOS("No valid file \"ImportExport\" found in " << aResDir.ToCString());
+ }
+ } else
+ isResourceFound = true;
+
+ if (myResMgrUser.IsNull()) {
+ char * dir = getenv("GEOM_ENGINE_RESOURCES_DIR");
+ TCollection_AsciiString aNull;
+ if ( dir )
+ {
+ aUserResDir = dir;
+ }
+ else
+ {
+ aUserResDir = getenv("HOME");
#ifdef WNT
- aUserResDir += "\\.salome\\resources";
+ aUserResDir += "\\.salome\\resources";
#else
- aUserResDir += "/.salome/resources";
+ aUserResDir += "/.salome/resources";
#endif
- myResMgr = new Resource_Manager ("ImportExport", aResDir, aUserResDir, Standard_True);
+ }
+
+ myResMgrUser = new Resource_Manager ("ImportExport", aNull, aUserResDir, Standard_False);
+
+ isResourceFoundUser = true;
+
+ if (!myResMgrUser->Find("Import") && !myResMgrUser->Find("Export")) {
+ // instead of complains in Resource_Manager
+ isResourceFoundUser = false;
+ }
+
+ } else
+ isResourceFoundUser = true;
+
+ if(!isResourceFound && !isResourceFoundUser){
+ INFOS("No valid file \"ImportExport\" found in " << aResDir.ToCString());
+ INFOS("No valid file \"ImportExport\" found in " << aUserResDir.ToCString() );
}
- if (myResMgr->Find("Import") || myResMgr->Find("Export"))
- return Standard_True;
+ return ( myResMgr->Find("Import") || myResMgr->Find("Export") ||
+ myResMgrUser->Find("Import") || myResMgrUser->Find("Export"));
+}
- return Standard_False;
+int GEOMImpl_IInsertOperations::LoadTexture(const TCollection_AsciiString& theTextureFile)
+{
+ SetErrorCode(KO);
+
+ if (theTextureFile.IsEmpty()) return 0;
+
+ Handle(TDataStd_HArray1OfByte) aTexture;
+
+ FILE* fp = fopen(theTextureFile.ToCString(), "r");
+ if (!fp) return 0;
+
+ std::list<std::string> lines;
+ char buffer[4096];
+ int maxlen = 0;
+ while (!feof(fp)) {
+ if ((fgets(buffer, 4096, fp)) == NULL) break;
+ int aLen = strlen(buffer);
+ if (buffer[aLen-1] == '\n') buffer[aLen-1] = '\0';
+ lines.push_back(buffer);
+ maxlen = std::max(maxlen, (int)strlen(buffer));
+ }
+
+ fclose(fp);
+
+ int lenbytes = maxlen/8;
+ if (maxlen%8) lenbytes++;
+
+ if (lenbytes == 0 || lines.empty())
+ return 0;
+
+ std::list<unsigned char> bytedata;
+ std::list<std::string>::const_iterator it;
+ for (it = lines.begin(); it != lines.end(); ++it) {
+ std::string line = *it;
+ int lenline = (line.size()/8 + (line.size()%8 ? 1 : 0)) * 8;
+ for (int i = 0; i < lenline/8; i++) {
+ unsigned char byte = 0;
+ for (int j = 0; j < 8; j++)
+ byte = (byte << 1) + ( i*8+j < line.size() && line[i*8+j] != '0' ? 1 : 0 );
+ bytedata.push_back(byte);
+ }
+ for (int i = lenline/8; i < lenbytes; i++)
+ bytedata.push_back((unsigned char)0);
+ }
+
+ if (bytedata.empty() || bytedata.size() != lines.size()*lenbytes)
+ return 0;
+
+ aTexture = new TDataStd_HArray1OfByte(1, lines.size()*lenbytes);
+ std::list<unsigned char>::iterator bdit;
+ int i;
+ for (i = 1, bdit = bytedata.begin(); bdit != bytedata.end(); ++bdit, ++i)
+ aTexture->SetValue(i, (Standard_Byte)(*bdit));
+
+ int aTextureId = GetEngine()->addTexture(GetDocID(), lenbytes*8, lines.size(), aTexture, theTextureFile);
+ if (aTextureId > 0) SetErrorCode(OK);
+ return aTextureId;
+}
+
+int GEOMImpl_IInsertOperations::AddTexture(int theWidth, int theHeight,
+ const Handle(TDataStd_HArray1OfByte)& theTexture)
+{
+ SetErrorCode(KO);
+ int aTextureId = GetEngine()->addTexture(GetDocID(), theWidth, theHeight, theTexture);
+ if (aTextureId > 0) SetErrorCode(OK);
+ return aTextureId;
+}
+
+Handle(TDataStd_HArray1OfByte) GEOMImpl_IInsertOperations::GetTexture(int theTextureId,
+ int& theWidth, int& theHeight)
+{
+ SetErrorCode(KO);
+
+ Handle(TDataStd_HArray1OfByte) aTexture;
+ theWidth = theHeight = 0;
+ TCollection_AsciiString aFileName;
+
+ if (theTextureId <= 0)
+ return aTexture;
+
+ aTexture = GetEngine()->getTexture(GetDocID(), theTextureId, theWidth, theHeight, aFileName);
+
+ if (theWidth > 0 && theHeight > 0 && aTexture->Length() > 0) SetErrorCode(OK);
+
+ return aTexture;
+}
+
+std::list<int> GEOMImpl_IInsertOperations::GetAllTextures()
+{
+ SetErrorCode(KO);
+ std::list<int> id_list = GetEngine()->getAllTextures(GetDocID());
+ SetErrorCode(OK);
+ return id_list;
}
