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
    var j = 0
    while(25 >= i)
    {
        j = 0
        while(25 >= j)
        {
            print(codingTable(i)(j)+" ")
            j += 1
        }
        print("\n")
        i += 1
    }

    var keyWord = new Array[Char](6)
    keyWord = Array('S', 'E', 'K', 'I', 'R', 'O')
    print("\n"+"\n"+"Podaj napis do zaszyfrowania z duzych liter: ")
    var n = io.StdIn.readLine()
    var codingWord = new Array[Char](n.length)
    var codingWord_length = codingWord.length

    i = 0
    while(n.length > i)
    {
        codingWord(i) = n(i)
        i += 1
    }

    i = 0
    while(codingWord_length > i)
    {
        if(codingWord(i) == ' ')
        {
            j = 0
            while(codingWord_length > (i + j + 1))
            {
                codingWord(i + j) = codingWord(i + j +1)
                j += 1
            }
            codingWord_length -= 1
        }
        i += 1
    }

    i = 0
    print("\n"+"Napis do zaszyfrowania: ")
    while(codingWord_length > i)
    {
        print(codingWord(i))
        i += 1
    }

    i = 0
    j = 0
    print("\n"+"Slowo szyfrujace:       ")
    while(codingWord_length > i)
    {
        print(keyWord(j))
        j += 1
        if(j == 6)
        {
            j = 0
        }
        i += 1
    }

    print("\n"+"\n"+"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")
}