mirror of https://github.com/ErsatzTV/ErsatzTV.git
Browse Source
* add script graphics element * pass template data as json to stdin * update changelogpull/2683/head
Jason Dove
1 month ago
committed by
GitHub
19 changed files with 377 additions and 53 deletions
@ -0,0 +1,27 @@
@@ -0,0 +1,27 @@
|
||||
using YamlDotNet.Serialization; |
||||
|
||||
namespace ErsatzTV.Core.Graphics; |
||||
|
||||
public class ScriptGraphicsElement : BaseGraphicsElement |
||||
{ |
||||
[YamlMember(Alias = "command", ApplyNamingConventions = false)] |
||||
public string Command { get; set; } |
||||
|
||||
[YamlMember(Alias = "args", ApplyNamingConventions = false)] |
||||
public List<string> Arguments { get; set; } |
||||
|
||||
[YamlMember(Alias = "z_index", ApplyNamingConventions = false)] |
||||
public int? ZIndex { get; set; } |
||||
|
||||
[YamlMember(Alias = "epg_entries", ApplyNamingConventions = false)] |
||||
public int EpgEntries { get; set; } |
||||
|
||||
[YamlMember(Alias = "start_seconds", ApplyNamingConventions = false)] |
||||
public double? StartSeconds { get; set; } |
||||
|
||||
[YamlMember(Alias = "duration_seconds", ApplyNamingConventions = false)] |
||||
public double? DurationSeconds { get; set; } |
||||
|
||||
[YamlMember(Alias = "pixel_format", ApplyNamingConventions = false)] |
||||
public string PixelFormat { get; set; } |
||||
} |
||||
@ -0,0 +1,173 @@
@@ -0,0 +1,173 @@
|
||||
using System.Buffers; |
||||
using System.IO.Pipelines; |
||||
using System.Text.Json; |
||||
using CliWrap; |
||||
using ErsatzTV.Core; |
||||
using ErsatzTV.Core.Graphics; |
||||
using ErsatzTV.Core.Interfaces.Streaming; |
||||
using Microsoft.Extensions.Logging; |
||||
using SkiaSharp; |
||||
|
||||
namespace ErsatzTV.Infrastructure.Streaming.Graphics; |
||||
|
||||
public class ScriptElement(ScriptGraphicsElement scriptElement, ILogger logger) |
||||
: GraphicsElement, IDisposable |
||||
{ |
||||
private CancellationTokenSource _cancellationTokenSource; |
||||
private CommandTask<CommandResult> _commandTask; |
||||
private int _frameSize; |
||||
private PipeReader _pipeReader; |
||||
private SKBitmap _canvasBitmap; |
||||
private TimeSpan _startTime; |
||||
private TimeSpan _endTime; |
||||
|
||||
public void Dispose() |
||||
{ |
||||
GC.SuppressFinalize(this); |
||||
|
||||
_pipeReader?.Complete(); |
||||
|
||||
_cancellationTokenSource?.Cancel(); |
||||
try |
||||
{ |
||||
#pragma warning disable VSTHRD002
|
||||
_commandTask?.Task.Wait(); |
||||
#pragma warning restore VSTHRD002
|
||||
} |
||||
catch (Exception) |
||||
{ |
||||
// do nothing
|
||||
} |
||||
|
||||
_cancellationTokenSource?.Dispose(); |
||||
|
||||
_canvasBitmap?.Dispose(); |
||||
} |
||||
|
||||
public override Task InitializeAsync(GraphicsEngineContext context, CancellationToken cancellationToken) |
||||
{ |
||||
try |
||||
{ |
||||
_startTime = TimeSpan.FromSeconds(scriptElement.StartSeconds ?? 0); |
||||
_endTime = _startTime + TimeSpan.FromSeconds(scriptElement.DurationSeconds ?? 0); |
||||
|
||||
// already past the time when this is supposed to play; don't do any more work
|
||||
if (_endTime < context.Seek) |
||||
{ |
||||
IsFinished = true; |
||||
return Task.CompletedTask; |
||||
} |
||||
|
||||
var options = new PipeOptions( |
||||
minimumSegmentSize: 1024 * 1024, |
||||
pauseWriterThreshold: 64 * 1024 * 1024, |
||||
resumeWriterThreshold: 32 * 1024 * 1024 |
||||
); |
||||
var pipe = new Pipe(options); |
||||
_pipeReader = pipe.Reader; |
||||
|
||||
_frameSize = context.FrameSize.Width * context.FrameSize.Height * 4; |
||||
|
||||
// default to bgra, but allow rgba when configured
|
||||
SKColorType pixelFormat = SKColorType.Bgra8888; |
||||
if (string.Equals(scriptElement.PixelFormat, "rgba", StringComparison.OrdinalIgnoreCase)) |
||||
{ |
||||
pixelFormat = SKColorType.Rgba8888; |
||||
} |
||||
|
||||
_canvasBitmap = new SKBitmap( |
||||
context.FrameSize.Width, |
||||
context.FrameSize.Height, |
||||
pixelFormat, |
||||
SKAlphaType.Unpremul); |
||||
|
||||
string json = JsonSerializer.Serialize(context.TemplateVariables); |
||||
|
||||
Command command = Cli.Wrap(scriptElement.Command) |
||||
.WithArguments(scriptElement.Arguments) |
||||
.WithWorkingDirectory(FileSystemLayout.TempFilePoolFolder) |
||||
.WithStandardInputPipe(PipeSource.FromString(json)) |
||||
.WithStandardOutputPipe(PipeTarget.ToStream(pipe.Writer.AsStream())); |
||||
|
||||
logger.LogDebug( |
||||
"script element command {Command} arguments {Arguments}", |
||||
command.TargetFilePath, |
||||
command.Arguments); |
||||
|
||||
_cancellationTokenSource = new CancellationTokenSource(); |
||||
var linkedToken = CancellationTokenSource.CreateLinkedTokenSource( |
||||
cancellationToken, |
||||
_cancellationTokenSource.Token); |
||||
|
||||
_commandTask = command.ExecuteAsync(linkedToken.Token); |
||||
|
||||
_ = _commandTask.Task.ContinueWith(_ => pipe.Writer.Complete(), TaskScheduler.Default); |
||||
} |
||||
catch (Exception ex) |
||||
{ |
||||
IsFinished = true; |
||||
logger.LogWarning(ex, "Failed to initialize motion element; will disable for this content"); |
||||
} |
||||
|
||||
return Task.CompletedTask; |
||||
} |
||||
|
||||
public override async ValueTask<Option<PreparedElementImage>> PrepareImage( |
||||
TimeSpan timeOfDay, |
||||
TimeSpan contentTime, |
||||
TimeSpan contentTotalTime, |
||||
TimeSpan channelTime, |
||||
CancellationToken cancellationToken) |
||||
{ |
||||
try |
||||
{ |
||||
if (contentTime < _startTime || contentTime > _endTime) |
||||
{ |
||||
return Option<PreparedElementImage>.None; |
||||
} |
||||
|
||||
while (true) |
||||
{ |
||||
ReadResult readResult = await _pipeReader.ReadAsync(cancellationToken); |
||||
ReadOnlySequence<byte> buffer = readResult.Buffer; |
||||
SequencePosition consumed = buffer.Start; |
||||
SequencePosition examined = buffer.End; |
||||
|
||||
try |
||||
{ |
||||
if (buffer.Length >= _frameSize) |
||||
{ |
||||
ReadOnlySequence<byte> sequence = buffer.Slice(0, _frameSize); |
||||
|
||||
using (SKPixmap pixmap = _canvasBitmap.PeekPixels()) |
||||
{ |
||||
sequence.CopyTo(pixmap.GetPixelSpan()); |
||||
} |
||||
|
||||
// mark this frame as consumed
|
||||
consumed = sequence.End; |
||||
|
||||
// we are done, return the frame
|
||||
return new PreparedElementImage(_canvasBitmap, SKPointI.Empty, 1.0f, ZIndex, false); |
||||
} |
||||
|
||||
if (readResult.IsCompleted) |
||||
{ |
||||
await _pipeReader.CompleteAsync(); |
||||
|
||||
return Option<PreparedElementImage>.None; |
||||
} |
||||
} |
||||
finally |
||||
{ |
||||
// advance the reader, consuming the processed frame and examining the entire buffer
|
||||
_pipeReader.AdvanceTo(consumed, examined); |
||||
} |
||||
} |
||||
} |
||||
catch (TaskCanceledException) |
||||
{ |
||||
return Option<PreparedElementImage>.None; |
||||
} |
||||
} |
||||
} |
||||
Loading…
Reference in new issue