component representation revolution! Vive la révolution

1 files changed, 154 insertions, 0 deletions

diff --git a/build/content/coso.py b/build/content/coso.py
new file mode 100644
index 0000000..e68d89d
--- /dev/null
+++ b/build/content/coso.py
@@ -0,0 +1,154 @@
+import json
+
+def vectorize (obj, forbidlist, _):
+    for c in obj:
+        for key in c:
+            if not(key in forbidlist):
+                val = c[key]
+                try: val = int(val);
+                except: pass
+                try: val = float(val);
+                except: pass
+                c[key] = [val for i in range(3)]
+    return obj
+    
+def query (obj, allowlist, op):
+    for c in obj:
+        for key in c:
+            if key in allowlist:
+                c[key] = op(c[key])
+    return obj
+                
+def update (obj, forbidlist, op):
+    return op(obj)
+
+def select (obj, allowlist, op):
+    l = []
+    for c in obj:
+        for key in c:
+            if key in allowlist:
+                l += [c]
+    return l
+                
+
+def comp(filename, l, fun, op=None, rewrite=None):
+    with open(filename+".txt", "r") as f: obj = json.loads(f.read())
+    nobj = fun(obj, l, op)
+    if rewrite is None:
+        with open(filename+".txt", "w") as f:
+            f.write(json.dumps(nobj, indent=4))
+    return nobj
+    
+def change_values():
+    amt = .5
+    perc_inc = lambda x: [x[0]*(1.0+(i/2.0)*amt) for i in range(3)]
+    perc_dec = lambda x: [x[0]*(1.0-(i/2.0)*amt) for i in range(3)]
+    
+    comp("weapons", ["damage"], query, perc_inc)
+    comp("weapons", ["cooldown"], query, perc_dec)
+    comp("weapons", ["weight"], query, perc_dec)
+    comp("weapons", ["range"], query, perc_inc)
+    
+    comp("armor", ["weight"], query, perc_dec)
+    comp("armor", ["pierce"], query, perc_inc)
+    comp("armor", ["spread"], query, perc_inc)
+    comp("armor", ["fusion"], query, perc_inc)
+    comp("armor", ["explosive"], query, perc_inc)
+    comp("armor", ["laser"], query, perc_inc)
+    comp("armor", ["emp"], query, perc_inc)
+    comp("armor", ["impact"], query, perc_inc)
+    
+    comp("batteries", ["weight"], query, perc_dec)
+    comp("batteries", ["capacity"], query, perc_inc)
+    
+    comp("chassis", ["weight"], query, perc_dec)
+    comp("chassis", ["weight_max"], query, perc_inc)
+    comp("chassis", ["hp"], query, perc_inc)
+    comp("chassis", ["speed"], query, perc_inc)
+    
+    comp("brains", ["upkeep"], query, perc_dec)
+    
+    comp("augments", ["weight"], query, perc_inc)
+    comp("augments", ["damage_pierce"], query, perc_inc)
+    comp("augments", ["damage_spread"], query, perc_inc)
+    comp("augments", ["damage_fusion"], query, perc_inc)
+    comp("augments", ["damage_explosive"], query, perc_inc)
+    comp("augments", ["damage_laser"], query, perc_inc)
+    comp("augments", ["damage_emp"], query, perc_inc)
+    comp("augments", ["damage_impact"], query, perc_inc)
+    comp("augments", ["armor_pierce"], query, perc_inc)
+    comp("augments", ["armor_spread"], query, perc_inc)
+    comp("augments", ["armor_fusion"], query, perc_inc)
+    comp("augments", ["armor_explosive"], query, perc_inc)
+    comp("augments", ["armor_laser"], query, perc_inc)
+    comp("augments", ["armor_emp"], query, perc_inc)
+    comp("augments", ["armor_impact"], query, perc_inc)
+    comp("augments", ["add_range"], query, perc_inc)
+    comp("augments", ["add_cooldown"], query, perc_inc)
+    comp("augments", ["recharge"], query, perc_inc)
+    comp("augments", ["add_hp"], query, perc_inc)
+
+def get_data ():
+    # static data analysis, compute image
+    get_first = lambda x: x
+    xsc = comp("weapons", ["damage"], select, get_first, False)
+    if len(xsc) < 2: pygame.quit(); return
+    wstep = 200
+    w = len(xsc)*wstep
+    hmax = 512
+    xs = [x["damage"][0]/x["cooldown"][0]*x["weight"][0]*x["range"][0] for x in xsc]
+    maxx = max(xs); minx = min(xs)
+    
+    import pygame
+    pygame.init()
+    pygame.display.set_mode((20, 20))
+    size = (w, hmax)
+    scr = pygame.Surface((size[0], size[1]+15)).convert_alpha()
+    scr.fill((0,0,0))
+    font = pygame.font.SysFont("Courier New", 10)
+    
+    for i in range(len(xs)):
+        x = xs[i]
+        h = (x-minx)/(maxx-minx) *size[1]
+        rect = wstep*i, 15+size[1]-h, wstep, h
+        pygame.draw.rect(scr, (255,255,255), rect)
+        txt = font.render(xsc[i]["name"], True, (255,255,255))
+        scr.blit(txt, (rect[0], rect[1]-12))
+    pygame.image.save(scr, "__results.png")
+    pygame.quit()
+
+if __name__ == "__main__":
+    try:
+        #comp("weapons", ["name", "damage_type"])
+        #comp("augments", ["name"])
+        #comp("batteries", ["name"])
+        #comp("armor", ["name"])
+        #comp("chassis", ["name"])
+        #comp("brains", ["name"])
+        
+        #change_values()
+        
+        #get_data();
+        
+        
+        forbidlist = ["name"]
+        def rename (obj):
+            nobj = []
+            for d in obj:
+                nd = {}
+                for (key, value) in d.items():
+                    if not(key in forbidlist): nd["base_"+key] = value
+                    else: nd[key] = value
+                nobj += [nd]
+            return nobj
+        #comp("chassis", ["name"], update, rename)
+        comp("weapons", ["name"], update, rename)
+        comp("augments", ["name"], update, rename)
+        comp("batteries", ["name"], update, rename)
+        comp("brains", ["name"], update, rename)
+        comp("armor", ["name"], update, rename)
+        
+    except:
+        import traceback
+        traceback.print_exc()
+        input()