MultiGroupMC/Flux_parser.py at master · geogunow/MultiGroupMC · GitHub

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
'''
 @file      Flux_parser.py
 @brief     Parses flux data from a text file
            and plots it as a heat map
 @author    Luke Eure
 @date      February 3, 2016
'''
import numpy as np
import matplotlib.pyplot as plt

num_groups = 2;
x_cells = 9;
y_cells = 9;
z_cells = 9;


'''
 @brief     Creates a heat map of the flux data
 @param     flux_data a 3d numpy array containing the flux information
 @param     index the z cell-value at which the heat map is displayed
'''
def plot_heat_map(flux_data, index, repeat = 100, title=''):

    img = flux_data
    lum_img = img[:, :, index]

    nx = len(lum_img)
    ny = len(lum_img[0])

    plot_array = np.zeros( (nx*repeat, ny*repeat) )
    for i in range(nx):
        for ii in range(i*repeat, (i+1)*repeat):
            for j in range(ny):
                for jj in range(j*repeat, (j+1)*repeat):
                    plot_array[ii, jj] = lum_img[j, i]
    plt.imshow(plot_array, origin='lower')
    plt.colorbar()
    plt.title(title)
    plt.show(title)

with open('flux_plot.txt', 'r') as fh:
    flux_file = fh.read().replace('\n', '')
    flux_list = flux_file.split()

    flux_to_plot = np.zeros((num_groups, x_cells,
        y_cells, z_cells))
    for g in range(num_groups):
        for i in range(x_cells):
            for j in range(y_cells):
                for k in range(z_cells):
                    flux_to_plot[g][i][j][k] = \
                            flux_list[g*x_cells*y_cells*z_cells \
                            + i*y_cells*z_cells + j*z_cells + k]

    index = 1
    for g in range(num_groups):
        plot_heat_map(flux_to_plot[g], index, repeat = 5,
                title = ('Group ' + str(g+1)))