TTS messages are now merged.

This commit is contained in:
Tom
2024-08-12 05:44:31 +00:00
parent 9f61295d2d
commit 2056b2cd48
12 changed files with 614 additions and 523 deletions

View File

@ -1,198 +1,218 @@
using NAudio.Wave;
using TwitchChatTTS.Twitch.Socket.Messages;
public class TTSPlayer
namespace TwitchChatTTS.Chat.Soeech
{
private readonly PriorityQueue<TTSMessage, int> _messages; // ready to play
private readonly PriorityQueue<TTSMessage, int> _buffer;
private readonly Mutex _mutex;
private readonly Mutex _mutex2;
public TTSMessage? Playing { get; set; }
public TTSPlayer()
public class TTSPlayer
{
_messages = new PriorityQueue<TTSMessage, int>(new DescendingOrder());
_buffer = new PriorityQueue<TTSMessage, int>(new DescendingOrder());
_mutex = new Mutex();
_mutex2 = new Mutex();
}
private readonly PriorityQueue<TTSGroupedMessage, int> _messages; // ready to play
private readonly PriorityQueue<TTSGroupedMessage, int> _buffer;
private readonly Mutex _mutex;
private readonly Mutex _mutex2;
public void Add(TTSMessage message)
{
try
//public TTSGroupedMessage? PlayingGroup { get; set; }
public TTSGroupedMessage? Playing { get; set; }
public TTSPlayer()
{
_mutex2.WaitOne();
_buffer.Enqueue(message, message.Priority);
_messages = new PriorityQueue<TTSGroupedMessage, int>(new DescendingOrder());
_buffer = new PriorityQueue<TTSGroupedMessage, int>(new DescendingOrder());
_mutex = new Mutex();
_mutex2 = new Mutex();
}
finally
{
_mutex2.ReleaseMutex();
}
}
public TTSMessage? ReceiveReady()
{
try
public void Add(TTSGroupedMessage message, int priority)
{
_mutex.WaitOne();
if (_messages.TryDequeue(out TTSMessage? message, out int _))
try
{
return message;
_mutex2.WaitOne();
_buffer.Enqueue(message, priority);
}
return null;
}
finally
{
_mutex.ReleaseMutex();
}
}
public TTSMessage? ReceiveBuffer()
{
try
{
_mutex2.WaitOne();
if (_buffer.TryDequeue(out TTSMessage? message, out int _))
finally
{
return message;
_mutex2.ReleaseMutex();
}
return null;
}
finally
{
_mutex2.ReleaseMutex();
}
}
public void Ready(TTSMessage message)
{
try
{
_mutex.WaitOne();
_messages.Enqueue(message, message.Priority);
}
finally
{
_mutex.ReleaseMutex();
}
}
public void RemoveAll()
{
try
{
_mutex2.WaitOne();
_buffer.Clear();
}
finally
{
_mutex2.ReleaseMutex();
}
try
public TTSGroupedMessage? ReceiveReady()
{
_mutex.WaitOne();
_messages.Clear();
}
finally
{
_mutex.ReleaseMutex();
}
}
public void RemoveAll(long broadcasterId, long chatterId)
{
try
{
_mutex2.WaitOne();
if (_buffer.UnorderedItems.Any(i => i.Element.RoomId == broadcasterId && i.Element.ChatterId == chatterId))
try
{
var list = _buffer.UnorderedItems.Where(i => i.Element.RoomId == broadcasterId && i.Element.ChatterId != chatterId).ToArray();
_mutex.WaitOne();
if (_messages.TryDequeue(out TTSGroupedMessage? message, out int _))
{
return message;
}
return null;
}
finally
{
_mutex.ReleaseMutex();
}
}
public TTSGroupedMessage? ReceiveBuffer()
{
try
{
_mutex2.WaitOne();
if (_buffer.TryDequeue(out TTSGroupedMessage? messages, out int _))
{
return messages;
}
return null;
}
finally
{
_mutex2.ReleaseMutex();
}
}
public void Ready(TTSGroupedMessage message)
{
try
{
_mutex.WaitOne();
_messages.Enqueue(message, message.Priority);
}
finally
{
_mutex.ReleaseMutex();
}
}
public void RemoveAll()
{
try
{
_mutex2.WaitOne();
_buffer.Clear();
foreach (var item in list)
_buffer.Enqueue(item.Element, item.Element.Priority);
}
}
finally
{
_mutex2.ReleaseMutex();
}
try
{
_mutex.WaitOne();
if (_messages.UnorderedItems.Any(i => i.Element.RoomId == broadcasterId && i.Element.ChatterId == chatterId))
finally
{
var list = _messages.UnorderedItems.Where(i => i.Element.RoomId == broadcasterId && i.Element.ChatterId != chatterId).ToArray();
_mutex2.ReleaseMutex();
}
try
{
_mutex.WaitOne();
_messages.Clear();
foreach (var item in list)
_messages.Enqueue(item.Element, item.Element.Priority);
}
}
finally
{
_mutex.ReleaseMutex();
}
}
public void RemoveMessage(string messageId)
{
try
{
_mutex2.WaitOne();
if (_buffer.UnorderedItems.Any(i => i.Element.MessageId == messageId))
finally
{
var list = _buffer.UnorderedItems.Where(i => i.Element.MessageId != messageId).ToArray();
_buffer.Clear();
foreach (var item in list)
_buffer.Enqueue(item.Element, item.Element.Priority);
return;
_mutex.ReleaseMutex();
}
}
finally
{
_mutex2.ReleaseMutex();
}
try
public void RemoveAll(long broadcasterId, long chatterId)
{
_mutex.WaitOne();
if (_messages.UnorderedItems.Any(i => i.Element.MessageId == messageId))
try
{
var list = _messages.UnorderedItems.Where(i => i.Element.MessageId != messageId).ToArray();
_messages.Clear();
foreach (var item in list)
_messages.Enqueue(item.Element, item.Element.Priority);
_mutex2.WaitOne();
if (_buffer.UnorderedItems.Any(i => i.Element.RoomId == broadcasterId && i.Element.ChatterId == chatterId))
{
var list = _buffer.UnorderedItems.Where(i => i.Element.RoomId == broadcasterId && i.Element.ChatterId != chatterId).ToArray();
_buffer.Clear();
foreach (var item in list)
_buffer.Enqueue(item.Element, item.Element.Priority);
}
}
finally
{
_mutex2.ReleaseMutex();
}
try
{
_mutex.WaitOne();
if (_messages.UnorderedItems.Any(i => i.Element.RoomId == broadcasterId && i.Element.ChatterId == chatterId))
{
var list = _messages.UnorderedItems.Where(i => i.Element.RoomId == broadcasterId && i.Element.ChatterId != chatterId).ToArray();
_messages.Clear();
foreach (var item in list)
_messages.Enqueue(item.Element, item.Element.Priority);
}
}
finally
{
_mutex.ReleaseMutex();
}
}
finally
public void RemoveMessage(string messageId)
{
_mutex.ReleaseMutex();
try
{
_mutex2.WaitOne();
if (_buffer.UnorderedItems.Any(i => i.Element.MessageId == messageId))
{
var list = _buffer.UnorderedItems.Where(i => i.Element.MessageId != messageId).ToArray();
_buffer.Clear();
foreach (var item in list)
_buffer.Enqueue(item.Element, item.Element.Priority);
return;
}
}
finally
{
_mutex2.ReleaseMutex();
}
try
{
_mutex.WaitOne();
if (_messages.UnorderedItems.Any(i => i.Element.MessageId == messageId))
{
var list = _messages.UnorderedItems.Where(i => i.Element.MessageId != messageId).ToArray();
_messages.Clear();
foreach (var item in list)
_messages.Enqueue(item.Element, item.Element.Priority);
}
}
finally
{
_mutex.ReleaseMutex();
}
}
public bool IsEmpty()
{
return _messages.Count == 0;
}
private class DescendingOrder : IComparer<int>
{
public int Compare(int x, int y) => y.CompareTo(x);
}
}
public bool IsEmpty()
public class TTSMessage
{
return _messages.Count == 0;
public string? Voice { get; set; }
public string? Message { get; set; }
public string? File { get; set; }
}
private class DescendingOrder : IComparer<int>
public class TTSGroupedMessage
{
public int Compare(int x, int y) => y.CompareTo(x);
}
}
public long RoomId { get; set; }
public long ChatterId { get; set; }
public string MessageId { get; set; }
public DateTime Timestamp { get; set; }
public int Priority { get; set; }
public IList<TTSMessage> Messages { get; set; }
//public IList<ISampleProvider> Audios { get; set; }
public ISampleProvider? Audio { get; set; }
public class TTSMessage
{
public string? Voice { get; set; }
public long RoomId { get; set; }
public long ChatterId { get; set; }
public string MessageId { get; set; }
public string? Message { get; set; }
public string? File { get; set; }
public DateTime Timestamp { get; set; }
public IEnumerable<TwitchBadge> Badges { get; set; }
public int Priority { get; set; }
public ISampleProvider? Audio { get; set; }
public TTSGroupedMessage(long broadcasterId, long chatterId, string messageId, IList<TTSMessage> messages, DateTime timestamp, int priority)
{
RoomId = broadcasterId;
ChatterId = chatterId;
MessageId = messageId;
Messages = messages;
Timestamp = timestamp;
Priority = priority;
//Audios = new List<ISampleProvider>();
}
}
}