Skip to content

Commit 227687d

Browse files
authored
Merge pull request #3867 from cudawarped:cudacodec_enable_win32_ffmpeg_features
`cudacodec`: Enable features available with updated ffmpeg dll
2 parents ac9ef6c + 5c544bc commit 227687d

File tree

2 files changed

+0
-18
lines changed

2 files changed

+0
-18
lines changed

modules/cudacodec/src/video_writer.cpp

-8
Original file line numberDiff line numberDiff line change
@@ -53,10 +53,6 @@ Ptr<cudacodec::VideoWriter> createVideoWriter(const String&, const Size, const C
5353

5454
#else // !defined HAVE_NVCUVENC
5555

56-
#if defined(WIN32) // remove when FFmpeg wrapper includes PR25874
57-
#define WIN32_WAIT_FOR_FFMPEG_WRAPPER_UPDATE
58-
#endif
59-
6056
NV_ENC_BUFFER_FORMAT EncBufferFormat(const ColorFormat colorFormat);
6157
int NChannels(const ColorFormat colorFormat);
6258
GUID CodecGuid(const Codec codec);
@@ -107,9 +103,7 @@ void FFmpegVideoWriter::onEncoded(const std::vector<std::vector<uint8_t>>& vPack
107103
Mat wrappedPacket(1, packet.size(), CV_8UC1, (void*)packet.data());
108104
const double ptsDouble = static_cast<double>(pts.at(i));
109105
CV_Assert(static_cast<uint64_t>(ptsDouble) == pts.at(i));
110-
#if !defined(WIN32_WAIT_FOR_FFMPEG_WRAPPER_UPDATE)
111106
CV_Assert(writer.set(VIDEOWRITER_PROP_PTS, ptsDouble));
112-
#endif
113107
writer.write(wrappedPacket);
114108
}
115109
}
@@ -337,11 +331,9 @@ void VideoWriterImpl::InitializeEncoder(const GUID codec, const double fps)
337331
initializeParams.encodeConfig->rcParams.maxBitRate = encoderParams.maxBitRate;
338332
initializeParams.encodeConfig->rcParams.targetQuality = encoderParams.targetQuality;
339333
initializeParams.encodeConfig->gopLength = encoderParams.gopLength;
340-
#if !defined(WIN32_WAIT_FOR_FFMPEG_WRAPPER_UPDATE)
341334
if (initializeParams.encodeConfig->frameIntervalP > 1) {
342335
CV_Assert(encoderCallback->setFrameIntervalP(initializeParams.encodeConfig->frameIntervalP));
343336
}
344-
#endif
345337
if (codec == NV_ENC_CODEC_H264_GUID)
346338
initializeParams.encodeConfig->encodeCodecConfig.h264Config.idrPeriod = encoderParams.idrPeriod;
347339
else if (codec == NV_ENC_CODEC_HEVC_GUID)

modules/cudacodec/test/test_video.cpp

-10
Original file line numberDiff line numberDiff line change
@@ -1011,10 +1011,6 @@ struct H264ToH265 : SetDevice
10111011
{
10121012
};
10131013

1014-
#if defined(WIN32) // remove when FFmpeg wrapper includes PR25874
1015-
#define WIN32_WAIT_FOR_FFMPEG_WRAPPER_UPDATE
1016-
#endif
1017-
10181014
CUDA_TEST_P(H264ToH265, Transcode)
10191015
{
10201016
const std::string inputFile = std::string(cvtest::TS::ptr()->get_data_path()) + "../highgui/video/big_buck_bunny.h264";
@@ -1054,10 +1050,8 @@ CUDA_TEST_P(H264ToH265, Transcode)
10541050
for (int i = 0; i < nFrames; ++i) {
10551051
cap >> frame;
10561052
ASSERT_FALSE(frame.empty());
1057-
#if !defined(WIN32_WAIT_FOR_FFMPEG_WRAPPER_UPDATE)
10581053
const int pts = static_cast<int>(cap.get(CAP_PROP_PTS));
10591054
ASSERT_EQ(i, pts > 0 ? pts : 0); // FFmpeg back end returns dts if pts is zero.
1060-
#endif
10611055
}
10621056
}
10631057
ASSERT_EQ(0, remove(outputFile.c_str()));
@@ -1201,10 +1195,8 @@ CUDA_TEST_P(Write, Writer)
12011195
for (int i = 0; i < nFrames; ++i) {
12021196
cap >> frame;
12031197
ASSERT_FALSE(frame.empty());
1204-
#if !defined(WIN32_WAIT_FOR_FFMPEG_WRAPPER_UPDATE)
12051198
const int pts = static_cast<int>(cap.get(CAP_PROP_PTS));
12061199
ASSERT_EQ(i, pts > 0 ? pts : 0); // FFmpeg back end returns dts if pts is zero.
1207-
#endif
12081200
}
12091201
}
12101202
ASSERT_EQ(0, remove(outputFile.c_str()));
@@ -1299,10 +1291,8 @@ CUDA_TEST_P(EncoderParams, Writer)
12991291
const bool keyFrameActual = capRaw.get(CAP_PROP_LRF_HAS_KEY_FRAME) == 1.0;
13001292
const bool keyFrameReference = i % idrPeriod == 0;
13011293
ASSERT_EQ(keyFrameActual, keyFrameReference);
1302-
#if !defined(WIN32_WAIT_FOR_FFMPEG_WRAPPER_UPDATE)
13031294
const int pts = static_cast<int>(cap.get(CAP_PROP_PTS));
13041295
ASSERT_EQ(i, pts > 0 ? pts : 0); // FFmpeg back end returns dts if pts is zero.
1305-
#endif
13061296
}
13071297
}
13081298
}

0 commit comments

Comments
 (0)