from dust import setup_dust

if not os.path.exists("models"):
    os.mkdir("models")

if not os.path.exists("models/indiv"):
    os.mkdir("models/indiv")

seed = -69102

for model_name in ["sky_updarm_gaussian_hole_morepah"]:

    for spectral_type in spectral_types:

        print "Processing %s..." % spectral_type

        m = Model("models/indiv/energy_%s_%s" % (model_name, spectral_type.replace(" ", "_")))

        seed += 1
        m.set_seed(seed)

        m = setup_sources(m, model_name, sky_model[model_name][0], spectral_type=spectral_type)
        m = setup_dust(m, **sky_model[model_name][1])

        m.set_n_initial_iterations(3)
        m.set_n_photons(initial=100000000, imaging=0)

        m.set_enforce_energy_range(False)

        m.write()
if not os.path.exists("models"):
    os.mkdir("models")

if not os.path.exists("models/basic"):
    os.mkdir("models/basic")

seed = -43912

for model_name in sky_model:

    m = Model("models/basic/basic_%s" % model_name)

    seed += 1
    m.set_seed(seed)

    m = setup_sources(m, model_name, sky_model[model_name][0])
    m = setup_dust(m, **sky_model[model_name][1])

    image = m.add_peeled_images(image=True, sed=False)
    image.set_image_size(1, 1)
    image.set_viewing_angles([90.0], [180.0])
    image.set_image_limits(65.0, -65.0, -1.0, 1.0)
    image.set_inside_observer((8.5 * kpc, 0.0 * kpc, 0.015 * kpc))
    image.set_wavelength_range(160, 3.0, 140.0)
    image.set_track_origin("detailed")

    image = m.add_peeled_images(image=True, sed=False)
    image.set_image_size(130, 1)
    image.set_viewing_angles([90.0], [180.0])
    image.set_image_limits(65.0, -65.0, -1.0, 1.0)
    image.set_inside_observer((8.5 * kpc, 0.0 * kpc, 0.015 * kpc))