global screen

started = False

block_list = pygame.sprite.Group()

all_sprites_list = pygame.sprite.Group()

all_sprites_list.add(player)

startBlock = Block(GREEN, 60, 60)

all_sprites_list.add(startBlock)

startBlock.rect.x = 630

startBlock.rect.y = 320

block_list.add(startBlock)

all_sprites_list.draw(screen)

clock = pygame.time.Clock()

while not started:

for puck in puck_movement():

player.update(puck[0],puck[1])

all_sprites_list.draw(screen)

if len(highscore_list) == 0:

Score = myfont.render("Score to beat:0",1, (0,50,0))

else:

Score = myfont.render("Score to beat set by "+highscore_list[0].keys()[0]+": "+str(highscore_list[0].values()[0]), 1, (0,50,0))

screen.blit(Score,(10,10))

# See if the player block has collided with anything.

#clock.tick(20)

pygame.display.flip()

blocks_hit_list = pygame.sprite.spritecollide(player, block_list, False,pygame.sprite.collide_rect)

#if blocks_hit_list is not None:

for block in blocks_hit_list:

print("HIT!!!!!")

started = True

global myfont

screen.fill(WHITE)

Score = myfont.render("Number of hits "+str(hitcount), 1, (0,50,0))

screen.blit(Score,(10,10))

cap = cv2.VideoCapture(1)

#cap = cv2.VideoCapture("/home/lars/dev/Videos/Webcam/2017-12-20-223500.webm")

#window = pyglet.window.Window()

#cursor= window.get_system_mouse_cursor(window.CURSOR_CROSSHAIR)

#window.set_mouse_cursor(cursor)

# Game size:

#700*1258 >840*1507

#outputsize

#40*1507

# Initialize Pygame

pygame.init()

# Set the height and width of the screen

screen_width = 1513

screen_height = 900

screen = pygame.Surface((1510,842))

gameoutput = pygame.Surface((1510,58))

screen.fill(WHITE)

gameoutput.fill(RED)

global myfont

myfont = pygame.font.SysFont("monospace", 30,True)

clock = pygame.time.Clock()

player = Player(BLACK, 60, 60)

topborder = Border(BLACK, 20, 142, 745,0)

bluehole = Border(BLUE, 20,142, 745,142)

midborder = Border(BLACK, 20, 274, 745,284)

redhole = Border(RED, 20, 142, 745, 557)

lowborder = Border(BLACK, 20, 142, 745,699)

targettl = Border(GREEN, 120, 120,312,220)

targetbl = Border(GREEN, 120, 120,312,500)

targettr = Border(GREEN, 120, 120,1077,220)

targetbr = Border(GREEN, 120, 120,1077,500)

targets = [targettl,targetbl,targettr,targetbr]

sprite_list = pygame.sprite.Group()

target_list = pygame.sprite.Group()

sprite_list.add(player)

target = copy.copy(targets[randint(0,3)])

sprite_list.add(redhole)

sprite_list.add(bluehole)

sprite_list.add(topborder)

sprite_list.add(midborder)

sprite_list.add(lowborder)

sprite_list.add(target)

target_list.add(target)

# -------- Main Program Loop -----------

started = False

done = False

score=0

dest = np.array([ [0,0],[1509,0],[1509,841],[0,841] ],np.float32)

undistinfo = pickle.load(open("undistinfo.p","rb"),fix_imports=True,encoding='latin1')

mtx = undistinfo['mtx']

dist = undistinfo['dist']

ret, frame = cap.read()

h, w = frame.shape[:2]

newcameramtx, roi=cv2.getOptimalNewCameraMatrix(mtx,dist,(w,h),1,(w,h))

print("foer kallibrer")

warp,mask,maskcorners,_ = kallibrer(cap,dest,newcameramtx,mtx,dist)

fullscreen = pygame.display.set_mode([screen_width, screen_height],pygame.FULLSCREEN)

hitcount = 0

hole = ""

print("am alive")

myfont = pygame.font.SysFont("monospace", 30,True)

time1 = time.clock()

puck_size=60

x=0

dircount = 0

pucksizelist = []

start_ticks=pygame.time.get_ticks()

for puck in puck_movement(lambda:circleposition(cap,warp,newcameramtx,mtx,dist,mask,maskcorners)):

loop_ticks=pygame.time.get_ticks() #starter tick

current_x=puck[0][2][0]

current_y=puck[0][2][1]

prev_x=puck[0][1][0]

prev_y=puck[0][1][1]

length=math.sqrt(pow(abs(current_x-prev_x),2)+pow(abs(current_y-prev_y),2))

missing_pucks=[]

if length > puck_size*2:

parts=int(length/puck_size)

part_x = (current_x -prev_x)/parts

part_y = (current_y -prev_y)/parts

for x in range(1,parts):

missing_pucks.append([prev_x + part_x*x,prev_y + part_y*x])

#seconds=(pygame.time.get_ticks()-start_ticks)/1000

time2 = time.clock()

print('clocktime %s' % str(loop_ticks-start_ticks))

#time1=time2,

#pucksizelist.append([puck[0][2][2],puck[0][2][3]])

x+=1

# Calls update() method on every sprite in the list

sprite_list.update(puck[0][2][0],puck[0][2][1])

if puck[1]:

dircount = dircount + 1

if (puck[0][2][0] > 660 and puck[0][1][0] < 640) or (puck[0][2][0] < 640 and puck[0][1][0] > 660):

if puck[0][2][1] > 350:

hole = "Red"

else:

hole = "Blue"

gameoutput.fill(RED)

screen.fill(WHITE)

sprite_list.draw(screen)

player_list = pygame.sprite.Group()

player_list.add(player)

for p in missing_pucks:

player_list.add(Player(RED,60,60).update(p[0],p[1]))

player_list.draw(screen)

# ~ if len(missing_pucks) > 0:

# ~ press=True

# ~ while press:

# ~ for event in pygame.event.get():

# ~ if event.type == pygame.KEYDOWN:

# ~ press = False

#if event.key == pygame.K_q:

# pygame.quit()

target_hit_list = pygame.sprite.groupcollide(player_list, target_list, False,True,pygame.sprite.collide_rect)

#target_hit_list = pygame.sprite.spritecollide(player, target_list, True,pygame.sprite.collide_rect)

#if blocks_hit_list is not None:

for extrapuck in target_hit_list:

if len(target_hit_list[extrapuck]) > 0:

target = target_hit_list[extrapuck][0]

print("HIT!!!!!")

#raise Exception('Hit')

hitcount += 1

#randint(0,3)

ctargets = [x for x in copy.copy(targets) if not x.rect.x == target.rect.x or not x.rect.y == target.rect.y]

target = ctargets[randint(0,2)]

sprite_list.add(target)

target_list.add(target)

break

# Draw all the spites

#screen.blit(player.image,player.rect)

# Limit to 20 frames per second

clock.tick(30)

# Go ahead and update the screen with what we've drawn.

#timelimit = 100

#timecount = timelimit - round(time2 - time1)

#Timecount = myfont.render("time "+ str(timecount), 1, (0,50,0))

countdown=int(2-(loop_ticks-start_ticks)/1000)

gametime = myfont.render('clocktime %s' % str(countdown), 1, (0,50,0))

Score = myfont.render("Count "+ str(hitcount), 1, (0,50,0))

Hole = myfont.render("Hole "+ str(hole), 1, (0,50,0))

#screen.blit(Timecount,(10,10))

gameoutput.blit(Score,(10,30))

gameoutput.blit(Hole,(200 ,30))

gameoutput.blit(gametime,(400,30))

fullscreen.blit(gameoutput,(0,0))

fullscreen.blit(screen,(0,55))

if countdown < 0:

return hitcount,fullscreen

pygame.display.flip()

#if timelimit <= 0:

# return hitcount

if cv2.waitKey(1) & 0xFF == ord('q'):

break

# = time.clock()

#if x>100:

#s break

#with open("pucksizelist","wb") as f: