147 lines
3.4 KiB
Go
147 lines
3.4 KiB
Go
package models
|
|
|
|
import (
|
|
"bytes"
|
|
"database/sql/driver"
|
|
"encoding/binary"
|
|
"fmt"
|
|
"image"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"github.com/photoview/photoview/api/database/drivers"
|
|
"github.com/photoview/photoview/api/scanner/media_encoding/media_utils"
|
|
"gorm.io/gorm"
|
|
"gorm.io/gorm/schema"
|
|
)
|
|
|
|
type FaceGroup struct {
|
|
Model
|
|
Label *string
|
|
ImageFaces []ImageFace `gorm:"constraint:OnDelete:CASCADE;"`
|
|
}
|
|
|
|
type ImageFace struct {
|
|
Model
|
|
FaceGroupID int `gorm:"not null;index"`
|
|
FaceGroup *FaceGroup
|
|
MediaID int `gorm:"not null;index"`
|
|
Media Media `gorm:"constraint:OnDelete:CASCADE;"`
|
|
Descriptor FaceDescriptor `gorm:"not null"`
|
|
Rectangle FaceRectangle `gorm:"not null"`
|
|
}
|
|
|
|
func (f *ImageFace) FillMedia(db *gorm.DB) error {
|
|
if f.Media.ID != 0 {
|
|
// media already exists
|
|
return nil
|
|
}
|
|
|
|
if err := db.Model(&f).Association("Media").Find(&f.Media); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
type FaceDescriptor [128]float32 // same as go-face's Descriptor
|
|
|
|
// GormDataType datatype used in database
|
|
func (FaceDescriptor) GormDBDataType(db *gorm.DB, field *schema.Field) string {
|
|
switch drivers.GetDatabaseDriverType(db) {
|
|
case drivers.MYSQL, drivers.SQLITE:
|
|
return "BLOB"
|
|
case drivers.POSTGRES:
|
|
return "BYTEA"
|
|
}
|
|
return ""
|
|
}
|
|
|
|
// Scan tells GORM how to convert database data to Go format
|
|
func (fd *FaceDescriptor) Scan(value interface{}) error {
|
|
byteValue := value.([]byte)
|
|
reader := bytes.NewReader(byteValue)
|
|
binary.Read(reader, binary.LittleEndian, fd)
|
|
return nil
|
|
}
|
|
|
|
// Value tells GORM how to save into the database
|
|
func (fd FaceDescriptor) Value() (driver.Value, error) {
|
|
buf := new(bytes.Buffer)
|
|
if err := binary.Write(buf, binary.LittleEndian, fd); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return buf.Bytes(), nil
|
|
}
|
|
|
|
type FaceRectangle struct {
|
|
MinX, MaxX float64
|
|
MinY, MaxY float64
|
|
}
|
|
|
|
// ToDBFaceRectangle converts a pixel absolute rectangle to a relative FaceRectangle to be saved in the database
|
|
func ToDBFaceRectangle(imgRec image.Rectangle, imagePath string) (*FaceRectangle, error) {
|
|
size, err := media_utils.GetPhotoDimensions(imagePath)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &FaceRectangle{
|
|
MinX: float64(imgRec.Min.X) / float64(size.Width),
|
|
MaxX: float64(imgRec.Max.X) / float64(size.Width),
|
|
MinY: float64(imgRec.Min.Y) / float64(size.Height),
|
|
MaxY: float64(imgRec.Max.Y) / float64(size.Height),
|
|
}, nil
|
|
}
|
|
|
|
// GormDataType datatype used in database
|
|
func (fr FaceRectangle) GormDataType() string {
|
|
return "VARCHAR(64)"
|
|
}
|
|
|
|
// Scan tells GORM how to convert database data to Go format
|
|
func (fr *FaceRectangle) Scan(value interface{}) error {
|
|
stringArray, ok := value.(string)
|
|
if !ok {
|
|
byteArray := value.([]uint8)
|
|
stringArray = string(byteArray)
|
|
}
|
|
|
|
slices := strings.Split(stringArray, ":")
|
|
|
|
if len(slices) != 4 {
|
|
return fmt.Errorf("Invalid face rectangle format, expected 4 values, got %d", len(slices))
|
|
}
|
|
|
|
var err error
|
|
|
|
fr.MinX, err = strconv.ParseFloat(slices[0], 32)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
fr.MaxX, err = strconv.ParseFloat(slices[1], 32)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
fr.MinY, err = strconv.ParseFloat(slices[2], 32)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
fr.MaxY, err = strconv.ParseFloat(slices[3], 32)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// Value tells GORM how to save into the database
|
|
func (fr FaceRectangle) Value() (driver.Value, error) {
|
|
result := fmt.Sprintf("%f:%f:%f:%f", fr.MinX, fr.MaxX, fr.MinY, fr.MaxY)
|
|
return result, nil
|
|
}
|