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