stream.go

package main

import (
	"fmt"

	"github.com/KarpelesLab/ffprobe"
)

const (
	AudioStream byte = 'a'
	VideoStream byte = 'v'
	SubsStream  byte = 's'
)

type hlsStream struct {
	id  int             // stream number
	lid int             // stream number per type
	typ byte            // 'v', 'a' or 's' depending if video/audio/subtitle stream
	src *ffprobe.Stream // source stream
}

// newStream return a new stream with the correct id set
func (hls *hlsBuilder) newStream(src *ffprobe.Stream) *hlsStream {
	s := &hlsStream{
		id:  len(hls.streams),
		typ: src.CodecType[0],
		src: src,
	}
	// compute lid
	for _, x := range hls.streams {
		if x.typ == s.typ {
			s.lid += 1
		}
	}
	hls.streams = append(hls.streams, s)
	return s
}

func (s *hlsStream) String() string {
	return fmt.Sprintf("%c:%d", s.typ, s.lid)
}

func (s *hlsStream) Filename() string {
	switch s.typ {
	case SubsStream:
		return fmt.Sprintf("stream_%d.vtt", s.id)
	default:
		return fmt.Sprintf("stream_%d.mp4", s.id)
	}
}

func (s *hlsStream) Typename() string {
	switch s.typ {
	case AudioStream:
		return "audio"
	case VideoStream:
		return "video"
	case SubsStream:
		return "text"
	default:
		return ""
	}
}