def render(self, filename):
self.scheduler.start()
# create globals
integrator = self.pmgr.create({'type': self.setup['integrator']})
emitter = self.pmgr.create({'type': self.setup['emitter']})
sensor = self.pmgr.create({
'type': self.setup['sensor'],
'film': {
'type': self.setup['film'],
'width': self.setup['width'],
'height': self.setup['height'],
'pixelFormat': self.setup['pixelFormat'],
'exposure': self.setup['exposure'],
'banner': self.setup['banner']
},
'sampler': {
'type': self.setup['sampler'],
'sampleCount': self.setup['sampleCount']
},
'fov': self.setup['fov'],
})
scene = Scene()
scene.addChild(integrator)
scene.addChild(emitter)
scene.addChild(sensor)
for mesh in self.mesh:
scene.addChild(mesh)
scene.configure()
scene.initialize() # needed to force build of kd-tree
transformCurr = Transform.lookAt(self.setup['eye'],
self.setup['target'],
self.setup['camera_up'])
sensor.setWorldTransform(transformCurr)
scene.setDestinationFile(filename)
job = RenderJob('job', scene, self.queue)
job.start()
self.queue.waitLeft(0)
self.queue.join()
self.scheduler.stop()
def construct_simple_scene(scene_objects, sensor) -> Scene:
"""
Construct a simple scene containing given objects and using the given sensor. Uses the path integrator and constant
emitter
:param scene_objects: All scene child objects to add
:param sensor: The mitsuba sensor definition to use for this scene
:return: The scene created, already configured and initialized
"""
pmgr = PluginManager.getInstance()
integrator = pmgr.create({'type': 'path'})
emitter = pmgr.create({'type': 'constant'})
scene = Scene()
scene.addChild(integrator)
scene.addChild(emitter)
scene.addChild(sensor)
for obj in scene_objects:
scene.addChild(obj)
scene.configure()
scene.initialize()
return scene
def init_scene():
pmgr = PluginManager.getInstance()
scene = Scene()
scene.setDestinationFile('renderedResult')
camera_pos = Vector(0, 0.0, -12)
model_pos = Vector(0.5, -0.5, -2.0)
a = MitsubaShape(shape_type=MitsubaShape.PLY_TYPE, to_world=Transform.translate(model_pos), filename=mesh_file)
b = MitsubaShape(shape_type=MitsubaShape.CUBE, to_world=Transform.translate(model_pos))
integrator = create_integrator(RenderTargetType.DEPTH, hide_emitters=False)
#integrator = create_integrator(IntegratorType.DIRECT, hide_emitters=True)
#integrator = create_integrator(RenderTargetType.NORMAL, hide_emitters=True)
print(integrator.config)
sampler = Sampler(SamplerType.HALTON, num_samples=num_samples)
film = Film(FilmType.HDR, image_width, image_height)
sensor = Sensor(SensorType.PERSPECTIVE, sampler=sampler, film=film, to_world=Transform.translate(camera_pos))
scene_config = {
'type' : 'scene',
'integrator' : {
'type' : 'multichannel',
# 'a': {
# 'type' : 'path'
# },
# 'b': {
# 'type' : 'field',
# 'field' : 'distance',
# 'undefined': 0.0
# },
'c': {
'type' : 'field',
'field' : 'distance',
'undefined': 0.0
}
},
'sphere' : {
'type' : 'sphere',
'bsdf' : {
'type' : 'dielectric',
'reflectance' : Spectrum(0.4)
}
},
'envmap' : {
'type' : 'sunsky',
'albedo' : Spectrum(0.5)
},
'sensor' : {
'type' : 'perspective',
'toWorld' : Transform.translate(Vector(0, 0, 0)),
'sampler' : {
'type' : 'halton',
'sampleCount' : 64
},
'film' : {
'type' : 'ldrfilm',
'width' : 500,
'height' : 500,
'pixelFormat': "rgb",
'channelNames': "normal"
}
},
}
# # Add a shape
# scene.addChild(pmgr.create({
# 'type' : 'sphere',
# 'center' : Point(0, 0, 0),
# 'radius' : 1.0,
# 'bsdf' : {
# 'type' : 'diffuse',
# 'reflectance' : Spectrum(0.4)
# }
# }))
scene.addChild(pmgr.create(scene_config))
scene.configure()
# scene_config['cube'] = create_object('cube', Transform.translate(model_pos), bsdf=create_bsdf())
# scene_node = pmgr.create(scene_config)
# scene.addChild(scene_node)
# scene.configure()
scene.initialize()
sceneResID = scheduler.registerResource(scene)
num_views = 1
step_size = 360/(num_views)
for i in range(num_views):
destination = 'results/result_%03i' % i
# Create a shallow copy of the scene so that the queue can tell apart the two
# rendering processes. This takes almost no extra memory
newScene = mitsuba.render.Scene(scene)
pmgr = PluginManager.getInstance()
newSensor = pmgr.createObject(scene.getSensor().getProperties())
# <change the position of 'newSensor' here>
rotationCur = Transform.rotate(Vector(0, 1, 0), i*step_size)
new_pos = rotationCur*camera_pos
new_transform = Transform.lookAt(Point(new_pos), Point(0, 0, 0), Vector(0, 1, 0))
newSensor.setWorldTransform(new_transform)
newFilm = pmgr.createObject(scene.getFilm().getProperties())
newFilm.configure()
newSensor.addChild(newFilm)
newSensor.configure()
newScene.addSensor(newSensor)
newScene.setSensor(newSensor)
newScene.setSampler(scene.getSampler())
newScene.setDestinationFile(destination)
# Create a render job and insert it into the queue. Note how the resource
# ID of the original scene is provided to avoid sending the full scene
# contents over the network multiple times.
job = RenderJob('myRenderJob' + str(i), scene, queue, sceneResID)
job.start()
# Wait for all jobs to finish and release resources
queue.waitLeft(0)
def init_scene():
pmgr = PluginManager.getInstance()
scene = Scene()
scene.setDestinationFile('renderedResult')
camera_pos = Vector(0, 0.0, -12)
model_pos = Vector(0.5, -0.5, -2.0)
a = MitsubaShape(shape_type=MitsubaShape.PLY_TYPE, to_world=Transform.translate(model_pos), filename='/media/adrian/Data/Datasets/train/02691156/model_0000003.obj')
b = MitsubaShape(shape_type=MitsubaShape.CUBE, to_world=Transform.translate(model_pos))
integrator = Integrator(Integrator.DIRECT, hide_emitters=True)
sampler = Sampler(SamplerType.HALTON, num_samples=num_samples)
film = Film(FilmType.LDR, image_width, image_height)
sensor = Sensor(SensorType.PERSPECTIVE, sampler=sampler, film=film, to_world=Transform.translate(camera_pos))
scene_config = {
'type' : 'scene',
'a': a.config,
'envmap' : {
'type' : 'sunsky',
'hour' : 12.0,
'albedo' : Spectrum(1.0),
'samplingWeight' : 1.0,
},
# 'envmap' : {
# 'type' : 'constant',
# #'hour' : 10.0,
# 'radiance' : Spectrum(1.0),
# 'samplingWeight' : 0.5
# },
# 'integrator' : {
# 'type' : 'multichannel',
# 'depth' : {
# 'type' : 'field',
# 'field' : 'distance'
# },
# },
'integrator' : integrator.config,
'sensor' : sensor.config
}
# scene_config['cube'] = create_object('cube', Transform.translate(model_pos), bsdf=create_bsdf())
scene_node = pmgr.create(scene_config)
scene.addChild(scene_node)
scene.configure()
scene.initialize()
sceneResID = scheduler.registerResource(scene)
num_views = 6
step_size = 360/(num_views)
for i in range(num_views):
destination = 'results/result_%03i' % i
# Create a shallow copy of the scene so that the queue can tell apart the two
# rendering processes. This takes almost no extra memory
newScene = mitsuba.render.Scene(scene)
pmgr = PluginManager.getInstance()
newSensor = pmgr.createObject(scene.getSensor().getProperties())
# <change the position of 'newSensor' here>
rotationCur = Transform.rotate(Vector(0, 1, 0), i*step_size)
new_pos = rotationCur*camera_pos
new_transform = Transform.lookAt(Point(new_pos), Point(0, 0, 0), Vector(0, 1, 0))
newSensor.setWorldTransform(new_transform)
newFilm = pmgr.createObject(scene.getFilm().getProperties())
newFilm.configure()
newSensor.addChild(newFilm)
newSensor.configure()
newScene.addSensor(newSensor)
newScene.setSensor(newSensor)
newScene.setSampler(scene.getSampler())
newScene.setDestinationFile(destination)
# Create a render job and insert it into the queue. Note how the resource
# ID of the original scene is provided to avoid sending the full scene
# contents over the network multiple times.
job = RenderJob('myRenderJob' + str(i), newScene, queue, sceneResID)
job.start()
# Wait for all jobs to finish and release resources
queue.waitLeft(0)
