Python vortex 예제들

예제 #1

파일: main2.py 프로젝트: devyeshtandon/ParticleMethods

#############################################################################
# Simulation
#############################################################################
for t in time:
	X = reflection(vort[1].xy)
	vort[2].xy = X
	vort[2].updatexy = X
	vort[2].strength = -vort[1].strength/abs(vort[1].xy)

	vort[0].strength = -vort[2].strength

	target = [vort[i].xy for i in range(num_of_targets)]

	for i in range(num_of_targets):
		if vort[i].ele_type != "tracer":
			field[i][:] = vortex(vort[i], target)
	print field

	for i in range(num_of_targets):
		for j in range(num_of_vortex):
			if vort[i].fixed == 0:
				vort[i].updatexy = field[j][i]*dt + vort[i].updatexy

	target = [vort[i].updatexy for i in range(num_of_targets)]

	print target

	for i in range(num_of_targets):
		if vort[i].ele_type != "tracer":
			field2[i][:] = vortex(vort[i], target, "update")
	

예제 #2

파일: main_euler.py 프로젝트: devyeshtandon/ParticleMethods

time     = pylab.arange(0, sim_time, dt)

X = []
Y = []

alpha = 0.5

#############################################################################
# Simulation
#############################################################################
for t in time:	
	target = [vort[i].xy for i in range(num_of_targets)]

	for i in range(num_of_targets):
		if vort[i].ele_type != "tracer":
			field[i][:] = vortex(vort[i], target)

	for i in range(num_of_targets):
		for j in range(num_of_vortex):
			if vort[i].fixed == 0:
				vort[i].xy = field[j][i]*dt + vort[i].xy
	
#	if (round(t/dt)%round(1/dt) == 0):
	X.extend(vort[i].xy.real for i in range(num_of_targets))
	Y.extend(vort[i].xy.imag for i in range(num_of_targets))

	line.set_xdata(X)
	line.set_ydata(Y)
	pylab.draw()

raw_input()