main.py

from planet import *
from map import *
from ship import *
from missile import *
from save import *
from Bar import *
from text import *
from trail import *
from sound import *
from explosion import *
import config

class Display:
	def __init__(self,dbcom=""):
		py.init()
		self.s = None
		if dbcom == "create":
			self.s = Saver("create")
		else:
			self.s = Saver()
		self.initdisplayvar()
		self.win = py.display.set_mode((self.winx,self.winy))
		self.sound = Sound()
		self.writer = Text(self.win)
		self.init_event_variables()
		self.run()
		self.spacewalk=np.array([0,0])


	def initdisplayvar(self):
		self.winx = 900
		self.winy = 800
		self.stopgame = False

	def init_event_variables(self):
		self.planets = []
		self.hero = py.sprite.Group()
		self.map = py.sprite.Group( )
		self.missiles = py.sprite.Group()
		self.satellites = py.sprite.Group()
		self.explosions = py.sprite.Group()
		self.trail = Trail()
		self.player = None
		self.fuelbar = None
		self.wkup = False
		self.wkdo = False
		self.wklf = False
		self.wkrh = False
		self.zoomin = False
		self.zoomout = False
		self.autotrail = False
		self.m_W_player = False
		self.m_S_player = False
		self.r_L_player = False
		self.r_R_player = False
		self.setorbital = False
		self.launchmissile = False
		pass

	def setplanets(self):
		#initates the game settings and the sprites positions
		PG = PlanetGenerator()
		for i in PG.planets:
			self.planets.append(i)
			#sat = i.get_satellite()
			#self.satellites.add(sat)

	def sethero(self):
		i=None
		for i in self.planets:
			break;
		self.player = Player((i.rect.x,i.rect.y-i.radius))
		if loadlatest == True and self.s.dataexist():
			t =self.s.loadrecent()
			self.player.pos = np.array([t[1],t[2]])
			self.player.vel = np.array([t[3],t[4]])
			self.player.angle = t[5]
			self.player.accel = np.array([t[6],t[7]])

		self.hero.add(self.player)
		self.fuelbar = Bar()

	def setmap(self):
		self.map.add(Map(self.planets))

	def sprites_handler(self):
		pass

	def eventhandler(self):
		for event in py.event.get():
			if event.type == py.QUIT:
				self.stopgame = True
			if event.type == 2:
				if event.key == py.K_UP:
					self.zoomin = True
				if event.key == py.K_DOWN:
					self.zoomout = True
				if event.key == py.K_LEFT:
					self.wklf = True
				if event.key == py.K_RIGHT:
					self.wkrh = True
				if event.key == py.K_i:
					self.player.invert()
				if event.key == py.K_a:
					self.r_L_player = True
				if event.key == py.K_d:
					self.r_R_player = True
				if event.key == py.K_w:
					self.m_W_player = True
					self.player.accelerating = True
					self.sound.thrusts()
				if event.key == py.K_s:
					self.m_S_player = True

				if event.key == py.K_o:
					self.setorbital = True
				if event.key == py.K_SPACE:
					self.launchmissile = True
					self.player.missilecount +=1
				if event.key == py.K_t:
					self.player.autorotate = not self.player.autorotate
				if event.key == py.K_j:
					self.trail.addpoint(self.player.pos)
				if event.key == py.K_c:
					self.trail.clear()
				if event.key == py.K_k:
					self.autotrail = not self.autotrail
				if event.key == py.K_r:
					self.player.pos = np.array([0,-22300-size/2],float)
					self.player.angu_vel = 0
					self.player.vel = np.array([0.0,0.0])
					self.player.angle = 0
					self.sound.offthrust.stop()

			if event.type == 3:
				if event.key == py.K_UP:
					self.zoomin = False
				if event.key == py.K_DOWN:
					self.zoomout = False
				if event.key == py.K_LEFT:
					self.wklf = False
				if event.key == py.K_RIGHT:
					self.wkrh = False
				if event.key == py.K_a:
					self.r_L_player = False
				if event.key == py.K_d:
					self.r_R_player = False
				if event.key == py.K_w:
					self.m_W_player = False
					self.player.accelerating = False
					self.sound.stopthrusts()
				if event.key == py.K_s:
					self.m_S_player = False
				if event.key == py.K_q:
					self.stopgame = True
				if event.key == py.K_o:
					self.setorbital = False


	def affectgravity(self):
		nonthingtrue = True
		count = 0
		min = 100000000000
		minindex = 0
		for i in self.planets:
			r = i.pos - self.player.pos
			dis = np.linalg.norm(r)
			if(dis<min):
				min = dis
				minindex = count
			count+=1

		nearplanet = self.planets[minindex]
		r = nearplanet.pos - self.player.pos
		dis = np.linalg.norm(r)
		orb_vel = np.sqrt(nearplanet.mass / dis)
		self.player.requireorbitalvel = orb_vel
		if self.setorbital:
			self.player.setvelocity(orb_vel)
			self.setorbital = False

		for i in self.planets:
			r = i.pos - self.player.pos
			dis = np.linalg.norm(r)
			if((dis > 1 and dis < i.radius+3000) and (not self.player.stopgravity)):
				g = i.mass/(dis)**2
				g_vec = g*r/dis
				self.player.vel += g_vec*dt

	def respondevents(self):
		mpt = [0,0]
		if self.wkup:
			mpt[1]=-10
		if self.wkdo:
			mpt[1]=10
		if self.wklf:
			mpt[0]=-10
		if self.wkrh:
			mpt[0]=10

		if self.m_W_player or self.m_S_player:
			if self.m_W_player:
				self.player.accelerating = True
				self.player.accelerate()

				#print("accelerate")

			if self.m_S_player:
				#print("decelerate")
				self.player.decelerate()
		else:
			self.player.accelerating = False
			self.player.nonfireimage()
			self.player.stopaccelerate()
			#print("stopping")

		if self.r_L_player:
			self.player.rotate(False)
		elif self.r_R_player:
			self.player.rotate(True)

		if self.launchmissile:
			#print("missile launched")
			self.launchmissile = False
			self.missiles.add(Missile(self.player,str(self.player.missilecount),self.player.zoom))

		for i in self.planets:
			i.update(2,self.player.pos)
		self.hero.update(self.planets)
		self.map.update(self.player.pos)
		self.missiles.update(self.planets)
		self.satellites.update(self.player.pos)

		for missile in self.missiles:
			if missile.killme:
				self.explosions.add(Explosion(missile.pos))
				missile.kill()
		self.explosions.update(self.player)

	def writeparameters(self):
		velocitystr = "Velocity:  " + str(int(np.linalg.norm(self.player.vel)))
		vestrlencen = len(velocitystr)/2
		orbitalstr  = "Req orbital: "+str(int(self.player.requireorbitalvel))
		orbitalstrlen = len(orbitalstr)/2
		angvelstr = "Angular Velocity:" + str("{0:.2f}".format(self.player.angu_vel))
		angvelstrlen = len(angvelstr)/2
		self.writer.draw((vestrlencen*7, 100), velocitystr)
		self.writer.draw((orbitalstrlen*7,120),orbitalstr)
		self.writer.draw((angvelstrlen*7,80),angvelstr)

	def draw(self):
		if self.zoomin :
			config.zoom_value+=0.2

		elif self.zoomout:
			if config.zoom_value > 1:
				config.zoom_value -= 0.2

		if self.zoomin or self.zoomout:
			#planets zoom
			self.trail.zoom = config.zoom_value
			for i in self.planets:
				i.zoom = config.zoom_value
			#ship zoom
			self.player.zoom = config.zoom_value
			#missile zoom
			for i in self.missiles:
				i.zoom = config.zoom_value

			print(config.zoom_value)
			for i in self.planets:
				for j in i.clouds:
					j.zoom = config.zoom_value
		for i in self.planets:
			self.win.blit(i.atmos_image,(i.atmos_rect.x,i.atmos_rect.	y))
			self.win.blit(i.image,(i.rect.x,i.rect.y))
		self.hero.draw(self.win)
		self.map.draw(self.win)
		self.missiles.draw(self.win)

		if self.win != None:
			for planet in self.planets:
				planet.clouds.draw(self.win)
		self.fuelbar.draw(self.win,self.player.fuel,100)
		self.trail.draw(self.win, self.player.pos)
		self.explosions.draw(self.win)
		self.writeparameters()
		if self.autotrail:
			self.trail.autotrail(self.player)

	def run(self):
		self.stopgame=False
		self.setplanets()
		self.sethero()
		self.setmap()
		self.sound.theme()
		#testing
		while(not self.stopgame):
			self.win.fill((0,30,40))
			self.eventhandler()
			self.respondevents()
			for i in self.planets:
				#print(np.array(self.player.pos),np.array([i.rect.x,i.rect.y]))
				break
			self.draw()
			self.affectgravity()
			self.trail.maintainlength()
			py.display.update()
			if self.stopgame == True:
				self.s.save(self.player)
				self.s.commit()
			elif self.player.fuel < 0.1:
				self.stopgame = True
				self.s.save(self.player)
				self.s.commit()


if __name__ == '__main__':
	import os
	files = os.listdir()
	if "mydb" in files:
		game = Display()
	else:
		game = Display("create")