89 lines
2.3 KiB
Go
89 lines
2.3 KiB
Go
package entity
|
|
|
|
import (
|
|
"errors"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/jinzhu/gorm"
|
|
"github.com/photoprism/photoprism/pkg/txt"
|
|
)
|
|
|
|
// EstimatePosition updates the photo with an estimated geolocation if possible.
|
|
func (m *Photo) EstimatePosition() {
|
|
var recentPhoto Photo
|
|
var dateExpr string
|
|
|
|
switch DbDialect() {
|
|
case MySQL:
|
|
dateExpr = "ABS(DATEDIFF(taken_at, ?)) ASC"
|
|
case SQLite:
|
|
dateExpr = "ABS(JulianDay(taken_at) - JulianDay(?)) ASC"
|
|
default:
|
|
log.Errorf("photo: unknown sql dialect %s", DbDialect())
|
|
return
|
|
}
|
|
|
|
if err := UnscopedDb().
|
|
Where("place_id <> '' AND place_id <> 'zz' AND loc_src <> '' AND loc_src <> ?", SrcEstimate).
|
|
Order(gorm.Expr(dateExpr, m.TakenAt)).
|
|
Preload("Place").First(&recentPhoto).Error; err != nil {
|
|
log.Errorf("photo: %s", err.Error())
|
|
} else {
|
|
if days := recentPhoto.TakenAt.Sub(m.TakenAt) / (time.Hour * 24); days < -7 {
|
|
log.Debugf("prism: can't estimate position of %s, %d days time difference", m.PhotoUID, -1*days)
|
|
return
|
|
} else if days > -7 {
|
|
log.Debugf("prism: can't estimate position of %s, %d days time difference", m.PhotoUID, days)
|
|
return
|
|
}
|
|
|
|
if recentPhoto.HasPlace() {
|
|
m.Place = recentPhoto.Place
|
|
m.PlaceID = recentPhoto.PlaceID
|
|
m.PhotoCountry = recentPhoto.PhotoCountry
|
|
m.LocSrc = SrcEstimate
|
|
log.Debugf("prism: approximate position of %s is %s", m.PhotoUID, recentPhoto.PlaceID)
|
|
} else if recentPhoto.HasCountry() {
|
|
m.PhotoCountry = recentPhoto.PhotoCountry
|
|
m.LocSrc = SrcEstimate
|
|
log.Debugf("prism: probable country for %s is %s", m.PhotoUID, recentPhoto.PhotoCountry)
|
|
}
|
|
}
|
|
}
|
|
|
|
// Maintain photo data, improve if possible.
|
|
func (m *Photo) Maintain() error {
|
|
if !m.HasID() {
|
|
return errors.New("photo: can't maintain, id is empty")
|
|
}
|
|
|
|
maintained := time.Now()
|
|
m.MaintainedAt = &maintained
|
|
|
|
if m.UnknownCountry() && m.LocSrc == SrcAuto || m.UnknownLocation() && m.LocSrc == SrcEstimate {
|
|
m.EstimatePosition()
|
|
}
|
|
|
|
labels := m.ClassifyLabels()
|
|
|
|
m.UpdateYearMonth()
|
|
|
|
if err := m.UpdateTitle(labels); err != nil {
|
|
log.Warn(err)
|
|
}
|
|
|
|
if m.DetailsLoaded() {
|
|
w := txt.UniqueWords(txt.Words(m.Details.Keywords))
|
|
w = append(w, labels.Keywords()...)
|
|
m.Details.Keywords = strings.Join(txt.UniqueWords(w), ", ")
|
|
}
|
|
|
|
if err := m.IndexKeywords(); err != nil {
|
|
log.Errorf("photo: %s", err.Error())
|
|
}
|
|
|
|
m.PhotoQuality = m.QualityScore()
|
|
|
|
return UnscopedDb().Save(m).Error
|
|
}
|