/*
NAME
  PES.go -
DESCRIPTION
  See Readme.md

AUTHOR
  Saxon Nelson-Milton <saxon.milton@gmail.com>

LICENSE
  PES.go is Copyright (C) 2017 the Australian Ocean Lab (AusOcean)

  It is free software: you can redistribute it and/or modify them
  under the terms of the GNU General Public License as published by the
  Free Software Foundation, either version 3 of the License, or (at your
  option) any later version.

  It is distributed in the hope that it will be useful, but WITHOUT
  ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
 for more details.

  You should have received a copy of the GNU General Public License
  along with revid in gpl.txt.  If not, see [GNU licenses](http://www.gnu.org/licenses).
*/

package nal

type NalAccessUnit struct {
  SPS []byte
  PPS []byte
  SEI [][]byte
  Data [][]byte
}

func (u *NalAccessUnit) AsAnnexB() (output []byte) {
  startCode := []byte{ 0x00,0x00,0x01}
  AUD := []byte{0x09, 0xF0}
  format := [][]byte{startCode, AUD, startCode, u.SPS, startCode, u.PPS }
  for i := range format {
    output = append(output,format[i]...)
  }
  for i := range u.SEI {
    output = append(output, append(startCode,u.SEI[i]...)...)
  }
  for i := range u.Data {
    output = append(output, append(startCode,u.Data[i]...)...)
  }
  return
}