120 lines
2.9 KiB
Go
120 lines
2.9 KiB
Go
package entity
|
|
|
|
import (
|
|
"path"
|
|
"time"
|
|
|
|
"github.com/photoprism/photoprism/internal/classify"
|
|
"github.com/photoprism/photoprism/internal/maps"
|
|
"github.com/photoprism/photoprism/pkg/txt"
|
|
"gopkg.in/ugjka/go-tz.v2/tz"
|
|
)
|
|
|
|
// GetTimeZone uses PhotoLat and PhotoLng to guess the time zone of the photo.
|
|
func (m *Photo) GetTimeZone() string {
|
|
result := "UTC"
|
|
|
|
if m.HasLatLng() {
|
|
zones, err := tz.GetZone(tz.Point{
|
|
Lat: float64(m.PhotoLat),
|
|
Lon: float64(m.PhotoLng),
|
|
})
|
|
|
|
if err == nil && len(zones) > 0 {
|
|
result = zones[0]
|
|
}
|
|
}
|
|
|
|
return result
|
|
}
|
|
|
|
// CountryName returns the photo country name.
|
|
func (m *Photo) CountryName() string {
|
|
return maps.CountryNames[m.PhotoCountry]
|
|
}
|
|
|
|
// CountryCode returns the photo country code.
|
|
func (m *Photo) CountryCode() string {
|
|
return m.PhotoCountry
|
|
}
|
|
|
|
// GetTakenAt returns UTC time for TakenAtLocal.
|
|
func (m *Photo) GetTakenAt() time.Time {
|
|
loc, err := time.LoadLocation(m.TimeZone)
|
|
|
|
if err != nil {
|
|
return m.TakenAt
|
|
}
|
|
|
|
if takenAt, err := time.ParseInLocation("2006-01-02T15:04:05", m.TakenAtLocal.Format("2006-01-02T15:04:05"), loc); err != nil {
|
|
return m.TakenAt
|
|
} else {
|
|
return takenAt.UTC()
|
|
}
|
|
}
|
|
|
|
// UpdateLocation updates location and labels based on latitude and longitude.
|
|
func (m *Photo) UpdateLocation(geoApi string) (keywords []string, labels classify.Labels) {
|
|
if m.HasLatLng() {
|
|
var location = NewLocation(m.PhotoLat, m.PhotoLng)
|
|
|
|
err := location.Find(geoApi)
|
|
|
|
if location.Place == nil {
|
|
log.Warnf("photo: location place is nil (uid %s, loc_uid %s) - bug?", m.PhotoUID, location.LocUID)
|
|
}
|
|
|
|
if err == nil && location.Place != nil && location.LocUID != UnknownLocation.LocUID {
|
|
m.Location = location
|
|
m.LocUID = location.LocUID
|
|
m.Place = location.Place
|
|
m.PlaceUID = location.PlaceUID
|
|
m.PhotoCountry = location.CountryCode()
|
|
|
|
if m.TakenSrc != SrcManual {
|
|
m.TimeZone = m.GetTimeZone()
|
|
m.TakenAt = m.GetTakenAt()
|
|
}
|
|
|
|
FirstOrCreateCountry(NewCountry(location.CountryCode(), location.CountryName()))
|
|
|
|
locCategory := location.Category()
|
|
keywords = append(keywords, location.Keywords()...)
|
|
|
|
// Append category from reverse location lookup
|
|
if locCategory != "" {
|
|
labels = append(labels, classify.LocationLabel(locCategory, 0, -1))
|
|
}
|
|
|
|
return keywords, labels
|
|
}
|
|
}
|
|
|
|
keywords = []string{}
|
|
labels = classify.Labels{}
|
|
|
|
if m.UnknownLocation() {
|
|
m.Location = &UnknownLocation
|
|
m.LocUID = UnknownLocation.LocUID
|
|
} else if err := m.LoadLocation(); err == nil {
|
|
m.Place = m.Location.Place
|
|
m.PlaceUID = m.Location.PlaceUID
|
|
}
|
|
|
|
if m.UnknownPlace() {
|
|
m.Place = &UnknownPlace
|
|
m.PlaceUID = UnknownPlace.PlaceUID
|
|
} else if err := m.LoadPlace(); err == nil {
|
|
m.PhotoCountry = m.Place.CountryCode()
|
|
}
|
|
|
|
if m.UnknownCountry() {
|
|
m.PhotoCountry = txt.CountryCode(path.Join(m.PhotoPath, m.PhotoName))
|
|
}
|
|
|
|
if m.HasCountry() {
|
|
FirstOrCreateCountry(NewCountry(m.CountryCode(), m.CountryName()))
|
|
}
|
|
|
|
return keywords, labels
|
|
}
|