TagTree.h

Go to the documentation of this file.

/*
 *    Copyright (C) 2016-2020 Grok Image Compression Inc.
 *
 *    This source code is free software: you can redistribute it and/or  modify
 *    it under the terms of the GNU Affero General Public License, version 3,
 *    as published by the Free Software Foundation.
 *
 *    This source code 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 Affero General Public License for more details.
 *
 *    You should have received a copy of the GNU Affero General Public License
 *    along with this program.  If not, see <http://www.gnu.org/licenses/>.
 *
 *
 *    This source code incorporates work covered by the BSD 2-clause license.
 *    Please see the LICENSE file in the root directory for details.
 *
 */
 
#pragma once
 
namespace grk {
 
const uint32_t tag_tree_uninitialized_node_value = 999;
 
struct TagTreeNode {
 
        TagTreeNode() :
                        parent(nullptr), value(0), low(0), known(0) {
        }
 
        TagTreeNode *parent;
        int64_t value;
        int64_t low;
        uint32_t known;
};
 
class TagTree {
 
public:
 
        TagTree(uint64_t numleafsh, uint64_t numleafsv);
        ~TagTree();
 
        bool init(uint64_t num_leafs_h, uint64_t num_leafs_v);
 
        void reset();
        void setvalue(uint64_t leafno, int64_t value);
        bool compress(BitIO *bio, uint64_t leafno, int64_t threshold);
        void decompress(BitIO *bio, uint64_t leafno, int64_t threshold,
                        uint8_t *decoded);
 
        void decodeValue(BitIO *bio, uint64_t leafno, int64_t threshold,
                        uint64_t *value);
 
private:
 
        uint64_t numleafsh;
        uint64_t numleafsv;
        uint64_t numnodes;
        TagTreeNode *nodes;
        uint64_t nodes_size; /* maximum size taken by nodes */
 
};
 
}