OLD | NEW |
| (Empty) |
1 /* | |
2 * Copyright (C) 2007, 2008, 2009, 2010 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 COMPUTER, 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 COMPUTER, 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 #include "config.h" | |
26 | |
27 #include "QTPixelBuffer.h" | |
28 | |
29 #include <CFNumber.h> | |
30 #include <CFString.h> | |
31 #include <CGColorSpace.h> | |
32 #include <CGImage.h> | |
33 #include <CVPixelBuffer.h> | |
34 #include <QuickDraw.h> | |
35 #include <memory.h> | |
36 | |
37 static OSStatus SetNumberValue(CFMutableDictionaryRef inDict, CFStringRef inKey,
SInt32 inValue) | |
38 { | |
39 CFNumberRef number; | |
40 | |
41 number = CFNumberCreate(kCFAllocatorDefault, kCFNumberSInt32Type, &inValue); | |
42 if (!number) | |
43 return coreFoundationUnknownErr; | |
44 | |
45 CFDictionarySetValue(inDict, inKey, number); | |
46 CFRelease(number); | |
47 | |
48 return noErr; | |
49 } | |
50 | |
51 CFDictionaryRef QTPixelBuffer::createPixelBufferAttributesDictionary(QTPixelBuff
er::Type contextType) | |
52 { | |
53 static const CFStringRef kDirect3DCompatibilityKey = CFSTR("Direct3DCompatib
ility"); | |
54 | |
55 CFMutableDictionaryRef pixelBufferAttributes = CFDictionaryCreateMutable(kCF
AllocatorDefault, 0, &kCFTypeDictionaryKeyCallBacks, &kCFTypeDictionaryValueCall
Backs); | |
56 if (contextType == QTPixelBuffer::ConfigureForCAImageQueue) { | |
57 // Ask for D3D compatible pixel buffers so no further work is needed. | |
58 CFDictionarySetValue(pixelBufferAttributes, kDirect3DCompatibilityKey, k
CFBooleanTrue); | |
59 } else { | |
60 // Use the k32BGRAPixelFormat, as QuartzCore will be able to use the pix
els directly, | |
61 // without needing an additional copy or rendering pass. | |
62 SetNumberValue(pixelBufferAttributes, kCVPixelBufferPixelFormatTypeKey,
k32BGRAPixelFormat); | |
63 | |
64 // Set kCVPixelBufferBytesPerRowAlignmentKey to 16 to ensure that each r
ow of pixels | |
65 // starts at a 16 byte aligned address for most efficient data reading. | |
66 SetNumberValue(pixelBufferAttributes, kCVPixelBufferBytesPerRowAlignment
Key, 16); | |
67 CFDictionarySetValue(pixelBufferAttributes, kCVPixelBufferCGImageCompati
bilityKey, kCFBooleanTrue); | |
68 } | |
69 return pixelBufferAttributes; | |
70 } | |
71 | |
72 QTPixelBuffer::QTPixelBuffer() | |
73 : m_pixelBuffer(0) | |
74 { | |
75 } | |
76 | |
77 QTPixelBuffer::QTPixelBuffer(const QTPixelBuffer& p) | |
78 : m_pixelBuffer(p.m_pixelBuffer) | |
79 { | |
80 CVPixelBufferRetain(m_pixelBuffer); | |
81 } | |
82 | |
83 QTPixelBuffer::QTPixelBuffer(CVPixelBufferRef ref) | |
84 : m_pixelBuffer(ref) | |
85 { | |
86 CVPixelBufferRetain(m_pixelBuffer); | |
87 } | |
88 | |
89 QTPixelBuffer::~QTPixelBuffer() | |
90 { | |
91 clear(); | |
92 } | |
93 | |
94 QTPixelBuffer& QTPixelBuffer::operator=(const QTPixelBuffer& p) | |
95 { | |
96 set(p.m_pixelBuffer); | |
97 return *this; | |
98 } | |
99 | |
100 void QTPixelBuffer::set(CVPixelBufferRef ref) | |
101 { | |
102 CVPixelBufferRetain(ref); | |
103 CVPixelBufferRelease(m_pixelBuffer); | |
104 m_pixelBuffer = ref; | |
105 } | |
106 | |
107 CVPixelBufferRef QTPixelBuffer::pixelBufferRef() | |
108 { | |
109 return m_pixelBuffer; | |
110 } | |
111 | |
112 void QTPixelBuffer::adopt(CVPixelBufferRef ref) | |
113 { | |
114 if (ref == m_pixelBuffer) | |
115 return; | |
116 CVPixelBufferRelease(m_pixelBuffer); | |
117 m_pixelBuffer = ref; | |
118 } | |
119 | |
120 void QTPixelBuffer::clear() | |
121 { | |
122 CVPixelBufferRelease(m_pixelBuffer); | |
123 m_pixelBuffer = 0; | |
124 } | |
125 | |
126 CVReturn QTPixelBuffer::lockBaseAddress() | |
127 { | |
128 return CVPixelBufferLockBaseAddress(m_pixelBuffer, 0); | |
129 } | |
130 | |
131 CVReturn QTPixelBuffer::unlockBaseAddress() | |
132 { | |
133 return CVPixelBufferUnlockBaseAddress(m_pixelBuffer, 0); | |
134 } | |
135 | |
136 void* QTPixelBuffer::baseAddress() | |
137 { | |
138 return CVPixelBufferGetBaseAddress(m_pixelBuffer); | |
139 } | |
140 | |
141 size_t QTPixelBuffer::width() const | |
142 { | |
143 return CVPixelBufferGetWidth(m_pixelBuffer); | |
144 } | |
145 | |
146 size_t QTPixelBuffer::height() const | |
147 { | |
148 return CVPixelBufferGetHeight(m_pixelBuffer); | |
149 } | |
150 | |
151 unsigned long QTPixelBuffer::pixelFormatType() const | |
152 { | |
153 return CVPixelBufferGetPixelFormatType(m_pixelBuffer); | |
154 } | |
155 | |
156 bool QTPixelBuffer::pixelFormatIs32ARGB() const | |
157 { | |
158 return CVPixelBufferGetPixelFormatType(m_pixelBuffer) == k32ARGBPixelFormat; | |
159 } | |
160 | |
161 bool QTPixelBuffer::pixelFormatIs32BGRA() const | |
162 { | |
163 return CVPixelBufferGetPixelFormatType(m_pixelBuffer) == k32BGRAPixelFormat; | |
164 } | |
165 | |
166 size_t QTPixelBuffer::bytesPerRow() const | |
167 { | |
168 return CVPixelBufferGetBytesPerRow(m_pixelBuffer); | |
169 } | |
170 | |
171 size_t QTPixelBuffer::dataSize() const | |
172 { | |
173 return CVPixelBufferGetDataSize(m_pixelBuffer); | |
174 } | |
175 | |
176 bool QTPixelBuffer::isPlanar() const | |
177 { | |
178 return CVPixelBufferIsPlanar(m_pixelBuffer); | |
179 } | |
180 | |
181 size_t QTPixelBuffer::planeCount() const | |
182 { | |
183 return CVPixelBufferGetPlaneCount(m_pixelBuffer); | |
184 } | |
185 | |
186 size_t QTPixelBuffer::widthOfPlane(size_t plane) const | |
187 { | |
188 return CVPixelBufferGetWidthOfPlane(m_pixelBuffer, plane); | |
189 } | |
190 | |
191 size_t QTPixelBuffer::heightOfPlane(size_t plane) const | |
192 { | |
193 return CVPixelBufferGetHeightOfPlane(m_pixelBuffer, plane); | |
194 } | |
195 | |
196 void* QTPixelBuffer::baseAddressOfPlane(size_t plane) const | |
197 { | |
198 return CVPixelBufferGetBaseAddressOfPlane(m_pixelBuffer, plane); | |
199 } | |
200 | |
201 size_t QTPixelBuffer::bytesPerRowOfPlane(size_t plane) const | |
202 { | |
203 return CVPixelBufferGetBytesPerRowOfPlane(m_pixelBuffer, plane); | |
204 } | |
205 | |
206 void QTPixelBuffer::getExtendedPixels(size_t* left, size_t* right, size_t* top,
size_t* bottom) const | |
207 { | |
208 return CVPixelBufferGetExtendedPixels(m_pixelBuffer, left, right, top, botto
m); | |
209 } | |
210 | |
211 CFDictionaryRef QTPixelBuffer::attachments() const | |
212 { | |
213 return CVBufferGetAttachments(m_pixelBuffer, kCVAttachmentMode_ShouldPropaga
te); | |
214 } | |
215 | |
216 void QTPixelBuffer::retainCallback(void* refcon) | |
217 { | |
218 CVPixelBufferRetain(static_cast<CVPixelBufferRef>(refcon)); | |
219 } | |
220 | |
221 void QTPixelBuffer::releaseCallback(void* refcon) | |
222 { | |
223 CVPixelBufferRelease(static_cast<CVPixelBufferRef>(refcon)); | |
224 } | |
225 | |
226 void QTPixelBuffer::imageQueueReleaseCallback(unsigned int type, uint64_t id, vo
id* refcon) | |
227 { | |
228 CVPixelBufferRelease(static_cast<CVPixelBufferRef>(refcon)); | |
229 } | |
230 | |
231 void QTPixelBuffer::dataProviderReleaseBytePointerCallback(void* refcon, const v
oid* pointer) | |
232 { | |
233 CVPixelBufferUnlockBaseAddress(static_cast<CVPixelBufferRef>(refcon), 0); | |
234 } | |
235 | |
236 const void* QTPixelBuffer::dataProviderGetBytePointerCallback(void* refcon) | |
237 { | |
238 CVPixelBufferLockBaseAddress(static_cast<CVPixelBufferRef>(refcon), 0); | |
239 return CVPixelBufferGetBaseAddress(static_cast<CVPixelBufferRef>(refcon)); | |
240 } | |
241 | |
242 size_t QTPixelBuffer::dataProviderGetBytesAtPositionCallback(void* refcon, void*
buffer, size_t position, size_t count) | |
243 { | |
244 char* data = (char*)CVPixelBufferGetBaseAddress(static_cast<CVPixelBufferRef
>(refcon)); | |
245 size_t size = CVPixelBufferGetDataSize(static_cast<CVPixelBufferRef>(refcon)
); | |
246 if (size - position < count) | |
247 count = size - position; | |
248 | |
249 memcpy(buffer, data+position, count); | |
250 return count; | |
251 } | |
252 | |
253 void QTPixelBuffer::dataProviderReleaseInfoCallback(void* refcon) | |
254 { | |
255 CVPixelBufferRelease(static_cast<CVPixelBufferRef>(refcon)); | |
256 } | |
OLD | NEW |