from Blender import Node
import math

class MyNode(Node.Scripted):

	def __init__(self, sockets):
		normal = Node.Socket('Normal', val = [1.0, 1.0, 1.0, 1.0])
		sockets.input = [normal]
		sockets.output = [normal]

	def __call__(self):
		ocol = list(self.input.normal)
		ocol2 = list(self.input.normal)

		ocol2[0] = ocol[0] * math.cos(-self.s.surfaceViewVector[0]) - ocol[2] * math.sin(-self.s.surfaceViewVector[0])
		ocol2[1] = ocol[1]
		ocol2[2] = ocol[0] * math.sin(-self.s.surfaceViewVector[0]) + ocol[2] * math.cos(-self.s.surfaceViewVector[0])

		ocol[0] = ocol2[0]
		ocol[1] = ocol2[1] * math.cos(self.s.surfaceViewVector[1]) + ocol2[2] * math.sin(self.s.surfaceViewVector[1])
		ocol[2] = ocol2[2] * math.cos(self.s.surfaceViewVector[1]) - ocol2[1] * math.sin(self.s.surfaceViewVector[1])

		self.output.normal = ocol

__node__ = MyNode