formulate.py

import pandas as pd
import numpy as np



class Dimension:
    Spatial = -3.0
    Angular = 0.
    Order = 0
    Flat = True


class Operation:
    def __init__(self, seed = "+"):
        self.op = seed
        self.a = "x"
        self.b = "x"
        self.c = ""

    def split_b(self, sign = "+"):
        self.b = ""
        opb = Operation(sign)
        return opb

    d_a = Dimension()
    d_b = Dimension()
    d_c = Dimension()
    a = ""
    b = ""
    c = ""
    op = ""




class Generator:
    def __init__(self, dim = 1, seed = "+"):
        op = Operation(seed)
        self.ops = [op]
    
    def split(self, sign):
        self.ops = self.ops + [self.ops[len(self.ops)-1].split_b(sign)]

    def formulate(self):
        formula = "p[0]+"
        ipar = 1
        for o in self.ops:
            p = ""
            if o.op == "+":
                p = "p[" + str(ipar) + "]*"
                ipar += 1
            formula = formula + o.a + o.op + p + o.b
        return formula




def funk(x, y, p = [], formula = "p[0] + ( (p[1]*x) + ( p[2]*(x**2) ) )"):
    return eval(formula)


def formulate():
    #   return "2. * np.cos(np.arctan2(y,x)) / np.sqrt(x**2+y**2)"
    return "2. * y*y/x / np.sqrt(x**3+y**2+x**2)"





myops = np.random.choice(["+","*"],2,p=[0.8,0.2])
gen = Generator("+")

for o in myops:
    gen.split(o)

print(gen.formulate())