|
@@ -48,13 +48,6 @@ namespace MediaBrowser.Server.Implementations.Persistence
|
|
return connection;
|
|
return connection;
|
|
}
|
|
}
|
|
|
|
|
|
- public static void BindGetSimilarityScore(IDbConnection connection, ILogger logger)
|
|
|
|
- {
|
|
|
|
- var sqlConnection = (SQLiteConnection)connection;
|
|
|
|
- SimiliarToFunction.Logger = logger;
|
|
|
|
- sqlConnection.BindFunction(new SimiliarToFunction());
|
|
|
|
- }
|
|
|
|
-
|
|
|
|
public static void BindFunction(this SQLiteConnection connection, SQLiteFunction function)
|
|
public static void BindFunction(this SQLiteConnection connection, SQLiteFunction function)
|
|
{
|
|
{
|
|
var attributes = function.GetType().GetCustomAttributes(typeof(SQLiteFunctionAttribute), true).Cast<SQLiteFunctionAttribute>().ToArray();
|
|
var attributes = function.GetType().GetCustomAttributes(typeof(SQLiteFunctionAttribute), true).Cast<SQLiteFunctionAttribute>().ToArray();
|
|
@@ -65,99 +58,4 @@ namespace MediaBrowser.Server.Implementations.Persistence
|
|
connection.BindFunction(attributes[0], function);
|
|
connection.BindFunction(attributes[0], function);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
-
|
|
|
|
- [SQLiteFunction(Name = "GetSimilarityScore", Arguments = 6, FuncType = FunctionType.Scalar)]
|
|
|
|
- public class SimiliarToFunction : SQLiteFunction
|
|
|
|
- {
|
|
|
|
- internal static ILogger Logger;
|
|
|
|
-
|
|
|
|
- private readonly Dictionary<string, int> _personTypeScores = new Dictionary<string, int>(StringComparer.OrdinalIgnoreCase)
|
|
|
|
- {
|
|
|
|
- { PersonType.Actor, 3},
|
|
|
|
- { PersonType.Director, 5},
|
|
|
|
- { PersonType.Composer, 2},
|
|
|
|
- { PersonType.GuestStar, 3},
|
|
|
|
- { PersonType.Writer, 2},
|
|
|
|
- { PersonType.Conductor, 2},
|
|
|
|
- { PersonType.Producer, 2},
|
|
|
|
- { PersonType.Lyricist, 2}
|
|
|
|
- };
|
|
|
|
-
|
|
|
|
- public override object Invoke(object[] args)
|
|
|
|
- {
|
|
|
|
- var score = 0;
|
|
|
|
-
|
|
|
|
- // Official rating equals
|
|
|
|
- if ((long)args[0] == 1)
|
|
|
|
- {
|
|
|
|
- score += 10;
|
|
|
|
- }
|
|
|
|
-
|
|
|
|
- // Year difference
|
|
|
|
- long? yearDifference = args[1] == null ? (long?)null : (long)args[1];
|
|
|
|
- if (yearDifference.HasValue)
|
|
|
|
- {
|
|
|
|
- var diff = Math.Abs(yearDifference.Value);
|
|
|
|
-
|
|
|
|
- // Add if they came out within the same decade
|
|
|
|
- if (diff < 10)
|
|
|
|
- {
|
|
|
|
- score += 2;
|
|
|
|
- }
|
|
|
|
-
|
|
|
|
- // And more if within five years
|
|
|
|
- if (diff < 5)
|
|
|
|
- {
|
|
|
|
- score += 2;
|
|
|
|
- }
|
|
|
|
- }
|
|
|
|
-
|
|
|
|
- // genres
|
|
|
|
- score += Convert.ToInt32((long)args[2]) * 10;
|
|
|
|
-
|
|
|
|
- // tags
|
|
|
|
- score += Convert.ToInt32((long)args[3]) * 10;
|
|
|
|
-
|
|
|
|
- // # of common keywords
|
|
|
|
- score += Convert.ToInt32((long)args[4]) *10;
|
|
|
|
-
|
|
|
|
- // # of common studios
|
|
|
|
- score += Convert.ToInt32((long)args[5]) * 3;
|
|
|
|
-
|
|
|
|
- // studios
|
|
|
|
- //score += GetListScore(args, 7, 8, 3);
|
|
|
|
-
|
|
|
|
- //var rowPeopleNamesText = (args[12] as string) ?? string.Empty;
|
|
|
|
- //var rowPeopleNames = rowPeopleNamesText.Split(new[] { '|' }, StringSplitOptions.RemoveEmptyEntries);
|
|
|
|
-
|
|
|
|
- //foreach (var name in rowPeopleNames)
|
|
|
|
- //{
|
|
|
|
- // // TODO: Send along person types
|
|
|
|
- // score += 3;
|
|
|
|
- //}
|
|
|
|
-
|
|
|
|
- //Logger.Debug("Returning score {0}", score);
|
|
|
|
- return score;
|
|
|
|
- }
|
|
|
|
-
|
|
|
|
- private int GetListScore(object[] args, int index1, int index2, int value = 10)
|
|
|
|
- {
|
|
|
|
- var score = 0;
|
|
|
|
-
|
|
|
|
- var inputGenres = args[index1] as string;
|
|
|
|
- var rowGenres = args[index2] as string;
|
|
|
|
- var inputGenreList = string.IsNullOrWhiteSpace(inputGenres) ? new string[] { } : inputGenres.Split(new[] { '|' }, StringSplitOptions.RemoveEmptyEntries);
|
|
|
|
- var rowGenresList = string.IsNullOrWhiteSpace(rowGenres) ? new string[] { } : rowGenres.Split(new[] { '|' }, StringSplitOptions.RemoveEmptyEntries);
|
|
|
|
-
|
|
|
|
- foreach (var genre in inputGenreList)
|
|
|
|
- {
|
|
|
|
- if (rowGenresList.Contains(genre, StringComparer.OrdinalIgnoreCase))
|
|
|
|
- {
|
|
|
|
- score += value;
|
|
|
|
- }
|
|
|
|
- }
|
|
|
|
-
|
|
|
|
- return score;
|
|
|
|
- }
|
|
|
|
- }
|
|
|
|
}
|
|
}
|