some more blender render engine stuff
This commit is contained in:
parent
6ed1d2a085
commit
4700f4dcc4
1 changed files with 49 additions and 9 deletions
|
@ -16,14 +16,39 @@ class BlenderEngine(Base3DEngine):
|
||||||
super().__init__(*args, **kwargs)
|
super().__init__(*args, **kwargs)
|
||||||
self.result = ''
|
self.result = ''
|
||||||
self._add_python('''
|
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):
|
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):
|
def add_ring(coords, minz, maxz):
|
||||||
if coords[0] == coords[-1]:
|
if coords[0] == coords[-1]:
|
||||||
coords = coords[:-1]
|
coords = coords[:-1]
|
||||||
if len(coords) < 3:
|
if len(coords) < 3:
|
||||||
raise ValueError('Ring with less than 3 points.')
|
raise ValueError('Ring with less than 3 points.')
|
||||||
|
|
||||||
|
# create ring
|
||||||
indices = tuple(range(len(coords)))
|
indices = tuple(range(len(coords)))
|
||||||
mesh = bpy.data.meshes.new(name='Test')
|
mesh = bpy.data.meshes.new(name='Test')
|
||||||
mesh.from_pydata(
|
mesh.from_pydata(
|
||||||
|
@ -31,9 +56,22 @@ class BlenderEngine(Base3DEngine):
|
||||||
tuple(zip(indices, indices[1:]+(0, ))),
|
tuple(zip(indices, indices[1:]+(0, ))),
|
||||||
(indices, ),
|
(indices, ),
|
||||||
)
|
)
|
||||||
|
|
||||||
|
# add ring to scene
|
||||||
obj = bpy.data.objects.new('Test', mesh)
|
obj = bpy.data.objects.new('Test', mesh)
|
||||||
scene = bpy.context.scene
|
scene = bpy.context.scene
|
||||||
scene.objects.link(obj)
|
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
|
return obj
|
||||||
''')
|
''')
|
||||||
|
|
||||||
|
@ -70,18 +108,20 @@ class BlenderEngine(Base3DEngine):
|
||||||
restricted_spaces = unary_union((restricted_spaces_indoors, restricted_spaces_outdoors)) # noqa
|
restricted_spaces = unary_union((restricted_spaces_indoors, restricted_spaces_outdoors)) # noqa
|
||||||
|
|
||||||
for altitudearea in geoms.altitudeareas:
|
for altitudearea in geoms.altitudeareas:
|
||||||
self._add_polygon(altitudearea.geometry.geom)
|
self._add_polygon(altitudearea.geometry.geom, min_altitude-1, altitudearea.altitude)
|
||||||
break
|
|
||||||
|
|
||||||
break
|
break
|
||||||
|
|
||||||
def _add_polygon(self, geometry):
|
def _add_polygon(self, geometry, minz, maxz):
|
||||||
for polygon in assert_multipolygon(geometry):
|
for polygon in assert_multipolygon(geometry):
|
||||||
self._add_python('add_polygon(exterior=%(exterior)r, interiors=%(interiors)r, minz=0, maxz=1)' % {
|
self._add_python(
|
||||||
'exterior': tuple(polygon.exterior.coords),
|
'add_polygon(exterior=%(exterior)r, interiors=%(interiors)r, minz=%(minz)f, maxz=%(maxz)f)' % {
|
||||||
'interiors': tuple(tuple(interior.coords) for interior in polygon.interiors),
|
'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):
|
def render(self, filename=None):
|
||||||
return self.result.encode()
|
return self.result.encode()
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue