mirror of
https://github.com/OpenDiablo2/OpenDiablo2
synced 2024-09-06 19:44:15 -04:00
*`StreamReader.Read` methods now return errors The other edits in this commit are related to cleaning up lint errors caused by the changes to StreamReader
236 lines
5.3 KiB
Go
236 lines
5.3 KiB
Go
package d2video
|
|
|
|
import (
|
|
"errors"
|
|
"log"
|
|
|
|
"github.com/OpenDiablo2/OpenDiablo2/d2common/d2datautils"
|
|
)
|
|
|
|
// BinkVideoMode is the video mode type
|
|
type BinkVideoMode uint32
|
|
|
|
const (
|
|
// BinkVideoModeNormal is a normal video
|
|
BinkVideoModeNormal BinkVideoMode = iota
|
|
|
|
// BinkVideoModeHeightDoubled is a height-doubled video
|
|
BinkVideoModeHeightDoubled
|
|
|
|
// BinkVideoModeHeightInterlaced is a height-interlaced video
|
|
BinkVideoModeHeightInterlaced
|
|
|
|
// BinkVideoModeWidthDoubled is a width-doubled video
|
|
BinkVideoModeWidthDoubled
|
|
|
|
// BinkVideoModeWidthAndHeightDoubled is a width and height-doubled video
|
|
BinkVideoModeWidthAndHeightDoubled
|
|
|
|
// BinkVideoModeWidthAndHeightInterlaced is a width and height interlaced video
|
|
BinkVideoModeWidthAndHeightInterlaced
|
|
)
|
|
|
|
const (
|
|
numHeaderBytes = 3
|
|
bikHeaderStr = "BIK"
|
|
numAudioTrackUnknownBytes = 2
|
|
)
|
|
|
|
// BinkAudioAlgorithm represents the type of bink audio algorithm
|
|
type BinkAudioAlgorithm uint32
|
|
|
|
const (
|
|
// BinkAudioAlgorithmFFT is the FTT audio algorithm
|
|
BinkAudioAlgorithmFFT BinkAudioAlgorithm = iota
|
|
|
|
// BinkAudioAlgorithmDCT is the DCT audio algorithm
|
|
BinkAudioAlgorithmDCT
|
|
)
|
|
|
|
// BinkAudioTrack represents an audio track
|
|
type BinkAudioTrack struct {
|
|
AudioChannels uint16
|
|
AudioSampleRateHz uint16
|
|
Stereo bool
|
|
Algorithm BinkAudioAlgorithm
|
|
AudioTrackID uint32
|
|
}
|
|
|
|
// BinkDecoder represents the bink decoder
|
|
type BinkDecoder struct {
|
|
AudioTracks []BinkAudioTrack
|
|
FrameIndexTable []uint32
|
|
streamReader *d2datautils.StreamReader
|
|
fileSize uint32
|
|
numberOfFrames uint32
|
|
largestFrameSizeBytes uint32
|
|
VideoWidth uint32
|
|
VideoHeight uint32
|
|
FPS uint32
|
|
FrameTimeMS uint32
|
|
VideoMode BinkVideoMode
|
|
frameIndex uint32
|
|
videoCodecRevision byte
|
|
HasAlphaPlane bool
|
|
Grayscale bool
|
|
|
|
// Mask bit 0, as this is defined as a keyframe
|
|
|
|
}
|
|
|
|
// CreateBinkDecoder returns a new instance of the bink decoder
|
|
func CreateBinkDecoder(source []byte) (*BinkDecoder, error) {
|
|
result := &BinkDecoder{
|
|
streamReader: d2datautils.CreateStreamReader(source),
|
|
}
|
|
|
|
err := result.loadHeaderInformation()
|
|
|
|
return result, err
|
|
}
|
|
|
|
// GetNextFrame gets the next frame
|
|
func (v *BinkDecoder) GetNextFrame() error {
|
|
//nolint:gocritic // v.streamReader.SetPosition(uint64(v.FrameIndexTable[i] & 0xFFFFFFFE))
|
|
lengthOfAudioPackets, err := v.streamReader.ReadUInt32()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
samplesInPacket, err := v.streamReader.ReadUInt32()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
v.streamReader.SkipBytes(int(lengthOfAudioPackets) - 4) //nolint:gomnd // decode magic
|
|
|
|
log.Printf("Frame %d:\tSamp: %d", v.frameIndex, samplesInPacket)
|
|
|
|
v.frameIndex++
|
|
|
|
return nil
|
|
}
|
|
|
|
//nolint:gomnd,funlen,gocyclo // Decoder magic, can't help the long function length for now
|
|
func (v *BinkDecoder) loadHeaderInformation() error {
|
|
v.streamReader.SetPosition(0)
|
|
|
|
var err error
|
|
|
|
headerBytes, err := v.streamReader.ReadBytes(numHeaderBytes)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if string(headerBytes) != bikHeaderStr {
|
|
return errors.New("invalid header for bink video")
|
|
}
|
|
|
|
v.videoCodecRevision, err = v.streamReader.ReadByte()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
v.fileSize, err = v.streamReader.ReadUInt32()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
v.numberOfFrames, err = v.streamReader.ReadUInt32()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
v.largestFrameSizeBytes, err = v.streamReader.ReadUInt32()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
const numBytesToSkip = 4 // Number of frames again?
|
|
|
|
v.streamReader.SkipBytes(numBytesToSkip)
|
|
|
|
v.VideoWidth, err = v.streamReader.ReadUInt32()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
v.VideoHeight, err = v.streamReader.ReadUInt32()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
fpsDividend, err := v.streamReader.ReadUInt32()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
fpsDivider, err := v.streamReader.ReadUInt32()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
v.FPS = uint32(float32(fpsDividend) / float32(fpsDivider))
|
|
v.FrameTimeMS = 1000 / v.FPS
|
|
|
|
videoFlags, err := v.streamReader.ReadUInt32()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
v.VideoMode = BinkVideoMode((videoFlags >> 28) & 0x0F)
|
|
v.HasAlphaPlane = ((videoFlags >> 20) & 0x1) == 1
|
|
v.Grayscale = ((videoFlags >> 17) & 0x1) == 1
|
|
|
|
numberOfAudioTracks, err := v.streamReader.ReadUInt32()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
v.AudioTracks = make([]BinkAudioTrack, numberOfAudioTracks)
|
|
|
|
for i := 0; i < int(numberOfAudioTracks); i++ {
|
|
v.streamReader.SkipBytes(numAudioTrackUnknownBytes)
|
|
|
|
v.AudioTracks[i].AudioChannels, err = v.streamReader.ReadUInt16()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
for i := 0; i < int(numberOfAudioTracks); i++ {
|
|
v.AudioTracks[i].AudioSampleRateHz, err = v.streamReader.ReadUInt16()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
var flags uint16
|
|
|
|
flags, err = v.streamReader.ReadUInt16()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
v.AudioTracks[i].Stereo = ((flags >> 13) & 0x1) == 1
|
|
v.AudioTracks[i].Algorithm = BinkAudioAlgorithm((flags >> 12) & 0x1)
|
|
}
|
|
|
|
for i := 0; i < int(numberOfAudioTracks); i++ {
|
|
v.AudioTracks[i].AudioTrackID, err = v.streamReader.ReadUInt32()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
v.FrameIndexTable = make([]uint32, v.numberOfFrames+1)
|
|
|
|
for i := 0; i < int(v.numberOfFrames+1); i++ {
|
|
v.FrameIndexTable[i], err = v.streamReader.ReadUInt32()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|