examples_test.go

package mkvparse

import (
	"bytes"
	"fmt"
	"image"
	"image/jpeg"
	"image/png"
	"log"
	"os"
)

func ExampleNameForElementID() {
	fmt.Println(NameForElementID(InfoElement))
	// Output:
	// Info
}

func ExampleCoverHandler() {
	file, err := os.Open("testdata/example-cover.mkv")
	if err != nil {
		panic(err)
	}
	defer file.Close()

	handler := CoverHandler{}
	if err := ParseSections(file, &handler, AttachmentsElement); err != nil {
		panic(err)
	}

	fmt.Printf("parsed cover: %s (%d bytes)\n", handler.MIMEType, len(handler.Data))

	// Output:
	// parsed cover: image/jpeg (41363 bytes)
}

func ExampleParseCover() {
	data, typ, err := ParseCover("testdata/example-cover.mkv")
	if err != nil {
		panic(err)
	}

	fmt.Printf("parsed cover: %s (%d bytes)\n", typ, len(data))

	// Output:
	// parsed cover: image/jpeg (41363 bytes)
}

func ExampleParseCover_image() {
	data, typ, err := ParseCover("testdata/example-cover.mkv")
	if err != nil {
		log.Panic(err)
	}
	if data == nil {
		log.Panic("no cover")
		return
	}
	var img image.Image
	switch typ {
	case "image/jpeg":
		img, err = jpeg.Decode(bytes.NewReader(data))
	case "image/png":
		img, err = png.Decode(bytes.NewReader(data))
	default:
		log.Panicf("unknown MIME type: %s", typ)
	}
	if err != nil {
		log.Panic(err)
	}

	fmt.Printf("parsed cover image: %dx%d\n", img.Bounds().Dx(), img.Bounds().Dy())

	// Output:
	// parsed cover image: 265x377
}

func ExampleTagsHandler() {
	file, err := os.Open("testdata/example.mkv")
	if err != nil {
		panic(err)
	}
	defer file.Close()

	handler := NewTagsHandler()
	if err := ParseSections(file, handler, TagsElement); err != nil {
		panic(err)
	}

	fmt.Printf("Artist: %s\n", handler.Tags()[Tag_Artist])

	// Output:
	// Artist: John Doe
}