from pysketcher import *

L = 12.
H = L/6
W = L/6

xmax = L
drawing_tool.set_coordinate_system(xmin=-L, xmax=xmax,
                                   ymin=-1, ymax=L+H,
                                   axis=False,
                                   instruction_file='tmp_mpl.py')
x = 0
drawing_tool.set_linecolor('black')

def make_dashpot(x):
    d_start = (-L,2*H)
    d = Dashpot(start=d_start, total_length=L+x, width=W,
                bar_length=3*H/2, dashpot_length=L/2, piston_pos=H+x)
    d.rotate(-90, d_start)
    return d

def make_spring(x):
    s_start = (-L,4*H)
    s = Spring(start=s_start, length=L+x, bar_length=3*H/2, teeth=True)
    s.rotate(-90, s_start)
    return s

d = make_dashpot(0)
s = make_spring(0)

M = Rectangle((0,H), 4*H, 4*H).set_linewidth(4)
left_wall = Rectangle((-L,0),H/10,L).set_filled_curves(pattern='/')
ground = Wall(x=[-L/2,L], y=[0,0], thickness=-H/10)
wheel1 = Circle((H,H/2), H/2)
wheel2 = wheel1.copy()
wheel2.translate(point(2*H, 0))

fontsize = 18
text_m = Text('$m$', (2*H, H+2*H), fontsize=fontsize)
text_kx = Text('$kx$', (-L/2, H+4*H), fontsize=fontsize)
text_bv = Text('$b\dot x$', (-L/2, H), fontsize=fontsize)
x_axis = Axis((2*H, L), H, '$x(t)$', fontsize=fontsize,
              label_spacing=(0.04, -0.01))
x_axis_start = Line((2*H, L-H/4), (2*H, L+H/4)).set_linewidth(4)

fig = Composition({
    'spring': s, 'mass': M, 'left wall': left_wall,
    'ground': ground, 'wheel1': wheel1, 'wheel2': wheel2,
    'text_m': text_m, 'text_kx': text_kx,
    'x_axis': x_axis, 'x_axis_start': x_axis_start})

fig.draw()
drawing_tool.display()
drawing_tool.savefig('tmp_oscillator_spring')

drawing_tool.erase()

fig['dashpot'] = d
fig['text_bv'] = text_bv

# or fig = Composition(dict(fig=fig, dashpot=d, text_bv=text_bv))
fig.draw()

drawing_tool.display()
drawing_tool.savefig('tmp_oscillator')

drawing_tool.erase()

text_kx = Text('$s(u)$', (-L/2, H+4*H), fontsize=fontsize)
text_bv = Text('$f(\dot u)$', (-L/2, H), fontsize=fontsize)
x_axis = Axis((2*H, L), H, '$u(t)$', fontsize=fontsize,
              label_spacing=(0.04, -0.01))
F_force = Force((4*H, H+2*H), (4*H+H, H+2*H), '$F(t)$',
                text_spacing=(0.035, -0.01), text_pos='end', fontsize=fontsize)
fig['text_kx'] = text_kx
fig['text_bv'] = text_bv
fig['x_axis'] = x_axis
fig['F_force'] = F_force
fig.draw()
drawing_tool.savefig('tmp_oscillator_general')

input()