moparisthebest/filebot
1
0
Fork 0
mirror of https://github.com/mitb-archive/filebot synced 2025-03-10 06:20:27 -04:00
Code Issues Releases Wiki Activity

Add RecentlyAired metric (at the very bottom though, when everything is lost already)

Browse Source
This commit is contained in:
Reinhard Pointner 2019-05-27 18:33:20 +07:00
parent 4b62417804
commit c3350bc8e2
1 changed files with 35 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
source/net/filebot/similarity/EpisodeMetrics.java
View File

@ -591,7 +591,9 @@ public class EpisodeMetrics {
}; };
// Match by file last modified and episode release dates // Match by file last modified and episode release dates
public final SimilarityMetric TimeStamp = new TimeStampMetric(10, ChronoUnit.YEARS) { public final TimeStampMetric TimeStamp = new TimeStampMetric(10, ChronoUnit.YEARS) {
private final Map<File, Long> cache = synchronizedMap(new HashMap<>());
@Override @Override
public float getSimilarity(Object o1, Object o2) { public float getSimilarity(Object o1, Object o2) {
@ -615,6 +617,7 @@ public class EpisodeMetrics {
} }
private long getTimeStamp(File file) { private long getTimeStamp(File file) {
return cache.computeIfAbsent(file, f -> {
if (VIDEO_FILES.accept(file) && file.length() > ONE_MEGABYTE) { if (VIDEO_FILES.accept(file) && file.length() > ONE_MEGABYTE) {
try (MediaCharacteristics mi = MediaCharacteristicsParser.DEFAULT.open(file)) { try (MediaCharacteristics mi = MediaCharacteristicsParser.DEFAULT.open(file)) {
Instant t = mi.getCreationTime(); Instant t = mi.getCreationTime();
@ -627,6 +630,7 @@ public class EpisodeMetrics {
} }
return super.getTimeStamp(file); // default to file creation date return super.getTimeStamp(file); // default to file creation date
});
} }
@Override @Override
@ -651,6 +655,25 @@ public class EpisodeMetrics {
} }
}; };
// Match by recently aired status
public final SimilarityMetric RecentlyAired = new TimeStampMetric(3, ChronoUnit.DAYS) {
@Override
public float getSimilarity(Object o1, Object o2) {
return super.getSimilarity(o1, o2) > 0 ? 1 : 0;
}
@Override
public long getTimeStamp(Object object) {
return object instanceof Episode || object instanceof File ? TimeStamp.getTimeStamp(object) : -1;
}
@Override
public String toString() {
return "RecentlyAired";
}
};
public final SimilarityMetric SeriesRating = new SimilarityMetric() { public final SimilarityMetric SeriesRating = new SimilarityMetric() {
@Override @Override
@ -827,11 +850,11 @@ public class EpisodeMetrics {
// 6 pass: divide by generic numeric similarity // 6 pass: divide by generic numeric similarity
// 7 pass: prefer episodes that were aired closer to the last modified date of the file // 7 pass: prefer episodes that were aired closer to the last modified date of the file
// 8 pass: resolve remaining collisions via absolute string similarity // 8 pass: resolve remaining collisions via absolute string similarity
return new SimilarityMetric[] { EpisodeFunnel, EpisodeBalancer, AirDate, MetaAttributes, SubstringFields, SeriesNameBalancer, SeriesName, RegionHint, SpecialNumber, Numeric, NumericSequence, SeriesRating, VoteRate, TimeStamp, FilePathBalancer, FilePath }; return new SimilarityMetric[] { EpisodeFunnel, EpisodeBalancer, AirDate, MetaAttributes, SubstringFields, SeriesNameBalancer, SeriesName, RegionHint, SpecialNumber, Numeric, NumericSequence, SeriesRating, VoteRate, TimeStamp, RecentlyAired, FilePathBalancer, FilePath };
} }
public SimilarityMetric[] matchFileSequence() { public SimilarityMetric[] matchFileSequence() {
return new SimilarityMetric[] { FileSize, new MetricCascade(FileName, EpisodeFunnel), EpisodeBalancer, AirDate, MetaAttributes, SubstringFields, SeriesNameBalancer, SeriesName, RegionHint, SpecialNumber, Numeric, NumericSequence, SeriesRating, VoteRate, TimeStamp, FilePathBalancer, FilePath }; return new SimilarityMetric[] { FileSize, new MetricCascade(FileName, EpisodeFunnel), EpisodeBalancer, AirDate, MetaAttributes, SubstringFields, SeriesNameBalancer, SeriesName, RegionHint, SpecialNumber, Numeric, NumericSequence, SeriesRating, VoteRate, TimeStamp, RecentlyAired, FilePathBalancer, FilePath };
} }
public SimilarityMetric numbers() { public SimilarityMetric numbers() {