OpenIGTLinkStreamer.hpp

Go to the documentation of this file.

#pragma once
 
//#include <boost/signals2.hpp>
#include <thread>
#include <unordered_map>
#include "FAST/Streamers/Streamer.hpp"
#include "FAST/ProcessObject.hpp"
#include <set>
#include "FASTExport.hpp"
 
// Forward declare
 
namespace fast {
 
class Image;
class IGTLSocketWrapper;
 
class FAST_EXPORT OpenIGTLinkStreamer : public Streamer {
    FAST_OBJECT(OpenIGTLinkStreamer)
    public:
        std::set<std::string> getImageStreamNames();
        std::set<std::string> getTransformStreamNames();
        std::vector<std::string> getActiveImageStreamNames();
        std::vector<std::string> getActiveTransformStreamNames();
        std::string getStreamDescription(std::string streamName);
        void setConnectionAddress(std::string address);
        void setConnectionPort(uint port);
        uint getNrOfFrames() const;
 
        DataChannel::pointer getOutputPort(uint portID = 0) override;
 
        template<class T>
        DataChannel::pointer getOutputPort(std::string deviceName);
 
        void generateStream() override;
 
        ~OpenIGTLinkStreamer();
        void loadAttributes() override;
    private:
        OpenIGTLinkStreamer();
 
        // Update the streamer if any parameters have changed
        void execute();
 
        uint mNrOfFrames;
        uint mMaximumNrOfFrames;
        bool mMaximumNrOfFramesSet;
 
        bool mInFreezeMode;
 
        std::string mAddress;
        uint mPort;
 
        IGTLSocketWrapper* mSocketWrapper;
        //igtl::ClientSocket::Pointer mSocket;
 
        std::set<std::string> mImageStreamNames;
        std::set<std::string> mTransformStreamNames;
        std::unordered_map<std::string, std::string> mStreamDescriptions;
        std::unordered_map<std::string, uint> mOutputPortDeviceNames;
 
        void updateFirstFrameSetFlag();
};
 
 
 
template<class T>
DataChannel::pointer OpenIGTLinkStreamer::getOutputPort(std::string deviceName) {
    uint portID;
    if(mOutputPortDeviceNames.count(deviceName) == 0) {
        portID = getNrOfOutputPorts();
        createOutputPort<T>(portID);
        mOutputPortDeviceNames[deviceName] = portID;
    } else {
        portID = mOutputPortDeviceNames[deviceName];
    }
    return ProcessObject::getOutputPort(portID);
}
 
 
 
} // end namespace