proresdsp: add idct_put_bayer

This commit adds a 12-bit DCT function to directly write to Bayer images. Will be used in the following commit.

Lynne committed 1 year ago

Commit 56748791 parent b2928972 files · +18 −0Browse files

Unified · Split

2 files

     }
 }
+static inline void put_pixel_bayer_12(uint16_t *dst, ptrdiff_t linesize,
+                                      const int16_t *in)
+{
+    for (int y = 0; y < 8; y++, dst += linesize) {
+        for (int x = 0; x < 8; x++)
+            dst[x*2] = CLIP_12(in[(y << 3) + x]) << 4;
+    }
+}
 static void put_pixels_10(uint16_t *dst, ptrdiff_t linesize, const int16_t *in)
 {
     put_pixel(dst, linesize, in, 10);
     put_pixels_12(out, linesize >> 1, block);
 }
+static void prores_idct_put_bayer_12_c(uint16_t *out, ptrdiff_t linesize,
+                                       int16_t *block, const int16_t *qmat)
+{
+    prores_idct_12(block, qmat);
+    put_pixel_bayer_12(out, linesize << 1, block);
+}
 av_cold void ff_proresdsp_init(ProresDSPContext *dsp, int bits_per_raw_sample)
 {
     if (bits_per_raw_sample == 10) {
     } else {
         av_assert1(bits_per_raw_sample == 12);
         dsp->idct_put = prores_idct_put_12_c;
+        dsp->idct_put_bayer = prores_idct_put_bayer_12_c;
         dsp->idct_permutation_type = FF_IDCT_PERM_NONE;
     }

@@ -30,6 +30,7 @@ typedef struct ProresDSPContext { · 29 unmodified lines
30	int idct_permutation_type;	30	int idct_permutation_type;
31	uint8_t idct_permutation[64];	31	uint8_t idct_permutation[64];
32	void (idct_put)(uint16_t out, ptrdiff_t linesize, int16_t block, const int16_t qmat);	32	void (idct_put)(uint16_t out, ptrdiff_t linesize, int16_t block, const int16_t qmat);
		33	void (idct_put_bayer)(uint16_t out, ptrdiff_t linesize, int16_t block, const int16_t qmat);
33	} ProresDSPContext;	34	} ProresDSPContext;
34		35
35	void ff_proresdsp_init(ProresDSPContext *dsp, int bits_per_raw_sample);	36	void ff_proresdsp_init(ProresDSPContext *dsp, int bits_per_raw_sample);

Keyboard shortcuts