HyperNet/Paperclip
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

🐛 Analyzer now can handle EXIF Orientation

Browse Source
This commit is contained in:
LittleSheep 2025-01-21 12:27:30 +08:00
parent d0d14d7c67
commit ac347540b8
1 changed files with 14 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
pkg/internal/services/analyzer.go
View File

@ -99,6 +99,13 @@ func ScanUnanalyzedFileFromDatabase() {
}() }()
} }
func calculateAspectRatio(width, height int, orientation int) float64 {
if orientation == 6 || orientation == 8 {
width, height = height, width
}
return float64(width) / float64(height)
}
func AnalyzeAttachment(file models.Attachment) error { func AnalyzeAttachment(file models.Attachment) error {
if file.Destination != models.AttachmentDstTemporary { if file.Destination != models.AttachmentDstTemporary {
return fmt.Errorf("attachment isn't in temporary storage, unable to analyze") return fmt.Errorf("attachment isn't in temporary storage, unable to analyze")
@ -133,7 +140,7 @@ func AnalyzeAttachment(file models.Attachment) error {
"Model", "ShutterSpeed", "ISO", "Megapixels", "Aperture", "Model", "ShutterSpeed", "ISO", "Megapixels", "Aperture",
"ColorSpace", "ColorTemperature", "ColorTone", "Contrast", "ColorSpace", "ColorTemperature", "ColorTone", "Contrast",
"ExposureTime", "FNumber", "FocalLength", "Flash", "HDREffect", "ExposureTime", "FNumber", "FocalLength", "Flash", "HDREffect",
"LensModel", "LensModel", "Orientation",
} }
switch strings.SplitN(file.MimeType, "/", 2)[0] { switch strings.SplitN(file.MimeType, "/", 2)[0] {
@ -165,6 +172,9 @@ func AnalyzeAttachment(file models.Attachment) error {
exif := et.ExtractMetadata(dst) exif := et.ExtractMetadata(dst)
for _, data := range exif { for _, data := range exif {
for k := range data.Fields { for k := range data.Fields {
if k == "Orientation" {
file.Metadata["ratio"] = calculateAspectRatio(width, height, data.Fields[k].(int))
}
if strings.HasPrefix(k, "GPS") { if strings.HasPrefix(k, "GPS") {
data.Clear(k) data.Clear(k)
} else if lo.Contains(exifWhitelist, k) { } else if lo.Contains(exifWhitelist, k) {
@ -206,6 +216,9 @@ func AnalyzeAttachment(file models.Attachment) error {
exif := et.ExtractMetadata(dst) exif := et.ExtractMetadata(dst)
for _, data := range exif { for _, data := range exif {
for k := range data.Fields { for k := range data.Fields {
if k == "Orientation" {
file.Metadata["ratio"] = calculateAspectRatio(stream.Width, stream.Height, data.Fields[k].(int))
}
if strings.HasPrefix(k, "GPS") { if strings.HasPrefix(k, "GPS") {
data.Clear(k) data.Clear(k)
} else if lo.Contains(exifWhitelist, k) { } else if lo.Contains(exifWhitelist, k) {