#include "ballsort.h"
#include <iostream>

using namespace CS128;
using namespace std;

int main(int argc, char **argv)
{
    if (argc < 3) {
        cerr << "usage: sort NUMBER DELAY\n";
        exit(1);
    }

    int size = atoi(argv[1]);
    int delay = atoi(argv[2]);

    Ball list[size];

    for (int i = 0; i < size; ++i) {
        initialize(list[i]);
    }

    bool flag;

    do {
        flag = false;
        for (int i = 0; i < size - 1; ++i) {
            if (radius(list[i]) < radius(list[i+1])) {
                Ball tmp = list[i];
                list[i] = list[i + 1];
                list[i + 1] = tmp;
                flag = true;
            }
            display(list, i, size, delay);
        }

    } while (flag);

    return 0;
}