object main extends App
{
    var alphabet = ('A' to 'Z').toArray
    var alphabetSize = alphabet.size
    var codingTable = Array.ofDim[Char](alphabetSize, alphabetSize)

    var a = 0
    var b = 0
    while(a < alphabetSize)
    {
        while(b < alphabetSize)
        {
            if(a + b < alphabetSize)
            {
                codingTable(a)(b) = alphabet(a + b)
            }
            else
            {
                codingTable(a)(b) = alphabet(a + b - alphabetSize)
            }
            b += 1
        }
        b = 0
        a += 1
    }

    var i = 0
    while(25 >= i)
    {
        var j = 0
        while(25 >= j)
        {
            print(codingTable(i)(j)+" ")
            j += 1
        }
        print("\n")
        i += 1
    }

    var keyWord = new Array[Char](5)
    keyWord = Array('K', 'A', 'T', 'A', 'R')
    print("\n"+"\n")

    print("Podaj napis do zaszyfrowania, z duzych liter i bez spacji: ")
    var n = io.StdIn.readLine()
    var codingWord = new Array[Char](n.length)
    var c = 0
    while(n.length > c)
    {
        codingWord(c) = n(c)
        c += 1
    }

    var d = 0
    print("\n"+"Napis do zaszyfrowania: ")
    while(n.length > d)
    {
        print(codingWord(d))
        d += 1
    }

    var h = 0
    var u = 0
    print("\n"+"Slowo szyfrujace:       ")
    while(n.length > h)
    {
        print(keyWord(u))
        u += 1
        if(u == 5)
        {
            u = 0
        }
        h += 1
    }
    print("\n"+"\n")

    print("Szyf:                   ")
    var x = 0
    var y = 0
    var z = codingTable(x)(y)
    while(codingWord.length > x)
    {
        if(y == 5)
        {
            y = 0
        }
        z = codingTable(codingWord(x) - 'A')(keyWord(y) - 'A')
        print(z)
        x += 1
        y += 1
    }
    print("\n"+"\n")
}