mosaic.F90

Go to the documentation of this file.

! Copyright (C) 2007-2012, 2016 Matthew West
! Licensed under the GNU General Public License version 2 or (at your
! option) any later version. See the file COPYING for details.
 
!> \file
!> The pmc_mosaic module.
 
!> Interface to the MOSAIC aerosol and gas phase chemistry code.
module pmc_mosaic
 
  use pmc_aero_data
  use pmc_aero_state
  use pmc_constants
  use pmc_env_state
  use pmc_gas_data
  use pmc_gas_state
  use pmc_output
  use pmc_util
 
contains
 
!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
 
  !> Whether MOSAIC support is compiled in.
  logical function mosaic_support()
 
#ifdef PMC_USE_MOSAIC
    mosaic_support = .true.
#else
    mosaic_support = .false.
#endif
 
  end function mosaic_support
 
!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
 
  !> Initialize all MOSAIC data-structures.
  subroutine mosaic_init(env_state, aero_data, del_t, do_optical)
 
#ifdef PMC_USE_MOSAIC
    use module_data_mosaic_aero, only: alpha_astem, rtol_eqb_astem, &
         ptol_mol_astem, mgas_aer_xfer, mdynamic_solver
 
    use module_data_mosaic_main, only: tbeg_sec, dt_sec, rlon, rlat, &
         zalt_m, rh, te, pr_atm, cair_mlc, cair_molm3, ppb, avogad, &
         mmode, mgas, maer, mcld, maeroptic, mshellcore, &
         msolar, mphoto, lun_aeroptic, naerbin
#endif
 
    !> Environment state.
    type(env_state_t), intent(inout) :: env_state
    !> Aerosol data.
    type(aero_data_t), intent(in) :: aero_data
    !> Timestep for coagulation.
    real(kind=dp), intent(in) :: del_t
    !> Whether to compute optical properties.
    logical, intent(in) :: do_optical
 
#ifdef PMC_USE_MOSAIC
    ! MOSAIC function interfaces
    interface
       subroutine loadperoxyparameters()
       end subroutine loadperoxyparameters
       subroutine init_data_modules()
       end subroutine init_data_modules
       subroutine allocatememory()
       end subroutine allocatememory
    end interface
 
    call init_data_modules  ! initialize indices and vars
 
    ! allocate one aerosol bin
    naerbin = 1
    call allocatememory()
 
    ! parameters
    mmode = 1               ! 1 = time integration, 2 = parametric analysis
    mgas = 1                ! 1 = gas chem on, 0 = gas chem off
    maer = 1                ! 1 = aer chem on, 0 = aer chem off
    mcld = 0                ! 1 = cld chem on, 0 = cld chem off
    if (do_optical) then
       maeroptic = 1        ! 1 = aer_optical on, 0 = aer_optical off
    else
       maeroptic = 0
    end if
    mshellcore = 1          ! 0 = no shellcore, 1 = core is BC only
                            ! 2 = core is BC and DUST
    msolar = 1              ! 1 = diurnally varying phot, 2 = fixed phot
    mphoto = 2              ! 1 = Rick's param, 2 = Yang's param
    mgas_aer_xfer = 1       ! 1 = gas-aerosol partitioning, 0 = no partition
    mdynamic_solver = 1     ! 1 = astem, 2 = lsodes
    alpha_astem = 0.5d0     ! solver parameter. range: 0.01 - 1.0
    rtol_eqb_astem = 0.01d0 ! relative eqb tolerance. range: 0.01 - 0.03
    ptol_mol_astem = 0.01d0 ! percent mol tolerance.  range: 0.01 - 1.0
 
    ! time variables
    dt_sec = del_t                                 ! time-step (s)
    tbeg_sec = env_state%start_day*24*3600 + &     ! time since the beg of
         nint(env_state%start_time)                ! year 00:00, UTC (s)
 
    ! geographic location
    rlon = deg2rad(env_state%longitude)            ! longitude
    rlat = deg2rad(env_state%latitude)             ! latitude
    zalt_m = env_state%altitude                    ! altitude (m)
 
    ! environmental parameters: map PartMC -> MOSAIC
    rh = env_state%rel_humid * 100.d0              ! relative humidity (%)
    te = env_state%temp                            ! temperature (K)
    pr_atm = env_state%pressure / const%air_std_press ! pressure (atm)
    cair_mlc = avogad*pr_atm/(82.056d0*te)         ! air conc [molec/cc]
    cair_molm3 = 1d6*pr_atm/(82.056d0*te)          ! air conc [mol/m^3]
    ppb = 1d9
 
    call loadperoxyparameters ! Aperox and Bperox only once
 
    ! get unit for aerosol optical output
    if (lun_aeroptic <= 0 ) lun_aeroptic = get_unit()
 
    ! ensure H2O is a valid species
    call assert_msg(111041803, aero_data%i_water > 0, &
         "MOSAIC requires H2O as an aerosol species")
 
#endif
 
  end subroutine mosaic_init
 
!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
 
  !> Clean-up after running MOSAIC, deallocating memory.
  subroutine mosaic_cleanup()
 
#ifdef PMC_USE_MOSAIC
    ! MOSAIC function interfaces
    interface
       subroutine deallocatememory()
       end subroutine deallocatememory
    end interface
 
    call deallocatememory()
#endif
 
  end subroutine mosaic_cleanup
 
!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
 
  !> Map all data PartMC -> MOSAIC.
  subroutine mosaic_from_partmc(env_state, aero_data, &
       aero_state, gas_data, gas_state)
 
#ifdef PMC_USE_MOSAIC
    use module_data_mosaic_aero, only: nbin_a, aer, num_a, jhyst_leg, &
#ifdef pmc_use_wrf
         particle_error, &
#endif
         jtotal, water_a
 
    use module_data_mosaic_main, only: tbeg_sec, tcur_sec, tmid_sec, &
         dt_sec, dt_min, dt_aeroptic_min, rh, te, pr_atm, cnn, cair_mlc, &
         cair_molm3, ppb, avogad, msolar, naerbin
#endif
 
    !> Environment state.
    type(env_state_t), intent(in) :: env_state
    !> Aerosol data.
    type(aero_data_t), intent(in) :: aero_data
    !> Aerosol state.
    type(aero_state_t), intent(in) :: aero_state
    !> Gas data.
    type(gas_data_t), intent(in) :: gas_data
    !> Gas state.
    type(gas_state_t), intent(in) :: gas_state
 
#ifdef PMC_USE_MOSAIC
    ! local variables
    real(kind=dp) :: time_utc    ! 24-hr UTC clock time (hr).
    real(kind=dp) :: tmar21_sec  ! Time at noon, march 21, UTC (s).
    real(kind=dp) :: conv_fac(aero_data_n_spec(aero_data)), dum_var
    integer :: i_part, i_spec, i_spec_mosaic
    real(kind=dp) :: num_conc
 
    ! MOSAIC function interfaces
    interface
       subroutine allocatememory()
       end subroutine allocatememory
       subroutine deallocatememory()
       end subroutine deallocatememory
    end interface
 
    ! update time variables
    tmar21_sec = real((79*24 + 12)*3600, kind=dp)    ! noon, mar 21, UTC
    tcur_sec = real(tbeg_sec, kind=dp) + env_state%elapsed_time
    ! current (old) time since the beg of year 00:00, UTC (s)
 
    time_utc = env_state%start_time/3600d0  ! 24-hr UTC clock time (hr)
    time_utc = time_utc + dt_sec/3600d0
 
    do while (time_utc >= 24d0)
       time_utc = time_utc - 24d0
    end do
 
    tmid_sec = tcur_sec + 0.5d0*dt_sec
    if(tmid_sec .ge. tmar21_sec)then
       tmid_sec = tmid_sec - tmar21_sec     ! seconds since noon, march 21
    else
       tmid_sec = tmid_sec &
            + dble(((365-79)*24 - 12)*3600) ! seconds since noon, march 21
    endif
 
    ! transport timestep (min)
    dt_min = dt_sec/60d0
    ! aerosol optics timestep (min)
    dt_aeroptic_min = 0d0
 
    ! compute aerosol conversion factors
    do i_spec = 1,aero_data_n_spec(aero_data)
       ! converts m^3(species) to nmol(species)/m^3(air)
       conv_fac(i_spec) = 1.d9 * aero_data%density(i_spec) &
            / aero_data%molec_weight(i_spec)
    enddo
 
    ! environmental parameters: map PartMC -> MOSAIC
    rh = env_state%rel_humid * 100.d0              ! relative humidity (%)
    te = env_state%temp                            ! temperature (K)
    pr_atm = env_state%pressure / const%air_std_press ! pressure (atm)
    cair_mlc = avogad*pr_atm/(82.056d0*te)   ! air conc [molec/cc]
    cair_molm3 = 1d6*pr_atm/(82.056d0*te)    ! air conc [mol/m^3]
    ppb = 1d9
 
    ! aerosol data: map PartMC -> MOSAIC
    nbin_a = aero_state_total_particles(aero_state)
    if (nbin_a > naerbin) then
       call deallocatememory()
       naerbin = nbin_a
       call allocatememory()
    end if
    aer = 0d0    ! initialize to zero
    ! work backwards for consistency with mosaic_to_partmc(), which
    ! has specific ordering requirements
    do i_part = aero_state_n_part(aero_state),1,-1
       num_conc = aero_weight_array_num_conc(aero_state%awa, &
            aero_state%apa%particle(i_part), aero_data)
       do i_spec = 1,aero_data_n_spec(aero_data)
          i_spec_mosaic = aero_data%mosaic_index(i_spec)
          if (i_spec_mosaic > 0) then
             ! convert m^3(species) to nmol(species)/m^3(air)
             aer(i_spec_mosaic, 3, i_part) &   ! nmol/m^3(air)
                  = aero_state%apa%particle(i_part)%vol(i_spec) &
                  * conv_fac(i_spec) * num_conc
          end if
       end do
       ! handle water specially
       ! convert m^3(water) to kg(water)/m^3(air)
       water_a(i_part) = &
            aero_state%apa%particle(i_part)%vol(aero_data%i_water) &
            * aero_data%density(aero_data%i_water) * num_conc
       num_a(i_part) = 1d-6 * num_conc ! num conc (#/cc(air))
       jhyst_leg(i_part) = aero_state%apa%particle(i_part)%water_hyst_leg
    end do
 
    ! gas chemistry: map PartMC -> MOSAIC
    cnn = 0d0
    do i_spec = 1,gas_data_n_spec(gas_data)
       i_spec_mosaic = gas_data%mosaic_index(i_spec)
       if (i_spec_mosaic > 0) then
          ! convert ppbv to molec/cc
          cnn(i_spec_mosaic) = gas_state%mix_rat(i_spec) * cair_mlc / ppb
       end if
    end do
#ifdef PMC_USE_WRF
    particle_error = .false.
#endif
#endif
 
  end subroutine mosaic_from_partmc
 
!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
 
  !> Map all data MOSAIC -> PartMC.
  subroutine mosaic_to_partmc(env_state, aero_data, aero_state, gas_data, &
       gas_state, uuid)
 
#ifdef PMC_USE_MOSAIC
    use module_data_mosaic_aero, only: nbin_a, aer, num_a, jhyst_leg, &
#ifdef pmc_use_wrf
         particle_error, &
#endif
         jtotal, water_a
 
    use module_data_mosaic_main, only: tbeg_sec, tcur_sec, tmid_sec, &
         dt_sec, dt_min, dt_aeroptic_min, rh, te, pr_atm, cnn, cair_mlc, &
         cair_molm3, ppb, avogad, msolar, cos_sza
#endif
 
    !> Environment state.
    type(env_state_t), intent(inout) :: env_state
    !> Aerosol data.
    type(aero_data_t), intent(in) :: aero_data
    !> Aerosol state.
    type(aero_state_t), intent(inout) :: aero_state
    !> Gas data.
    type(gas_data_t), intent(in) :: gas_data
    !> Gas state.
    type(gas_state_t), intent(inout) :: gas_state
    !> UUID for this simulation.
    character(len=PMC_UUID_LEN), intent(in) :: uuid
 
#ifdef PMC_USE_MOSAIC
    ! local variables
    real(kind=dp) :: conv_fac(aero_data_n_spec(aero_data)), dum_var, num_conc
    integer :: i_part, i_spec, i_spec_mosaic
    real(kind=dp), allocatable :: reweight_num_conc(:)
 
#ifdef PMC_USE_WRF
    if (any(particle_error) .eqv. .true.) then
       call output_state('before_error', output_type_dist, aero_data, &
            aero_state, gas_data, gas_state, env_state, &
            int(env_state%elapsed_time / dt_sec), tcur_sec, dt_sec, 1, &
            .false., .false., uuid)
    end if
#endif
 
    ! compute aerosol conversion factors
    do i_spec = 1,aero_data_n_spec(aero_data)
       ! converts m^3(species) to nmol(species)/m^3(air)
       conv_fac(i_spec) = 1d9 * aero_data%density(i_spec) &
            / aero_data%molec_weight(i_spec)
    enddo
 
    ! environmental parameters: map MOSAIC -> PartMC
    env_state%rel_humid = rh / 100d0
    env_state%temp = te
    env_state%pressure = pr_atm * const%air_std_press
    if (msolar == 1) then
       env_state%solar_zenith_angle = acos(cos_sza)
    end if
    cair_mlc = avogad*pr_atm/(82.056d0*te)   ! air conc [molec/cc]
    cair_molm3 = 1d6*pr_atm/(82.056d0*te)    ! air conc [mol/m^3]
    ppb = 1d9
 
    ! gas chemistry: map MOSAIC -> PartMC
    do i_spec = 1,gas_data_n_spec(gas_data)
       i_spec_mosaic = gas_data%mosaic_index(i_spec)
       if (i_spec_mosaic > 0) then
          ! convert molec/cc to ppbv
          gas_state%mix_rat(i_spec) = cnn(i_spec_mosaic) / cair_mlc * ppb
       end if
    end do
 
    ! We're modifying particle diameters, so the bin sort is now invalid
    aero_state%valid_sort = .false.
 
    ! aerosol data: map MOSAIC -> PartMC
    allocate(reweight_num_conc(aero_state_n_part(aero_state)))
    call aero_state_num_conc_for_reweight(aero_state, aero_data, &
         reweight_num_conc)
    do i_part = 1,aero_state_n_part(aero_state),1
       num_conc = aero_weight_array_num_conc(aero_state%awa, &
            aero_state%apa%particle(i_part), aero_data)
       do i_spec = 1,aero_data_n_spec(aero_data)
          i_spec_mosaic = aero_data%mosaic_index(i_spec)
          if (i_spec_mosaic > 0) then
             aero_state%apa%particle(i_part)%vol(i_spec) = &
                  ! convert nmol(species)/m^3(air) to m^3(species)
                  aer(i_spec_mosaic, 3, i_part) / (conv_fac(i_spec) * num_conc)
          end if
       end do
       aero_state%apa%particle(i_part)%water_hyst_leg = jhyst_leg(i_part)
       ! handle water specially
       ! convert kg(water)/m^3(air) to m^3(water)
       aero_state%apa%particle(i_part)%vol(aero_data%i_water) = &
            water_a(i_part) / aero_data%density(aero_data%i_water) / num_conc
    end do
 
#ifdef PMC_USE_WRF
    if (any(particle_error) .eqv. .true.) then
       call output_state('after_error', output_type_dist, aero_data, &
            aero_state, gas_data, gas_state, env_state, &
            int(env_state%elapsed_time / dt_sec), tcur_sec, dt_sec, 1,&
            .false., .false., uuid)
       do i_part = aero_state_n_part(aero_state),1,-1
          if (particle_error(i_part)) then
             ! Move last element to the current spot if it is removed.
             if (i_part < aero_state_n_part(aero_state)) then
                reweight_num_conc(i_part) = reweight_num_conc( &
                     aero_state_n_part(aero_state))
             end if
             call aero_state_remove_particle_no_info(aero_state, i_part)
          end if
       end do
       ! Shrink the reweight_num_conc array to the number of particles
       reweight_num_conc = reweight_num_conc(1:aero_state_n_part(aero_state))
    end if
#endif
 
    ! adjust particles to account for weight changes
    call aero_state_reweight(aero_state, aero_data, reweight_num_conc)
 
#endif
 
  end subroutine mosaic_to_partmc
 
!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
 
  !> Do one timestep with MOSAIC.
  !!
  !! We currently also compute aerosol optical properties within this
  !! subroutine. In principle this could be done at data analysis
  !! time, rather than inside the timestepper. It's not clear if this
  !! really matters, however. Because of this mosaic_aero_optical() is
  !! currently disabled.
  subroutine mosaic_timestep(env_state, aero_data, aero_state, gas_data, &
       gas_state, do_optical, uuid)
 
#ifdef PMC_USE_MOSAIC
    use module_data_mosaic_main, only: msolar
#endif
 
    !> Environment state.
    type(env_state_t), intent(inout) :: env_state
    !> Aerosol data.
    type(aero_data_t), intent(in) :: aero_data
    !> Aerosol state.
    type(aero_state_t), intent(inout) :: aero_state
    !> Gas data.
    type(gas_data_t), intent(in) :: gas_data
    !> Gas state.
    type(gas_state_t), intent(inout) :: gas_state
    !> Whether to compute optical properties.
    logical, intent(in) :: do_optical
    !> UUID for this simulation.
    character(len=PMC_UUID_LEN), intent(in) :: uuid
 
#ifdef PMC_USE_MOSAIC
    ! MOSAIC function interfaces
    interface
       subroutine solarzenithangle()
       end subroutine solarzenithangle
       subroutine integratechemistry()
       end subroutine integratechemistry
#ifdef PMC_USE_MOSAIC_MULTI_OPT
       subroutine aerosol_optical(i_wavelength)
         integer, optional :: i_wavelength
       end subroutine aerosol_optical
#else
       subroutine aerosol_optical()
       end subroutine aerosol_optical
#endif
    end interface
 
    ! map PartMC -> MOSAIC
    call mosaic_from_partmc(env_state, aero_data, aero_state, gas_data, &
         gas_state)
 
    if (msolar == 1) then
      call solarzenithangle
    end if
 
    call integratechemistry
 
    ! map MOSAIC -> PartMC
    if (do_optical) then
       ! must do optical properties first, as mosaic_to_partmc() may
       ! change the number of particles
       call aerosol_optical
       call mosaic_aero_optical(env_state, aero_data, &
            aero_state, gas_data, gas_state)
    end if
 
    call mosaic_to_partmc(env_state, aero_data, aero_state, gas_data, &
         gas_state, uuid)
#endif
 
  end subroutine mosaic_timestep
 
!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
 
  !> Compute the optical properties of each aerosol particle.
  !> FIXME: currently disabled.
  !!
  !! At the moment we are computing the aerosol optical properties
  !! every timestep from withing mosaic_timestep. This decision should
  !! be re-evaluated at some point in the future.
  subroutine mosaic_aero_optical(env_state, aero_data, &
       aero_state, gas_data, gas_state)
 
#ifdef PMC_USE_MOSAIC
    use module_data_mosaic_aero, only: ri_shell_a, ri_core_a, &
         ext_cross, scat_cross, asym_particle, dp_core_a
#endif
 
    !> Environment state.
    type(env_state_t), intent(in) :: env_state
    !> Aerosol data.
    type(aero_data_t), intent(in) :: aero_data
    !> Aerosol state.
    type(aero_state_t), intent(inout) :: aero_state
    !> Gas data.
    type(gas_data_t), intent(in) :: gas_data
    !> Gas state.
    type(gas_state_t), intent(in) :: gas_state
 
#ifdef PMC_USE_MOSAIC
    ! MOSAIC function interfaces
    interface
#ifdef PMC_USE_MOSAIC_MULTI_OPT
       subroutine aerosol_optical(i_wavelength)
         integer, optional :: i_wavelength
       end subroutine aerosol_optical
#else
       subroutine aerosol_optical()
       end subroutine aerosol_optical
#endif
    end interface
 
    integer :: i_part
 
    ! map PartMC -> MOSAIC
!    call mosaic_from_partmc(env_state, aero_data, aero_state, &
!         gas_data, gas_state)
 
!    call aerosol_optical
 
    ! map MOSAIC -> PartMC
    ! work backwards for consistency with mosaic_to_partmc(), which
    ! has specific ordering requirements
    do i_part = aero_state_n_part(aero_state),1,-1
#ifdef PMC_USE_MOSAIC_MULTI_OPT
       aero_state%apa%particle(i_part)%absorb_cross_sect = &
            (ext_cross(i_part,:) - scat_cross(i_part,:)) / 1d4 ! (m^2)
       aero_state%apa%particle(i_part)%scatter_cross_sect = &
            scat_cross(i_part,:) / 1d4 ! (m^2)
       aero_state%apa%particle(i_part)%asymmetry = &
            asym_particle(i_part,:) ! (1)
       aero_state%apa%particle(i_part)%refract_shell = &
            cmplx(ri_shell_a(i_part,:), kind=dc) ! (1)
       aero_state%apa%particle(i_part)%refract_core =&
            cmplx(ri_core_a(i_part,:), kind=dc) ! (1)
       aero_state%apa%particle(i_part)%core_vol = &
            aero_data_diam2vol(aero_data, dp_core_a(i_part)) / 1d6 ! (m^3)
#else
       aero_state%apa%particle(i_part)%absorb_cross_sect = (ext_cross(i_part) &
            - scat_cross(i_part)) / 1d4                       ! (m^2)
       aero_state%apa%particle(i_part)%scatter_cross_sect = &
            scat_cross(i_part) / 1d4 ! (m^2)
       aero_state%apa%particle(i_part)%asymmetry = asym_particle(i_part) ! (1)
       aero_state%apa%particle(i_part)%refract_shell = &
            cmplx(ri_shell_a(i_part), kind=dc) ! (1)
       aero_state%apa%particle(i_part)%refract_core =&
            cmplx(ri_core_a(i_part), kind=dc) ! (1)
       aero_state%apa%particle(i_part)%core_vol = &
            aero_data_diam2vol(aero_data, dp_core_a(i_part)) / 1d6 ! (m^3)
#endif
    end do
#endif
 
  end subroutine mosaic_aero_optical
 
!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
 
  !> Compute the optical properties of each aerosol particle for initial
  !> timestep.
  subroutine mosaic_aero_optical_init(env_state, aero_data, &
       aero_state, gas_data, gas_state)
 
#ifdef PMC_USE_MOSAIC
    use module_data_mosaic_aero, only: ri_shell_a, ri_core_a, &
         ext_cross, scat_cross, asym_particle, dp_core_a
#endif
 
    !> Environment state.
    type(env_state_t), intent(in) :: env_state
    !> Aerosol data.
    type(aero_data_t), intent(in) :: aero_data
    !> Aerosol state.
    type(aero_state_t), intent(inout) :: aero_state
    !> Gas data.
    type(gas_data_t), intent(in) :: gas_data
    !> Gas state.
    type(gas_state_t), intent(in) :: gas_state
 
#ifdef PMC_USE_MOSAIC
    ! MOSAIC function interfaces
    interface
       subroutine load_mosaic_parameters()
       end subroutine load_mosaic_parameters
#ifdef PMC_USE_MOSAIC_MULTI_OPT
       subroutine aerosol_optical(i_wavelength)
         integer, optional :: i_wavelength
       end subroutine aerosol_optical
#else
       subroutine aerosol_optical()
       end subroutine aerosol_optical
#endif
    end interface
 
    call load_mosaic_parameters
 
    ! map PartMC -> MOSAIC
    call mosaic_from_partmc(env_state, aero_data, aero_state, &
         gas_data, gas_state)
 
    call aerosol_optical
 
    call mosaic_aero_optical(env_state, aero_data, &
         aero_state, gas_data, gas_state)
#endif
 
  end subroutine mosaic_aero_optical_init
 
!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
 
  !> Compute the optical properties of each aerosol particle for a single
  !> wavelength selected by index when MOSAIC has multiple wavelengths enabled.
  subroutine mosaic_aero_optical_single_wavelength(env_state, aero_data, &
       aero_state, gas_data, gas_state, i_wavelength)
 
#ifdef PMC_USE_MOSAIC
    use module_data_mosaic_main, only: rh, pr_atm, te
    use module_data_mosaic_aero, only: ri_shell_a, ri_core_a, &
         ext_cross, scat_cross, asym_particle, dp_core_a, &
         p_atm, rh_pc, ah2o, t_k
#endif
 
    !> Environment state.
    type(env_state_t), intent(in) :: env_state
    !> Aerosol data.
    type(aero_data_t), intent(in) :: aero_data
    !> Aerosol state.
    type(aero_state_t), intent(inout) :: aero_state
    !> Gas data.
    type(gas_data_t), intent(in) :: gas_data
    !> Gas state.
    type(gas_state_t), intent(in) :: gas_state
    !> Wavelength index to compute properties.
    integer, intent(in) :: i_wavelength
 
#ifdef PMC_USE_MOSAIC
    ! MOSAIC function interfaces
    interface
#ifdef PMC_USE_MOSAIC_MULTI_OPT
       subroutine aerosol_optical(i_wavelength)
         integer, optional :: i_wavelength
       end subroutine aerosol_optical
#else
       subroutine aerosol_optical()
       end subroutine aerosol_optical
#endif
       subroutine load_mosaic_parameters()
       end subroutine load_mosaic_parameters
    end interface
 
    integer :: i_part
 
    call load_mosaic_parameters
 
    ! map PartMC -> MOSAIC
    call mosaic_from_partmc(env_state, aero_data, aero_state, &
         gas_data, gas_state)
 
    rh_pc = rh                                ! RH(%)
    ah2o = 0.01*rh_pc                         ! aH2O (aerosol water activity)
    p_atm = pr_atm                            ! P(atm)
    t_k = te                                  ! T(K)
#ifdef PMC_USE_MOSAIC_MULTI_OPT
    call aerosol_optical(i_wavelength)
#endif
    call mosaic_aero_optical(env_state, aero_data, &
         aero_state, gas_data, gas_state)
#endif
 
  end subroutine mosaic_aero_optical_single_wavelength
 
!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
 
  !> Fetch the wavelengths that optical cross-sections are calculated at in
  !> MOSAIC. If not using multiple wavelengths, the wavelength is the default
  !> value of 550 nm.
  subroutine mosaic_optical_wavelengths(aero_data)
 
#ifdef PMC_USE_MOSAIC
#ifdef PMC_USE_MOSAIC_MULTI_OPT
    use module_data_mosaic_aero, only: wavelength
#endif
#endif
    type(aero_data_t), intent(inout) :: aero_data
    integer :: i
 
    if(allocated(aero_data%wavelengths)) deallocate(aero_data%wavelengths)
    allocate(aero_data%wavelengths(n_swbands))
    aero_data%wavelengths = 0.0d0
#ifdef PMC_USE_MOSAIC_MULTI_OPT
    do i = 1,n_swbands
       aero_data%wavelengths(i) = wavelength(i) * 1d-9
    end do
#else
    aero_data%wavelengths = 550.0d0 * 1d-9
#endif
 
  end subroutine mosaic_optical_wavelengths
 
!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
 
end module pmc_mosaic