VideoQuality.java

package net.filebot.media;

import static java.util.Comparator.*;
import static net.filebot.MediaTypes.*;
import static net.filebot.media.CachedMediaCharacteristics.*;
import static net.filebot.media.MediaDetection.*;
import static net.filebot.media.MediaFileUtilities.*;
import static net.filebot.util.StringUtilities.*;

import java.io.File;
import java.util.Comparator;
import java.util.regex.Pattern;

public class VideoQuality implements Comparator<File> {

    public static final Comparator<File> DESCENDING_ORDER = new VideoQuality(VideoFormat.DEFAULT_GROUPS, releaseInfo.getRepackPattern()).reversed();

    public static boolean isBetter(File f1, File f2) {
        return DESCENDING_ORDER.compare(f1, f2) < 0;
    }

    private final VideoFormat quality;
    private final Pattern repack;

    public VideoQuality(VideoFormat quality, Pattern repack) {
        this.quality = quality;
        this.repack = repack;
    }

    private final Comparator<File> chain = comparingInt(this::getQuality)
            .thenComparing(VideoCodec::compare)
            .thenComparingInt(this::getRepack)
            .thenComparingDouble(this::getScore)
            .thenComparingLong(File::length);

    @Override
    public int compare(File f1, File f2) {
        return chain.compare(getPrimaryFile(f1), getPrimaryFile(f2));
    }

    private File getPrimaryFile(File f) {
        return findPrimaryFile(f, VIDEO_FILES).orElse(f);
    }

    private int getRepack(File f) {
        return find(f.getName(), repack) ? 1 : 0;
    }

    private int getQuality(File f) {
        // use primary video file when checking video resolution of subtitle files or disk folders
        return getMediaCharacteristics(f, mi -> {
            Integer w = mi.getWidth();
            Integer h = mi.getHeight();

            // invalid media file
            if (w == null || h == null) {
                return null;
            }

            return quality.guessFormat(w, h);
        }).orElse(Integer.MIN_VALUE);
    }

    private double getScore(File f) {
        // use primary video file when checking video resolution of subtitle files or disk folders
        return getMediaCharacteristics(f, mi -> {
            Integer w = mi.getWidth();
            Integer h = mi.getHeight();
            Double br = mi.getBitRate();

            // invalid media file
            if (w == null || h == null || br == null) {
                return null;
            }

            return w * h * br;
        }).orElse(Double.MIN_VALUE);
    }

    public enum VideoCodec {

        MPEG, AVC, HEVC;

        public static VideoCodec get(MediaCharacteristics mi) {
            switch (mi.getVideoCodec()) {
                case "HEVC":
                    return HEVC;
                case "AVC":
                    return AVC;
                case "MPEG-4 Visual":
                    return MPEG;
            }

            // print missing codec name
            throw new UnsupportedOperationException("Unknown Video Codec: " + mi.getVideoCodec());
        }

        public static int compare(File f1, File f2) {
            VideoCodec vc1 = getMediaCharacteristics(f1, VideoCodec::get).orElse(null);
            VideoCodec vc2 = getMediaCharacteristics(f2, VideoCodec::get).orElse(null);

            if (vc1 == null || vc2 == null) {
                return 0;
            }

            return vc1.compareTo(vc2);
        }

    }

}