mirror of
https://github.com/nophead/Mendel90.git
synced 2025-01-16 20:38:15 +01:00
48 lines
1.4 KiB
Python
Executable File
48 lines
1.4 KiB
Python
Executable File
#!/usr/bin/env python
|
|
|
|
from __future__ import print_function
|
|
import os
|
|
import sys
|
|
import shutil
|
|
import subprocess
|
|
from time import *
|
|
|
|
def render(machine, parts = None):
|
|
stl_dir = machine + os.sep + "stls"
|
|
render_dir = machine + os.sep + "render"
|
|
|
|
if os.path.isdir(render_dir):
|
|
if not parts:
|
|
shutil.rmtree(render_dir) #if doing them all clear the directory first
|
|
sleep(0.1)
|
|
os.makedirs(render_dir)
|
|
else:
|
|
os.makedirs(render_dir)
|
|
|
|
#
|
|
# List of individual part files
|
|
#
|
|
if parts:
|
|
stls = [i[:-4] for i in parts]
|
|
else:
|
|
stls = [i[:-4] for i in os.listdir(stl_dir) if i[-4:] == ".stl"]
|
|
#
|
|
# Add the multipart files
|
|
#
|
|
for i in os.listdir(stl_dir + os.sep + "printed"):
|
|
if i[-4:] == ".stl" and not i[:-4] in stls:
|
|
stls.append("printed" + os.sep + i[:-4])
|
|
|
|
for i in stls:
|
|
command = 'blender -b utils' + os.sep + 'render.blend -P utils' + os.sep + 'viz.py -- ' + \
|
|
stl_dir + os.sep + i + '.stl ' + render_dir + os.sep + i + '.png'
|
|
print(command)
|
|
subprocess.check_output(command.split())
|
|
|
|
if __name__ == '__main__':
|
|
if len(sys.argv) > 1:
|
|
render(sys.argv[1], sys.argv[2:])
|
|
else:
|
|
print("usage: render dibond|mendel|sturdy|huxley|your_machine, [part.stl ...]")
|
|
sys.exit(1)
|