-
Notifications
You must be signed in to change notification settings - Fork 11
/
proc_video3.py
114 lines (97 loc) · 3.34 KB
/
proc_video3.py
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
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
#-Begin-preamble-------------------------------------------------------
#
# CERN
#
# European Organization for Nuclear Research
#
#
# This file is part of the code:
#
# PyECLOUD Version 8.7.1
#
#
# Main author: Giovanni IADAROLA
# BE-ABP Group
# CERN
# CH-1211 GENEVA 23
# SWITZERLAND
# giovanni.iadarola@cern.ch
#
# Contributors: Eleonora Belli
# Philipp Dijkstal
# Lorenzo Giacomel
# Lotta Mether
# Annalisa Romano
# Giovanni Rumolo
# Eric Wulff
#
#
# Copyright CERN, Geneva 2011 - Copyright and any other
# appropriate legal protection of this computer program and
# associated documentation reserved in all countries of the
# world.
#
# Organizations collaborating with CERN may receive this program
# and documentation freely and without charge.
#
# CERN undertakes no obligation for the maintenance of this
# program, nor responsibility for its correctness, and accepts
# no liability whatsoever resulting from its use.
#
# Program and documentation are provided solely for the use of
# the organization to which they are distributed.
#
# This program may not be copied or otherwise distributed
# without permission. This message must be retained on this and
# any other authorized copies.
#
# The material cannot be sold. CERN should be given credit in
# all references.
#
#-End-preamble---------------------------------------------------------
import scipy.io as sio
from pylab import imshow, colorbar, show, savefig, clf, squeeze, title, xlabel, ylabel, floor
import subprocess
import numpy as np
firs_pass = 0
last_pass = 3
flag_log = True
N_dec = 2
i_photog = 0
for pass_ind in range(firs_pass, last_pass):
filename_rho = 'rho_video/rho_pass%d.mat'%pass_ind
dict_ecl_video = sio.loadmat(filename_rho)
dict_pyecltest = sio.loadmat('Pyecltest.mat')
rho_video = dict_ecl_video['rho_video']
t_video = squeeze(dict_ecl_video['t_video'].real)
b_spac = squeeze(dict_pyecltest['b_spac'].real)
(nphotog, _, _) = rho_video.shape
#subprocess.check_call(('rm', '*.png'))
for ii in range(0, nphotog, N_dec):
print('Pass %d %d/%d'%(pass_ind, ii, nphotog))
imm = np.squeeze(rho_video[ii, :, :])
if flag_log:
imm = np.log10(np.abs(imm))
imshow(imm.T, cmap=None, norm=None, aspect=None, interpolation=None,
alpha=None, vmin=None, vmax=None, origin='lower', extent=None)
colorbar()
title(('passage = %d' % floor(t_video[ii] / b_spac)))
#xlabel('x [m]')
#ylabel('y [m]')
filename = str('Pass%05d_%05d' % (pass_ind, ii)) + '.png'
savefig(filename, dpi=100)
clf()
i_photog += 1
command = ('mencoder',
'mf://*.png',
'-mf',
'type=png:w=800:h=600:fps=5',
'-ovc',
'lavc',
'-lavcopts',
'vcodec=mpeg4',
'-oac',
'copy',
'-o',
'output.avi')
subprocess.check_call(command)