You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

160 lines
4.4KB

  1. /*
  2. * Copyright (c) 2014 Luca Barbato <lu_zero@gentoo.org>
  3. * Copyright (c) 2014 Michael Niedermayer <michaelni@gmx.at>
  4. *
  5. * This file is part of FFmpeg.
  6. *
  7. * FFmpeg is free software; you can redistribute it and/or
  8. * modify it under the terms of the GNU Lesser General Public
  9. * License as published by the Free Software Foundation; either
  10. * version 2.1 of the License, or (at your option) any later version.
  11. *
  12. * FFmpeg is distributed in the hope that it will be useful,
  13. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  15. * Lesser General Public License for more details.
  16. *
  17. * You should have received a copy of the GNU Lesser General Public
  18. * License along with FFmpeg; if not, write to the Free Software
  19. * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  20. */
  21. #include "swresample_internal.h"
  22. #include "libavutil/frame.h"
  23. #include "libavutil/opt.h"
  24. int swr_config_frame(SwrContext *s, const AVFrame *out, const AVFrame *in)
  25. {
  26. swr_close(s);
  27. if (in) {
  28. if (av_opt_set_int(s, "icl", in->channel_layout, 0) < 0)
  29. goto fail;
  30. if (av_opt_set_int(s, "isf", in->format, 0) < 0)
  31. goto fail;
  32. if (av_opt_set_int(s, "isr", in->sample_rate, 0) < 0)
  33. goto fail;
  34. }
  35. if (out) {
  36. if (av_opt_set_int(s, "ocl", out->channel_layout, 0) < 0)
  37. goto fail;
  38. if (av_opt_set_int(s, "osf", out->format, 0) < 0)
  39. goto fail;
  40. if (av_opt_set_int(s, "osr", out->sample_rate, 0) < 0)
  41. goto fail;
  42. }
  43. return 0;
  44. fail:
  45. av_log(s, AV_LOG_ERROR, "Failed to set option\n");
  46. return AVERROR(EINVAL);
  47. }
  48. static int config_changed(SwrContext *s,
  49. const AVFrame *out, const AVFrame *in)
  50. {
  51. int ret = 0;
  52. if (in) {
  53. if (s->in_ch_layout != in->channel_layout ||
  54. s->in_sample_rate != in->sample_rate ||
  55. s->in_sample_fmt != in->format) {
  56. ret |= AVERROR_INPUT_CHANGED;
  57. }
  58. }
  59. if (out) {
  60. if (s->out_ch_layout != out->channel_layout ||
  61. s->out_sample_rate != out->sample_rate ||
  62. s->out_sample_fmt != out->format) {
  63. ret |= AVERROR_OUTPUT_CHANGED;
  64. }
  65. }
  66. return ret;
  67. }
  68. static inline int convert_frame(SwrContext *s,
  69. AVFrame *out, const AVFrame *in)
  70. {
  71. int ret;
  72. uint8_t **out_data = NULL;
  73. const uint8_t **in_data = NULL;
  74. int out_nb_samples = 0, in_nb_samples = 0;
  75. if (out) {
  76. out_data = out->extended_data;
  77. out_nb_samples = out->nb_samples;
  78. }
  79. if (in) {
  80. in_data = (const uint8_t **)in->extended_data;
  81. in_nb_samples = in->nb_samples;
  82. }
  83. ret = swr_convert(s, out_data, out_nb_samples, in_data, in_nb_samples);
  84. if (ret < 0) {
  85. if (out)
  86. out->nb_samples = 0;
  87. return ret;
  88. }
  89. if (out)
  90. out->nb_samples = ret;
  91. return 0;
  92. }
  93. static inline int available_samples(AVFrame *out)
  94. {
  95. int bytes_per_sample = av_get_bytes_per_sample(out->format);
  96. int samples = out->linesize[0] / bytes_per_sample;
  97. if (av_sample_fmt_is_planar(out->format)) {
  98. return samples;
  99. } else {
  100. int channels = av_get_channel_layout_nb_channels(out->channel_layout);
  101. return samples / channels;
  102. }
  103. }
  104. int swr_convert_frame(SwrContext *s,
  105. AVFrame *out, const AVFrame *in)
  106. {
  107. int ret, setup = 0;
  108. if (!swr_is_initialized(s)) {
  109. if ((ret = swr_config_frame(s, out, in)) < 0)
  110. return ret;
  111. if ((ret = swr_init(s)) < 0)
  112. return ret;
  113. setup = 1;
  114. } else {
  115. // return as is or reconfigure for input changes?
  116. if ((ret = config_changed(s, out, in)))
  117. return ret;
  118. }
  119. if (out) {
  120. if (!out->linesize[0]) {
  121. out->nb_samples = swr_get_delay(s, s->out_sample_rate) + 3;
  122. if (in) {
  123. out->nb_samples += in->nb_samples*(int64_t)s->out_sample_rate / s->in_sample_rate;
  124. }
  125. if ((ret = av_frame_get_buffer(out, 0)) < 0) {
  126. if (setup)
  127. swr_close(s);
  128. return ret;
  129. }
  130. } else {
  131. if (!out->nb_samples)
  132. out->nb_samples = available_samples(out);
  133. }
  134. }
  135. return convert_frame(s, out, in);
  136. }