Skip to content

Commit d72bcfd

Browse files
committed
Merge branch 4.x
2 parents ff95d08 + 0658ab2 commit d72bcfd

File tree

4 files changed

+29
-9
lines changed

4 files changed

+29
-9
lines changed

modules/cudacodec/src/NvEncoder.cpp

Lines changed: 26 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -100,9 +100,12 @@ void NvEncoder::CreateDefaultEncoderParams(NV_ENC_INITIALIZE_PARAMS* pIntializeP
100100
pIntializeParams->enableEncodeAsync = GetCapabilityValue(codecGuid, NV_ENC_CAPS_ASYNC_ENCODE_SUPPORT);
101101
#endif
102102
pIntializeParams->tuningInfo = tuningInfo;
103-
NV_ENC_PRESET_CONFIG presetConfig = {};
104-
presetConfig.version = NV_ENC_PRESET_CONFIG_VER;
105-
presetConfig.presetCfg.version = NV_ENC_CONFIG_VER;
103+
pIntializeParams->encodeConfig->rcParams.rateControlMode = NV_ENC_PARAMS_RC_CONSTQP;
104+
#if (NVENCAPI_MAJOR_VERSION >= 12 && NVENCAPI_MINOR_VERSION >= 2)
105+
NV_ENC_PRESET_CONFIG presetConfig = { NV_ENC_PRESET_CONFIG_VER, 0, { NV_ENC_CONFIG_VER } };
106+
#else
107+
NV_ENC_PRESET_CONFIG presetConfig = { NV_ENC_PRESET_CONFIG_VER, { NV_ENC_CONFIG_VER } };
108+
#endif
106109
m_nvenc.nvEncGetEncodePresetConfigEx(m_hEncoder, codecGuid, presetGuid, tuningInfo, &presetConfig);
107110
memcpy(pIntializeParams->encodeConfig, &presetConfig.presetCfg, sizeof(NV_ENC_CONFIG));
108111

@@ -116,8 +119,13 @@ void NvEncoder::CreateDefaultEncoderParams(NV_ENC_INITIALIZE_PARAMS* pIntializeP
116119
}
117120
else if (pIntializeParams->encodeGUID == NV_ENC_CODEC_HEVC_GUID)
118121
{
122+
#if (NVENCAPI_MAJOR_VERSION >= 12 && NVENCAPI_MINOR_VERSION >= 2)
123+
pIntializeParams->encodeConfig->encodeCodecConfig.hevcConfig.inputBitDepth = pIntializeParams->encodeConfig->encodeCodecConfig.hevcConfig.outputBitDepth =
124+
(m_eBufferFormat == NV_ENC_BUFFER_FORMAT_YUV420_10BIT || m_eBufferFormat == NV_ENC_BUFFER_FORMAT_YUV444_10BIT) ? NV_ENC_BIT_DEPTH_10 : NV_ENC_BIT_DEPTH_8;
125+
#else
119126
pIntializeParams->encodeConfig->encodeCodecConfig.hevcConfig.pixelBitDepthMinus8 =
120127
(m_eBufferFormat == NV_ENC_BUFFER_FORMAT_YUV420_10BIT || m_eBufferFormat == NV_ENC_BUFFER_FORMAT_YUV444_10BIT) ? 2 : 0;
128+
#endif
121129
if (m_eBufferFormat == NV_ENC_BUFFER_FORMAT_YUV444 || m_eBufferFormat == NV_ENC_BUFFER_FORMAT_YUV444_10BIT)
122130
{
123131
pIntializeParams->encodeConfig->encodeCodecConfig.hevcConfig.chromaFormatIDC = 3;
@@ -171,7 +179,11 @@ void NvEncoder::CreateEncoder(const NV_ENC_INITIALIZE_PARAMS* pEncoderParams)
171179
if (pEncoderParams->encodeGUID == NV_ENC_CODEC_HEVC_GUID)
172180
{
173181
bool yuv10BitFormat = (m_eBufferFormat == NV_ENC_BUFFER_FORMAT_YUV420_10BIT || m_eBufferFormat == NV_ENC_BUFFER_FORMAT_YUV444_10BIT) ? true : false;
182+
#if (NVENCAPI_MAJOR_VERSION >= 12 && NVENCAPI_MINOR_VERSION >= 2)
183+
if (yuv10BitFormat && pEncoderParams->encodeConfig->encodeCodecConfig.hevcConfig.inputBitDepth != NV_ENC_BIT_DEPTH_10)
184+
#else
174185
if (yuv10BitFormat && pEncoderParams->encodeConfig->encodeCodecConfig.hevcConfig.pixelBitDepthMinus8 != 2)
186+
#endif
175187
{
176188
NVENC_THROW_ERROR("Invalid PixelBitdepth", NV_ENC_ERR_INVALID_PARAM);
177189
}
@@ -193,12 +205,20 @@ void NvEncoder::CreateEncoder(const NV_ENC_INITIALIZE_PARAMS* pEncoderParams)
193205
}
194206
else
195207
{
196-
NV_ENC_PRESET_CONFIG presetConfig = {};
197-
presetConfig.version = NV_ENC_PRESET_CONFIG_VER;
198-
presetConfig.presetCfg.version = NV_ENC_CONFIG_VER;
208+
#if (NVENCAPI_MAJOR_VERSION >= 12 && NVENCAPI_MINOR_VERSION >= 2)
209+
NV_ENC_PRESET_CONFIG presetConfig = { NV_ENC_PRESET_CONFIG_VER, 0, { NV_ENC_CONFIG_VER } };
210+
#else
211+
NV_ENC_PRESET_CONFIG presetConfig = { NV_ENC_PRESET_CONFIG_VER, { NV_ENC_CONFIG_VER } };
212+
#endif
199213
m_nvenc.nvEncGetEncodePresetConfigEx(m_hEncoder, pEncoderParams->encodeGUID, pEncoderParams->presetGUID, pEncoderParams->tuningInfo, &presetConfig);
200214
memcpy(&m_encodeConfig, &presetConfig.presetCfg, sizeof(NV_ENC_CONFIG));
201215
}
216+
217+
if (((uint32_t)m_encodeConfig.frameIntervalP) > m_encodeConfig.gopLength)
218+
{
219+
m_encodeConfig.frameIntervalP = m_encodeConfig.gopLength;
220+
}
221+
202222
m_initializeParams.encodeConfig = &m_encodeConfig;
203223
NVENC_API_CALL(m_nvenc.nvEncInitializeEncoder(m_hEncoder, &m_initializeParams));
204224
m_bEncoderInitialized = true;

modules/xfeatures2d/misc/java/test/MSDFeatureDetectorTest.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -60,7 +60,7 @@ public void testWriteYml() {
6060

6161
detector.write(filename);
6262

63-
String truth = "%YAML:1.0\n---\nname: \"Feature2D.MSD\"\npatch_radius: 3\nsearch_area_radius: 5\nnms_radius: 5\nnms_scale_radius: 0\nth_saliency: 250.\nkNN: 4\nscale_factor: 1.2500000000000000e+00\nn_scales: -1\ncompute_orientation: 0\n";
63+
String truth = "%YAML:1.0\n---\nname: \"Feature2D.MSD\"\npatch_radius: 3\nsearch_area_radius: 5\nnms_radius: 5\nnms_scale_radius: 0\nth_saliency: 250.\nkNN: 4\nscale_factor: 1.25\nn_scales: -1\ncompute_orientation: 0\n";
6464
String actual = readFile(filename);
6565
actual = actual.replaceAll("e([+-])0(\\d\\d)", "e$1$2"); // NOTE: workaround for different platforms double representation
6666
assertEquals(truth, actual);

modules/ximgproc/src/edge_drawing.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2592,7 +2592,7 @@ void EdgeDrawingImpl::detectEllipses(OutputArray ellipses)
25922592
}
25932593

25942594
// This is how much space we will allocate for circles buffers
2595-
int maxNoOfCircles = (int)lines.size() / 3 + noCircles1 * 2;
2595+
int maxNoOfCircles = (int)lines.size() / 3 + noCircles1 * 2 + 2;
25962596

25972597
edarcs1 = new EDArcs(maxNoOfCircles);
25982598
DetectArcs(); // Detect all arcs

modules/ximgproc/src/fast_line_detector.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -581,7 +581,7 @@ void FastLineDetectorImpl::lineDetection(const Mat& src, std::vector<SEGMENT>& s
581581
bool is_merged = false;
582582
int ith = (int)segments_tmp.size() - 1;
583583
int jth = ith - 1;
584-
while(ith > 1 || jth > 0)
584+
while(ith > 1 || jth >= 0)
585585
{
586586
seg1 = segments_tmp[ith];
587587
seg2 = segments_tmp[jth];

0 commit comments

Comments
 (0)