Replace if-else chain with switch statement.

leela-zero · Jul 23, 2018 · 8e5aab3 · 8e5aab3
1 parent d267299
commit 8e5aab3
Showing 1 changed file with 19 additions and 30 deletions.
diff --git a/src/Network.cpp b/src/Network.cpp
@@ -206,39 +206,28 @@ std::pair<int, int> Network::load_v1_network(std::istream& wtfile) {
                 process_bn_var(weights);
                 m_batchnorm_stddevs.emplace_back(weights);
             }
-        } else if (linecount == plain_conv_wts) {
-            m_conv_pol_w = std::move(weights);
-        } else if (linecount == plain_conv_wts + 1) {
-            m_conv_pol_b = std::move(weights);
-        } else if (linecount == plain_conv_wts + 2) {
-            std::copy(cbegin(weights), cend(weights), begin(m_bn_pol_w1));
-        } else if (linecount == plain_conv_wts + 3) {
-            process_bn_var(weights);
-            std::copy(cbegin(weights), cend(weights), begin(m_bn_pol_w2));
-        } else if (linecount == plain_conv_wts + 4) {
-            std::copy(cbegin(weights), cend(weights), begin(m_ip_pol_w));
-        } else if (linecount == plain_conv_wts + 5) {
-            std::copy(cbegin(weights), cend(weights), begin(m_ip_pol_b));
-        } else if (linecount == plain_conv_wts + 6) {
-            m_conv_val_w = std::move(weights);
-        } else if (linecount == plain_conv_wts + 7) {
-            m_conv_val_b = std::move(weights);
-        } else if (linecount == plain_conv_wts + 8) {
-            std::copy(cbegin(weights), cend(weights), begin(m_bn_val_w1));
-        } else if (linecount == plain_conv_wts + 9) {
-            process_bn_var(weights);
-            std::copy(cbegin(weights), cend(weights), begin(m_bn_val_w2));
-        } else if (linecount == plain_conv_wts + 10) {
-            std::copy(cbegin(weights), cend(weights), begin(m_ip1_val_w));
-        } else if (linecount == plain_conv_wts + 11) {
-            std::copy(cbegin(weights), cend(weights), begin(m_ip1_val_b));
-        } else if (linecount == plain_conv_wts + 12) {
-            std::copy(cbegin(weights), cend(weights), begin(m_ip2_val_w));
-        } else if (linecount == plain_conv_wts + 13) {
-            std::copy(cbegin(weights), cend(weights), begin(m_ip2_val_b));
+        } else {
+            switch (linecount - plain_conv_wts) {
+                case  0: m_conv_pol_w = std::move(weights); break;
+                case  1: m_conv_pol_b = std::move(weights); break;
+                case  2: std::copy(cbegin(weights), cend(weights), begin(m_bn_pol_w1)); break;
+                case  3: std::copy(cbegin(weights), cend(weights), begin(m_bn_pol_w2)); break;
+                case  4: std::copy(cbegin(weights), cend(weights), begin(m_ip_pol_w)); break;
+                case  5: std::copy(cbegin(weights), cend(weights), begin(m_ip_pol_b)); break;
+                case  6: m_conv_val_w = std::move(weights); break;
+                case  7: m_conv_val_b = std::move(weights); break;
+                case  8: std::copy(cbegin(weights), cend(weights), begin(m_bn_val_w1)); break;
+                case  9: std::copy(cbegin(weights), cend(weights), begin(m_bn_val_w2)); break;
+                case 10: std::copy(cbegin(weights), cend(weights), begin(m_ip1_val_w)); break;
+                case 11: std::copy(cbegin(weights), cend(weights), begin(m_ip1_val_b)); break;
+                case 12: std::copy(cbegin(weights), cend(weights), begin(m_ip2_val_w)); break;
+                case 13: std::copy(cbegin(weights), cend(weights), begin(m_ip2_val_b)); break;
+            }
         }
         linecount++;
     }
+    process_bn_var(m_bn_pol_w2);
+    process_bn_var(m_bn_val_w2);
 
     return {channels, static_cast<int>(residual_blocks)};
 }