1 |
gezelter |
4407 |
#!/opt/local/bin/python2.7 |
2 |
|
|
|
3 |
|
|
'''An example script that generates a picture of the field lines for an electric dipole.''' |
4 |
|
|
|
5 |
|
|
import math |
6 |
|
|
import VectorFieldPlot as vfp |
7 |
|
|
|
8 |
|
|
|
9 |
|
|
# create a document. we specify the file name and image size here |
10 |
|
|
doc = vfp.FieldplotDocument( 'PointCharges', width=800,height=600,unit=100) |
11 |
|
|
|
12 |
|
|
# create a field opbject |
13 |
|
|
field = vfp.Field() |
14 |
|
|
# add the point charges |
15 |
|
|
field.add_element('monopoles' , [ [ 1,0, 1] # the positive charge |
16 |
|
|
, [-1,0,-1] # the negative charge |
17 |
|
|
] ) |
18 |
|
|
|
19 |
|
|
# draw the charges for the field on the document |
20 |
|
|
doc.draw_charges(field) |
21 |
|
|
|
22 |
|
|
# start drawing the field lines |
23 |
|
|
# we are going to draw 20 field lines comming off of the positive charge at uniformly spaced angles. |
24 |
|
|
N = 20 |
25 |
|
|
for i in range(N): |
26 |
|
|
# compute the angle that the line will start off at |
27 |
|
|
angle = i * 2.*math.pi / (N-1) |
28 |
|
|
# generate the line |
29 |
|
|
# this takes the initial position, the x and y components of the initial direction, and whether or not should go forward |
30 |
|
|
# or backward. |
31 |
|
|
line = vfp.FieldLine( field, [1,0], start_v=[math.cos( angle ) , math.sin( angle )],directions='forward' ) |
32 |
|
|
# draw the lin on the document |
33 |
|
|
doc.draw_line(line,arrows_style={'min_arrows':1,'scale':0.4,'max_arrows':3}) |
34 |
|
|
|
35 |
|
|
# write the document |
36 |
|
|
doc.write() |