photoprism/internal/crop/area_test.go

package crop

import (
	"fmt"
	"testing"

	"github.com/stretchr/testify/assert"
)

func TestArea_String(t *testing.T) {
	t.Run("3e814d0011f4", func(t *testing.T) {
		expected := fmt.Sprintf("%x%x00%x%x", 1000, 333, 1, 500)
		m := NewArea("face", 1.000, 0.33333, 0.001, 0.5)
		assert.Equal(t, expected, m.String())
	})
	t.Run("3360a7064042_face", func(t *testing.T) {
		m := NewArea("face", 0.822059, 0.167969, 0.1, 0.0664062)
		assert.Equal(t, "3360a7064042", m.String())
	})
	t.Run("3360a7064042_back", func(t *testing.T) {
		m := NewArea("back", 0.822059, 0.167969, 0.1, 0.0664062)
		assert.Equal(t, "3360a7064042", m.String())
	})
	t.Run("0c93e801e000", func(t *testing.T) {
		m := NewArea("face", 0.201, 1.000, 0.03, 0.00000001)
		assert.Equal(t, "0c93e801e000", m.String())
	})
	t.Run("0003e8000000", func(t *testing.T) {
		m := NewArea("face", 0.0001, 1.000, 0, 0.00000001)
		assert.Equal(t, "0003e8000000", m.String())
	})
	t.Run("00007b0003e8", func(t *testing.T) {
		m := NewArea("", -2.0001, 0.123, -0.1, 4.00000001)
		assert.Equal(t, "00007b0003e8", m.String())
	})
	t.Run("00007b0003e8", func(t *testing.T) {
		m := NewArea("", 0, 0, 0, 0)
		assert.Equal(t, "", m.String())
	})
}

func TestArea_Thumb(t *testing.T) {
	t.Run("3e814d0011f4", func(t *testing.T) {
		expected := fmt.Sprintf("346b3897eec9ef75e35fbf0bbc4c83c55ca41e31-%x%x00%x%x", 1000, 333, 1, 500)
		m := NewArea("face", 1.000, 0.33333, 0.001, 0.5)
		assert.Equal(t, expected, m.Thumb("346b3897eec9ef75e35fbf0bbc4c83c55ca41e31"))
	})
	t.Run("3360a7064042_face", func(t *testing.T) {
		m := NewArea("face", 0.822059, 0.167969, 0.1, 0.0664062)
		assert.Equal(t, "346b3897eec9ef75e35fbf0bbc4c83c55ca41e31-3360a7064042", m.Thumb("346b3897eec9ef75e35fbf0bbc4c83c55ca41e31"))
	})
	t.Run("3360a7064042_back", func(t *testing.T) {
		m := NewArea("back", 0.822059, 0.167969, 0.1, 0.0664062)
		assert.Equal(t, "346b3897eec9ef75e35fbf0bbc4c83c55ca41e31-3360a7064042", m.Thumb("346b3897eec9ef75e35fbf0bbc4c83c55ca41e31"))
	})
	t.Run("0c93e801e000", func(t *testing.T) {
		m := NewArea("face", 0.201, 1.000, 0.03, 0.00000001)
		assert.Equal(t, "346b3897eec9ef75e35fbf0bbc4c83c55ca41e31-0c93e801e000", m.Thumb("346b3897eec9ef75e35fbf0bbc4c83c55ca41e31"))
	})
	t.Run("0003e8000000", func(t *testing.T) {
		m := NewArea("face", 0.0001, 1.000, 0, 0.00000001)
		assert.Equal(t, "346b3897eec9ef75e35fbf0bbc4c83c55ca41e31-0003e8000000", m.Thumb("346b3897eec9ef75e35fbf0bbc4c83c55ca41e31"))
	})
	t.Run("ShortHash", func(t *testing.T) {
		m := NewArea("", -2.0001, 0.123, -0.1, 4.00000001)
		assert.Equal(t, "00007b0003e8", m.Thumb("346b3897eec9ef75e35fbf0bbc4c83c55ca41"))
	})
	t.Run("Empty", func(t *testing.T) {
		m := NewArea("", 0, 0, 0, 0)
		assert.Equal(t, "", m.Thumb("346b3897eec9ef75e35fbf0bbc4c83c55ca41"))
	})
}

func TestArea_FileWidth(t *testing.T) {
	t.Run("Tile50", func(t *testing.T) {
		m := NewArea("face", 1.000, 0.33333, 0.001, 0.5)
		assert.Equal(t, 49999, m.FileWidth(Size{Tile50, Tile320, "Lists", 50, 50, DefaultOptions}))
	})
	t.Run("Tile500", func(t *testing.T) {
		m := NewArea("face", 1.000, 0.33333, 0.001, 0.5)
		assert.Equal(t, 499999, m.FileWidth(Size{Tile500, "", "FaceNet", 500, 500, DefaultOptions}))
	})
}

func TestAreaFromString(t *testing.T) {
	t.Run("3e814d0011f4", func(t *testing.T) {
		a := AreaFromString("3e814d0011f4")
		assert.Equal(t, float32(1), a.X)
		assert.Equal(t, float32(0.333), a.Y)
		assert.Equal(t, float32(0.001), a.W)
		assert.Equal(t, float32(0.5), a.H)
	})
	t.Run("3360a7064042", func(t *testing.T) {
		a := AreaFromString("3360a7064042")
		assert.Equal(t, NewArea("crop", 0.822, 0.167, 0.1, 0.066), a)
	})
}

func TestIsCroppedThumb(t *testing.T) {
	t.Run("True", func(t *testing.T) {
		assert.Equal(t, 40, IsCroppedThumb("346b3897eec9ef75e35fbf0bbc4c83c55ca41e31-00007b0003e8"))
	})
	t.Run("Invalid", func(t *testing.T) {
		assert.Equal(t, -1, IsCroppedThumb("346b3897eec9ef75e35fbf0bbc4c83c55ca41e3100007b0003e8-"))
	})
	t.Run("CropArea", func(t *testing.T) {
		assert.Equal(t, -1, IsCroppedThumb("00007b0003e8"))
	})
	t.Run("ShortHash", func(t *testing.T) {
		assert.Equal(t, -1, IsCroppedThumb("346b3897eec9ef75e35fbf0bbc4c83c55ca41"))
	})
	t.Run("HashOnly", func(t *testing.T) {
		assert.Equal(t, -1, IsCroppedThumb("346b3897eec9ef75e35fbf0bbc4c83c55ca41e31"))
	})
}

func TestParseThumb(t *testing.T) {
	t.Run("True", func(t *testing.T) {
		h, a := ParseThumb("346b3897eec9ef75e35fbf0bbc4c83c55ca41e31-00007b0003e8")
		assert.Equal(t, 40, len(h))
		assert.Equal(t, 12, len(a))
		assert.False(t, AreaFromString(a).Empty())
	})
	t.Run("Invalid", func(t *testing.T) {
		h, a := ParseThumb("346b3897eec9ef75e35fbf0bbc4c83c55ca41e3100007b0003e8-")
		assert.Equal(t, "346b3897eec9ef75e35fbf0bbc4c83c55ca41e3100007b0003e8", h)
		assert.Equal(t, 0, len(a))
		assert.True(t, AreaFromString(a).Empty())
	})
	t.Run("CropArea", func(t *testing.T) {
		h, a := ParseThumb("00007b0003e8")
		assert.Equal(t, 0, len(h))
		assert.Equal(t, 12, len(a))
		assert.False(t, AreaFromString(a).Empty())
	})
	t.Run("ShortHash", func(t *testing.T) {
		h, a := ParseThumb("346b3897eec9ef75e35fbf0bbc4c83c55ca41")
		assert.Equal(t, 37, len(h))
		assert.Equal(t, "346b3897eec9ef75e35fbf0bbc4c83c55ca41", h)
		assert.Equal(t, 0, len(a))
		assert.True(t, AreaFromString(a).Empty())
	})
	t.Run("HashOnly", func(t *testing.T) {
		h, a := ParseThumb("346b3897eec9ef75e35fbf0bbc4c83c55ca41e31")
		assert.Equal(t, 40, len(h))
		assert.Equal(t, "346b3897eec9ef75e35fbf0bbc4c83c55ca41e31", h)
		assert.Equal(t, 0, len(a))
		assert.True(t, AreaFromString(a).Empty())
	})
}

func TestArea_SurfaceRatio(t *testing.T) {
	var a1 = Area{Name: "face", X: 0.308333, Y: 0.206944, W: 0.355556, H: 0.355556}
	var a2 = Area{Name: "face", X: 0.208313, Y: 0.156914, W: 0.655556, H: 0.655556}
	var a3 = Area{Name: "face", X: 0.998133, Y: 0.816944, W: 0.0001, H: 0.0001}
	var a4 = Area{Name: "face", X: 0.298133, Y: 0.216944, W: 0.255556, H: 0.155556}
	var a5 = Area{Name: "face", X: 0.298133, Y: 0.216944, W: 0, H: 0}

	assert.Equal(t, 99, int(a1.SurfaceRatio(a1.OverlapArea(a1))*100))
	assert.Equal(t, 99, int(a1.SurfaceRatio(a1.OverlapArea(a2))*100))
	assert.Equal(t, 29, int(a2.SurfaceRatio(a2.OverlapArea(a1))*100))
	assert.Equal(t, 0, int(a1.SurfaceRatio(a1.OverlapArea(a3))*100))
	assert.Equal(t, 30, int(a1.SurfaceRatio(a1.OverlapArea(a4))*100))
	assert.Equal(t, 0, int(a1.SurfaceRatio(a3.OverlapArea(a1))*100))
	assert.Equal(t, 30, int(a1.SurfaceRatio(a4.OverlapArea(a1))*100))
	assert.Equal(t, 0, int(a1.SurfaceRatio(a5.OverlapArea(a1))*100))
}

func TestArea_OverlapArea(t *testing.T) {
	var a1 = Area{Name: "face", X: 0.308333, Y: 0.206944, W: 0.355556, H: 0.355556}
	var a2 = Area{Name: "face", X: 0.208313, Y: 0.156914, W: 0.655556, H: 0.655556}
	var a3 = Area{Name: "face", X: 0.998133, Y: 0.816944, W: 0.0001, H: 0.0001}
	var a4 = Area{Name: "face", X: 0.298133, Y: 0.216944, W: 0.255556, H: 0.155556}

	assert.Equal(t, 0.1264200823986168, a1.OverlapArea(a1))
	assert.Equal(t, int(a1.Surface()*10000), int(a1.OverlapArea(a1)*10000))
	assert.Equal(t, 0.1264200823986168, a1.OverlapArea(a2))
	assert.Equal(t, 0.1264200823986168, a2.OverlapArea(a1))
	assert.Equal(t, 0.0, a1.OverlapArea(a3))
	assert.Equal(t, 0.038166598943088825, a1.OverlapArea(a4))
}

func TestArea_OverlapPercent(t *testing.T) {
	var a1 = Area{Name: "face", X: 0.308333, Y: 0.206944, W: 0.355556, H: 0.355556}
	var a2 = Area{Name: "face", X: 0.208313, Y: 0.156914, W: 0.655556, H: 0.655556}
	var a3 = Area{Name: "face", X: 0.998133, Y: 0.816944, W: 0.0001, H: 0.0001}
	var a4 = Area{Name: "face", X: 0.298133, Y: 0.216944, W: 0.255556, H: 0.155556}

	assert.Equal(t, 100, a1.OverlapPercent(a1))
	assert.Equal(t, 29, a1.OverlapPercent(a2))
	assert.Equal(t, 100, a2.OverlapPercent(a1))
	assert.Equal(t, 0, a1.OverlapPercent(a3))
	assert.Equal(t, 96, a1.OverlapPercent(a4))
}