464745a3ac6666dddd61a8177c8781ccea68a6fb
[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>, public CanMakeWeakPtr<MediaRecorderPrivateWriter> {
51 public:
52     MediaRecorderPrivateWriter() = default;
53     ~MediaRecorderPrivateWriter();
54     
55     bool setupWriter();
56     bool setVideoInput(int width, int height);
57     bool setAudioInput();
58     void appendVideoSampleBuffer(CMSampleBufferRef);
59     void appendAudioSampleBuffer(const PlatformAudioData&, const AudioStreamDescription&, const WTF::MediaTime&, size_t);
60     void stopRecording();
61     RefPtr<SharedBuffer> fetchData();
62     
63 private:
64     void clear();
65
66     RetainPtr<AVAssetWriter> m_writer;
67     RetainPtr<AVAssetWriterInput> m_videoInput;
68     RetainPtr<AVAssetWriterInput> m_audioInput;
69
70     String m_path;
71     Lock m_videoLock;
72     Lock m_audioLock;
73     BinarySemaphore m_finishWritingSemaphore;
74     BinarySemaphore m_finishWritingAudioSemaphore;
75     BinarySemaphore m_finishWritingVideoSemaphore;
76     bool m_hasStartedWriting { false };
77     bool m_isStopped { false };
78     bool m_isFirstAudioSample { true };
79     dispatch_queue_t m_audioPullQueue;
80     dispatch_queue_t m_videoPullQueue;
81     Deque<RetainPtr<CMSampleBufferRef>> m_videoBufferPool;
82     Deque<RetainPtr<CMSampleBufferRef>> m_audioBufferPool;
83 };
84
85 } // namespace WebCore
86
87 #endif // ENABLE(MEDIA_STREAM)