tetris.lua

local gpu=require("component").gpu
local event=require("event")

fieldWidth=20
fieldHeight=26

width=30 --width of mas NOT field

mas={}

StockBlocks = { { start_position = { x = 4, y = 1 },
                  color = { 175, 175, 0 },
                  symbol = "T",
                  transformations = { { { x = -1, y = 0 },
                                        { x = 0, y = 0 },
                                        { x = 1, y = 0 },
                                        { x = 0, y = -1 } },
                                      { { x = 0, y = 0 },
                                        { x = 0, y = -1 },
                                        { x = 0, y = 1 },
                                        { x = 1, y = 0 } },
                                      { { x = -1, y = 0 },
                                        { x = 0, y = 0 },
                                        { x = 1, y = 0 },
                                        { x = 0, y = 1 } },
                                      { { x = 0, y = 0 },
                                        { x = 0, y = -1 },
                                        { x = 0, y = 1 },
                                        { x = -1, y = 0 } } } },
                { start_position = { x = 4, y = 1 },
                  color = { 0, 255, 0 },
                  symbol = "Z",
                  transformations = { { { x = -1, y = 0 },
                                        { x = 0, y = 0 },
                                        { x = 0, y = -1 },
                                        { x = 1, y = -1 } },
                                      { { x = 0, y = -1 },
                                        { x = 0, y = 0 },
                                        { x = 1, y = 0 },
                                        { x = 1, y = 1 } } } },
                { start_position = { x = 4, y = 1 },
                  color = { 100, 100, 175 },
                  symbol = "S",
                  transformations = { { { x = 1, y = 0 },
                                        { x = 0, y = 0 },
                                        { x = 0, y = -1 },
                                        { x = -1, y = -1 } },
                                      { { x = 0, y = 1 },
                                        { x = 0, y = 0 },
                                        { x = 1, y = 0 },
                                        { x = 1, y = -1 } } } },
                { start_position = { x = 4, y = 0 },
                  color = { 255, 255, 100 },
                  symbol = "J",
                  transformations = { { { x = -1, y = 0 },
                                        { x = 0, y = 0 },
                                        { x = 1, y = 0 },
                                        { x = -1, y = 1 } },
                                      { { x = 0, y = -1 },
                                        { x = 0, y = 0 },
                                        { x = 0, y = 1 },
                                        { x = 1, y = 1 } },
                                      { { x = -1, y = 0 },
                                        { x = 0, y = 0 },
                                        { x = 1, y = 0 },
                                        { x = 1, y = -1 } },
                                      { { x = 0, y = -1 },
                                        { x = 0, y = 0 },
                                        { x = 0, y = 1 },
                                        { x = -1, y = -1 } } } },
                { start_position = { x = 4, y = 0 },
                  color = { 255, 0, 255 },
                  symbol = "L",
                  transformations = { { { x = -1, y = 0 },
                                        { x = 0, y = 0 },
                                        { x = 1, y = 0 },
                                        { x = 1, y = 1 } },
                                      { { x = 0, y = -1 },
                                        { x = 0, y = 0 },
                                        { x = 0, y = 1 },
                                        { x = 1, y = -1 } },
                                      { { x = -1, y = 0 },
                                        { x = 0, y = 0 },
                                        { x = 1, y = 0 },
                                        { x = -1, y = -1 } },
                                      { { x = 0, y = -1 },
                                        { x = 0, y = 0 },
                                        { x = 0, y = 1 },
                                        { x = -1, y = 1 } } } },
                { start_position = { x = 4, y = 0 },
                  color = { 255, 0, 0 },
                  symbol = "I",
                  transformations = { { { x = -1, y = 0 },
                                        { x = 0, y = 0 },
                                        { x = 1, y = 0 },
                                        { x = 2, y = 0 } },
                                      { { x = 0, y = -1 },
                                        { x = 0, y = 0 },
                                        { x = 0, y = 1 },
                                        { x = 0, y = 2 } } } },
                { start_position = { x = 4, y = 0 },
                  color = { 0, 0, 255 },
                  symbol = "[]",
                  transformations = { { { x = 0, y = 0 },
                                        { x = 0, y = 1 },
                                        { x = 1, y = 0 },
                                        { x = 1, y = 1 } } } }
             }

function toXY(num,wid)
  wid=wid or width
  y=math.floor((num-1)/wid)+1
  return num-(y-1)*width,y
end
function fromXY(x,y,wid)
  wid=wid or width
  return (y-1)*wid+x
end

for i=-1000,2500 do
  mas[i]=0
end
--print(StockBlocks[1].transformations[1][1].y)
--[[f=7
tr=1
for i=1,4 do
  gpu.set(StockBlocks[f].transformations[tr][i].x+5,StockBlocks[f].transformations[tr][i].y+5,"!")
end]]--
obj={
  id=-1,
  x=-1,
  y=-1,
  dX=0,
  dY=0,
  dR=0,
  rotate=-1,
}
function newTile(tile)
  tile=tile or (math.random(1,7))
  obj.id=tile
  obj.x,obj.y=StockBlocks[tile].start_position.x+5,StockBlocks[tile].start_position.y
  obj.rotate=1
  if test() then
    gpu.setForeground(0xff0000)
    gpu.set(10,1,"GAME OVER")
    os.sleep(0.5)
    gpu.setForeground(0xffffff)
    isRunning=false
  else
    clearLines()
  end
--  print("!")
end
function drawBlock(x,y,color)
  gpu.set((x-1)*2+1,y,"  ")
end
function drawTile(color)
  gpu.setBackground(color or 0x005500)
--  print(obj.id,obj.rotate)
  for i=1,4 do
    drawBlock(StockBlocks[obj.id].transformations[obj.rotate][i].x+obj.x,StockBlocks[obj.id].transformations[obj.rotate][i].y+obj.y)
  end
  gpu.setBackground(0x000000)
end
function testLine(line)
  for x=1,fieldWidth-1 do
    if mas[fromXY(x,line)]==0 then
      return false
    end
  end
  return true
end
function clearLines()
--  gpu.set(90,1,"!")
  for i=1,fieldHeight do
    if testLine(i) then
      downTiles(i)
--      gpu.set(90,2,"!")
      gpu.set(1,i,string.rep(" ",fieldWidth*2-2))
      os.sleep(0.5)
      gpu.copy(1,i-20,fieldWidth*2-1,20,0,1)
    end
  end
end
function downTiles(line)
  for i=line,1,-1 do
    for j=1,fieldWidth+2 do
      mas[fromXY(j,i)]=mas[fromXY(j,i-1)]
    end
  end
end
function test()
  for i=1,4 do
    local x,y=StockBlocks[obj.id].transformations[obj.rotate][i].x+obj.x,StockBlocks[obj.id].transformations[obj.rotate][i].y+obj.y
    if (y==fieldHeight)or(mas[fromXY(x,y)]~=0)or(x>=fieldWidth)or(x<1) then
      return true
    end
  end
  return false
end
function putInMap()
  for j=1,4 do
    mas[fromXY(StockBlocks[obj.id].transformations[obj.rotate][j].x+obj.x,StockBlocks[obj.id].transformations[obj.rotate][j].y+obj.y)]=1
  end
end
function mapUpdate()
  drawTile(0x000000)
  if step2 then
    obj.y=obj.y+1
  end
--  obj.x=obj.x+obj.dX
  if (test())and(step2) then
--    obj.x=obj.x-obj.dX
      obj.y=obj.y-1
    drawTile()
    putInMap()
    newTile()
  else
    if obj.dY==1 then
      drawTile(0x000000)
      obj.y=obj.y+1
      if test() then
        obj.y=obj.y-1
        drawTile()
        putInMap()
        newTile()
      end
    end
    if obj.dX~=0 then
      drawTile(0x000000)
      obj.x=obj.x+obj.dX
      if test() then
        obj.x=obj.x-obj.dX
      end
    end
    if obj.dR~=0 then
      local tmp=obj.rotate
      drawTile(0x000000)
      obj.rotate=(obj.rotate+#StockBlocks[obj.id].transformations)%(#StockBlocks[obj.id].transformations)+1
      if test() then
        obj.rotate=tmp
      end
--      obj.rotate=obj.rotate+1
    end
  end
  obj.dX=0
  obj.dY=0
  obj.dR=0
  drawTile()
end
function onKey(ev,_,code1,code2,player)
  if (code1==113)and(code2==16) then
    isRunning=false
  elseif (code1==0) then
    if code2==203 then--left
      obj.dX=-1
    elseif code2==200 then--up
      obj.dR=1
    elseif code2==205 then--right
      obj.dX=1
    elseif code2==208 then--down
      obj.dY=1
    end
  end
end
newTile()
drawTile()
isRunning=true
event.ignore("key_down",onKey)
event.listen("key_down",onKey)
step2=false
gpu.setBackground(0x222222)
require("term").clear()
gpu.setBackground(0x000000)
gpu.fill(1,1,fieldWidth*2-2,fieldHeight-1," ")
while isRunning do
  step2=not step2
  os.sleep(0.15)
  mapUpdate()
--  for i=1,fieldWidth do
--    gpu.set(70+i,3,tostring(mas[fromXY(i,25)]))
--  end
end
event.ignore("key_down",onKey)