Changeset 847

Timestamp:

05/05/19 00:08:28 (5 years ago)

Author:

dubos

Message:

devel : split etat0 into etat0, etat0_collocated and etat0_isothermal

Location:

codes/icosagcm/devel/src/initial

Files:

• etat0.f90 (modified) (3 diffs)
• etat0_collocated.f90 (added)
• etat0_isothermal.f90 (added)

codes/icosagcm/devel/src/initial/etat0.f90

@@ -1 +1 @@
 MODULE etat0_mod
   USE icosa
-  USE omp_para
-  USE caldyn_vars_mod
+  USE etat0_collocated_mod
   IMPLICIT NONE         
   PRIVATE
 
-    CHARACTER(len=255),SAVE :: etat0_type
-!$OMP THREADPRIVATE(etat0_type)
-
-    REAL(rstd) :: etat0_temp
-
-    PUBLIC :: etat0, init_etat0, etat0_type
+  PUBLIC :: etat0, init_etat0
 
 ! Important notes for OpenMP
@@ -67 +61 @@
 
   SUBROUTINE etat0(f_ps,f_mass,f_phis,f_theta_rhodz,f_u, f_geopot,f_w, f_q)
+    USE omp_para, ONLY : is_omp_level_master
     USE mpipara, ONLY : is_mpi_root
     USE disvert_mod
     ! Generic interface
+    USE etat0_isothermal_mod, ONLY : getin_etat0_isothermal=>getin_etat0
     USE etat0_dcmip1_mod, ONLY : getin_etat0_dcmip1=>getin_etat0
     USE etat0_dcmip2_mod, ONLY : getin_etat0_dcmip2=>getin_etat0
@@ -183 +179 @@
   END SUBROUTINE etat0
 
-  SUBROUTINE etat0_collocated(f_phis,f_ps,f_mass,f_theta_rhodz,f_u, f_geopot,f_W, f_q)
-    USE theta2theta_rhodz_mod
-    TYPE(t_field),POINTER :: f_ps(:)
-    TYPE(t_field),POINTER :: f_mass(:)
-    TYPE(t_field),POINTER :: f_phis(:)
-    TYPE(t_field),POINTER :: f_theta_rhodz(:)
-    TYPE(t_field),POINTER :: f_u(:)
-    TYPE(t_field),POINTER :: f_geopot(:)
-    TYPE(t_field),POINTER :: f_W(:)
-    TYPE(t_field),POINTER :: f_q(:)
-  
-    TYPE(t_field),POINTER,SAVE :: f_temp(:)
-    REAL(rstd),POINTER :: ps(:)
-    REAL(rstd),POINTER :: mass(:,:)
-    REAL(rstd),POINTER :: phis(:)
-    REAL(rstd),POINTER :: theta_rhodz(:,:,:)
-    REAL(rstd),POINTER :: temp(:,:)
-    REAL(rstd),POINTER :: u(:,:)
-    REAL(rstd),POINTER :: geopot(:,:)
-    REAL(rstd),POINTER :: W(:,:)
-    REAL(rstd),POINTER :: q(:,:,:)
-    INTEGER :: ind
-
-    CALL allocate_field(f_temp,field_t,type_real,llm,name='temp')
-
-    DO ind=1,ndomain
-      IF (.NOT. assigned_domain(ind) .OR. .NOT. is_omp_level_master) CYCLE
-!      IF (.NOT. assigned_domain(ind)) CYCLE
-      CALL swap_dimensions(ind)
-      CALL swap_geometry(ind)
-      ps=f_ps(ind)
-      mass=f_mass(ind)
-      phis=f_phis(ind)
-      theta_rhodz=f_theta_rhodz(ind)
-      temp=f_temp(ind)
-      u=f_u(ind)
-      geopot=f_geopot(ind)
-      w=f_w(ind)
-      q=f_q(ind)
-
-      IF( TRIM(etat0_type)=='williamson91.6' ) THEN
-         CALL compute_etat0_collocated(ps,mass, phis, theta_rhodz(:,:,1), u, geopot, W, q)
-      ELSE
-         CALL compute_etat0_collocated(ps,mass, phis, temp, u, geopot, W, q)
-      ENDIF
-
-      IF( TRIM(etat0_type)/='williamson91.6' ) CALL compute_temperature2entropy(ps,temp,q,theta_rhodz, 1)
-    
-    ENDDO
-    
-    CALL deallocate_field(f_temp)
-    
-  END SUBROUTINE etat0_collocated
-
-  SUBROUTINE compute_temperature2entropy(ps,temp,q,theta_rhodz,offset)
-    USE icosa
-    USE pression_mod
-    USE exner_mod
-    USE omp_para
-    REAL(rstd),INTENT(IN)  :: ps(iim*jjm)
-    REAL(rstd),INTENT(IN)  :: temp(iim*jjm,llm)
-    REAL(rstd),INTENT(IN)  :: q(iim*jjm,llm,nqtot)
-    REAL(rstd),INTENT(OUT) :: theta_rhodz(iim*jjm,llm)
-    INTEGER,INTENT(IN) :: offset
-
-    REAL(rstd) :: p(iim*jjm,llm+1)
-    REAL(rstd) :: cppd,Rd, mass, p_ij, q_ij,r_ij, chi,nu, entropy, theta
-    INTEGER :: i,j,ij,l
-
-    cppd=cpp
-    Rd=kappa*cppd
-
-    CALL compute_pression(ps,p,offset)
-    ! flush p
-    DO    l    = ll_begin, ll_end
-       DO j=jj_begin-offset,jj_end+offset
-          DO i=ii_begin-offset,ii_end+offset
-             ij=(j-1)*iim+i
-             mass = (p(ij,l)-p(ij,l+1))/g ! dry+moist mass
-             p_ij = .5*(p(ij,l)+p(ij,l+1))  ! pressure at full level
-             SELECT CASE(caldyn_thermo)
-             CASE(thermo_theta)
-                theta = temp(ij,l)*(p_ij/preff)**(-kappa) 
-                theta_rhodz(ij,l) = mass * theta
-             CASE(thermo_entropy)
-                nu = log(p_ij/preff)
-                chi = log(temp(ij,l)/Treff)
-                entropy = cppd*chi-Rd*nu
-                theta_rhodz(ij,l) = mass * entropy
-!             CASE(thermo_moist)
-!                q_ij=q(ij,l,1)
-!                r_ij=1.-q_ij
-!                mass=mass*(1-q_ij) ! dry mass
-!                nu = log(p_ij/preff)
-!                chi = log(temp(ij,l)/Treff)
-!                entropy = r_ij*(cppd*chi-Rd*nu) + q_ij*(cppv*chi-Rv*nu)
-!                theta_rhodz(ij,l) = mass * entropy                
-                CASE DEFAULT
-                   STOP
-             END SELECT
-          ENDDO
-       ENDDO
-    ENDDO
-  END SUBROUTINE compute_temperature2entropy
-
-  SUBROUTINE compute_etat0_collocated(ps,mass,phis,temp_i,u, geopot,W, q)
-    USE wind_mod
-    USE disvert_mod
-    USE etat0_jablonowsky06_mod, ONLY : compute_jablonowsky06 => compute_etat0
-    USE etat0_dcmip1_mod, ONLY : compute_dcmip1 => compute_etat0
-    USE etat0_dcmip2_mod, ONLY : compute_dcmip2 => compute_etat0
-    USE etat0_dcmip3_mod, ONLY : compute_dcmip3 => compute_etat0
-    USE etat0_dcmip4_mod, ONLY : compute_dcmip4 => compute_etat0
-    USE etat0_dcmip5_mod, ONLY : compute_dcmip5 => compute_etat0
-    USE etat0_bubble_mod, ONLY : compute_bubble => compute_etat0  
-    USE etat0_williamson_mod, ONLY : compute_w91_6 => compute_etat0
-    USE etat0_temperature_mod, ONLY: compute_etat0_temperature => compute_etat0
-    USE etat0_dcmip2016_baroclinic_wave_mod, ONLY : compute_dcmip2016_baroclinic_wave => compute_etat0
-    USE etat0_dcmip2016_cyclone_mod, ONLY : compute_dcmip2016_cyclone => compute_etat0
-    USE etat0_dcmip2016_supercell_mod, ONLY : compute_dcmip2016_supercell => compute_etat0
-    REAL(rstd),INTENT(INOUT) :: ps(iim*jjm)
-    REAL(rstd),INTENT(INOUT) :: mass(iim*jjm,llm)
-    REAL(rstd),INTENT(OUT) :: phis(iim*jjm)
-    REAL(rstd),INTENT(OUT) :: temp_i(iim*jjm,llm)
-    REAL(rstd),INTENT(OUT) :: u(3*iim*jjm,llm)
-    REAL(rstd),INTENT(OUT) :: W(iim*jjm,llm+1)
-    REAL(rstd),INTENT(OUT) :: geopot(iim*jjm,llm+1)
-    REAL(rstd),INTENT(OUT) :: q(iim*jjm,llm,nqtot)
-
-    REAL(rstd) :: ulon_i(iim*jjm,llm)
-    REAL(rstd) :: ulat_i(iim*jjm,llm)
-
-    REAL(rstd) :: ps_e(3*iim*jjm)
-    REAL(rstd) :: mass_e(3*iim*jjm,llm)
-    REAL(rstd) :: phis_e(3*iim*jjm)
-    REAL(rstd) :: temp_e(3*iim*jjm,llm)
-    REAL(rstd) :: geopot_e(3*iim*jjm,llm+1)
-    REAL(rstd) :: ulon_e(3*iim*jjm,llm)
-    REAL(rstd) :: ulat_e(3*iim*jjm,llm)
-    REAL(rstd) :: q_e(3*iim*jjm,llm,nqtot)
-
-    INTEGER :: l,i,j,ij
-    REAL :: p_ik, v_ik, mass_ik
-    LOGICAL :: autoinit_mass, autoinit_NH
-
-    ! For NH geopotential and vertical momentum must be initialized.
-    ! Unless autoinit_NH is set to .FALSE. , they will be initialized
-    ! to hydrostatic geopotential and zero
-    autoinit_mass = .TRUE.
-    autoinit_NH = .NOT. hydrostatic
-    w(:,:) = 0
-
-    SELECT CASE (TRIM(etat0_type))
-    CASE ('isothermal')
-       CALL compute_etat0_isothermal(iim*jjm, phis, ps, temp_i, ulon_i, ulat_i, q)
-       CALL compute_etat0_isothermal(3*iim*jjm, phis_e, ps_e, temp_e, ulon_e, ulat_e, q_e)
-    CASE ('temperature_profile')
-       CALL compute_etat0_temperature(iim*jjm, phis, ps, temp_i, ulon_i, ulat_i, q)
-       CALL compute_etat0_temperature(3*iim*jjm, phis_e, ps_e, temp_e, ulon_e, ulat_e, q_e)
-    CASE('jablonowsky06')
-       CALL compute_jablonowsky06(iim*jjm,lon_i,lat_i, phis, ps, temp_i, ulon_i, ulat_i)
-       CALL compute_jablonowsky06(3*iim*jjm,lon_e,lat_e, phis_e, ps_e, temp_e, ulon_e, ulat_e)
-    CASE('dcmip1')
-       CALL compute_dcmip1(iim*jjm,lon_i,lat_i, phis, ps, temp_i, ulon_i, ulat_i, q)
-       CALL compute_dcmip1(3*iim*jjm,lon_e,lat_e, phis_e, ps_e, temp_e, ulon_e, ulat_e, q_e)
-    CASE ('dcmip2_mountain','dcmip2_schaer_noshear','dcmip2_schaer_shear')
-       CALL compute_dcmip2(iim*jjm,lon_i,lat_i, phis, ps, temp_i, ulon_i, ulat_i)
-       CALL compute_dcmip2(3*iim*jjm,lon_e,lat_e, phis_e, ps_e, temp_e, ulon_e, ulat_e)      
-    CASE('dcmip3')
-       CALL compute_dcmip3(iim*jjm,lon_i,lat_i, phis, ps, temp_i, ulon_i, ulat_i, geopot, q)
-       CALL compute_dcmip3(3*iim*jjm,lon_e,lat_e, phis_e, ps_e, temp_e, ulon_e, ulat_e, geopot_e, q_e)
-       autoinit_NH = .FALSE. ! compute_dcmip3 initializes geopot
-    CASE('dcmip4')
-       CALL compute_dcmip4(iim*jjm,lon_i,lat_i, phis, ps, temp_i, ulon_i, ulat_i, q)
-       CALL compute_dcmip4(3*iim*jjm,lon_e,lat_e, phis_e, ps_e, temp_e, ulon_e, ulat_e, q_e)
-    CASE('dcmip5')
-       CALL compute_dcmip5(iim*jjm,lon_i,lat_i, phis, ps, temp_i, ulon_i, ulat_i, q)
-       CALL compute_dcmip5(3*iim*jjm,lon_e,lat_e, phis_e, ps_e, temp_e, ulon_e, ulat_e, q_e)
-    CASE('bubble')
-       CALL compute_bubble(iim*jjm,lon_i,lat_i, phis, ps, temp_i, ulon_i, ulat_i, geopot, q)
-       CALL compute_bubble(3*iim*jjm,lon_e,lat_e, phis_e, ps_e, temp_e, ulon_e, ulat_e, geopot_e, q_e)
-!       autoinit_NH = .FALSE. ! compute_bubble initializes geopot
-    CASE('williamson91.6')
-       CALL compute_w91_6(iim*jjm,lon_i,lat_i, phis, mass(:,1), temp_i(:,1), ulon_i(:,1), ulat_i(:,1))
-       CALL compute_w91_6(3*iim*jjm,lon_e,lat_e, phis_e, mass_e(:,1), temp_e(:,1), ulon_e(:,1), ulat_e(:,1))
-       autoinit_mass = .FALSE. ! do not overwrite mass
-    CASE('dcmip2016_baroclinic_wave')
-       CALL compute_dcmip2016_baroclinic_wave(iim*jjm,lon_i,lat_i, phis, ps, temp_i, ulon_i, ulat_i, q)
-       CALL compute_dcmip2016_baroclinic_wave(3*iim*jjm,lon_e,lat_e, phis_e, ps_e, temp_e, ulon_e, ulat_e, q_e)
-    CASE('dcmip2016_cyclone')
-       CALL compute_dcmip2016_cyclone(iim*jjm,lon_i,lat_i, phis, ps, temp_i, ulon_i, ulat_i, q)
-       CALL compute_dcmip2016_cyclone(3*iim*jjm,lon_e,lat_e, phis_e, ps_e, temp_e, ulon_e, ulat_e, q_e)
-    CASE('dcmip2016_supercell')
-       CALL compute_dcmip2016_supercell(iim*jjm,lon_i,lat_i, phis, ps, temp_i, ulon_i, ulat_i, q)
-       CALL compute_dcmip2016_supercell(3*iim*jjm,lon_e,lat_e, phis_e, ps_e, temp_e, ulon_e, ulat_e, q_e)
-    END SELECT
-
-    IF(autoinit_mass) CALL compute_rhodz(.TRUE., ps, mass) ! initialize mass distribution using ps
-    IF(autoinit_NH) THEN
-       geopot(:,1) = phis(:) ! surface geopotential
-       DO l = 1, llm
-          DO ij=1,iim*jjm
-             ! hybrid pressure coordinate
-             p_ik = ptop + mass_ak(l) + mass_bk(l)*ps(ij)
-             mass_ik = (mass_dak(l) + mass_dbk(l)*ps(ij))/g
-             ! v=R.T/p, R=kappa*cpp
-             v_ik = kappa*cpp*temp_i(ij,l)/p_ik
-             geopot(ij,l+1) = geopot(ij,l) + mass_ik*v_ik*g
-          END DO
-       END DO
-    END IF
-
-    CALL compute_wind_perp_from_lonlat_compound(ulon_e, ulat_e, u)
-
-  END SUBROUTINE compute_etat0_collocated
-
-!----------------------------- Resting isothermal state --------------------------------
-
-  SUBROUTINE getin_etat0_isothermal
-    etat0_temp=300
-    CALL getin("etat0_isothermal_temp",etat0_temp)
-  END SUBROUTINE getin_etat0_isothermal
-
-  SUBROUTINE compute_etat0_isothermal(ngrid, phis, ps, temp, ulon, ulat, q)
-    INTEGER, INTENT(IN)    :: ngrid
-    REAL(rstd),INTENT(OUT) :: phis(ngrid)
-    REAL(rstd),INTENT(OUT) :: ps(ngrid)
-    REAL(rstd),INTENT(OUT) :: temp(ngrid,llm)
-    REAL(rstd),INTENT(OUT) :: ulon(ngrid,llm)
-    REAL(rstd),INTENT(OUT) :: ulat(ngrid,llm)
-    REAL(rstd),INTENT(OUT) :: q(ngrid,llm,nqtot)
-    phis(:)=0
-    ps(:)=preff
-    temp(:,:)=etat0_temp
-    ulon(:,:)=0
-    ulat(:,:)=0
-    q(:,:,:)=0
-  END SUBROUTINE compute_etat0_isothermal
-
 END MODULE etat0_mod