Unreviewed, add myself to some watch lists.
[WebKit-https.git] / Source / WebCore / platform / mediarecorder / cocoa / MediaRecorderPrivateWriterCocoa.h
1 /*
2  * Copyright (C) 2018 Apple Inc. All rights reserved.
3  *
4  * Redistribution and use in source and binary forms, with or without
5  * modification, are permitted provided that the following conditions
6  * are met:
7  * 1.  Redistributions of source code must retain the above copyright
8  *     notice, this list of conditions and the following disclaimer.
9  * 2.  Redistributions in binary form must reproduce the above copyright
10  *     notice, this list of conditions and the following disclaimer in the
11  *     documentation and/or other materials provided with the distribution.
12  *
13  * THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS'' AND ANY
14  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
15  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
16  * DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS BE LIABLE FOR ANY
17  * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
18  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
19  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON
20  * ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
21  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
22  * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
23  */
24
25 #pragma once
26
27 #if ENABLE(MEDIA_STREAM)
28
29 #include "SharedBuffer.h"
30 #include <wtf/Deque.h>
31 #include <wtf/Lock.h>
32 #include <wtf/RetainPtr.h>
33 #include <wtf/ThreadSafeRefCounted.h>
34 #include <wtf/threads/BinarySemaphore.h>
35
36 typedef struct opaqueCMSampleBuffer *CMSampleBufferRef;
37
38 OBJC_CLASS AVAssetWriter;
39 OBJC_CLASS AVAssetWriterInput;
40
41 namespace WTF {
42 class MediaTime;
43 }
44
45 namespace WebCore {
46
47 class AudioStreamDescription;
48 class PlatformAudioData;
49
50 class MediaRecorderPrivateWriter : public ThreadSafeRefCounted<MediaRecorderPrivateWriter, WTF::DestructionThread::Main> {
51 public:
52     MediaRecorderPrivateWriter() = default;
53     
54     bool setupWriter();
55     bool setVideoInput(int width, int height);
56     bool setAudioInput();
57     void appendVideoSampleBuffer(CMSampleBufferRef);
58     void appendAudioSampleBuffer(const PlatformAudioData&, const AudioStreamDescription&, const WTF::MediaTime&, size_t);
59     void stopRecording();
60     RefPtr<SharedBuffer> fetchData();
61
62 private:
63     RetainPtr<AVAssetWriter> m_writer;
64     RetainPtr<AVAssetWriterInput> m_videoInput;
65     RetainPtr<AVAssetWriterInput> m_audioInput;
66
67     String m_path;
68     Lock m_videoLock;
69     Lock m_audioLock;
70     BinarySemaphore m_finishWritingSemaphore;
71     BinarySemaphore m_finishWritingAudioSemaphore;
72     BinarySemaphore m_finishWritingVideoSemaphore;
73     bool m_hasStartedWriting { false };
74     bool m_isStopped { false };
75     bool m_isFirstAudioSample { true };
76     dispatch_queue_t m_audioPullQueue;
77     dispatch_queue_t m_videoPullQueue;
78     Deque<RetainPtr<CMSampleBufferRef>> m_videoBufferPool;
79     Deque<RetainPtr<CMSampleBufferRef>> m_audioBufferPool;
80 };
81
82 } // namespace WebCore
83
84 #endif // ENABLE(MEDIA_STREAM)