private void DoUpload(int iterations, int timeout, Func <uint, uint, byte[]> frameBytes)
{
UploadJobWorker worker = null;
AtemMockServerWrapper.Each(_output, _pool, (a, b) => worker?.HandleCommand(a, b), DeviceTestCases.MediaPlayer, helper =>
{
helper.DisposeSdkClient = true;
IBMDSwitcherMediaPool pool = GetMediaPool(helper);
IBMDSwitcherStills stills = GetStillsPool(helper);
for (int i = 0; i < iterations; i++)
{
AtemState stateBefore = helper.Helper.BuildLibState();
Tuple <uint, uint> resolution = stateBefore.Settings.VideoMode.GetResolution().GetSize();
uint index = Randomiser.RangeInt((uint)stateBefore.MediaPool.Stills.Count);
string name = Guid.NewGuid().ToString();
worker = new UploadJobWorker(resolution.Item1 * resolution.Item2 * 4, _output,
(uint)MediaPoolFileType.Still, index, DataTransferUploadRequestCommand.TransferMode.Write);
var cb = new LockCallback();
helper.SendAndWaitForChange(stateBefore, () => { stills.Lock(cb); });
Assert.True(cb.Wait.WaitOne(2000));
pool.CreateFrame(_BMDSwitcherPixelFormat.bmdSwitcherPixelFormat10BitYUVA, resolution.Item1,
resolution.Item2, out IBMDSwitcherFrame frame);
byte[] bytes = frameBytes(resolution.Item1, resolution.Item2);
if (bytes.Length > 0)
{
MediaPoolUtil.FillSdkFrame(frame, bytes);
}
var stillState = stateBefore.MediaPool.Stills[(int)index];
stillState.IsUsed = true;
stillState.Filename = name;
var uploadCb = new TransferCallback();
stills.AddCallback(uploadCb);
stills.Upload(index, name, frame);
helper.HandleUntil(uploadCb.Wait, timeout);
Assert.True(uploadCb.Wait.WaitOne(500));
Assert.Equal(_BMDSwitcherMediaPoolEventType.bmdSwitcherMediaPoolEventTypeTransferCompleted,
uploadCb.Result);
Assert.Equal(BitConverter.ToString(bytes), BitConverter.ToString(worker.Buffer));
helper.SendAndWaitForChange(stateBefore, () =>
{
stills.Unlock(cb);
});
}
});
}
public void TestAbortingStillUpload()
{
AbortedUploadJobWorker worker = null;
AtemMockServerWrapper.Each(_output, _pool, (a, b) => worker?.HandleCommand(a, b), DeviceTestCases.MediaPlayerStillTransfer, helper =>
{
helper.DisposeSdkClient = true;
IBMDSwitcherMediaPool pool = GetMediaPool(helper);
IBMDSwitcherStills stills = GetStillsPool(helper);
for (int i = 0; i < 3; i++)
{
AtemState stateBefore = helper.Helper.BuildLibState();
Tuple <uint, uint> resolution = stateBefore.Settings.VideoMode.GetResolution().GetSize();
uint index = Randomiser.RangeInt((uint)stateBefore.MediaPool.Stills.Count);
string name = Guid.NewGuid().ToString();
worker = new AbortedUploadJobWorker(_output);
var cb = new LockCallback();
helper.SendAndWaitForChange(stateBefore, () => { stills.Lock(cb); });
Assert.True(cb.Wait.WaitOne(2000));
pool.CreateFrame(_BMDSwitcherPixelFormat.bmdSwitcherPixelFormat10BitYUVA, resolution.Item1,
resolution.Item2, out IBMDSwitcherFrame frame);
MediaPoolUtil.FillSdkFrame(frame, MediaPoolUtil.RandomFrame(resolution.Item1 * resolution.Item2));
var uploadCb = new TransferCallback();
stills.AddCallback(uploadCb);
stills.Upload(index, name, frame);
// Short bit of work before the abort
helper.HandleUntil(uploadCb.Wait, 1000);
stills.CancelTransfer();
helper.HandleUntil(uploadCb.Wait, 1000);
Assert.True(uploadCb.Wait.WaitOne(500));
Assert.Equal(_BMDSwitcherMediaPoolEventType.bmdSwitcherMediaPoolEventTypeTransferCancelled,
uploadCb.Result);
helper.SendAndWaitForChange(stateBefore, () =>
{
stills.Unlock(cb);
});
}
});
}
private void StartUploadWithLock()
{
dispatcher.Invoke(() =>
{
if (m_stillsmonitor != null)
{
m_stills.RemoveCallback(m_stillsmonitor);
}
m_stillsmonitor = new StillsMonitor()
{
OnTransferCompleted = OnTransferCompleted, OnTransferFailled = OnTransferFailled
};
m_stills.AddCallback(m_stillsmonitor);
m_stills.Upload(m_slot, m_imgname, m_frame);
});
}