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);
		}

	}

}