osd-alsa.c

   1 /*
   2  *      A Simple ALSA Volume Control via OSD
   3  *
   4  *      (c) 2012 Martin Mares <mj@ucw.cz>
   5  */
   6
   7 #undef DEBUG
   8
   9 #include <stdio.h>
  10 #include <stdlib.h>
  11 #include <string.h>
  12 #include <getopt.h>
  13 #include <alsa/asoundlib.h>
  14
  15 #include "util.h"
  16 #include "osd.h"
  17
  18 static char *alsa_device = "default";
  19 static char *mixer_control = "Master";
  20 static int adjust_by;
  21 static int want_mute = -1;
  22
  23 static snd_mixer_t *mixer;
  24 static snd_mixer_elem_t *elem;
  25
  26 static void init_mixer(void)
  27 {
  28   int err;
  29
  30   if (err = snd_mixer_open(&mixer, 0))
  31     die("snd_mixer_open failed: error %d", err);
  32
  33   if (err = snd_mixer_attach(mixer, alsa_device))
  34     die("snd_mixer_attach failed: error %d", err);
  35
  36   if (err = snd_mixer_selem_register(mixer, NULL, NULL))
  37     die("snd_mixer_selem_register failed: error %d", err);
  38
  39   if (err = snd_mixer_load(mixer))
  40     die("snd_mixer_load: error %d", err);
  41
  42   for (elem = snd_mixer_first_elem(mixer); elem; elem = snd_mixer_elem_next(elem))
  43     {
  44       const char *name = snd_mixer_selem_get_name(elem);
  45       int index = snd_mixer_selem_get_index(elem);
  46       if (!strcmp(name, mixer_control) || index)
  47         {
  48           if (snd_mixer_elem_get_type(elem) != SND_MIXER_ELEM_SIMPLE)
  49             die("Unable to handle non-simple mixer controls");
  50           if (!snd_mixer_selem_is_active(elem))
  51             die("Selected mixer control is not active");
  52           DBG("Found mixer control %s[%d]\n", name, index);
  53           return;
  54         }
  55     }
  56
  57   die("Unable to find mixer control %s", mixer_control);
  58 }
  59
  60 static int get_mute(void)
  61 {
  62   int mute_on = 0, mute_off = 0;
  63
  64   if (snd_mixer_selem_has_playback_switch(elem))
  65     {
  66       for (snd_mixer_selem_channel_id_t ch=0; ch < SND_MIXER_SCHN_LAST; ch++)
  67         {
  68           int val;
  69           if (!snd_mixer_selem_get_playback_switch(elem, ch, &val))
  70             {
  71               if (val)
  72                 mute_off++;
  73               else
  74                 mute_on++;
  75             }
  76         }
  77     }
  78   DBG("Mute: on=%d off=%d\n", mute_on, mute_off);
  79   return !!mute_on;
  80 }
  81
  82 static long get_volume(long *pmin, long *pmax)
  83 {
  84   long min, max, curr=0;
  85   if (!snd_mixer_selem_has_playback_volume(elem) ||
  86       snd_mixer_selem_get_playback_volume_range(elem, &min, &max))
  87     {
  88       *pmin = *pmax = 0;
  89       return 0;
  90     }
  91
  92   for (snd_mixer_selem_channel_id_t ch=0; ch < SND_MIXER_SCHN_LAST; ch++)
  93     {
  94       long val;
  95       if (!snd_mixer_selem_get_playback_volume(elem, ch, &val))
  96         {
  97           if (val > curr)
  98             curr = val;
  99         }
 100     }
 101
 102   DBG("Volume: min=%ld max=%ld curr=%ld\n", min, max, curr);
 103   *pmin = min;
 104   *pmax = max;
 105   return curr;
 106 }
 107
 108 static int vol_to_perc(long curr, long min, long max)
 109 {
 110   return (100LL * (curr-min) + (max-min)/2) / (max-min);
 111 }
 112
 113 static long perc_to_vol(int perc, long min, long max)
 114 {
 115   return ((long long) perc * (max-min) + 50) / 100;
 116 }
 117
 118 static void show_mixer(void)
 119 {
 120   long min, max;
 121   long curr = get_volume(&min, &max);
 122   int muted = get_mute();
 123
 124   struct osd_msg *msg = osd_new_msg();
 125   osd_add_line(msg, "min-duration", "0");
 126   char buf[256];
 127   snprintf(buf, sizeof(buf), "%s volume", mixer_control);
 128   osd_add_line(msg, NULL, buf);
 129   if (muted)
 130     osd_add_line(msg, NULL, "[mute]");
 131   else if (min < max)
 132     {
 133       snprintf(buf, sizeof(buf), "%d", vol_to_perc(curr, min, max));
 134       osd_add_line(msg, "slider", buf);
 135     }
 136   osd_send(msg);
 137 }
 138
 139
 140 static void set_mute(void)
 141 {
 142   if (want_mute < 0)
 143     return;
 144
 145   if (want_mute == 2)
 146     want_mute = !get_mute();
 147
 148   if (snd_mixer_selem_has_playback_switch(elem))
 149     {
 150       for (snd_mixer_selem_channel_id_t ch=0; ch < SND_MIXER_SCHN_LAST; ch++)
 151         snd_mixer_selem_set_playback_switch(elem, ch, !want_mute);
 152     }
 153 }
 154
 155 static void set_volume(void)
 156 {
 157   if (!adjust_by)
 158     return;
 159
 160   long min, max;
 161   long curr = get_volume(&min, &max);
 162   int perc = vol_to_perc(curr, min, max);
 163
 164   DBG("Volume: have %d %ld\n", perc, curr);
 165   perc += adjust_by;
 166   if (perc < 0)
 167     perc = 0;
 168   if (perc > 100)
 169     perc = 100;
 170   curr = perc_to_vol(perc, min, max);
 171   curr = min + (((long long) perc * (max-min) + 50) / 100);
 172   DBG("Volume: want %d %ld\n", perc, curr);
 173
 174   for (snd_mixer_selem_channel_id_t ch=0; ch < SND_MIXER_SCHN_LAST; ch++)
 175     snd_mixer_selem_set_playback_volume(elem, ch, curr);
 176 }
 177
 178 static void NONRET
 179 usage(void)
 180 {
 181   fprintf(stderr, "\
 182 Usage: osd-alsa <options>\n\
 183 \n\
 184 Options:\n\
 185 -a, --adjust=<percent>  Adjust the control by a given amount\n\
 186 -D, --device=<name>     ALSA device (default: `default')\n\
 187 -m, --mixer=<name>      Name of mixer control (default: `Master')\n\
 188 -0, --mute              Mute the control\n\
 189 -t, --toggle            Mute/unmute the control\n\
 190 -1, --unmute            Unmute the control\n\
 191 ");
 192   exit(1);
 193 }
 194
 195 static const char short_opts[] = "01a:c:D:t";
 196
 197 static const struct option long_opts[] = {
 198   { "adjust",           required_argument,      NULL,   'a' },
 199   { "device",           required_argument,      NULL,   'D' },
 200   { "mixer",            required_argument,      NULL,   'm' },
 201   { "mute",             no_argument,            NULL,   '0' },
 202   { "toggle",           no_argument,            NULL,   't' },
 203   { "unmute",           no_argument,            NULL,   '1' },
 204   { NULL,               0,                      NULL,   0   },
 205 };
 206
 207 int main(int argc, char **argv)
 208 {
 209   int opt;
 210   while ((opt = getopt_long(argc, argv, short_opts, long_opts, NULL)) >= 0)
 211     switch (opt)
 212       {
 213       case '0':
 214         want_mute = 1;
 215         break;
 216       case '1':
 217         want_mute = 0;
 218         break;
 219       case 'a':
 220         adjust_by = atoi(optarg);
 221         break;
 222       case 'D':
 223         alsa_device = optarg;
 224         break;
 225       case 'm':
 226         mixer_control = optarg;
 227         break;
 228       case 't':
 229         want_mute = 2;
 230         break;
 231       default:
 232         usage();
 233       }
 234   if (optind < argc)
 235     usage();
 236
 237   init_mixer();
 238   osd_init();
 239   set_mute();
 240   set_volume();
 241   show_mixer();
 242   return 0;
 243 }