audio/wavcapture.c

   1 #include "hw/hw.h"
   2 #include "monitor/monitor.h"
   3 #include "audio.h"
   4
   5 typedef struct {
   6     FILE *f;
   7     int bytes;
   8     char *path;
   9     int freq;
  10     int bits;
  11     int nchannels;
  12     CaptureVoiceOut *cap;
  13 } WAVState;
  14
  15 /* VICE code: Store number as little endian. */
  16 static void le_store (uint8_t *buf, uint32_t val, int len)
  17 {
  18     int i;
  19     for (i = 0; i < len; i++) {
  20         buf[i] = (uint8_t) (val & 0xff);
  21         val >>= 8;
  22     }
  23 }
  24
  25 static void wav_notify (void *opaque, audcnotification_e cmd)
  26 {
  27     (void) opaque;
  28     (void) cmd;
  29 }
  30
  31 static void wav_destroy (void *opaque)
  32 {
  33     WAVState *wav = opaque;
  34     uint8_t rlen[4];
  35     uint8_t dlen[4];
  36     uint32_t datalen = wav->bytes;
  37     uint32_t rifflen = datalen + 36;
  38     Monitor *mon = cur_mon;
  39
  40     if (wav->f) {
  41         le_store (rlen, rifflen, 4);
  42         le_store (dlen, datalen, 4);
  43
  44         if (fseek (wav->f, 4, SEEK_SET)) {
  45             monitor_printf (mon, "wav_destroy: rlen fseek failed\nReason: %s\n",
  46                             strerror (errno));
  47             goto doclose;
  48         }
  49         if (fwrite (rlen, 4, 1, wav->f) != 1) {
  50             monitor_printf (mon, "wav_destroy: rlen fwrite failed\nReason %s\n",
  51                             strerror (errno));
  52             goto doclose;
  53         }
  54         if (fseek (wav->f, 32, SEEK_CUR)) {
  55             monitor_printf (mon, "wav_destroy: dlen fseek failed\nReason %s\n",
  56                             strerror (errno));
  57             goto doclose;
  58         }
  59         if (fwrite (dlen, 1, 4, wav->f) != 4) {
  60             monitor_printf (mon, "wav_destroy: dlen fwrite failed\nReason %s\n",
  61                             strerror (errno));
  62             goto doclose;
  63         }
  64     doclose:
  65         if (fclose (wav->f)) {
  66             fprintf (stderr, "wav_destroy: fclose failed: %s",
  67                      strerror (errno));
  68         }
  69     }
  70
  71     g_free (wav->path);
  72 }
  73
  74 static void wav_capture (void *opaque, void *buf, int size)
  75 {
  76     WAVState *wav = opaque;
  77
  78     if (fwrite (buf, size, 1, wav->f) != 1) {
  79         monitor_printf (cur_mon, "wav_capture: fwrite error\nReason: %s",
  80                         strerror (errno));
  81     }
  82     wav->bytes += size;
  83 }
  84
  85 static void wav_capture_destroy (void *opaque)
  86 {
  87     WAVState *wav = opaque;
  88
  89     AUD_del_capture (wav->cap, wav);
  90 }
  91
  92 static void wav_capture_info (void *opaque)
  93 {
  94     WAVState *wav = opaque;
  95     char *path = wav->path;
  96
  97     monitor_printf (cur_mon, "Capturing audio(%d,%d,%d) to %s: %d bytes\n",
  98                     wav->freq, wav->bits, wav->nchannels,
  99                     path ? path : "<not available>", wav->bytes);
 100 }
 101
 102 static struct capture_ops wav_capture_ops = {
 103     .destroy = wav_capture_destroy,
 104     .info = wav_capture_info
 105 };
 106
 107 int wav_start_capture (CaptureState *s, const char *path, int freq,
 108                        int bits, int nchannels)
 109 {
 110     Monitor *mon = cur_mon;
 111     WAVState *wav;
 112     uint8_t hdr[] = {
 113         0x52, 0x49, 0x46, 0x46, 0x00, 0x00, 0x00, 0x00, 0x57, 0x41, 0x56,
 114         0x45, 0x66, 0x6d, 0x74, 0x20, 0x10, 0x00, 0x00, 0x00, 0x01, 0x00,
 115         0x02, 0x00, 0x44, 0xac, 0x00, 0x00, 0x10, 0xb1, 0x02, 0x00, 0x04,
 116         0x00, 0x10, 0x00, 0x64, 0x61, 0x74, 0x61, 0x00, 0x00, 0x00, 0x00
 117     };
 118     struct audsettings as;
 119     struct audio_capture_ops ops;
 120     int stereo, bits16, shift;
 121     CaptureVoiceOut *cap;
 122
 123     if (bits != 8 && bits != 16) {
 124         monitor_printf (mon, "incorrect bit count %d, must be 8 or 16\n", bits);
 125         return -1;
 126     }
 127
 128     if (nchannels != 1 && nchannels != 2) {
 129         monitor_printf (mon, "incorrect channel count %d, must be 1 or 2\n",
 130                         nchannels);
 131         return -1;
 132     }
 133
 134     stereo = nchannels == 2;
 135     bits16 = bits == 16;
 136
 137     as.freq = freq;
 138     as.nchannels = 1 << stereo;
 139     as.fmt = bits16 ? AUD_FMT_S16 : AUD_FMT_U8;
 140     as.endianness = 0;
 141
 142     ops.notify = wav_notify;
 143     ops.capture = wav_capture;
 144     ops.destroy = wav_destroy;
 145
 146     wav = g_malloc0 (sizeof (*wav));
 147
 148     shift = bits16 + stereo;
 149     hdr[34] = bits16 ? 0x10 : 0x08;
 150
 151     le_store (hdr + 22, as.nchannels, 2);
 152     le_store (hdr + 24, freq, 4);
 153     le_store (hdr + 28, freq << shift, 4);
 154     le_store (hdr + 32, 1 << shift, 2);
 155
 156     wav->f = fopen (path, "wb");
 157     if (!wav->f) {
 158         monitor_printf (mon, "Failed to open wave file `%s'\nReason: %s\n",
 159                         path, strerror (errno));
 160         g_free (wav);
 161         return -1;
 162     }
 163
 164     wav->path = g_strdup (path);
 165     wav->bits = bits;
 166     wav->nchannels = nchannels;
 167     wav->freq = freq;
 168
 169     if (fwrite (hdr, sizeof (hdr), 1, wav->f) != 1) {
 170         monitor_printf (mon, "Failed to write header\nReason: %s\n",
 171                         strerror (errno));
 172         goto error_free;
 173     }
 174
 175     cap = AUD_add_capture (&as, &ops, wav);
 176     if (!cap) {
 177         monitor_printf (mon, "Failed to add audio capture\n");
 178         goto error_free;
 179     }
 180
 181     wav->cap = cap;
 182     s->opaque = wav;
 183     s->ops = wav_capture_ops;
 184     return 0;
 185
 186 error_free:
 187     g_free (wav->path);
 188     if (fclose (wav->f)) {
 189         monitor_printf (mon, "Failed to close wave file\nReason: %s\n",
 190                         strerror (errno));
 191     }
 192     g_free (wav);
 193     return -1;
 194 }