#include <iostream>
#include "zasobnik.h"
using namespace std;

Zasobnik::Zasobnik(int maxVelkost, int cislo)
{
    pocetVagonu = 0;
    vagony = new Vagon*[maxVelkost];
    kapacita = maxVelkost;
    cisloZasobniku = cislo;
}

Zasobnik::~Zasobnik()
{
    delete vagony;
}

void Zasobnik::Vloz(Vagon* vagon)
{
    if (pocetVagonu + 1 <= kapacita)
    {
        vagony[pocetVagonu] = vagon;
        pocetVagonu++;
        cout << "Z" << vagon->Cislo() << "[" << cisloZasobniku << "]" << " ";
    }
    else
        cout << "Zasobnik " << cisloZasobniku << " je plny, vagon sa nepridal!" << endl;
}

Vagon* Zasobnik::Vyber()
{
    if (pocetVagonu > 0)
    {
        Vagon* vagon = vagony[pocetVagonu - 1];
        cout << "X" << vagon->Cislo() << "[" << cisloZasobniku << "]" << " ";
        pocetVagonu--;
        return vagon;
    }
    else
    {
        cout << "Zasobnik " << cisloZasobniku << " je prazdny!" << endl;
        return NULL;
    }
}

Vagon* Zasobnik::Nazri()
{
    return pocetVagonu > 0 ? vagony[pocetVagonu - 1] : NULL;
}

int Zasobnik::PocetVagonu()
{
    return pocetVagonu;
}

bool Zasobnik::Prazdny()
{
    return pocetVagonu == 0;
}

bool Zasobnik::Plny()
{
    return pocetVagonu == kapacita;
}

int Zasobnik::Cislo()
{
    return cisloZasobniku;
}