File: player.ml

package info (click to toggle)
ocaml-ffmpeg 1.2.8-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 924 kB
  • sloc: ansic: 6,448; ml: 6,294; makefile: 3
file content (53 lines) | stat: -rw-r--r-- 1,563 bytes parent folder | download | duplicates (2)
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
let () =
  if Array.length Sys.argv < 2 then (
    Printf.printf "usage: %s input_file\n" Sys.argv.(0);
    exit 0);

  Avutil.Log.set_level `Debug;
  Avutil.Log.set_callback print_string;

  let src = Av.open_input Sys.argv.(1) in

  let audio, audio_frame =
    try
      let audio_idx, audio_src, _ = Av.find_best_audio_stream src in

      let audio_dst = Avdevice.open_default_audio_output () in
      let _, audio_stream, _ = List.hd (Av.get_audio_streams audio_dst) in

      (Some (audio_idx, audio_dst, audio_stream), [audio_src])
    with Avutil.Error _ -> (None, [])
  in

  let video, video_frame =
    try
      let video_idx, video_src, _ = Av.find_best_video_stream src in

      let video_dst = Avdevice.open_video_output "xv" in
      let _, video_stream, _ = List.hd (Av.get_video_streams video_dst) in

      (Some (video_idx, video_dst, video_stream), [video_src])
    with Avutil.Error _ -> (None, [])
  in

  let rec f () =
    match (Av.read_input ~audio_frame ~video_frame src, audio, video) with
      | `Audio_frame (i, frame), Some (idx, _, stream), _ when i = idx ->
          Av.write_frame stream frame;
          f ()
      | `Video_frame (i, frame), _, Some (idx, _, stream) when i = idx ->
          Av.write_frame stream frame;
          f ()
      | exception Avutil.Error `Eof -> ()
      | _ -> f ()
  in
  f ();

  Av.close src;

  let () = match audio with Some (_, dst, _) -> Av.close dst | None -> () in

  let () = match video with Some (_, dst, _) -> Av.close dst | None -> () in

  Gc.full_major ();
  Gc.full_major ()