peak_icv_mono_conversion_module.hpp

Go to the documentation of this file.

 
#pragma once
 
#include <peak_common/pipeline/modules/peak_common_imodule.hpp>
#include <peak_icv/pipeline/detail/peak_icv_pipeline_utils.hpp>
#include <peak_icv/types/peak_icv_image.hpp>
#include <peak_icv_c/algorithms/preprocessing/peak_icv_image_converter.h>
 
namespace peak
{
namespace pipeline
{
namespace detail
{
 
class MonoConversionModule : public modules::IModule
{
public:
    MonoConversionModule();
 
    ~MonoConversionModule() override;
 
    MonoConversionModule(const MonoConversionModule& other) = delete;
 
    MonoConversionModule(MonoConversionModule&& other) noexcept;
 
    MonoConversionModule& operator=(const MonoConversionModule& other) = delete;
 
    MonoConversionModule& operator=(MonoConversionModule&& other) noexcept;
 
    void Serialize(peak::common::serialization::IArchive& archive) const override;
 
 
    void Deserialize(const peak::common::serialization::IArchive& archive) override;
 
    void SetEnabled(bool enabled) override;
 
    PEAK_COMMON_NO_DISCARD bool IsEnabled() const override;
 
    PEAK_COMMON_NO_DISCARD const char* GetType() const override;
 
    void ResetToDefault() override
    {}
 
    PEAK_COMMON_NO_DISCARD peak::icv::Image Process(const peak::icv::Image& input) const;
 
    PEAK_COMMON_NO_DISCARD peak::common::Any Process(const peak::common::Any& input) const override;
 
    void ReleaseBuffers() const;
 
private:
    PEAK_COMMON_NO_DISCARD static peak::common::PixelFormat GetOutputFormat(peak::common::PixelFormat inputFormat);
 
    constexpr static int moduleVersion{ 1 };
 
    bool m_enabled{ true };
    peak_icv_image_converter_handle m_handle{};
};
 
inline MonoConversionModule::MonoConversionModule()
{
    peak::icv::detail::ExecuteAndMapReturnCodes([&] {
        return PEAK_ICV_C_ABI_PREFIX peak_icv_Preprocessing_ImageConverter_Create(&m_handle);
    });
}
 
inline MonoConversionModule::~MonoConversionModule()
{
    if (m_handle != nullptr)
    {
        std::ignore = PEAK_ICV_C_ABI_PREFIX peak_icv_Preprocessing_ImageConverter_Destroy(m_handle);
    }
}
 
inline MonoConversionModule::MonoConversionModule(MonoConversionModule&& other) noexcept
{
    m_enabled = other.m_enabled;
    std::swap(m_handle, other.m_handle);
}
 
inline MonoConversionModule& MonoConversionModule::operator=(MonoConversionModule&& other) noexcept
{
    if (this != &other)
    {
        m_enabled = other.m_enabled;
        std::swap(m_handle, other.m_handle);
    }
 
    return *this;
}
 
inline void MonoConversionModule::Serialize(peak::common::serialization::IArchive& archive) const
{
    archive.SetBool("Enabled", IsEnabled());
    archive.SetInt("Version", moduleVersion);
}
 
inline void MonoConversionModule::Deserialize(const peak::common::serialization::IArchive& archive)
{
    const auto version = archive.GetInt("Version");
    detail::ValidateVersion(version, GetType());
 
    SetEnabled(archive.GetBool("Enabled"));
}
 
inline void MonoConversionModule::SetEnabled(bool enabled)
{
    m_enabled = enabled;
}
 
inline bool MonoConversionModule::IsEnabled() const
{
    return m_enabled;
}
 
inline const char* MonoConversionModule::GetType() const
{
    return "MonoConversion";
}
 
inline peak::common::PixelFormat MonoConversionModule::GetOutputFormat(peak::common::PixelFormat inputFormat)
{
    const peak::common::PixelFormatInfo info(inputFormat);
 
    if (info.IsSingleChannel() && info.HasIntensityChannel())
    {
        return inputFormat;
    }
 
    const auto bitDepth = info.GetStorageBitsPerChannel();
 
    if (bitDepth <= 8)
    {
        return peak::common::PixelFormat::Mono8;
    }
    if (bitDepth <= 10)
    {
        return peak::common::PixelFormat::Mono10;
    }
    if (bitDepth <= 12)
    {
        return peak::common::PixelFormat::Mono12;
    }
    return peak::common::PixelFormat::Mono16;
}
 
inline peak::icv::Image MonoConversionModule::Process(const peak::icv::Image& input) const
{
    return Process(peak::common::Any(input)).AnyCast<peak::icv::Image>();
}
 
inline peak::common::Any MonoConversionModule::Process(const peak::common::Any& input) const
{
    if (!IsEnabled())
    {
        return input;
    }
 
    const auto& inputImage = input.AnyCast<peak::icv::Image>();
 
    const auto inputFormat = inputImage.GetPixelFormat();
    const auto outputFormat = GetOutputFormat(inputFormat);
 
    if (inputFormat == outputFormat)
    {
        return input;
    }
 
    auto* const inputImageHandle = peak::common::detail::BackendAccessor<peak::icv::Image>::BackendHandle(inputImage);
    peak_icv_image_handle outputHandle = nullptr;
 
    peak::icv::detail::ExecuteAndMapReturnCodes([&] {
        return PEAK_ICV_C_ABI_PREFIX peak_icv_Preprocessing_ImageConverter_Convert(
            m_handle, inputImageHandle, peak::common::detail::CppPixelFormatToCPixelFormat(outputFormat), &outputHandle);
    });
 
    return peak::common::Any(peak::common::detail::BackendAccessor<peak::icv::Image>::CreateInstance(outputHandle));
}
 
inline void MonoConversionModule::ReleaseBuffers() const
{
    peak::icv::detail::ExecuteAndMapReturnCodes([&] {
        return PEAK_ICV_C_ABI_PREFIX peak_icv_Preprocessing_ImageConverter_ReleaseBuffers(m_handle);
    });
}
 
} // namespace detail
} // namespace pipeline 
} // namespace peak