1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98
|
(*
* Copyright 2003-2011 Savonet team
*
* This file is part of Ocaml-speex.
*
* Ocaml-speex is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2 of the License, or
* (at your option) any later version.
*
* Ocaml-speex is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with Ocaml-speex; if not, write to the Free Software
* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
*)
let check p =
try
let _ = Speex.Header.header_of_packet p in
true
with _ -> false
let decoder ~fill:_ os =
let dec_p = ref None in
let decoder = ref None in
let com_p = ref None in
let os = ref os in
let init () =
match !decoder with
| None ->
let dec_p =
match !dec_p with
| None ->
let packet = Ogg.Stream.get_packet !os in
dec_p := Some packet;
packet
| Some p -> p
in
let com_p =
match !com_p with
| None ->
let packet = Ogg.Stream.get_packet !os in
com_p := Some packet;
packet
| Some p -> p
in
let header = Speex.Header.header_of_packet dec_p in
let meta = Speex.Header.comments_of_packet com_p in
let mode = header.Speex.Header.mode in
let dec = Speex.Decoder.init mode in
let sample_freq = header.Speex.Header.rate in
let chans = header.Speex.Header.nb_channels in
Speex.Decoder.set dec Speex.SPEEX_SET_SAMPLING_RATE sample_freq;
decoder := Some (dec, sample_freq, chans, meta);
(dec, sample_freq, chans, meta)
| Some d -> d
in
let info () =
let _, rate, chans, meta = init () in
({ Ogg_decoder.channels = chans; sample_rate = rate }, meta)
in
let decode feed =
let dec, _, chans, _ = init () in
let len = ref 0 in
let feed buf =
let buf = Array.map (Array.map (fun x -> float x /. 32768.)) buf in
len := !len + Array.length buf.(0);
feed buf
in
try
let decode dec os feed =
if chans = 2 then Speex.Decoder.decode_int_feed_stereo dec os feed
else (
let feed x = feed [| x |] in
Speex.Decoder.decode_int_feed dec os feed)
in
decode dec !os feed
with Ogg.Not_enough_data -> if !len = 0 then raise Ogg.Not_enough_data
in
let restart ~fill:_ new_os =
os := new_os;
let d, _, _, _ = init () in
Speex.Decoder.set d Speex.SPEEX_RESET_STATE 0
in
Ogg_decoder.Audio
{
Ogg_decoder.name = "speex";
info;
restart;
decode;
samples_of_granulepos = (fun x -> x);
}
let register () = Hashtbl.add Ogg_decoder.ogg_decoders "speex" (check, decoder)
|