udpthread.h

/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at https://mozilla.org/MPL/2.0/. */
 
#pragma once
 
// External includes
#include <nap/resourceptr.h>
#include <nap/resource.h>
#include <nap/device.h>
#include <thread>
 
// NAP includes
#include <nap/numeric.h>
#include <concurrentqueue.h>
#include <rtti/factory.h>
 
// ASIO forward declaration
namespace asio
{
    class io_context;
}
 
namespace nap
{
 
    enum EUDPThreadUpdateMethod : int
    {
        UDP_MAIN_THREAD         = 0,            
        UDP_SPAWN_OWN_THREAD    = 1,            
        UDP_MANUAL              = 2             
    };
 
    // forward declares
    class UDPAdapter;
    class UDPService;
 
    class NAPAPI UDPThread : public Device
    {
        friend class UDPService;
        friend class UDPAdapter;
        RTTI_ENABLE(Device)
    public:
        UDPThread();
 
        UDPThread(UDPService& service);
 
        virtual ~UDPThread();
 
        virtual bool start(utility::ErrorState& errorState) override final;
 
        virtual void stop() override final;
 
        asio::io_context& getIOContext();
 
        void manualProcess();
 
        // properties
        EUDPThreadUpdateMethod mUpdateMethod = EUDPThreadUpdateMethod::UDP_MAIN_THREAD; 
    private:
        void thread();
 
        void process();
 
        void registerAdapter(UDPAdapter* adapter);
 
        void removeAdapter(UDPAdapter* adapter);
 
        // threading
        std::thread                                         mThread;
        std::mutex                                          mMutex;
        std::atomic_bool                                    mRun = { false };
        std::function<void()>                               mManualProcessFunc;
 
        // service
        UDPService& mService;
 
        // adapters
        std::vector<UDPAdapter*>    mAdapters;
 
        struct Impl;
        std::unique_ptr<Impl> mImpl;
    };
 
    // Object creator used for constructing the UDP thread
    using UDPThreadObjectCreator = rtti::ObjectCreator<UDPThread, UDPService>;
}