|
|
@ -68,7 +68,7 @@ namespace MediaBrowser.Api
|
|
|
|
/// <param name="includeInSearch">The include in search.</param>
|
|
|
|
/// <param name="includeInSearch">The include in search.</param>
|
|
|
|
/// <param name="getSimilarityScore">The get similarity score.</param>
|
|
|
|
/// <param name="getSimilarityScore">The get similarity score.</param>
|
|
|
|
/// <returns>ItemsResult.</returns>
|
|
|
|
/// <returns>ItemsResult.</returns>
|
|
|
|
internal static ItemsResult GetSimilarItemsResult(DtoOptions dtoOptions, IUserManager userManager, IItemRepository itemRepository, ILibraryManager libraryManager, IUserDataManager userDataRepository, IDtoService dtoService, ILogger logger, BaseGetSimilarItemsFromItem request, Func<BaseItem, bool> includeInSearch, Func<BaseItem, BaseItem, ILibraryManager, int> getSimilarityScore)
|
|
|
|
internal static ItemsResult GetSimilarItemsResult(DtoOptions dtoOptions, IUserManager userManager, IItemRepository itemRepository, ILibraryManager libraryManager, IUserDataManager userDataRepository, IDtoService dtoService, ILogger logger, BaseGetSimilarItemsFromItem request, Func<BaseItem, bool> includeInSearch, Func<BaseItem, List<PersonInfo>, List<PersonInfo>, BaseItem, int> getSimilarityScore)
|
|
|
|
{
|
|
|
|
{
|
|
|
|
var user = !string.IsNullOrWhiteSpace(request.UserId) ? userManager.GetUserById(request.UserId) : null;
|
|
|
|
var user = !string.IsNullOrWhiteSpace(request.UserId) ? userManager.GetUserById(request.UserId) : null;
|
|
|
|
|
|
|
|
|
|
|
@ -110,12 +110,17 @@ namespace MediaBrowser.Api
|
|
|
|
/// <param name="inputItems">The input items.</param>
|
|
|
|
/// <param name="inputItems">The input items.</param>
|
|
|
|
/// <param name="getSimilarityScore">The get similarity score.</param>
|
|
|
|
/// <param name="getSimilarityScore">The get similarity score.</param>
|
|
|
|
/// <returns>IEnumerable{BaseItem}.</returns>
|
|
|
|
/// <returns>IEnumerable{BaseItem}.</returns>
|
|
|
|
internal static IEnumerable<BaseItem> GetSimilaritems(BaseItem item, ILibraryManager libraryManager, IEnumerable<BaseItem> inputItems, Func<BaseItem, BaseItem, ILibraryManager, int> getSimilarityScore)
|
|
|
|
internal static IEnumerable<BaseItem> GetSimilaritems(BaseItem item, ILibraryManager libraryManager, IEnumerable<BaseItem> inputItems, Func<BaseItem, List<PersonInfo>, List<PersonInfo>, BaseItem, int> getSimilarityScore)
|
|
|
|
{
|
|
|
|
{
|
|
|
|
var itemId = item.Id;
|
|
|
|
var itemId = item.Id;
|
|
|
|
inputItems = inputItems.Where(i => i.Id != itemId);
|
|
|
|
inputItems = inputItems.Where(i => i.Id != itemId);
|
|
|
|
|
|
|
|
var itemPeople = libraryManager.GetPeople(item);
|
|
|
|
|
|
|
|
var allPeople = libraryManager.GetPeople(new InternalPeopleQuery
|
|
|
|
|
|
|
|
{
|
|
|
|
|
|
|
|
AppearsInItemId = item.Id
|
|
|
|
|
|
|
|
});
|
|
|
|
|
|
|
|
|
|
|
|
return inputItems.Select(i => new Tuple<BaseItem, int>(i, getSimilarityScore(item, i, libraryManager)))
|
|
|
|
return inputItems.Select(i => new Tuple<BaseItem, int>(i, getSimilarityScore(item, itemPeople, allPeople, i)))
|
|
|
|
.Where(i => i.Item2 > 2)
|
|
|
|
.Where(i => i.Item2 > 2)
|
|
|
|
.OrderByDescending(i => i.Item2)
|
|
|
|
.OrderByDescending(i => i.Item2)
|
|
|
|
.Select(i => i.Item1);
|
|
|
|
.Select(i => i.Item1);
|
|
|
@ -147,9 +152,11 @@ namespace MediaBrowser.Api
|
|
|
|
/// Gets the similiarity score.
|
|
|
|
/// Gets the similiarity score.
|
|
|
|
/// </summary>
|
|
|
|
/// </summary>
|
|
|
|
/// <param name="item1">The item1.</param>
|
|
|
|
/// <param name="item1">The item1.</param>
|
|
|
|
|
|
|
|
/// <param name="item1People">The item1 people.</param>
|
|
|
|
|
|
|
|
/// <param name="allPeople">All people.</param>
|
|
|
|
/// <param name="item2">The item2.</param>
|
|
|
|
/// <param name="item2">The item2.</param>
|
|
|
|
/// <returns>System.Int32.</returns>
|
|
|
|
/// <returns>System.Int32.</returns>
|
|
|
|
internal static int GetSimiliarityScore(BaseItem item1, BaseItem item2, ILibraryManager libraryManager)
|
|
|
|
internal static int GetSimiliarityScore(BaseItem item1, List<PersonInfo> item1People, List<PersonInfo> allPeople, BaseItem item2)
|
|
|
|
{
|
|
|
|
{
|
|
|
|
var points = 0;
|
|
|
|
var points = 0;
|
|
|
|
|
|
|
|
|
|
|
@ -170,11 +177,13 @@ namespace MediaBrowser.Api
|
|
|
|
// Find common studios
|
|
|
|
// Find common studios
|
|
|
|
points += item1.Studios.Where(i => item2.Studios.Contains(i, StringComparer.OrdinalIgnoreCase)).Sum(i => 3);
|
|
|
|
points += item1.Studios.Where(i => item2.Studios.Contains(i, StringComparer.OrdinalIgnoreCase)).Sum(i => 3);
|
|
|
|
|
|
|
|
|
|
|
|
var item2PeopleNames = libraryManager.GetPeople(item2).Select(i => i.Name)
|
|
|
|
var item2PeopleNames = allPeople.Where(i => i.ItemId == item2.Id)
|
|
|
|
|
|
|
|
.Select(i => i.Name)
|
|
|
|
|
|
|
|
.Where(i => !string.IsNullOrWhiteSpace(i))
|
|
|
|
.DistinctNames()
|
|
|
|
.DistinctNames()
|
|
|
|
.ToDictionary(i => i, StringComparer.OrdinalIgnoreCase);
|
|
|
|
.ToDictionary(i => i, StringComparer.OrdinalIgnoreCase);
|
|
|
|
|
|
|
|
|
|
|
|
points += libraryManager.GetPeople(item1).Where(i => item2PeopleNames.ContainsKey(i.Name)).Sum(i =>
|
|
|
|
points += item1People.Where(i => item2PeopleNames.ContainsKey(i.Name)).Sum(i =>
|
|
|
|
{
|
|
|
|
{
|
|
|
|
if (string.Equals(i.Type, PersonType.Director, StringComparison.OrdinalIgnoreCase) || string.Equals(i.Role, PersonType.Director, StringComparison.OrdinalIgnoreCase))
|
|
|
|
if (string.Equals(i.Type, PersonType.Director, StringComparison.OrdinalIgnoreCase) || string.Equals(i.Role, PersonType.Director, StringComparison.OrdinalIgnoreCase))
|
|
|
|
{
|
|
|
|
{
|
|
|
|