06eb4684d891cb4dd06c7173e7fe86536a2d90a5
[WebKit-https.git] / Source / WebCore / platform / graphics / avfoundation / objc / MediaSampleAVFObjC.h
1 /*
2  * Copyright (C) 2016 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. ``AS IS'' AND ANY
14  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
15  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
16  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL APPLE INC. OR
17  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
18  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
19  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
20  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
21  * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
22  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
23  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
24  */
25
26 #pragma once
27
28 #include "MediaSample.h"
29 #include <JavaScriptCore/Uint8ClampedArray.h>
30 #include <pal/avfoundation/MediaTimeAVFoundation.h>
31 #include <wtf/Forward.h>
32
33 namespace WebCore {
34
35 class MediaSampleAVFObjC : public MediaSample {
36 public:
37     static Ref<MediaSampleAVFObjC> create(CMSampleBufferRef sample, int trackID) { return adoptRef(*new MediaSampleAVFObjC(sample, trackID)); }
38     static Ref<MediaSampleAVFObjC> create(CMSampleBufferRef sample, AtomicString trackID) { return adoptRef(*new MediaSampleAVFObjC(sample, trackID)); }
39     static Ref<MediaSampleAVFObjC> create(CMSampleBufferRef sample, VideoRotation rotation = VideoRotation::None, bool mirrored = false) { return adoptRef(*new MediaSampleAVFObjC(sample, rotation, mirrored)); }
40     static RefPtr<MediaSampleAVFObjC> createImageSample(Vector<uint8_t>&&, unsigned long width, unsigned long height);
41
42     RefPtr<JSC::Uint8ClampedArray> getRGBAImageData() const override;
43
44     MediaTime presentationTime() const override;
45     MediaTime outputPresentationTime() const override;
46     MediaTime decodeTime() const override;
47     MediaTime duration() const override;
48     MediaTime outputDuration() const override;
49
50     AtomicString trackID() const override { return m_id; }
51     void setTrackID(const String& id) override { m_id = id; }
52
53     size_t sizeInBytes() const override;
54     FloatSize presentationSize() const override;
55
56     SampleFlags flags() const override;
57     PlatformSample platformSample() override;
58     void dump(PrintStream&) const override;
59     void offsetTimestampsBy(const MediaTime&) override;
60     void setTimestamps(const MediaTime&, const MediaTime&) override;
61     bool isDivisable() const override;
62     std::pair<RefPtr<MediaSample>, RefPtr<MediaSample>> divide(const MediaTime& presentationTime) override;
63     Ref<MediaSample> createNonDisplayingCopy() const override;
64
65     VideoRotation videoRotation() const override { return m_rotation; }
66     bool videoMirrored() const override { return m_mirrored; }
67     uint32_t videoPixelFormat() const final;
68
69     CMSampleBufferRef sampleBuffer() const { return m_sample.get(); }
70
71     String toJSONString() const override;
72
73 protected:
74     MediaSampleAVFObjC(RetainPtr<CMSampleBufferRef>&& sample)
75         : m_sample(WTFMove(sample))
76     {
77     }
78     MediaSampleAVFObjC(CMSampleBufferRef sample)
79         : m_sample(sample)
80     {
81     }
82     MediaSampleAVFObjC(CMSampleBufferRef sample, AtomicString trackID)
83         : m_sample(sample)
84         , m_id(trackID)
85     {
86     }
87     MediaSampleAVFObjC(CMSampleBufferRef sample, int trackID)
88         : m_sample(sample)
89         , m_id(AtomicString::number(trackID))
90     {
91     }
92     MediaSampleAVFObjC(CMSampleBufferRef sample, VideoRotation rotation, bool mirrored)
93         : m_sample(sample)
94         , m_rotation(rotation)
95         , m_mirrored(mirrored)
96     {
97     }
98
99     virtual ~MediaSampleAVFObjC() = default;
100     RetainPtr<CMSampleBufferRef> m_sample;
101     AtomicString m_id;
102     VideoRotation m_rotation { VideoRotation::None };
103     bool m_mirrored { false };
104 };
105
106 } // namespace WebCore
107