Alc/converter.c

   1
   2 #include "config.h"
   3
   4 #include "converter.h"
   5
   6 #include "mixer_defs.h"
   7
   8
   9 SampleConverter *CreateSampleConverter(enum DevFmtType srcType, enum DevFmtType dstType, ALsizei numchans, ALsizei srcRate, ALsizei dstRate)
  10 {
  11     SampleConverter *converter;
  12     FPUCtl oldMode;
  13
  14     if(numchans <= 0 || srcRate <= 0 || dstRate <= 0)
  15         return NULL;
  16
  17     converter = al_calloc(16, FAM_SIZE(SampleConverter, Chan, numchans));
  18     converter->mSrcType = srcType;
  19     converter->mDstType = dstType;
  20     converter->mNumChannels = numchans;
  21     converter->mSrcTypeSize = BytesFromDevFmt(srcType);
  22     converter->mDstTypeSize = BytesFromDevFmt(dstType);
  23
  24     converter->mSrcPrepCount = 0;
  25     converter->mFracOffset = 0;
  26
  27     /* Have to set the mixer FPU mode since that's what the resampler code expects. */
  28     SetMixerFPUMode(&oldMode);
  29     converter->mIncrement = (ALsizei)clampu64((ALuint64)srcRate*FRACTIONONE/dstRate,
  30                                               1, MAX_PITCH*FRACTIONONE);
  31     if(converter->mIncrement == FRACTIONONE)
  32         converter->mResample = Resample_copy32_C;
  33     else
  34     {
  35         /* TODO: Allow other resamplers. */
  36         BsincPrepare(converter->mIncrement, &converter->mState.bsinc);
  37         converter->mResample = SelectResampler(BSincResampler);
  38     }
  39     RestoreFPUMode(&oldMode);
  40
  41     return converter;
  42 }
  43
  44 void DestroySampleConverter(SampleConverter **converter)
  45 {
  46     if(converter)
  47     {
  48         al_free(*converter);
  49         *converter = NULL;
  50     }
  51 }
  52
  53
  54 static inline ALfloat Sample_ALbyte(ALbyte val)
  55 { return val * (1.0f/128.0f); }
  56 static inline ALfloat Sample_ALubyte(ALubyte val)
  57 { return Sample_ALbyte((ALint)val - 128); }
  58
  59 static inline ALfloat Sample_ALshort(ALshort val)
  60 { return val * (1.0f/32768.0f); }
  61 static inline ALfloat Sample_ALushort(ALushort val)
  62 { return Sample_ALshort((ALint)val - 32768); }
  63
  64 static inline ALfloat Sample_ALint(ALint val)
  65 { return (val>>7) * (1.0f/16777216.0f); }
  66 static inline ALfloat Sample_ALuint(ALuint val)
  67 { return Sample_ALint(val - INT_MAX - 1); }
  68
  69 static inline ALfloat Sample_ALfloat(ALfloat val)
  70 { return val; }
  71
  72 #define DECL_TEMPLATE(T)                                                      \
  73 static inline void Load_##T(ALfloat *restrict dst, const T *restrict src,     \
  74                             ALint srcstep, ALsizei samples)                   \
  75 {                                                                             \
  76     ALsizei i;                                                                \
  77     for(i = 0;i < samples;i++)                                                \
  78         dst[i] = Sample_##T(src[i*srcstep]);                                  \
  79 }
  80
  81 DECL_TEMPLATE(ALbyte)
  82 DECL_TEMPLATE(ALubyte)
  83 DECL_TEMPLATE(ALshort)
  84 DECL_TEMPLATE(ALushort)
  85 DECL_TEMPLATE(ALint)
  86 DECL_TEMPLATE(ALuint)
  87 DECL_TEMPLATE(ALfloat)
  88
  89 #undef DECL_TEMPLATE
  90
  91 static void LoadSamples(ALfloat *dst, const ALvoid *src, ALint srcstep, enum DevFmtType srctype, ALsizei samples)
  92 {
  93     switch(srctype)
  94     {
  95         case DevFmtByte:
  96             Load_ALbyte(dst, src, srcstep, samples);
  97             break;
  98         case DevFmtUByte:
  99             Load_ALubyte(dst, src, srcstep, samples);
 100             break;
 101         case DevFmtShort:
 102             Load_ALshort(dst, src, srcstep, samples);
 103             break;
 104         case DevFmtUShort:
 105             Load_ALushort(dst, src, srcstep, samples);
 106             break;
 107         case DevFmtInt:
 108             Load_ALint(dst, src, srcstep, samples);
 109             break;
 110         case DevFmtUInt:
 111             Load_ALuint(dst, src, srcstep, samples);
 112             break;
 113         case DevFmtFloat:
 114             Load_ALfloat(dst, src, srcstep, samples);
 115             break;
 116     }
 117 }
 118
 119
 120 static inline ALbyte ALbyte_Sample(ALfloat val)
 121 { return fastf2i(clampf(val*128.0f, -128.0f, 127.0f)); }
 122 static inline ALubyte ALubyte_Sample(ALfloat val)
 123 { return ALbyte_Sample(val)+128; }
 124
 125 static inline ALshort ALshort_Sample(ALfloat val)
 126 { return fastf2i(clampf(val*32768.0f, -32768.0f, 32767.0f)); }
 127 static inline ALushort ALushort_Sample(ALfloat val)
 128 { return ALshort_Sample(val)+32768; }
 129
 130 static inline ALint ALint_Sample(ALfloat val)
 131 { return fastf2i(clampf(val*16777216.0f, -16777216.0f, 16777215.0f)) << 7; }
 132 static inline ALuint ALuint_Sample(ALfloat val)
 133 { return ALint_Sample(val)+INT_MAX+1; }
 134
 135 static inline ALfloat ALfloat_Sample(ALfloat val)
 136 { return val; }
 137
 138 #define DECL_TEMPLATE(T)                                                      \
 139 static inline void Store_##T(T *restrict dst, const ALfloat *restrict src,    \
 140                              ALint dststep, ALsizei samples)                  \
 141 {                                                                             \
 142     ALsizei i;                                                                \
 143     for(i = 0;i < samples;i++)                                                \
 144         dst[i*dststep] = T##_Sample(src[i]);                                  \
 145 }
 146
 147 DECL_TEMPLATE(ALbyte)
 148 DECL_TEMPLATE(ALubyte)
 149 DECL_TEMPLATE(ALshort)
 150 DECL_TEMPLATE(ALushort)
 151 DECL_TEMPLATE(ALint)
 152 DECL_TEMPLATE(ALuint)
 153 DECL_TEMPLATE(ALfloat)
 154
 155 #undef DECL_TEMPLATE
 156
 157 static void StoreSamples(ALvoid *dst, const ALfloat *src, ALint dststep, enum DevFmtType dsttype, ALsizei samples)
 158 {
 159     switch(dsttype)
 160     {
 161         case DevFmtByte:
 162             Store_ALbyte(dst, src, dststep, samples);
 163             break;
 164         case DevFmtUByte:
 165             Store_ALubyte(dst, src, dststep, samples);
 166             break;
 167         case DevFmtShort:
 168             Store_ALshort(dst, src, dststep, samples);
 169             break;
 170         case DevFmtUShort:
 171             Store_ALushort(dst, src, dststep, samples);
 172             break;
 173         case DevFmtInt:
 174             Store_ALint(dst, src, dststep, samples);
 175             break;
 176         case DevFmtUInt:
 177             Store_ALuint(dst, src, dststep, samples);
 178             break;
 179         case DevFmtFloat:
 180             Store_ALfloat(dst, src, dststep, samples);
 181             break;
 182     }
 183 }
 184
 185
 186 ALsizei SampleConverterAvailableOut(SampleConverter *converter, ALsizei srcframes)
 187 {
 188     ALint prepcount = converter->mSrcPrepCount;
 189     ALsizei increment = converter->mIncrement;
 190     ALsizei DataPosFrac = converter->mFracOffset;
 191     ALuint64 DataSize64;
 192
 193     if(prepcount < 0)
 194     {
 195         /* Negative prepcount means we need to skip that many input samples. */
 196         if(-prepcount >= srcframes)
 197             return 0;
 198         srcframes += prepcount;
 199         prepcount = 0;
 200     }
 201
 202     if(srcframes < 1)
 203     {
 204         /* No output samples if there's no input samples. */
 205         return 0;
 206     }
 207
 208     if(prepcount < MAX_POST_SAMPLES+MAX_PRE_SAMPLES &&
 209        MAX_POST_SAMPLES+MAX_PRE_SAMPLES-prepcount >= srcframes)
 210     {
 211         /* Not enough input samples to generate an output sample. */
 212         return 0;
 213     }
 214
 215     DataSize64  = prepcount;
 216     DataSize64 += srcframes;
 217     DataSize64 -= MAX_POST_SAMPLES+MAX_PRE_SAMPLES;
 218     DataSize64 <<= FRACTIONBITS;
 219     DataSize64 -= DataPosFrac;
 220
 221     /* If we have a full prep, we can generate at least one sample. */
 222     return (ALsizei)clampu64(DataSize64/increment, 1, INT_MAX);
 223 }
 224
 225
 226 ALsizei SampleConverterInput(SampleConverter *converter, const ALvoid **src, ALsizei *srcframes, ALvoid *dst, ALsizei dstframes)
 227 {
 228     const ALsizei SrcFrameSize = converter->mNumChannels * converter->mSrcTypeSize;
 229     const ALsizei DstFrameSize = converter->mNumChannels * converter->mDstTypeSize;
 230     const ALsizei increment = converter->mIncrement;
 231     ALsizei pos = 0;
 232     FPUCtl oldMode;
 233
 234     SetMixerFPUMode(&oldMode);
 235     while(pos < dstframes && *srcframes > 0)
 236     {
 237         ALfloat *restrict SrcData = ASSUME_ALIGNED(converter->mSrcSamples, 16);
 238         ALfloat *restrict DstData = ASSUME_ALIGNED(converter->mDstSamples, 16);
 239         ALint prepcount = converter->mSrcPrepCount;
 240         ALsizei DataPosFrac = converter->mFracOffset;
 241         ALuint64 DataSize64;
 242         ALsizei DstSize;
 243         ALint toread;
 244         ALsizei chan;
 245
 246         if(prepcount < 0)
 247         {
 248             /* Negative prepcount means we need to skip that many input samples. */
 249             if(-prepcount >= *srcframes)
 250             {
 251                 converter->mSrcPrepCount = prepcount + *srcframes;
 252                 *srcframes = 0;
 253                 break;
 254             }
 255             *src = (const ALbyte*)*src + SrcFrameSize*-prepcount;
 256             *srcframes += prepcount;
 257             converter->mSrcPrepCount = 0;
 258             continue;
 259         }
 260         toread = mini(*srcframes, BUFFERSIZE-(MAX_POST_SAMPLES+MAX_PRE_SAMPLES));
 261
 262         if(prepcount < MAX_POST_SAMPLES+MAX_PRE_SAMPLES &&
 263            MAX_POST_SAMPLES+MAX_PRE_SAMPLES-prepcount >= toread)
 264         {
 265             /* Not enough input samples to generate an output sample. Store
 266              * what we're given for later.
 267              */
 268             for(chan = 0;chan < converter->mNumChannels;chan++)
 269                 LoadSamples(&converter->Chan[chan].mPrevSamples[prepcount],
 270                     (const ALbyte*)*src + converter->mSrcTypeSize*chan,
 271                     converter->mNumChannels, converter->mSrcType, toread
 272                 );
 273
 274             converter->mSrcPrepCount = prepcount + toread;
 275             *srcframes = 0;
 276             break;
 277         }
 278
 279         DataSize64  = prepcount;
 280         DataSize64 += toread;
 281         DataSize64 -= MAX_POST_SAMPLES+MAX_PRE_SAMPLES;
 282         DataSize64 <<= FRACTIONBITS;
 283         DataSize64 -= DataPosFrac;
 284
 285         /* If we have a full prep, we can generate at least one sample. */
 286         DstSize = (ALsizei)clampu64(DataSize64/increment, 1, BUFFERSIZE);
 287         DstSize = mini(DstSize, dstframes-pos);
 288
 289         for(chan = 0;chan < converter->mNumChannels;chan++)
 290         {
 291             const ALbyte *SrcSamples = (const ALbyte*)*src + converter->mSrcTypeSize*chan;
 292             ALbyte *DstSamples = (ALbyte*)dst + converter->mDstTypeSize*chan;
 293             const ALfloat *ResampledData;
 294             ALsizei SrcDataEnd;
 295
 296             /* Load the previous samples into the source data first, then the
 297              * new samples from the input buffer.
 298              */
 299             memcpy(SrcData, converter->Chan[chan].mPrevSamples,
 300                    prepcount*sizeof(ALfloat));
 301             LoadSamples(SrcData + prepcount, SrcSamples,
 302                 converter->mNumChannels, converter->mSrcType, toread
 303             );
 304
 305             /* Store as many prep samples for next time as possible, given the
 306              * number of output samples being generated.
 307              */
 308             SrcDataEnd = (DataPosFrac + increment*DstSize)>>FRACTIONBITS;
 309             if(SrcDataEnd >= prepcount+toread)
 310                 memset(converter->Chan[chan].mPrevSamples, 0,
 311                        sizeof(converter->Chan[chan].mPrevSamples));
 312             else
 313             {
 314                 size_t len = mini(MAX_PRE_SAMPLES+MAX_POST_SAMPLES, prepcount+toread-SrcDataEnd);
 315                 memcpy(converter->Chan[chan].mPrevSamples, &SrcData[SrcDataEnd],
 316                        len*sizeof(ALfloat));
 317                 memset(converter->Chan[chan].mPrevSamples+len, 0,
 318                        sizeof(converter->Chan[chan].mPrevSamples) - len*sizeof(ALfloat));
 319             }
 320
 321             /* Now resample, and store the result in the output buffer. */
 322             ResampledData = converter->mResample(&converter->mState,
 323                 SrcData+MAX_PRE_SAMPLES, DataPosFrac, increment,
 324                 DstData, DstSize
 325             );
 326
 327             StoreSamples(DstSamples, ResampledData, converter->mNumChannels,
 328                          converter->mDstType, DstSize);
 329         }
 330
 331         /* Update the number of prep samples still available, as well as the
 332          * fractional offset.
 333          */
 334         DataPosFrac += increment*DstSize;
 335         converter->mSrcPrepCount = mini(MAX_PRE_SAMPLES+MAX_POST_SAMPLES,
 336                                         prepcount+toread-(DataPosFrac>>FRACTIONBITS));
 337         converter->mFracOffset = DataPosFrac & FRACTIONMASK;
 338
 339         /* Update the src and dst pointers in case there's still more to do. */
 340         *src = (const ALbyte*)*src + SrcFrameSize*(DataPosFrac>>FRACTIONBITS);
 341         *srcframes -= mini(*srcframes, (DataPosFrac>>FRACTIONBITS));
 342
 343         dst = (ALbyte*)dst + DstFrameSize*DstSize;
 344         pos += DstSize;
 345     }
 346     RestoreFPUMode(&oldMode);
 347
 348     return pos;
 349 }
 350
 351
 352 ChannelConverter *CreateChannelConverter(enum DevFmtType srcType, enum DevFmtChannels srcChans, enum DevFmtChannels dstChans)
 353 {
 354     ChannelConverter *converter;
 355
 356     if(srcChans != dstChans && !((srcChans == DevFmtMono && dstChans == DevFmtStereo) ||
 357                                  (srcChans == DevFmtStereo && dstChans == DevFmtMono)))
 358         return NULL;
 359
 360     converter = al_calloc(DEF_ALIGN, sizeof(*converter));
 361     converter->mSrcType = srcType;
 362     converter->mSrcChans = srcChans;
 363     converter->mDstChans = dstChans;
 364
 365     return converter;
 366 }
 367
 368 void DestroyChannelConverter(ChannelConverter **converter)
 369 {
 370     if(converter)
 371     {
 372         al_free(*converter);
 373         *converter = NULL;
 374     }
 375 }
 376
 377
 378 #define DECL_TEMPLATE(T)                                                       \
 379 static void Mono2Stereo##T(ALfloat *restrict dst, const T *src, ALsizei frames)\
 380 {                                                                              \
 381     ALsizei i;                                                                 \
 382     for(i = 0;i < frames;i++)                                                  \
 383         dst[i*2 + 1] = dst[i*2 + 0] = Sample_##T(src[i]) * 0.707106781187f;    \
 384 }                                                                              \
 385                                                                                \
 386 static void Stereo2Mono##T(ALfloat *restrict dst, const T *src, ALsizei frames)\
 387 {                                                                              \
 388     ALsizei i;                                                                 \
 389     for(i = 0;i < frames;i++)                                                  \
 390         dst[i] = (Sample_##T(src[i*2 + 0])+Sample_##T(src[i*2 + 1])) *         \
 391                  0.707106781187f;                                              \
 392 }
 393
 394 DECL_TEMPLATE(ALbyte)
 395 DECL_TEMPLATE(ALubyte)
 396 DECL_TEMPLATE(ALshort)
 397 DECL_TEMPLATE(ALushort)
 398 DECL_TEMPLATE(ALint)
 399 DECL_TEMPLATE(ALuint)
 400 DECL_TEMPLATE(ALfloat)
 401
 402 #undef DECL_TEMPLATE
 403
 404 void ChannelConverterInput(ChannelConverter *converter, const ALvoid *src, ALfloat *dst, ALsizei frames)
 405 {
 406     if(converter->mSrcChans == converter->mDstChans)
 407     {
 408         LoadSamples(dst, src, 1, converter->mSrcType,
 409                     frames*ChannelsFromDevFmt(converter->mSrcChans, 0));
 410         return;
 411     }
 412
 413     if(converter->mSrcChans == DevFmtStereo && converter->mDstChans == DevFmtMono)
 414     {
 415         switch(converter->mSrcType)
 416         {
 417             case DevFmtByte:
 418                 Stereo2MonoALbyte(dst, src, frames);
 419                 break;
 420             case DevFmtUByte:
 421                 Stereo2MonoALubyte(dst, src, frames);
 422                 break;
 423             case DevFmtShort:
 424                 Stereo2MonoALshort(dst, src, frames);
 425                 break;
 426             case DevFmtUShort:
 427                 Stereo2MonoALushort(dst, src, frames);
 428                 break;
 429             case DevFmtInt:
 430                 Stereo2MonoALint(dst, src, frames);
 431                 break;
 432             case DevFmtUInt:
 433                 Stereo2MonoALuint(dst, src, frames);
 434                 break;
 435             case DevFmtFloat:
 436                 Stereo2MonoALfloat(dst, src, frames);
 437                 break;
 438         }
 439     }
 440     else /*if(converter->mSrcChans == DevFmtMono && converter->mDstChans == DevFmtStereo)*/
 441     {
 442         switch(converter->mSrcType)
 443         {
 444             case DevFmtByte:
 445                 Mono2StereoALbyte(dst, src, frames);
 446                 break;
 447             case DevFmtUByte:
 448                 Mono2StereoALubyte(dst, src, frames);
 449                 break;
 450             case DevFmtShort:
 451                 Mono2StereoALshort(dst, src, frames);
 452                 break;
 453             case DevFmtUShort:
 454                 Mono2StereoALushort(dst, src, frames);
 455                 break;
 456             case DevFmtInt:
 457                 Mono2StereoALint(dst, src, frames);
 458                 break;
 459             case DevFmtUInt:
 460                 Mono2StereoALuint(dst, src, frames);
 461                 break;
 462             case DevFmtFloat:
 463                 Mono2StereoALfloat(dst, src, frames);
 464                 break;
 465         }
 466     }
 467 }