include/fsdk/Types/Detection.h

#pragma once

#include <fsdk/Types/Rect.h>

namespace fsdk {
    struct Detection {
        Detection() noexcept
            : m_rect{}
            , m_w{0}
            , m_h{0}
            , m_score{0.0f} {
        }

        Detection(FloatRect rect, float score = 1.f) noexcept
            : m_rect{rect}
            , m_w{static_cast<int>(rect.x + rect.width)}
            , m_h{static_cast<int>(rect.y + rect.height)}
            , m_score{score} {
        }

        Detection(FloatRect rect, int width, int height, float score = 1.f) noexcept
            : m_rect{rect}
            , m_w{width}
            , m_h{height}
            , m_score{score} {
        }

        Detection(FloatRect rect, Rect imageRect, float score = 1.f) noexcept
            : m_rect{rect}
            , m_w{imageRect.x + imageRect.width}
            , m_h{imageRect.y + imageRect.height}
            , m_score{score} {
        }

        Detection(const Detection& rhs) noexcept
            : m_rect{rhs.m_rect}
            , m_w{rhs.m_w}
            , m_h{rhs.m_h}
            , m_score{rhs.m_score} {
        }

        Detection& operator=(const Detection& rhs) noexcept {
            m_rect = rhs.m_rect;
            m_w = rhs.m_w;
            m_h = rhs.m_h;
            m_score = rhs.m_score;
            return *this;
        }

        Rect getRect() const noexcept {
            return m_rect & fsdk::Rect{0, 0, m_w, m_h};
        }

        FloatRect getRawRect() const noexcept {
            return m_rect;
        }

        void setRawRect(fsdk::FloatRect rect) noexcept {
            m_rect = rect;
        }

        float getScore() const noexcept {
            return m_score;
        }

        void setScore(float score) noexcept {
            m_score = score;
        }

        bool isValid() const noexcept {
            return m_rect.isValid() && m_score && m_w && m_h;
        }

    private:
        FloatRect m_rect; // Detection rect
        int m_w, m_h;     // Bounding width and height of the source image
        float m_score;    // Detection score
    };
} // namespace fsdk