2
0

SimilarItemsHelper.cs 6.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182
  1. using System;
  2. using System.Collections.Generic;
  3. using System.Linq;
  4. using MediaBrowser.Controller.Dto;
  5. using MediaBrowser.Controller.Entities;
  6. using MediaBrowser.Controller.Library;
  7. using MediaBrowser.Controller.Persistence;
  8. using MediaBrowser.Model.Dto;
  9. using MediaBrowser.Model.Entities;
  10. using MediaBrowser.Model.Querying;
  11. namespace Jellyfin.Api.Helpers
  12. {
  13. /// <summary>
  14. /// The similar items helper class.
  15. /// </summary>
  16. public static class SimilarItemsHelper
  17. {
  18. internal static QueryResult<BaseItemDto> GetSimilarItemsResult(
  19. DtoOptions dtoOptions,
  20. IUserManager userManager,
  21. ILibraryManager libraryManager,
  22. IDtoService dtoService,
  23. Guid userId,
  24. string id,
  25. string excludeArtistIds,
  26. int? limit,
  27. Type[] includeTypes,
  28. Func<BaseItem, List<PersonInfo>, List<PersonInfo>, BaseItem, int> getSimilarityScore)
  29. {
  30. var user = !userId.Equals(Guid.Empty) ? userManager.GetUserById(userId) : null;
  31. var item = string.IsNullOrEmpty(id) ?
  32. (!userId.Equals(Guid.Empty) ? libraryManager.GetUserRootFolder() :
  33. libraryManager.RootFolder) : libraryManager.GetItemById(id);
  34. var query = new InternalItemsQuery(user)
  35. {
  36. IncludeItemTypes = includeTypes.Select(i => i.Name).ToArray(),
  37. Recursive = true,
  38. DtoOptions = dtoOptions
  39. };
  40. query.ExcludeArtistIds = RequestHelpers.GetGuids(excludeArtistIds);
  41. var inputItems = libraryManager.GetItemList(query);
  42. var items = GetSimilaritems(item, libraryManager, inputItems, getSimilarityScore)
  43. .ToList();
  44. var returnItems = items;
  45. if (limit.HasValue)
  46. {
  47. returnItems = returnItems.Take(limit.Value).ToList();
  48. }
  49. var dtos = dtoService.GetBaseItemDtos(returnItems, dtoOptions, user);
  50. return new QueryResult<BaseItemDto>
  51. {
  52. Items = dtos,
  53. TotalRecordCount = items.Count
  54. };
  55. }
  56. /// <summary>
  57. /// Gets the similaritems.
  58. /// </summary>
  59. /// <param name="item">The item.</param>
  60. /// <param name="libraryManager">The library manager.</param>
  61. /// <param name="inputItems">The input items.</param>
  62. /// <param name="getSimilarityScore">The get similarity score.</param>
  63. /// <returns>IEnumerable{BaseItem}.</returns>
  64. private static IEnumerable<BaseItem> GetSimilaritems(
  65. BaseItem item,
  66. ILibraryManager libraryManager,
  67. IEnumerable<BaseItem> inputItems,
  68. Func<BaseItem, List<PersonInfo>, List<PersonInfo>, BaseItem, int> getSimilarityScore)
  69. {
  70. var itemId = item.Id;
  71. inputItems = inputItems.Where(i => i.Id != itemId);
  72. var itemPeople = libraryManager.GetPeople(item);
  73. var allPeople = libraryManager.GetPeople(new InternalPeopleQuery
  74. {
  75. AppearsInItemId = item.Id
  76. });
  77. return inputItems.Select(i => new Tuple<BaseItem, int>(i, getSimilarityScore(item, itemPeople, allPeople, i)))
  78. .Where(i => i.Item2 > 2)
  79. .OrderByDescending(i => i.Item2)
  80. .Select(i => i.Item1);
  81. }
  82. private static IEnumerable<string> GetTags(BaseItem item)
  83. {
  84. return item.Tags;
  85. }
  86. /// <summary>
  87. /// Gets the similiarity score.
  88. /// </summary>
  89. /// <param name="item1">The item1.</param>
  90. /// <param name="item1People">The item1 people.</param>
  91. /// <param name="allPeople">All people.</param>
  92. /// <param name="item2">The item2.</param>
  93. /// <returns>System.Int32.</returns>
  94. internal static int GetSimiliarityScore(BaseItem item1, List<PersonInfo> item1People, List<PersonInfo> allPeople, BaseItem item2)
  95. {
  96. var points = 0;
  97. if (!string.IsNullOrEmpty(item1.OfficialRating) && string.Equals(item1.OfficialRating, item2.OfficialRating, StringComparison.OrdinalIgnoreCase))
  98. {
  99. points += 10;
  100. }
  101. // Find common genres
  102. points += item1.Genres.Where(i => item2.Genres.Contains(i, StringComparer.OrdinalIgnoreCase)).Sum(i => 10);
  103. // Find common tags
  104. points += GetTags(item1).Where(i => GetTags(item2).Contains(i, StringComparer.OrdinalIgnoreCase)).Sum(i => 10);
  105. // Find common studios
  106. points += item1.Studios.Where(i => item2.Studios.Contains(i, StringComparer.OrdinalIgnoreCase)).Sum(i => 3);
  107. var item2PeopleNames = allPeople.Where(i => i.ItemId == item2.Id)
  108. .Select(i => i.Name)
  109. .Where(i => !string.IsNullOrWhiteSpace(i))
  110. .DistinctNames()
  111. .ToDictionary(i => i, StringComparer.OrdinalIgnoreCase);
  112. points += item1People.Where(i => item2PeopleNames.ContainsKey(i.Name)).Sum(i =>
  113. {
  114. if (string.Equals(i.Type, PersonType.Director, StringComparison.OrdinalIgnoreCase) || string.Equals(i.Role, PersonType.Director, StringComparison.OrdinalIgnoreCase))
  115. {
  116. return 5;
  117. }
  118. if (string.Equals(i.Type, PersonType.Actor, StringComparison.OrdinalIgnoreCase) || string.Equals(i.Role, PersonType.Actor, StringComparison.OrdinalIgnoreCase))
  119. {
  120. return 3;
  121. }
  122. if (string.Equals(i.Type, PersonType.Composer, StringComparison.OrdinalIgnoreCase) || string.Equals(i.Role, PersonType.Composer, StringComparison.OrdinalIgnoreCase))
  123. {
  124. return 3;
  125. }
  126. if (string.Equals(i.Type, PersonType.GuestStar, StringComparison.OrdinalIgnoreCase) || string.Equals(i.Role, PersonType.GuestStar, StringComparison.OrdinalIgnoreCase))
  127. {
  128. return 3;
  129. }
  130. if (string.Equals(i.Type, PersonType.Writer, StringComparison.OrdinalIgnoreCase) || string.Equals(i.Role, PersonType.Writer, StringComparison.OrdinalIgnoreCase))
  131. {
  132. return 2;
  133. }
  134. return 1;
  135. });
  136. if (item1.ProductionYear.HasValue && item2.ProductionYear.HasValue)
  137. {
  138. var diff = Math.Abs(item1.ProductionYear.Value - item2.ProductionYear.Value);
  139. // Add if they came out within the same decade
  140. if (diff < 10)
  141. {
  142. points += 2;
  143. }
  144. // And more if within five years
  145. if (diff < 5)
  146. {
  147. points += 2;
  148. }
  149. }
  150. return points;
  151. }
  152. }
  153. }