#include <iostream>
#include "time.h"
#include "dice.h"
#include "print.h"

using namespace std;

int main(int, char**){
    srand (time(NULL));

    int seitenzahl1 = 0;
    int seitenzahl2 = 0;
    int wuerfe = 0;
    int anzahl_wuerfel = 1;

    cout << "WÜRFEL" << endl;
    cout << "Sollen 1 oder 2 Würfel geworfen werden? ";
    cin >> anzahl_wuerfel;

    if (anzahl_wuerfel == 1)
    {
        cout << "Seitenzahl: " << endl;
        cin >> seitenzahl1;
        cout << "Anzahl der Würfe: " << endl;
        cin >> wuerfe;


        int *ergebnisse = new int[seitenzahl1]();

        for (int i = 1; i <= wuerfe; i++)
        {
            ergebnisse[roll_dice(seitenzahl1)-1] += 1;
        }

        // Ausgabe
        for (int i = 1; i <= seitenzahl1; i++)
        {
            print(i, ergebnisse[i-1]);
        }
    }
    
    else if (anzahl_wuerfel == 2)
    {
        cout << "WÜRFEL 1: Seitenzahl: ";
        cin >> seitenzahl1;
        cout << "WÜRFEL 2: Seitenzahl: ";
        cin >> seitenzahl2;
        cout << "Anzahl der Würfe: " << endl;
        cin >> wuerfe;

        int array_length = seitenzahl1 + seitenzahl2;
        int *ergebnisse = new int[array_length]();

        for (int i = 1; i <= wuerfe; i++)
        {
            ergebnisse[roll_2_dice(seitenzahl1, seitenzahl2)-1] += 1;
        }

        // Ausgabe
        for (int i = 1; i <= array_length; i++)
        {
            print(i, ergebnisse[i-1]);
        }
    }
    else
    {
        cout << "Sie haben keine gültige Anzahl angegeben!" << endl;
    }

}