! *********************************************************************** ! ! Copyright (C) 2010 Bill Paxton ! ! this file is part of mesa. ! ! mesa is free software; you can redistribute it and/or modify ! it under the terms of the gnu general library public license as published ! by the free software foundation; either version 2 of the license, or ! (at your option) any later version. ! ! mesa is distributed in the hope that it will be useful, ! but without any warranty; without even the implied warranty of ! merchantability or fitness for a particular purpose. see the ! gnu library general public license for more details. ! ! you should have received a copy of the gnu library general public license ! along with this software; if not, write to the free software ! foundation, inc., 59 temple place, suite 330, boston, ma 02111-1307 usa ! ! *********************************************************************** module run_star_extras use star_lib use star_def use const_def implicit none integer :: time0, time1, clock_rate double precision, parameter :: expected_runtime = 7 ! minutes ! these routines are called by the standard run_star check_model contains subroutine extras_controls(s, ierr) type (star_info), pointer :: s integer, intent(out) :: ierr ierr = 0 ! this is the place to set any procedure pointers you want to change ! e.g., other_wind, other_mixing, other_energy (see star_data.dek) end subroutine extras_controls integer function extras_startup(s, id, restart, ierr) type (star_info), pointer :: s integer, intent(in) :: id logical, intent(in) :: restart integer, intent(out) :: ierr ierr = 0 extras_startup = 0 call system_clock(time0,clock_rate) if (.not. restart) then call alloc_extra_info(s) else ! it is a restart call unpack_extra_info(s) end if end function extras_startup subroutine extras_after_evolve(s, id, id_extra, ierr) type (star_info), pointer :: s integer, intent(in) :: id, id_extra integer, intent(out) :: ierr double precision :: dt ierr = 0 call system_clock(time1,clock_rate) dt = dble(time1 - time0) / clock_rate / 60 if (dt > 10*expected_runtime) then write(*,'(/,a30,2f18.6,a,/)') '>>>>>>> EXCESSIVE runtime', & dt, expected_runtime, ' <<<<<<<<< ERROR' else write(*,'(/,a30,99f18.6,/)') 'runtime (minutes)', dt, expected_runtime end if end subroutine extras_after_evolve ! returns either keep_going, retry, backup, or terminate. integer function extras_check_model(s, id, id_extra) type (star_info), pointer :: s integer, intent(in) :: id, id_extra extras_check_model = keep_going if (.false. .and. s% star_mass_h1 < 0.35d0) then ! stop when star hydrogen mass drops to specified level extras_check_model = terminate write(*, *) 'have reached desired hydrogen mass' return end if if (s% log_surface_luminosity > 4d0) then s% mixing_length_alpha = 1d0 end if if (s% log_surface_luminosity > 4d0 .and. s% Teff < 5d5) then s% mixing_length_alpha = 0.5d0 s% mass_change = 0d0 end if end function extras_check_model integer function how_many_extra_log_columns(s, id, id_extra) type (star_info), pointer :: s integer, intent(in) :: id, id_extra how_many_extra_log_columns = 3 end function how_many_extra_log_columns subroutine data_for_extra_log_columns(s, id, id_extra, n, names, vals, ierr) use chem_def, only: ih1 type (star_info), pointer :: s integer, intent(in) :: id, id_extra, n character (len=maxlen_log_column_name) :: names(n) double precision :: vals(n) integer, intent(out) :: ierr integer :: k double precision :: dq00, dqm1, dqsum, T, rho, h1, Tmax, rhomax ierr = 0 names(1) = 'T_max' names(2) = 'rho_max' names(3) = 'v_surf' h1 = s% net_iso(ih1) Tmax = 0d0 rhomax = 0d0 do k = 2, s% nz if (s% xa(h1,k) < 1d-4 .and. k > 1) cycle ! interpolate values at face dq00 = s% dq(k) dqm1 = s% dq(k-1) dqsum = dq00 + dqm1 rho = (dqm1*s% rho(k) + dq00*s% rho(k-1))/dqsum T = (dqm1*s% T(k) + dq00*s% T(k-1))/dqsum if ( T > Tmax) then Tmax = T rhomax = rho end if end do vals(1) = Tmax vals(2) = rhomax vals(3) = s% v_surf end subroutine data_for_extra_log_columns integer function how_many_extra_profile_columns(s, id, id_extra) type (star_info), pointer :: s integer, intent(in) :: id, id_extra how_many_extra_profile_columns = 0 end function how_many_extra_profile_columns subroutine data_for_extra_profile_columns(s, id, id_extra, n, nz, names, vals, ierr) type (star_info), pointer :: s integer, intent(in) :: id, id_extra, n, nz character (len=maxlen_profile_column_name) :: names(n) double precision :: vals(nz,n) integer, intent(out) :: ierr integer :: k ierr = 0 ! here is an example for adding a profile column !if (n /= 1) stop 'data_for_extra_profile_columns' !names(1) = 'beta' !do k = 1, nz ! vals(k,1) = s% Pgas(k)/s% P(k) !end do end subroutine data_for_extra_profile_columns ! returns either keep_going or terminate. integer function extras_finish_step(s, id, id_extra) type (star_info), pointer :: s integer, intent(in) :: id, id_extra integer :: ierr extras_finish_step = keep_going call store_extra_info(s) ! if (.false. .and. s% model_number == 1000) then ! save FGONG file for model 1000 ! call star_write_fgong(id, 'mod1000.fgong', ierr) ! if (ierr /= 0) then ! extras_finish_step = terminate ! write(*, *) 'star_write_fgong failed: ierr', ierr ! return ! end if ! end if end function extras_finish_step ! routines for saving and restoring extra data so can do restarts ! put these defs at the top and delete from the following routines !integer, parameter :: extra_info_alloc = 1 !integer, parameter :: extra_info_get = 2 !integer, parameter :: extra_info_put = 3 subroutine alloc_extra_info(s) integer, parameter :: extra_info_alloc = 1 type (star_info), pointer :: s call move_extra_info(s,extra_info_alloc) end subroutine alloc_extra_info subroutine unpack_extra_info(s) integer, parameter :: extra_info_get = 2 type (star_info), pointer :: s call move_extra_info(s,extra_info_get) end subroutine unpack_extra_info subroutine store_extra_info(s) integer, parameter :: extra_info_put = 3 type (star_info), pointer :: s call move_extra_info(s,extra_info_put) end subroutine store_extra_info subroutine move_extra_info(s,op) integer, parameter :: extra_info_alloc = 1 integer, parameter :: extra_info_get = 2 integer, parameter :: extra_info_put = 3 type (star_info), pointer :: s integer, intent(in) :: op integer :: i, j, num_ints, num_dbls, ierr i = 0 ! call move_int or move_flg num_ints = i i = 0 ! call move_dbl num_dbls = i if (op /= extra_info_alloc) return if (num_ints == 0 .and. num_dbls == 0) return ierr = 0 call star_alloc_extras(s% id, num_ints, num_dbls, ierr) if (ierr /= 0) then write(*,*) 'failed in star_alloc_extras' write(*,*) 'alloc_extras num_ints', num_ints write(*,*) 'alloc_extras num_dbls', num_dbls stop 1 end if contains subroutine move_dbl(dbl) double precision :: dbl i = i+1 select case (op) case (extra_info_get) dbl = s% extra_work(i) case (extra_info_put) s% extra_work(i) = dbl end select end subroutine move_dbl subroutine move_int(int) integer :: int i = i+1 select case (op) case (extra_info_get) int = s% extra_iwork(i) case (extra_info_put) s% extra_iwork(i) = int end select end subroutine move_int subroutine move_flg(flg) logical :: flg i = i+1 select case (op) case (extra_info_get) flg = (s% extra_iwork(i) /= 0) case (extra_info_put) if (flg) then s% extra_iwork(i) = 1 else s% extra_iwork(i) = 0 end if end select end subroutine move_flg end subroutine move_extra_info end module run_star_extras