Python myloadmat_to_obj 예제들

예제 #1

파일: 000_plot_main_output.py 프로젝트: jsopouse/PyECLOUD

import sys, os
BIN = os.path.expanduser("../../../")
sys.path.append(BIN)

import pylab as pl
import numpy as np

import PyECLOUD.myloadmat_to_obj as mlo
import PyECLOUD.mystyle as ms

pl.close('all')
ms.mystyle_arial(fontsz=16)
dpiset = 200

ob=mlo.myloadmat_to_obj('../../testing/tests_buildup/LHC_ArcDipReal_450GeV_sey1.60_2.5e11ppb_bl_1.00ns/Pyecltest_ref.mat')

ifig = 0

#################################
# Variables saved per time step #
#################################

#8.
ifig+=1; pl.figure(ifig)
pl.plot(ob.t, ob.lam_t_array, linewidth=2)
pl.xlabel('Time [s]')
pl.ylabel('Beam profile [p/m]') 
ms.scix(); pl.grid('on')
pl.suptitle('Var. name: lam_t_array\nBeam density at each time step')
pl.subplots_adjust(top=.82, bottom=.14)

예제 #2

lim_dens = [10, 14]
flag_log = True
outp_filname = 'pass%d.avi'%passlist[0]

########

########

qe = 1.60217657e-19
c = 299792458.

i_photog = 0

ms.mystyle_arial(fontsz=fontsz, dist_tick_lab=10)

chamber = mlo.myloadmat_to_obj(folder_sim + '/LHC_chm_ver.mat')
Vx = chamber.Vx
Vy = chamber.Vy

##load output file
obout = mlo.myloadmat_to_obj(folder_sim + '/' + main_outp_filename)
t = obout.t
lam_b1 = obout.lam_t_array
cendens = np.squeeze(obout.cen_density)

#every bunch passage
for pass_ind in passlist:

    filename_rho = 'rho_video/rho_pass%d.mat'%pass_ind
    filename_sc_forces = 'sc_forces/sc_forces_pass%d.mat'%pass_ind
    #new code

예제 #3

import sys
import os
BIN = os.path.expanduser("../../../../")
sys.path.append(BIN)

import pylab as pl
import numpy as np

import PyECLOUD.myloadmat_to_obj as mlo
import PyECLOUD.mystyle as ms

pl.close('all')
ms.mystyle_arial(fontsz=16)
dpiset = 200

ob = mlo.myloadmat_to_obj('./Pyecltest.mat')

#################################
# Variables saved per time step #
#################################

fig = pl.figure()
pl.plot(ob.t, ob.Nel_timep, linewidth=2)
pl.plot(ob.t, ob.Nelectrons, 'r--', linewidth=2)

pl.xlabel('Time [s]')
pl.ylabel('Number of $e^-$ per unit length [$m^{-1}$]')
ms.scix()
pl.grid('on')
pl.suptitle(
    'Var. name: Nel_timep\nNumber of electrons in the chamber at each time step'

예제 #4

import os, sys

BIN = os.path.expanduser(
    "../../../")  # folder containing PyECLOUD, PyPIC, PyKLU
if BIN not in sys.path:
    sys.path.append(BIN)

import PyECLOUD.myloadmat_to_obj as mlm

import matplotlib.pyplot as plt

# obu = mlm.myloadmat_to_obj('LHC_ArcDipReal_450GeV_sey1.70_2.5e11ppb_bl_1.00ns/Pyecltest_angle3D_ref.mat')
# obn = mlm.myloadmat_to_obj('LHC_ArcDipReal_450GeV_sey1.70_2.5e11ppb_bl_1.00ns_nonuniftime/Pyecltest_angle3D.mat')

obu = mlm.myloadmat_to_obj(
    'LHC_ArcDipReal_450GeV_sey1.00_2.5e11ppb_bl_1.00ns_gas_ionization/Pyecltest_angle3D_ref.mat'
)
obn = mlm.myloadmat_to_obj(
    'LHC_ArcDipReal_450GeV_sey1.00_2.5e11ppb_bl_1.00ns_gas_ionization/Pyecltest_angle3D.mat'
)

# obu = mlm.myloadmat_to_obj('LHC_ArcDipReal_6500GeV_sey_1.70_1.1e11ppb_b1_1.00ns/Pyecltest_angle3D_ref.mat')
# obn = mlm.myloadmat_to_obj('LHC_ArcDipReal_6500GeV_sey_1.70_1.1e11ppb_b1_1.00ns/Pyecltest_angle3D.mat')

plt.close('all')
plt.figure(1)
plt.plot(obu.t, obu.Nel_timep, '.-')
plt.plot(obn.t, obn.Nel_timep, '.r')
plt.show()

예제 #5

파일: 001_plot_result.py 프로젝트: lgiacome/PyECLOUD

BIN = os.path.expanduser("../../../../")
sys.path.append(BIN)

import argparse
import matplotlib.pyplot as plt
import numpy as np
#from colorsys import hsv_to_rgb
import os
import PyECLOUD.myloadmat_to_obj as mlm
import matplotlib.gridspec as gridspec
import PyECLOUD.mystyle as ms

plt.close('all')
ms.mystyle_arial(fontsz=16)

ob = mlm.myloadmat_to_obj(
    './test_saving__iter0.mat')  # load dictionary of the current simulation

N_pass = ob.nel_hist.shape[0]

fig1 = plt.figure(1)
fig1.set_facecolor('w')
plt.pcolormesh(ob.xg_hist, ob.t_hist * 1e6, ob.nel_hist, shading='Gouraud')
plt.axis('tight')

x_bun = 0 * ob.t_hist
x_bun[30:] += 3e-3
plt.plot(x_bun, ob.t_hist * 1e6, '.w', markersize=10.)
plt.xlabel('x [m]')
plt.ylabel('Time [us]')
fig1.subplots_adjust(bottom=.12)

예제 #6

import numpy as np
import matplotlib.pyplot as pl
import sys
sys.path.append('../../../')

import PyECLOUD.myloadmat_to_obj as mlo
import PyECLOUD.mystyle as ms

pyecl_output_file = '../../testing/tests_buildup/LHC_ArcDipReal_450GeV_sey1.70_2.5e11ppb_bl_1.00ns/Pyecltest_angle3D.mat'
#~ pyecl_output_file = '../../testing/tests_buildup/LHC_ArcDipReal_450GeV_sey1.70_2.5e11ppb_bl_1.00ns_change_s_and_E0/Pyecltest_angle3D.mat'
pyecl_output_file = '../../testing/tests_buildup/LHC_ArcDipReal_450GeV_sey1.70_2.5e11ppb_bl_1.00ns_seyfromfile/Pyecltest_angle3D.mat'

ob = mlo.myloadmat_to_obj(pyecl_output_file)

pl.close('all')
ms.mystyle_arial()

fig1 = pl.figure(1, figsize=(2 * 8, 6))
fig1.set_facecolor('w')
sp1 = fig1.add_subplot(1, 2, 1)
sp2 = fig1.add_subplot(1, 2, 2, sharex=sp1)
for i_ct, ct in enumerate(ob.sey_test_cos_theta):
    thiscol = ms.colorprog(i_ct, len(ob.sey_test_cos_theta))
    label = 'costheta=%.2f' % ct
    sp1.plot(ob.sey_test_E_impact_eV,
             ob.sey_test_del_true_mat[i_ct, :],
             color=thiscol,
             label=label)
    sp2.plot(ob.sey_test_E_impact_eV,
             ob.sey_test_del_elast_mat[i_ct, :],
             color=thiscol,

예제 #7

import sys, os
BIN = os.path.expanduser("../../../")
sys.path.append(BIN)

import pylab as pl
import numpy as np

import PyECLOUD.myloadmat_to_obj as mlo
import PyECLOUD.mystyle as ms

pl.close('all')
ms.mystyle_arial(fontsz=16)
dpiset = 200

ob = mlo.myloadmat_to_obj('../../testing/tests_buildup/LHC_ArcDipReal_450GeV_sey1.70_2.5e11ppb_bl_1.00ns/Pyecltest_angle3D_ref.mat')

ifig = 0

#################################
# Variables saved per time step #
#################################

#8.
ifig += 1; pl.figure(ifig)
pl.plot(ob.t, ob.lam_t_array, linewidth=2)
pl.xlabel('Time [s]')
pl.ylabel('Beam profile [p/m]')
ms.scix(); pl.grid('on')
pl.suptitle('Var. name: lam_t_array\nBeam density at each time step')
pl.subplots_adjust(top=.82, bottom=.14)

예제 #8

파일: 001_test_output_plots.py 프로젝트: lgiacome/PyECLOUD

import sys
import os
BIN = os.path.expanduser("../../../../")
sys.path.append(BIN)

import pylab as pl
import numpy as np

import PyECLOUD.myloadmat_to_obj as mlo
import PyECLOUD.mystyle as ms

pl.close('all')
ms.mystyle_arial(fontsz=16)
dpiset = 200

ob = mlo.myloadmat_to_obj('./Pyecltest.mat')
ob2 = mlo.myloadmat_to_obj('./Pyecltest.mat.ref')
# ob2 = mlo.myloadmat_to_obj('./Pyecltest_excl.mat')

ifig = 0

#################################
# Variables saved per time step #
#################################

# 8.
ifig += 1
pl.figure(ifig)
pl.plot(ob.t, ob.lam_t_array, linewidth=2, label='Furman-Pivi model')
pl.plot(ob2.t, ob2.lam_t_array, '--', linewidth=2, label='ECLOUD model')
pl.xlabel('Time [s]')