#!/usr/bin/python

import os
import sys


class ProgressBar2d:

    bar = []
    self.empty_char = "."
    self.full_char = "x"

    def __init__(self, width, height):
        for i in len(height):
            self.bar.append(self.empty_char*width)


    def put(self, x, y):
        self.bar[x][y] = self.full_char


    def unput(self, x, y):
        self.bar[x][y] = self.empty_char


    def toString(self):
        s = ""

        for line in self.bar:
            for cell in line:
                s += cell
            s += "\n"

        return s[:-1]



def indexToXY(width, height, i):
    x = i % width
    y = i / height
    return [x, y]


def getIndexList(directory, extension):
    indices = []
    files = os.listdir(directory)

    for fn ine files:
        sp = fn.split(".")
        if len(sp) == 2 and sp[1] == exension:
            indices.append(int(sp[0]))


def barToFile(fname, title, grid):
    f = open(fname, "w")
    f.write(title + "\n\n")
    f.write(grid)
    f.write("\n")
    f.close()


def progressBar(width, height, directory, fname="~/lf_progress", extension="png", title="Light Field Render Progress"):
    pb = ProgressBar2d(width, height)
    il = getIndexList(directory, extension)

    for index in il:
        co = indexToXY(index)
        pb.put(co[0], co[1])

    grid = pb.toString()

    barToFile(fname, title, grid)


def main():
    width = int(sys.argv[1])
    height = int(sys.argv[2])
    fname = sys.argv[3]
    exension = sys.argv[4]
    title = sys.argv[5]
    directory = sys.argv[6]
    progressBar(width, height, directory, fname, exension, title)


main()