【Minecraft】タートルの一人で建築できるもん！【CCの道も一行から＃2】

-- const

-- コマンドライン引数を取得
local args = {...}

-- 建築サイズを定義
local X_SIZE, Y_SIZE, Z_SIZE = 16, 4 * (args[1] or 1), 16

-- 使用ブロックを指定
local BLOCKS = {
    corner   = 'minecraft:polished_deepslate',
    wall     = 'minecraft:stone_bricks',
    floorEdge= 'minecraft:smooth_stone',
    floor    = 'minecraft:smooth_stone_slab',
    window   = 'minecraft:glass_pane',
}

-- 窓ガラスの位置を指定
local WINDOWS_I = {2, 3, 5, 6, 9, 10, 12, 13}


-- function

-- テーブルがある値を含むか
table.includes = function (t, x)
    for k, v in pairs(t) do
        if v == x then
            return true
        end
    end
    return false
end

-- 壁に窓をつけるか
function isWindow(i)
    -- i: 端を 1 として、端から何ブロック目か
    return table.includes(WINDOWS_I, i)
end

-- 建築家位置からの相対座標から、設置すべきブロックを返す
function getBlock(x, y, z)
    -- X軸・Z軸 に平行な壁面であるか
    local isEdgeX, isEdgeZ = x == 0 or x == X_SIZE - 1, z == 0 or z == Z_SIZE - 1
    
    -- X軸・Z軸 両方の壁面 ⇔ 四隅の柱
    if isEdgeX and isEdgeZ then
        return BLOCKS.corner
    -- 四ブロックごとに床を張る
    elseif y % 4 == 3 then
        if isEdgeX or isEdgeZ then
            return BLOCKS.floorEdge
        else
            return BLOCKS.floor
        end
    elseif isEdgeX or isEdgeZ then
        local i -- 端から何ブロック目か
        if isEdgeX then
            i = z
        else
            i = x
        end
        if y % 4 == 1 and isWindow(i) then
            return BLOCKS.window
        else
            return BLOCKS.wall
        end
    end
    return nil
end

-- タートルが持っているアイテムを数える
function countItems()
    local counts = {}
    local detail
    for slot = 1, 16 do
        detail = turtle.getItemDetail(slot)
        if detail ~= nil then
            counts[detail.name] = (counts[detail.name] or 0) + detail.count
        end
    end
    return counts
end

-- 必要な建材の量を数える
function countMaterials()
    local counts = {}
    for y = 0, Y_SIZE - 1 do
        for z = 0, Z_SIZE - 1 do
            for x = 0, X_SIZE - 1 do
                block = getBlock(x, y, z)
                if block ~= nil then
                    counts[block] = (counts[block] or 0) + 1
                end
            end
        end
    end
    return counts
end

-- 総移動回数を求める
function countMove()
    return (X_SIZE - 1) * 2 * Z_SIZE * Y_SIZE + (Z_SIZE - 1) * 2 * Y_SIZE + Y_SIZE
end

-- 建築を実行可能か確かめる
function canBuild()
    local itemCounts = countItems()
    local materials = countMaterials()
    
    -- すべての建材が必要数用意されているかをチェックする
    for name, needs in pairs(materials) do
        local extra = (itemCounts[name] or 0) - needs
        if extra < 0 then
            -- 何か一つでも足りなかったら、false を返す
            return false, ("%s need more %d."):format(name, -extra)
        end
    end

    local moveCount = countMove()
    local fuelLevel = turtle.getFuelLevel()
    if fuelLevel ~= 'unlimited' and moveCount > fuelLevel then
        -- 燃料が足りなければ、false を返す
        return false, ("need more fuel. (+%d)"):format(moveCount - fuelLevel)
    end
    return true
end

-- あるアイテムを選択する
function selectItem(name)
    for slot = 1, 16 do
        detail = turtle.getItemDetail(slot)
        if detail ~= nil then
            if detail.name == name then
                turtle.select(slot)
                return true
            end
        end
    end
    return false
end

-- 建築する
function build()
    for y = 0, Y_SIZE - 1 do
        turtle.up()
        for z = 0, Z_SIZE - 1 do
            for x = 0, X_SIZE - 1 do
                block = getBlock(x, y, z)
                if block ~= nil then
                    selectItem(block)
                    turtle.placeDown()
                end
                if x ~= X_SIZE - 1 then
                    turtle.forward()
                end
            end
            for i = 1, X_SIZE - 1 do
                turtle.back()
            end
            if z ~= Z_SIZE - 1 then
                turtle.turnRight()
                turtle.forward()
                turtle.turnLeft()
            end
        end
        turtle.turnLeft()
        for i = 1, Z_SIZE - 1 do
            turtle.forward()
        end
        turtle.turnRight()
    end
end

function main()
    local isPossible, reason = canBuild()

    -- 建築不可能なら理由を表示
    if not isPossible then
        print(reason)
        return false
    end

    print("OK")
    build()
    return true
end


-- main

main()