RawTracklets.hpp

Go to the documentation of this file.

#pragma once
 
#include <iostream>
 
#include "depthai-shared/common/Point3f.hpp"
#include "depthai-shared/common/Rect.hpp"
#include "depthai-shared/datatype/RawBuffer.hpp"
#include "depthai-shared/datatype/RawImgDetections.hpp"
#include "depthai-shared/utility/Serialization.hpp"
 
namespace dai {
 
struct Tracklet {
    enum class TrackingStatus : std::int32_t {
        NEW,     
        TRACKED, 
        LOST,   
        REMOVED 
    };
    Rect roi;
    std::int32_t id = 0;
    std::int32_t label = 0;
    std::int32_t age = 0;
    TrackingStatus status = TrackingStatus::LOST;
 
    ImgDetection srcImgDetection;
    Point3f spatialCoordinates;
    DEPTHAI_SERIALIZE(Tracklet, roi, id, label, age, status, srcImgDetection, spatialCoordinates);
};
 
struct RawTracklets : public RawBuffer {
    std::vector<Tracklet> tracklets;
 
    void serialize(std::vector<std::uint8_t>& metadata, DatatypeEnum& datatype) const override {
        metadata = utility::serialize(*this);
        datatype = DatatypeEnum::Tracklets;
    };
 
    DatatypeEnum getType() const override {
        return DatatypeEnum::Tracklets;
    }
 
    DEPTHAI_SERIALIZE(RawTracklets, tracklets, RawBuffer::sequenceNum, RawBuffer::ts, RawBuffer::tsDevice);
};
 
}  // namespace dai