package main

import (
	"github.com/lkmio/avformat/utils"
	"github.com/lkmio/lkm/flv"
	"github.com/lkmio/lkm/hls"
	"github.com/lkmio/lkm/rtsp"
	"github.com/lkmio/lkm/stream"
)

// 处理不同包不能相互引用的需求

func NewStreamEndInfo(source string, tracks []*stream.Track, streams map[stream.TransStreamID]stream.TransStream) *stream.StreamEndInfo {
	if len(tracks) < 1 || len(streams) < 1 {
		return nil
	}

	info := stream.StreamEndInfo{
		ID:         source,
		Timestamps: make(map[utils.AVCodecID][2]int64, len(tracks)),
	}

	for _, track := range tracks {
		var timestamp [2]int64
		timestamp[0] = track.Dts + int64(track.FrameDuration)
		timestamp[1] = track.Pts + int64(track.FrameDuration)

		info.Timestamps[track.Stream.CodecID] = timestamp
	}

	for _, transStream := range streams {
		// 获取ts切片序号
		if stream.TransStreamHls == transStream.GetProtocol() {
			if hls := transStream.(*hls.TransStream); hls.M3U8Writer.Size() > 0 {
				info.M3U8Writer = hls.M3U8Writer
				info.PlaylistFormat = hls.PlaylistFormatPtr
			}
		} else if stream.TransStreamRtsp == transStream.GetProtocol() {
			if rtsp := transStream.(*rtsp.TransStream); len(rtsp.Tracks) > 0 {
				info.RtspTracks = make(map[int]uint16, len(tracks))
				for _, track := range rtsp.RtspTracks {
					info.RtspTracks[int(track.CodecID)] = track.EndSeq
				}
			}
		} else if stream.TransStreamFlv == transStream.GetProtocol() {
			stream := transStream.(*flv.TransStream)
			info.FLVPrevTagSize = stream.Muxer.PrevTagSize()
		}
	}

	return &info
}