mostly rewritten music, doesn't work on large number of concurrent streams properly >.>
This commit is contained in:
parent
6f29450aae
commit
cb5079a384
@ -3,6 +3,7 @@ using NadekoBot.Classes;
|
|||||||
using NadekoBot.Extensions;
|
using NadekoBot.Extensions;
|
||||||
using System;
|
using System;
|
||||||
using System.Diagnostics;
|
using System.Diagnostics;
|
||||||
|
using System.Diagnostics.Contracts;
|
||||||
using System.IO;
|
using System.IO;
|
||||||
using System.Linq;
|
using System.Linq;
|
||||||
using System.Text.RegularExpressions;
|
using System.Text.RegularExpressions;
|
||||||
@ -74,7 +75,7 @@ namespace NadekoBot.Modules.Music.Classes
|
|||||||
return this;
|
return this;
|
||||||
}
|
}
|
||||||
|
|
||||||
private Task BufferSong(CancellationToken cancelToken) =>
|
private Task BufferSong(string filename, CancellationToken cancelToken) =>
|
||||||
Task.Factory.StartNew(async () =>
|
Task.Factory.StartNew(async () =>
|
||||||
{
|
{
|
||||||
Process p = null;
|
Process p = null;
|
||||||
@ -89,33 +90,10 @@ namespace NadekoBot.Modules.Music.Classes
|
|||||||
RedirectStandardError = false,
|
RedirectStandardError = false,
|
||||||
CreateNoWindow = true,
|
CreateNoWindow = true,
|
||||||
});
|
});
|
||||||
const int blockSize = 3840;
|
using (var outStream = new FileStream(filename, FileMode.Append, FileAccess.Write, FileShare.Read))
|
||||||
var buffer = new byte[blockSize];
|
await p.StandardOutput.BaseStream.CopyToAsync(outStream, 81920, cancelToken);
|
||||||
var attempt = 0;
|
|
||||||
while (!cancelToken.IsCancellationRequested)
|
|
||||||
{
|
|
||||||
var read = 0;
|
|
||||||
try
|
|
||||||
{
|
|
||||||
read = await p.StandardOutput.BaseStream.ReadAsync(buffer, 0, blockSize, cancelToken)
|
|
||||||
.ConfigureAwait(false);
|
|
||||||
}
|
|
||||||
catch
|
|
||||||
{
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
if (read == 0)
|
|
||||||
if (attempt++ == 50)
|
|
||||||
break;
|
|
||||||
else
|
|
||||||
await Task.Delay(100, cancelToken).ConfigureAwait(false);
|
|
||||||
else
|
|
||||||
attempt = 0;
|
|
||||||
await songBuffer.WriteAsync(buffer, read, cancelToken).ConfigureAwait(false);
|
|
||||||
if (songBuffer.ContentLength > 2.MB())
|
|
||||||
prebufferingComplete = true;
|
prebufferingComplete = true;
|
||||||
}
|
}
|
||||||
}
|
|
||||||
catch (Exception ex)
|
catch (Exception ex)
|
||||||
{
|
{
|
||||||
Console.WriteLine($"Buffering errored: {ex.Message}");
|
Console.WriteLine($"Buffering errored: {ex.Message}");
|
||||||
@ -137,25 +115,23 @@ namespace NadekoBot.Modules.Music.Classes
|
|||||||
|
|
||||||
internal async Task Play(IAudioClient voiceClient, CancellationToken cancelToken)
|
internal async Task Play(IAudioClient voiceClient, CancellationToken cancelToken)
|
||||||
{
|
{
|
||||||
// initialize the buffer here because if this song was playing before (requeued), we must delete old buffer data
|
var filename = Path.Combine(MusicModule.MusicDataPath, DateTime.Now.UnixTimestamp().ToString());
|
||||||
songBuffer = new PoopyBuffer(NadekoBot.Config.BufferSize);
|
|
||||||
|
|
||||||
var bufferTask = BufferSong(cancelToken).ConfigureAwait(false);
|
var bufferTask = BufferSong(filename, cancelToken).ConfigureAwait(false);
|
||||||
var bufferAttempts = 0;
|
|
||||||
const int waitPerAttempt = 500;
|
var inStream = new FileStream(filename, FileMode.OpenOrCreate, FileAccess.Read, FileShare.Write);
|
||||||
var toAttemptTimes = SongInfo.ProviderType != MusicType.Normal ? 5 : 9;
|
try
|
||||||
while (!prebufferingComplete && bufferAttempts++ < toAttemptTimes)
|
|
||||||
{
|
{
|
||||||
await Task.Delay(waitPerAttempt, cancelToken).ConfigureAwait(false);
|
await Task.Delay(1000);
|
||||||
}
|
|
||||||
Console.WriteLine($"Prebuffering done? in {waitPerAttempt * bufferAttempts}");
|
|
||||||
const int blockSize = 3840;
|
const int blockSize = 3840;
|
||||||
var attempt = 0;
|
var attempt = 0;
|
||||||
while (!cancelToken.IsCancellationRequested)
|
while (!cancelToken.IsCancellationRequested)
|
||||||
{
|
{
|
||||||
//Console.WriteLine($"Read: {songBuffer.ReadPosition}\nWrite: {songBuffer.WritePosition}\nContentLength:{songBuffer.ContentLength}\n---------");
|
//Console.WriteLine($"Read: {songBuffer.ReadPosition}\nWrite: {songBuffer.WritePosition}\nContentLength:{songBuffer.ContentLength}\n---------");
|
||||||
byte[] buffer = new byte[blockSize];
|
byte[] buffer = new byte[blockSize];
|
||||||
var read = await songBuffer.ReadAsync(buffer, blockSize).ConfigureAwait(false);
|
var read = inStream.Read(buffer, 0, buffer.Length);
|
||||||
|
//await inStream.CopyToAsync(voiceClient.OutputStream);
|
||||||
unchecked
|
unchecked
|
||||||
{
|
{
|
||||||
bytesSent += (ulong)read;
|
bytesSent += (ulong)read;
|
||||||
@ -163,8 +139,9 @@ namespace NadekoBot.Modules.Music.Classes
|
|||||||
if (read == 0)
|
if (read == 0)
|
||||||
if (attempt++ == 20)
|
if (attempt++ == 20)
|
||||||
{
|
{
|
||||||
|
Console.WriteLine("blocking");
|
||||||
voiceClient.Wait();
|
voiceClient.Wait();
|
||||||
Console.WriteLine($"Song finished. [{songBuffer.ContentLength}]");
|
Console.WriteLine("unblocking");
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
@ -177,13 +154,17 @@ namespace NadekoBot.Modules.Music.Classes
|
|||||||
buffer = AdjustVolume(buffer, MusicPlayer.Volume);
|
buffer = AdjustVolume(buffer, MusicPlayer.Volume);
|
||||||
voiceClient.Send(buffer, 0, read);
|
voiceClient.Send(buffer, 0, read);
|
||||||
}
|
}
|
||||||
Console.WriteLine("Awiting buffer task");
|
|
||||||
await bufferTask;
|
await bufferTask;
|
||||||
Console.WriteLine("Buffer task done.");
|
|
||||||
voiceClient.Clear();
|
voiceClient.Clear();
|
||||||
cancelToken.ThrowIfCancellationRequested();
|
cancelToken.ThrowIfCancellationRequested();
|
||||||
}
|
}
|
||||||
|
finally {
|
||||||
|
inStream.Dispose();
|
||||||
|
try { File.Delete(filename); } catch { }
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
//stackoverflow ftw
|
//stackoverflow ftw
|
||||||
private static byte[] AdjustVolume(byte[] audioSamples, float volume)
|
private static byte[] AdjustVolume(byte[] audioSamples, float volume)
|
||||||
{
|
{
|
||||||
@ -210,6 +191,33 @@ namespace NadekoBot.Modules.Music.Classes
|
|||||||
}
|
}
|
||||||
return array;
|
return array;
|
||||||
}
|
}
|
||||||
|
*/
|
||||||
|
|
||||||
|
//aidiakapi ftw
|
||||||
|
public unsafe static byte[] AdjustVolume(byte[] audioSamples, float volume)
|
||||||
|
{
|
||||||
|
Contract.Requires(audioSamples != null);
|
||||||
|
Contract.Requires(audioSamples.Length % 2 == 0);
|
||||||
|
Contract.Requires(volume >= 0f && volume <= 1f);
|
||||||
|
Contract.Assert(BitConverter.IsLittleEndian);
|
||||||
|
|
||||||
|
if (Math.Abs(volume - 1f) < 0.0001f) return audioSamples;
|
||||||
|
|
||||||
|
// 16-bit precision for the multiplication
|
||||||
|
int volumeFixed = (int)Math.Round(volume * 65536d);
|
||||||
|
|
||||||
|
int count = audioSamples.Length / 2;
|
||||||
|
|
||||||
|
fixed (byte* srcBytes = audioSamples)
|
||||||
|
{
|
||||||
|
short* src = (short*)srcBytes;
|
||||||
|
|
||||||
|
for (int i = count; i != 0; i--, src++)
|
||||||
|
*src = (short)(((*src) * volumeFixed) >> 16);
|
||||||
|
}
|
||||||
|
|
||||||
|
return audioSamples;
|
||||||
|
}
|
||||||
|
|
||||||
public static async Task<Song> ResolveSong(string query, MusicType musicType = MusicType.Normal)
|
public static async Task<Song> ResolveSong(string query, MusicType musicType = MusicType.Normal)
|
||||||
{
|
{
|
||||||
|
@ -18,11 +18,16 @@ namespace NadekoBot.Modules.Music
|
|||||||
{
|
{
|
||||||
internal class MusicModule : DiscordModule
|
internal class MusicModule : DiscordModule
|
||||||
{
|
{
|
||||||
|
|
||||||
public static ConcurrentDictionary<Server, MusicPlayer> MusicPlayers = new ConcurrentDictionary<Server, MusicPlayer>();
|
public static ConcurrentDictionary<Server, MusicPlayer> MusicPlayers = new ConcurrentDictionary<Server, MusicPlayer>();
|
||||||
|
|
||||||
|
public const string MusicDataPath = "data/musicdata";
|
||||||
|
|
||||||
public MusicModule()
|
public MusicModule()
|
||||||
{
|
{
|
||||||
|
//it can fail if its currenctly opened or doesn't exist. Either way i don't care
|
||||||
|
try { Directory.Delete(MusicDataPath, true); } catch { }
|
||||||
|
|
||||||
|
Directory.CreateDirectory(MusicDataPath);
|
||||||
}
|
}
|
||||||
|
|
||||||
public override string Prefix { get; } = NadekoBot.Config.CommandPrefixes.Music;
|
public override string Prefix { get; } = NadekoBot.Config.CommandPrefixes.Music;
|
||||||
|
Loading…
Reference in New Issue
Block a user