Added mapping of streams

Author: djthorpe

cmd/transcode/main.go

@@ -118,7 +118,9 @@ func main() {
 		fmt.Fprintln(os.Stderr, err)
 		os.Exit(-2)
 	}
-	fmt.Println(media_map)
+
+	// Print the map
+	media_map.PrintMap(os.Stdout)
 
 	// For the output, we can output to a device, a URL, or a file
 	var out Media

media.go

@@ -107,6 +107,9 @@ type Map interface {
 
 	// Return streams which are mapped for decoding
 	Streams() []Stream
+
+	// Print a summary of the mapping
+	PrintMap(w io.Writer)
 }
 
 // Media is a source or destination of media

pkg/media/map.go

@@ -2,6 +2,8 @@ package media
 
 import (
 	"context"
+	"fmt"
+	"io"
 
 	// Packages
 	multierror "github.com/hashicorp/go-multierror"
@@ -172,9 +174,35 @@ func (m *decodemap) Demux(ctx context.Context, p Packet, fn DemuxFn) error {
 	}
 }
 
+// PrintMap will print out a summary of the mapping
+func (m *decodemap) PrintMap(w io.Writer) {
+	for id := range m.context {
+		stream := m.input.streams[id]
+		fmt.Fprintf(w, "Stream %2d (%s): %s\n", id, toMediaType(stream.Flags()), stream)
+	}
+}
+
 ////////////////////////////////////////////////////////////////////////////////
 // PRIVATE METHODS
 
+// Return media type (audio, video, subtitle, etc)
+func toMediaType(flag MediaFlag) string {
+	switch {
+	case flag.Is(MEDIA_FLAG_AUDIO):
+		return "audio"
+	case flag.Is(MEDIA_FLAG_VIDEO):
+		return "video"
+	case flag.Is(MEDIA_FLAG_SUBTITLE):
+		return "subtitle"
+	case flag.Is(MEDIA_FLAG_DATA):
+		return "data"
+	case flag.Is(MEDIA_FLAG_ATTACHMENT):
+		return "attachment"
+	default:
+		return "other"
+	}
+}
+
 // Return streams of a given type for input media
 func streamsByType(input *input, media_type MediaFlag) map[int]*mapentry {
 	if input == nil || input.ctx == nil {

pkg/media/stream.go

@@ -65,10 +65,22 @@ func (stream *stream) Flags() MediaFlag {
 		return flags
 	}
 
-	// TODO: Add codec flags
-	//if stream.ctx.CodecPar()codec != nil {
-	//	flags |= s.codec.Flags()
-	//}
+	// Add codec parameter flags
+	if stream.ctx != nil {
+		t := stream.ctx.CodecPar().CodecType()
+		switch t {
+		case ffmpeg.AVMEDIA_TYPE_AUDIO:
+			flags |= MEDIA_FLAG_AUDIO
+		case ffmpeg.AVMEDIA_TYPE_VIDEO:
+			flags |= MEDIA_FLAG_VIDEO
+		case ffmpeg.AVMEDIA_TYPE_SUBTITLE:
+			flags |= MEDIA_FLAG_SUBTITLE
+		case ffmpeg.AVMEDIA_TYPE_DATA:
+			flags |= MEDIA_FLAG_DATA
+		case ffmpeg.AVMEDIA_TYPE_ATTACHMENT:
+			flags |= MEDIA_FLAG_ATTACHMENT
+		}
+	}
 
 	// Remove encoder/decoder flags
 	flags &^= (MEDIA_FLAG_ENCODER | MEDIA_FLAG_DECODER)

sys/ffmpeg51/avformat.go

@@ -43,7 +43,7 @@ const (
 	// Needs '%d' in filename.
 	AVFMT_NEEDNUMBER AVFormat = C.AVFMT_NEEDNUMBER
 	// The muxer/demuxer is experimental and should be used with caution
-	//AVFMT_EXPERIMENTAL AVFormat = C.AVFMT_EXPERIMENTAL
+	AVFMT_EXPERIMENTAL AVFormat = C.AVFMT_EXPERIMENTAL
 	// Show format stream IDs numbers.
 	AVFMT_SHOWIDS AVFormat = C.AVFMT_SHOW_IDS
 	// Format wants global header.
@@ -116,14 +116,14 @@ const (
 	AV_DISPOSITION_CLEAN_EFFECTS    AVDisposition = C.AV_DISPOSITION_CLEAN_EFFECTS
 	AV_DISPOSITION_ATTACHED_PIC     AVDisposition = C.AV_DISPOSITION_ATTACHED_PIC
 	AV_DISPOSITION_TIMED_THUMBNAILS AVDisposition = C.AV_DISPOSITION_TIMED_THUMBNAILS
-	//AV_DISPOSITION_NON_DIEGETIC     AVDisposition = C.AV_DISPOSITION_NON_DIEGETIC
-	AV_DISPOSITION_CAPTIONS     AVDisposition = C.AV_DISPOSITION_CAPTIONS
-	AV_DISPOSITION_DESCRIPTIONS AVDisposition = C.AV_DISPOSITION_DESCRIPTIONS
-	AV_DISPOSITION_METADATA     AVDisposition = C.AV_DISPOSITION_METADATA
-	AV_DISPOSITION_DEPENDENT    AVDisposition = C.AV_DISPOSITION_DEPENDENT
-	AV_DISPOSITION_STILL_IMAGE  AVDisposition = C.AV_DISPOSITION_STILL_IMAGE
-	AV_DISPOSITION_NONE         AVDisposition = 0
-	AV_DISPOSITION_MAX                        = AV_DISPOSITION_STILL_IMAGE
+	AV_DISPOSITION_NON_DIEGETIC     AVDisposition = C.AV_DISPOSITION_NON_DIEGETIC
+	AV_DISPOSITION_CAPTIONS         AVDisposition = C.AV_DISPOSITION_CAPTIONS
+	AV_DISPOSITION_DESCRIPTIONS     AVDisposition = C.AV_DISPOSITION_DESCRIPTIONS
+	AV_DISPOSITION_METADATA         AVDisposition = C.AV_DISPOSITION_METADATA
+	AV_DISPOSITION_DEPENDENT        AVDisposition = C.AV_DISPOSITION_DEPENDENT
+	AV_DISPOSITION_STILL_IMAGE      AVDisposition = C.AV_DISPOSITION_STILL_IMAGE
+	AV_DISPOSITION_NONE             AVDisposition = 0
+	AV_DISPOSITION_MAX                            = AV_DISPOSITION_STILL_IMAGE
 )
 
 const (