samplefmt.c 4.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130
  1. /*
  2. * This file is part of FFmpeg.
  3. *
  4. * FFmpeg is free software; you can redistribute it and/or
  5. * modify it under the terms of the GNU Lesser General Public
  6. * License as published by the Free Software Foundation; either
  7. * version 2.1 of the License, or (at your option) any later version.
  8. *
  9. * FFmpeg is distributed in the hope that it will be useful,
  10. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  12. * Lesser General Public License for more details.
  13. *
  14. * You should have received a copy of the GNU Lesser General Public
  15. * License along with FFmpeg; if not, write to the Free Software
  16. * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  17. */
  18. #include "samplefmt.h"
  19. #include <stdio.h>
  20. #include <stdlib.h>
  21. #include <string.h>
  22. typedef struct SampleFmtInfo {
  23. const char *name;
  24. int bits;
  25. } SampleFmtInfo;
  26. /** this table gives more information about formats */
  27. static const SampleFmtInfo sample_fmt_info[AV_SAMPLE_FMT_NB] = {
  28. [AV_SAMPLE_FMT_U8] = { .name = "u8", .bits = 8 },
  29. [AV_SAMPLE_FMT_S16] = { .name = "s16", .bits = 16 },
  30. [AV_SAMPLE_FMT_S32] = { .name = "s32", .bits = 32 },
  31. [AV_SAMPLE_FMT_FLT] = { .name = "flt", .bits = 32 },
  32. [AV_SAMPLE_FMT_DBL] = { .name = "dbl", .bits = 64 },
  33. };
  34. const char *av_get_sample_fmt_name(enum AVSampleFormat sample_fmt)
  35. {
  36. if (sample_fmt < 0 || sample_fmt >= AV_SAMPLE_FMT_NB)
  37. return NULL;
  38. return sample_fmt_info[sample_fmt].name;
  39. }
  40. enum AVSampleFormat av_get_sample_fmt(const char *name)
  41. {
  42. int i;
  43. for (i = 0; i < AV_SAMPLE_FMT_NB; i++)
  44. if (!strcmp(sample_fmt_info[i].name, name))
  45. return i;
  46. return AV_SAMPLE_FMT_NONE;
  47. }
  48. char *av_get_sample_fmt_string (char *buf, int buf_size, enum AVSampleFormat sample_fmt)
  49. {
  50. /* print header */
  51. if (sample_fmt < 0)
  52. snprintf(buf, buf_size, "name " " depth");
  53. else if (sample_fmt < AV_SAMPLE_FMT_NB) {
  54. SampleFmtInfo info = sample_fmt_info[sample_fmt];
  55. snprintf (buf, buf_size, "%-6s" " %2d ", info.name, info.bits);
  56. }
  57. return buf;
  58. }
  59. int av_get_bytes_per_sample(enum AVSampleFormat sample_fmt)
  60. {
  61. return sample_fmt < 0 || sample_fmt >= AV_SAMPLE_FMT_NB ?
  62. 0 : sample_fmt_info[sample_fmt].bits >> 3;
  63. }
  64. #if FF_API_GET_BITS_PER_SAMPLE_FMT
  65. int av_get_bits_per_sample_fmt(enum AVSampleFormat sample_fmt)
  66. {
  67. return sample_fmt < 0 || sample_fmt >= AV_SAMPLE_FMT_NB ?
  68. 0 : sample_fmt_info[sample_fmt].bits;
  69. }
  70. #endif
  71. int av_samples_fill_arrays(uint8_t *pointers[8], int linesizes[8],
  72. uint8_t *buf, int nb_channels, int nb_samples,
  73. enum AVSampleFormat sample_fmt, int planar, int align)
  74. {
  75. int i, linesize;
  76. int sample_size = av_get_bytes_per_sample(sample_fmt);
  77. if (nb_channels * (uint64_t)nb_samples * sample_size >= INT_MAX - align*(uint64_t)nb_channels)
  78. return AVERROR(EINVAL);
  79. linesize = planar ? FFALIGN(nb_samples*sample_size, align) :
  80. FFALIGN(nb_samples*sample_size*nb_channels, align);
  81. if (pointers) {
  82. pointers[0] = buf;
  83. for (i = 1; planar && i < nb_channels; i++) {
  84. pointers[i] = pointers[i-1] + linesize;
  85. }
  86. memset(&pointers[i], 0, (8-i) * sizeof(pointers[0]));
  87. }
  88. if (linesizes) {
  89. linesizes[0] = linesize;
  90. for (i = 1; planar && i < nb_channels; i++)
  91. linesizes[i] = linesizes[0];
  92. memset(&linesizes[i], 0, (8-i) * sizeof(linesizes[0]));
  93. }
  94. return planar ? linesize * nb_channels : linesize;
  95. }
  96. int av_samples_alloc(uint8_t *pointers[8], int linesizes[8],
  97. int nb_channels, int nb_samples,
  98. enum AVSampleFormat sample_fmt, int planar,
  99. int align)
  100. {
  101. uint8_t *buf;
  102. int size = av_samples_fill_arrays(NULL, NULL,
  103. NULL, nb_channels, nb_samples,
  104. sample_fmt, planar, align);
  105. buf = av_mallocz(size);
  106. if (!buf)
  107. return AVERROR(ENOMEM);
  108. return av_samples_fill_arrays(pointers, linesizes,
  109. buf, nb_channels, nb_samples,
  110. sample_fmt, planar, align);
  111. }