mirror of http://CODE.RHODECODE.COM/u/O/O/O
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
488 lines
20 KiB
488 lines
20 KiB
import bpy, os |
|
import numpy as np |
|
import mathutils |
|
from mathutils import Vector |
|
from math import pi |
|
from bpy.types import ( |
|
Operator, |
|
Panel, |
|
PropertyGroup, |
|
) |
|
from bpy.props import ( |
|
BoolProperty, |
|
EnumProperty, |
|
FloatProperty, |
|
IntProperty, |
|
StringProperty, |
|
PointerProperty |
|
) |
|
from .utils import * |
|
|
|
def change_speed_mode(self, context): |
|
props = context.scene.tissue_gcode |
|
if props.previous_speed_mode != props.speed_mode: |
|
if props.speed_mode == 'SPEED': |
|
props.speed = props.feed/60 |
|
props.speed_vertical = props.feed_vertical/60 |
|
props.speed_horizontal = props.feed_horizontal/60 |
|
else: |
|
props.feed = props.speed*60 |
|
props.feed_vertical = props.speed_vertical*60 |
|
props.feed_horizontal = props.speed_horizontal*60 |
|
props.previous_speed_mode == props.speed_mode |
|
return |
|
|
|
class tissue_gcode_prop(PropertyGroup): |
|
last_e : FloatProperty(name="Pull", default=5.0, min=0, soft_max=10) |
|
path_length : FloatProperty(name="Pull", default=5.0, min=0, soft_max=10) |
|
|
|
folder : StringProperty( |
|
name="File", default="", subtype='FILE_PATH', |
|
description = 'Destination folder.\nIf missing, the file folder will be used' |
|
) |
|
pull : FloatProperty( |
|
name="Pull", default=5.0, min=0, soft_max=10, |
|
description='Pull material before lift' |
|
) |
|
push : FloatProperty( |
|
name="Push", default=5.0, min=0, soft_max=10, |
|
description='Push material before start extruding' |
|
) |
|
dz : FloatProperty( |
|
name="dz", default=2.0, min=0, soft_max=20, |
|
description='Z movement for lifting the nozzle before travel' |
|
) |
|
flow_mult : FloatProperty( |
|
name="Flow Mult", default=1.0, min=0, soft_max=3, |
|
description = 'Flow multiplier.\nUse a single value or a list of values for changing it during the printing path' |
|
) |
|
feed : IntProperty( |
|
name="Feed Rate (F)", default=3600, min=0, soft_max=20000, |
|
description='Printing speed' |
|
) |
|
feed_horizontal : IntProperty( |
|
name="Feed Horizontal", default=7200, min=0, soft_max=20000, |
|
description='Travel speed' |
|
) |
|
feed_vertical : IntProperty( |
|
name="Feed Vertical", default=3600, min=0, soft_max=20000, |
|
description='Lift movements speed' |
|
) |
|
|
|
speed : IntProperty( |
|
name="Speed", default=60, min=0, soft_max=100, |
|
description='Printing speed' |
|
) |
|
speed_horizontal : IntProperty( |
|
name="Travel", default=120, min=0, soft_max=200, |
|
description='Travel speed' |
|
) |
|
speed_vertical : IntProperty( |
|
name="Z-Lift", default=60, min=0, soft_max=200, |
|
description='Lift movements speed' |
|
) |
|
|
|
esteps : FloatProperty( |
|
name="E Steps/Unit", default=5, min=0, soft_max=100) |
|
start_code : StringProperty( |
|
name="Start", default='', description = 'Text block for starting code' |
|
) |
|
end_code : StringProperty( |
|
name="End", default='', description = 'Text block for ending code' |
|
) |
|
auto_sort_layers : BoolProperty( |
|
name="Auto Sort Layers", default=True, |
|
description = 'Sort layers according to the Z of the median point' |
|
) |
|
auto_sort_points : BoolProperty( |
|
name="Auto Sort Points", default=False, |
|
description = 'Shift layer points trying to automatically reduce needed travel movements' |
|
) |
|
close_all : BoolProperty( |
|
name="Close Shapes", default=False, |
|
description = 'Repeat the starting point at the end of the vertices list for each layer' |
|
) |
|
nozzle : FloatProperty( |
|
name="Nozzle", default=0.4, min=0, soft_max=10, |
|
description='Nozzle diameter' |
|
) |
|
layer_height : FloatProperty( |
|
name="Layer Height", default=0.1, min=0, soft_max=10, |
|
description = 'Average layer height, needed for a correct extrusion' |
|
) |
|
filament : FloatProperty( |
|
name="Filament (\u03A6)", default=1.75, min=0, soft_max=120, |
|
description='Filament (or material container) diameter' |
|
) |
|
|
|
gcode_mode : EnumProperty(items=[ |
|
("CONT", "Continuous", ""), |
|
("RETR", "Retraction", "") |
|
], default='CONT', name="Mode", |
|
description = 'If retraction is used, then each separated list of vertices\nwill be considered as a different layer' |
|
) |
|
speed_mode : EnumProperty(items=[ |
|
("SPEED", "Speed (mm/s)", ""), |
|
("FEED", "Feed (mm/min)", "") |
|
], default='SPEED', name="Speed Mode", |
|
description = 'Speed control mode', |
|
update = change_speed_mode |
|
) |
|
previous_speed_mode : StringProperty( |
|
name="previous_speed_mode", default='', description = '' |
|
) |
|
retraction_mode : EnumProperty(items=[ |
|
("FIRMWARE", "Firmware", ""), |
|
("GCODE", "Gcode", "") |
|
], default='GCODE', name="Retraction Mode", |
|
description = 'If firmware retraction is used, then the retraction parameters will be controlled by the printer' |
|
) |
|
animate : BoolProperty( |
|
name="Animate", default=False, |
|
description = 'Show print progression according to current frame' |
|
) |
|
|
|
|
|
class TISSUE_PT_gcode_exporter(Panel): |
|
bl_category = "Tissue Gcode" |
|
bl_space_type = "VIEW_3D" |
|
bl_region_type = "UI" |
|
#bl_space_type = 'PROPERTIES' |
|
#bl_region_type = 'WINDOW' |
|
#bl_context = "data" |
|
bl_label = "Tissue Gcode Export" |
|
#bl_options = {'DEFAULT_CLOSED'} |
|
|
|
@classmethod |
|
def poll(cls, context): |
|
try: return context.object.type in ('CURVE','MESH') |
|
except: return False |
|
|
|
def draw(self, context): |
|
props = context.scene.tissue_gcode |
|
|
|
#addon = context.user_preferences.addons.get(sverchok.__name__) |
|
#over_sized_buttons = addon.preferences.over_sized_buttons |
|
layout = self.layout |
|
col = layout.column(align=True) |
|
row = col.row() |
|
row.prop(props, 'folder', toggle=True, text='') |
|
col = layout.column(align=True) |
|
row = col.row() |
|
row.prop(props, 'gcode_mode', expand=True, toggle=True) |
|
#col = layout.column(align=True) |
|
col = layout.column(align=True) |
|
col.label(text="Extrusion:", icon='MOD_FLUIDSIM') |
|
#col.prop(self, 'esteps') |
|
col.prop(props, 'filament') |
|
col.prop(props, 'nozzle') |
|
col.prop(props, 'layer_height') |
|
col.separator() |
|
col.label(text="Speed (Feed Rate F):", icon='DRIVER') |
|
col.prop(props, 'speed_mode', text='') |
|
speed_prefix = 'feed' if props.speed_mode == 'FEED' else 'speed' |
|
col.prop(props, speed_prefix, text='Print') |
|
if props.gcode_mode == 'RETR': |
|
col.prop(props, speed_prefix + '_vertical', text='Z Lift') |
|
col.prop(props, speed_prefix + '_horizontal', text='Travel') |
|
col.separator() |
|
if props.gcode_mode == 'RETR': |
|
col = layout.column(align=True) |
|
col.label(text="Retraction Mode:", icon='NOCURVE') |
|
row = col.row() |
|
row.prop(props, 'retraction_mode', expand=True, toggle=True) |
|
if props.retraction_mode == 'GCODE': |
|
col.separator() |
|
col.label(text="Retraction:", icon='PREFERENCES') |
|
col.prop(props, 'pull', text='Retraction') |
|
col.prop(props, 'dz', text='Z Hop') |
|
col.prop(props, 'push', text='Preload') |
|
col.separator() |
|
#col.label(text="Layers options:", icon='ALIGN_JUSTIFY') |
|
col.separator() |
|
col.prop(props, 'auto_sort_layers', text="Sort Layers (Z)") |
|
col.prop(props, 'auto_sort_points', text="Sort Points (XY)") |
|
#col.prop(props, 'close_all') |
|
col.separator() |
|
col.label(text='Custom Code:', icon='TEXT') |
|
col.prop_search(props, 'start_code', bpy.data, 'texts') |
|
col.prop_search(props, 'end_code', bpy.data, 'texts') |
|
col.separator() |
|
row = col.row(align=True) |
|
row.scale_y = 2.0 |
|
row.operator('scene.tissue_gcode_export') |
|
#col.separator() |
|
#col.prop(props, 'animate', icon='TIME') |
|
|
|
|
|
class tissue_gcode_export(Operator): |
|
bl_idname = "scene.tissue_gcode_export" |
|
bl_label = "Export Gcode" |
|
bl_description = ("Export selected curve object as Gcode file") |
|
bl_options = {'REGISTER', 'UNDO'} |
|
|
|
@classmethod |
|
def poll(cls, context): |
|
try: |
|
return context.object.type in ('CURVE', 'MESH') |
|
except: |
|
return False |
|
|
|
def execute(self, context): |
|
scene = context.scene |
|
props = scene.tissue_gcode |
|
# manage data |
|
if props.speed_mode == 'SPEED': |
|
props.feed = props.speed*60 |
|
props.feed_vertical = props.speed_vertical*60 |
|
props.feed_horizontal = props.speed_horizontal*60 |
|
feed = props.feed |
|
feed_v = props.feed_vertical |
|
feed_h = props.feed_horizontal |
|
layer = props.layer_height |
|
flow_mult = props.flow_mult |
|
#if context.object.type != 'CURVE': |
|
# self.report({'ERROR'}, 'Please select a Curve object') |
|
# return {'CANCELLED'} |
|
ob = context.object |
|
matr = ob.matrix_world |
|
if ob.type == 'MESH': |
|
dg = context.evaluated_depsgraph_get() |
|
mesh = ob.evaluated_get(dg).data |
|
edges = [list(e.vertices) for e in mesh.edges] |
|
verts = [v.co for v in mesh.vertices] |
|
ordered_verts = find_curves(edges, len(mesh.vertices)) |
|
ob = curve_from_pydata(verts, ordered_verts, name='__temp_curve__', merge_distance=0.1, set_active=False) |
|
|
|
vertices = [[matr @ p.co.xyz for p in s.points] for s in ob.data.splines] |
|
cyclic_u = [s.use_cyclic_u for s in ob.data.splines] |
|
|
|
if ob.name == '__temp_curve__': bpy.data.objects.remove(ob) |
|
|
|
if len(vertices) == 1: props.gcode_mode = 'CONT' |
|
export = True |
|
|
|
# open file |
|
if(export): |
|
if props.folder == '': |
|
folder = '//' + os.path.splitext(bpy.path.basename(bpy.context.blend_data.filepath))[0] |
|
else: |
|
folder = props.folder |
|
if '.gcode' not in folder: folder += '.gcode' |
|
path = bpy.path.abspath(folder) |
|
file = open(path, 'w') |
|
try: |
|
for line in bpy.data.texts[props.start_code].lines: |
|
file.write(line.body + '\n') |
|
except: |
|
pass |
|
|
|
#if props.gcode_mode == 'RETR': |
|
|
|
# sort layers (Z) |
|
if props.auto_sort_layers: |
|
sorted_verts = [] |
|
for curve in vertices: |
|
# mean z |
|
listz = [v[2] for v in curve] |
|
meanz = np.mean(listz) |
|
# store curve and meanz |
|
sorted_verts.append((curve, meanz)) |
|
vertices = [data[0] for data in sorted(sorted_verts, key=lambda height: height[1])] |
|
|
|
# sort vertices (XY) |
|
if props.auto_sort_points: |
|
# curves median point |
|
median_points = [np.mean(verts,axis=0) for verts in vertices] |
|
|
|
# chose starting point for each curve |
|
for j, curve in enumerate(vertices): |
|
# for closed curves finds the best starting point |
|
if cyclic_u[j]: |
|
# create kd tree |
|
kd = mathutils.kdtree.KDTree(len(curve)) |
|
for i, v in enumerate(curve): |
|
kd.insert(v, i) |
|
kd.balance() |
|
|
|
if props.gcode_mode == 'RETR': |
|
if j==0: |
|
# close to next two curves median point |
|
co_find = np.mean(median_points[j+1:j+3],axis=0) |
|
elif j < len(vertices)-1: |
|
co_find = np.mean([median_points[j-1],median_points[j+1]],axis=0) |
|
else: |
|
co_find = np.mean(median_points[j-2:j],axis=0) |
|
#flow_mult[j] = flow_mult[j][index:]+flow_mult[j][:index] |
|
#layer[j] = layer[j][index:]+layer[j][:index] |
|
else: |
|
if j==0: |
|
# close to next two curves median point |
|
co_find = np.mean(median_points[j+1:j+3],axis=0) |
|
else: |
|
co_find = vertices[j-1][-1] |
|
co, index, dist = kd.find(co_find) |
|
vertices[j] = vertices[j][index:]+vertices[j][:index+1] |
|
else: |
|
if j > 0: |
|
p0 = curve[0] |
|
p1 = curve[-1] |
|
last = vertices[j-1][-1] |
|
d0 = (last-p0).length |
|
d1 = (last-p1).length |
|
if d1 < d0: vertices[j].reverse() |
|
|
|
|
|
|
|
''' |
|
# close shapes |
|
if props.close_all: |
|
for i in range(len(vertices)): |
|
vertices[i].append(vertices[i][0]) |
|
#flow_mult[i].append(flow_mult[i][0]) |
|
#layer[i].append(layer[i][0]) |
|
''' |
|
# calc bounding box |
|
min_corner = np.min(vertices[0],axis=0) |
|
max_corner = np.max(vertices[0],axis=0) |
|
for i in range(1,len(vertices)): |
|
eval_points = vertices[i] + [min_corner] |
|
min_corner = np.min(eval_points,axis=0) |
|
eval_points = vertices[i] + [max_corner] |
|
max_corner = np.max(eval_points,axis=0) |
|
|
|
# initialize variables |
|
e = 0 |
|
last_vert = Vector((0,0,0)) |
|
maxz = 0 |
|
path_length = 0 |
|
travel_length = 0 |
|
|
|
printed_verts = [] |
|
printed_edges = [] |
|
travel_verts = [] |
|
travel_edges = [] |
|
|
|
# write movements |
|
for i in range(len(vertices)): |
|
curve = vertices[i] |
|
first_id = len(printed_verts) |
|
for j in range(len(curve)): |
|
v = curve[j] |
|
v_flow_mult = flow_mult#[i][j] |
|
v_layer = layer#[i][j] |
|
|
|
# record max z |
|
maxz = np.max((maxz,v[2])) |
|
#maxz = max(maxz,v[2]) |
|
|
|
# first point of the gcode |
|
if i == j == 0: |
|
printed_verts.append(v) |
|
if(export): |
|
file.write('G92 E0 \n') |
|
params = v[:3] + (feed,) |
|
to_write = 'G1 X{0:.4f} Y{1:.4f} Z{2:.4f} F{3:.0f}\n'.format(*params) |
|
file.write(to_write) |
|
else: |
|
# start after retraction |
|
if j == 0 and props.gcode_mode == 'RETR': |
|
if(export): |
|
params = v[:2] + (maxz+props.dz,) + (feed_h,) |
|
to_write = 'G1 X{0:.4f} Y{1:.4f} Z{2:.4f} F{3:.0f}\n'.format(*params) |
|
file.write(to_write) |
|
params = v[:3] + (feed_v,) |
|
to_write = 'G1 X{0:.4f} Y{1:.4f} Z{2:.4f} F{3:.0f}\n'.format(*params) |
|
file.write(to_write) |
|
to_write = 'G1 F{:.0f}\n'.format(feed) |
|
file.write(to_write) |
|
if props.retraction_mode == 'GCODE': |
|
e += props.push |
|
file.write( 'G1 E' + format(e, '.4f') + '\n') |
|
else: |
|
file.write('G11\n') |
|
printed_verts.append((v[0], v[1], maxz+props.dz)) |
|
travel_edges.append((len(printed_verts)-1, len(printed_verts)-2)) |
|
travel_length += (Vector(printed_verts[-1])-Vector(printed_verts[-2])).length |
|
printed_verts.append(v) |
|
travel_edges.append((len(printed_verts)-1, len(printed_verts)-2)) |
|
travel_length += maxz+props.dz - v[2] |
|
# regular extrusion |
|
else: |
|
printed_verts.append(v) |
|
v1 = Vector(v) |
|
v0 = Vector(curve[j-1]) |
|
dist = (v1-v0).length |
|
area = v_layer * props.nozzle + pi*(v_layer/2)**2 # rectangle + circle |
|
cylinder = pi*(props.filament/2)**2 |
|
flow = area / cylinder * (0 if j == 0 else 1) |
|
e += dist * v_flow_mult * flow |
|
params = v[:3] + (e,) |
|
if(export): |
|
to_write = 'G1 X{0:.4f} Y{1:.4f} Z{2:.4f} E{3:.4f}\n'.format(*params) |
|
file.write(to_write) |
|
path_length += dist |
|
printed_edges.append([len(printed_verts)-1, len(printed_verts)-2]) |
|
if props.gcode_mode == 'RETR': |
|
v0 = Vector(curve[-1]) |
|
if props.close_all and False: |
|
#printed_verts.append(v0) |
|
printed_edges.append([len(printed_verts)-1, first_id]) |
|
|
|
v1 = Vector(curve[0]) |
|
dist = (v0-v1).length |
|
area = v_layer * props.nozzle + pi*(v_layer/2)**2 # rectangle + circle |
|
cylinder = pi*(props.filament/2)**2 |
|
flow = area / cylinder |
|
e += dist * v_flow_mult * flow |
|
params = v1[:3] + (e,) |
|
if(export): |
|
to_write = 'G1 X{0:.4f} Y{1:.4f} Z{2:.4f} E{3:.4f}\n'.format(*params) |
|
file.write(to_write) |
|
path_length += dist |
|
v0 = v1 |
|
if i < len(vertices)-1: |
|
if(export): |
|
if props.retraction_mode == 'GCODE': |
|
e -= props.pull |
|
file.write('G0 E' + format(e, '.4f') + '\n') |
|
else: |
|
file.write('G10\n') |
|
params = v0[:2] + (maxz+props.dz,) + (feed_v,) |
|
to_write = 'G1 X{0:.4f} Y{1:.4f} Z{2:.4f} F{3:.0f}\n'.format(*params) |
|
file.write(to_write) |
|
printed_verts.append(v0.to_tuple()) |
|
printed_verts.append((v0.x, v0.y, maxz+props.dz)) |
|
travel_edges.append((len(printed_verts)-1, len(printed_verts)-2)) |
|
travel_length += maxz+props.dz - v0.z |
|
if(export): |
|
# end code |
|
try: |
|
for line in bpy.data.texts[props.end_code].lines: |
|
file.write(line.body + '\n') |
|
except: |
|
pass |
|
file.close() |
|
print("Saved gcode to " + path) |
|
bb = list(min_corner) + list(max_corner) |
|
info = 'Bounding Box:\n' |
|
info += '\tmin\tX: {0:.1f}\tY: {1:.1f}\tZ: {2:.1f}\n'.format(*bb) |
|
info += '\tmax\tX: {3:.1f}\tY: {4:.1f}\tZ: {5:.1f}\n'.format(*bb) |
|
info += 'Extruded Filament: ' + format(e, '.2f') + '\n' |
|
info += 'Extruded Volume: ' + format(e*pi*(props.filament/2)**2, '.2f') + '\n' |
|
info += 'Printed Path Length: ' + format(path_length, '.2f') + '\n' |
|
info += 'Travel Length: ' + format(travel_length, '.2f') |
|
''' |
|
# animate |
|
if scene.animate: |
|
scene = bpy.context.scene |
|
try: |
|
param = (scene.frame_current - scene.frame_start)/(scene.frame_end - scene.frame_start) |
|
except: |
|
param = 1 |
|
last_vert = max(int(param*len(printed_verts)),1) |
|
printed_verts = printed_verts[:last_vert] |
|
printed_edges = [e for e in printed_edges if e[0] < last_vert and e[1] < last_vert] |
|
travel_edges = [e for e in travel_edges if e[0] < last_vert and e[1] < last_vert] |
|
''' |
|
return {'FINISHED'}
|
|
|