Skip to content

Instantly share code, notes, and snippets.

@tamask
Created July 28, 2019 20:29
Show Gist options
  • Save tamask/1fa1cca7b2c68bcd1f3a2f3fdc61b378 to your computer and use it in GitHub Desktop.
Save tamask/1fa1cca7b2c68bcd1f3a2f3fdc61b378 to your computer and use it in GitHub Desktop.
Blender 2.80 plotter
import bpy
import bmesh
NAME = '(Plot)'
def line(gen, name=NAME):
"""Plot arbitrary x,y values as line"""
with Plotter(name) as p:
for x, y in gen:
p.extend(x, y)
p.link(bpy.context.scene)
def plot(fn, res=1024, extents=1, name=NAME):
"""Plot y values as a function of x"""
with Plotter(name) as p:
for x in range(res * (extents * 2 + 1) + 1):
x -= res * extents
x = float(x) / float(res)
try:
y = fn(x)
except ZeroDivisionError:
p.clip()
continue
except ValueError as e:
if str(e) == 'math domain error':
p.clip()
continue
else:
raise e
p.extend(x, y)
p.link(bpy.context.scene)
class Plotter:
def __init__(self, name):
self.name = name
def __enter__(self):
if self.name in bpy.data.objects:
self.obj = bpy.data.objects[self.name]
self.data = self.obj.data
else:
self.data = bpy.data.meshes.new(self.name)
self.obj = bpy.data.objects.new(self.name, self.data)
self.mesh = bmesh.new()
self.mesh.from_mesh(self.data)
self.mesh.clear()
self.last = None
return self
def __exit__(self, t, v, e):
self.mesh.to_mesh(self.data)
self.mesh.free()
self.obj.data.update()
def link(self, scene):
if self.obj.name not in scene.objects:
scene.collection.objects.link(self.obj)
layer = bpy.context.view_layer
layer.update()
self.obj.select_set(True)
def extend(self, x, y):
vert = self.mesh.verts.new((x, 0, y))
if self.last is not None:
self.mesh.edges.new((self.last, vert))
self.last = vert
def clip(self):
self.last = None
__all__ = ('line', 'plot', 'Plotter')
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment