collisiondisplay2d.h

#ifndef COLLISIONDISPLAY2D_H
#define COLLISIONDISPLAY2D_H

#include <QGraphicsView>
#include <QGraphicsScene>
#include <QTimer>
#include <unordered_map>
#include "collisiondisplay.h"
#include "qalgorithm.h"
#include "qparticle2d.h"

class CollisionDisplay2D : public QGraphicsView, public CollisionDisplay
{
    Q_OBJECT
public:

    explicit CollisionDisplay2D(QWidget *parent = 0);
    ~CollisionDisplay2D();

    QGraphicsScene* getScene();

    QTimer* getTimer();

    void setQAlgorithm(QAlgorithm *qalgorithm);

//    int getNumCollisions();
//    int getDisplayTimestep();

signals:
    void finished();

public slots:

    void start();

    void stop();

    void step();

private:
    QGraphicsScene* scene;

    std::vector<QParticle2D*> drawnParticles;
    std::unordered_map<Particle*, QParticle2D*> particleMap;
};

#endif // COLLISIONDISPLAY2D_H