import gpanel  #um bilder einzulesen
import os  #um Verzeichnisse zu listen
import csv #um CSV-Dateien zu lesen.

# Pfad zu den Bilddateien
digitsdirectory = 'C:/temp/digits/train/'


def getPixeListFromFilePath(filepath):
    img = gpanel.getImage(filepath)
    w = img.getWidth()
    h = img.getHeight()
    pixellist = []
    for y in range(h):
        for x in range(w):

            # color is ein Objekt mit verschiedenen Attributen, u.a. red, green, blue.
            # bei grau sind rot=gruen=blau, d.h., eine Farbe auslesen reicht.
            # siehe auch https://docs.oracle.com/javase/7/docs/api/java/awt/Color.html
            color = img.getPixelColor(x, y)

            # umlegen auf das Intervall [-1,1] zwecks Normalisierung
            value = color.red / 255 * 2 - 1
            # an liste anhaengen
            pixellist.append(value)
            
    return pixellist

# Lese Ziffer aus Dateiname aus.
def getDigitFromFileName(filename):
    return int(filename.split('_', 3)[2])


# leere Liste fuer alle Trainingsdaten der Form [-0.93,0.331,....,0.99,3]

trainingset = []

# durch alle files im Ziffernverzeichnis loopen
for filename in [filename for filename in os.listdir(digitsdirectory) if filename.endswith("gif")]:

    # Ziffer auslesen
    currdigit = getDigitFromFileName(filename)

    # Pixelliste von Datei auslesen
    currpixellist = getPixeListFromFilePath(digitsdirectory + filename)

    # Der Pixelliste die Ziffer anhaengen
    currpixellist.append(currdigit)

    # Gesamte Liste dem trainingsset anhaengen.
    trainingset.append(currpixellist)

# Das Trainingsset kann jetzt verwendet werden
# print(trainingsset)