def display(flags, bases, obstacles, field_function):
try:
from Gnuplot import GnuplotProcess
except ImportError:
print "Sorry. You don't have the Gnuplot module installed."
import sys
sys.exit(-1)
gp = GnuplotProcess(persist=True)
gp.write(gnuplot_header(-WORLDSIZE / 2, WORLDSIZE / 2))
gp.write(unset_arrow())
gp.write(draw_flags(flags))
gp.write(draw_bases(bases))
gp.write(draw_obstacles(obstacles))
gp.write(plot_field(field_function))
def visualize(self):
time.sleep(1)
forward_list = list(
linspace(self.ANIMATION_MIN, self.ANIMATION_MAX,
self.ANIMATION_FRAMES / 2))
backward_list = list(
linspace(self.ANIMATION_MAX, self.ANIMATION_MIN,
self.ANIMATION_FRAMES / 2))
anim_points = forward_list + backward_list
lastTime = time.time()
gp = GnuplotProcess(persist=False)
gp.write(self.gnuplot_header(-self.WORLDSIZE / 2, self.WORLDSIZE / 2))
gp.write(self.draw_obstacles(self.OBSTACLES))
for scale in cycle(anim_points):
self.mytanks, self.othertanks, self.flags, shots = self.BZRC.get_lots_o_stuff(
)
field_function = self.generate_field_function(scale)
gp.write(self.plot_field(field_function))
print >> sys.stderr, '%s: incorrect number of arguments' % execname
print >> sys.stderr, 'usage: %s hostname port' % sys.argv[0]
sys.exit(-1)
x = 0
for f in flags:
if f.color == flagColor:
FLAG_INT = x
x += 1
forward_list = list(
linspace(ANIMATION_MIN, ANIMATION_MAX, ANIMATION_FRAMES / 2))
backward_list = list(
linspace(ANIMATION_MAX, ANIMATION_MIN, ANIMATION_FRAMES / 2))
anim_points = forward_list + backward_list
lastTime = time.time()
gp = GnuplotProcess(persist=False)
gp.write(gnuplot_header(-WORLDSIZE / 2, WORLDSIZE / 2))
gp.write(draw_obstacles(OBSTACLES))
for scale in cycle(anim_points):
# if ((time.time() - lastTime) > 1):
# print 'hello world'
# lastTime = time.time()
time.sleep(.3)
mytanks, othertanks, flags, shots = BZRC.get_lots_o_stuff()
f = flags[FLAG_INT]
# print 'hello world'
field_function = generate_field_function(scale)
gp.write(plot_field(field_function))
# vim: et sw=4 sts=4
def __init__(self, world_size):
self.gp = GnuplotProcess(persist=False)
self.gp.write(self.gnuplot_header(-world_size / 2, world_size / 2))
def plot(obstacleList):
########################################################################
# Constants
# Output file:
FILENAME = 'fields.gpi'
# Size of the world (one of the "constants" in bzflag):
WORLDSIZE = 800
# How many samples to take along each dimension:
SAMPLES = 25
# Change spacing by changing the relative length of the vectors. It looks
# like scaling by 0.75 is pretty good, but this is adjustable:
VEC_LEN = 0.75 * WORLDSIZE / SAMPLES
# Animation parameters:
ANIMATION_MIN = 0
ANIMATION_MAX = 500
ANIMATION_FRAMES = 50
########################################################################
# Field and Obstacle Definitions
def generate_field_function(scale):
def function(x, y):
'''User-defined field function.'''
sqnorm = (x**2 + y**2)
if sqnorm == 0.0:
return 0, 0
else:
return x * scale / sqnorm, y * scale / sqnorm
return function
OBSTACLES = obstacleList
########################################################################
# Helper Functions
def gpi_point(x, y, vec_x, vec_y):
'''Create the centered gpi data point (4-tuple) for a position and
vector. The vectors are expected to be less than 1 in magnitude,
and larger values will be scaled down.'''
r = (vec_x**2 + vec_y**2)**0.5
if r > 1:
vec_x /= r
vec_y /= r
return (x - vec_x * VEC_LEN / 2, y - vec_y * VEC_LEN / 2,
vec_x * VEC_LEN, vec_y * VEC_LEN)
def gnuplot_header(minimum, maximum):
'''Return a string that has all of the gnuplot sets and unsets.'''
s = ''
s += 'set terminal png\n'
s += "set output 'fields.png'\n"
s += 'set xrange [%s: %s]\n' % (minimum, maximum)
s += 'set yrange [%s: %s]\n' % (minimum, maximum)
# The key is just clutter. Get rid of it:
s += 'unset key\n'
# Make sure the figure is square since the world is square:
s += 'set size square\n'
# Add a pretty title (optional):
#s += "set title 'Potential Fields'\n"
return s
def draw_line(p1, p2):
'''Return a string to tell Gnuplot to draw a line from point p1 to
point p2 in the form of a set command.'''
x1, y1 = p1
x2, y2 = p2
return 'set arrow from %s, %s to %s, %s nohead lt 3\n' % (x1, y1, x2,
y2)
def draw_obstacles(obstacles):
'''Return a string which tells Gnuplot to draw all of the obstacles.'''
s = 'unset arrow\n'
for obs in obstacles:
last_point = obs[0]
for cur_point in obs[1:]:
s += draw_line(last_point, cur_point)
last_point = cur_point
s += draw_line(last_point, obs[0])
return s
def plot_field(function):
'''Return a Gnuplot command to plot a field.'''
s = "plot '-' with vectors head\n"
separation = WORLDSIZE / SAMPLES
end = WORLDSIZE / 2 - separation / 2
start = -end
points = ((x, y) for x in linspace(start, end, SAMPLES)
for y in linspace(start, end, SAMPLES))
for x, y in points:
f_x, f_y = function(x, y)
plotvalues = gpi_point(x, y, f_x, f_y)
if plotvalues is not None:
x1, y1, x2, y2 = plotvalues
s += '%s %s %s %s\n' % (x1, y1, x2, y2)
s += 'e\n'
return s
########################################################################
# Plot the potential fields to a file
outfile = open(FILENAME, 'w')
print >> outfile, gnuplot_header(-WORLDSIZE / 2, WORLDSIZE / 2)
print >> outfile, draw_obstacles(OBSTACLES)
#field_function = generate_field_function(150)
#print >>outfile, plot_field(field_function)
########################################################################
# Animate a changing field, if the Python Gnuplot library is present
try:
from Gnuplot import GnuplotProcess
except ImportError:
print "Sorry. You don't have the Gnuplot module installed."
import sys
sys.exit(-1)
forward_list = list(
linspace(ANIMATION_MIN, ANIMATION_MAX, ANIMATION_FRAMES / 2))
backward_list = list(
linspace(ANIMATION_MAX, ANIMATION_MIN, ANIMATION_FRAMES / 2))
anim_points = forward_list + backward_list
gp = GnuplotProcess(persist=False)
gp.write(gnuplot_header(-WORLDSIZE / 2, WORLDSIZE / 2))
gp.write(draw_obstacles(OBSTACLES))
for scale in cycle(anim_points):
field_function = generate_field_function(scale)
gp.write(plot_field(field_function))
print >> outfile, draw_obstacles(OBSTACLES)
field_function = generate_field_function(150, type)
print >> outfile, plot_field(field_function)
########################################################################
# Animate a changing field, if the Python Gnuplot library is present
try:
from Gnuplot import GnuplotProcess
except ImportError:
print "Sorry. You don't have the Gnuplot module installed."
import sys
sys.exit(-1)
forward_list = list(
linspace(ANIMATION_MIN, ANIMATION_MAX, ANIMATION_FRAMES / 2))
backward_list = list(
linspace(ANIMATION_MAX, ANIMATION_MIN, ANIMATION_FRAMES / 2))
anim_points = forward_list + backward_list
gp = GnuplotProcess(persist=True)
gp.write(gnuplot_header(-WORLDSIZE / 2, WORLDSIZE / 2))
gp.write(draw_obstacles(OBSTACLES))
#for scale in cycle(anim_points):
# field_function = generate_field_function(scale)
# gp.write(plot_field(field_function))
field_function = generate_field_function(150, type)
gp.write(plot_field(field_function))
