Colorbar not displayed properly in the Multiplot class about plonk

[dmentipl]

Hi @benedettaveronesi, thanks for the bug report!

I don't think this should be too hard to fix.

I'll have a look this week.

[dmentipl]

For future reference, a more minimal bug report code snippet would be

import plonk

dump = plonk.Dump('dump_00000.h5')

ndust = plonk.header['ndustsmall']

options = list()
options.append({'render': 'density'})

for idx in range(ndust):
    options.append({'render': dump.density * dump.particles.arrays['dustfrac'][:, idx]})

multiplot = plonk.visualization.MultiPlot(
    dump, options, shape=(3, 2), cbar_mode='each', cbar_location='top', cbar_pad=0.2
)

[dmentipl]

Hi @benedettaveronesi, sorry for taking so long to respond. I've been busy with a new release of Plonk (0.2.0). This release is a major overhaul of the internals of Plonk, and of the API.

I've actually removed MultiPlot from this release.

Having said that, I don't think it's too hard to do what you require:

snap = plonk.load_snap('path_to_your_file')
ndust = len(snap.properties['grain size'])

# Say ndust is 6
ncols = 3
nrows = 2

fig, axes = plt.subplots(nrows=nrows, ncols=ncols, sharex=True, sharey=True)
for idx in range(ndust):
    plonk.visualize.plot(
        scalar_data=snap['rho'] * snap['dustfrac'][:, idx],
        x_coordinate=snap['x'],
        y_coordinate=snap['y'],
        extent=(-100, 100, -100, 100),
        particle_mass=snap['m'],
        smoothing_length=snap['h'],
        axis=axes.ravel()[idx],
    )

Please, let me know if you have any further issues.