FilesExpand file tree

 master

/

potential.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

58 lines (47 loc) · 1.65 KB

 master

/

potential.py

Copy path

Top

File metadata and controls

• Code
• Blame

58 lines (47 loc) · 1.65 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

#-----------------------------------------------------------------------

# potential.py

#-----------------------------------------------------------------------

import stddraw

import stdio

import stdarray

from charge import Charge

from color import Color

from picture import Picture

# Read values from standard input to create an array of charged

# particles. Set each pixel color in an image to a grayscale value

# proportional to the total of the potentials due to the particles at

# corresponding points. Draw the resulting image to standard draw.

MAX_GRAY_SCALE = 255

# Read charges from standard input into an array.

n = stdio.readInt()

charges = stdarray.create1D(n)

for i in range(n):

x0 = stdio.readFloat()

y0 = stdio.readFloat()

q0 = stdio.readFloat()

charges[i] = Charge(x0, y0, q0)

# Create a Picture depicting potential values.

pic = Picture()

for col in range(pic.width()):

for row in range(pic.height()):

# Compute pixel color.

x = 1.0 * col / pic.width()

y = 1.0 * row / pic.height()

v = 0.0

for i in range(n):

v += charges[i].potentialAt(x, y)

v = (MAX_GRAY_SCALE / 2.0) + (v / 2.0e10)

if v < 0:

grayScale = 0

elif v > MAX_GRAY_SCALE:

grayScale = MAX_GRAY_SCALE

else:

grayScale = int(v)

color = Color(grayScale, grayScale, grayScale)

pic.set(col, pic.height()-1-row, color)

# Draw the Picture.

stddraw.setCanvasSize(pic.width(), pic.height())

stddraw.picture(pic)

stddraw.show()

#-----------------------------------------------------------------------

# python potential.py < charges.txt