debayer_cpu.h

/* SPDX-License-Identifier: LGPL-2.1-or-later */
/*
 * Copyright (C) 2023, Linaro Ltd
 * Copyright (C) 2023-2026 Red Hat Inc.
 *
 * Authors:
 * Hans de Goede <hdegoede@redhat.com>
 *
 * CPU based debayering header
 */
 
#pragma once
 
#include <memory>
#include <stdint.h>
#include <vector>
 
#include <libcamera/base/mutex.h>
#include <libcamera/base/object.h>
 
#include <libcamera/camera_manager.h>
 
#include "libcamera/internal/bayer_format.h"
#include "libcamera/internal/software_isp/debayer_params.h"
#include "libcamera/internal/software_isp/swstats_cpu.h"
 
#include "debayer.h"
 
namespace libcamera {
 
class DebayerCpuThread;
class DebayerCpu : public Debayer
{
public:
    DebayerCpu(std::unique_ptr<SwStatsCpu> stats, const CameraManager &cm);
    ~DebayerCpu();
 
    int configure(const StreamConfiguration &inputCfg,
              const std::vector<std::reference_wrapper<const StreamConfiguration>> &outputCfgs,
              bool ccmEnabled) override;
    Size patternSize(PixelFormat inputFormat) override;
    std::vector<PixelFormat> formats(PixelFormat input) override;
    std::tuple<unsigned int, unsigned int>
    strideAndFrameSize(const PixelFormat &outputFormat, const Size &size) override;
    void process(uint32_t frame, FrameBuffer *input, FrameBuffer *output, const DebayerParams &params) override;
    int start() override;
    void stop() override;
    SizeRange sizes(PixelFormat inputFormat, const Size &inputSize) override;
    const SharedFD &getStatsFD() override { return stats_->getStatsFD(); }
 
private:
    friend class DebayerCpuThread;
 
    using debayerFn = void (DebayerCpu::*)(uint8_t *dst, const uint8_t *src[]);
 
    void debayer0(uint8_t *dst, const uint8_t *src[]) { (this->*debayer0_)(dst, src); }
    void debayer1(uint8_t *dst, const uint8_t *src[]) { (this->*debayer1_)(dst, src); }
    void debayer2(uint8_t *dst, const uint8_t *src[]) { (this->*debayer2_)(dst, src); }
    void debayer3(uint8_t *dst, const uint8_t *src[]) { (this->*debayer3_)(dst, src); }
 
    /* 8-bit raw bayer format */
    template<bool addAlphaByte, bool ccmEnabled>
    void debayer8_BGBG_BGR888(uint8_t *dst, const uint8_t *src[]);
    template<bool addAlphaByte, bool ccmEnabled>
    void debayer8_GRGR_BGR888(uint8_t *dst, const uint8_t *src[]);
    /* unpacked 10-bit raw bayer format */
    template<bool addAlphaByte, bool ccmEnabled>
    void debayer10_BGBG_BGR888(uint8_t *dst, const uint8_t *src[]);
    template<bool addAlphaByte, bool ccmEnabled>
    void debayer10_GRGR_BGR888(uint8_t *dst, const uint8_t *src[]);
    /* unpacked 12-bit raw bayer format */
    template<bool addAlphaByte, bool ccmEnabled>
    void debayer12_BGBG_BGR888(uint8_t *dst, const uint8_t *src[]);
    template<bool addAlphaByte, bool ccmEnabled>
    void debayer12_GRGR_BGR888(uint8_t *dst, const uint8_t *src[]);
    /* CSI-2 packed 10-bit raw bayer format (all the 4 orders) */
    template<bool addAlphaByte, bool ccmEnabled>
    void debayer10P_BGBG_BGR888(uint8_t *dst, const uint8_t *src[]);
    template<bool addAlphaByte, bool ccmEnabled>
    void debayer10P_GRGR_BGR888(uint8_t *dst, const uint8_t *src[]);
    template<bool addAlphaByte, bool ccmEnabled>
    void debayer10P_GBGB_BGR888(uint8_t *dst, const uint8_t *src[]);
    template<bool addAlphaByte, bool ccmEnabled>
    void debayer10P_RGRG_BGR888(uint8_t *dst, const uint8_t *src[]);
    /* CSI-2 packed 12-bit raw bayer format (all the 4 orders) */
    template<bool addAlphaByte, bool ccmEnabled>
    void debayer12P_BGBG_BGR888(uint8_t *dst, const uint8_t *src[]);
    template<bool addAlphaByte, bool ccmEnabled>
    void debayer12P_GRGR_BGR888(uint8_t *dst, const uint8_t *src[]);
    template<bool addAlphaByte, bool ccmEnabled>
    void debayer12P_GBGB_BGR888(uint8_t *dst, const uint8_t *src[]);
    template<bool addAlphaByte, bool ccmEnabled>
    void debayer12P_RGRG_BGR888(uint8_t *dst, const uint8_t *src[]);
 
    static int getInputConfig(PixelFormat inputFormat, DebayerInputConfig &config);
    static int getOutputConfig(PixelFormat outputFormat, DebayerOutputConfig &config);
    int setupStandardBayerOrder(BayerFormat::Order order);
    int setDebayerFunctions(PixelFormat inputFormat,
                PixelFormat outputFormat,
                bool ccmEnabled);
    void updateGammaTable(const DebayerParams &params);
    void updateLookupTables(const DebayerParams &params);
 
    static constexpr unsigned int kRGBLookupSize = 256;
    static constexpr unsigned int kGammaLookupSize = 1024;
    struct CcmColumn {
        int16_t r;
        int16_t g;
        int16_t b;
    };
    using LookupTable = std::array<uint8_t, kRGBLookupSize>;
    using CcmLookupTable = std::array<CcmColumn, kRGBLookupSize>;
    LookupTable red_;
    LookupTable green_;
    LookupTable blue_;
    CcmLookupTable redCcm_;
    CcmLookupTable greenCcm_;
    CcmLookupTable blueCcm_;
    std::array<double, kGammaLookupSize> gammaTable_;
    LookupTable gammaLut_;
    bool ccmEnabled_;
    DebayerParams params_;
 
    debayerFn debayer0_;
    debayerFn debayer1_;
    debayerFn debayer2_;
    debayerFn debayer3_;
    Rectangle window_;
    std::unique_ptr<SwStatsCpu> stats_;
    unsigned int xShift_; /* Offset of 0/1 applied to window_.x */
 
    static constexpr unsigned int kMinThreads = 1;
    static constexpr unsigned int kMaxThreads = 8;
    static constexpr unsigned int kDefaultThreads = 2;
 
    unsigned int workPending_ LIBCAMERA_TSA_GUARDED_BY(workPendingMutex_);
    Mutex workPendingMutex_;
    ConditionVariable workPendingCv_;
    std::vector<std::unique_ptr<DebayerCpuThread>> threads_;
};
 
} /* namespace libcamera */