def make_snapshot(input_file, output_file, size=48):
from pivy import coin
ext = os.path.splitext(input_file)[1][1:]
mod = FreeCAD.getImportType(ext)
if len(mod) == 0:
print("Cannot load file {}".format(input_file))
return
# use the first listed module
module = importlib.import_module(mod[0])
module.open(input_file)
doc = FreeCAD.ActiveDocument
if doc is None:
print("No active document")
return
init_gui()
nodes = [FreeCADGui.subgraphFromObject(obj) for obj in doc.Objects]
# add light and camera so that the rendered geometry is visible
root = coin.SoSeparator()
light = coin.SoDirectionalLight()
cam = coin.SoOrthographicCamera()
root.addChild(cam)
root.addChild(light)
for node in nodes:
root.addChild(node)
# do the rendering now
axo = coin.SbRotation(-0.353553, -0.146447, -0.353553, -0.853553)
width = size
height = size
viewport = coin.SbViewportRegion(width, height)
cam.orientation.setValue(axo)
cam.viewAll(root, viewport)
off = FreeCADGui.SoQtOffscreenRenderer(width, height)
off.setBackgroundColor(1, 1, 1)
root.ref()
# A QGuiApplication is needed to create an OpenGL context
if QtGui.QGuiApplication.instance() is None:
app = QtGui.QGuiApplication(sys.argv)
off.render(root)
off.writeToImage(output_file)
root.unref()
def isOpenableByFreeCAD(filename):
"check if FreeCAD can handle this file type"
if os.path.isdir(filename):
return False
if os.path.basename(filename)[0] == ".":
return False
extensions = [key.lower() for key in FreeCAD.getImportType().keys()]
ext = os.path.splitext(filename)[1].lower()
if ext:
if ext[0] == ".":
ext = ext[1:]
if ext in extensions:
return True
return False
def isOpenableByFreeCAD(filename):
"check if FreeCAD can handle this file type"
if os.path.isdir(filename):
return False
if os.path.basename(filename)[0] == ".":
return False
extensions = [key.lower() for key in FreeCAD.getImportType().keys()]
ext = os.path.splitext(filename)[1].lower()
if ext:
if ext[0] == ".":
ext = ext[1:]
if ext in extensions:
return True
return False
def reverseimporttypes():
'''allows to search for supported filetypes by module'''
def getsetfromdict(dict1, index):
if index in dict1:
return dict1[index]
else:
set1 = set()
dict1[index] = set1
return set1
importtypes = {}
import FreeCAD
for key, value in FreeCAD.getImportType().iteritems():
if type(value) is str:
getsetfromdict(importtypes, value).add(key)
else:
for vitem in value:
getsetfromdict(importtypes, vitem).add(key)
return importtypes
def reverseimporttypes():
'''allows to search for supported filetypes by module'''
def getsetfromdict(dict1,index):
if index in dict1:
return dict1[index]
else:
set1=set()
dict1[index]=set1
return set1
importtypes={}
import FreeCAD
for key,value in FreeCAD.getImportType().iteritems():
if type(value) is str:
getsetfromdict(importtypes,value).add(key)
else:
for vitem in value:
getsetfromdict(importtypes,vitem).add(key)
return importtypes
