1 #include <stdio.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 39 if (wav->f) { 40 le_store (rlen, rifflen, 4); 41 le_store (dlen, datalen, 4); 42 43 fseek (wav->f, 4, SEEK_SET); 44 fwrite(rlen, 4, 1, wav->f); 45 46 fseek (wav->f, 32, SEEK_CUR); 47 fwrite(dlen, 4, 1, wav->f); 48 fclose (wav->f); 49 } 50 51 g_free (wav->path); 52 } 53 54 static void wav_capture (void *opaque, void *buf, int size) 55 { 56 WAVState *wav = opaque; 57 58 fwrite(buf, size, 1, wav->f); 59 wav->bytes += size; 60 } 61 62 static void wav_capture_destroy (void *opaque) 63 { 64 WAVState *wav = opaque; 65 66 AUD_del_capture (wav->cap, wav); 67 } 68 69 static void wav_capture_info (void *opaque) 70 { 71 WAVState *wav = opaque; 72 char *path = wav->path; 73 74 monitor_printf(cur_mon, "Capturing audio(%d,%d,%d) to %s: %d bytes\n", 75 wav->freq, wav->bits, wav->nchannels, 76 path ? path : "<not available>", wav->bytes); 77 } 78 79 static struct capture_ops wav_capture_ops = { 80 .destroy = wav_capture_destroy, 81 .info = wav_capture_info 82 }; 83 84 int wav_start_capture (CaptureState *s, const char *path, int freq, 85 int bits, int nchannels) 86 { 87 Monitor *mon = cur_mon; 88 WAVState *wav; 89 uint8_t hdr[] = { 90 0x52, 0x49, 0x46, 0x46, 0x00, 0x00, 0x00, 0x00, 0x57, 0x41, 0x56, 91 0x45, 0x66, 0x6d, 0x74, 0x20, 0x10, 0x00, 0x00, 0x00, 0x01, 0x00, 92 0x02, 0x00, 0x44, 0xac, 0x00, 0x00, 0x10, 0xb1, 0x02, 0x00, 0x04, 93 0x00, 0x10, 0x00, 0x64, 0x61, 0x74, 0x61, 0x00, 0x00, 0x00, 0x00 94 }; 95 struct audsettings as; 96 struct audio_capture_ops ops; 97 int stereo, bits16, shift; 98 CaptureVoiceOut *cap; 99 100 if (bits != 8 && bits != 16) { 101 monitor_printf(mon, "incorrect bit count %d, must be 8 or 16\n", bits); 102 return -1; 103 } 104 105 if (nchannels != 1 && nchannels != 2) { 106 monitor_printf(mon, "incorrect channel count %d, must be 1 or 2\n", 107 nchannels); 108 return -1; 109 } 110 111 stereo = nchannels == 2; 112 bits16 = bits == 16; 113 114 as.freq = freq; 115 as.nchannels = 1 << stereo; 116 as.fmt = bits16 ? AUD_FMT_S16 : AUD_FMT_U8; 117 as.endianness = 0; 118 119 ops.notify = wav_notify; 120 ops.capture = wav_capture; 121 ops.destroy = wav_destroy; 122 123 wav = g_malloc0 (sizeof (*wav)); 124 125 shift = bits16 + stereo; 126 hdr[34] = bits16 ? 0x10 : 0x08; 127 128 le_store (hdr + 22, as.nchannels, 2); 129 le_store (hdr + 24, freq, 4); 130 le_store (hdr + 28, freq << shift, 4); 131 le_store (hdr + 32, 1 << shift, 2); 132 133 wav->f = fopen (path, "wb"); 134 if (!wav->f) { 135 monitor_printf(mon, "Failed to open wave file `%s'\nReason: %s\n", 136 path, strerror (errno)); 137 g_free (wav); 138 return -1; 139 } 140 141 wav->path = g_strdup (path); 142 wav->bits = bits; 143 wav->nchannels = nchannels; 144 wav->freq = freq; 145 146 fwrite(hdr, sizeof(hdr), 1, wav->f); 147 148 cap = AUD_add_capture (&as, &ops, wav); 149 if (!cap) { 150 monitor_printf(mon, "Failed to add audio capture\n"); 151 g_free (wav->path); 152 fclose (wav->f); 153 g_free (wav); 154 return -1; 155 } 156 157 wav->cap = cap; 158 s->opaque = wav; 159 s->ops = wav_capture_ops; 160 return 0; 161 } 162