Grok 10.0.5
TileProcessor.h
Go to the documentation of this file.
1/*
2 * Copyright (C) 2016-2023 Grok Image Compression Inc.
3 *
4 * This source code is free software: you can redistribute it and/or modify
5 * it under the terms of the GNU Affero General Public License, version 3,
6 * as published by the Free Software Foundation.
7 *
8 * This source code is distributed in the hope that it will be useful,
9 * but WITHOUT ANY WARRANTY; without even the implied warranty of
10 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11 * GNU Affero General Public License for more details.
12 *
13 * You should have received a copy of the GNU Affero General Public License
14 * along with this program. If not, see <http://www.gnu.org/licenses/>.
15 *
16 *
17 * This source code incorporates work covered by the BSD 2-clause license.
18 * Please see the LICENSE file in the root directory for details.
19 *
20 */
21
22#pragma once
23#include "grk_includes.h"
24#include <queue>
25#include <mutex>
26
27namespace grk
28{
29/*
30 * Tile structure.
31 *
32 * Tile bounds are in canvas coordinates, and are equal to the
33 * full, non-windowed, unreduced tile dimensions,
34 * while the component dimensions are reduced
35 * if there is a resolution reduction.
36 *
37 */
38struct Tile : public grk_rect32
39{
40 Tile();
41 explicit Tile(uint16_t numcomps);
42 virtual ~Tile();
43 uint16_t numcomps_;
44 TileComponent* comps;
45 double distortion;
46 double layerDistoration[maxCompressLayersGRK];
47};
48
49struct PacketTracker
50{
51 PacketTracker();
52 ~PacketTracker();
53 void init(uint32_t numcomps, uint32_t numres, uint64_t numprec, uint32_t numlayers);
54 void clear(void);
55 void packet_encoded(uint32_t comps, uint32_t res, uint64_t prec, uint32_t layer);
56 bool is_packet_encoded(uint32_t comps, uint32_t res, uint64_t prec, uint32_t layer);
57
58 private:
59 uint8_t* bits;
60
61 uint32_t numcomps_;
62 uint32_t numres_;
63 uint64_t numprec_;
64 uint32_t numlayers_;
65
66 uint64_t get_buffer_len(uint32_t numcomps, uint32_t numres, uint64_t numprec,
67 uint32_t numlayers);
68 uint64_t index(uint32_t comps, uint32_t res, uint64_t prec, uint32_t layer);
69};
70
75class mct;
76
77struct TileProcessor
78{
79 explicit TileProcessor(uint16_t index, CodeStream* codeStream, BufferedStream* stream,
80 bool isCompressor, StripCache* stripCache);
81 ~TileProcessor();
82 bool init(void);
83 bool createWindowBuffers(const GrkImage* outputImage);
84 void deallocBuffers();
85 bool preCompressTile(void);
86 bool canWritePocMarker(void);
87 bool writeTilePartT2(uint32_t* tileBytesWritten);
88 bool doCompress(void);
89 bool decompressT2T1(GrkImage* outputImage);
90 bool ingestUncompressedData(uint8_t* p_src, uint64_t src_length);
91 bool needsRateControl();
92 void ingestImage();
93 bool cacheTilePartPackets(CodeStreamDecompress* codeStream);
94 void generateImage(GrkImage* src_image, Tile* src_tile);
95 GrkImage* getImage(void);
96 void release(GRK_TILE_CACHE_STRATEGY strategy);
97 void setCorruptPacket(void);
98 PacketTracker* getPacketTracker(void);
99 grk_rect32 getUnreducedTileWindow(void);
100 TileCodingParams* getTileCodingParams(void);
101 uint8_t getMaxNumDecompressResolutions(void);
102 BufferedStream* getStream(void);
103 uint32_t getPreCalculatedTileLen(void);
104 bool canPreCalculateTileLen(void);
105 uint16_t getIndex(void) const;
106 void incrementIndex(void);
107 Tile* getTile(void);
108 Scheduler* getScheduler(void);
109 bool isCompressor(void);
110
121 uint32_t pino;
122 GrkImage* headerImage;
124 CodingParams* cp_;
126 uint64_t getTilePartDataLength(void);
127 bool subtractMarkerSegmentLength(uint16_t markerLen);
128 bool setTilePartDataLength(uint16_t tilePart, uint32_t tilePartLength,
130 uint64_t getNumProcessedPackets(void);
131 void incNumProcessedPackets(void);
132 void incNumProcessedPackets(uint64_t numPackets);
133 uint64_t getNumDecompressedPackets(void);
134 void incNumDecompressedPackets(void);
135
136 private:
137 bool isWholeTileDecompress(uint16_t compno);
138 bool needsMctDecompress(uint16_t compno);
139 bool needsMctDecompress(void);
140 bool mctDecompress(FlowComponent* flow);
141 bool dcLevelShiftCompress();
142 bool mct_encode();
143 bool dwt_encode();
144 void t1_encode();
145 bool encodeT2(uint32_t* packet_bytes_written);
146 bool rateAllocate(uint32_t* allPacketBytes);
147 bool layerNeedsRateControl(uint32_t layno);
148 bool makeSingleLosslessLayer();
149 void makeLayerFinal(uint32_t layno);
150 bool pcrdBisectSimple(uint32_t* p_data_written);
151 void makeLayerSimple(uint32_t layno, double thresh, bool finalAttempt);
152 bool pcrdBisectFeasible(uint32_t* p_data_written);
153 void makeLayerFeasible(uint32_t layno, uint16_t thresh, bool finalAttempt);
154
155 Tile* tile;
156 Scheduler* scheduler_;
158 std::atomic<uint64_t> numDecompressedPackets;
159 // Decompressing Only
162 uint16_t tileIndex_;
163 // Compressing only - track which packets have already been written
164 // to the code stream
165 PacketTracker packetTracker_;
166 BufferedStream* stream_;
170 // coding/decoding parameters for this tile
171 TileCodingParams* tcp_;
173 GrkImage* image_;
178};
179
180} // namespace grk
BufferedStream * stream
Definition BitIO.h:88
BufferedStream * stream_
Definition CodeStream.h:155
CodingParams cp_
Definition CodeStream.h:147
grk_plugin_tile * current_plugin_tile
Definition CodeStream.h:157
uint32_t newTilePartProgressionPosition
Position of tile part flag in progression order.
Definition CodingParams.h:212
double distortion[maxCompressLayersGRK]
fixed_quality
Definition CodingParams.h:161
uint8_t tilePartCounter_
Definition CodingParams.h:174
uint16_t numlayers
number of layers
Definition CodingParams.h:138
bool lastTilePartInCodeStream
Indicate that the current tile-part is assumed to be the last tile part of the code stream.
Definition CodingParams.h:332
uint8_t mct
multi-component transform identifier
Definition CodingParams.h:141
bool needsRateControl
Definition CompressScheduler.h:37
uint16_t numcomps_
Definition DecompressScheduler.h:54
ComponentInfo * comps
Definition FileFormat.h:162
uint16_t numcomps
Definition FileFormat.h:149
CodeStreamCompress * codeStream
Definition FileFormatCompress.h:60
uint16_t layno
Definition PacketIter.h:259
bool isCompressor_
Definition Precinct.h:59
std::atomic< uint64_t > numDecompressedPackets
Definition TileProcessor.h:158
bool corrupt_packet_
Definition TileProcessor.h:167
uint64_t numprec_
Definition TileProcessor.h:63
uint16_t tileIndex_
index of tile being currently compressed/decompressed
Definition TileProcessor.h:162
uint32_t numres_
Definition TileProcessor.h:62
bool first_poc_tile_part_
Compression Only true for first POC tile part, otherwise false.
Definition TileProcessor.h:113
double layerDistoration[maxCompressLayersGRK]
Definition TileProcessor.h:46
uint64_t tilePartDataLength
Definition TileProcessor.h:160
mct * mct_
Definition TileProcessor.h:177
bool truncated
Definition TileProcessor.h:172
uint32_t pino
Compression Only Current packet iterator number.
Definition TileProcessor.h:121
Scheduler * scheduler_
Definition TileProcessor.h:156
uint8_t * bits
Definition TileProcessor.h:59
grk_rect32 unreducedImageWindow
Definition TileProcessor.h:175
PacketTracker packetTracker_
Definition TileProcessor.h:165
PLCache packetLengthCache
Definition TileProcessor.h:125
uint32_t preCalculatedTileLen
Definition TileProcessor.h:176
uint64_t numProcessedPackets
Definition TileProcessor.h:157
uint32_t numlayers_
Definition TileProcessor.h:64
GrkImage * headerImage
Definition TileProcessor.h:122
Definition FlowComponent.h:21
enum _GRK_TILE_CACHE_STRATEGY GRK_TILE_CACHE_STRATEGY
GrkImage * image_
Definition mct.h:130
TileCodingParams * tcp_
Definition mct.h:131
grk_rect< uint32_t > grk_rect32
Definition TileCache.h:61
Copyright (C) 2016-2023 Grok Image Compression Inc.
Definition ICacheable.h:20
grk_rect< uint32_t > grk_rect32
Definition geometry.h:61
grk_plugin_tile * tile
Definition plugin_interface.h:64
uint16_t compno
Definition plugin_interface.h:43
Plugin tile.
Definition grok.h:739