@slithy/prim-lib 0.3.0 → 0.3.1

package/dist/index.d.ts

@@ -38,6 +38,7 @@ interface Cfg {
     alpha: number;
     mutateAlpha: boolean;
     shapeTypes: Array<new (w: number, h: number) => ShapeInterface>;
+    shapeWeights?: number[];
     shapes: number;
     mutations: number;
     steps: number;

package/dist/index.js

@@ -349,6 +349,16 @@ var Shape = class {
   }
   static create(cfg) {
     const ctors = cfg.shapeTypes;
+    const weights = cfg.shapeWeights;
+    if (weights && weights.length === ctors.length) {
+      const total = weights.reduce((sum, w) => sum + w, 0);
+      let r = Math.random() * total;
+      for (let i = 0; i < weights.length; i++) {
+        r -= weights[i];
+        if (r <= 0) return new ctors[i](cfg.width, cfg.height);
+      }
+      return new ctors[ctors.length - 1](cfg.width, cfg.height);
+    }
     const index = Math.floor(Math.random() * ctors.length);
     const ctor = ctors[index];
     return new ctor(cfg.width, cfg.height);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@slithy/prim-lib",
-  "version": "0.3.0",
+  "version": "0.3.1",
   "description": "Core engine for primitive-based image reconstruction.",
   "type": "module",
   "exports": {