Austin/BililiveRecorder
1
0
Fork 0
mirror of https://github.com/BililiveRecorder/BililiveRecorder.git synced 2024-11-16 03:32:20 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

~

Browse Source
This commit is contained in:
Genteure 2018-03-19 17:35:27 +08:00
parent 0abe31752f
commit bab05ec0d7
2 changed files with 55 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
BililiveRecorder.FlvProcessor/FlvClipProcessor.cs
View File

@ -15,7 +15,7 @@ namespace BililiveRecorder.FlvProcessor
{
Header = header;
Tags = past;
target = Tags[Tags.Count - 1].TimeStamp + future;
target = Tags[Tags.Count - 1].TimeStamp + (future * FlvStreamProcessor.SEC_TO_MS);
}
public void AddTag(FlvTag tag)

65
BililiveRecorder.FlvProcessor/FlvStreamProcessor.cs
View File

@ -10,7 +10,8 @@ namespace BililiveRecorder.FlvProcessor
{
public class FlvStreamProcessor : IDisposable
{
private const int MIN_BUFFER_SIZE = 1024 * 2;
internal const int SEC_TO_MS = 1000; // 1 second = 1000 ms
internal const int MIN_BUFFER_SIZE = 1024 * 2;
internal static readonly byte[] FLV_HEADER_BYTES = new byte[]
{
0x46, // F
@ -22,10 +23,10 @@ namespace BililiveRecorder.FlvProcessor
0x00, // |
0x00, // |
0x09, // total of 9 bytes
0x00, // ---
0x00, // |
0x00, // |
0x00, // the "0th" tag has a length of 0
// 0x00, // ---
// 0x00, // |
// 0x00, // |
// 0x00, // the "0th" tag has a length of 0
};
public RecordInfo Info; // not used for now.
@ -33,6 +34,9 @@ namespace BililiveRecorder.FlvProcessor
public event TagProcessedEvent TagProcessed;
public event StreamFinalizedEvent StreamFinalized;
public int Clip_Past = 90;
public int Clip_Future = 30;
private bool _headerParsed = false;
private readonly List<FlvTag> Tags = new List<FlvTag>();
private readonly MemoryStream _buffer = new MemoryStream();
@ -43,7 +47,11 @@ namespace BililiveRecorder.FlvProcessor
private readonly FileStream _fs;
public int MaxTimeStamp { get; private set; }
public int MaxTimeStamp { get; private set; } = -1;
public int BaseTimeStamp { get; private set; } = 0;
public int TagVideoCount { get; private set; } = 0;
public int TagAudioCount { get; private set; } = 0;
private bool hasOffset = false;
public FlvStreamProcessor(RecordInfo info, string path)
{
@ -132,14 +140,45 @@ namespace BililiveRecorder.FlvProcessor
}
else
{
tag.TimeStamp -= 0; // TODO: 修复时间戳
Tags.Add(tag);
// TODO: remove old tag
switch (tag.TagType)
{
case TagType.AUDIO:
TagAudioCount++;
if (!hasOffset && TagAudioCount == 2)
{
hasOffset = true;
BaseTimeStamp = tag.TimeStamp;
Debug.Write(string.Format("Reseting to this base timestamp {0} ms\n", BaseTimeStamp));
tag.TimeStamp = 0;
MaxTimeStamp = 0;
}
break;
case TagType.VIDEO:
TagVideoCount++;
if (!hasOffset && TagVideoCount == 2)
{
hasOffset = true;
BaseTimeStamp = tag.TimeStamp;
Debug.Write(string.Format("Reseting to this base timestamp {0} ms\n", BaseTimeStamp));
tag.TimeStamp = 0;
MaxTimeStamp = 0;
}
break;
case TagType.DATA:
default:
break;
}
tag.TimeStamp -= BaseTimeStamp; // 修复时间戳
MaxTimeStamp = Math.Max(MaxTimeStamp, tag.TimeStamp);
Tags.Add(tag); // Clip 缓存
Tags.Where(x => (MaxTimeStamp - x.TimeStamp) > (Clip_Past * SEC_TO_MS)).Any(x => Tags.Remove(x)); // 移除过旧的数据
// 写入硬盘
var b = tag.ToBytes();
_fs.Write(b, 0, b.Length);
_fs.Write(tag.Data, 0, tag.Data.Length);
_fs.Write(BitConverter.GetBytes(tag.Data.Length + b.Length).ToBE(), 0, 4);
_fs.Write(BitConverter.GetBytes(tag.Data.Length + b.Length).ToBE(), 0, 4); // Last Tag Size
TagProcessed?.Invoke(this, new TagProcessedArgs() { Tag = tag });
}
@ -147,12 +186,16 @@ namespace BililiveRecorder.FlvProcessor
private void _ParseTag(byte[] data)
{
byte[] b = { 0, 0, 0, 0, };
byte[] b = new byte[4];
_buffer.Write(data, 0, data.Length);
long dataLen = _buffer.Position;
_buffer.Position = 0;
FlvTag tag = new FlvTag();
// Previous Tag Size
_buffer.Read(b, 0, 4);
b = new byte[4];
// TagType UI8
tag.TagType = (TagType)_buffer.ReadByte();
Debug.Write(string.Format("Tag Type: {0}\n", tag.TagType));