TopicPublisher.h 1.47 KB
Newer Older
Valentin Platzgummer's avatar
Valentin Platzgummer committed
1 2 3 4 5 6 7 8 9 10 11
#pragma once

#include "ros_bridge/include/TypeFactory.h"
#include "ros_bridge/include/CasePacker.h"
#include "ros_bridge/include/ComPrivateInclude.h"
#include "ros_bridge/include/RosBridgeClient.h"

#include <thread>
#include <deque>
#include <atomic>
#include <mutex>
Valentin Platzgummer's avatar
Valentin Platzgummer committed
12
#include <set>
13
#include <condition_variable>
Valentin Platzgummer's avatar
Valentin Platzgummer committed
14 15 16 17

namespace ROSBridge {
namespace ComPrivate {

18 19
struct ThreadData;

Valentin Platzgummer's avatar
Valentin Platzgummer committed
20
class TopicPublisher
Valentin Platzgummer's avatar
Valentin Platzgummer committed
21 22
{
    typedef std::unique_ptr<std::thread> ThreadPtr;
23
    using CondVar = std::condition_variable;
Valentin Platzgummer's avatar
Valentin Platzgummer committed
24 25
public:

Valentin Platzgummer's avatar
Valentin Platzgummer committed
26
    TopicPublisher() = delete;
27 28
    TopicPublisher(CasePacker &casePacker,
                   RosbridgeWsClient &rbc);
Valentin Platzgummer's avatar
Valentin Platzgummer committed
29 30

    ~TopicPublisher();
Valentin Platzgummer's avatar
Valentin Platzgummer committed
31

32
    //! @brief Starts the publisher.
Valentin Platzgummer's avatar
Valentin Platzgummer committed
33
    void start();
34 35 36 37

    //! @brief Resets the publisher.
    void reset();

Valentin Platzgummer's avatar
Valentin Platzgummer committed
38
    void publish(JsonDocUPtr docPtr){
39
        {
Valentin Platzgummer's avatar
Valentin Platzgummer committed
40 41
        std::lock_guard<std::mutex> lock(_queueMutex);
        _queue.push_back(std::move(docPtr));
42 43
        }
        _cv.notify_one();  // Wake publisher thread.
Valentin Platzgummer's avatar
Valentin Platzgummer committed
44 45
    }

46 47
    template<class T>
    void publish(const T &msg, const std::string &topic){
48
        JsonDocUPtr docPtr(_casePacker.pack(msg, topic));
49 50 51
        publish(std::move(docPtr));
    }

Valentin Platzgummer's avatar
Valentin Platzgummer committed
52 53 54
private:
    JsonQueue           _queue;
    std::mutex          _queueMutex;
55
    std::atomic<bool>   _running;
56 57
    CasePacker         &_casePacker;
    RosbridgeWsClient  &_rbc;
58 59
    CondVar             _cv;
    ThreadPtr           _pThread;
Valentin Platzgummer's avatar
Valentin Platzgummer committed
60 61 62
};


63
void transmittLoop(ThreadData data);
Valentin Platzgummer's avatar
Valentin Platzgummer committed
64 65 66 67


} // namespace CommunicatorDetail
} // namespace ROSBridge