slBlenderVirtualInfrastructure.py

import bpy
import math
import sys
import os
import json


def getTuple(json, angle = False, default = 0):
	if json == None:
		return (default, default, default)

	if angle:
		return (math.radians(json[0]), math.radians(json[1]), math.radians(json[2]))
	
	return tuple(json)

def getNumber(json):
	if json == None:
		return 0
	
	return json

if __name__ == "__main__":
	argv = sys.argv
	argv = argv[argv.index("--") + 1:] 

	patternPath = os.path.abspath(argv[0])
	capturePath = os.path.abspath(argv[1])
	outputPath = os.path.abspath(argv[2])
	cameraWidth = int(argv[3])
	cameraHeight = int(argv[4])

	jsonData = json.loads(open(os.path.abspath("./slVirtualScene.json")).read())

	bpy.ops.scene.new(type='EMPTY')
        	
	for sceneObjectJson in jsonData['objects']:
		objectType = sceneObjectJson.get('type')

		location = getTuple(sceneObjectJson.get('location'))
		rotation = getTuple(sceneObjectJson.get('rotation'), True)

		print(rotation)
		sceneObject = None

		if objectType == "plane":
			bpy.ops.mesh.primitive_plane_add(location=location, rotation=rotation)
		elif objectType == "sphere":
			bpy.ops.mesh.primitive_uv_sphere_add(location=location, rotation=rotation)
		elif objectType == "cube":
			bpy.ops.mesh.primitive_cube_add(location=location, rotation=rotation)
		
		sceneObject = bpy.context.object
		sceneObjectData = sceneObject.data

		sceneObject.scale = getTuple(sceneObjectJson.get('scale'), default=1)
		#sceneObject.rotation_euler = getTuple(sceneObjectJson.get('rotation_euler'), True)


	#bpy.ops.import_scene.obj(filepath="/Users/ehendrikd/Downloads/GI Joe - The Rock/TheRock2.obj")
	#obj = bpy.context.object
	#obj.scale = (0.2, 0.2, 0.2)
		

	projectorObjectJson = jsonData['projector']

	bpy.ops.object.lamp_add(type='SPOT', location=getTuple(projectorObjectJson['location']), rotation=getTuple(projectorObjectJson['rotation'], True))


	projectorObject = bpy.context.object
	projectorObject.delta_rotation_euler = getTuple(projectorObjectJson['rotation_euler'], True)

	projector = projectorObject.data

	projector.distance = getNumber(projectorObjectJson['focal_length'])
	projector.spot_size = math.radians(getNumber(projectorObjectJson['projection_size']))
	projector.use_square = True
	projector.show_cone = True
	projector.energy = getNumber(projectorObjectJson['intensity'])
	projector.falloff_type = 'CONSTANT'
	projector.spot_blend = 0

	image = bpy.data.images.load(patternPath)
	texture = bpy.data.textures.new('ColorTex', type = 'IMAGE')
	texture.image = image    
	projector.active_texture = texture
	projector.texture_slots[0].texture_coords = 'VIEW'

	#print(type(cTex))

	#realpath = os.path.expanduser('~/snippets/textures/color.png')
	#try:
	#    img = bpy.data.images.load(realpath)
	#except:
	#    raise NameError("Cannot load image %s" % realpath)
    
	cameraObjectJson = jsonData['camera']

	bpy.ops.object.camera_add(location=getTuple(cameraObjectJson['location']), rotation=getTuple(cameraObjectJson['rotation'], True))
	camera = bpy.context.object
	#camera.data.type = 'ORTHO'
	#camera.data.ortho_scale = 8.69
	bpy.context.scene.camera = camera


	bpy.context.scene.render.resolution_x = cameraWidth
	bpy.context.scene.render.resolution_y = cameraHeight
	#bpy.context.scene.render.resolution_x = 65536
	#bpy.context.scene.render.resolution_y = 100
	bpy.context.scene.render.resolution_percentage = 100

	bpy.context.scene.render.filepath = capturePath

	if jsonData['save_scene_file']:
		bpy.ops.wm.save_as_mainfile(filepath=outputPath)

	bpy.ops.render.render( write_still=True )