#!/usr/bin/python

# Usage: render_nff.py [scene.nff] [output.png]
# reads scene from nff file and render it to PNG file
# if no output file name is given, it makes one from input name
# reads from stdin by default

# see http://tog.acm.org/resources/SPD/ for nff format description
# cylinders are currently not implemented

from pyrit import *
from math import pi
import sys

rt = Raytracer()
top = KdTree()
rt.setTop(top)
rt.setCamera(Camera())

imagesize = (800, 600)

mat = Material(colour=(1.0, 1.0, 1.0))

f = sys.stdin
fout = "render_nff.png"
if len(sys.argv) > 1:
	f = open(sys.argv[1])
	fout = sys.argv[1].rsplit('.',1)[0] + '.png'
if len(sys.argv) > 2:
	fout = sys.argv[2]

while True:
	line = f.readline()
	if line == "":
		break;
	ln = line.split()
	if ln[0] == 'v':	# Viewpoint location
		# from
		ln = f.readline().split()
		assert ln[0] == 'from'
		eye = (float(ln[1]), float(ln[2]), float(ln[3]))
		# at
		ln = f.readline().split()
		assert ln[0] == 'at'
		lookat = (float(ln[1]), float(ln[2]), float(ln[3]))
		# up
		ln = f.readline().split()
		assert ln[0] == 'up'
		up = (float(ln[1]), float(ln[2]), float(ln[3]))
		# angle
		ln = f.readline().split()
		assert ln[0] == 'angle'
		angle = float(ln[1])
		# hither
		ln = f.readline().split()
		assert ln[0] == 'hither'
		hither = float(ln[1])
		# resolution
		ln = f.readline().split()
		assert ln[0] == 'resolution'
		imagesize = (int(ln[1]), int(ln[2]))
		# set camera as specified
		cam = Camera(eye=eye, lookat=lookat, up=up)
		cam.setAngle(angle/180*pi)
		rt.setCamera(cam)
	elif ln[0] == 'b':	# Background color
		rt.setBgColour((float(ln[1]), float(ln[2]), float(ln[3])))
	elif ln[0] == 'l':	# Light
		pos = (float(ln[1]), float(ln[2]), float(ln[3]))
		rt.addLight(Light(position=pos))
	elif ln[0] == 'f':	# Fill color and shading parameters
		colour = (float(ln[1]), float(ln[2]), float(ln[3]))
		mat = Material(colour=colour)
		mat.setPhong(0,float(ln[4]),float(ln[5]),float(ln[6]))
		mat.setTransmissivity(float(ln[7]),float(ln[8]))
	elif ln[0] == 's':	# Sphere
		center = (float(ln[1]), float(ln[2]), float(ln[3]))
		radius = float(ln[4])
		rt.addShape(Sphere(center=center, radius=radius, material=mat))
	elif ln[0] == 'p':	# Polygon
		vertex_count = int(ln[1])
		vertices = []
		for i in range(vertex_count):
			ln = f.readline().split()
			vertex = (float(ln[0]), float(ln[1]), float(ln[2]))
			vertices.append(NormalVertex(vertex))
		rt.addShape(Triangle(vertices[0], vertices[1], vertices[2], mat))
		for i in range(vertex_count)[3:]:
			rt.addShape(Triangle(vertices[0], vertices[i-1], vertices[i], mat))
	elif ln[0] == 'pp':	# Polygonal patch
		mat.setSmooth(True)
		vertex_count = int(ln[1])
		vertices = []
		for i in range(vertex_count):
			ln = f.readline().split()
			vertex = (float(ln[0]), float(ln[1]), float(ln[2]))
			normal = (float(ln[3]), float(ln[4]), float(ln[5]))
			vertices.append(NormalVertex(vertex, normal))
		rt.addShape(Triangle(vertices[0], vertices[1], vertices[2], mat))
		for i in range(vertex_count)[3:]:
			rt.addShape(Triangle(vertices[0], vertices[i-1], vertices[i], mat))
	elif ln[0] == '#':	# Comment
		pass
	else:
		print "Not implemented:", line
f.close()

top.optimize()

sampler = DefaultSampler(imagesize)
rt.setSampler(sampler)
rt.render()
sampler.getPixmap().writePNG(fout)