开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > 多媒体 > 流媒体/Mpeg4/MP4 > 查看源码
ratecontrol.c
资源名称：NETVIDEO.rar　[点击查看]
上传用户：sun1608
上传日期：2007-02-02
资源大小：6116k
文件大小：3k
源码类别：

流媒体/Mpeg4/MP4

开发平台：

Visual C++

ratecontrol.c：源码内容
  1. #include "ratecontrol.h"
  3. #include <math.h> /* fabs */
  5. #define ABS(X) (X < 0) ? (-X) : (X)
  7. typedef struct {
  8. int64_t size;
  9. int32_t count;
  10. } QuantInfo;
  12. typedef struct
  13. {
  14. int32_t rtn_quant;
  15. int64_t frames;
  16. int64_t total_size;
  17. double framerate;
  18. int32_t target_rate;
  19. int16_t max_quant;
  20. int16_t min_quant;
  21. int64_t last_change;
  22. int64_t quant_sum;
  23. double quant_error[32];
  24. double avg_framesize;
  25. double target_framesize;
  26. int32_t reaction_delay_factor;
  27. } RateControl;
  29. RateControl rate_control;
  31. int get_initial_quant(int bpp) {
  32. return 5;
  33. }
  35. void RateControlInit(uint32_t target_rate, uint32_t reaction_delay_factor, int framerate,
  36.      int max_quant, int min_quant)
  37. {
  38. int i;
  40. rate_control.frames = 0;
  41. rate_control.total_size = 0;
  43. rate_control.framerate = framerate / 1000.0;
  44. rate_control.target_rate = target_rate;
  46. rate_control.rtn_quant = get_initial_quant(0);
  47. rate_control.max_quant = max_quant;
  48. rate_control.min_quant = min_quant;
  50. for (i=0 ; i<32 ; ++i)
  51. {
  52. rate_control.quant_error[i] = 0.0;
  53. }
  55. rate_control.target_framesize = target_rate / rate_control.framerate;
  56. rate_control.avg_framesize = target_rate / rate_control.framerate;
  58. rate_control.reaction_delay_factor = reaction_delay_factor;
  59. }
  61. int RateControlGetQ(int keyframe)
  62. {
  63. return rate_control.rtn_quant;
  64. }
  66. void RateControlUpdate(int16_t quant, int frame_size, int keyframe)
  67. {
  68. int64_t deviation;
  69. double overflow, cur_target;
  70. int32_t rtn_quant;
  72. rate_control.frames++;
  73. rate_control.total_size += frame_size;
  75. deviation = (int64_t) ((double) rate_control.total_size - 
  76.        ((double) ((double) rate_control.target_rate / 8.0 /
  77.  (double) rate_control.framerate) * (double) rate_control.frames));
  79. DEBUGCBR((int32_t)(rate_control.frames - 1), rate_control.rtn_quant, (int32_t)deviation);
  81. rate_control.avg_framesize -= rate_control.avg_framesize / (double)rate_control.reaction_delay_factor;
  82. rate_control.avg_framesize += frame_size * rate_control.rtn_quant * 0.5 / (double)rate_control.reaction_delay_factor;
  84. if (rate_control.target_framesize > rate_control.avg_framesize)
  85. cur_target = rate_control.avg_framesize;
  86. else
  87. cur_target = rate_control.target_framesize;
  89. deviation /= -rate_control.reaction_delay_factor;
  91. overflow = (double)deviation * rate_control.avg_framesize / rate_control.target_framesize;
  93. if (overflow > cur_target)
  94. overflow = cur_target;
  95. else if (overflow < cur_target * -0.935)
  96. overflow = cur_target * -0.935;
  98. rtn_quant = (int32_t)(rate_control.avg_framesize * 2.0 / (cur_target + overflow));
  100. if (rtn_quant < 31)
  101. {
  102. rate_control.quant_error[rtn_quant] += rate_control.avg_framesize * 2.0 / (cur_target + overflow) - rtn_quant;
  103. if (rate_control.quant_error[rtn_quant] >= 1.0)
  104. {
  105. rate_control.quant_error[rtn_quant] -= 1.0;
  106. rtn_quant++;
  107. }
  108. }
  110. if (rtn_quant > rate_control.rtn_quant + 1)
  111. rtn_quant = rate_control.rtn_quant + 1;
  112. else if (rtn_quant < rate_control.rtn_quant - 1)
  113. rtn_quant = rate_control.rtn_quant - 1;
  115. if (rtn_quant > rate_control.max_quant)
  116. rtn_quant = rate_control.max_quant;
  117. else if (rtn_quant < rate_control.min_quant)
  118. rtn_quant = rate_control.min_quant;
  120. rate_control.rtn_quant = rtn_quant;
  121. }