kilobotexperiment.h

#ifndef KILOBOTEXPERIMENT_H
#define KILOBOTEXPERIMENT_H
 
#include <QObject>
#include <QDebug>
#include <QLayout>
#include "kilobotenvironment.h"
#include "kilobot.h"
#include <QColor>
#include <opencv2/core/core.hpp>
 
class KilobotExperiment : public QObject
{
    Q_OBJECT
public:
    KilobotExperiment() {}
    virtual ~KilobotExperiment()
    {
    }
 
    int serviceInterval = 100; // ms
 
    virtual QWidget * createGUI() {return NULL;}
 
signals:
    void updateKilobotStates();
    void getInitialKilobotStates();
    void experimentComplete();
    void saveImage(QString);
    void saveVideoFrames(QString,unsigned int);
    void signalKilobot(kilobot_message);
    void broadcastMessage(kilobot_broadcast);
    void setTrackingType(int);
    void sendBroadcastingState(bool);
 
    // drawing
    void drawCircle(QPointF pos, float r, QColor col, int thickness, std::string text, bool transparent=false);
    void drawLine(std::vector<cv::Point> pos, QColor col, int thickness, std::string text, bool transparent);
    void clearDrawings();
    void drawCircleOnRecordedImage(QPointF pos, float r, QColor col, int thickness, std::string text);
    void clearDrawingsOnRecordedImage();
 
 
public slots:
    virtual void initialise(bool) = 0;
    virtual void stopExperiment() {}
    virtual void run() {}
    void setRuntimeIdentificationLock(bool lock) {this->runtimeIdentificationLock = lock;}
    void GetMsgsQueueState(bool state){ ThereIsMsgsToSend=state;}

    void updateStateRequiredCode(Kilobot* kilobot, Kilobot kilobotCopy)
    {
        //qDebug() << "pre set state 2";
        // store pointer for connecting
        this->currKilobot = kilobot;
        updateKilobotState(kilobotCopy);
    }

    void setupInitialStateRequiredCode(Kilobot* kilobot, Kilobot kilobotCopy)
    {
        //qDebug() << "pre set state";
        // store pointer for connecting
        this->currKilobot = kilobot;
        // switch the signal from setup to standard
        kilobot->disconnect(SIGNAL(sendUpdateToExperiment(Kilobot*,Kilobot)));
        connect(kilobot,SIGNAL(sendUpdateToExperiment(Kilobot*,Kilobot)), this, SLOT(updateStateRequiredCode(Kilobot*,Kilobot)));
        setupInitialKilobotState(kilobotCopy);
    }
 
    void signalKilobotExpt(kilobot_message msg)
    {
        emit signalKilobot(msg);
    }
 
protected:
    double time;
    bool runtimeIdentificationLock = false;
    bool ThereIsMsgsToSend=false;
 
    void setCurrentKilobotEnvironment(KilobotEnvironment * environment) {
        if (currKilobot != NULL && environment != NULL) {
            QObject::disconnect(currKilobot,SIGNAL(sendUpdateToHardware(Kilobot)), 0, 0);
            QObject::connect(currKilobot,SIGNAL(sendUpdateToHardware(Kilobot)), environment, SLOT(updateVirtualSensor(Kilobot)));
        }
    }
 
    virtual void updateKilobotState(Kilobot) {} // provided in derived class to implement experiment logic for Kilobot state updates
    virtual void setupInitialKilobotState(Kilobot) {}
 
private:
    Kilobot * currKilobot = NULL;
 
 
};
 
 
 
#endif // KILOBOTEXPERIMENT_H