#!/usr/bin/python
"""
	copyright 2008  - Johannes 'josch' Schauer <j.schauer@email.de>

	This program is free software: you can redistribute it and/or modify
	it under the terms of the GNU General Public License as published by
	the Free Software Foundation, either version 3 of the License, or
	(at your option) any later version.

	This program is distributed in the hope that it will be useful,
	but WITHOUT ANY WARRANTY; without even the implied warranty of
	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
	GNU General Public License for more details.

	You should have received a copy of the GNU General Public License
	along with this program.  If not, see <http://www.gnu.org/licenses/>.
"""

import pyglet
import random

class Animation(object):
	def __init__(self, frames):
		self.frames = frames
		self.animation = 0
	
	@property
	def current_frame(self):
		return self.frames[self.animation]
	
	@property
	def next_frame(self):
		self.animation = (self.animation+1)%len(self.frames)
		return self.frames[self.animation]

class WaterTile(Animation):
	def __init__(self, tiles):
		super(WaterTile, self).__init__(tiles)
		self.tiles = tiles
	
	@property
	def current_tex_coords(self):
		return self.current_frame.tex_coords
	
	@property
	def next_tex_coords(self):
		return self.next_frame.tex_coords

class MapSet(object):
	def __init__(self, loaded_map):
		self.width = len(loaded_map[0])
		self.height = len(loaded_map)
		
		self.alt_water = pyglet.image.TextureGrid(pyglet.image.ImageGrid(pyglet.image.load("water.png"), 20, 16))
		self.water_texture = self.alt_water
		
		self.tiles = [[None for i in xrange(self.width)] for j in xrange(self.height)]
		for y, line in enumerate(loaded_map):
			for x, tile in enumerate(line):
				#self.tiles[y][x] = WaterTile(self.water_tiles[tile])
				self.tiles[y][x] = WaterTile(self.alt_water[(0, tile):(19, tile+1)])
	
	def __getitem__(self, coords):
		x, y = coords
		return self.tiles[y][x]

class MapView(object):
	def __init__(self, mapset, window):
		self.window = window
		self.mapset = mapset
		
		self.batch = pyglet.graphics.Batch()
		
		self.x = 0
		self.y = 0
		self.dx = 0
		self.dy = 0
		
		self.vertex_lists = [[] for x in xrange((self.window.height-48)//32)]
		
		self._init_view()
		
		pyglet.clock.schedule_interval(self.animate_water, 1/6.0)
		pyglet.clock.schedule_interval(self.update, 1/60.0)
	
	def _init_view(self):
		try:
			for row in self.vertex_lists:
				for vlist in row:
					vlist.delete()
		except:
			raise
		self.vertex_lists = [[] for x in xrange((self.window.height-48)//32)]
		group = pyglet.graphics.TextureGroup(self.mapset.water_texture)
		for y in xrange((self.window.height-48)//32):
			y1 = y*32+48
			y2 = y1+32
			for x in xrange((self.window.width-192)//32):
				x1 = x*32
				x2 = x1+32
				tile = self.mapset[x, y]
				self.vertex_lists[y].append(self.batch.add(4, pyglet.gl.GL_QUADS, group,
											('v2i', [x1, y1, x2, y1, x2, y2, x1, y2]),
											('t3f', tile.current_tex_coords),
											('c4B', (255,255,255,255) * 4)))
	
	def on_draw(self):
		pyglet.gl.glClear(pyglet.gl.GL_COLOR_BUFFER_BIT | pyglet.gl.GL_DEPTH_BUFFER_BIT)
		self.batch.draw()
	
	def _move(self, dx, dy):
		div_y, mod_y = divmod(self.y+dy, 32)
		div_x, mod_x = divmod(self.x+dx, 32)
		
		if div_x != self.x//32 or div_y != self.y//32:
			update_texture = True
		else:   
			update_texture = False
		
		self.x += dx
		self.y += dy
		
		for y in xrange((self.window.height-48)//32):
			y1 = mod_y+y*32+48
			y2 = y1+32
			for x in xrange((self.window.width-192)//32):
				x1 = mod_x+x*32
				x2 = x1+32
				self.vertex_lists[y][x].vertices = [x1, y1, x2, y1, x2, y2, x1, y2]
				
				if update_texture:
					tile = self.mapset[x-div_x, y-div_y]
					self.vertex_lists[y][x].tex_coords=tile.current_tex_coords
	
	def update(self, dt):
		try:
			if self.window.keys[pyglet.window.key.LCTRL] and \
			   any([self.window.keys[pyglet.window.key.UP],
					self.window.keys[pyglet.window.key.DOWN],
					self.window.keys[pyglet.window.key.LEFT],
					self.window.keys[pyglet.window.key.RIGHT]]):
				
				if self.window.keys[pyglet.window.key.LEFT]:
					x = 1
				elif self.window.keys[pyglet.window.key.RIGHT]:
					x = -1
				else:
					x = 0
				
				if self.window.keys[pyglet.window.key.UP]:
					y = -1
				elif self.window.keys[pyglet.window.key.DOWN]:
					y = 1
				else:
					y = 0
				self.dx += x*8
				self.dy += y*8
		except KeyError:
			pass
		if self.dx or self.dy:
			self._move(self.dx, self.dy)
			self.dx = 0
			self.dy = 0
	
	def on_mouse_drag(self, x, y, dx, dy, buttons, modifiers):
		self.dx += dx
		self.dy += dy
		return pyglet.event.EVENT_HANDLED
	
	def on_resize(self, width, height):
		self._init_view()
	
	def animate_water(self, dt):
		for y in xrange((self.window.height-48)//32):
			for x in xrange((self.window.width-192)//32):
				tile = self.mapset[x-self.x//32, y-self.y//32]
				self.vertex_lists[y][x].tex_coords=tile.next_tex_coords

class Window(pyglet.window.Window):
	def __init__(self, *args, **kwargs):
		super(Window, self).__init__(800, 600, resizable=True, vsync=False )
		#img = pyglet.resource.image("cursor.png")
		#self.set_mouse_cursor(pyglet.window.ImageMouseCursor(img, 0, 40))
		self.keys = pyglet.window.key.KeyStateHandler()
		self.push_handlers(self.keys)
		self.fps = pyglet.clock.ClockDisplay()
	
	def on_draw(self):
		self.fps.draw()
	
	def on_key_press(self, symbol, modifiers):
		if symbol == pyglet.window.key.F11:
			self.set_fullscreen(fullscreen=not self.fullscreen)
	
	def empty(self, dt):
		pass

def main():
	pyglet.gl.glEnable(pyglet.gl.GL_BLEND)
	pyglet.gl.glBlendFunc(pyglet.gl.GL_SRC_ALPHA, pyglet.gl.GL_ONE_MINUS_SRC_ALPHA)
	window = Window()
	
	loaded_map = [[random.randint(0,11) for x in xrange(128)] for y in xrange(128)]
	mapset = MapSet(loaded_map)
	
	mapview = MapView(mapset, window)
	window.push_handlers(mapview)
	pyglet.clock.schedule(window.empty)
	pyglet.app.run()

if __name__ == '__main__':
	main()