Add torus topology.

diff --git a/deployment.cpp b/deployment.cpp
index c5a1f9d40decaab7bc73e6f05e50c87bf6ca845a..d582dc66e6b5327b3544e7100276eec61ae1fe38 100644 (file)
--- a/deployment.cpp
+++ b/deployment.cpp
@@ -122,6 +122,66 @@ void deployment_star::generate()
         set_link(0, i);
 }
 
-// void deployment_torus::generate()
-// {
-// }
+void deployment_torus::generate()
+{
+    unsigned a = 0;
+    unsigned b = size();
+    while (a + 1 < b) {
+        unsigned c = (a + b) / 2;
+        if (c * c < size())
+            a = c;
+        else
+            b = c;
+    }
+    unsigned width = b;
+
+    unsigned first_on_last_line = (size() - 1) - (size() - 1) % width;
+    DEBUG4("torus size = %u ; width = %u ; height = %u ; foll = %u",
+           (unsigned )size(), width,
+           (unsigned )(size() / width + !!(size() % width)),
+           first_on_last_line);
+    for (unsigned i = 0; i < size(); i++) {
+        unsigned next_line;
+        unsigned prev_line;
+        unsigned next_column;
+        unsigned prev_column;
+
+        next_line = i + width;
+        if (next_line >= size())
+            next_line %= width; // rewind
+
+        if (i >= width) {
+            prev_line = i - width;
+        } else {
+            prev_line = first_on_last_line + i; // rewind
+            if (prev_line >= size())
+                prev_line -= width; // need to go at last but one line
+        }
+
+        if (i != size() - 1) {
+            next_column = i + 1;
+            if (next_column % width == 0)
+                next_column -= width; // rewind
+        } else {
+            next_column = first_on_last_line; // special case for last cell
+        }
+
+        if (i % width != 0) {
+            prev_column = i - 1;
+        } else if (i < first_on_last_line) {
+            prev_column = i + width - 1; // rewind
+        } else {
+            prev_column = size() - 1; // special case for 1st cell of last line
+        }
+        if (next_line != i) {
+            set_neighbor(i, next_line);
+            if (prev_line != next_line)
+                set_neighbor(i, prev_line);
+        }
+        if (next_column != i) {
+            set_neighbor(i, next_column);
+            if (prev_column != next_column)
+                set_neighbor(i, prev_column);
+        }
+    }
+}

diff --git a/deployment.h b/deployment.h
index 2009b36c8a8d28e58f6823fc5a6bbe4378a8599b..cba191b62536b568c5a4853cecb783cde74b6840 100644 (file)
--- a/deployment.h
+++ b/deployment.h
@@ -37,7 +37,7 @@ DEPLOYMENT(hcube);
 DEPLOYMENT(line);
 DEPLOYMENT(ring);
 DEPLOYMENT(star);
-// DEPLOYMENT(torus);
+DEPLOYMENT(torus);
 
 #undef DEPLOYMENT
 

diff --git a/options.cpp b/options.cpp
index 1b12ce7e8a1136811c3e45c2dc58ddfe2ffc0240..668aed3e044ae27c3b5df48beb392d76dd9de462 100644 (file)
--- a/options.cpp
+++ b/options.cpp
@@ -61,7 +61,7 @@ namespace opt {
         NOL_INSERT("ring", "ring topology", deployment_ring);
         NOL_INSERT("star", "star topology, initial load at center",
                    deployment_star);
-        // NOL_INSERT("torus", "torus topology", deployment_torus);
+        NOL_INSERT("torus", "torus topology", deployment_torus);
     }
 
 } // namespace opt