mpegpicture.h 3.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114
  1. /*
  2. * Mpeg video formats-related defines and utility functions
  3. *
  4. * This file is part of FFmpeg.
  5. *
  6. * FFmpeg is free software; you can redistribute it and/or
  7. * modify it under the terms of the GNU Lesser General Public
  8. * License as published by the Free Software Foundation; either
  9. * version 2.1 of the License, or (at your option) any later version.
  10. *
  11. * FFmpeg is distributed in the hope that it will be useful,
  12. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  14. * Lesser General Public License for more details.
  15. *
  16. * You should have received a copy of the GNU Lesser General Public
  17. * License along with FFmpeg; if not, write to the Free Software
  18. * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  19. */
  20. #ifndef AVCODEC_MPEGPICTURE_H
  21. #define AVCODEC_MPEGPICTURE_H
  22. #include <stdint.h>
  23. #include "libavutil/frame.h"
  24. #include "avcodec.h"
  25. #include "motion_est.h"
  26. #include "thread.h"
  27. #define MAX_PICTURE_COUNT 36
  28. #define EDGE_WIDTH 16
  29. typedef struct ScratchpadContext {
  30. uint8_t *edge_emu_buffer; ///< temporary buffer for if MVs point to out-of-frame data
  31. uint8_t *rd_scratchpad; ///< scratchpad for rate distortion mb decision
  32. uint8_t *obmc_scratchpad;
  33. uint8_t *b_scratchpad; ///< scratchpad used for writing into write only buffers
  34. } ScratchpadContext;
  35. /**
  36. * Picture.
  37. */
  38. typedef struct Picture {
  39. struct AVFrame *f;
  40. ThreadFrame tf;
  41. AVBufferRef *qscale_table_buf;
  42. int8_t *qscale_table;
  43. AVBufferRef *motion_val_buf[2];
  44. int16_t (*motion_val[2])[2];
  45. AVBufferRef *mb_type_buf;
  46. uint32_t *mb_type; ///< types and macros are defined in mpegutils.h
  47. AVBufferRef *mbskip_table_buf;
  48. uint8_t *mbskip_table;
  49. AVBufferRef *ref_index_buf[2];
  50. int8_t *ref_index[2];
  51. AVBufferRef *mb_var_buf;
  52. uint16_t *mb_var; ///< Table for MB variances
  53. AVBufferRef *mc_mb_var_buf;
  54. uint16_t *mc_mb_var; ///< Table for motion compensated MB variances
  55. int alloc_mb_width; ///< mb_width used to allocate tables
  56. int alloc_mb_height; ///< mb_height used to allocate tables
  57. AVBufferRef *mb_mean_buf;
  58. uint8_t *mb_mean; ///< Table for MB luminance
  59. AVBufferRef *hwaccel_priv_buf;
  60. void *hwaccel_picture_private; ///< Hardware accelerator private data
  61. int field_picture; ///< whether or not the picture was encoded in separate fields
  62. int64_t mb_var_sum; ///< sum of MB variance for current frame
  63. int64_t mc_mb_var_sum; ///< motion compensated MB variance for current frame
  64. int b_frame_score;
  65. int needs_realloc; ///< Picture needs to be reallocated (eg due to a frame size change)
  66. int reference;
  67. int shared;
  68. uint64_t encoding_error[AV_NUM_DATA_POINTERS];
  69. } Picture;
  70. /**
  71. * Allocate a Picture.
  72. * The pixels are allocated/set by calling get_buffer() if shared = 0.
  73. */
  74. int ff_alloc_picture(AVCodecContext *avctx, Picture *pic, MotionEstContext *me,
  75. ScratchpadContext *sc, int shared, int encoding,
  76. int chroma_x_shift, int chroma_y_shift, int out_format,
  77. int mb_stride, int mb_width, int mb_height, int b8_stride,
  78. ptrdiff_t *linesize, ptrdiff_t *uvlinesize);
  79. int ff_mpeg_framesize_alloc(AVCodecContext *avctx, MotionEstContext *me,
  80. ScratchpadContext *sc, int linesize);
  81. int ff_mpeg_ref_picture(AVCodecContext *avctx, Picture *dst, Picture *src);
  82. void ff_mpeg_unref_picture(AVCodecContext *avctx, Picture *picture);
  83. void ff_free_picture_tables(Picture *pic);
  84. int ff_update_picture_tables(Picture *dst, Picture *src);
  85. int ff_find_unused_picture(AVCodecContext *avctx, Picture *picture, int shared);
  86. #endif /* AVCODEC_MPEGPICTURE_H */