#!/usr/bin/python

import math

#hp = {'Dx':1.,'Dy':1.,'r':1.,'L':1.,'phi':0.}
hp = {'Dx':1.,'Dy':1.,'r':1.,'L':1.,'phi':math.pi/4}

## ____________________
def frange(start,end,nintervals):
    step = (float(end)-float(start))/float(nintervals-1)
    return [start+step*i for i in range(nintervals)]

from ROOT import *
from clean_canvas import *
gStyle = makegstyle(gStyle)
#g = TGraph2D()
g = TPolyLine3D()
g2 = TGraph()
for i,z in enumerate(frange(0,4.,1000)):
    fz = 2*math.pi*(z/hp['L'])+hp['phi']
    x = hp['Dx']+hp['r']*math.cos(fz)
    y = hp['Dy']+hp['r']*math.sin(fz)
    g.SetPoint(i,z,x,y)
#    while fz>2*math.pi: fz -= 2*math.pi
    g2.SetPoint(i,z,fz/math.pi)
c1 = TCanvas()
c1.Divide(2,1)
#g.Draw('P')
c1.cd(1)
g.Draw()
c1.cd(2)
g.Draw()
#g.DrawOutlineCube(0,-3,-3,10,3,3)
c2 = TCanvas()
c2.cd()
g2.Draw('AL')
au = raw_input('>')