GOMODULES
/
go-exif
mirror of https://github.com/dsoprea/go-exif.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
go-exif/ifd_enumerate_test.go

507 lines
14 KiB
Go
Raw Blame History

package exif
import (
"path"
"testing"
"bytes"
"fmt"
"reflect"
"encoding/binary"
"io/ioutil"
"github.com/dsoprea/go-logging"
)
func TestIfdTagEntry_ValueBytes(t *testing.T) {
byteOrder := binary.BigEndian
ve := NewValueEncoder(byteOrder)
original := []byte("original text")
ed, err := ve.encodeBytes(original)
log.PanicIf(err)
// Now, pass the raw encoded value as if it was the entire addressable area
// and provide an offset of 0 as if it was a real block of data and this
// value happened to be recorded at the beginning.
ite := IfdTagEntry{
TagType: TypeByte,
UnitCount: uint32(len(original)),
ValueOffset: 0,
}
decodedBytes, err := ite.ValueBytes(ed.Encoded, byteOrder)
log.PanicIf(err)
if bytes.Compare(decodedBytes, original) != 0 {
t.Fatalf("Bytes not decoded correctly.")
}
}
func TestIfdTagEntry_ValueBytes_RealData(t *testing.T) {
defer func() {
if state := recover(); state != nil {
err := log.Wrap(state.(error))
log.PrintErrorf(err, "Test failure.")
}
}()
filepath := path.Join(assetsPath, "NDM_8901.jpg")
rawExif, err := SearchFileAndExtractExif(filepath)
log.PanicIf(err)
eh, index, err := Collect(rawExif)
log.PanicIf(err)
var ite *IfdTagEntry
for _, thisIte := range index.RootIfd.Entries {
if thisIte.TagId == 0x0110 {
ite = thisIte
break
}
}
if ite == nil {
t.Fatalf("Tag not found.")
}
addressableData := rawExif[ExifAddressableAreaStart:]
decodedBytes, err := ite.ValueBytes(addressableData, eh.ByteOrder)
log.PanicIf(err)
expected := []byte("Canon EOS 5D Mark III")
expected = append(expected, 0)
if len(decodedBytes) != int(ite.UnitCount) {
t.Fatalf("Decoded bytes not the right count.")
} else if bytes.Compare(decodedBytes, expected) != 0 {
t.Fatalf("Decoded bytes not correct.")
}
}
func TestIfdTagEntry_Resolver_ValueBytes(t *testing.T) {
defer func() {
if state := recover(); state != nil {
err := log.Wrap(state.(error))
log.PrintErrorf(err, "Test failure.")
}
}()
filepath := path.Join(assetsPath, "NDM_8901.jpg")
rawExif, err := SearchFileAndExtractExif(filepath)
log.PanicIf(err)
eh, index, err := Collect(rawExif)
log.PanicIf(err)
var ite *IfdTagEntry
for _, thisIte := range index.RootIfd.Entries {
if thisIte.TagId == 0x0110 {
ite = thisIte
break
}
}
if ite == nil {
t.Fatalf("Tag not found.")
}
itevr := NewIfdTagEntryValueResolver(rawExif, eh.ByteOrder)
decodedBytes, err := itevr.ValueBytes(ite)
log.PanicIf(err)
expected := []byte("Canon EOS 5D Mark III")
expected = append(expected, 0)
if len(decodedBytes) != int(ite.UnitCount) {
t.Fatalf("Decoded bytes not the right count.")
} else if bytes.Compare(decodedBytes, expected) != 0 {
t.Fatalf("Decoded bytes not correct.")
}
}
func TestIfdTagEntry_Resolver_ValueBytes__Unknown_Field_And_Nonroot_Ifd(t *testing.T) {
defer func() {
if state := recover(); state != nil {
err := log.Wrap(state.(error))
log.PrintErrorf(err, "Test failure.")
}
}()
filepath := path.Join(assetsPath, "NDM_8901.jpg")
rawExif, err := SearchFileAndExtractExif(filepath)
log.PanicIf(err)
eh, index, err := Collect(rawExif)
log.PanicIf(err)
ii, _ := IfdIdOrFail(IfdStandard, IfdExif)
ifdExif := index.Lookup[ii][0]
var ite *IfdTagEntry
for _, thisIte := range ifdExif.Entries {
if thisIte.TagId == 0x9000 {
ite = thisIte
break
}
}
if ite == nil {
t.Fatalf("Tag not found.")
}
itevr := NewIfdTagEntryValueResolver(rawExif, eh.ByteOrder)
decodedBytes, err := itevr.ValueBytes(ite)
log.PanicIf(err)
expected := []byte { '0', '2', '3', '0' }
if len(decodedBytes) != int(ite.UnitCount) {
t.Fatalf("Decoded bytes not the right count.")
} else if bytes.Compare(decodedBytes, expected) != 0 {
t.Fatalf("Recovered unknown value is not correct.")
}
}
func Test_Ifd_FindTagWithId_Hit(t *testing.T) {
filepath := path.Join(assetsPath, "NDM_8901.jpg")
rawExif, err := SearchFileAndExtractExif(filepath)
log.PanicIf(err)
_, index, err := Collect(rawExif)
log.PanicIf(err)
ifd := index.RootIfd
results, err := ifd.FindTagWithId(0x011b)
if len(results) != 1 {
t.Fatalf("Exactly one result was not found: (%d)", len(results))
} else if results[0].TagId != 0x011b {
t.Fatalf("The result was not expected: %v", results[0])
}
}
func Test_Ifd_FindTagWithId_Miss(t *testing.T) {
filepath := path.Join(assetsPath, "NDM_8901.jpg")
rawExif, err := SearchFileAndExtractExif(filepath)
log.PanicIf(err)
_, index, err := Collect(rawExif)
log.PanicIf(err)
ifd := index.RootIfd
_, err = ifd.FindTagWithId(0xffff)
if err == nil {
t.Fatalf("Expected error for not-found tag.")
} else if log.Is(err, ErrTagNotFound) == false {
log.Panic(err)
}
}
func Test_Ifd_FindTagWithName_Hit(t *testing.T) {
filepath := path.Join(assetsPath, "NDM_8901.jpg")
rawExif, err := SearchFileAndExtractExif(filepath)
log.PanicIf(err)
_, index, err := Collect(rawExif)
log.PanicIf(err)
ifd := index.RootIfd
results, err := ifd.FindTagWithName("YResolution")
if len(results) != 1 {
t.Fatalf("Exactly one result was not found: (%d)", len(results))
} else if results[0].TagId != 0x011b {
t.Fatalf("The result was not expected: %v", results[0])
}
}
func Test_Ifd_FindTagWithName_Miss(t *testing.T) {
filepath := path.Join(assetsPath, "NDM_8901.jpg")
rawExif, err := SearchFileAndExtractExif(filepath)
log.PanicIf(err)
_, index, err := Collect(rawExif)
log.PanicIf(err)
ifd := index.RootIfd
_, err = ifd.FindTagWithName("PlanarConfiguration")
if err == nil {
t.Fatalf("Expected error for not-found tag.")
} else if log.Is(err, ErrTagNotFound) == false {
log.Panic(err)
}
}
func Test_Ifd_FindTagWithName_NonStandard(t *testing.T) {
filepath := path.Join(assetsPath, "NDM_8901.jpg")
rawExif, err := SearchFileAndExtractExif(filepath)
log.PanicIf(err)
_, index, err := Collect(rawExif)
log.PanicIf(err)
ifd := index.RootIfd
_, err = ifd.FindTagWithName("GeorgeNotAtHome")
if err == nil {
t.Fatalf("Expected error for not-found tag.")
} else if log.Is(err, ErrTagNotStandard) == false {
log.Panic(err)
}
}
func Test_Ifd_Thumbnail(t *testing.T) {
filepath := path.Join(assetsPath, "NDM_8901.jpg")
rawExif, err := SearchFileAndExtractExif(filepath)
log.PanicIf(err)
_, index, err := Collect(rawExif)
log.PanicIf(err)
ifd := index.RootIfd
if ifd.NextIfd == nil {
t.Fatalf("There is no IFD1.")
}
// The thumbnail is in IFD1 (The second root IFD).
actual, err := ifd.NextIfd.Thumbnail()
log.PanicIf(err)
expectedFilepath := path.Join(assetsPath, "NDM_8901.jpg.thumbnail")
expected, err := ioutil.ReadFile(expectedFilepath)
log.PanicIf(err)
if bytes.Compare(actual, expected) != 0 {
t.Fatalf("thumbnail not correct")
}
}
func TestIfd_GpsInfo(t *testing.T) {
defer func() {
if state := recover(); state != nil {
err := log.Wrap(state.(error))
log.PrintErrorf(err, "Test failure.")
}
}()
filepath := path.Join(assetsPath, "gps.jpg")
rawExif, err := SearchFileAndExtractExif(filepath)
log.PanicIf(err)
_, index, err := Collect(rawExif)
log.PanicIf(err)
ifd, err := index.RootIfd.ChildWithIfdIdentity(GpsIi)
log.PanicIf(err)
gi, err := ifd.GpsInfo()
log.PanicIf(err)
if gi.Latitude.Orientation != 'N' || gi.Latitude.Degrees != 26 || gi.Latitude.Minutes != 35 || gi.Latitude.Seconds != 12 {
t.Fatalf("latitude not correct")
} else if gi.Longitude.Orientation != 'W' || gi.Longitude.Degrees != 80 || gi.Longitude.Minutes != 3 || gi.Longitude.Seconds != 13 {
t.Fatalf("longitude not correct")
} else if gi.Altitude != 0 {
t.Fatalf("altitude not correct")
} else if gi.Timestamp.Unix() != 1524964977 {
t.Fatalf("timestamp not correct")
} else if gi.Altitude != 0 {
t.Fatalf("altitude not correct")
}
}
func TestIfdEnumerate_EnumerateTagsRecursively(t *testing.T) {
filepath := path.Join(assetsPath, "NDM_8901.jpg")
rawExif, err := SearchFileAndExtractExif(filepath)
log.PanicIf(err)
_, index, err := Collect(rawExif)
log.PanicIf(err)
collected := make([][2]interface{}, 0)
cb := func(ifd *Ifd, ite *IfdTagEntry) error {
item := [2]interface{} {
ifd.Ii.IfdName,
int(ite.TagId),
}
collected = append(collected, item)
return nil
}
err = index.RootIfd.EnumerateTagsRecursively(cb)
log.PanicIf(err)
expected := [][2]interface{} {
[2]interface{} { "IFD", 0x010f },
[2]interface{} { "IFD", 0x0110 },
[2]interface{} { "IFD", 0x0112 },
[2]interface{} { "IFD", 0x011a },
[2]interface{} { "IFD", 0x011b },
[2]interface{} { "IFD", 0x0128 },
[2]interface{} { "IFD", 0x0132 },
[2]interface{} { "IFD", 0x013b },
[2]interface{} { "IFD", 0x0213 },
[2]interface{} { "IFD", 0x8298 },
[2]interface{} { "Exif", 0x829a },
[2]interface{} { "Exif", 0x829d },
[2]interface{} { "Exif", 0x8822 },
[2]interface{} { "Exif", 0x8827 },
[2]interface{} { "Exif", 0x8830 },
[2]interface{} { "Exif", 0x8832 },
[2]interface{} { "Exif", 0x9000 },
[2]interface{} { "Exif", 0x9003 },
[2]interface{} { "Exif", 0x9004 },
[2]interface{} { "Exif", 0x9101 },
[2]interface{} { "Exif", 0x9201 },
[2]interface{} { "Exif", 0x9202 },
[2]interface{} { "Exif", 0x9204 },
[2]interface{} { "Exif", 0x9207 },
[2]interface{} { "Exif", 0x9209 },
[2]interface{} { "Exif", 0x920a },
[2]interface{} { "Exif", 0x927c },
[2]interface{} { "Exif", 0x9286 },
[2]interface{} { "Exif", 0x9290 },
[2]interface{} { "Exif", 0x9291 },
[2]interface{} { "Exif", 0x9292 },
[2]interface{} { "Exif", 0xa000 },
[2]interface{} { "Exif", 0xa001 },
[2]interface{} { "Exif", 0xa002 },
[2]interface{} { "Exif", 0xa003 },
[2]interface{} { "Iop", 0x0001 },
[2]interface{} { "Iop", 0x0002 },
[2]interface{} { "Exif", 0xa20e },
[2]interface{} { "Exif", 0xa20f },
[2]interface{} { "Exif", 0xa210 },
[2]interface{} { "Exif", 0xa401 },
[2]interface{} { "Exif", 0xa402 },
[2]interface{} { "Exif", 0xa403 },
[2]interface{} { "Exif", 0xa406 },
[2]interface{} { "Exif", 0xa430 },
[2]interface{} { "Exif", 0xa431 },
[2]interface{} { "Exif", 0xa432 },
[2]interface{} { "Exif", 0xa434 },
[2]interface{} { "Exif", 0xa435 },
[2]interface{} { "GPSInfo", 0x0000 },
[2]interface{} { "IFD", 0x010f },
[2]interface{} { "IFD", 0x0110 },
[2]interface{} { "IFD", 0x0112 },
[2]interface{} { "IFD", 0x011a },
[2]interface{} { "IFD", 0x011b },
[2]interface{} { "IFD", 0x0128 },
[2]interface{} { "IFD", 0x0132 },
[2]interface{} { "IFD", 0x013b },
[2]interface{} { "IFD", 0x0213 },
[2]interface{} { "IFD", 0x8298 },
[2]interface{} { "Exif", 0x829a },
[2]interface{} { "Exif", 0x829d },
[2]interface{} { "Exif", 0x8822 },
[2]interface{} { "Exif", 0x8827 },
[2]interface{} { "Exif", 0x8830 },
[2]interface{} { "Exif", 0x8832 },
[2]interface{} { "Exif", 0x9000 },
[2]interface{} { "Exif", 0x9003 },
[2]interface{} { "Exif", 0x9004 },
[2]interface{} { "Exif", 0x9101 },
[2]interface{} { "Exif", 0x9201 },
[2]interface{} { "Exif", 0x9202 },
[2]interface{} { "Exif", 0x9204 },
[2]interface{} { "Exif", 0x9207 },
[2]interface{} { "Exif", 0x9209 },
[2]interface{} { "Exif", 0x920a },
[2]interface{} { "Exif", 0x927c },
[2]interface{} { "Exif", 0x9286 },
[2]interface{} { "Exif", 0x9290 },
[2]interface{} { "Exif", 0x9291 },
[2]interface{} { "Exif", 0x9292 },
[2]interface{} { "Exif", 0xa000 },
[2]interface{} { "Exif", 0xa001 },
[2]interface{} { "Exif", 0xa002 },
[2]interface{} { "Exif", 0xa003 },
[2]interface{} { "Iop", 0x0001 },
[2]interface{} { "Iop", 0x0002 },
[2]interface{} { "Exif", 0xa20e },
[2]interface{} { "Exif", 0xa20f },
[2]interface{} { "Exif", 0xa210 },
[2]interface{} { "Exif", 0xa401 },
[2]interface{} { "Exif", 0xa402 },
[2]interface{} { "Exif", 0xa403 },
[2]interface{} { "Exif", 0xa406 },
[2]interface{} { "Exif", 0xa430 },
[2]interface{} { "Exif", 0xa431 },
[2]interface{} { "Exif", 0xa432 },
[2]interface{} { "Exif", 0xa434 },
[2]interface{} { "Exif", 0xa435 },
[2]interface{} { "GPSInfo", 0x0000 },
}
if reflect.DeepEqual(collected, expected) != true {
fmt.Printf("ACTUAL:\n")
fmt.Printf("\n")
for _, item := range collected {
fmt.Printf("[2]interface{} { \"%s\", 0x%04x },\n", item[0], item[1])
}
fmt.Printf("\n")
fmt.Printf("EXPECTED:\n")
fmt.Printf("\n")
for _, item := range expected {
fmt.Printf("[2]interface{} { \"%s\", 0x%04x },\n", item[0], item[1])
}
fmt.Printf("\n")
t.Fatalf("tags not visited correctly: (%d) != (%d)", len(collected), len(expected))
}
}
func ExampleIfd_GpsInfo() {
filepath := path.Join(assetsPath, "gps.jpg")
rawExif, err := SearchFileAndExtractExif(filepath)
log.PanicIf(err)
_, index, err := Collect(rawExif)
log.PanicIf(err)
ifd, err := index.RootIfd.ChildWithIfdIdentity(GpsIi)
log.PanicIf(err)
gi, err := ifd.GpsInfo()
log.PanicIf(err)
fmt.Printf("%s\n", gi)
// Output:
// GpsInfo<LAT=(26.58667) LON=(-80.05361) ALT=(0) TIME=[2018-04-29 01:22:57 +0000 UTC]>
}
Reference in New Issue View Git Blame Copy Permalink