1
0
mirror of https://github.com/danog/ton.git synced 2024-12-13 01:27:28 +01:00
ton/crypto/vm/cells/CellUsageTree.h

76 lines
2.3 KiB
C
Raw Normal View History

2019-09-07 12:03:22 +02:00
/*
This file is part of TON Blockchain Library.
TON Blockchain Library is free software: you can redistribute it and/or modify
it under the terms of the GNU Lesser General Public License as published by
the Free Software Foundation, either version 2 of the License, or
(at your option) any later version.
TON Blockchain Library 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 Lesser General Public License for more details.
You should have received a copy of the GNU Lesser General Public License
along with TON Blockchain Library. If not, see <http://www.gnu.org/licenses/>.
Copyright 2017-2019 Telegram Systems LLP
*/
#pragma once
#include "vm/cells/CellTraits.h"
#include "td/utils/int_types.h"
#include "td/utils/logging.h"
namespace vm {
class CellUsageTree : public std::enable_shared_from_this<CellUsageTree> {
public:
using NodeId = td::uint32;
struct NodePtr {
public:
NodePtr() = default;
NodePtr(std::weak_ptr<CellUsageTree> tree_weak, NodeId node_id)
: tree_weak_(std::move(tree_weak)), node_id_(node_id) {
}
bool empty() const {
return node_id_ == 0 || tree_weak_.expired();
}
bool on_load() const;
NodePtr create_child(unsigned ref_id) const;
bool mark_path(CellUsageTree* master_tree) const;
bool is_from_tree(const CellUsageTree* master_tree) const;
private:
std::weak_ptr<CellUsageTree> tree_weak_;
NodeId node_id_{0};
};
NodePtr root_ptr();
NodeId root_id() const;
bool is_loaded(NodeId node_id) const;
bool has_mark(NodeId node_id) const;
void set_mark(NodeId node_id, bool mark = true);
void mark_path(NodeId node_id);
NodeId get_parent(NodeId node_id);
NodeId get_child(NodeId node_id, unsigned ref_id);
void set_use_mark_for_is_loaded(bool use_mark = true);
NodeId create_child(NodeId node_id, unsigned ref_id);
private:
struct Node {
bool is_loaded{false};
bool has_mark{false};
NodeId parent{0};
std::array<td::uint32, CellTraits::max_refs> children{};
};
bool use_mark_{false};
std::vector<Node> nodes_{2};
void on_load(NodeId node_id);
NodeId create_node(NodeId parent);
};
} // namespace vm