extra/audio/aiff/aiff.factor

   1 ! Copyright (C) 2009 Joe Groff.
   2 ! See http://factorcode.org/license.txt for BSD license.
   3 USING: accessors alien alien.c-types alien.data audio
   4 audio.chunked-file classes.struct combinators
   5 combinators.short-circuit endian io io.binary
   6 io.encodings.binary io.files kernel locals math sequences
   7 audio.loader ;
   8 IN: audio.aiff
   9
  10 CONSTANT: FORM-MAGIC "FORM"
  11 CONSTANT: AIFF-MAGIC "AIFF"
  12 CONSTANT: COMM-MAGIC "COMM"
  13 CONSTANT: SSND-MAGIC "SSND"
  14
  15 STRUCT: aiff-chunk-header
  16     { id char[4] }
  17     { size char[4] } ;
  18
  19 STRUCT: form-chunk
  20     { header aiff-chunk-header }
  21     { form-type char[4] } ;
  22
  23 STRUCT: common-chunk
  24     { header aiff-chunk-header }
  25     { num-channels uchar[2] }
  26     { num-sample-frames uchar[4] }
  27     { sample-size uchar[2] }
  28     { sample-rate uchar[10] } ;
  29
  30 STRUCT: sound-data-chunk
  31     { header aiff-chunk-header }
  32     { offset uchar[4] }
  33     { block-size uchar[4] }
  34     { waveform-data uchar[0] } ;
  35
  36 ! cheesy long-double>integer converter that assumes the long double is a positive integer
  37 : sample-rate>integer ( byte[10] -- sample-rate )
  38     2 cut-slice [ be> ] bi@ swap 16383 - 63 - shift ;
  39
  40 : read-form-chunk ( -- byte-array/f )
  41     form-chunk heap-size ensured-read* ;
  42
  43 : verify-aiff ( chunk -- )
  44     {
  45         [ FORM-MAGIC id= ]
  46         [ form-chunk memory>struct form-type>> 4 memory>byte-array AIFF-MAGIC id= ]
  47     } 1&&
  48     [ invalid-audio-file ] unless ;
  49
  50 :: read-aiff-chunks ( -- comm ssnd )
  51     f :> comm! f :> ssnd!
  52     [ { [ comm ssnd and not ] [ read-chunk ] } 0&& dup ]
  53     [ {
  54         {
  55             [ dup COMM-MAGIC common-chunk check-chunk ]
  56             [ common-chunk memory>struct comm! ]
  57         }
  58         {
  59             [ dup SSND-MAGIC sound-data-chunk check-chunk ]
  60             [ sound-data-chunk memory>struct ssnd! ]
  61         }
  62         [ drop ]
  63     } cond ] while drop
  64     comm ssnd 2dup and [ invalid-audio-file ] unless ;
  65
  66 : (read-aiff) ( -- audio )
  67     read-aiff-chunks
  68     [
  69         [ num-channels>>    2 memory>byte-array be> ]
  70         [ sample-size>>     2 memory>byte-array be> ]
  71         [ sample-rate>>     sample-rate>integer ] tri
  72     ] [
  73         [ header>> size>> 4 memory>byte-array be> 8 - dup ]
  74         [ waveform-data>> >c-ptr ] bi swap memory>byte-array
  75     ] bi*
  76     <audio> convert-data-endian ;
  77
  78 : read-aiff ( filename -- audio )
  79     big-endian [
  80         binary [
  81             read-form-chunk verify-aiff (read-aiff)
  82         ] with-file-reader
  83     ] with-endianness ;
  84
  85 "aif"  [ read-aiff ] register-audio-extension
  86 "aiff" [ read-aiff ] register-audio-extension