From 0a447d6791b3e1a1db57f5b70faf763f3b1fbbb2 Mon Sep 17 00:00:00 2001
From: lordwelch <timmy@narnian.us>
Date: Mon, 8 Apr 2019 20:22:48 -0700
Subject: [PATCH] Finished the PlayItem now onto the SubPath. Design of these
 structs my become simplified later.

---
 main.go | 111 ++++++++++++++++++++++++++++++++++++++++++++++++++++++--
 1 file changed, 107 insertions(+), 4 deletions(-)

diff --git a/main.go b/main.go
index cce42e1..a3b3d44 100644
--- a/main.go
+++ b/main.go
@@ -202,6 +202,8 @@ type STNTable struct {
 	PrimaryAudioStreams       []PrimaryStream
 	PrimaryPGStreams          []PrimaryStream
 	PrimaryIGStreams          []PrimaryStream
+	SecondaryAudioStreams     []SecondaryAudioStream
+	SecondaryVideoStreams     []SecondaryVideoStream
 }
 
 // PrimaryStream holds a stream entry and attributes
@@ -210,6 +212,25 @@ type PrimaryStream struct {
 	StreamAttributes
 }
 
+// SecondaryStream holds stream references
+type SecondaryStream struct {
+	RefrenceEntryCount byte
+	StreamIDs          []byte
+}
+
+// SecondaryAudioStream holds a primary stream and a secondary stream
+type SecondaryAudioStream struct {
+	PrimaryStream
+	ExtraAttributes SecondaryStream
+}
+
+// SecondaryVideoStream holds a primary stream and a secondary stream for the video and a secondary stream for the Presentation Graphics/pip
+type SecondaryVideoStream struct {
+	PrimaryStream
+	ExtraAttributes SecondaryStream
+	PGStream        SecondaryStream
+}
+
 // StreamEntry holds the information for the data stream
 type StreamEntry struct {
 	Len       byte
@@ -236,6 +257,33 @@ type CLPI struct {
 	STCID    byte
 }
 
+type SubPath struct {
+	Len           uint32
+	Type          byte
+	flags         uint16
+	PlayItemCount byte
+}
+
+// SubPlayItem contains information about a PlayItem in the subpath
+type SubPlayItem struct {
+	Len              uint16
+	Clpi             CLPI
+	Flags            byte // multiangle/connection condition
+	Type             byte
+	InTime           int
+	OutTime          int
+	PlayItemID       uint16
+	StartOfPlayitem  uint64
+	UOMask           uint64
+	RandomAccessFlag byte
+	StillMode        byte
+	StillTime        uint16
+	AngleCount       byte
+	AngleFlags       byte
+	Angles           []CLPI
+	StreamTable      STNTable
+}
+
 type errReader struct {
 	RS  *bytes.Reader
 	err error
@@ -394,6 +442,10 @@ func (p *Playlist) parse(reader *errReader) error {
 		p.PlayItems = append(p.PlayItems, item)
 	}
 
+	for i := 0; i < int(p.SubPathCount); i++ {
+
+	}
+
 	return reader.err
 }
 
@@ -470,7 +522,7 @@ func (clpi *CLPI) parse(reader *errReader) error {
 	return reader.err
 }
 
-// parse reads Stream data from an *errReader #nosec G104
+// parse reads PrimaryStream data from an *errReader #nosec G104
 func (stnt *STNTable) parse(reader *errReader) error {
 	var (
 		buf [10]byte
@@ -509,6 +561,15 @@ func (stnt *STNTable) parse(reader *errReader) error {
 		stnt.PrimaryAudioStreams = append(stnt.PrimaryAudioStreams, stream)
 	}
 
+	for i := 0; i < int(stnt.PrimaryPGStreamCount); i++ {
+		var stream PrimaryStream
+		err = stream.parse(reader)
+		if err != nil {
+			return err
+		}
+		stnt.PrimaryPGStreams = append(stnt.PrimaryPGStreams, stream)
+	}
+
 	for i := 0; i < int(stnt.PrimaryIGStreamCount); i++ {
 		var stream PrimaryStream
 		err = stream.parse(reader)
@@ -518,15 +579,57 @@ func (stnt *STNTable) parse(reader *errReader) error {
 		stnt.PrimaryIGStreams = append(stnt.PrimaryIGStreams, stream)
 	}
 
-	for i := 0; i < int(stnt.PrimaryAudioStreamCount); i++ {
-		var stream PrimaryStream
+	for i := 0; i < int(stnt.SecondaryAudioStreamCount); i++ {
+		var stream SecondaryAudioStream
 		err = stream.parse(reader)
 		if err != nil {
 			return err
 		}
-		stnt.PrimaryAudioStreams = append(stnt.PrimaryAudioStreams, stream)
+		stnt.SecondaryAudioStreams = append(stnt.SecondaryAudioStreams, stream)
 	}
 
+	for i := 0; i < int(stnt.SecondaryVideoStreamCount); i++ {
+		var stream SecondaryVideoStream
+		err = stream.parse(reader)
+		if err != nil {
+			return err
+		}
+		stnt.SecondaryVideoStreams = append(stnt.SecondaryVideoStreams, stream)
+	}
+
+	return reader.err
+}
+
+// parse reads SecondaryStream data from an *errReader #nosec G104
+func (ss *SecondaryStream) parse(reader *errReader) error {
+	var (
+		buf [10]byte
+	)
+
+	reader.Read(buf[:2])
+	ss.RefrenceEntryCount = buf[0]
+	ss.StreamIDs = make([]byte, ss.RefrenceEntryCount, ss.RefrenceEntryCount)
+	reader.Read(ss.StreamIDs)
+	if ss.RefrenceEntryCount%2 != 0 {
+		reader.Seek(1, io.SeekCurrent)
+	}
+	return reader.err
+}
+
+// parse reads SecondaryAudioStream data from an *errReader #nosec G104
+func (sas *SecondaryAudioStream) parse(reader *errReader) error {
+	sas.PrimaryStream.parse(reader)
+	sas.ExtraAttributes.parse(reader)
+
+	return reader.err
+}
+
+// parse reads SecondaryVideoStream data from an *errReader #nosec G104
+func (svs *SecondaryVideoStream) parse(reader *errReader) error {
+	svs.PrimaryStream.parse(reader)
+	svs.ExtraAttributes.parse(reader)
+	svs.PGStream.parse(reader)
+
 	return reader.err
 }