mirror of
https://git.vectorsigma.ru/public/photoview.git
synced 2026-08-03 20:49:02 +00:00
a378522af4
Fix #1204 * Add bad medias to scanner_test. * Fix the script. * Add GenerateThumbnail function. * Add identify dimension * Use new API. Cover json condition. * Use magick to generate thumbnails. * Update go mod. * Remove dependency of executable_worker -> test_utils * Reduce dependency of magick. * Use magick to get dimension. * Remove almost all code around `image` package. * Fix thumbnailing with multi-frame gif. * Remove the struct in wrapper. * Fix typo. * Build images. * Update logs * Add jpeg decoder. * Remove code dependency. * Add log to confirm scan in parallel. * Update log * Add reformat image perf tests. * Rollback the build workflow.
131 lines
2.8 KiB
Go
131 lines
2.8 KiB
Go
package models
|
|
|
|
import (
|
|
"bytes"
|
|
"database/sql/driver"
|
|
"encoding/binary"
|
|
"fmt"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"github.com/photoview/photoview/api/database/drivers"
|
|
"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
|
|
}
|
|
|
|
// FaceRectangle stores a relative rectangle of a face in an image.
|
|
type FaceRectangle struct {
|
|
MinX, MaxX float64
|
|
MinY, MaxY float64
|
|
}
|
|
|
|
// 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
|
|
}
|