Fix #4442: Autotag image metadata.

Autotag `greyscale`, `non-repeating_animation`, and `exif_rotation`.

Note that this does not detect all (or even most) greyscale images.
Artists often save greyscale images as RGB instead of as greyscale.
This commit is contained in:
evazion
2021-09-21 10:58:51 -05:00
parent d5981754c4
commit c69ba54b5a
6 changed files with 82 additions and 2 deletions

View File

@@ -30,4 +30,48 @@ class MediaAsset < ApplicationRecord
def is_animated_png?
file_ext == "png" && metadata.fetch("PNG:AnimationFrames", 1) > 1
end
end
# @see https://exiftool.org/TagNames/JPEG.html
# @see https://exiftool.org/TagNames/PNG.html
# @see https://danbooru.donmai.us/posts?tags=exif:File:ColorComponents=1
# @see https://danbooru.donmai.us/posts?tags=exif:PNG:ColorType=Grayscale
def is_greyscale?
metadata["File:ColorComponents"] == 1 ||
metadata["PNG:ColorType"] == "Grayscale" ||
metadata["PNG:ColorType"] == "Grayscale with Alpha"
# Not always accurate:
# metadata["ICC-header:ColorSpaceData"] == "GRAY" ||
# metadata["XMP-photoshop:ColorMode"] == "Grayscale" ||
# metadata["XMP-photoshop:ICCProfileName"] == "EPSON Gray - Gamma 2.2" ||
# metadata["XMP-photoshop:ICCProfileName"] == "Gray Gamma 2.2"
end
# https://exiftool.org/TagNames/EXIF.html
def is_rotated?
metadata["IFD0:Orientation"].in?(["Rotate 90 CW", "Rotate 270 CW", "Rotate 180"]) ||
metadata["IFD1:Orientation"].in?(["Rotate 90 CW", "Rotate 270 CW", "Rotate 180"])
end
# Some animations technically have a finite loop count, but loop for hundreds
# or thousands of times. Only count animations with a low loop count as non-repeating.
def is_non_repeating_animation?
loop_count.in?(0..10)
end
# @see https://exiftool.org/TagNames/GIF.html
# @see https://exiftool.org/TagNames/PNG.html
# @see https://danbooru.donmai.us/posts?tags=-exif:GIF:AnimationIterations=Infinite+animated_gif
# @see https://danbooru.donmai.us/posts?tags=-exif:PNG:AnimationPlays=inf+animated_png
def loop_count
return Float::INFINITY if metadata["GIF:AnimationIterations"] == "Infinite"
return Float::INFINITY if metadata["PNG:AnimationPlays"] == "inf"
return metadata["GIF:AnimationIterations"] if metadata["GIF:AnimationIterations"].present?
return metadata["PNG:AnimationPlays"] if metadata["PNG:AnimationPlays"].present?
# If the AnimationIterations tag isn't present, then it's counted as a loop count of 0.
return 0 if is_animated_gif? && metadata["GIF:AnimationIterations"].nil?
nil
end
end

View File

@@ -498,7 +498,7 @@ class Post < ApplicationRecord
end
def add_automatic_tags(tags)
tags -= %w[incredibly_absurdres absurdres highres lowres huge_filesize flash video ugoira animated_gif animated_png]
tags -= %w[incredibly_absurdres absurdres highres lowres huge_filesize flash video ugoira animated_gif animated_png exif_rotation non-repeating_animation]
if image_width >= 10_000 || image_height >= 10_000
tags << "incredibly_absurdres"
@@ -543,6 +543,10 @@ class Post < ApplicationRecord
tags << "animated_gif" if media_asset.is_animated_gif?
tags << "animated_png" if media_asset.is_animated_png?
tags << "greyscale" if media_asset.is_greyscale?
tags << "exif_rotation" if media_asset.is_rotated?
tags << "non-repeating_animation" if media_asset.is_non_repeating_animation?
tags
end