ref: a32c50862831c9f74ad38acf5764dac30a15a718
dir: /src/ipred_prepare.c/
/* * Copyright © 2018, VideoLAN and dav1d authors * Copyright © 2018, Two Orioles, LLC * All rights reserved. * * Redistribution and use in source and binary forms, with or without * modification, are permitted provided that the following conditions are met: * * 1. Redistributions of source code must retain the above copyright notice, this * list of conditions and the following disclaimer. * * 2. Redistributions in binary form must reproduce the above copyright notice, * this list of conditions and the following disclaimer in the documentation * and/or other materials provided with the distribution. * * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. */ #include "config.h" #include <assert.h> #include <stdint.h> #include <string.h> #include "common/intops.h" #include "src/ipred_prepare.h" static const uint8_t av1_mode_conv[N_INTRA_PRED_MODES] [2 /* have_left */][2 /* have_top */] = { [DC_PRED] = { { DC_128_PRED, TOP_DC_PRED }, { LEFT_DC_PRED, DC_PRED } }, [PAETH_PRED] = { { DC_128_PRED, VERT_PRED }, { HOR_PRED, PAETH_PRED } }, }; static const uint8_t av1_mode_to_angle_map[8] = { 90, 180, 45, 135, 113, 157, 203, 67 }; static const struct { uint8_t needs_left:1; uint8_t needs_top:1; uint8_t needs_topleft:1; uint8_t needs_topright:1; uint8_t needs_bottomleft:1; } av1_intra_prediction_edges[N_IMPL_INTRA_PRED_MODES] = { [DC_PRED] = { .needs_top = 1, .needs_left = 1 }, [VERT_PRED] = { .needs_top = 1 }, [HOR_PRED] = { .needs_left = 1 }, [LEFT_DC_PRED] = { .needs_left = 1 }, [TOP_DC_PRED] = { .needs_top = 1 }, [DC_128_PRED] = { 0 }, [Z1_PRED] = { .needs_top = 1, .needs_topright = 1, .needs_topleft = 1 }, [Z2_PRED] = { .needs_left = 1, .needs_top = 1, .needs_topleft = 1 }, [Z3_PRED] = { .needs_left = 1, .needs_bottomleft = 1, .needs_topleft = 1 }, [SMOOTH_PRED] = { .needs_left = 1, .needs_top = 1 }, [SMOOTH_V_PRED] = { .needs_left = 1, .needs_top = 1 }, [SMOOTH_H_PRED] = { .needs_left = 1, .needs_top = 1 }, [PAETH_PRED] = { .needs_left = 1, .needs_top = 1, .needs_topleft = 1 }, [FILTER_PRED] = { .needs_left = 1, .needs_top = 1, .needs_topleft = 1 }, }; enum IntraPredMode bytefn(dav1d_prepare_intra_edges)(const int x, const int have_left, const int y, const int have_top, const int w, const int h, const enum EdgeFlags edge_flags, const pixel *const dst, const ptrdiff_t stride, const pixel *prefilter_toplevel_sb_edge, enum IntraPredMode mode, int *const angle, const int tw, const int th, pixel *const topleft_out) { assert(y < h && x < w); switch (mode) { case VERT_PRED: case HOR_PRED: case DIAG_DOWN_LEFT_PRED: case DIAG_DOWN_RIGHT_PRED: case VERT_RIGHT_PRED: case HOR_DOWN_PRED: case HOR_UP_PRED: case VERT_LEFT_PRED: { *angle = av1_mode_to_angle_map[mode - VERT_PRED] + 3 * *angle; if (*angle < 90) { mode = have_top ? Z1_PRED : VERT_PRED; } else if (*angle == 90) { mode = VERT_PRED; } else if (*angle < 180) { mode = Z2_PRED; } else if (*angle == 180) { mode = HOR_PRED; } else { mode = have_left ? Z3_PRED : HOR_PRED; } break; } case DC_PRED: case PAETH_PRED: mode = av1_mode_conv[mode][have_left][have_top]; break; default: break; } const pixel *dst_top; if (have_top && (av1_intra_prediction_edges[mode].needs_top || av1_intra_prediction_edges[mode].needs_topleft || (av1_intra_prediction_edges[mode].needs_left && !have_left))) { if (prefilter_toplevel_sb_edge) { dst_top = &prefilter_toplevel_sb_edge[x * 4]; } else { dst_top = &dst[-PXSTRIDE(stride)]; } } if (av1_intra_prediction_edges[mode].needs_left) { const int sz = th << 2; pixel *const left = &topleft_out[-sz]; if (have_left) { const int px_have = imin(sz, (h - y) << 2); for (int i = 0; i < px_have; i++) left[sz - 1 - i] = dst[PXSTRIDE(stride) * i - 1]; if (px_have < sz) pixel_set(left, left[sz - px_have], sz - px_have); } else { pixel_set(left, have_top ? *dst_top : ((1 << BITDEPTH) >> 1) + 1, sz); } if (av1_intra_prediction_edges[mode].needs_bottomleft) { const int have_bottomleft = (!have_left || y + th >= h) ? 0 : (edge_flags & EDGE_I444_LEFT_HAS_BOTTOM); if (have_bottomleft) { const int px_have = imin(sz, (h - y - th) << 2); for (int i = 0; i < px_have; i++) left[-(i + 1)] = dst[(sz + i) * PXSTRIDE(stride) - 1]; if (px_have < sz) pixel_set(left - sz, left[-px_have], sz - px_have); } else { pixel_set(left - sz, left[0], sz); } } } if (av1_intra_prediction_edges[mode].needs_top) { const int sz = tw << 2; pixel *const top = &topleft_out[1]; if (have_top) { const int px_have = imin(sz, (w - x) << 2); pixel_copy(top, dst_top, px_have); if (px_have < sz) pixel_set(top + px_have, top[px_have - 1], sz - px_have); } else { pixel_set(top, have_left ? dst[-1] : ((1 << BITDEPTH) >> 1) - 1, sz); } if (av1_intra_prediction_edges[mode].needs_topright) { const int have_topright = (!have_top || x + tw >= w) ? 0 : (edge_flags & EDGE_I444_TOP_HAS_RIGHT); if (have_topright) { const int px_have = imin(sz, (w - x - tw) << 2); pixel_copy(top + sz, &dst_top[sz], px_have); if (px_have < sz) pixel_set(top + sz + px_have, top[sz + px_have - 1], sz - px_have); } else { pixel_set(top + sz, top[sz - 1], sz); } } } if (av1_intra_prediction_edges[mode].needs_topleft) { if (have_left) { *topleft_out = have_top ? dst_top[-1] : dst[-1]; } else { *topleft_out = have_top ? *dst_top : (1 << BITDEPTH) >> 1; } if (mode == Z2_PRED && tw + th >= 6) *topleft_out = (topleft_out[-1] * 5 + topleft_out[0] * 6 + topleft_out[1] * 5 + 8) >> 4; } return mode; }