36 lines
663 B
Java
36 lines
663 B
Java
|
|
package net.filebot.similarity;
|
|
|
|
|
|
import static net.filebot.util.FileUtilities.*;
|
|
|
|
import java.io.File;
|
|
|
|
|
|
public class FileNameMetric implements SimilarityMetric {
|
|
|
|
@Override
|
|
public float getSimilarity(Object o1, Object o2) {
|
|
String s1 = getFileName(o1);
|
|
if (s1 == null || s1.isEmpty())
|
|
return 0;
|
|
|
|
String s2 = getFileName(o2);
|
|
if (s2 == null || s2.isEmpty())
|
|
return 0;
|
|
|
|
return s1.startsWith(s2) || s2.startsWith(s1) ? 1 : 0;
|
|
}
|
|
|
|
|
|
protected String getFileName(Object object) {
|
|
if (object instanceof File) {
|
|
// name without extension normalized to lower-case
|
|
return getName((File) object).trim().toLowerCase();
|
|
}
|
|
|
|
return null;
|
|
}
|
|
|
|
}
|