From 1e20c23729f120e34a0197093fe6ef932ca416d2 Mon Sep 17 00:00:00 2001
From: Dmitry Kokorin <dmitry@kokorin.org>
Date: Wed, 16 Mar 2016 15:37:48 +0300
Subject: [PATCH] grains: with std::conditional

---
 cpp/grains/grains.cpp | 42 ++++++++++++++++++++++++++++++++++++++++--
 1 file changed, 40 insertions(+), 2 deletions(-)

diff --git a/cpp/grains/grains.cpp b/cpp/grains/grains.cpp
index d213d8f..b7646ff 100644
--- a/cpp/grains/grains.cpp
+++ b/cpp/grains/grains.cpp
@@ -1,18 +1,56 @@
 #include "grains.h"
 
+#include <limits>
+#include <stdexcept>
+
+
 namespace grains {
 
+using namespace std;
+
+
+namespace {
+    
 static const index_t number_of_squares = 64;
 
+template <typename T>
+struct total_as_max
+{
+    static inline T total()
+    {
+        return numeric_limits<T>::max();
+    }
+};
+
+template <typename T>
+struct total_by_shift
+{
+    static inline T total()
+    {
+        return T(1) << number_of_squares - 1;
+    }
+};
+
+static_assert(number_of_squares <= numeric_limits<ulong_t>::digits,
+              "grains::ulong_t is too small for grains::total");
+
+typedef conditional<number_of_squares == numeric_limits<ulong_t>::digits,
+                    total_as_max<ulong_t>,
+                    total_by_shift<ulong_t>>::type total_policy_t;
+}
+
 
 ulong_t square(index_t index)
 {
-    return 1ULL << (index - 1);
+    if (index < 1 || index > number_of_squares)
+        throw domain_error("invalid square number");
+
+    return ulong_t(1) << (index - 1);
 }
 
 ulong_t total()
 {
-    return (1ULL << number_of_squares) - 1;
+    return total_policy_t::total();
 }
 
 }