Alc/ambdec.c

   1
   2 #include "config.h"
   3
   4 #include "ambdec.h"
   5
   6 #include <stdio.h>
   7 #include <string.h>
   8 #include <ctype.h>
   9
  10 #include "compat.h"
  11
  12
  13 static char *lstrip(char *line)
  14 {
  15     while(isspace(line[0]))
  16         line++;
  17     return line;
  18 }
  19
  20 static char *rstrip(char *line)
  21 {
  22     size_t len = strlen(line);
  23     while(len > 0 && isspace(line[len-1]))
  24         len--;
  25     line[len] = 0;
  26     return line;
  27 }
  28
  29 static int readline(FILE *f, char **output, size_t *maxlen)
  30 {
  31     size_t len = 0;
  32     int c;
  33
  34     while((c=fgetc(f)) != EOF && (c == '\r' || c == '\n'))
  35         ;
  36     if(c == EOF)
  37         return 0;
  38
  39     do {
  40         if(len+1 >= *maxlen)
  41         {
  42             void *temp = NULL;
  43             size_t newmax;
  44
  45             newmax = (*maxlen ? (*maxlen)<<1 : 32);
  46             if(newmax > *maxlen)
  47                 temp = realloc(*output, newmax);
  48             if(!temp)
  49             {
  50                 ERR("Failed to realloc "SZFMT" bytes from "SZFMT"!\n", newmax, *maxlen);
  51                 return 0;
  52             }
  53
  54             *output = temp;
  55             *maxlen = newmax;
  56         }
  57         (*output)[len++] = c;
  58         (*output)[len] = '\0';
  59     } while((c=fgetc(f)) != EOF && c != '\r' && c != '\n');
  60
  61     return 1;
  62 }
  63
  64
  65 /* Custom strtok_r, since we can't rely on it existing. */
  66 static char *my_strtok_r(char *str, const char *delim, char **saveptr)
  67 {
  68     /* Sanity check and update internal pointer. */
  69     if(!saveptr || !delim) return NULL;
  70     if(str) *saveptr = str;
  71     str = *saveptr;
  72
  73     /* Nothing more to do with this string. */
  74     if(!str) return NULL;
  75
  76     /* Find the first non-delimiter character. */
  77     while(*str != '\0' && strchr(delim, *str) != NULL)
  78         str++;
  79     if(*str == '\0')
  80     {
  81         /* End of string. */
  82         *saveptr = NULL;
  83         return NULL;
  84     }
  85
  86     /* Find the next delimiter character. */
  87     *saveptr = strpbrk(str, delim);
  88     if(*saveptr) *((*saveptr)++) = '\0';
  89
  90     return str;
  91 }
  92
  93 static char *read_uint(ALuint *num, const char *line, int base)
  94 {
  95     char *end;
  96     *num = strtoul(line, &end, base);
  97     if(end && *end != '\0')
  98         end = lstrip(end);
  99     return end;
 100 }
 101
 102 static char *read_float(ALfloat *num, const char *line)
 103 {
 104     char *end;
 105 #ifdef HAVE_STRTOF
 106     *num = strtof(line, &end);
 107 #else
 108     *num = strtod(line, &end);
 109 #endif
 110     if(end && *end != '\0')
 111         end = lstrip(end);
 112     return end;
 113 }
 114
 115
 116 char *read_clipped_line(FILE *f, char **buffer, size_t *maxlen)
 117 {
 118     while(readline(f, buffer, maxlen))
 119     {
 120         char *line, *comment;
 121
 122         line = lstrip(*buffer);
 123         comment = strchr(line, '#');
 124         if(comment) *(comment++) = 0;
 125
 126         line = rstrip(line);
 127         if(line[0]) return line;
 128     }
 129     return NULL;
 130 }
 131
 132 static int load_speakers(AmbDecConf *conf, FILE *f, char **buffer, size_t *maxlen, char **saveptr)
 133 {
 134     ALuint cur = 0;
 135     while(cur < conf->NumSpeakers)
 136     {
 137         const char *cmd = my_strtok_r(NULL, " \t", saveptr);
 138         if(!cmd)
 139         {
 140             char *line = read_clipped_line(f, buffer, maxlen);
 141             if(!line)
 142             {
 143                 ERR("Unexpected end of file\n");
 144                 return 0;
 145             }
 146             cmd = my_strtok_r(line, " \t", saveptr);
 147         }
 148
 149         if(strcmp(cmd, "add_spkr") == 0)
 150         {
 151             const char *name = my_strtok_r(NULL, " \t", saveptr);
 152             const char *dist = my_strtok_r(NULL, " \t", saveptr);
 153             const char *az = my_strtok_r(NULL, " \t", saveptr);
 154             const char *elev = my_strtok_r(NULL, " \t", saveptr);
 155             const char *conn = my_strtok_r(NULL, " \t", saveptr);
 156
 157             if(!name) ERR("Name not specified for speaker %u\n", cur+1);
 158             else al_string_copy_cstr(&conf->Speakers[cur].Name, name);
 159             if(!dist) ERR("Distance not specified for speaker %u\n", cur+1);
 160             else read_float(&conf->Speakers[cur].Distance, dist);
 161             if(!az) ERR("Azimuth not specified for speaker %u\n", cur+1);
 162             else read_float(&conf->Speakers[cur].Azimuth, az);
 163             if(!elev) ERR("Elevation not specified for speaker %u\n", cur+1);
 164             else read_float(&conf->Speakers[cur].Elevation, elev);
 165             if(!conn) ERR("Connection not specified for speaker %u\n", cur+1);
 166             else al_string_copy_cstr(&conf->Speakers[cur].Connection, conn);
 167             cur++;
 168         }
 169         else
 170         {
 171             ERR("Unexpected speakers command: %s\n", cmd);
 172             return 0;
 173         }
 174
 175         cmd = my_strtok_r(NULL, " \t", saveptr);
 176         if(cmd)
 177         {
 178             ERR("Unexpected junk on line: %s\n", cmd);
 179             return 0;
 180         }
 181     }
 182
 183     return 1;
 184 }
 185
 186 static int load_matrix(ALfloat *gains, ALfloat (*matrix)[MAX_AMBI_COEFFS], ALuint maxrow, FILE *f, char **buffer, size_t *maxlen, char **saveptr)
 187 {
 188     int gotgains = 0;
 189     ALuint cur = 0;
 190     while(cur < maxrow)
 191     {
 192         const char *cmd = my_strtok_r(NULL, " \t", saveptr);
 193         if(!cmd)
 194         {
 195             char *line = read_clipped_line(f, buffer, maxlen);
 196             if(!line)
 197             {
 198                 ERR("Unexpected end of file\n");
 199                 return 0;
 200             }
 201             cmd = my_strtok_r(line, " \t", saveptr);
 202         }
 203
 204         if(strcmp(cmd, "order_gain") == 0)
 205         {
 206             ALuint curgain = 0;
 207             char *line;
 208             while((line=my_strtok_r(NULL, " \t", saveptr)) != NULL)
 209             {
 210                 ALfloat value;
 211                 line = read_float(&value, line);
 212                 if(line && *line != '\0')
 213                 {
 214                     ERR("Extra junk on gain %u: %s\n", curgain+1, line);
 215                     return 0;
 216                 }
 217                 if(curgain < MAX_AMBI_ORDER+1)
 218                     gains[curgain] = value;
 219                 curgain++;
 220             }
 221             while(curgain < MAX_AMBI_ORDER+1)
 222                 gains[curgain++] = 0.0f;
 223             gotgains = 1;
 224         }
 225         else if(strcmp(cmd, "add_row") == 0)
 226         {
 227             ALuint curidx = 0;
 228             char *line;
 229             while((line=my_strtok_r(NULL, " \t", saveptr)) != NULL)
 230             {
 231                 ALfloat value;
 232                 line = read_float(&value, line);
 233                 if(line && *line != '\0')
 234                 {
 235                     ERR("Extra junk on matrix element %ux%u: %s\n", cur, curidx, line);
 236                     return 0;
 237                 }
 238                 if(curidx < MAX_AMBI_COEFFS)
 239                     matrix[cur][curidx] = value;
 240                 curidx++;
 241             }
 242             while(curidx < MAX_AMBI_COEFFS)
 243                 matrix[cur][curidx++] = 0.0f;
 244             cur++;
 245         }
 246         else
 247         {
 248             ERR("Unexpected speakers command: %s\n", cmd);
 249             return 0;
 250         }
 251
 252         cmd = my_strtok_r(NULL, " \t", saveptr);
 253         if(cmd)
 254         {
 255             ERR("Unexpected junk on line: %s\n", cmd);
 256             return 0;
 257         }
 258     }
 259
 260     if(!gotgains)
 261     {
 262         ERR("Matrix order_gain not specified\n");
 263         return 0;
 264     }
 265
 266     return 1;
 267 }
 268
 269 void ambdec_init(AmbDecConf *conf)
 270 {
 271     ALuint i;
 272
 273     memset(conf, 0, sizeof(*conf));
 274     AL_STRING_INIT(conf->Description);
 275     for(i = 0;i < MAX_OUTPUT_CHANNELS;i++)
 276     {
 277         AL_STRING_INIT(conf->Speakers[i].Name);
 278         AL_STRING_INIT(conf->Speakers[i].Connection);
 279     }
 280 }
 281
 282 void ambdec_deinit(AmbDecConf *conf)
 283 {
 284     ALuint i;
 285
 286     al_string_deinit(&conf->Description);
 287     for(i = 0;i < MAX_OUTPUT_CHANNELS;i++)
 288     {
 289         al_string_deinit(&conf->Speakers[i].Name);
 290         al_string_deinit(&conf->Speakers[i].Connection);
 291     }
 292     memset(conf, 0, sizeof(*conf));
 293 }
 294
 295 int ambdec_load(AmbDecConf *conf, const char *fname)
 296 {
 297     char *buffer = NULL;
 298     size_t maxlen = 0;
 299     FILE *f;
 300
 301     f = al_fopen(fname, "r");
 302     if(!f)
 303     {
 304         ERR("Failed to open: %s\n", fname);
 305         return 0;
 306     }
 307
 308     while(read_clipped_line(f, &buffer, &maxlen))
 309     {
 310         char *line, *saveptr;
 311         char *command;
 312
 313         command = my_strtok_r(buffer, "/ \t", &saveptr);
 314         if(!command)
 315         {
 316             ERR("Malformed line: %s\n", line);
 317             goto fail;
 318         }
 319
 320         if(strcmp(command, "description") == 0)
 321         {
 322             char *value = my_strtok_r(NULL, "", &saveptr);
 323             al_string_copy_cstr(&conf->Description, lstrip(value));
 324         }
 325         else if(strcmp(command, "version") == 0)
 326         {
 327             line = my_strtok_r(NULL, "", &saveptr);
 328             line = read_uint(&conf->Version, line, 10);
 329             if(line && *line != '\0')
 330             {
 331                 ERR("Extra junk after version: %s\n", line);
 332                 goto fail;
 333             }
 334             if(conf->Version != 3)
 335             {
 336                 ERR("Unsupported version: %u\n", conf->Version);
 337                 goto fail;
 338             }
 339         }
 340         else if(strcmp(command, "dec") == 0)
 341         {
 342             const char *dec = my_strtok_r(NULL, "/ \t", &saveptr);
 343             if(strcmp(dec, "chan_mask") == 0)
 344             {
 345                 line = my_strtok_r(NULL, "", &saveptr);
 346                 line = read_uint(&conf->ChanMask, line, 16);
 347                 if(line && *line != '\0')
 348                 {
 349                     ERR("Extra junk after mask: %s\n", line);
 350                     goto fail;
 351                 }
 352             }
 353             else if(strcmp(dec, "freq_bands") == 0)
 354             {
 355                 line = my_strtok_r(NULL, "", &saveptr);
 356                 line = read_uint(&conf->FreqBands, line, 10);
 357                 if(line && *line != '\0')
 358                 {
 359                     ERR("Extra junk after freq_bands: %s\n", line);
 360                     goto fail;
 361                 }
 362                 if(conf->FreqBands != 1 && conf->FreqBands != 2)
 363                 {
 364                     ERR("Invalid freq_bands value: %u\n", conf->FreqBands);
 365                     goto fail;
 366                 }
 367             }
 368             else if(strcmp(dec, "speakers") == 0)
 369             {
 370                 line = my_strtok_r(NULL, "", &saveptr);
 371                 line = read_uint(&conf->NumSpeakers, line, 10);
 372                 if(line && *line != '\0')
 373                 {
 374                     ERR("Extra junk after speakers: %s\n", line);
 375                     goto fail;
 376                 }
 377                 if(conf->NumSpeakers > MAX_OUTPUT_CHANNELS)
 378                 {
 379                     ERR("Unsupported speaker count: %u\n", conf->NumSpeakers);
 380                     goto fail;
 381                 }
 382             }
 383             else if(strcmp(dec, "coeff_scale") == 0)
 384             {
 385                 line = my_strtok_r(NULL, " \t", &saveptr);
 386                 if(strcmp(line, "n3d") == 0)
 387                     conf->CoeffScale = ADS_N3D;
 388                 else if(strcmp(line, "sn3d") == 0)
 389                     conf->CoeffScale = ADS_SN3D;
 390                 else if(strcmp(line, "fuma") == 0)
 391                     conf->CoeffScale = ADS_FuMa;
 392                 else
 393                 {
 394                     ERR("Unsupported coeff scale: %s\n", line);
 395                     goto fail;
 396                 }
 397             }
 398             else
 399             {
 400                 ERR("Unexpected /dec option: %s\n", dec);
 401                 goto fail;
 402             }
 403         }
 404         else if(strcmp(command, "opt") == 0)
 405         {
 406             const char *opt = my_strtok_r(NULL, "/ \t", &saveptr);
 407             if(strcmp(opt, "xover_freq") == 0)
 408             {
 409                 line = my_strtok_r(NULL, "", &saveptr);
 410                 line = read_float(&conf->XOverFreq, line);
 411                 if(line && *line != '\0')
 412                 {
 413                     ERR("Extra junk after xover_freq: %s\n", line);
 414                     goto fail;
 415                 }
 416             }
 417             else if(strcmp(opt, "xover_ratio") == 0)
 418             {
 419                 line = my_strtok_r(NULL, "", &saveptr);
 420                 line = read_float(&conf->XOverRatio, line);
 421                 if(line && *line != '\0')
 422                 {
 423                     ERR("Extra junk after xover_ratio: %s\n", line);
 424                     goto fail;
 425                 }
 426             }
 427             else if(strcmp(opt, "input_scale") == 0 || strcmp(opt, "nfeff_comp") == 0 ||
 428                     strcmp(opt, "delay_comp") == 0 || strcmp(opt, "level_comp") == 0)
 429             {
 430                 /* Unused */
 431                 my_strtok_r(NULL, " \t", &saveptr);
 432             }
 433             else
 434             {
 435                 ERR("Unexpected /opt option: %s\n", opt);
 436                 goto fail;
 437             }
 438         }
 439         else if(strcmp(command, "speakers") == 0)
 440         {
 441             const char *value = my_strtok_r(NULL, "/ \t", &saveptr);
 442             if(strcmp(value, "{") != 0)
 443             {
 444                 ERR("Expected { after speakers command, got %s\n", value);
 445                 goto fail;
 446             }
 447             if(!load_speakers(conf, f, &buffer, &maxlen, &saveptr))
 448                 goto fail;
 449             value = my_strtok_r(NULL, "/ \t", &saveptr);
 450             if(!value)
 451             {
 452                 line = read_clipped_line(f, &buffer, &maxlen);
 453                 if(!line)
 454                 {
 455                     ERR("Unexpected end of file\n");
 456                     goto fail;
 457                 }
 458                 value = my_strtok_r(line, "/ \t", &saveptr);
 459             }
 460             if(strcmp(value, "}") != 0)
 461             {
 462                 ERR("Expected } after speaker definitions, got %s\n", value);
 463                 goto fail;
 464             }
 465         }
 466         else if(strcmp(command, "lfmatrix") == 0 || strcmp(command, "hfmatrix") == 0 ||
 467                 strcmp(command, "matrix") == 0)
 468         {
 469             const char *value = my_strtok_r(NULL, "/ \t", &saveptr);
 470             if(strcmp(value, "{") != 0)
 471             {
 472                 ERR("Expected { after speakers command, got %s\n", value);
 473                 goto fail;
 474             }
 475             if(conf->FreqBands == 1)
 476             {
 477                 if(strcmp(command, "matrix") != 0)
 478                 {
 479                     ERR("Unexpected \"%s\" type for a single-band decoder\n", command);
 480                     goto fail;
 481                 }
 482                 if(!load_matrix(conf->HFOrderGain, conf->HFMatrix, conf->NumSpeakers,
 483                                 f, &buffer, &maxlen, &saveptr))
 484                     goto fail;
 485             }
 486             else
 487             {
 488                 if(strcmp(command, "lfmatrix") == 0)
 489                 {
 490                     if(!load_matrix(conf->LFOrderGain, conf->LFMatrix, conf->NumSpeakers,
 491                                     f, &buffer, &maxlen, &saveptr))
 492                         goto fail;
 493                 }
 494                 else if(strcmp(command, "hfmatrix") == 0)
 495                 {
 496                     if(!load_matrix(conf->HFOrderGain, conf->HFMatrix, conf->NumSpeakers,
 497                                     f, &buffer, &maxlen, &saveptr))
 498                         goto fail;
 499                 }
 500                 else
 501                 {
 502                     ERR("Unexpected \"%s\" type for a dual-band decoder\n", command);
 503                     goto fail;
 504                 }
 505             }
 506             value = my_strtok_r(NULL, "/ \t", &saveptr);
 507             if(!value)
 508             {
 509                 line = read_clipped_line(f, &buffer, &maxlen);
 510                 if(!line)
 511                 {
 512                     ERR("Unexpected end of file\n");
 513                     goto fail;
 514                 }
 515                 value = my_strtok_r(line, "/ \t", &saveptr);
 516             }
 517             if(strcmp(value, "}") != 0)
 518             {
 519                 ERR("Expected } after matrix definitions, got %s\n", value);
 520                 goto fail;
 521             }
 522         }
 523         else if(strcmp(command, "end") == 0)
 524         {
 525             line = my_strtok_r(NULL, "/ \t", &saveptr);
 526             if(line)
 527             {
 528                 ERR("Unexpected junk on end: %s\n", line);
 529                 goto fail;
 530             }
 531
 532             fclose(f);
 533             free(buffer);
 534             return 1;
 535         }
 536         else
 537         {
 538             ERR("Unexpected command: %s\n", command);
 539             goto fail;
 540         }
 541
 542         line = my_strtok_r(NULL, "/ \t", &saveptr);
 543         if(line)
 544         {
 545             ERR("Unexpected junk on line: %s\n", line);
 546             goto fail;
 547         }
 548     }
 549     ERR("Unexpected end of file\n");
 550
 551 fail:
 552     fclose(f);
 553     free(buffer);
 554     return 0;
 555 }