From 81e9a604c763c16c288f83d4df3235da545c246a Mon Sep 17 00:00:00 2001
From: Reinhard Pointner <rednoah@users.noreply.github.com>
Date: Tue, 29 Nov 2011 03:07:44 +0000
Subject: [PATCH] * color highlighting indicating match probability

---
 .../ui/subtitle/SubtitleAutoMatchDialog.java  | 50 ++++++++++++++++---
 1 file changed, 43 insertions(+), 7 deletions(-)

diff --git a/source/net/sourceforge/filebot/ui/subtitle/SubtitleAutoMatchDialog.java b/source/net/sourceforge/filebot/ui/subtitle/SubtitleAutoMatchDialog.java
index fe7d9f54..4cc6f8de 100644
--- a/source/net/sourceforge/filebot/ui/subtitle/SubtitleAutoMatchDialog.java
+++ b/source/net/sourceforge/filebot/ui/subtitle/SubtitleAutoMatchDialog.java
@@ -76,10 +76,12 @@ import net.sourceforge.tuned.ui.RoundBorder;
 
 class SubtitleAutoMatchDialog extends JDialog {
 	
-	private final JPanel hashMatcherServicePanel = createServicePanel(0xFAFAD2); // LightGoldenRodYellow
-	private final JPanel nameMatcherServicePanel = createServicePanel(0xFFEBCD); // BlanchedAlmond
-	private final List<SubtitleServiceBean> services = new ArrayList<SubtitleServiceBean>();
+	private static final Color hashMatchColor = new Color(0xFAFAD2); // LightGoldenRodYellow
+	private static final Color nameMatchColor = new Color(0xFFEBCD); // BlanchedAlmond
+	private final JPanel hashMatcherServicePanel = createServicePanel(hashMatchColor);
+	private final JPanel nameMatcherServicePanel = createServicePanel(nameMatchColor);
 	
+	private final List<SubtitleServiceBean> services = new ArrayList<SubtitleServiceBean>();
 	private final JTable subtitleMappingTable = createTable();
 	
 	private ExecutorService queryService;
@@ -101,11 +103,11 @@ class SubtitleAutoMatchDialog extends JDialog {
 	}
 	
 
-	protected JPanel createServicePanel(int color) {
+	protected JPanel createServicePanel(Color color) {
 		JPanel panel = new JPanel(new MigLayout("hidemode 3"));
 		panel.setBorder(new RoundBorder());
 		panel.setOpaque(false);
-		panel.setBackground(new Color(color));
+		panel.setBackground(color);
 		panel.setVisible(false);
 		return panel;
 	}
@@ -412,6 +414,18 @@ class SubtitleAutoMatchDialog extends JDialog {
 			setText(subtitleBean.getText());
 			setIcon(subtitleBean.getError() == null ? subtitleBean.getIcon() : ResourceManager.getIcon("status.warning"));
 			
+			if (!isSelected) {
+				float f = subtitleBean.getMatchProbability();
+				if (f < 1) {
+					setOpaque(true);
+					setBackground(nameMatchColor);
+				}
+				if (f < 0.9f) {
+					setOpaque(true);
+					setBackground(derive(Color.RED, 1 - (f * 0.75f)));
+				}
+			}
+			
 			return this;
 		}
 	}
@@ -628,6 +642,7 @@ class SubtitleAutoMatchDialog extends JDialog {
 
 	private static class SubtitleDescriptorBean extends AbstractBean {
 		
+		private final File videoFile;
 		private final SubtitleDescriptor descriptor;
 		private final SubtitleServiceBean service;
 		
@@ -635,12 +650,18 @@ class SubtitleAutoMatchDialog extends JDialog {
 		private Exception error;
 		
 
-		public SubtitleDescriptorBean(SubtitleDescriptor descriptor, SubtitleServiceBean service) {
+		public SubtitleDescriptorBean(File videoFile, SubtitleDescriptor descriptor, SubtitleServiceBean service) {
+			this.videoFile = videoFile;
 			this.descriptor = descriptor;
 			this.service = service;
 		}
 		
 
+		public float getMatchProbability() {
+			return service.getMatchProbabilty(videoFile, descriptor);
+		}
+		
+
 		public String getText() {
 			return formatSubtitle(descriptor.getName(), getLanguageName(), getType());
 		}
@@ -739,7 +760,7 @@ class SubtitleAutoMatchDialog extends JDialog {
 						
 						// associate subtitles with services
 						for (SubtitleDescriptor subtitleDescriptor : result.getValue()) {
-							subtitles.add(new SubtitleDescriptorBean(subtitleDescriptor, service));
+							subtitles.add(new SubtitleDescriptorBean(result.getKey(), subtitleDescriptor, service));
 						}
 						
 						subtitleSet.put(result.getKey(), subtitles);
@@ -853,6 +874,9 @@ class SubtitleAutoMatchDialog extends JDialog {
 		}
 		
 
+		public abstract float getMatchProbabilty(File videoFile, SubtitleDescriptor descriptor);
+		
+
 		protected abstract Map<File, List<SubtitleDescriptor>> getSubtitleList(Collection<File> files, String languageName, Component parent) throws Exception;
 		
 
@@ -905,6 +929,12 @@ class SubtitleAutoMatchDialog extends JDialog {
 		protected Map<File, List<SubtitleDescriptor>> getSubtitleList(Collection<File> files, String languageName, Component parent) throws Exception {
 			return service.getSubtitleList(files.toArray(new File[0]), languageName);
 		}
+		
+
+		@Override
+		public float getMatchProbabilty(File videoFile, SubtitleDescriptor descriptor) {
+			return 1;
+		}
 	}
 	
 
@@ -969,6 +999,12 @@ class SubtitleAutoMatchDialog extends JDialog {
 			
 			return subtitlesByFile;
 		}
+		
+
+		@Override
+		public float getMatchProbabilty(File videoFile, SubtitleDescriptor descriptor) {
+			return EpisodeMetrics.verificationMetric().getSimilarity(videoFile, descriptor) * 0.9f;
+		}
 	}
 	
 }