Advertisement
jee7

TIA Tetris - Versioon 37

Nov 22nd, 2014
149
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 14.43 KB | None | 0 0
  1. import pygame, random
  2.  
  3. def looKaart(laius, kõrgus, sümbol = 0):
  4.     kaart = []
  5.     for x in range(laius):
  6.             kaart.append([])
  7.             for y in range(kõrgus):
  8.                 kaart[x].append(sümbol)
  9.  
  10.     return kaart
  11.  
  12.  
  13. def kopeeriKaart(kaart1, kaart2):
  14.     for x in range(len(kaart2)):
  15.         for y in range(len(kaart2[0])):
  16.             kaart1[x][y] = kaart2[x][y]
  17.  
  18.     return kaart1
  19.  
  20.  
  21. class Ruut:
  22.     laius = 15
  23.  
  24.     def __init__(self, värv = 0):
  25.         self.värv = värv
  26.  
  27.     def joonista(self, x, y):
  28.         äärX = Kaart.äärX
  29.         äärY = Kaart.äärY
  30.        
  31.         if self.värv == 0:
  32.             pygame.draw.rect(ekraan, (20, 20, 20), (äärX + x * Ruut.laius, äärY + y * Ruut.laius, Ruut.laius, Ruut.laius))
  33.             pygame.draw.rect(ekraan, (0, 0, 0), (äärX + x * Ruut.laius + 1, äärY + y * Ruut.laius + 1, Ruut.laius - 2, Ruut.laius - 2))
  34.         else:
  35.             värv = self.värv
  36.             alumineVärv = (max(0, värv[0] - 10), max(0, värv[1] - 10), max(0, värv[2] - 10))
  37.             pygame.draw.rect(ekraan, alumineVärv, (äärX + x * Ruut.laius, äärY + y * Ruut.laius, Ruut.laius, Ruut.laius))
  38.  
  39.             ülemineVärv = [min(üksVärv + 10, 255) for üksVärv in värv]
  40.             pygame.draw.rect(ekraan, ülemineVärv, (äärX + x * Ruut.laius, äärY + y * Ruut.laius, Ruut.laius - 1, Ruut.laius - 1))
  41.  
  42.             pygame.draw.rect(ekraan, värv, (äärX + x * Ruut.laius + 1, äärY + y * Ruut.laius + 1, Ruut.laius - 2, Ruut.laius - 2))
  43.                
  44.        
  45.  
  46. class Skoor:
  47.  
  48.     def __init__(self):
  49.         self.skoor = 0
  50.         self.font = pygame.font.SysFont("monospace", 14)
  51.  
  52.     def suurenda(self, ridadeArv):
  53.         if ridadeArv == 1:
  54.             self.skoor = self.skoor + 1
  55.         elif ridadeArv == 2:
  56.             self.skoor = self.skoor + 3
  57.         elif ridadeArv == 3:
  58.             self.skoor = self.skoor + 5
  59.         elif ridadeArv == 4:
  60.             self.skoor = self.skoor + 8
  61.  
  62.     def joonista(self, ekraan):
  63.         tekst = self.font.render("Skoor: " + str(self.skoor), True, (200, 200, 200))
  64.         ekraan.blit(tekst, (5, 2))
  65.    
  66.  
  67. #See klass hoiab väljaku seisu ja oskab seda joonistada
  68. class Kaart:
  69.     laius = 15  #niipalju ruute laiuses
  70.     kõrgus = 25 #niipalju ruute kõrguses
  71.     äärX = 0
  72.     äärY = 20
  73.     tühiRuut = Ruut(0)
  74.  
  75.     def __init__(self):
  76.         #Teeme 2-mõõtmelise listi
  77.         self.kaart = looKaart(self.laius, self.kõrgus)
  78.         self.täisRead = []
  79.        
  80.  
  81.     def joonista(self):
  82.         #Joonistame ruudud
  83.         for x, rida in enumerate(self.kaart):
  84.             for y, lahter in enumerate(rida):
  85.                 kasJoonistas = False
  86.                 if lahter != 0:
  87.                     kasJoonistas = lahter.joonista(x, y)
  88.                 if lahter == 0:
  89.                     Kaart.tühiRuut.joonista(x, y)
  90.  
  91.  
  92.     def lisaKujund(self, kujund):
  93.         for lokaalneX, rida in enumerate(kujund.kujund):
  94.             for lokaalneY, lahter in enumerate(rida):
  95.                 if kujund.kujund[lokaalneX][lokaalneY]:
  96.                     self.kaart[kujund.x + lokaalneX][kujund.y + lokaalneY] = kujund;
  97.  
  98.  
  99.     def kaotaRead(self):
  100.         kaardiKoopia = looKaart(self.laius, self.kõrgus)
  101.         kaardiKoopia = kopeeriKaart(kaardiKoopia, self.kaart)
  102.  
  103.         koopiaNihe = 0
  104.         self.täisRead.sort()
  105.         print(self.täisRead)
  106.         for y in self.täisRead:
  107.             print("Kaotame rea: " + str(y))
  108.             for x in range(len(self.kaart)):
  109.                 for koopiaY in range(y, -1, -1):
  110.                     if koopiaY - koopiaNihe <= 0:
  111.                         kaardiKoopia[x][koopiaY] = 0
  112.                     else:
  113.                         kaardiKoopia[x][koopiaY] = kaardiKoopia[x][koopiaY - 1]
  114.  
  115.             for x in range(len(self.kaart)):
  116.                 self.kaart[x][y] = 0
  117.  
  118.         self.täisRead = []
  119.         print(self.täisRead)
  120.         print("Ridade kaotamine läbi")
  121.         self.kaart = kopeeriKaart(self.kaart, kaardiKoopia)
  122.  
  123.  
  124.     def uuenda(self, kujund):
  125.         kasSaabLiikuda = kujund.y < len(self.kaart[0]) - len(kujund.kujund[0])
  126.         if kasSaabLiikuda:
  127.             for x in range(len(kujund.kujund)):
  128.                 for y in range(len(kujund.kujund[0])):
  129.                     if (kujund.kujund[x][y] and (y + 1 == len(kujund.kujund[0]) or not kujund.kujund[x][y + 1]) and self.kaart[x + kujund.x][y + kujund.y + 1] != 0):
  130.                         kasSaabLiikuda = False
  131.  
  132.                     if not kasSaabLiikuda:
  133.                         break
  134.  
  135.         #Kui kujundi all on veel tühjasi ridu:
  136.         if kasSaabLiikuda:
  137.             kujund.liigu(self, pygame.K_DOWN)
  138.         else:
  139.             self.lukusta(kujund)
  140.  
  141.  
  142.     def lukusta(self, kujund):
  143.         #Kujund ei saa enam liikuda, lukustame selle kujundi (kirjutame värvid väljakule)
  144.         print("Lukusta")
  145.         kujund.kasElus = False
  146.         for x in range(len(kujund.kujund)):
  147.             for y in range(len(kujund.kujund[0])):
  148.                 if kujund.kujund[x][y]:
  149.                     self.kaart[kujund.x + x][kujund.y + y] = kujund.kujund[x][y]
  150.  
  151.         for y in range(len(kujund.kujund[0])):
  152.             kasRidaTäis = True
  153.             for x in range(len(self.kaart)):
  154.                 kasRidaTäis &= self.kaart[x][kujund.y + y] != 0
  155.  
  156.             if kasRidaTäis:
  157.                 self.täisRead.append(kujund.y + y)
  158.                
  159.        
  160. #Klass, mille objektid on erinevad kujundid
  161. class Kujund:
  162.  
  163.     def __init__(self, x, y, kujund, värv):
  164.         self.x = x             #Kujundi asukoht kaardil
  165.         self.y = y          
  166.         self.kujund = kujund[0]   #2-mõõtmeline list, mis hoiab kujundit
  167.         self.pöörded = kujund
  168.         self.pööre = 0
  169.         self.värv = värv
  170.         self.kasElus = True
  171.  
  172.         for pööre in self.pöörded:
  173.             for x in range(len(pööre)):
  174.                 for y in range(len(pööre[x])):
  175.                     if pööre[x][y]:
  176.                         pööre[x][y] = Ruut(self.värv)
  177.                    
  178.  
  179.     def joonista(self, kaardiX, kaardiY):
  180.         #Joonistame kujundi vastavatele kordinaatidele, kui kujund on seal
  181.         x = kaardiX - self.x
  182.         y = kaardiY - self.y
  183.         if x >= 0 and y >= 0 and self.kujund[x][y]:
  184.             self.kujund[x][y].joonista(kaardiX, kaardiY)
  185.  
  186.             return True
  187.         else:
  188.  
  189.             return False
  190.  
  191.     def liigu(self, kaart, suund):
  192.         deltaX = 0
  193.         deltaY = 0
  194.  
  195.         #Leiame, kui palju liikuda mingites suundades
  196.         if suund == pygame.K_LEFT:
  197.             deltaX = -1
  198.         elif suund == pygame.K_RIGHT:
  199.             deltaX = 1
  200.         elif suund == pygame.K_DOWN:
  201.             deltaY = 1
  202.         else:
  203.             return
  204.  
  205.         #Vasakul võib olla tühjasi tulpi (et kujund oleks tsentris)
  206.         vasakulTühjasiTulpi = 0
  207.         for x in range(len(self.kujund)):
  208.             kasTühi = True
  209.             for y in range(len(self.kujund[0])):
  210.                 kasTühi &= not self.kujund[x][y]
  211.  
  212.             if kasTühi:
  213.                 vasakulTühjasiTulpi += 1
  214.             else:
  215.                 break;
  216.  
  217.         #Vaatame, kas üldse saab liikuda vastavas suunas
  218.         kasSaabLiikuda = True
  219.         for x in range(len(self.kujund)):
  220.             for y in range(len(self.kujund[0])):
  221.                 uusX = self.x + deltaX + x
  222.                 uusY = self.y + deltaY + y
  223.                 kasSaabLiikuda &= uusX >= -vasakulTühjasiTulpi
  224.                 kasSaabLiikuda &= uusY < len(kaart.kaart[0])
  225.                 kasSaabLiikuda &= uusX < len(kaart.kaart)
  226.                 if kasSaabLiikuda and self.kujund[x][y]:
  227.                     #Liikuda saab siis, kui kõik uued kujundi ruudud on hetkel tühjad
  228.                     kasSaabLiikuda &= (kaart.kaart[uusX][uusY] == 0 or kaart.kaart[uusX][uusY] == self)
  229.  
  230.         if kasSaabLiikuda:
  231.             self.kustutaKaardilt(kaart)
  232.             self.x += deltaX
  233.             self.y += deltaY
  234.             self.asetaKaardile(kaart)
  235.  
  236.  
  237.     def pööra(self, kaart, kasVastupäeva):
  238.         if kasVastupäeva:
  239.             indeks = self.pööre + 1
  240.         else:
  241.             indeks = self.pööre - 1
  242.  
  243.         indeks %= len(self.pöörded)
  244.         uusKujund = self.pöörded[indeks]
  245.         kasSaabPöörata = True
  246.  
  247.         #Vaatame, kas üldse saab pöörata
  248.         for x in range(len(uusKujund)):
  249.             for y in range(len(uusKujund[0])):
  250.                 if self.x + x < 0 or self.x + x >= len(kaart.kaart) or self.y + y < 0 or self.y + y >= len(kaart.kaart[0]):
  251.                     #Pööre viib kaardilt välja, ei saa pöörata
  252.                     kasSaabPöörata &= False
  253.                     continue
  254.                
  255.                 if uusKujund[x][y] and (kaart.kaart[self.x + x][self.y + y] != 0 and kaart.kaart[self.x + x][self.y + y] != self):
  256.                     #Uus ruut ei ole vaba ruut, ei saa pöörata
  257.                     kasSaabPöörata &= False
  258.  
  259.         print("Saab pöörata: " + str(kasSaabPöörata))
  260.         if kasSaabPöörata:
  261.             self.kustutaKaardilt(kaart)
  262.             self.pööre = indeks
  263.             self.kujund = uusKujund
  264.             self.asetaKaardile(kaart)
  265.            
  266.  
  267.     def kustutaKaardilt(self, kaart):
  268.         for x in range(len(self.kujund)):
  269.                 for y in range(len(self.kujund[0])):
  270.                     if self.kujund[x][y]:
  271.                         kaart.kaart[self.x + x][self.y + y] = 0
  272.  
  273.  
  274.     def asetaKaardile(self, kaart):
  275.         for x in range(len(self.kujund)):
  276.                 for y in range(len(self.kujund[0])):
  277.                     if self.kujund[x][y]:
  278.                         kaart.kaart[self.x + x][self.y + y] = self
  279.  
  280.        
  281.  
  282. class KujundiGeneraator:
  283.  
  284.     def __init__(self):
  285.         self.kujundid = [
  286.             [ # --- kujund
  287.                 [[False, False, False, False], [True, True, True, True]],
  288.                 [[False, True], [False, True], [False, True], [False, True]]
  289.             ],
  290.             [ # _|_ kujund
  291.                 [[False, True, False], [True, True, True]],
  292.                 [[False, True, False], [False, True, True], [False, True, False]],
  293.                 [[False, False, False], [True, True, True], [False, True, False]],
  294.                 [[False, True], [True, True], [False, True]]
  295.             ],
  296.             [ # kasti kujund
  297.                 [[True, True], [True, True]]
  298.             ],
  299.             [ # |__ kujund
  300.                 [[True, False, False], [True, True, True]],
  301.                 [[True, True], [True, False], [True, False]],
  302.                 [[True, True, True], [False, False, True]],
  303.                 [[False, True], [False, True], [True, True]]
  304.             ],
  305.             [ # __| kujund
  306.                 [[True, True, True], [True, False, False]],
  307.                 [[True, True], [False, True], [False, True]],
  308.                 [[False, False, True], [True, True, True]],
  309.                 [[True, False], [True, False], [True, True]]
  310.             ],
  311.             [ # _|- kujund
  312.                 [[True, True, False], [False, True, True]],
  313.                 [[False, True], [True, True], [True, False]],
  314.             ],
  315.             [ # -|_ kujund
  316.                 [[False, True, True], [True, True, False]],
  317.                 [[True, False], [True, True], [False, True]],
  318.             ],
  319.         ]
  320.         self.värvid = [
  321.             (100, 0, 0),
  322.             (100, 100, 0),
  323.             (0, 100, 0),
  324.             (0, 100, 100),
  325.             (0, 0, 100),
  326.             (100, 0, 100)
  327.         ]
  328.  
  329.     def genereeri(self):
  330.  
  331.         return Kujund(10, 0, random.choice(self.kujundid), random.choice(self.värvid))
  332.    
  333. #Alustame Pygame asju
  334. pygame.init()
  335. #Vastava suurusega aken
  336. ekraan = pygame.display.set_mode([Ruut.laius * Kaart.laius, Ruut.laius * Kaart.kõrgus + 20])
  337. #Meie kaart
  338. kaart = Kaart()
  339. skoor = Skoor()
  340.  
  341. kujundiGeneraator = KujundiGeneraator()
  342. kujund = kujundiGeneraator.genereeri()
  343. kujund.x = len(kaart.kaart) // 2
  344. kaart.lisaKujund(kujund)
  345.  
  346. UUENDUS = pygame.USEREVENT + 1
  347. pygame.time.set_timer(UUENDUS, 500)
  348.  
  349. LIIGU = pygame.USEREVENT + 2
  350. PÖÖRA_VP = pygame.USEREVENT + 3
  351. PÖÖRA_PP = pygame.USEREVENT + 4
  352.  
  353.  
  354. while True:
  355.     ekraan.fill((0,0,0))
  356.  
  357.     kaart.joonista()
  358.     skoor.joonista(ekraan)
  359.  
  360.     pygame.display.flip()
  361.     pygame.time.delay(10)
  362.  
  363.     events = pygame.event.get()
  364.     for oneEvent in events:
  365.         if oneEvent.type == UUENDUS and len(kaart.täisRead) > 0:
  366.             pygame.time.set_timer(UUENDUS, 0)
  367.             skoor.suurenda(len(kaart.täisRead))
  368.             kaart.kaotaRead()
  369.             kaart.joonista()
  370.             pygame.time.set_timer(UUENDUS, 500)
  371.         else:
  372.             if oneEvent.type == UUENDUS:
  373.                 if not kujund.kasElus and len(kaart.täisRead) == 0:
  374.                     kujund = kujundiGeneraator.genereeri()
  375.                     kujund.x = len(kaart.kaart) // 2
  376.                     kaart.lisaKujund(kujund)
  377.                     print("Lisa kujund")
  378.                 else:
  379.                     kaart.uuenda(kujund)
  380.  
  381.             if oneEvent.type == pygame.KEYDOWN:
  382.                 if oneEvent.key == pygame.K_q:
  383.                     kujund.pööra(kaart, True)
  384.                     pygame.time.set_timer(PÖÖRA_VP, 100)
  385.                 elif oneEvent.key == pygame.K_w:
  386.                     kujund.pööra(kaart, False)
  387.                     pygame.time.set_timer(PÖÖRA_PP, 100)
  388.                 else:
  389.                     kujund.liigu(kaart, oneEvent.key)
  390.                     pygame.time.set_timer(LIIGU, 100)
  391.  
  392.             klahvid = pygame.key.get_pressed()
  393.             if oneEvent.type == PÖÖRA_VP:
  394.                 if klahvid[pygame.K_q]:
  395.                     kujund.pööra(kaart, True)
  396.                 else:
  397.                     pygame.time.set_timer(PÖÖRA_VP, 0)
  398.             elif oneEvent.type == PÖÖRA_PP:
  399.                 if klahvid[pygame.K_w]:
  400.                     kujund.pööra(kaart, False)
  401.                 else:
  402.                     pygame.time.set_timer(PÖÖRA_PP, 0)
  403.             elif oneEvent.type == LIIGU:
  404.                 if klahvid[pygame.K_DOWN]:
  405.                     kujund.liigu(kaart, pygame.K_DOWN)
  406.                 elif klahvid[pygame.K_LEFT]:
  407.                     kujund.liigu(kaart, pygame.K_LEFT)
  408.                 elif klahvid[pygame.K_RIGHT]:
  409.                     kujund.liigu(kaart, pygame.K_RIGHT)
  410.                 else:
  411.                     pygame.time.set_timer(LIIGU, 0)
  412.                
  413.         if oneEvent.type == pygame.QUIT:
  414.             #quit()
  415.             exit()
  416.             break;
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement