Changeset 3bf812b for doc/theses/thierry_delisle_PhD/code

Timestamp:

May 22, 2020, 4:32:11 PM (4 years ago)

Author:

Thierry Delisle <tdelisle@…>

Branches:

ADT, arm-eh, ast-experimental, enum, forall-pointer-decay, jacob/cs343-translation, master, new-ast, new-ast-unique-expr, pthread-emulation, qualifiedEnum

Children:

8e1b1bb

Parents:

8f4f3e0

Message:

snzi creator now supports m-ary trees

File:

• doc/theses/thierry_delisle_PhD/code/snzi.hpp (modified) (2 diffs)

doc/theses/thierry_delisle_PhD/code/snzi.hpp

@@ -11 +11 @@
         std::unique_ptr<snzi_t::node[]> nodes;
 
-        snzi_t(unsigned depth);
+        snzi_t(unsigned depth, unsigned base = 2);
 
         void arrive(int idx) {
-                idx &= mask;
+                idx %= mask;
                 nodes[idx].arrive();
         }
 
         void depart(int idx) {
-                idx &= mask;
+                idx %= mask;
                 nodes[idx].depart();
         }
@@ -145 +145 @@
 };
 
-snzi_t::snzi_t(unsigned depth)
-        : mask((1 << depth) - 1)
-        , root( ((1 << depth) * 2) - 2)
+snzi_t::snzi_t(unsigned depth, unsigned base)
+        : mask( std::pow(base, depth) )
+        , root( ((std::pow(base, depth + 1) - 1) / (base -1)) - 1 )
         , nodes(new node[ root + 1 ]())
 {
-        int width = (1 << depth);
+        int width = std::pow(base, depth);
         for(int i = 0; i < root; i++) {
-                nodes[i].parent = &nodes[(i / 2) + width ];
+                nodes[i].parent = &nodes[(i / base) + width ];
         }
 }