//
// Created by Nik on 11/04/2022.
//

#ifndef NALOGA0602_VLAK_H
#define NALOGA0602_VLAK_H

#include <vector>
#include "Prikljucek.h"
#include "Lokomotiva.h"
#include "Vagon.h"
#include <random>

class Vlak {
private:
    std::vector<Prikljucek*> vlak;
public:
    void addPriklucek(Prikljucek *p);
    std::string toString() const;
    static std::vector<Prikljucek*> generate(unsigned int amount);
    unsigned int lenghtOfTranin() const;
};


#endif //NALOGA0602_VLAK_H