mirror of https://github.com/ErsatzTV/ErsatzTV.git
Jason Dove
4 years ago
committed by
GitHub
17 changed files with 431 additions and 5 deletions
@ -0,0 +1,11 @@ |
|||||||
|
using ErsatzTV.Core.Domain; |
||||||
|
|
||||||
|
namespace ErsatzTV.Core.Interfaces.Scheduling; |
||||||
|
|
||||||
|
public interface IMultiEpisodeShuffleCollectionEnumeratorFactory |
||||||
|
{ |
||||||
|
IMediaCollectionEnumerator Create( |
||||||
|
string luaTemplatePath, |
||||||
|
IList<MediaItem> mediaItems, |
||||||
|
CollectionEnumeratorState state); |
||||||
|
} |
||||||
@ -0,0 +1,199 @@ |
|||||||
|
using ErsatzTV.Core.Domain; |
||||||
|
using ErsatzTV.Core.Interfaces.Scheduling; |
||||||
|
using ErsatzTV.Core.Scheduling; |
||||||
|
using Microsoft.Extensions.Logging; |
||||||
|
using NLua; |
||||||
|
|
||||||
|
namespace ErsatzTV.Infrastructure.Scheduling; |
||||||
|
|
||||||
|
public class MultiEpisodeShuffleCollectionEnumerator : IMediaCollectionEnumerator |
||||||
|
{ |
||||||
|
private readonly ILogger _logger; |
||||||
|
private readonly int _mediaItemCount; |
||||||
|
private readonly Dictionary<int, List<MediaItem>> _mediaItemGroups; |
||||||
|
private readonly List<MediaItem> _ungrouped; |
||||||
|
private CloneableRandom _random; |
||||||
|
private IList<MediaItem> _shuffled; |
||||||
|
|
||||||
|
public MultiEpisodeShuffleCollectionEnumerator( |
||||||
|
IList<MediaItem> mediaItems, |
||||||
|
CollectionEnumeratorState state, |
||||||
|
string templateFile, |
||||||
|
ILogger logger) |
||||||
|
{ |
||||||
|
_logger = logger; |
||||||
|
using var lua = new Lua(); |
||||||
|
lua.DoFile(templateFile); |
||||||
|
|
||||||
|
var numGroups = (int)(double)lua["numParts"]; |
||||||
|
|
||||||
|
_mediaItemGroups = new Dictionary<int, List<MediaItem>>(); |
||||||
|
for (var i = 1; i <= numGroups; i++) |
||||||
|
{ |
||||||
|
_mediaItemGroups.Add(i, new List<MediaItem>()); |
||||||
|
} |
||||||
|
|
||||||
|
_ungrouped = new List<MediaItem>(); |
||||||
|
_mediaItemCount = mediaItems.Count; |
||||||
|
|
||||||
|
var groupForEpisode = (LuaFunction)lua["partNumberForEpisode"]; |
||||||
|
IList<Episode> validEpisodes = mediaItems |
||||||
|
.OfType<Episode>() |
||||||
|
.Filter(e => e.Season is not null && e.EpisodeMetadata is not null && e.EpisodeMetadata.Count == 1) |
||||||
|
.ToList(); |
||||||
|
foreach (Episode episode in validEpisodes) |
||||||
|
{ |
||||||
|
// prep lua params
|
||||||
|
int seasonNumber = episode.Season.SeasonNumber; |
||||||
|
int episodeNumber = episode.EpisodeMetadata[0].EpisodeNumber; |
||||||
|
|
||||||
|
// call the lua fn
|
||||||
|
object[] result = groupForEpisode.Call(seasonNumber, episodeNumber); |
||||||
|
|
||||||
|
// if we get a group number back, use it
|
||||||
|
if (result[0] is long groupNumber) |
||||||
|
{ |
||||||
|
_mediaItemGroups[(int)groupNumber].Add(episode); |
||||||
|
} |
||||||
|
else |
||||||
|
{ |
||||||
|
_ungrouped.Add(episode); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// add everything else
|
||||||
|
_ungrouped.AddRange(mediaItems.Except(validEpisodes)); |
||||||
|
|
||||||
|
if (state.Index >= _mediaItemCount) |
||||||
|
{ |
||||||
|
state.Index = 0; |
||||||
|
state.Seed = new Random(state.Seed).Next(); |
||||||
|
} |
||||||
|
|
||||||
|
_random = new CloneableRandom(state.Seed); |
||||||
|
_shuffled = Shuffle(_random); |
||||||
|
|
||||||
|
State = new CollectionEnumeratorState { Seed = state.Seed }; |
||||||
|
while (State.Index < state.Index) |
||||||
|
{ |
||||||
|
MoveNext(); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
public CollectionEnumeratorState State { get; } |
||||||
|
|
||||||
|
public Option<MediaItem> Current => _shuffled.Any() ? _shuffled[State.Index % _mediaItemCount] : None; |
||||||
|
|
||||||
|
public void MoveNext() |
||||||
|
{ |
||||||
|
if ((State.Index + 1) % _mediaItemCount == 0) |
||||||
|
{ |
||||||
|
Option<MediaItem> tail = Current; |
||||||
|
|
||||||
|
State.Index = 0; |
||||||
|
do |
||||||
|
{ |
||||||
|
State.Seed = _random.Next(); |
||||||
|
_random = new CloneableRandom(State.Seed); |
||||||
|
_shuffled = Shuffle(_random); |
||||||
|
} while (_mediaItemCount > 1 && Current == tail); |
||||||
|
} |
||||||
|
else |
||||||
|
{ |
||||||
|
State.Index++; |
||||||
|
} |
||||||
|
|
||||||
|
State.Index %= _mediaItemCount; |
||||||
|
} |
||||||
|
|
||||||
|
public Option<MediaItem> Peek(int offset) |
||||||
|
{ |
||||||
|
if (offset == 0) |
||||||
|
{ |
||||||
|
return Current; |
||||||
|
} |
||||||
|
|
||||||
|
if ((State.Index + offset) % _mediaItemCount == 0) |
||||||
|
{ |
||||||
|
IList<MediaItem> shuffled; |
||||||
|
Option<MediaItem> tail = Current; |
||||||
|
|
||||||
|
// clone the random
|
||||||
|
CloneableRandom randomCopy = _random.Clone(); |
||||||
|
|
||||||
|
do |
||||||
|
{ |
||||||
|
int newSeed = randomCopy.Next(); |
||||||
|
randomCopy = new CloneableRandom(newSeed); |
||||||
|
shuffled = Shuffle(randomCopy); |
||||||
|
} while (_mediaItemCount > 1 && shuffled[0] == tail); |
||||||
|
|
||||||
|
return shuffled.Any() ? shuffled[0] : None; |
||||||
|
} |
||||||
|
|
||||||
|
return _shuffled.Any() ? _shuffled[(State.Index + offset) % _mediaItemCount] : None; |
||||||
|
} |
||||||
|
|
||||||
|
private IList<MediaItem> Shuffle(CloneableRandom random) |
||||||
|
{ |
||||||
|
int maxGroupNumber = _mediaItemGroups.Max(a => a.Key); |
||||||
|
var shuffledGroups = new List<IList<MediaItem>>(); |
||||||
|
for (var i = 1; i <= maxGroupNumber; i++) |
||||||
|
{ |
||||||
|
shuffledGroups.Add(Shuffle(_mediaItemGroups[i], random)); |
||||||
|
} |
||||||
|
|
||||||
|
int minItems = shuffledGroups.Min(g => g.Count); |
||||||
|
if (shuffledGroups.Any(g => g.Count != minItems)) |
||||||
|
{ |
||||||
|
_logger.LogError("Multi Episode Groups are different sizes; shuffle will not perform correctly!"); |
||||||
|
} |
||||||
|
|
||||||
|
// convert shuffled "groups" into groups that can be used for scheduling
|
||||||
|
var copy = new GroupedMediaItem[minItems + _ungrouped.Count]; |
||||||
|
for (var i = 0; i < minItems; i++) |
||||||
|
{ |
||||||
|
var group = new GroupedMediaItem(shuffledGroups[0][i], null); |
||||||
|
for (var j = 1; j < shuffledGroups.Count; j++) |
||||||
|
{ |
||||||
|
group.Additional.Add(shuffledGroups[j][i]); |
||||||
|
} |
||||||
|
|
||||||
|
copy[i] = group; |
||||||
|
} |
||||||
|
|
||||||
|
// convert all ungrouped into groups that can be used for scheduling
|
||||||
|
for (var i = 0; i < _ungrouped.Count; i++) |
||||||
|
{ |
||||||
|
MediaItem ungrouped = _ungrouped[i]; |
||||||
|
copy[minItems + i] = new GroupedMediaItem(ungrouped, null); |
||||||
|
} |
||||||
|
|
||||||
|
// perform shuffle
|
||||||
|
int n = copy.Length; |
||||||
|
while (n > 1) |
||||||
|
{ |
||||||
|
n--; |
||||||
|
int k = random.Next(n + 1); |
||||||
|
(copy[k], copy[n]) = (copy[n], copy[k]); |
||||||
|
} |
||||||
|
|
||||||
|
// flatten
|
||||||
|
return GroupedMediaItem.FlattenGroups(copy, _mediaItemCount); |
||||||
|
} |
||||||
|
|
||||||
|
private static IList<MediaItem> Shuffle(IEnumerable<MediaItem> mediaItems, CloneableRandom random) |
||||||
|
{ |
||||||
|
MediaItem[] copy = mediaItems.ToArray(); |
||||||
|
|
||||||
|
int n = copy.Length; |
||||||
|
while (n > 1) |
||||||
|
{ |
||||||
|
n--; |
||||||
|
int k = random.Next(n + 1); |
||||||
|
(copy[k], copy[n]) = (copy[n], copy[k]); |
||||||
|
} |
||||||
|
|
||||||
|
return copy; |
||||||
|
} |
||||||
|
} |
||||||
@ -0,0 +1,23 @@ |
|||||||
|
using ErsatzTV.Core.Domain; |
||||||
|
using ErsatzTV.Core.Interfaces.Scheduling; |
||||||
|
using Microsoft.Extensions.Logging; |
||||||
|
|
||||||
|
namespace ErsatzTV.Infrastructure.Scheduling; |
||||||
|
|
||||||
|
public class MultiEpisodeShuffleCollectionEnumeratorFactory |
||||||
|
: IMultiEpisodeShuffleCollectionEnumeratorFactory |
||||||
|
{ |
||||||
|
private readonly ILogger<MultiEpisodeShuffleCollectionEnumeratorFactory> _logger; |
||||||
|
|
||||||
|
public MultiEpisodeShuffleCollectionEnumeratorFactory( |
||||||
|
ILogger<MultiEpisodeShuffleCollectionEnumeratorFactory> logger) |
||||||
|
{ |
||||||
|
_logger = logger; |
||||||
|
} |
||||||
|
|
||||||
|
public IMediaCollectionEnumerator Create( |
||||||
|
string luaTemplatePath, |
||||||
|
IList<MediaItem> mediaItems, |
||||||
|
CollectionEnumeratorState state) => |
||||||
|
new MultiEpisodeShuffleCollectionEnumerator(mediaItems, state, luaTemplatePath, _logger); |
||||||
|
} |
||||||
@ -0,0 +1,13 @@ |
|||||||
|
-- the number of parts that each un-split file typically contains |
||||||
|
numParts = 3 |
||||||
|
|
||||||
|
-- return the part number for the given season number and episode number |
||||||
|
function partNumberForEpisode(seasonNumber, episodeNumber) |
||||||
|
local mod = episodeNumber % 3 |
||||||
|
if mod == 0 |
||||||
|
then |
||||||
|
return 3 |
||||||
|
else |
||||||
|
return mod |
||||||
|
end |
||||||
|
end |
||||||
Loading…
Reference in new issue