some more blender render engine stuff

This commit is contained in:
Laura Klünder 2018-12-04 01:15:52 +01:00
parent 6ed1d2a085
commit 4700f4dcc4

View file

@ -16,14 +16,39 @@ class BlenderEngine(Base3DEngine):
super().__init__(*args, **kwargs)
self.result = ''
self._add_python('''
import bpy
def deselect_all():
bpy.ops.object.select_all(action='DESELECT')
def select_object(obj):
deselect_all()
obj.select = True
bpy.context.scene.objects.active = obj
def add_polygon(exterior, interiors, minz, maxz):
add_ring(exterior, minz, maxz)
bpy.ops.object.mode_set(mode='OBJECT')
deselect_all()
exterior = add_ring(exterior, minz, maxz)
for interior_coords in interiors:
interior = add_ring(interior_coords, minz-1, maxz+1)
select_object(exterior)
bpy.ops.object.modifier_add(type='BOOLEAN')
mod = exterior.modifiers
mod[0].name = 'Difference'
mod[0].operation = 'DIFFERENCE'
mod[0].object = interior
bpy.ops.object.modifier_apply(apply_as='DATA', modifier=mod[0].name)
select_object(interior)
bpy.ops.object.delete()
def add_ring(coords, minz, maxz):
if coords[0] == coords[-1]:
coords = coords[:-1]
if len(coords) < 3:
raise ValueError('Ring with less than 3 points.')
# create ring
indices = tuple(range(len(coords)))
mesh = bpy.data.meshes.new(name='Test')
mesh.from_pydata(
@ -31,9 +56,22 @@ class BlenderEngine(Base3DEngine):
tuple(zip(indices, indices[1:]+(0, ))),
(indices, ),
)
# add ring to scene
obj = bpy.data.objects.new('Test', mesh)
scene = bpy.context.scene
scene.objects.link(obj)
# extrude it
select_object(obj)
bpy.ops.object.mode_set(mode='EDIT')
bpy.ops.mesh.select_mode(type='FACE')
bpy.ops.mesh.select_all(action='SELECT')
bpy.ops.mesh.extrude_region_move(
TRANSFORM_OT_translate={'value': (0, 0, maxz-minz)}
)
bpy.ops.object.mode_set(mode='OBJECT')
obj.select = False
return obj
''')
@ -70,18 +108,20 @@ class BlenderEngine(Base3DEngine):
restricted_spaces = unary_union((restricted_spaces_indoors, restricted_spaces_outdoors)) # noqa
for altitudearea in geoms.altitudeareas:
self._add_polygon(altitudearea.geometry.geom)
break
self._add_polygon(altitudearea.geometry.geom, min_altitude-1, altitudearea.altitude)
break
def _add_polygon(self, geometry):
def _add_polygon(self, geometry, minz, maxz):
for polygon in assert_multipolygon(geometry):
self._add_python('add_polygon(exterior=%(exterior)r, interiors=%(interiors)r, minz=0, maxz=1)' % {
self._add_python(
'add_polygon(exterior=%(exterior)r, interiors=%(interiors)r, minz=%(minz)f, maxz=%(maxz)f)' % {
'exterior': tuple(polygon.exterior.coords),
'interiors': tuple(tuple(interior.coords) for interior in polygon.interiors),
})
continue
'minz': minz/1000,
'maxz': maxz/1000,
}
)
def render(self, filename=None):
return self.result.encode()