Commit 0e2ff9c2 authored by cianciosa's avatar cianciosa
Browse files

Add utility to replace NaNs in files.

parent 5c3bb1b3
Loading
Loading
Loading
Loading
+1 −1
Original line number Diff line number Diff line
@@ -106,7 +106,7 @@ void trace_ray(const size_t num_times,
            //auto eq = equilibrium::make_slab_field<T, SAFE_MATH> ();
            //auto eq = equilibrium::make_no_magnetic_field<T, SAFE_MATH> ();

            const T endtime = static_cast<T> (1.5);
            const T endtime = static_cast<T> (2.0);
            const T dt = endtime/static_cast<T> (num_times);

            //auto dt_var = graph::variable(num_rays, static_cast<T> (dt), "dt");

utilities/fix_NaN.py

0 → 100644
+27 −0
Original line number Diff line number Diff line
import netCDF4
import argparse
import numpy

def main(**args):
    for i in range(0, 12):
        with netCDF4.Dataset('{}/result{}.nc'.format(args['directory'], i), 'r+') as result:
            result.variables['kamp'][:] = numpy.where(numpy.isnan(result.variables['kamp'][:]), 0.0, result.variables['kamp'][:])

if __name__ == '__main__':
    command_line_parser = argparse.ArgumentParser()

    command_line_parser.add_argument('-d',
                                     '--directory',
                                     action='store',
                                     required=True,
                                     dest='directory',
                                     help='Directory',
                                     metavar='DIRECTORY')

    args = vars(command_line_parser.parse_args())

#  Remove empty arguments
    for key in [key for key in args if args[key] == None]:
        del args[key]

    main(**args)