77import os
88import numpy as np
99
10- image_folder = 'd:\Profile\oqb\Desktop\presentations\POF2025\Schaefer\pics '
10+ image_folder = 'd:\Profile\oqb\Desktop\presentations\POF2025\Forces\Plots '
1111
12- vids = 'interpolate'
1312
14- if vids == 'non_interpolate' :
13+
14+ for result in ['forcevectorsapple_h' ,
15+ 'forcevectorsapple_v' ,
16+ 'forcevectorscompapple_h' ,
17+ 'forcevectorscompapple_v' ]:
1518
16- for result in ['bfield' ,
17- 'flux_density' ,
18- 'trajectory' ]:
19-
20-
21-
22- image_ref = '{}/{}_gap_6.00_shift_0.00.png' .format (image_folder , result )
23- frame = cv2 .imread (image_ref )
24- height , width , layers = frame .shape
25-
26-
27- for s in ['0.00' ,'-16.00' ,'16.00' ,'32.00' ,'-32.00' ]:
28- video_name = '{}_s{}_gap.avi' .format (result ,s )
29- video = cv2 .VideoWriter (os .path .join (image_folder ,'vids' ,video_name ), 0 , 0.625 , (width ,height ))
30-
31- for i in ['6.00' ,'15.26' ,'24.53' ,'33.79' ,'43.05' ]:
32- image = '{}/{}_gap_{}_shift_{}.png' .format (image_folder ,result ,i , s )
33- frame = cv2 .imread (image )
34- resized_frame = cv2 .resize (frame ,(width ,height ))
35- #cv2.imwrite('')
36- print ('{} gives frame size {}, originally {}' .format (i ,resized_frame .shape , frame .shape ))
37- video .write (resized_frame )
3819
39- cv2 .destroyAllWindows ()
40- video .release ()
4120
42- else :
43- for result in ['flux_density_distribution' ,
44- 'power_distro' ]:
45-
46-
47-
48- image_ref = '{}/{}_gap_6.00_shift_0.00_interpolated.png' .format (image_folder , result )
49- frame = cv2 .imread (image_ref )
50- height , width , layers = frame .shape
51-
52-
53- for s in ['0.00' ,'-16.00' ,'16.00' ,'32.00' ,'-32.00' ]:
54- video_name = '{}_s{}_gap_interpolated.avi' .format (result ,s )
55- video = cv2 .VideoWriter (os .path .join (image_folder ,'vids' ,video_name ), 0 , 0.625 , (width ,height ))
56-
57- for i in ['6.00' ,'15.26' ,'24.53' ,'33.79' ,'43.05' ]:
58- image = '{}/{}_gap_{}_shift_{}_interpolated.png' .format (image_folder ,result ,i , s )
59- frame = cv2 .imread (image )
60- resized_frame = cv2 .resize (frame ,(width ,height ))
61- #cv2.imwrite('')
62- print ('{} gives frame size {}, originally {}' .format (i ,resized_frame .shape , frame .shape ))
63- video .write (resized_frame )
21+ image_ref = '{}\\ forcevectorsapple_v6.png' .format (image_folder , result )
22+ frame = cv2 .imread (image_ref )
23+ height , width , layers = frame .shape
6424
65- cv2 .destroyAllWindows ()
66- video .release ()
67-
25+
26+ video_name = '{}gap.avi' .format (result )
27+ video = cv2 .VideoWriter (os .path .join (image_folder ,video_name ), 0 , 5 , (width ,height ))
28+
29+ for i in range (41 ):
30+ image = '{}\\ {}{}.png' .format (image_folder ,result ,i + 6 )
31+ frame = cv2 .imread (image )
32+ # resized_frame = cv2.resize(frame,(width,height))
33+ #cv2.imwrite('')
34+ # print('{} gives frame size {}, originally {}'.format(i,resized_frame.shape, frame.shape))
35+ video .write (frame )
36+
37+ cv2 .destroyAllWindows ()
38+ video .release ()
39+
0 commit comments