IFightsEstimator.h

#pragma once

#include <fsdk/Def.h>
#include <fsdk/FSDKError.h>
#include <fsdk/IRefCounted.h>
#include <fsdk/Types/Image.h>
#include <fsdk/Types/Ref.h>
#include <fsdk/Types/Result.h>
#include <fsdk/Types/ResultValue.h>
#include <fsdk/Types/Span.h>

namespace fsdk {

#ifndef DOXYGEN_SHOULD_SKIP_THIS
    DECLARE_SMARTPTR(IFightsEstimator);
    DECLARE_SMARTPTR(IFightsEstimatorContext);
#endif

    struct FightsEstimation {
        enum class State {
            NoFight, 
            Fight    
        };
        State state; 
        float score; 
    };

    struct IFightsEstimatorContext : IRefCounted {
        enum class State {
            NotReady,
            Ready
        };
        virtual State getState() const noexcept = 0;

        virtual FightsEstimation getResult() const noexcept = 0;
    };

    struct IFightsEstimator : IRefCounted {

        virtual size_t getBatchSize() const noexcept = 0;

        virtual size_t getMinBatchCount() const noexcept = 0;

        virtual ResultValue<FSDKError, IFightsEstimatorContextPtr>
        estimate(Span<const Image> framesBatch, IFightsEstimatorContextPtr context) const noexcept = 0;

        virtual Result<FSDKError>
        validate(Span<const Image> framesBatch, Span<Result<FSDKError>> errors) const noexcept = 0;
    };

} // namespace fsdk