minetris

python2-scripts/mcpipy/minetris.py

@@ -0,0 +1,186 @@
+#
+# requires Windows and pywin32
+#
+from mc import *
+from time import sleep,time
+from random import randint
+import win32con,win32api
+
+HEIGHT = 22
+WIDTH = 9
+BORDER = WOOL_BLACK
+PIECES = (  (('XXXX',), ('.X','.X','.X','.X')),
+			(('XX','XX'),),
+			(('XXX', '..X'), ('.X', '.X', 'XX'), ('X','XXX'), ('XX', 'X', 'X')),
+			(('XXX', 'X'), ('XX', '.X', '.X'), ('..X', 'XXX'), ('X.', 'X.', 'XX')),
+			(('XX', '.XX'), ('.X','XX', 'X.')),
+			(('.XX', 'XX'), ('X.', 'XX', '.X')) )
+
+def drawBoard():
+	mc.setBlocks(left-1, bottom-1, plane, left+WIDTH, bottom-1, plane, BORDER)
+	mc.setBlocks(left-1, bottom+HEIGHT, plane, left+WIDTH, bottom+HEIGHT, plane, BORDER)
+	mc.setBlocks(left-1, bottom, plane, left, bottom+HEIGHT-1, plane, BORDER)
+	mc.setBlocks(left+WIDTH, bottom, plane, left+WIDTH, bottom+HEIGHT-1, plane, BORDER)
+	mc.setBlocks(left, bottom, plane, left+WIDTH-1, bottom+HEIGHT-1, plane, AIR)
+
+def pieceWidth(piece):
+	return max((len(a) for a in piece[orientation]))
+
+def enumeratePiece(x, y, piece):
+	for row in range(len(piece)):
+		for col in range(len(piece[row])):
+			if piece[row][col] == 'X':
+				yield (x+col,y-row)
+
+def erasePiece(buffer, x, y, piece):
+	for (xx,yy) in enumeratePiece(x,y, piece):
+		buffer[(left+xx,bottom+yy)] = AIR
+
+def drawPiece(buffer, x, y, piece, color):
+	for (xx,yy) in enumeratePiece(x,y, piece):
+		if (left+xx,bottom+yy) in buffer and buffer[(left+xx,bottom+yy)] == AIR:
+			del buffer[(left+xx,bottom+yy)]
+		else:
+			buffer[(left+xx,bottom+yy)] = color
+
+def drawBuffer(buffer):
+	for x,y in buffer:
+		mc.setBlock(x,y,plane,buffer[(x,y)])
+
+def placePiece():
+	global pieceNum, color, family, orientation, x, y, fall, descendDelay
+	pieceNum = randint(0, len(PIECES)-1)
+	color = Block(WOOL.id, pieceNum % 16)
+	family = PIECES[pieceNum]
+	orientation = 0
+	x = WIDTH // 2 - pieceWidth(family[orientation])
+	y = HEIGHT - 1		
+	descendDelay = currentDescendDelay
+	fall = False
+
+def fit(x, y, piece):
+	for (xx,yy) in enumeratePiece(x, y, piece):
+		if yy >= HEIGHT or yy < 0 or xx >= WIDTH or xx < 0 or board[xx][yy] is not None:
+			return False
+	return True
+
+
+def descend():
+	global descendTimer
+	if descendTimer + descendDelay <= time():
+		descendTimer += descendDelay
+		return True
+	return False
+
+def moveDown():
+	return (win32api.GetAsyncKeyState(win32con.VK_DOWN)&1) or (win32api.GetAsyncKeyState(ord(' '))&1)
+
+def moveLeft():
+	return (win32api.GetAsyncKeyState(win32con.VK_LEFT)&1)
+
+def moveRight():
+	return (win32api.GetAsyncKeyState(win32con.VK_RIGHT)&1)
+
+def rotateLeft():
+	return (win32api.GetAsyncKeyState(win32con.VK_PRIOR)&1)
+
+def rotateRight():
+	return (win32api.GetAsyncKeyState(win32con.VK_NEXT)&1) or (win32api.GetAsyncKeyState(win32con.VK_UP)&1)
+
+def addPiece(x, y, piece, color):
+	for (xx,yy) in enumeratePiece(x, y, piece):
+		board[xx][yy] = color
+	while True:
+		foundRow = False
+		for y in range(HEIGHT):
+			full = True
+			for x in range(WIDTH):
+				if board[x][y] is None:
+					full = False
+					break
+			if full:
+				foundRow = True
+				for y2 in range(y, HEIGHT-1):
+					for x in range(WIDTH):
+						b = board[x][y2+1]
+						board[x][y2] = b
+						mc.setBlock(left+x,bottom+y2,plane,b if b is not None else AIR)
+				for x in range(WIDTH):
+					board[x][HEIGHT-1] = None
+					mc.setBlock(left+x,bottom+HEIGHT-1,plane,AIR)
+
+		if not foundRow:
+			break
+
+
+mc = Minecraft()
+
+playerPos = mc.player.getTilePos()
+mc.player.setRotation(180)
+mc.player.setTilePos(playerPos.x, playerPos.y, playerPos.z + 10)
+left = playerPos.x - WIDTH // 2
+plane = playerPos.z
+bottom = playerPos.y + 1
+board = [[None for i in range(HEIGHT)] for j in range(WIDTH)]
+currentDescendDelay = 0.5
+
+drawBoard()
+
+newPiece = True
+
+while True:
+	if newPiece:
+		placePiece()
+		oldPiece = None
+		if not fit(x, y, family[orientation]):
+			mc.postToChat("Doesn't fit: End of game")
+			print "no fit"
+			break
+		draw = True
+		newPiece = False
+		descendTimer = time()
+	else:
+		oldPiece = family[orientation]
+		draw = False
+	oldX = x
+	oldY = y
+
+	if not fall:
+		if moveLeft() and fit(x-1, y, family[orientation]):
+			x -= 1
+			draw = True
+
+		if moveRight() and fit(x+1, y, family[orientation]):
+			x += 1
+			draw = True
+
+		if rotateLeft() and fit(x, y, family[(orientation-1)%len(family)]):
+			orientation = (orientation-1)%len(family)
+			draw = True
+
+		if rotateRight() and fit(x, y, family[(orientation+1)%len(family)]):
+			orientation = (orientation+1)%len(family)
+			draw = True
+
+		if moveDown():
+			fall = True
+			descendDelay = 0.05
+
+	if descend():
+		if not fit(x, y-1, family[orientation]):
+			addPiece(x, y, family[orientation], color)
+			newPiece = True
+		else:
+			draw = True
+			y -= 1
+
+	if draw:
+		buffer = {}
+		if oldPiece:
+			erasePiece(buffer, oldX, oldY, oldPiece)
+		drawPiece(buffer, x, y, family[orientation], color)
+		drawBuffer(buffer)
+
+	sleep(0.1)
+
+print "done"