libaf/af_comp.c

   1 /*
   2  * Copyright (C) 2002 Anders Johansson ajh@atri.curtin.edu.au
   3  *
   4  * This file is part of MPlayer.
   5  *
   6  * MPlayer is free software; you can redistribute it and/or modify
   7  * it under the terms of the GNU General Public License as published by
   8  * the Free Software Foundation; either version 2 of the License, or
   9  * (at your option) any later version.
  10  *
  11  * MPlayer is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  * GNU General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU General Public License along
  17  * with MPlayer; if not, write to the Free Software Foundation, Inc.,
  18  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  19  */
  20
  21 #include <stdio.h>
  22 #include <stdlib.h>
  23 #include <string.h>
  24
  25 #include <inttypes.h>
  26 #include <math.h>
  27 #include <limits.h>
  28
  29 #include "af.h"
  30
  31 // Data for specific instances of this filter
  32 typedef struct af_comp_s
  33 {
  34   int   enable[AF_NCH];         // Enable/disable / channel
  35   float time[AF_NCH];           // Forgetting factor for power estimate
  36   float pow[AF_NCH];            // Estimated power level [dB]
  37   float tresh[AF_NCH];          // Threshold [dB]
  38   int   attack[AF_NCH];         // Attack time [ms]
  39   int   release[AF_NCH];        // Release time [ms]
  40   float ratio[AF_NCH];          // Compression ratio
  41 }af_comp_t;
  42
  43 // Initialization and runtime control
  44 static int control(struct af_instance_s* af, int cmd, void* arg)
  45 {
  46   af_comp_t* s   = (af_comp_t*)af->setup;
  47   int i;
  48
  49   switch(cmd){
  50   case AF_CONTROL_REINIT:
  51     // Sanity check
  52     if(!arg) return AF_ERROR;
  53
  54     af->data->rate   = ((af_data_t*)arg)->rate;
  55     af->data->nch    = ((af_data_t*)arg)->nch;
  56     af->data->format = AF_FORMAT_FLOAT_NE;
  57     af->data->bps    = 4;
  58
  59     // Time constant set to 0.1s
  60     //    s->alpha = (1.0/0.2)/(2.0*M_PI*(float)((af_data_t*)arg)->rate);
  61     return af_test_output(af,(af_data_t*)arg);
  62   case AF_CONTROL_COMMAND_LINE:{
  63 /*     float v=-10.0; */
  64 /*     float vol[AF_NCH]; */
  65 /*     float s=0.0; */
  66 /*     float clipp[AF_NCH]; */
  67 /*     int i; */
  68 /*     sscanf((char*)arg,"%f:%f", &v, &s); */
  69 /*     for(i=0;i<AF_NCH;i++){ */
  70 /*       vol[i]=v; */
  71 /*       clipp[i]=s; */
  72 /*     } */
  73 /*     if(AF_OK != control(af,AF_CONTROL_VOLUME_SOFTCLIP | AF_CONTROL_SET, clipp)) */
  74 /*       return AF_ERROR; */
  75 /*     return control(af,AF_CONTROL_VOLUME_LEVEL | AF_CONTROL_SET, vol); */
  76   }
  77   case AF_CONTROL_COMP_ON_OFF | AF_CONTROL_SET:
  78     memcpy(s->enable,(int*)arg,AF_NCH*sizeof(int));
  79     return AF_OK;
  80   case AF_CONTROL_COMP_ON_OFF | AF_CONTROL_GET:
  81     memcpy((int*)arg,s->enable,AF_NCH*sizeof(int));
  82     return AF_OK;
  83   case AF_CONTROL_COMP_THRESH | AF_CONTROL_SET:
  84     return af_from_dB(AF_NCH,(float*)arg,s->tresh,20.0,-60.0,-1.0);
  85   case AF_CONTROL_COMP_THRESH | AF_CONTROL_GET:
  86     return af_to_dB(AF_NCH,s->tresh,(float*)arg,10.0);
  87   case AF_CONTROL_COMP_ATTACK | AF_CONTROL_SET:
  88     return af_from_ms(AF_NCH,(float*)arg,s->attack,af->data->rate,500.0,0.1);
  89   case AF_CONTROL_COMP_ATTACK | AF_CONTROL_GET:
  90     return af_to_ms(AF_NCH,s->attack,(float*)arg,af->data->rate);
  91   case AF_CONTROL_COMP_RELEASE | AF_CONTROL_SET:
  92     return af_from_ms(AF_NCH,(float*)arg,s->release,af->data->rate,3000.0,10.0);
  93   case AF_CONTROL_COMP_RELEASE | AF_CONTROL_GET:
  94     return af_to_ms(AF_NCH,s->release,(float*)arg,af->data->rate);
  95   case AF_CONTROL_COMP_RATIO | AF_CONTROL_SET:
  96     for(i=0;i<AF_NCH;i++)
  97       s->ratio[i] = clamp(((float*)arg)[i],1.0,10.0);
  98     return AF_OK;
  99   case AF_CONTROL_COMP_RATIO | AF_CONTROL_GET:
 100     for(i=0;i<AF_NCH;i++)
 101       ((float*)arg)[i] = s->ratio[i];
 102     return AF_OK;
 103   }
 104   return AF_UNKNOWN;
 105 }
 106
 107 // Deallocate memory
 108 static void uninit(struct af_instance_s* af)
 109 {
 110   if(af->data)
 111     free(af->data);
 112   if(af->setup)
 113     free(af->setup);
 114 }
 115
 116 // Filter data through filter
 117 static af_data_t* play(struct af_instance_s* af, af_data_t* data)
 118 {
 119   af_data_t*    c   = data;                     // Current working data
 120   af_comp_t*    s   = (af_comp_t*)af->setup;    // Setup for this instance
 121   float*        a   = (float*)c->audio;         // Audio data
 122   int           len = c->len/4;                 // Number of samples
 123   int           ch  = 0;                        // Channel counter
 124   register int  nch = c->nch;                   // Number of channels
 125   register int  i   = 0;
 126
 127   // Compress/expand
 128   for(ch = 0; ch < nch ; ch++){
 129     if(s->enable[ch]){
 130       float     t   = 1.0 - s->time[ch];
 131       for(i=ch;i<len;i+=nch){
 132         register float x        = a[i];
 133         register float pow      = x*x;
 134         s->pow[ch] = t*s->pow[ch] +
 135           pow*s->time[ch]; // LP filter
 136         if(pow < s->pow[ch]){
 137           ;
 138         }
 139         else{
 140           ;
 141         }
 142         a[i] = x;
 143       }
 144     }
 145   }
 146   return c;
 147 }
 148
 149 // Allocate memory and set function pointers
 150 static int af_open(af_instance_t* af){
 151   af->control=control;
 152   af->uninit=uninit;
 153   af->play=play;
 154   af->mul=1;
 155   af->data=calloc(1,sizeof(af_data_t));
 156   af->setup=calloc(1,sizeof(af_comp_t));
 157   if(af->data == NULL || af->setup == NULL)
 158     return AF_ERROR;
 159   return AF_OK;
 160 }
 161
 162 // Description of this filter
 163 af_info_t af_info_comp = {
 164     "Compressor/expander audio filter",
 165     "comp",
 166     "Anders",
 167     "",
 168     AF_FLAGS_NOT_REENTRANT,
 169     af_open
 170 };