blob: 3a66a4de1412183f9e3d5a862a34f95240c17872
1 | /* |
2 | * DirectShow capture interface |
3 | * Copyright (c) 2010 Ramiro Polla |
4 | * |
5 | * This file is part of FFmpeg. |
6 | * |
7 | * FFmpeg is free software; you can redistribute it and/or |
8 | * modify it under the terms of the GNU Lesser General Public |
9 | * License as published by the Free Software Foundation; either |
10 | * version 2.1 of the License, or (at your option) any later version. |
11 | * |
12 | * FFmpeg is distributed in the hope that it will be useful, |
13 | * but WITHOUT ANY WARRANTY; without even the implied warranty of |
14 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
15 | * Lesser General Public License for more details. |
16 | * |
17 | * You should have received a copy of the GNU Lesser General Public |
18 | * License along with FFmpeg; if not, write to the Free Software |
19 | * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
20 | */ |
21 | |
22 | #include "dshow_capture.h" |
23 | |
24 | DECLARE_QUERYINTERFACE(libAVEnumMediaTypes, |
25 | { {&IID_IUnknown,0}, {&IID_IEnumMediaTypes,0} }) |
26 | DECLARE_ADDREF(libAVEnumMediaTypes) |
27 | DECLARE_RELEASE(libAVEnumMediaTypes) |
28 | |
29 | long WINAPI |
30 | libAVEnumMediaTypes_Next(libAVEnumMediaTypes *this, unsigned long n, |
31 | AM_MEDIA_TYPE **types, unsigned long *fetched) |
32 | { |
33 | int count = 0; |
34 | dshowdebug("libAVEnumMediaTypes_Next(%p)\n", this); |
35 | if (!types) |
36 | return E_POINTER; |
37 | if (!this->pos && n == 1) { |
38 | if (!IsEqualGUID(&this->type.majortype, &GUID_NULL)) { |
39 | AM_MEDIA_TYPE *type = av_malloc(sizeof(AM_MEDIA_TYPE)); |
40 | if (!type) |
41 | return E_OUTOFMEMORY; |
42 | ff_copy_dshow_media_type(type, &this->type); |
43 | *types = type; |
44 | count = 1; |
45 | } |
46 | this->pos = 1; |
47 | } |
48 | if (fetched) |
49 | *fetched = count; |
50 | if (!count) |
51 | return S_FALSE; |
52 | return S_OK; |
53 | } |
54 | long WINAPI |
55 | libAVEnumMediaTypes_Skip(libAVEnumMediaTypes *this, unsigned long n) |
56 | { |
57 | dshowdebug("libAVEnumMediaTypes_Skip(%p)\n", this); |
58 | if (n) /* Any skip will always fall outside of the only valid type. */ |
59 | return S_FALSE; |
60 | return S_OK; |
61 | } |
62 | long WINAPI |
63 | libAVEnumMediaTypes_Reset(libAVEnumMediaTypes *this) |
64 | { |
65 | dshowdebug("libAVEnumMediaTypes_Reset(%p)\n", this); |
66 | this->pos = 0; |
67 | return S_OK; |
68 | } |
69 | long WINAPI |
70 | libAVEnumMediaTypes_Clone(libAVEnumMediaTypes *this, libAVEnumMediaTypes **enums) |
71 | { |
72 | libAVEnumMediaTypes *new; |
73 | dshowdebug("libAVEnumMediaTypes_Clone(%p)\n", this); |
74 | if (!enums) |
75 | return E_POINTER; |
76 | new = libAVEnumMediaTypes_Create(&this->type); |
77 | if (!new) |
78 | return E_OUTOFMEMORY; |
79 | new->pos = this->pos; |
80 | *enums = new; |
81 | return S_OK; |
82 | } |
83 | |
84 | static int |
85 | libAVEnumMediaTypes_Setup(libAVEnumMediaTypes *this, const AM_MEDIA_TYPE *type) |
86 | { |
87 | IEnumMediaTypesVtbl *vtbl = this->vtbl; |
88 | SETVTBL(vtbl, libAVEnumMediaTypes, QueryInterface); |
89 | SETVTBL(vtbl, libAVEnumMediaTypes, AddRef); |
90 | SETVTBL(vtbl, libAVEnumMediaTypes, Release); |
91 | SETVTBL(vtbl, libAVEnumMediaTypes, Next); |
92 | SETVTBL(vtbl, libAVEnumMediaTypes, Skip); |
93 | SETVTBL(vtbl, libAVEnumMediaTypes, Reset); |
94 | SETVTBL(vtbl, libAVEnumMediaTypes, Clone); |
95 | |
96 | if (!type) { |
97 | this->type.majortype = GUID_NULL; |
98 | } else { |
99 | ff_copy_dshow_media_type(&this->type, type); |
100 | } |
101 | |
102 | return 1; |
103 | } |
104 | DECLARE_CREATE(libAVEnumMediaTypes, libAVEnumMediaTypes_Setup(this, type), const AM_MEDIA_TYPE *type) |
105 | DECLARE_DESTROY(libAVEnumMediaTypes, nothing) |
106 |