/* * ScreenPressor decoder * * Copyright (c) 2017 Paul B Mahol * * This file is part of FFmpeg. * * FFmpeg is free software; you can redistribute it and/or * modify it under the terms of the GNU Lesser General Public * License as published by the Free Software Foundation; either * version 2.1 of the License, or (at your option) any later version. * * FFmpeg is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU * Lesser General Public License for more details. * * You should have received a copy of the GNU Lesser General Public * License along with FFmpeg; if not, write to the Free Software * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA */ #ifndef AVCODEC_SCPR_H #define AVCODEC_SCPR_H #include #include #include #include "avcodec.h" #include "bytestream.h" #include "internal.h" #include "scpr3.h" typedef struct RangeCoder { uint32_t code; uint32_t range; uint32_t code1; } RangeCoder; typedef struct PixelModel { uint32_t freq[256]; uint32_t lookup[16]; uint32_t total_freq; } PixelModel; typedef struct SCPRContext { int version; AVFrame *last_frame; AVFrame *current_frame; GetByteContext gb; RangeCoder rc; PixelModel pixel_model[3][4096]; uint32_t op_model[6][7]; uint32_t run_model[6][257]; uint32_t range_model[257]; uint32_t count_model[257]; uint32_t fill_model[6]; uint32_t sxy_model[4][17]; uint32_t mv_model[2][513]; uint32_t nbx, nby; uint32_t nbcount; uint32_t *blocks; uint32_t cbits; int cxshift; PixelModel3 pixel_model3[3][4096]; RunModel3 run_model3[6]; RunModel3 range_model3; RunModel3 count_model3; FillModel3 fill_model3; SxyModel3 sxy_model3[4]; MVModel3 mv_model3[2]; OpModel3 op_model3[6]; int (*get_freq)(RangeCoder *rc, uint32_t total_freq, uint32_t *freq); int (*decode)(GetByteContext *gb, RangeCoder *rc, uint32_t cumFreq, uint32_t freq, uint32_t total_freq); } SCPRContext; static int decode_run_i(AVCodecContext *avctx, uint32_t ptype, int run, int *x, int *y, uint32_t clr, uint32_t *dst, int linesize, uint32_t *lx, uint32_t *ly, uint32_t backstep, int off, int *cx, int *cx1) { uint32_t r, g, b; int z; switch (ptype) { case 0: while (run-- > 0) { if (*y >= avctx->height) return AVERROR_INVALIDDATA; dst[*y * linesize + *x] = clr; *lx = *x; *ly = *y; (*x)++; if (*x >= avctx->width) { *x = 0; (*y)++; } } break; case 1: while (run-- > 0) { if (*y >= avctx->height) return AVERROR_INVALIDDATA; dst[*y * linesize + *x] = dst[*ly * linesize + *lx]; *lx = *x; *ly = *y; (*x)++; if (*x >= avctx->width) { *x = 0; (*y)++; } } clr = dst[*ly * linesize + *lx]; break; case 2: while (run-- > 0) { if (*y < 1 || *y >= avctx->height) return AVERROR_INVALIDDATA; clr = dst[*y * linesize + *x + off + 1]; dst[*y * linesize + *x] = clr; *lx = *x; *ly = *y; (*x)++; if (*x >= avctx->width) { *x = 0; (*y)++; } } break; case 4: while (run-- > 0) { uint8_t *odst = (uint8_t *)dst; if (*y < 1 || *y >= avctx->height || (*y == 1 && *x == 0)) return AVERROR_INVALIDDATA; if (*x == 0) { z = backstep; } else { z = 0; } r = odst[(*ly * linesize + *lx) * 4] + odst[((*y * linesize + *x) + off) * 4 + 4] - odst[((*y * linesize + *x) + off - z) * 4]; g = odst[(*ly * linesize + *lx) * 4 + 1] + odst[((*y * linesize + *x) + off) * 4 + 5] - odst[((*y * linesize + *x) + off - z) * 4 + 1]; b = odst[(*ly * linesize + *lx) * 4 + 2] + odst[((*y * linesize + *x) + off) * 4 + 6] - odst[((*y * linesize + *x) + off - z) * 4 + 2]; clr = ((b & 0xFF) << 16) + ((g & 0xFF) << 8) + (r & 0xFF); dst[*y * linesize + *x] = clr; *lx = *x; *ly = *y; (*x)++; if (*x >= avctx->width) { *x = 0; (*y)++; } } break; case 5: while (run-- > 0) { if (*y < 1 || *y >= avctx->height || (*y == 1 && *x == 0)) return AVERROR_INVALIDDATA; if (*x == 0) { z = backstep; } else { z = 0; } clr = dst[*y * linesize + *x + off - z]; dst[*y * linesize + *x] = clr; *lx = *x; *ly = *y; (*x)++; if (*x >= avctx->width) { *x = 0; (*y)++; } } break; } if (avctx->bits_per_coded_sample == 16) { *cx1 = (clr & 0x3F00) >> 2; *cx = (clr & 0x3FFFFF) >> 16; } else { *cx1 = (clr & 0xFC00) >> 4; *cx = (clr & 0xFFFFFF) >> 18; } return 0; } static int decode_run_p(AVCodecContext *avctx, uint32_t ptype, int run, int x, int y, uint32_t clr, uint32_t *dst, uint32_t *prev, int linesize, int plinesize, uint32_t *bx, uint32_t *by, uint32_t backstep, int sx1, int sx2, int *cx, int *cx1) { uint32_t r, g, b; int z; switch (ptype) { case 0: while (run-- > 0) { if (*by >= avctx->height) return AVERROR_INVALIDDATA; dst[*by * linesize + *bx] = clr; (*bx)++; if (*bx >= x * 16 + sx2 || *bx >= avctx->width) { *bx = x * 16 + sx1; (*by)++; } } break; case 1: while (run-- > 0) { if (*bx == 0) { if (*by < 1) return AVERROR_INVALIDDATA; z = backstep; } else { z = 0; } if (*by >= avctx->height) return AVERROR_INVALIDDATA; clr = dst[*by * linesize + *bx - 1 - z]; dst[*by * linesize + *bx] = clr; (*bx)++; if (*bx >= x * 16 + sx2 || *bx >= avctx->width) { *bx = x * 16 + sx1; (*by)++; } } break; case 2: while (run-- > 0) { if (*by < 1 || *by >= avctx->height) return AVERROR_INVALIDDATA; clr = dst[(*by - 1) * linesize + *bx]; dst[*by * linesize + *bx] = clr; (*bx)++; if (*bx >= x * 16 + sx2 || *bx >= avctx->width) { *bx = x * 16 + sx1; (*by)++; } } break; case 3: while (run-- > 0) { if (*by >= avctx->height) return AVERROR_INVALIDDATA; clr = prev[*by * plinesize + *bx]; dst[*by * linesize + *bx] = clr; (*bx)++; if (*bx >= x * 16 + sx2 || *bx >= avctx->width) { *bx = x * 16 + sx1; (*by)++; } } break; case 4: while (run-- > 0) { uint8_t *odst = (uint8_t *)dst; if (*by < 1 || *by >= avctx->height) return AVERROR_INVALIDDATA; if (*bx == 0) { if (*by < 2) return AVERROR_INVALIDDATA; z = backstep; } else { z = 0; } r = odst[((*by - 1) * linesize + *bx) * 4] + odst[(*by * linesize + *bx - 1 - z) * 4] - odst[((*by - 1) * linesize + *bx - 1 - z) * 4]; g = odst[((*by - 1) * linesize + *bx) * 4 + 1] + odst[(*by * linesize + *bx - 1 - z) * 4 + 1] - odst[((*by - 1) * linesize + *bx - 1 - z) * 4 + 1]; b = odst[((*by - 1) * linesize + *bx) * 4 + 2] + odst[(*by * linesize + *bx - 1 - z) * 4 + 2] - odst[((*by - 1) * linesize + *bx - 1 - z) * 4 + 2]; clr = ((b & 0xFF) << 16) + ((g & 0xFF) << 8) + (r & 0xFF); dst[*by * linesize + *bx] = clr; (*bx)++; if (*bx >= x * 16 + sx2 || *bx >= avctx->width) { *bx = x * 16 + sx1; (*by)++; } } break; case 5: while (run-- > 0) { if (*by < 1 || *by >= avctx->height) return AVERROR_INVALIDDATA; if (*bx == 0) { if (*by < 2) return AVERROR_INVALIDDATA; z = backstep; } else { z = 0; } clr = dst[(*by - 1) * linesize + *bx - 1 - z]; dst[*by * linesize + *bx] = clr; (*bx)++; if (*bx >= x * 16 + sx2 || *bx >= avctx->width) { *bx = x * 16 + sx1; (*by)++; } } break; } if (avctx->bits_per_coded_sample == 16) { *cx1 = (clr & 0x3F00) >> 2; *cx = (clr & 0x3FFFFF) >> 16; } else { *cx1 = (clr & 0xFC00) >> 4; *cx = (clr & 0xFFFFFF) >> 18; } return 0; } #endif /* AVCODEC_SCPR_H */