Random Game

from tkinter import *
from math import *
from time import time
from random import *

##############################
#Defining variables

#~Nenyúljhozzá részleg

timeborder=0.03

r=3
cr=40
br=5
d=br
kulso=500

x, y = 350, 350
change=90.0
point=[]

gosmaller=90
gosmperiod=35
getpts=[]
delay=60
acc=200

color='red'
simplecolor='black'
fullcolor='#00ffff'
outlinecolor='white'
pressing=False
first=True
direction=0
shoots=[]

#*****************************************
#~Hozzányúlhatsz részleg

db=17 # töltények száma
plus=5# lőtorony forgási sebesség

enenum=20 # enemyk száma
blackene=3 # fekete enemyk száma
enespeed=0.7 # enemyk sebessége
startdist=500 # első enemy milyen korán jön
between=50# enemyk közötti táv

#Ezekkel vicces játszani ;)
min=20 # enemyk mininum mérete
max=40 # enemyk maximum mérete

########
eredb=db
##############################
#Defining good functions

def cd(which, i):
	if not which: #tehát x coord
		return x+cos(radians(360*(i/db) -change))*cr
	else: #tehát y coord
		return y+sin(radians(360*(i/db) -change))*cr

def cdp(which, i, rad):
	if not which: #tehát x coord
		return x+cos(radians(360*(i/db) -change))*rad
	else: #tehát y coord
		return y+sin(radians(360*(i/db) -change))*rad

#~~~~~~~~~~~~~~~~~~~~~~~~~~

def rotate(dir):
	global pressing, direction
	direction=dir
	pressing=True

def stop_rot():
	global pressing
	pressing=False

def eneinside(xy):
	if xy[0] >= 0 and xy[0] <= 700 and xy[1] >= 0 and xy[1] <= 700:
		return True
	else: return False

def ng():
	global newgame
	newgame=True

#~~~~~~~~~~~~~~~~~~~~~~~~~~

def fire():
	global shoots, alive, enedied, lab
	if db==1 or not alive:
		return 0
	shoots.append([rajz.create_line(cdp(0, 0, cr+d), cdp(1, 0, cr+d), cdp(0, 0, kulso), cdp(1, 0, kulso), fill='#ff0000', width=br), time()])

	for enemy in enearr:
		enexy=[350+cos(enemy[1]) *enemy[2], 350+sin(enemy[1])*enemy[2]]
		diff=radians(change*-1) - enemy[1]
		if abs(sin(diff)*enemy[2])< min and eneinside(enexy) and cos(diff)>0 and enemy[3]==1:
			enemy[3]=0

	changepts(db-1)

#~~~~~~~~~~~~~~~~~~~~~~~~~~

def changepts(how):
	global point, db
	for pt in point:
		rajz.delete(pt)
	db=how
	point=[rajz.create_oval(cd(0, i)-r, cd(1, i)-r, cd(0, i)+r, cd(1, i)+r, width=r, outline=outlinecolor) for i in range(db)]
	rad=br
	rajz.coords(point[0], cd(0, 0)-rad, cd(1, 0)-rad, cd(0, 0)+rad, cd(1, 0)+rad)
	rajz.itemconfig(point[0], outline=color, width=br)

#~~~~~~~~~~~~~~~~~~~~~~~~~~

def randdeg():
	return uniform(0, 2*pi)

def help(i):
	global min, max
	r=radians(uniform(0, 90)+i*90)
	d=uniform(min, max)
	#alseg.append([r, d])
	enedeg=deg[ind]
	if first:
		enedist=dist[ind]
	else:
		enedist=enearr[ind][2]
	kx, ky= x+cos(enedeg)*enedist, y+sin(enedeg)* enedist
	a, b = cos(r)*d+kx, sin(r)*d+ky
	return [a, b]

def getshape(index):
	global ind
	ind=index
	return help(0)+help(1)+help(2)+help(3)

#~~~~~~~~~~~~~~~~~~~~~~~~~~

def calcDeg(start, Ln):
	def easy(x):
		if x >= 0:
			return 1
		elif x < 0:
			return -1
	"""def correct(x):
		if x<0:
			return 360+x
		else:
			return x"""
	try:
		c = (Ln[2]-Ln[0])/hypot(Ln[1]-Ln[3], Ln[0]-Ln[2])
		s = (Ln[3]-Ln[1])/hypot(Ln[1]-Ln[3], Ln[0]-Ln[2])
	except:
		c, s = 0, 0
	recovery=acos(c)*easy(s)-radians(start)
	dgc=acos(cos(recovery))
	elojel=sin(recovery)
	return dgc*easy(elojel)

#~~~~~~~~~~~~~~~~~~~~~~~~~~

def animation():
	for enemy in enearr:
		if enemy[3]<1:
			enexy=[350+cos(enemy[1]) *enemy[2], 350+sin(enemy[1])*enemy[2]]
			if (enemy[4]==simplecolor and enemy[3] > -gosmperiod) or (enemy[4]==fullcolor and enemy[3] > -gosmperiod/2):
				eneshape= [[rajz.coords(enemy[0])[2*j+i] for i in range(2)] for j in range(4)]
				enedegs=[calcDeg(0, enexy+eneshape[i]) for i in range(4)]
				eneptdists=[hypot(enexy[0]- eneshape[i][0], enexy[1]-eneshape[i][1])* gosmaller/100 for i in range(4)]
				def shape(i):
					return [enexy[0]+ cos(enedegs[i])* eneptdists[i], enexy[1]+ sin(enedegs[i])*eneptdists[i]]
				newshape=[]
				for i in range(4):
					newshape+=shape(i)
				rajz.coords(enemy[0], newshape)
				enemy[3]-=1
			else:
				rajz.coords(enemy[0], enexy+ enexy+enexy+enexy)
				enemy[3]=2
				if enemy[4]==fullcolor:
					for i in range(randint(2, 5)):
						getpts.append([ rajz.create_oval(enexy[0]-r, enexy[1]-r, enexy[0]+r, enexy[1]+r, width=r, outline=outlinecolor), -i*delay, time(), enemy[1]+pi, rajz.coords(enemy[0])[:2]])

		rem=[]
		if len(getpts)>0:
			for pt in getpts:
				t=pow(time()-pt[2], 2)
				npx, npy = pt[4][0]+ cos(pt[3])*acc*t, pt[4][1]+sin(pt[3])*acc*t
				pdist=hypot(350-npx, 350-npy)
				if pt[1]<0:
					pt[1]+=1
					pt[2]=time()
				elif pdist > cr:
					rajz.coords(pt[0], npx-r, npy-r, npx+r, npy+r)
				else:
					changepts(db+1)
					rem.append(pt)
					rajz.delete(pt[0])
			for item in rem:
				getpts.remove(item)

def stop(arg):
	f=open('sth.txt', 'w')
	f.write(str(arg))
	f.close()
	while 1:
		pass

"""hatar1=2
hatar2=2

def mozgat():
	global alak
	for i in range(4):
		r=alseg[i][0]+uniform(-hatar1, hatar1)
		d=alseg[i][1]+uniform(-hatar2, hatar2)
		if r>90+i*90:
			r=90+i*90
		elif r<i*90:
			r=i*90
		if d < min:
			d=min
		elif d > max:
			d=max
		a, b = cos(r)*d+x, sin(r)*d+y
		alak[i][0]=a
		alak[i][1]=b"""

##############################
#Setting program

ablak=Tk()

rajz=Canvas(width=700, height=700, bg=simplecolor)
rajz.grid(columnspan=2)

def setnewgame():
	global centralline, deg, dist, enearr, first
	first=True
	changepts(eredb)

	centralline=rajz.create_line(350, 350, 350, 350-cr, fill='red', width=4)

	deg=[randdeg() for i in range(enenum)]
	dist=[startdist+i*between for i in range(enenum)]
	shuffle(dist)
	enearr=[[rajz.create_polygon( getshape(i), fill=simplecolor, outline=outlinecolor), deg[i],dist[i], 1, simplecolor] for i in range(enenum)]
	for i in range(blackene):
		enearr[i][4]=fullcolor
		rajz.itemconfig(enearr[i][0], fill=fullcolor)
	first=False

dir=["Balra", "Jobbra"]
dirbut=[]
for i in range(2):
	dirbut.append(Button(text=dir[i], width=14, height=3))
	dirbut[i].grid(column=i, row=1)
	dirbut[i].bind('<ButtonPress-1>', lambda event, i=i: rotate(i))
	dirbut[i].bind('<ButtonRelease-1>', lambda event: stop_rot())

firebut=Button(text='FIRE!!!!', command=fire, width=20, height=3)
firebut.grid(row=2, columnspan=2)

newgamebut=Button(text='New Game', command=ng, width=20, height=3)

alive=True
rogzit=time()
lab=Label()
lab.grid(columnspan=2)

setnewgame()

#~~~~~~~~~Main loop~~~~~~~~~~~
"""
STATISZTIKA:
Alap várási idő: 0.02 sec
Iterációk 0.025 sec feletti idővel: 45%
Iterációk 0.03 sec feletti idővel: 14%
Iterációk 0.035 sec feletti idővel: 5%

Ebbe beleszámítandó az iterációnkénti
időellenőrzés
"""
while alive:
	while time()-rogzit < timeborder:
		pass
	rogzit=time()
	animation()
	ablak.update()
	if len(shoots)!=0:
		for sh in shoots:
			kul=time()-sh[1]
			if kul <= 1:
				ertek=str(hex(int(255- 255*kul)))[2:]
				if len(ertek)==1:
					ertek='0'+ertek
				#rajz.itemconfig(sh[0], fill='#ff'+ertek+ertek)
				rajz.itemconfig(sh[0], fill='#'+ertek+'0000')
			else:
				rajz.delete(sh[0])
				shoots.remove(sh)
	if pressing:
		if direction:
			plus=abs(plus)*-1
		else:
			plus=abs(plus)
		change+=plus
		for i in range(db):
			if i==0: rad=br
			else: rad=r
			rajz.coords(point[i], cd(0, i)-rad, cd(1, i)-rad, cd(0, i)+rad, cd(1, i)+rad)
		rajz.coords(centralline, 350, 350, cd(0, 0), cd(1, 0))

	h=0
	for enemy in enearr:
		if enemy[2]<cr:
			alive=False
			break
		elif enemy[3]==1:
			enemy[2] -= enespeed
			rajz.coords(enemy[0], getshape(h))
		h+=1
	if not alive:
		rajz.delete(ALL)
		rajz.create_text(350, 350, text='GAME OVER')
		for i in [0,1]:
			dirbut[i].grid_forget()
		firebut.grid_forget()
		newgamebut.grid(row=1, columnspan=2)

		newgame=False
		while not newgame:
			ablak.update()
		rajz.delete(ALL)

		newgamebut.grid_forget()
		for i in [0,1]:
			dirbut[i].grid(column=i, row=1)
		firebut.grid(row=2, columnspan=2)

		change=90.0
		setnewgame()
		alive=True