from salome.geom import geomBuilder
geompy = geomBuilder.New(salome.myStudy)
import salome
import math


ptList = []

pi = 3.1415926
i = -4.0 * pi
step1 = 4.0 / 100.0
z = -2.0
theta = -4.0 * pi
step = 8.0 * pi / 100.0
while i <= 4 * pi:
	i += step
	theta += step
	z += step1
	r = z*z + 1.0
	x = r * math.sin(theta)
	y = r * math.cos(theta)
	v = geompy.MakeVertex(x, y, z)
	geompy.addToStudy(v, "Helix_%f"%i )
	ptList.append(v)
print "Done"
polyline = geompy.MakePolyline(ptList)
interpol = geompy.MakeInterpol(ptList)

geompy.addToStudy(polyline, "polyline" )
geompy.addToStudy(interpol, "interpol" )


startX = r * math.sin(-4.0 * pi)
startY = r * math.cos(-4.0 * pi)
startZ = -2.0
nextX =  r * math.sin(-4.0 * pi + step)
nextY =  r * math.sin(-4.0 * pi + step)
nextZ = -2.0 + step1


p1 = geompy.MakeVertex(startX, startY, startZ)
p2 = geompy.MakeVertex(nextX, nextY, nextZ)
vxy  = geompy.MakeVector(p1, p2)
disk2 = geompy.MakeDiskPntVecR(p1, vxy, 0.3)
geompy.addToStudy( disk2, 'Disk' )
Pipe_1 = geompy.MakePipe(disk2, interpol)
geompy.addToStudy( Pipe_1, 'Pipe_1' )

if salome.sg.hasDesktop():
  salome.sg.updateObjBrowser(True)