def test_recon_mpi(): from imars3d.recon.mpi import recon_mpi recon_template = os.path.join(outdir, "test_recon_mpi", "recon_%05i.tiff") recon_series = imars3d.io.ImageFileSeries( recon_template, mode = 'w', name = "Reconstructed", identifiers=layers, ) recon_mpi(sinograms, theta, recon_series, stepsize=3) return
def test_recon_mpi(): from imars3d.recon.mpi import recon_mpi recon_template = os.path.join(outdir, "test_recon_mpi", "recon_%05i.tiff") recon_series = imars3d.io.ImageFileSeries( recon_template, mode='w', name="Reconstructed", identifiers=layers, ) recon_mpi(sinograms, theta, recon_series, stepsize=3) return
def test_recon_mpi_2cpu(): workdir = dir pycode = """ import os, imars3d, numpy as np datadir = os.path.join(%(workdir)r, "..", "..", "iMars3D_data_set")""" % locals( ) pycode += """ outdir = "_tmp/test_mpi/test_recon_mpi_2cpu" # sinogram layers = range(900, 905) sinogram_template = os.path.join(datadir, "turbine", "sinogram_%i.tiff") sinograms = imars3d.io.ImageFileSeries( sinogram_template, name = "Sinogram", identifiers = layers, ) # recon output recon_template = os.path.join(outdir, "recon_%05i.tiff") # recon_series = imars3d.io.ImageFileSeries( recon_template, mode = 'w', name = "Reconstructed", identifiers=layers, ) # angles angles = np.arange(0, 182, .85) theta = angles * np.pi / 180. # process from imars3d.recon.mpi import recon_mpi recon_mpi(sinograms, theta, recon_series, stepsize=1) """ pypath = os.path.join(outdir, "test_recon_mpi_2cpu.py") open(pypath, 'wt').write(pycode) cmd = "mpirun -np 2 python %s" % pypath if os.system(cmd): raise RuntimeError("%s failed" % cmd) return
def test_recon_mpi_2cpu(): workdir = dir pycode = """ import os, imars3d, numpy as np datadir = os.path.join(%(workdir)r, "..", "..", "iMars3D_data_set")""" % locals() pycode += """ outdir = "_tmp/test_mpi/test_recon_mpi_2cpu" # sinogram layers = range(900, 905) sinogram_template = os.path.join(datadir, "turbine", "sinogram_%i.tiff") sinograms = imars3d.io.ImageFileSeries( sinogram_template, name = "Sinogram", identifiers = layers, ) # recon output recon_template = os.path.join(outdir, "recon_%05i.tiff") # recon_series = imars3d.io.ImageFileSeries( recon_template, mode = 'w', name = "Reconstructed", identifiers=layers, ) # angles angles = np.arange(0, 182, .85) theta = angles * np.pi / 180. # process from imars3d.recon.mpi import recon_mpi recon_mpi(sinograms, theta, recon_series, stepsize=1) """ pypath = os.path.join(outdir, "test_recon_mpi_2cpu.py") open(pypath, 'wt').write(pycode) cmd = "mpirun -np 2 python %s" % pypath if os.system(cmd): raise RuntimeError("%s failed" % cmd) return