Add back Clone() method to the Tree class (#235)

dmlc · Dec 17, 2020 · deed6c7 · deed6c7
1 parent 9fb79c3
commit deed6c7
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/include/treelite/tree.h b/include/treelite/tree.h
@@ -259,6 +259,8 @@ class Tree {
   Tree(Tree&&) noexcept = default;
   Tree& operator=(Tree&&) noexcept = default;
 
+  inline Tree<ThresholdType, LeafOutputType> Clone() const;
+
   inline const char* GetFormatStringForNode();
   inline void GetPyBuffer(std::vector<PyBufferFrame>* dest);
   inline void InitFromPyBuffer(std::vector<PyBufferFrame>::iterator begin,

diff --git a/include/treelite/tree_impl.h b/include/treelite/tree_impl.h
@@ -396,6 +396,19 @@ inline void InitScalarFromPyBuffer(T* scalar, PyBufferFrame buffer) {
   *scalar = *t;
 }
 
+template <typename ThresholdType, typename LeafOutputType>
+inline Tree<ThresholdType, LeafOutputType>
+Tree<ThresholdType, LeafOutputType>::Clone() const {
+  Tree<ThresholdType, LeafOutputType> tree;
+  tree.num_nodes = num_nodes;
+  tree.nodes_ = nodes_.Clone();
+  tree.leaf_vector_ = leaf_vector_.Clone();
+  tree.leaf_vector_offset_ = leaf_vector_offset_.Clone();
+  tree.matching_categories_ = matching_categories_.Clone();
+  tree.matching_categories_offset_ = matching_categories_offset_.Clone();
+  return tree;
+}
+
 template <typename ThresholdType, typename LeafOutputType>
 inline const char*
 Tree<ThresholdType, LeafOutputType>::GetFormatStringForNode() {