Game 3.0

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'
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) 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]=='white' and enemy[3] > -gosmperiod) or (enemy[4]=='black' 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]=='black':
                    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), -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='white')
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='white', outline='black'), deg[i],dist[i], 1, 'white'] for i in range(enenum)]
    for i in range(blackene):
        enearr[i][4]='black'
        rajz.itemconfig(enearr[i][0], fill='black')
    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*kul)))[2:]
                if len(ertek)==1:
                    ertek='0'+ertek
                rajz.itemconfig(sh[0], fill='#ff'+ertek+ertek)
            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