from BondGraphTools import *

#%%

model = new(name= 'RC')

#%%
C = new("C", value= 1)
R = new("R", value= 1)
KCL = new("0")

#%%

add(model, R, C, KCL)
connect(R, KCL)
connect(KCL, C)

#%%

Sf = new("Sf")
add(model, Sf)
connect(Sf, KCL)
draw(model)


#%%
timespan = [0, 5]
x0 = {'x_0':1}

step_fn = {'u_0': 't < 1 ? 1 : 0'} # if t < 0 then 1 else 0
t, x = simulate(model, timespan=timespan, x0=x0, control_vars= step_fn)
plot(t,x)