7afac5f206
The getTeighaConverter function did not work on Windows. We don't deal with the actual TeighaFileConverter anymore so there is no need to check for that file. Although the name of the function and the name of the parameter have been kept.
313 lines
10 KiB
Python
313 lines
10 KiB
Python
# -*- coding: utf8 -*-
|
|
## @package importDWG
|
|
# \ingroup DRAFT
|
|
# \brief DWG file importer & exporter
|
|
'''
|
|
@package importDWG
|
|
ingroup DRAFT
|
|
\brief DWG file importer & exporter
|
|
|
|
This module provides support for importing and exporting Autodesk DWG files.
|
|
This module is only a thin layer that uses the ODA (formerly Teigha) File
|
|
Converter application to convert to/from DXF. Then the real work is done by
|
|
importDXF
|
|
|
|
The converter may be called
|
|
/usr/bin/TeighaFileConverter
|
|
/usr/bin/ODAFileConverter
|
|
|
|
Test files
|
|
https://knowledge.autodesk.com/support/autocad/downloads/
|
|
caas/downloads/content/autocad-sample-files.html
|
|
'''
|
|
# Check code quality with
|
|
# flake8 --ignore=E226,E266,E401,W503
|
|
|
|
# ***************************************************************************
|
|
# * Copyright (c) 2009 Yorik van Havre <yorik@uncreated.net> *
|
|
# * *
|
|
# * This program is free software; you can redistribute it and/or modify *
|
|
# * it under the terms of the GNU Lesser General Public License (LGPL) *
|
|
# * as published by the Free Software Foundation; either version 2 of *
|
|
# * the License, or (at your option) any later version. *
|
|
# * for detail see the LICENCE text file. *
|
|
# * *
|
|
# * This program 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 program; if not, write to the Free Software *
|
|
# * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 *
|
|
# * USA *
|
|
# * *
|
|
# ***************************************************************************
|
|
|
|
# TODO: use subprocess.popen() instead of subprocess.call()
|
|
|
|
import six
|
|
import FreeCAD
|
|
from FreeCAD import Console as FCC
|
|
|
|
if FreeCAD.GuiUp:
|
|
from DraftTools import translate
|
|
else:
|
|
def translate(context, txt):
|
|
return txt
|
|
|
|
# Save the native open function to avoid collisions
|
|
if open.__module__ == '__builtin__':
|
|
pythonopen = open
|
|
|
|
|
|
def open(filename):
|
|
"""Open filename and parse using importDXF.open().
|
|
|
|
Parameters
|
|
----------
|
|
filename : str
|
|
The path to the filename to be opened.
|
|
|
|
Returns
|
|
-------
|
|
App::Document
|
|
The new FreeCAD document object created, with the parsed information.
|
|
"""
|
|
dxf = convertToDxf(filename)
|
|
if dxf:
|
|
import importDXF
|
|
doc = importDXF.open(dxf)
|
|
return doc
|
|
return
|
|
|
|
|
|
def insert(filename, docname):
|
|
"""Imports a file using importDXF.insert().
|
|
|
|
If no document exist, it is created.
|
|
|
|
Parameters
|
|
----------
|
|
filename : str
|
|
The path to the filename to be opened.
|
|
docname : str
|
|
The name of the active App::Document if one exists, or
|
|
of the new one created.
|
|
|
|
Returns
|
|
-------
|
|
App::Document
|
|
The active FreeCAD document, or the document created if none exists,
|
|
with the parsed information.
|
|
"""
|
|
dxf = convertToDxf(filename)
|
|
if dxf:
|
|
import importDXF
|
|
# Warning: function doesn't return?
|
|
doc = importDXF.insert(dxf, docname)
|
|
return doc
|
|
return
|
|
|
|
|
|
def export(objectslist, filename):
|
|
"""Export the DWG file with a given list of objects.
|
|
|
|
The objects are exported with importDXF.export().
|
|
Then the result is converted to DWG.
|
|
|
|
Parameters
|
|
----------
|
|
exportList : list
|
|
List of document objects to export.
|
|
filename : str
|
|
Path to the new file.
|
|
|
|
Returns
|
|
-------
|
|
str
|
|
The same `filename` input.
|
|
"""
|
|
import importDXF, os, tempfile
|
|
outdir = tempfile.mkdtemp()
|
|
_basename = os.path.splitext(os.path.basename(filename))[0]
|
|
dxf = outdir + os.sep + _basename + ".dxf"
|
|
importDXF.export(objectslist, dxf)
|
|
convertToDwg(dxf, filename)
|
|
return filename
|
|
|
|
|
|
def getTeighaConverter():
|
|
"""Find the ODA (formerly Teigha) executable.
|
|
|
|
It searches the FreeCAD parameters database, then searches for common
|
|
paths in Linux and Windows systems.
|
|
|
|
Parameters
|
|
----------
|
|
None
|
|
|
|
Returns
|
|
-------
|
|
str
|
|
The full path of the converter executable
|
|
'/usr/bin/ODAFileConverter'
|
|
"""
|
|
import FreeCAD, os, platform
|
|
p = FreeCAD.ParamGet("User parameter:BaseApp/Preferences/Mod/Draft")
|
|
path = p.GetString("TeighaFileConverter")
|
|
if path:
|
|
# path set manually
|
|
return path
|
|
elif platform.system() == "Linux":
|
|
path = "/usr/bin/ODAFileConverter"
|
|
if os.path.exists(path):
|
|
return path
|
|
elif platform.system() == "Windows":
|
|
odadir = os.path.expandvars("%ProgramFiles%\ODA")
|
|
if os.path.exists(odadir):
|
|
for sub in os.listdir(odadir):
|
|
path = os.path.join(odadir, sub, "ODAFileConverter.exe")
|
|
if os.path.exists(path):
|
|
return path
|
|
return None
|
|
|
|
|
|
def convertToDxf(dwgfilename):
|
|
"""Convert a DWG file to a DXF file.
|
|
|
|
If the converter is found it is used, otherwise the conversion fails.
|
|
|
|
Parameters
|
|
----------
|
|
dwgfilename : str
|
|
The input filename.
|
|
|
|
Returns
|
|
-------
|
|
str
|
|
The new file produced.
|
|
"""
|
|
import os, tempfile, subprocess, sys
|
|
|
|
p = FreeCAD.ParamGet("User parameter:BaseApp/Preferences/Mod/Draft")
|
|
conv = p.GetInt("DWGConversion",0)
|
|
path = p.GetString("TeighaFileConverter","")
|
|
|
|
if conv in [0,1]: # LibreDWG dxf2dwg/dwg2dxf
|
|
if not path:
|
|
path = "dwg2dxf"
|
|
if path.endswith("dxf2dwg"):
|
|
path = os.path.join(os.path.dirname,"dwg2dxf")
|
|
try:
|
|
outdir = tempfile.mkdtemp()
|
|
basename = os.path.basename(dwgfilename)
|
|
result = outdir + os.sep + os.path.splitext(basename)[0] + ".dxf"
|
|
proc = subprocess.Popen((path, dwgfilename, "-o", result))
|
|
proc.communicate()
|
|
return result
|
|
except Exception:
|
|
if conv != 0:
|
|
FCC.PrintError(translate("draft", "LibreDWG error") + "\n")
|
|
|
|
if conv in [0,2]: # ODA
|
|
teigha = getTeighaConverter()
|
|
if teigha:
|
|
indir = os.path.dirname(dwgfilename)
|
|
outdir = tempfile.mkdtemp()
|
|
basename = os.path.basename(dwgfilename)
|
|
cmdline = ('"%s" "%s" "%s" "ACAD2000" "DXF" "0" "1" "%s"' % (teigha, indir, outdir, basename))
|
|
FCC.PrintMessage(translate("draft", "Converting:") + " " + cmdline + "\n")
|
|
if six.PY2:
|
|
if isinstance(cmdline, six.text_type):
|
|
encoding = sys.getfilesystemencoding()
|
|
cmdline = cmdline.encode(encoding)
|
|
subprocess.call(cmdline, shell=True) # os.system(cmdline)
|
|
result = outdir + os.sep + os.path.splitext(basename)[0] + ".dxf"
|
|
if os.path.exists(result):
|
|
FCC.PrintMessage(translate("draft", "Conversion successful") + "\n")
|
|
return result
|
|
else:
|
|
FCC.PrintError(translate("draft","Error during DWG conversion. Try moving the DWG file to a directory path without spaces and non-english characters, or try saving to a lower DWG version.") + "\n")
|
|
else:
|
|
if conv != 0:
|
|
FCC.PrintError(translate("draft", "ODA File Converter not found") + "\n")
|
|
|
|
if conv in [0,3]: # QCAD
|
|
if not path:
|
|
path = "dwg2dwg"
|
|
try:
|
|
outdir = tempfile.mkdtemp()
|
|
basename = os.path.basename(dwgfilename)
|
|
result = outdir + os.sep + os.path.splitext(basename)[0] + ".dxf"
|
|
proc = subprocess.Popen((path, "-f", "-o", result, dwgfilename))
|
|
proc.communicate()
|
|
return result
|
|
except Exception:
|
|
FCC.PrintError(translate("draft", "QCAD error") + "\n")
|
|
|
|
return None
|
|
|
|
|
|
def convertToDwg(dxffilename, dwgfilename):
|
|
"""Convert a DXF file to a DWG file.
|
|
|
|
If the converter is found it is used, otherwise the conversion fails.
|
|
|
|
Parameters
|
|
----------
|
|
dxffilename : str
|
|
The input DXF file
|
|
dwgfilename : str
|
|
The output DWG file
|
|
|
|
Returns
|
|
-------
|
|
str
|
|
The same `dwgfilename` file path.
|
|
"""
|
|
import os, subprocess, shutil
|
|
|
|
p = FreeCAD.ParamGet("User parameter:BaseApp/Preferences/Mod/Draft")
|
|
conv = p.GetInt("DWGConversion",0)
|
|
path = p.GetString("TeighaFileConverter","")
|
|
|
|
if conv in [0,1]: # LibreDWG dxf2dwg/dwg2dxf
|
|
if not path:
|
|
path = "dxf2dwg"
|
|
if path.endswith("dwg2dxf"):
|
|
path = os.path.join(os.path.dirname,"dxf2dwg")
|
|
try:
|
|
proc = subprocess.Popen((path, dxffilename, "-y", "-o", dwgfilename))
|
|
proc.communicate()
|
|
return dwgfilename
|
|
except Exception:
|
|
if conv != 0:
|
|
FCC.PrintError(translate("draft", "LibreDWG error") + "\n")
|
|
|
|
if conv in [0,2]: # ODA
|
|
teigha = getTeighaConverter()
|
|
if teigha:
|
|
indir = os.path.dirname(dxffilename)
|
|
outdir = os.path.dirname(dwgfilename)
|
|
basename = os.path.basename(dxffilename)
|
|
cmdline = ('"%s" "%s" "%s" "ACAD2000" "DWG" "0" "1" "%s"' % (teigha, indir, outdir, basename))
|
|
FCC.PrintMessage(translate("draft", "Converting:") + " " + cmdline + "\n")
|
|
subprocess.call(cmdline, shell=True) # os.system(cmdline)
|
|
return dwgfilename
|
|
else:
|
|
if conv != 0:
|
|
FCC.PrintError(translate("draft", "ODA File Converter not found") + "\n")
|
|
|
|
if conv in [0,3]: # QCAD
|
|
if not path:
|
|
path = "dwg2dwg"
|
|
try:
|
|
proc = subprocess.Popen((path, "-o", dwgfilename, dxffilename))
|
|
proc.communicate()
|
|
return dwgfilename
|
|
except Exception:
|
|
FCC.PrintError(translate("draft", "QCAD error") + "\n")
|
|
|
|
return None
|