alc/converter.cpp

   1
   2 #include "config.h"
   3
   4 #include "converter.h"
   5
   6 #include <algorithm>
   7 #include <cstdint>
   8 #include <iterator>
   9
  10 #include "AL/al.h"
  11
  12 #include "albyte.h"
  13 #include "alu.h"
  14 #include "fpu_ctrl.h"
  15 #include "mixer/defs.h"
  16
  17 struct CTag;
  18 struct CopyTag;
  19
  20
  21 namespace {
  22
  23 /* Base template left undefined. Should be marked =delete, but Clang 3.8.1
  24  * chokes on that given the inline specializations.
  25  */
  26 template<DevFmtType T>
  27 inline float LoadSample(typename DevFmtTypeTraits<T>::Type val) noexcept;
  28
  29 template<> inline float LoadSample<DevFmtByte>(DevFmtTypeTraits<DevFmtByte>::Type val) noexcept
  30 { return val * (1.0f/128.0f); }
  31 template<> inline float LoadSample<DevFmtShort>(DevFmtTypeTraits<DevFmtShort>::Type val) noexcept
  32 { return val * (1.0f/32768.0f); }
  33 template<> inline float LoadSample<DevFmtInt>(DevFmtTypeTraits<DevFmtInt>::Type val) noexcept
  34 { return static_cast<float>(val) * (1.0f/2147483648.0f); }
  35 template<> inline float LoadSample<DevFmtFloat>(DevFmtTypeTraits<DevFmtFloat>::Type val) noexcept
  36 { return val; }
  37
  38 template<> inline float LoadSample<DevFmtUByte>(DevFmtTypeTraits<DevFmtUByte>::Type val) noexcept
  39 { return LoadSample<DevFmtByte>(static_cast<ALbyte>(val - 128)); }
  40 template<> inline float LoadSample<DevFmtUShort>(DevFmtTypeTraits<DevFmtUShort>::Type val) noexcept
  41 { return LoadSample<DevFmtShort>(static_cast<ALshort>(val - 32768)); }
  42 template<> inline float LoadSample<DevFmtUInt>(DevFmtTypeTraits<DevFmtUInt>::Type val) noexcept
  43 { return LoadSample<DevFmtInt>(static_cast<ALint>(val - 2147483648u)); }
  44
  45
  46 template<DevFmtType T>
  47 inline void LoadSampleArray(float *RESTRICT dst, const void *src, const size_t srcstep,
  48     const size_t samples) noexcept
  49 {
  50     using SampleType = typename DevFmtTypeTraits<T>::Type;
  51
  52     const SampleType *ssrc = static_cast<const SampleType*>(src);
  53     for(size_t i{0u};i < samples;i++)
  54         dst[i] = LoadSample<T>(ssrc[i*srcstep]);
  55 }
  56
  57 void LoadSamples(float *dst, const void *src, const size_t srcstep, const DevFmtType srctype,
  58     const size_t samples) noexcept
  59 {
  60 #define HANDLE_FMT(T)                                                         \
  61     case T: LoadSampleArray<T>(dst, src, srcstep, samples); break
  62     switch(srctype)
  63     {
  64         HANDLE_FMT(DevFmtByte);
  65         HANDLE_FMT(DevFmtUByte);
  66         HANDLE_FMT(DevFmtShort);
  67         HANDLE_FMT(DevFmtUShort);
  68         HANDLE_FMT(DevFmtInt);
  69         HANDLE_FMT(DevFmtUInt);
  70         HANDLE_FMT(DevFmtFloat);
  71     }
  72 #undef HANDLE_FMT
  73 }
  74
  75
  76 template<DevFmtType T>
  77 inline typename DevFmtTypeTraits<T>::Type StoreSample(float) noexcept;
  78
  79 template<> inline ALfloat StoreSample<DevFmtFloat>(float val) noexcept
  80 { return val; }
  81 template<> inline ALint StoreSample<DevFmtInt>(float val) noexcept
  82 { return fastf2i(clampf(val*2147483648.0f, -2147483648.0f, 2147483520.0f)); }
  83 template<> inline ALshort StoreSample<DevFmtShort>(float val) noexcept
  84 { return static_cast<ALshort>(fastf2i(clampf(val*32768.0f, -32768.0f, 32767.0f))); }
  85 template<> inline ALbyte StoreSample<DevFmtByte>(float val) noexcept
  86 { return static_cast<ALbyte>(fastf2i(clampf(val*128.0f, -128.0f, 127.0f))); }
  87
  88 /* Define unsigned output variations. */
  89 template<> inline ALuint StoreSample<DevFmtUInt>(float val) noexcept
  90 { return static_cast<ALuint>(StoreSample<DevFmtInt>(val)) + 2147483648u; }
  91 template<> inline ALushort StoreSample<DevFmtUShort>(float val) noexcept
  92 { return static_cast<ALushort>(StoreSample<DevFmtShort>(val) + 32768); }
  93 template<> inline ALubyte StoreSample<DevFmtUByte>(float val) noexcept
  94 { return static_cast<ALubyte>(StoreSample<DevFmtByte>(val) + 128); }
  95
  96 template<DevFmtType T>
  97 inline void StoreSampleArray(void *dst, const float *RESTRICT src, const size_t dststep,
  98     const size_t samples) noexcept
  99 {
 100     using SampleType = typename DevFmtTypeTraits<T>::Type;
 101
 102     SampleType *sdst = static_cast<SampleType*>(dst);
 103     for(size_t i{0u};i < samples;i++)
 104         sdst[i*dststep] = StoreSample<T>(src[i]);
 105 }
 106
 107
 108 void StoreSamples(void *dst, const float *src, const size_t dststep, const DevFmtType dsttype,
 109     const size_t samples) noexcept
 110 {
 111 #define HANDLE_FMT(T)                                                         \
 112     case T: StoreSampleArray<T>(dst, src, dststep, samples); break
 113     switch(dsttype)
 114     {
 115         HANDLE_FMT(DevFmtByte);
 116         HANDLE_FMT(DevFmtUByte);
 117         HANDLE_FMT(DevFmtShort);
 118         HANDLE_FMT(DevFmtUShort);
 119         HANDLE_FMT(DevFmtInt);
 120         HANDLE_FMT(DevFmtUInt);
 121         HANDLE_FMT(DevFmtFloat);
 122     }
 123 #undef HANDLE_FMT
 124 }
 125
 126
 127 template<DevFmtType T>
 128 void Mono2Stereo(float *RESTRICT dst, const void *src, const size_t frames) noexcept
 129 {
 130     using SampleType = typename DevFmtTypeTraits<T>::Type;
 131
 132     const SampleType *ssrc = static_cast<const SampleType*>(src);
 133     for(size_t i{0u};i < frames;i++)
 134         dst[i*2 + 1] = dst[i*2 + 0] = LoadSample<T>(ssrc[i]) * 0.707106781187f;
 135 }
 136
 137 template<DevFmtType T>
 138 void Stereo2Mono(float *RESTRICT dst, const void *src, const size_t frames) noexcept
 139 {
 140     using SampleType = typename DevFmtTypeTraits<T>::Type;
 141
 142     const SampleType *ssrc = static_cast<const SampleType*>(src);
 143     for(size_t i{0u};i < frames;i++)
 144         dst[i] = (LoadSample<T>(ssrc[i*2 + 0])+LoadSample<T>(ssrc[i*2 + 1])) *
 145                  0.707106781187f;
 146 }
 147
 148 } // namespace
 149
 150 SampleConverterPtr CreateSampleConverter(DevFmtType srcType, DevFmtType dstType, size_t numchans,
 151     ALuint srcRate, ALuint dstRate, Resampler resampler)
 152 {
 153     if(numchans < 1 || srcRate < 1 || dstRate < 1)
 154         return nullptr;
 155
 156     SampleConverterPtr converter{new(FamCount(numchans)) SampleConverter{numchans}};
 157     converter->mSrcType = srcType;
 158     converter->mDstType = dstType;
 159     converter->mSrcTypeSize = BytesFromDevFmt(srcType);
 160     converter->mDstTypeSize = BytesFromDevFmt(dstType);
 161
 162     converter->mSrcPrepCount = 0;
 163     converter->mFracOffset = 0;
 164
 165     /* Have to set the mixer FPU mode since that's what the resampler code expects. */
 166     FPUCtl mixer_mode{};
 167     auto step = static_cast<ALuint>(
 168         mind(srcRate*double{FRACTIONONE}/dstRate + 0.5, MAX_PITCH*FRACTIONONE));
 169     converter->mIncrement = maxu(step, 1);
 170     if(converter->mIncrement == FRACTIONONE)
 171         converter->mResample = Resample_<CopyTag,CTag>;
 172     else
 173         converter->mResample = PrepareResampler(resampler, converter->mIncrement,
 174             &converter->mState);
 175
 176     return converter;
 177 }
 178
 179 ALuint SampleConverter::availableOut(ALuint srcframes) const
 180 {
 181     int prepcount{mSrcPrepCount};
 182     if(prepcount < 0)
 183     {
 184         /* Negative prepcount means we need to skip that many input samples. */
 185         if(static_cast<ALuint>(-prepcount) >= srcframes)
 186             return 0;
 187         srcframes -= static_cast<ALuint>(-prepcount);
 188         prepcount = 0;
 189     }
 190
 191     if(srcframes < 1)
 192     {
 193         /* No output samples if there's no input samples. */
 194         return 0;
 195     }
 196
 197     if(prepcount < MAX_RESAMPLER_PADDING
 198         && static_cast<ALuint>(MAX_RESAMPLER_PADDING - prepcount) >= srcframes)
 199     {
 200         /* Not enough input samples to generate an output sample. */
 201         return 0;
 202     }
 203
 204     auto DataSize64 = static_cast<uint64_t>(prepcount);
 205     DataSize64 += srcframes;
 206     DataSize64 -= MAX_RESAMPLER_PADDING;
 207     DataSize64 <<= FRACTIONBITS;
 208     DataSize64 -= mFracOffset;
 209
 210     /* If we have a full prep, we can generate at least one sample. */
 211     return static_cast<ALuint>(clampu64((DataSize64 + mIncrement-1)/mIncrement, 1, BUFFERSIZE));
 212 }
 213
 214 ALuint SampleConverter::convert(const void **src, ALuint *srcframes, void *dst, ALuint dstframes)
 215 {
 216     const ALuint SrcFrameSize{static_cast<ALuint>(mChan.size()) * mSrcTypeSize};
 217     const ALuint DstFrameSize{static_cast<ALuint>(mChan.size()) * mDstTypeSize};
 218     const ALuint increment{mIncrement};
 219     auto SamplesIn = static_cast<const al::byte*>(*src);
 220     ALuint NumSrcSamples{*srcframes};
 221
 222     FPUCtl mixer_mode{};
 223     ALuint pos{0};
 224     while(pos < dstframes && NumSrcSamples > 0)
 225     {
 226         int prepcount{mSrcPrepCount};
 227         if(prepcount < 0)
 228         {
 229             /* Negative prepcount means we need to skip that many input samples. */
 230             if(static_cast<ALuint>(-prepcount) >= NumSrcSamples)
 231             {
 232                 mSrcPrepCount = static_cast<int>(NumSrcSamples) + prepcount;
 233                 NumSrcSamples = 0;
 234                 break;
 235             }
 236             SamplesIn += SrcFrameSize*static_cast<ALuint>(-prepcount);
 237             NumSrcSamples -= static_cast<ALuint>(-prepcount);
 238             mSrcPrepCount = 0;
 239             continue;
 240         }
 241         ALuint toread{minu(NumSrcSamples, BUFFERSIZE - MAX_RESAMPLER_PADDING)};
 242
 243         if(prepcount < MAX_RESAMPLER_PADDING
 244             && static_cast<ALuint>(MAX_RESAMPLER_PADDING - prepcount) >= toread)
 245         {
 246             /* Not enough input samples to generate an output sample. Store
 247              * what we're given for later.
 248              */
 249             for(size_t chan{0u};chan < mChan.size();chan++)
 250                 LoadSamples(&mChan[chan].PrevSamples[prepcount], SamplesIn + mSrcTypeSize*chan,
 251                     mChan.size(), mSrcType, toread);
 252
 253             mSrcPrepCount = prepcount + static_cast<int>(toread);
 254             NumSrcSamples = 0;
 255             break;
 256         }
 257
 258         float *RESTRICT SrcData{mSrcSamples};
 259         float *RESTRICT DstData{mDstSamples};
 260         ALuint DataPosFrac{mFracOffset};
 261         auto DataSize64 = static_cast<uint64_t>(prepcount);
 262         DataSize64 += toread;
 263         DataSize64 -= MAX_RESAMPLER_PADDING;
 264         DataSize64 <<= FRACTIONBITS;
 265         DataSize64 -= DataPosFrac;
 266
 267         /* If we have a full prep, we can generate at least one sample. */
 268         auto DstSize = static_cast<ALuint>(
 269             clampu64((DataSize64 + increment-1)/increment, 1, BUFFERSIZE));
 270         DstSize = minu(DstSize, dstframes-pos);
 271
 272         for(size_t chan{0u};chan < mChan.size();chan++)
 273         {
 274             const al::byte *SrcSamples{SamplesIn + mSrcTypeSize*chan};
 275             al::byte *DstSamples = static_cast<al::byte*>(dst) + mDstTypeSize*chan;
 276
 277             /* Load the previous samples into the source data first, then the
 278              * new samples from the input buffer.
 279              */
 280             std::copy_n(mChan[chan].PrevSamples, prepcount, SrcData);
 281             LoadSamples(SrcData + prepcount, SrcSamples, mChan.size(), mSrcType, toread);
 282
 283             /* Store as many prep samples for next time as possible, given the
 284              * number of output samples being generated.
 285              */
 286             ALuint SrcDataEnd{(DstSize*increment + DataPosFrac)>>FRACTIONBITS};
 287             if(SrcDataEnd >= static_cast<ALuint>(prepcount)+toread)
 288                 std::fill(std::begin(mChan[chan].PrevSamples),
 289                     std::end(mChan[chan].PrevSamples), 0.0f);
 290             else
 291             {
 292                 const size_t len{minz(al::size(mChan[chan].PrevSamples),
 293                     static_cast<ALuint>(prepcount)+toread-SrcDataEnd)};
 294                 std::copy_n(SrcData+SrcDataEnd, len, mChan[chan].PrevSamples);
 295                 std::fill(std::begin(mChan[chan].PrevSamples)+len,
 296                     std::end(mChan[chan].PrevSamples), 0.0f);
 297             }
 298
 299             /* Now resample, and store the result in the output buffer. */
 300             const float *ResampledData{mResample(&mState, SrcData+(MAX_RESAMPLER_PADDING>>1),
 301                 DataPosFrac, increment, {DstData, DstSize})};
 302
 303             StoreSamples(DstSamples, ResampledData, mChan.size(), mDstType, DstSize);
 304         }
 305
 306         /* Update the number of prep samples still available, as well as the
 307          * fractional offset.
 308          */
 309         DataPosFrac += increment*DstSize;
 310         mSrcPrepCount = mini(prepcount + static_cast<int>(toread - (DataPosFrac>>FRACTIONBITS)),
 311             MAX_RESAMPLER_PADDING);
 312         mFracOffset = DataPosFrac & FRACTIONMASK;
 313
 314         /* Update the src and dst pointers in case there's still more to do. */
 315         SamplesIn += SrcFrameSize*(DataPosFrac>>FRACTIONBITS);
 316         NumSrcSamples -= minu(NumSrcSamples, (DataPosFrac>>FRACTIONBITS));
 317
 318         dst = static_cast<al::byte*>(dst) + DstFrameSize*DstSize;
 319         pos += DstSize;
 320     }
 321
 322     *src = SamplesIn;
 323     *srcframes = NumSrcSamples;
 324
 325     return pos;
 326 }
 327
 328
 329 void ChannelConverter::convert(const void *src, float *dst, ALuint frames) const
 330 {
 331     if(mSrcChans == DevFmtStereo && mDstChans == DevFmtMono)
 332     {
 333         switch(mSrcType)
 334         {
 335 #define HANDLE_FMT(T) case T: Stereo2Mono<T>(dst, src, frames); break
 336         HANDLE_FMT(DevFmtByte);
 337         HANDLE_FMT(DevFmtUByte);
 338         HANDLE_FMT(DevFmtShort);
 339         HANDLE_FMT(DevFmtUShort);
 340         HANDLE_FMT(DevFmtInt);
 341         HANDLE_FMT(DevFmtUInt);
 342         HANDLE_FMT(DevFmtFloat);
 343 #undef HANDLE_FMT
 344         }
 345     }
 346     else if(mSrcChans == DevFmtMono && mDstChans == DevFmtStereo)
 347     {
 348         switch(mSrcType)
 349         {
 350 #define HANDLE_FMT(T) case T: Mono2Stereo<T>(dst, src, frames); break
 351         HANDLE_FMT(DevFmtByte);
 352         HANDLE_FMT(DevFmtUByte);
 353         HANDLE_FMT(DevFmtShort);
 354         HANDLE_FMT(DevFmtUShort);
 355         HANDLE_FMT(DevFmtInt);
 356         HANDLE_FMT(DevFmtUInt);
 357         HANDLE_FMT(DevFmtFloat);
 358 #undef HANDLE_FMT
 359         }
 360     }
 361     else
 362         LoadSamples(dst, src, 1u, mSrcType, frames * ChannelsFromDevFmt(mSrcChans, 0));
 363 }