FrameQueue.h
2.58 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
/*
* Copyright 1993-2015 NVIDIA Corporation. All rights reserved.
*
* Please refer to the NVIDIA end user license agreement (EULA) associated
* with this source code for terms and conditions that govern your use of
* this software. Any use, reproduction, disclosure, or distribution of
* this software and related documentation outside the terms of the EULA
* is strictly prohibited.
*
*/
#ifndef FRAMEQUEUE_H
#define FRAMEQUEUE_H
#include <nvcuvid.h>
#if defined(WIN32) || defined(_WIN32) || defined(WIN64) || defined(_WIN64)
#include <windows.h>
//#define sleep(x) Sleep(x)
#else
#include <unistd.h>
#include <string.h>
#endif
#ifdef __linux__
#include "DxStd.h" //modified by Junlin 190221
#endif
class FrameQueue
{
public:
static const unsigned int cnMaximumSize = 20; // MAX_FRM_CNT;
FrameQueue();
virtual
~FrameQueue();
void
enter_CS(CRITICAL_SECTION *pCS);
void
leave_CS(CRITICAL_SECTION *pCS);
/*
void
set_event(HANDLE event);
void
reset_event(HANDLE event);
*/
void
enqueue(const CUVIDPARSERDISPINFO *pPicParams);
// Dequeue the next frame.
// Parameters:
// pDisplayInfo - New frame info gets placed into this object.
// Note: This pointer must point to a valid struct. The method
// does not create memory for this.
// Returns:
// true, if a new frame was returned,
// false, if the queue was empty and no new frame could be returned.
// In that case, pPicParams doesn't contain valid data.
bool
dequeue(CUVIDPARSERDISPINFO *pDisplayInfo);
void
releaseFrame(const CUVIDPARSERDISPINFO *pPicParams);
bool
isInUse(int nPictureIndex)
const;
bool
isDecodeFinished()
const;
void
endDecode();
// Spins until frame becomes available or decoding
// gets canceled.
// If the requested frame is available the method returns true.
// If decoding was interrupted before the requested frame becomes
// available, the method returns false.
bool
waitUntilFrameAvailable(int nPictureIndex);
private:
CRITICAL_SECTION oCriticalSection_;
volatile int nReadPosition_;
volatile int nFramesInQueue_;
CUVIDPARSERDISPINFO aDisplayQueue_[cnMaximumSize];
volatile int aIsFrameInUse_[cnMaximumSize];
volatile int bEndOfDecode_;
};
#endif // FRAMEQUEUE_H