crown-engine/tools/pycrown/Compiler.py

# Copyright (c) 2013 Daniele Bartolini, Michele Rossi
# Copyright (c) 2012 Daniele Bartolini, Simone Boscaratto
#
# Permission is hereby granted, free of charge, to any person
# obtaining a copy of this software and associated documentation
# files (the "Software"), to deal in the Software without
# restriction, including without limitation the rights to use,
# copy, modify, merge, publish, distribute, sublicense, and/or sell
# copies of the Software, and to permit persons to whom the
# Software is furnished to do so, subject to the following
# conditions:
# 
# The above copyright notice and this permission notice shall be
# included in all copies or substantial portions of the Software.
# 
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
# EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES
# OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
# NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
# HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
# WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
# FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
# OTHER DEALINGS IN THE SOFTWARE.

import subprocess
import os

LUAJIT = "luajit-2.0.1"
BC_G = "bytecode-generator.lua"
TXT_C = "txt-compiler"
TGA_C = "tga-compiler"
LUA_C = "lua-compiler"
VS_C = "vs-compiler"
PS_C = "ps-compiler"
RES_H = "resource-hash"

ROOT_P = "--root-path"
DEST_P = "--dest-path"
RES_IN = "--resource-in"
SEED = "--seed"

# Compiler is an helper for compiling resources in a repository
# either one by one or in a whole
class Compiler:
	def __init__(self, repository, dest_path):
		self.m_repository = repository
		self.m_perfect_seed = -1
		self.m_dest_path = dest_path

	# Returns the destination path where the compiled
	# resources will be put
	def dest_path(self):
		return self.m_dest_path

	# Returns the perfect seed
	def perfect_seed(self):
		return self.m_perfect_seed

	# Computes a perfect seed for the resources contained
	# in the repository
	def compute_perfect_seed(self):
		# Obtain resources from repository
		resources = self.m_repository.all_resources()

		resource_hashes = []

		seed = 0

		# Calculate the hash resource by resource
		# Note: we can speed-up this process by modifying resource-hash to
		# calculate more than a hash at time...
		while True:
			for res in resources:
				p = subprocess.check_output([RES_H, RES_IN, res, SEED, str(seed)])
				resource_hashes.append(str(p))

			if len(resource_hashes) == len(set(resource_hashes)):
				self.m_perfect_seed = seed
				return

			resource_hashes.clear()
			seed = seed + 1;

	# Compiles all the texture resources in the repository
	def compile_textures(self):
		textures = self.m_repository.texture_resources();

		for res in textures:
			self.compile(res)

	# Compiles all the mesh resources in the repository
	def compile_meshes(self):
		meshes = self.m_repository.mesh_resources();

		for res in meshes:
			self.compile(res)

	# Compiles all the text resources in the repository
	def compile_texts(self):
		texts = self.m_repository.text_resources();

		for res in texts:
			self.compile(res)

	# Compiles all the script resources in the repository
	def compile_scripts(self):
		scripts = self.m_repository.script_resources();

		for res in scripts:
			self.compile(res)

	# Compiles all the vertex shader resources in the repository
	def compile_vertex_shaders(self):
		vss = self.m_repository.vertex_shader_resources();

		for res in vss:
			self.compile(res)

	# Compiles all the vertex shader resources in the repository
	def compile_pixel_shaders(self):
		pss = self.m_repository.pixel_shader_resources();

		for res in pss:
			self.compile(res)

	# Compiles all the resources in the repository
	def compile_all(self):
		print("Compiling textures...")
		self.compile_textures()

		print("Compiling meshes...")
		self.compile_meshes()

		print("Compiling texts...")
		self.compile_texts()

		print("Compiling scripts...")
		self.compile_scripts()

		print("Compiling vertex shaders...")
		self.compile_vertex_shaders()

		print("Compiling pixel shaders...")
		self.compile_pixel_shaders()

	# Compile a single resource from the repository
	def compile(self, resource):
		# Compute perfect seed if necessary
		if (self.m_perfect_seed == -1):
			self.compute_perfect_seed()

		root_path = self.m_repository.root_path()

		print(resource + " => ???")

		# Call appropriate compiler based on resource extension
		if resource.endswith('.txt'):
			p = subprocess.call([TXT_C, ROOT_P, root_path, DEST_P, self.m_dest_path, RES_IN, resource, SEED, str(self.m_perfect_seed)]);
		
		if resource.endswith('.tga'):
			p = subprocess.call([TGA_C, ROOT_P, root_path, DEST_P, self.m_dest_path, RES_IN, resource, SEED, str(self.m_perfect_seed)]);
		
		if resource.endswith('.lua'):
			path = os.path.normpath(root_path + "/" + resource)
			f = subprocess.call([LUAJIT, BC_G, path]);
			p = subprocess.call([LUA_C, ROOT_P, root_path, DEST_P, self.m_dest_path, RES_IN, resource, SEED, str(self.m_perfect_seed)]);
		
		if resource.endswith('.vs'):
			p = subprocess.call([VS_C, ROOT_P, root_path, DEST_P, self.m_dest_path, RES_IN, resource, SEED, str(self.m_perfect_seed)]);
		
		if resource.endswith('.ps'):
			p = subprocess.call([PS_C, ROOT_P, root_path, DEST_P, self.m_dest_path, RES_IN, resource, SEED, str(self.m_perfect_seed)]);