RasterHelperFunctions.h

#pragma once
#include "RasterFrame.h"
#include "Base/Headers/RGBColor.h"
namespace NDEVR
{
    # define Y_OFFSET   16
    # define UV_OFFSET 128
    # define YUV2RGB_11  298
    # define YUV2RGB_12   -1
    # define YUV2RGB_13  409
    # define YUV2RGB_22 -100
    # define YUV2RGB_23 -210
    # define YUV2RGB_32  519
    # define YUV2RGB_33    0
    static inline RGBColor PullAsYUV(const VideoFrameDetails&, int Text_Index, const uint01* memory)
    {
        int value = ((Text_Index - (Text_Index % 2)) * 2);
        int y, u, v;
        int uv_r, uv_g, uv_b;
 
        if (Text_Index % 2 == 0)
        {
            y = YUV2RGB_11 * (memory[value] - Y_OFFSET);
        }
        else
        {
            y = YUV2RGB_11 * (memory[value + 2] - Y_OFFSET);
        }
        u = memory[value + 1] - UV_OFFSET;
        v = memory[value + 3] - UV_OFFSET;
 
        uv_r = YUV2RGB_12 * u + YUV2RGB_13 * v;
        uv_g = YUV2RGB_22 * u + YUV2RGB_23 * v;
        uv_b = YUV2RGB_32 * u + YUV2RGB_33 * v;
 
        return RGBColor(
            clip((y + uv_r) >> 8, 0, 255)
            , clip((y + uv_g) >> 8, 0, 255)
            , clip((y + uv_b) >> 8, 0, 255));
 
 
    }

    static inline RGBColor PullAsRGB(const VideoFrameDetails& texture, int Text_Index, const uint01* memory)
    {
        int value = Text_Index * texture.bytes_per_pixel;
        return RGBColor(memory[value], memory[value + 1], memory[value + 2], 255U);
    }

    static inline RGBColor PullAsBGR(const VideoFrameDetails& texture, int Text_Index, const uint01* memory)
    {
        int value = Text_Index * texture.bytes_per_pixel;
        return RGBColor(memory[value + 2], memory[value + 1], memory[value + 0], 255U);
    }

    static inline RGBColor PullAsY8(const VideoFrameDetails& texture, int Text_Index, const uint01* memory)
    {
        int value = Text_Index * texture.bytes_per_pixel;
        return RGBColor(memory[value], memory[value], memory[value], 255U);
    }

    static inline RGBColor PullRGB(const VideoFrameDetails& texture, int index, const uint01* memory)
    {
        switch (texture.format)
        {
        case RasterColorFormat::e_bgr: return PullAsBGR(texture, index, memory);
        case RasterColorFormat::e_rgb: return PullAsRGB(texture, index, memory);
        case RasterColorFormat::e_y8:  return PullAsY8(texture, index, memory);
        case RasterColorFormat::e_yuv: return PullAsYUV(texture, index, memory);
        default: lib_assert(false, "unknown format"); return Constant<RGBColor>::Invalid;
        }
    }

    static inline fltp04 PullDepth(const VideoFrameDetails& texture, int index, const uint01* memory)
    {
        uint64_t pixel = 0;
        switch (texture.bytes_per_pixel)  // bits per pixel
        {
        case 1:
            pixel = memory[index];
            break;
        case 2:
            pixel = reinterpret_cast<const uint16_t*>(memory)[index];
            break;
        case 4:
            pixel = reinterpret_cast<const uint32_t*>(memory)[index];
            break;
        case 8:
            pixel = reinterpret_cast<const uint64_t*>(memory)[index];
            break;
        }
        return cast<fltp04>(pixel);
    }
}