Add media duration to lpms_get_codec_info for GetCodecInfo

cmd/test/get_fps_dur.go

@@ -0,0 +1,32 @@
+package main
+
+import (
+	"fmt"
+	"os"
+	"text/tabwriter"
+	"time"
+
+	"github.com/livepeer/lpms/ffmpeg"
+)
+
+func main() {
+	p := "/opt/livepeer/test-videos"
+	items, _ := os.ReadDir(p)
+	w := tabwriter.NewWriter(os.Stdout, 1, 1, 1, ' ', tabwriter.Debug)
+
+	fmt.Fprintln(w, "File\tTook (ms)\tFPS\tDur\tACodec")
+	for _, item := range items {
+		vid_path := p + "/" + item.Name()
+		start := time.Now()
+		_, mfi, err := ffmpeg.GetCodecInfo(vid_path)
+		if err == nil {
+			took := time.Since(start).Milliseconds()
+			line := fmt.Sprintf("%v\t%v\t%v\t%v\t%v", item.Name(), took, mfi.FPS, mfi.Dur, mfi.Acodec)
+			fmt.Fprintln(w, line)
+		} else {
+			fmt.Printf("error getting codec info: %w\n", err)
+		}
+	}
+
+	w.Flush()
+}

ffmpeg/extras.c

@@ -130,6 +130,28 @@ int lpms_rtmp2hls(char *listen, char *outf, char *ts_tmpl, char* seg_time, char
   return ret == AVERROR_EOF ? 0 : ret;
 }
 
+double calculate_duration(AVFormatContext *ic, int astream) {
+  AVPacket pkt;
+  av_init_packet(&pkt);
+  double duration = 0;
+  int64_t last_pts = AV_NOPTS_VALUE;
+  while (av_read_frame(ic, &pkt) >= 0) {
+    if (pkt.stream_index == astream) {
+      if (pkt.pts != AV_NOPTS_VALUE) {
+        if (last_pts != AV_NOPTS_VALUE) {
+          // Calculate the difference between the current and last PTS
+          int64_t pts_diff = pkt.pts - last_pts;
+          // Convert the PTS difference to seconds and add to duration
+          duration += pts_diff * av_q2d(ic->streams[astream]->time_base);
+        }
+        last_pts = pkt.pts;
+      }
+      av_packet_unref(&pkt);
+    }
+  }
+  return duration;
+}
+
 #define GET_CODEC_INTERNAL_ERROR -1
 #define GET_CODEC_OK 0
 #define GET_CODEC_NEEDS_BYPASS 1
@@ -149,6 +171,9 @@ int lpms_get_codec_info(char *fname, pcodec_info out)
 #define MIN(x, y) (((x) < (y)) ? (x) : (y))
   AVFormatContext *ic = NULL;
   AVCodec *ac, *vc;
+
+  int64_t last_pts = AV_NOPTS_VALUE;
+
   int ret = GET_CODEC_OK, vstream = 0, astream = 0;
 
   ret = avformat_open_input(&ic, fname, NULL, NULL);
@@ -163,6 +188,13 @@ int lpms_get_codec_info(char *fname, pcodec_info out)
   if(!audio_present && !video_present) {
     // instead of returning -1
     ret = GET_CODEC_STREAMS_MISSING;
+  } else if (video_present) {
+    out->dur = ic->duration / AV_TIME_BASE;
+  } else if (audio_present && !video_present) {
+    out->dur = calculate_duration(ic, astream);
+    if (out->dur == 0) {
+      out->dur = ic->duration / AV_TIME_BASE;
+    }
   }
   // Return
   if (video_present && vc->name) {
@@ -176,6 +208,7 @@ int lpms_get_codec_info(char *fname, pcodec_info out)
       }
       out->width  = ic->streams[vstream]->codecpar->width;
       out->height = ic->streams[vstream]->codecpar->height;
+      out->fps = av_q2d(ic->streams[vstream]->avg_frame_rate);
   } else {
       // Indicate failure to extract video codec from given container
       out->video_codec[0] = 0;
@@ -186,6 +219,7 @@ int lpms_get_codec_info(char *fname, pcodec_info out)
       // Indicate failure to extract audio codec from given container
       out->audio_codec[0] = 0;
   }
+
 #undef MIN
 close_format_context:
   if (ic) avformat_close_input(&ic);

ffmpeg/extras.h

@@ -7,6 +7,8 @@ typedef struct s_codec_info {
   int    pixel_format;
   int    width;
   int    height;
+  double fps;
+  double dur;
 } codec_info, *pcodec_info;
 
 int lpms_rtmp2hls(char *listen, char *outf, char *ts_tmpl, char *seg_time, char *seg_start);

ffmpeg/ffmpeg.go

@@ -245,6 +245,8 @@ type MediaFormatInfo struct {
 	Acodec, Vcodec string
 	PixFormat      PixelFormat
 	Width, Height  int
+	FPS            float32
+	Dur            float32
 }
 
 func (f *MediaFormatInfo) ScaledHeight(width int) int {
@@ -277,6 +279,9 @@ func GetCodecInfo(fname string) (CodecStatus, MediaFormatInfo, error) {
 	format.PixFormat = PixelFormat{int(params_c.pixel_format)}
 	format.Width = int(params_c.width)
 	format.Height = int(params_c.height)
+	format.FPS = float32(params_c.fps)
+	format.Dur = float32(params_c.dur)
+
 	return status, format, nil
 }