| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "media/video/capture/win/sink_input_pin_win.h" | 5 #include "media/video/capture/win/sink_input_pin_win.h" |
| 6 | 6 |
| 7 #include <cstring> | 7 #include <cstring> |
| 8 | 8 |
| 9 // Avoid including strsafe.h via dshow as it will cause build warnings. | 9 // Avoid including strsafe.h via dshow as it will cause build warnings. |
| 10 #define NO_DSHOW_STRSAFE | 10 #define NO_DSHOW_STRSAFE |
| 11 #include <dshow.h> | 11 #include <dshow.h> |
| 12 | 12 |
| 13 #include "base/logging.h" | 13 #include "base/logging.h" |
| 14 | 14 |
| 15 namespace media { | 15 namespace media { |
| 16 | 16 |
| 17 const REFERENCE_TIME kSecondsToReferenceTime = 10000000; | 17 const REFERENCE_TIME kSecondsToReferenceTime = 10000000; |
| 18 | 18 |
| 19 SinkInputPin::SinkInputPin(IBaseFilter* filter, | 19 SinkInputPin::SinkInputPin(IBaseFilter* filter, |
| 20 SinkFilterObserver* observer) | 20 SinkFilterObserver* observer) |
| 21 : observer_(observer), | 21 : observer_(observer), |
| 22 PinBase(filter) { | 22 PinBase(filter) { |
| 23 memset(&requested_capability_, 0, sizeof(requested_capability_)); | |
| 24 memset(&resulting_capability_, 0, sizeof(resulting_capability_)); | |
| 25 } | 23 } |
| 26 | 24 |
| 27 SinkInputPin::~SinkInputPin() {} | 25 SinkInputPin::~SinkInputPin() {} |
| 28 | 26 |
| 29 bool SinkInputPin::GetValidMediaType(int index, AM_MEDIA_TYPE* media_type) { | 27 bool SinkInputPin::GetValidMediaType(int index, AM_MEDIA_TYPE* media_type) { |
| 30 if (media_type->cbFormat < sizeof(VIDEOINFOHEADER)) | 28 if (media_type->cbFormat < sizeof(VIDEOINFOHEADER)) |
| 31 return false; | 29 return false; |
| 32 | 30 |
| 33 VIDEOINFOHEADER* pvi = | 31 VIDEOINFOHEADER* pvi = |
| 34 reinterpret_cast<VIDEOINFOHEADER*>(media_type->pbFormat); | 32 reinterpret_cast<VIDEOINFOHEADER*>(media_type->pbFormat); |
| 35 | 33 |
| 36 ZeroMemory(pvi, sizeof(VIDEOINFOHEADER)); | 34 ZeroMemory(pvi, sizeof(VIDEOINFOHEADER)); |
| 37 pvi->bmiHeader.biSize = sizeof(BITMAPINFOHEADER); | 35 pvi->bmiHeader.biSize = sizeof(BITMAPINFOHEADER); |
| 38 pvi->bmiHeader.biPlanes = 1; | 36 pvi->bmiHeader.biPlanes = 1; |
| 39 pvi->bmiHeader.biClrImportant = 0; | 37 pvi->bmiHeader.biClrImportant = 0; |
| 40 pvi->bmiHeader.biClrUsed = 0; | 38 pvi->bmiHeader.biClrUsed = 0; |
| 41 if (requested_capability_.frame_rate > 0) { | 39 if (requested_format_.frame_rate > 0) { |
| 42 pvi->AvgTimePerFrame = kSecondsToReferenceTime / | 40 pvi->AvgTimePerFrame = |
| 43 requested_capability_.frame_rate; | 41 kSecondsToReferenceTime / requested_format_.frame_rate; |
| 44 } | 42 } |
| 45 | 43 |
| 46 media_type->majortype = MEDIATYPE_Video; | 44 media_type->majortype = MEDIATYPE_Video; |
| 47 media_type->formattype = FORMAT_VideoInfo; | 45 media_type->formattype = FORMAT_VideoInfo; |
| 48 media_type->bTemporalCompression = FALSE; | 46 media_type->bTemporalCompression = FALSE; |
| 49 | 47 |
| 50 switch (index) { | 48 switch (index) { |
| 51 case 0: { | 49 case 0: { |
| 52 pvi->bmiHeader.biCompression = MAKEFOURCC('I', '4', '2', '0'); | 50 pvi->bmiHeader.biCompression = MAKEFOURCC('I', '4', '2', '0'); |
| 53 pvi->bmiHeader.biBitCount = 12; // bit per pixel | 51 pvi->bmiHeader.biBitCount = 12; // bit per pixel |
| 54 pvi->bmiHeader.biWidth = requested_capability_.width; | 52 pvi->bmiHeader.biWidth = requested_format_.frame_size.width(); |
| 55 pvi->bmiHeader.biHeight = requested_capability_.height; | 53 pvi->bmiHeader.biHeight = requested_format_.frame_size.height(); |
| 56 pvi->bmiHeader.biSizeImage = 3 * requested_capability_.height * | 54 pvi->bmiHeader.biSizeImage = |
| 57 requested_capability_.width / 2; | 55 requested_format_.frame_size.GetArea() * 3 / 2; |
| 58 media_type->subtype = kMediaSubTypeI420; | 56 media_type->subtype = kMediaSubTypeI420; |
| 59 break; | 57 break; |
| 60 } | 58 } |
| 61 case 1: { | 59 case 1: { |
| 62 pvi->bmiHeader.biCompression = MAKEFOURCC('Y', 'U', 'Y', '2'); | 60 pvi->bmiHeader.biCompression = MAKEFOURCC('Y', 'U', 'Y', '2'); |
| 63 pvi->bmiHeader.biBitCount = 16; | 61 pvi->bmiHeader.biBitCount = 16; |
| 64 pvi->bmiHeader.biWidth = requested_capability_.width; | 62 pvi->bmiHeader.biWidth = requested_format_.frame_size.width(); |
| 65 pvi->bmiHeader.biHeight = requested_capability_.height; | 63 pvi->bmiHeader.biHeight = requested_format_.frame_size.height(); |
| 66 pvi->bmiHeader.biSizeImage = 2 * requested_capability_.width * | 64 pvi->bmiHeader.biSizeImage = requested_format_.frame_size.GetArea() * 2; |
| 67 requested_capability_.height; | |
| 68 media_type->subtype = MEDIASUBTYPE_YUY2; | 65 media_type->subtype = MEDIASUBTYPE_YUY2; |
| 69 break; | 66 break; |
| 70 } | 67 } |
| 71 case 2: { | 68 case 2: { |
| 72 pvi->bmiHeader.biCompression = BI_RGB; | 69 pvi->bmiHeader.biCompression = BI_RGB; |
| 73 pvi->bmiHeader.biBitCount = 24; | 70 pvi->bmiHeader.biBitCount = 24; |
| 74 pvi->bmiHeader.biWidth = requested_capability_.width; | 71 pvi->bmiHeader.biWidth = requested_format_.frame_size.width(); |
| 75 pvi->bmiHeader.biHeight = requested_capability_.height; | 72 pvi->bmiHeader.biHeight = requested_format_.frame_size.height(); |
| 76 pvi->bmiHeader.biSizeImage = 3 * requested_capability_.height * | 73 pvi->bmiHeader.biSizeImage = requested_format_.frame_size.GetArea() * 3; |
| 77 requested_capability_.width; | |
| 78 media_type->subtype = MEDIASUBTYPE_RGB24; | 74 media_type->subtype = MEDIASUBTYPE_RGB24; |
| 79 break; | 75 break; |
| 80 } | 76 } |
| 81 default: | 77 default: |
| 82 return false; | 78 return false; |
| 83 } | 79 } |
| 84 | 80 |
| 85 media_type->bFixedSizeSamples = TRUE; | 81 media_type->bFixedSizeSamples = TRUE; |
| 86 media_type->lSampleSize = pvi->bmiHeader.biSizeImage; | 82 media_type->lSampleSize = pvi->bmiHeader.biSizeImage; |
| 87 return true; | 83 return true; |
| 88 } | 84 } |
| 89 | 85 |
| 90 bool SinkInputPin::IsMediaTypeValid(const AM_MEDIA_TYPE* media_type) { | 86 bool SinkInputPin::IsMediaTypeValid(const AM_MEDIA_TYPE* media_type) { |
| 91 GUID type = media_type->majortype; | 87 GUID type = media_type->majortype; |
| 92 if (type != MEDIATYPE_Video) | 88 if (type != MEDIATYPE_Video) |
| 93 return false; | 89 return false; |
| 94 | 90 |
| 95 GUID format_type = media_type->formattype; | 91 GUID format_type = media_type->formattype; |
| 96 if (format_type != FORMAT_VideoInfo) | 92 if (format_type != FORMAT_VideoInfo) |
| 97 return false; | 93 return false; |
| 98 | 94 |
| 99 // Check for the sub types we support. | 95 // Check for the sub types we support. |
| 100 GUID sub_type = media_type->subtype; | 96 GUID sub_type = media_type->subtype; |
| 101 VIDEOINFOHEADER* pvi = | 97 VIDEOINFOHEADER* pvi = |
| 102 reinterpret_cast<VIDEOINFOHEADER*>(media_type->pbFormat); | 98 reinterpret_cast<VIDEOINFOHEADER*>(media_type->pbFormat); |
| 103 if (pvi == NULL) | 99 if (pvi == NULL) |
| 104 return false; | 100 return false; |
| 105 | 101 |
| 106 // Store the incoming width and height. | 102 // Store the incoming width and height. |
| 107 resulting_capability_.width = pvi->bmiHeader.biWidth; | 103 resulting_format_.frame_size.SetSize(pvi->bmiHeader.biWidth, |
| 108 resulting_capability_.height = abs(pvi->bmiHeader.biHeight); | 104 abs(pvi->bmiHeader.biHeight)); |
| 109 if (pvi->AvgTimePerFrame > 0) { | 105 if (pvi->AvgTimePerFrame > 0) { |
| 110 resulting_capability_.frame_rate = | 106 resulting_format_.frame_rate = |
| 111 static_cast<int>(kSecondsToReferenceTime / pvi->AvgTimePerFrame); | 107 static_cast<int>(kSecondsToReferenceTime / pvi->AvgTimePerFrame); |
| 112 } else { | 108 } else { |
| 113 resulting_capability_.frame_rate = requested_capability_.frame_rate; | 109 resulting_format_.frame_rate = requested_format_.frame_rate; |
| 114 } | 110 } |
| 115 if (sub_type == kMediaSubTypeI420 && | 111 if (sub_type == kMediaSubTypeI420 && |
| 116 pvi->bmiHeader.biCompression == MAKEFOURCC('I', '4', '2', '0')) { | 112 pvi->bmiHeader.biCompression == MAKEFOURCC('I', '4', '2', '0')) { |
| 117 resulting_capability_.color = PIXEL_FORMAT_I420; | 113 resulting_format_.pixel_format = PIXEL_FORMAT_I420; |
| 118 return true; // This format is acceptable. | 114 return true; // This format is acceptable. |
| 119 } | 115 } |
| 120 if (sub_type == MEDIASUBTYPE_YUY2 && | 116 if (sub_type == MEDIASUBTYPE_YUY2 && |
| 121 pvi->bmiHeader.biCompression == MAKEFOURCC('Y', 'U', 'Y', '2')) { | 117 pvi->bmiHeader.biCompression == MAKEFOURCC('Y', 'U', 'Y', '2')) { |
| 122 resulting_capability_.color = PIXEL_FORMAT_YUY2; | 118 resulting_format_.pixel_format = PIXEL_FORMAT_YUY2; |
| 123 return true; // This format is acceptable. | 119 return true; // This format is acceptable. |
| 124 } | 120 } |
| 125 if (sub_type == MEDIASUBTYPE_RGB24 && | 121 if (sub_type == MEDIASUBTYPE_RGB24 && |
| 126 pvi->bmiHeader.biCompression == BI_RGB) { | 122 pvi->bmiHeader.biCompression == BI_RGB) { |
| 127 resulting_capability_.color = PIXEL_FORMAT_RGB24; | 123 resulting_format_.pixel_format = PIXEL_FORMAT_RGB24; |
| 128 return true; // This format is acceptable. | 124 return true; // This format is acceptable. |
| 129 } | 125 } |
| 130 return false; | 126 return false; |
| 131 } | 127 } |
| 132 | 128 |
| 133 HRESULT SinkInputPin::Receive(IMediaSample* sample) { | 129 HRESULT SinkInputPin::Receive(IMediaSample* sample) { |
| 134 const int length = sample->GetActualDataLength(); | 130 const int length = sample->GetActualDataLength(); |
| 135 uint8* buffer = NULL; | 131 uint8* buffer = NULL; |
| 136 if (FAILED(sample->GetPointer(&buffer))) | 132 if (FAILED(sample->GetPointer(&buffer))) |
| 137 return S_FALSE; | 133 return S_FALSE; |
| 138 | 134 |
| 139 observer_->FrameReceived(buffer, length); | 135 observer_->FrameReceived(buffer, length); |
| 140 return S_OK; | 136 return S_OK; |
| 141 } | 137 } |
| 142 | 138 |
| 143 void SinkInputPin::SetRequestedMediaCapability( | 139 void SinkInputPin::SetRequestedMediaFormat(const VideoCaptureFormat& format) { |
| 144 const VideoCaptureCapability& capability) { | 140 requested_format_ = format; |
| 145 requested_capability_ = capability; | 141 resulting_format_.frame_size.SetSize(0, 0); |
| 146 resulting_capability_.width = 0; | 142 resulting_format_.frame_rate = 0; |
| 147 resulting_capability_.height = 0; | 143 resulting_format_.pixel_format = PIXEL_FORMAT_UNKNOWN; |
| 148 resulting_capability_.frame_rate = 0; | |
| 149 resulting_capability_.color = PIXEL_FORMAT_UNKNOWN; | |
| 150 } | 144 } |
| 151 | 145 |
| 152 const VideoCaptureCapability& SinkInputPin::ResultingCapability() { | 146 const VideoCaptureFormat& SinkInputPin::ResultingFormat() { |
| 153 return resulting_capability_; | 147 return resulting_format_; |
| 154 } | 148 } |
| 155 | 149 |
| 156 } // namespace media | 150 } // namespace media |
| OLD | NEW |