from jhplot.io import *
from jhplot import *

print "Write YAML data"
f=HFileYAML("test.yaml","w")
h1=H1D("test1",4,-2,2)
h1.fillGauss(100,0,2)
h2=H1D("test2",[-1, -0.75, -0.5, 0.5])
h2.fillGauss(200,0,2)

c1=HPlot("yaml files",600,400,2,1)
c1.visible()
c1.setAutoRange(1)
c1.setName("Original data")

c1.cd(1,1)
c1.draw(h1)
c1.draw(h2)

f.write("t0",h1)
f.write("t2",h2)
f.close()

print "Read YAML data:"
f=HFileYAML("test.yaml")
h11=f.read("t0")
h21=f.read("t2")
c1.cd(2,1)
c1.setName("Data from yaml")
c1.setAutoRange(1)
c1.draw(h11)
c1.draw(h21)
f.close()