]> git.sur5r.net Git - i3/i3status/blobdiff - src/print_volume.c
OpenBSD: fix volume display on some hardware
[i3/i3status] / src / print_volume.c
index 080dea2a4c01f9312ba8a766c462d5d430ca1983..be6a1d7be2613a5463d85107a2c0d337482903b0 100644 (file)
@@ -1,8 +1,10 @@
-// vim:ts=8:expandtab
+// vim:ts=4:sw=4:expandtab
 #include <time.h>
 #include <string.h>
 #include <stdlib.h>
 #include <stdio.h>
+#include <err.h>
+#include <ctype.h>
 #include <yajl/yajl_gen.h>
 #include <yajl/yajl_version.h>
 
 #ifdef __OpenBSD__
 #include <fcntl.h>
 #include <unistd.h>
-#include <soundcard.h>
+#include <sys/audioio.h>
+#include <sys/ioctl.h>
 #endif
 
 #include "i3status.h"
 #include "queue.h"
 
-#ifdef LINUX
-struct mixer_hdl {
-       char *device;
-       char *mixer;
-       int mixer_idx;
-       snd_mixer_selem_id_t *sid;
-       snd_mixer_t *m;
-       snd_mixer_elem_t *elem;
-       long min;
-       long max;
-
-       TAILQ_ENTRY(mixer_hdl) handles;
-};
-
-TAILQ_HEAD(handles_head, mixer_hdl) cached = TAILQ_HEAD_INITIALIZER(cached);
-
-static void free_hdl(struct mixer_hdl *hdl) {
-       free(hdl->device);
-       free(hdl->mixer);
-       free(hdl);
+static char *apply_volume_format(const char *fmt, char *outwalk, int ivolume) {
+    const char *walk = fmt;
+
+    for (; *walk != '\0'; walk++) {
+        if (*walk != '%') {
+            *(outwalk++) = *walk;
+            continue;
+        }
+        if (BEGINS_WITH(walk + 1, "%")) {
+            outwalk += sprintf(outwalk, "%s", pct_mark);
+            walk += strlen("%");
+        }
+        if (BEGINS_WITH(walk + 1, "volume")) {
+            outwalk += sprintf(outwalk, "%d%s", ivolume, pct_mark);
+            walk += strlen("volume");
+        }
+    }
+    return outwalk;
 }
-#endif
 
-void print_volume(yajl_gen json_gen, char *buffer, const char *fmt, const char *device, const char *mixer, int mixer_idx) {
-        char *outwalk = buffer;
-       int pbval = 1;
+void print_volume(yajl_gen json_gen, char *buffer, const char *fmt, const char *fmt_muted, const char *device, const char *mixer, int mixer_idx) {
+    char *outwalk = buffer;
+    int pbval = 1;
 
-        /* Printing volume only works with ALSA at the moment */
-        if (output_format == O_I3BAR) {
-                char *instance;
-                asprintf(&instance, "%s.%s.%d", device, mixer, mixer_idx);
-                INSTANCE(instance);
-                free(instance);
+    /* Printing volume works with ALSA and PulseAudio at the moment */
+    if (output_format == O_I3BAR) {
+        char *instance;
+        asprintf(&instance, "%s.%s.%d", device, mixer, mixer_idx);
+        INSTANCE(instance);
+        free(instance);
+    }
+
+#ifndef __OpenBSD__
+    /* Try PulseAudio first */
+
+    /* If the device name has the format "pulse[:N]" where N is the
+     * index of the PulseAudio sink then force PulseAudio, optionally
+     * overriding the default sink */
+    if (!strncasecmp(device, "pulse", strlen("pulse"))) {
+        uint32_t sink_idx = device[strlen("pulse")] == ':' ? (uint32_t)atoi(device + strlen("pulse:")) : DEFAULT_SINK_INDEX;
+        const char *sink_name = device[strlen("pulse")] == ':' &&
+                                        !isdigit(device[strlen("pulse:")])
+                                    ? device + strlen("pulse:")
+                                    : NULL;
+        int cvolume = pulse_initialize() ? volume_pulseaudio(sink_idx, sink_name) : 0;
+        int ivolume = DECOMPOSE_VOLUME(cvolume);
+        bool muted = DECOMPOSE_MUTED(cvolume);
+        if (muted) {
+            START_COLOR("color_degraded");
+            pbval = 0;
+        }
+        /* negative result means error, stick to 0 */
+        if (ivolume < 0)
+            ivolume = 0;
+        outwalk = apply_volume_format(muted ? fmt_muted : fmt,
+                                      outwalk,
+                                      ivolume);
+        goto out;
+    } else if (!strcasecmp(device, "default") && pulse_initialize()) {
+        /* no device specified or "default" set */
+        int cvolume = volume_pulseaudio(DEFAULT_SINK_INDEX, NULL);
+        int ivolume = DECOMPOSE_VOLUME(cvolume);
+        bool muted = DECOMPOSE_MUTED(cvolume);
+        if (ivolume >= 0) {
+            if (muted) {
+                START_COLOR("color_degraded");
+                pbval = 0;
+            }
+            outwalk = apply_volume_format(muted ? fmt_muted : fmt,
+                                          outwalk,
+                                          ivolume);
+            goto out;
         }
+        /* negative result means error, fail PulseAudio attempt */
+    }
+/* If some other device was specified or PulseAudio is not detected,
+ * proceed to ALSA / OSS */
+#endif
+
 #ifdef LINUX
-       /* Check if we already opened the mixer and get the handle
-        * from cache if so */
-       bool found = false;
-       int err;
-       struct mixer_hdl *hdl;
-       TAILQ_FOREACH(hdl, &cached, handles) {
-               if (strcmp(hdl->device, device) != 0 ||
-                   strcmp(hdl->mixer, mixer) != 0 ||
-                   hdl->mixer_idx != mixer_idx)
-                       continue;
-               found = true;
-               break;
-       }
-
-       if (!found) {
-               if ((hdl = calloc(sizeof(struct mixer_hdl), 1)) == NULL)
-                       return;
-
-               if ((hdl->device = strdup(device)) == NULL) {
-                       free(hdl);
-                       return;
-               }
-
-               if ((hdl->mixer = strdup(mixer)) == NULL) {
-                       free(hdl->device);
-                       free(hdl);
-                       return;
-               }
-
-               hdl->mixer_idx = mixer_idx;
-               snd_mixer_selem_id_malloc(&(hdl->sid));
-               if (hdl->sid == NULL) {
-                       free_hdl(hdl);
-                       return;
-               }
-
-               if ((err = snd_mixer_open(&(hdl->m), 0)) < 0) {
-                       fprintf(stderr, "i3status: ALSA: Cannot open mixer: %s\n", snd_strerror(err));
-                       free_hdl(hdl);
-                       return;
-               }
-
-               /* Attach this mixer handle to the given device */
-               if ((err = snd_mixer_attach(hdl->m, device)) < 0) {
-                       fprintf(stderr, "i3status: ALSA: Cannot attach mixer to device: %s\n", snd_strerror(err));
-                       snd_mixer_close(hdl->m);
-                       free_hdl(hdl);
-                       return;
-               }
-
-               /* Register this mixer */
-               if ((err = snd_mixer_selem_register(hdl->m, NULL, NULL)) < 0) {
-                       fprintf(stderr, "i3status: ALSA: snd_mixer_selem_register: %s\n", snd_strerror(err));
-                       snd_mixer_close(hdl->m);
-                       free_hdl(hdl);
-                       return;
-               }
-
-               if ((err = snd_mixer_load(hdl->m)) < 0) {
-                       fprintf(stderr, "i3status: ALSA: snd_mixer_load: %s\n", snd_strerror(err));
-                       snd_mixer_close(hdl->m);
-                       free_hdl(hdl);
-                       return;
-               }
-
-               /* Find the given mixer */
-               snd_mixer_selem_id_set_index(hdl->sid, mixer_idx);
-               snd_mixer_selem_id_set_name(hdl->sid, mixer);
-               if (!(hdl->elem = snd_mixer_find_selem(hdl->m, hdl->sid))) {
-                       fprintf(stderr, "i3status: ALSA: Cannot find mixer %s (index %i)\n",
-                               snd_mixer_selem_id_get_name(hdl->sid), snd_mixer_selem_id_get_index(hdl->sid));
-                       snd_mixer_close(hdl->m);
-                       free_hdl(hdl);
-                       return;
-               }
-
-               /* Get the volume range to convert the volume later */
-               snd_mixer_selem_get_playback_volume_range(hdl->elem, &(hdl->min), &(hdl->max));
-               TAILQ_INSERT_TAIL(&cached, hdl, handles);
-       }
-
-       long val;
-       snd_mixer_handle_events (hdl->m);
-       snd_mixer_selem_get_playback_volume (hdl->elem, 0, &val);
-       int avg;
-       if (hdl->max != 100) {
-               float avgf = ((float)val / hdl->max) * 100;
-               avg = (int)avgf;
-               avg = (avgf - avg < 0.5 ? avg : (avg+1));
-       } else avg = (int)val;
-
-       /* Check for mute */
-       if (snd_mixer_selem_has_playback_switch(hdl->elem)) {
-               if ((err = snd_mixer_selem_get_playback_switch(hdl->elem, 0, &pbval)) < 0)
-                       fprintf (stderr, "i3status: ALSA: playback_switch: %s\n", snd_strerror(err));
-               if (!pbval)  {
-                       START_COLOR("color_bad");
-                       avg = 0;
-               }
-       }
-
-       const char *walk = fmt;
-       for (; *walk != '\0'; walk++) {
-               if (*walk != '%') {
-                        *(outwalk++) = *walk;
-                       continue;
-               }
-               if (BEGINS_WITH(walk+1, "volume")) {
-                       outwalk += sprintf(outwalk, "%d%%", avg);
-                       walk += strlen("volume");
-               }
-       }
+    int err;
+    snd_mixer_t *m;
+    snd_mixer_selem_id_t *sid;
+    snd_mixer_elem_t *elem;
+    long min, max, val;
+    int avg;
+
+    if ((err = snd_mixer_open(&m, 0)) < 0) {
+        fprintf(stderr, "i3status: ALSA: Cannot open mixer: %s\n", snd_strerror(err));
+        goto out;
+    }
+
+    /* Attach this mixer handle to the given device */
+    if ((err = snd_mixer_attach(m, device)) < 0) {
+        fprintf(stderr, "i3status: ALSA: Cannot attach mixer to device: %s\n", snd_strerror(err));
+        snd_mixer_close(m);
+        goto out;
+    }
+
+    /* Register this mixer */
+    if ((err = snd_mixer_selem_register(m, NULL, NULL)) < 0) {
+        fprintf(stderr, "i3status: ALSA: snd_mixer_selem_register: %s\n", snd_strerror(err));
+        snd_mixer_close(m);
+        goto out;
+    }
+
+    if ((err = snd_mixer_load(m)) < 0) {
+        fprintf(stderr, "i3status: ALSA: snd_mixer_load: %s\n", snd_strerror(err));
+        snd_mixer_close(m);
+        goto out;
+    }
+
+    snd_mixer_selem_id_malloc(&sid);
+    if (sid == NULL) {
+        snd_mixer_close(m);
+        goto out;
+    }
+
+    /* Find the given mixer */
+    snd_mixer_selem_id_set_index(sid, mixer_idx);
+    snd_mixer_selem_id_set_name(sid, mixer);
+    if (!(elem = snd_mixer_find_selem(m, sid))) {
+        fprintf(stderr, "i3status: ALSA: Cannot find mixer %s (index %i)\n",
+                snd_mixer_selem_id_get_name(sid), snd_mixer_selem_id_get_index(sid));
+        snd_mixer_close(m);
+        snd_mixer_selem_id_free(sid);
+        goto out;
+    }
+
+    /* Get the volume range to convert the volume later */
+    snd_mixer_selem_get_playback_volume_range(elem, &min, &max);
+
+    snd_mixer_handle_events(m);
+    snd_mixer_selem_get_playback_volume(elem, 0, &val);
+    if (max != 100) {
+        float avgf = ((float)val / max) * 100;
+        avg = (int)avgf;
+        avg = (avgf - avg < 0.5 ? avg : (avg + 1));
+    } else
+        avg = (int)val;
+
+    /* Check for mute */
+    if (snd_mixer_selem_has_playback_switch(elem)) {
+        if ((err = snd_mixer_selem_get_playback_switch(elem, 0, &pbval)) < 0)
+            fprintf(stderr, "i3status: ALSA: playback_switch: %s\n", snd_strerror(err));
+        if (!pbval) {
+            START_COLOR("color_degraded");
+            fmt = fmt_muted;
+        }
+    }
+
+    snd_mixer_close(m);
+    snd_mixer_selem_id_free(sid);
+
+    outwalk = apply_volume_format(fmt, outwalk, avg);
+
 #endif
 #if defined(__FreeBSD__) || defined(__OpenBSD__) || defined(__DragonFly__)
-        char mixerpath[] = "/dev/mixer";
-        int mixfd, vol, devmask = 0;
-
-        if ((mixfd = open(mixerpath, O_RDWR)) < 0)
-                return;
-        if (ioctl(mixfd, SOUND_MIXER_READ_DEVMASK, &devmask) == -1)
-                return;
-        if (ioctl(mixfd, MIXER_READ(0),&vol) == -1)
-                return;
-
-        const char *walk = fmt;
-        for (; *walk != '\0'; walk++) {
-                if (*walk != '%') {
-                        *(outwalk++) = *walk;
-                        continue;
-                }
-                if (BEGINS_WITH(walk+1, "volume")) {
-                        outwalk += sprintf(outwalk, "%d%%", vol & 0x7f);
-                        walk += strlen("volume");
-                }
-        }
-        close(mixfd);
+    char *mixerpath;
+    char defaultmixer[] = "/dev/mixer";
+    int mixfd, vol, devmask = 0;
+    pbval = 1;
+
+    if (mixer_idx > 0)
+        asprintf(&mixerpath, "/dev/mixer%d", mixer_idx);
+    else
+        mixerpath = defaultmixer;
+
+    if ((mixfd = open(mixerpath, O_RDWR)) < 0) {
+#if defined(__OpenBSD__)
+        warn("audioio: Cannot open mixer");
+#else
+        warn("OSS: Cannot open mixer");
 #endif
+        goto out;
+    }
+
+    if (mixer_idx > 0)
+        free(mixerpath);
+
+#if defined(__OpenBSD__)
+    int oclass_idx = -1, master_idx = -1, master_mute_idx = -1;
+    int master_next = AUDIO_MIXER_LAST;
+    mixer_devinfo_t devinfo, devinfo2;
+    mixer_ctrl_t vinfo;
+
+    devinfo.index = 0;
+    while (ioctl(mixfd, AUDIO_MIXER_DEVINFO, &devinfo) >= 0) {
+        if (devinfo.type != AUDIO_MIXER_CLASS) {
+            devinfo.index++;
+            continue;
+        }
+        if (strncmp(devinfo.label.name, AudioCoutputs, MAX_AUDIO_DEV_LEN) == 0)
+            oclass_idx = devinfo.index;
+
+        devinfo.index++;
+    }
+
+    devinfo2.index = 0;
+    while (ioctl(mixfd, AUDIO_MIXER_DEVINFO, &devinfo2) >= 0) {
+        if ((devinfo2.type == AUDIO_MIXER_VALUE) && (devinfo2.mixer_class == oclass_idx) && (strncmp(devinfo2.label.name, AudioNmaster, MAX_AUDIO_DEV_LEN) == 0)) {
+            master_idx = devinfo2.index;
+            master_next = devinfo2.next;
+        }
+
+        if ((devinfo2.type == AUDIO_MIXER_ENUM) && (devinfo2.mixer_class == oclass_idx) && (strncmp(devinfo2.label.name, AudioNmute, MAX_AUDIO_DEV_LEN) == 0))
+            if (master_next == devinfo2.index)
+                master_mute_idx = devinfo2.index;
 
-        *outwalk = '\0';
-        OUTPUT_FULL_TEXT(buffer);
+        if (master_next != AUDIO_MIXER_LAST)
+            master_next = devinfo2.next;
+        devinfo2.index++;
+    }
+
+    if (master_idx == -1)
+        goto out;
+
+    devinfo.index = master_idx;
+    if (ioctl(mixfd, AUDIO_MIXER_DEVINFO, &devinfo) == -1)
+        goto out;
+
+    vinfo.dev = master_idx;
+    vinfo.type = AUDIO_MIXER_VALUE;
+    vinfo.un.value.num_channels = devinfo.un.v.num_channels;
+    if (ioctl(mixfd, AUDIO_MIXER_READ, &vinfo) == -1)
+        goto out;
+
+    if (AUDIO_MAX_GAIN != 100) {
+        float avgf = ((float)vinfo.un.value.level[AUDIO_MIXER_LEVEL_MONO] / AUDIO_MAX_GAIN) * 100;
+        vol = (int)avgf;
+        vol = (avgf - vol < 0.5 ? vol : (vol + 1));
+    } else {
+        vol = (int)vinfo.un.value.level[AUDIO_MIXER_LEVEL_MONO];
+    }
+
+    vinfo.dev = master_mute_idx;
+    vinfo.type = AUDIO_MIXER_ENUM;
+    if (ioctl(mixfd, AUDIO_MIXER_READ, &vinfo) == -1)
+        goto out;
+
+    if (master_mute_idx != -1 && vinfo.un.ord) {
+        START_COLOR("color_degraded");
+        fmt = fmt_muted;
+        pbval = 0;
+    }
+
+#else
+    if (ioctl(mixfd, SOUND_MIXER_READ_DEVMASK, &devmask) == -1) {
+        warn("OSS: Cannot read mixer information");
+        goto out;
+    }
+    if (ioctl(mixfd, MIXER_READ(0), &vol) == -1) {
+        warn("OSS: Cannot read mixer information");
+        goto out;
+    }
+
+    if (((vol & 0x7f) == 0) && (((vol >> 8) & 0x7f) == 0)) {
+        START_COLOR("color_degraded");
+        pbval = 0;
+    }
+
+#endif
+    outwalk = apply_volume_format(fmt, outwalk, vol & 0x7f);
+    close(mixfd);
+#endif
 
-        if (!pbval)
-               END_COLOR;
+out:
+    *outwalk = '\0';
+    if (!pbval)
+        END_COLOR;
+    OUTPUT_FULL_TEXT(buffer);
 }