mirror of https://github.com/ErsatzTV/ErsatzTV.git
Jason Dove
3 years ago
committed by
GitHub
17 changed files with 431 additions and 5 deletions
@ -0,0 +1,11 @@
@@ -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 @@
@@ -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 @@
@@ -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 @@
@@ -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