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
103
104
105
106
|
/*****************************************************************************
* Copyright (C) 2013-2020 MulticoreWare, Inc
*
* Authors: Steve Borho <steve@borho.org>
* Min Chen <chenm003@163.com>
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, write to the Free Software
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02111, USA.
*
* This program is also available under a commercial proprietary license.
* For more information, contact us at license @ x265.com
*****************************************************************************/
#ifndef X265_WAVEFRONT_H
#define X265_WAVEFRONT_H
#include "common.h"
#include "threadpool.h"
namespace X265_NS {
// x265 private namespace
// Generic wave-front scheduler, manages busy-state of CU rows as a priority
// queue (higher CU rows have priority over lower rows)
//
// Derived classes must implement ProcessRow().
class WaveFront : public JobProvider
{
private:
// bitmaps of rows queued for processing, uses atomic intrinsics
// Dependencies are categorized as internal and external. Internal dependencies
// are caused by neighbor block availability. External dependencies are generally
// reference frame reconstructed pixels being available.
uint32_t volatile *m_internalDependencyBitmap;
uint32_t volatile *m_externalDependencyBitmap;
// number of words in the bitmap
int m_numWords;
int m_numRows;
int m_sLayerId;
protected:
uint32_t *m_row_to_idx;
uint32_t *m_idx_to_row;
public:
WaveFront()
: m_internalDependencyBitmap(NULL)
, m_externalDependencyBitmap(NULL)
{}
virtual ~WaveFront();
// If returns false, the frame must be encoded in series.
bool init(int numRows);
// Enqueue a row to be processed (mark its internal dependencies as resolved).
// A worker thread will later call processRow(row).
// This provider must be enqueued in the pool before enqueuing a row
void enqueueRow(int row);
// Mark a row as no longer having internal dependencies resolved. Returns
// true if bit clear was successful, false otherwise.
bool dequeueRow(int row);
// Mark the row's external dependencies as being resolved
void enableRow(int row);
// Mark all row external dependencies as being resolved. Some wavefront
// implementations (lookahead, for instance) have no recon pixel dependencies.
void enableAllRows();
// Mark all rows as having external dependencies which must be
// resolved before each row may proceed.
void clearEnabledRowMask();
// WaveFront's implementation of JobProvider::findJob. Consults
// m_queuedBitmap and calls ProcessRow(row) for lowest numbered queued row
// processes available rows and returns when no work remains
void findJob(int threadId);
// Start or resume encode processing of this row, must be implemented by
// derived classes.
virtual void processRow(int row, int threadId, int layer) = 0;
void setLayerId(int layer);
};
} // end namespace X265_NS
#endif // ifndef X265_WAVEFRONT_H
|
