fix: number of dimensions

sho/bit.py

@@ -8,11 +8,12 @@ from . import x,y,pb
 # Objective functions
 ########################################################################
 
-def cover_sum(sol, domain_width, sensor_range):
+def cover_sum(sol, domain_width, sensor_range, dim):
     """Compute the coverage quality of the given array of bits."""
     assert(0 < sensor_range <= math.sqrt(2))
     assert(0 < domain_width)
-    assert(len(sol)>0)
+    assert(dim > 0)
+    assert(len(sol) >= dim)
     domain = np.zeros((domain_width,domain_width))
     sensors = to_sensors(sol)
     cov = pb.coverage(domain, sensors, sensor_range*domain_width)

sho/num.py

@@ -21,11 +21,12 @@ def to_sensors(sol):
     return sensors
 
 
-def cover_sum(sol, domain_width, sensor_range):
+def cover_sum(sol, domain_width, sensor_range, dim):
     """Compute the coverage quality of the given vector."""
     assert(0 < sensor_range <= domain_width * math.sqrt(2))
     assert(0 < domain_width)
-    assert(len(sol)>0)
+    assert(dim > 0)
+    assert(len(sol) >= dim)
     domain = np.zeros((domain_width,domain_width))
     sensors = to_sensors(sol)
     cov = pb.coverage(domain, sensors, sensor_range*domain_width)

snp.py

@@ -93,7 +93,8 @@ if __name__=="__main__":
         val,sol = algo.greedy(
                 make.func(num.cover_sum,
                     domain_width = the.domain_width,
-                    sensor_range = the.sensor_range),
+                    sensor_range = the.sensor_range,
+                    dim = d * the.nb_sensors),
                 make.init(num.rand,
                     dim = d * the.nb_sensors,
                     scale = the.domain_width),
@@ -108,7 +109,8 @@ if __name__=="__main__":
         val,sol = algo.greedy(
                 make.func(bit.cover_sum,
                     domain_width = the.domain_width,
-                    sensor_range = the.sensor_range),
+                    sensor_range = the.sensor_range,
+                    dim = d * the.nb_sensors),
                 make.init(bit.rand,
                     domain_width = the.domain_width,
                     nb_sensors = the.nb_sensors),