helpers.py

from random import randint
from config import *

def boundName(num, bounds=[("1 zeros",0,100),("2 zeros",100,1000),("3 zeros",1000,10000)]):
    for bound in bounds:
        if bound[0]<=num<bound[1]:
            return bound[2]
    return False

def boundNum(name, bounds=[("1 zeros",0,100),("2 zeros",100,1000),("3 zeros",1000,10000)]):
    for bound in bounds:
        if bound[2]==name:
            return randint(bound[0],bound[1])
    return False

def randColor(min=(0,0,0),max=(255,255,255)):
    out=""
    for val in [hex(randint(min[i],max[i])) for i in range(3)]:
        out+=val[2:]
        if len(val[2:])<2:
            out+="0"
    return out

def randfloat(mn,mx):
    if type(mn)==type(mx)==int:
        return randint(mn,mx)

    decimals=max(len(str(mn))-2,len(str(mx))-2)
    return randint(int(mn*(10**decimals)),int(mx*(10**decimals)))/(10**decimals)

def randitem(lst):
    if lst:
        return lst[randint(0,len(lst)-1)]
    return None

def trykey(ky,dct,excep=None):
    if ky in dct.keys():
        return dct[ky]
    return excep

blocks=[]

class Block:
    def __init__(self,features,args,extends=None,extras=[],children=[],base=False):
        self.extends=False
        self.extendBlock=None
        if extends:
            self.extends=True
            self.extendBlock=extends

        self.args=args
        self.features=features
        self.id=len(blocks)+1+kbaseSettings["id offset"]
        self.extras=extras
        self.children=children

        self.actualized=False
        self.base=base

    def actualize(self,childize=True):
        # the main point of actualize() is to finish all features and prepare the block for randomization
        # and also verify self integrity so we don't gave duplicate features/args

        self.sverify()

        if (not self.actualized) or self.extras:#shhhhhh
            for extra in self.extras:
                if randfloat(0.0,1.0)<=extra[0]:#chance
                    self.features.append(extra[1])
                    if len(extra)!=2:
                        for arg in extra[2]:
                            self.args.append(arg)

            new_children=[]
            if childize and self.children!=[] and type(self.children[0])==list:
                for child in self.children:
                    blk=Block(
                        trykey("features",child,excep=[]),
                        child["args"],
                        extends=False,
                        extras=trykey("extras",child,excep=[]),
                    )
                    blk.actualize(childize=False)
                    new_child=(child["name"],blk)
                    new_children.append(new_child)

            self.children=new_children
            self.actualized=True

            self.sverify()

        # print(str(self.id)+"  act args: "+str(self.args))

    def sverify(self):
        nargs=[]
        for arg in self.args:
            if not arg in nargs:
                nargs.append(arg)
        self.args=nargs
        nfeat=[]
        for feat in self.features:
            if not feat in nfeat:
                nfeat.append(feat)
        self.features=nfeat
        next=[]
        for ext in self.extras:
            if not ext in next:
                next.append(ext)
        self.extras=next
        nkid=[]
        for kid in self.children:
            if not kid in nkid:
                nkid.append(kid)
        self.children=nkid

    def text(self,out="str",id=True):#out can be str or list
        features=self.features
        args=self.args
        # print(args)

        if self.extends:
            features+=self.extendBlock.features#append their features

        lines=["{"]
        if id:
            lines.append("\t"+str(self.id)+",")
        if self.extends:
            lines.append("\textends="+str(self.extendBlock.id)+",")
        if features:
            if self.base:
                lines+=["\tfeatures=NOPALETTE,\n\tname=\"Base Block\","]
            else:
                lines+=["\tfeatures="+"|".join(features)+","]#{id,feats=x|y|z,

        for arg in args:#arg=value,
            if len(arg)==3:
                lines.append("\t"+arg[0]+"="+str(randfloat(arg[1],arg[2]))+",")
            elif len(arg)==2:
                if arg[1] in ["0x","0xff"]:
                    lines.append("\t"+arg[0]+"="+arg[1]+randColor()+",")
                elif type(arg[1]) in [list,tuple]:
                    lines.append("\t"+arg[0]+"="+randitem(arg[1])+",")
                else:
                    lines.append("\t"+arg+",")
            else:
                lines.append("\t"+arg+",")

        # print(self.children)
        # self.actualize()
        # print(self.children)
        for child in self.children:
            lines.append("\t"+child[0]+"={\n"+"".join(["\t\t"+i+"\n" for i in child[1].text(out="list",id=False)[1:-1]])+"\n\t}")
        lines.append("}\n")

        if out=="str":
            return "\n".join(lines)
        return lines

def genBases(num=kbaseSettings["colors"],features=kbaseSettings["features"],args=kbaseSettings["args"]):
    global blocks
    out=[]
    for i in range(num):
        block=Block(features,args+[("sort=500")],base=True)
        block.actualize()
        blocks.append(block)
        out.append(block)
    return out