Ignore:
Timestamp:
Nov 16, 2017, 12:12:53 PM (7 years ago)
Author:
Laurent Fairhead
Message:

Suite de la synchro

File:
1 edited

Legend:

Unmodified
Added
Removed

  • LMDZ6/branches/DYNAMICO-conv/libf/phylmd/iophy.F90

    r3055 r3077  
    4545                                mpi_size, mpi_rank, klon_mpi, &
    4646                                is_sequential, is_south_pole_dyn
    47   USE mod_grid_phy_lmdz, ONLY: nbp_lon, nbp_lat, klon_glo
     47  USE mod_grid_phy_lmdz, ONLY: nbp_lon, nbp_lat, klon_glo, grid_type, unstructured
    4848  USE print_control_mod, ONLY: prt_level,lunout
    4949#ifdef CPP_IOIPSL
    … …  
    5151#endif
    5252#ifdef CPP_XIOS
    53   USE wxios, ONLY: wxios_domain_param
     53  use wxios, ONLY: wxios_domain_param, wxios_domain_param_unstructured, wxios_context_init
    5454#endif
    5555  IMPLICIT NONE
    … …  
    7070    INTEGER :: data_ibegin, data_iend
    7171
    72     CALL gather(rlat,rlat_glo)
    73     CALL bcast(rlat_glo)
    74     CALL gather(rlon,rlon_glo)
    75     CALL bcast(rlon_glo)
     72!#ifdef CPP_XIOS
     73!      CALL wxios_context_init
     74!#endif
     75   
     76
     77    IF (grid_type==unstructured) THEN
     78   
     79#ifdef CPP_XIOS
     80      CALL wxios_domain_param_unstructured("dom_glo")
     81#endif
     82
     83    ELSE
     84
     85      CALL gather(rlat,rlat_glo)
     86      CALL bcast(rlat_glo)
     87      CALL gather(rlon,rlon_glo)
     88      CALL bcast(rlon_glo)
    7689   
    7790!$OMP MASTER 
    … …  
    126139#endif
    127140#ifdef CPP_XIOS
    128     ! Set values for the mask:
    129     IF (mpi_rank == 0) THEN
    130         data_ibegin = 0
    131     ELSE
    132         data_ibegin = ii_begin - 1
    133     ENDIF
    134 
    135     IF (mpi_rank == mpi_size-1) THEN
    136         data_iend = nbp_lon
    137     ELSE
    138         data_iend = ii_end + 1
    139     ENDIF
    140 
    141     IF (prt_level>=10) THEN
    142       write(lunout,*) "init_iophy_new: mpirank=",mpi_rank," iibegin=",ii_begin , " ii_end=",ii_end," jjbegin=",jj_begin," jj_nb=",jj_nb," jj_end=",jj_end
    143       write(lunout,*) "init_iophy_new: mpirank=",mpi_rank," nbp_lon=",nbp_lon," nbp_lat=",nbp_lat
    144       write(lunout,*) "init_iophy_new: mpirank=",mpi_rank," data_ibegin=",data_ibegin," data_iend=",data_iend
    145       write(lunout,*) "init_iophy_new: mpirank=",mpi_rank," data_ibegin=",data_ibegin," data_iend=",data_iend
    146       write(lunout,*) "init_iophy_new: mpirank=",mpi_rank," is_south_pole=",is_south_pole_dyn
    147     ENDIF
    148 
    149     ! Initialize the XIOS domain coreesponding to this process:
    150     CALL wxios_domain_param("dom_glo", is_sequential, nbp_lon, jj_nb, nbp_lon, nbp_lat, &
    151                             1, nbp_lon, ii_begin, ii_end, jj_begin, jj_end,             &
    152                             klon_mpi+2*(nbp_lon-1), data_ibegin, data_iend,             &
    153                             io_lat, io_lon,is_south_pole_dyn,mpi_rank)
     141      ! Set values for the mask:
     142      IF (mpi_rank == 0) THEN
     143          data_ibegin = 0
     144      ELSE
     145          data_ibegin = ii_begin - 1
     146      END IF
     147
     148      IF (mpi_rank == mpi_size-1) THEN
     149          data_iend = nbp_lon
     150      ELSE
     151          data_iend = ii_end + 1
     152      END IF
     153
     154      IF (prt_level>=10) THEN
     155        write(lunout,*) "init_iophy_new: mpirank=",mpi_rank," iibegin=",ii_begin , " ii_end=",ii_end," jjbegin=",jj_begin," jj_nb=",jj_nb," jj_end=",jj_end
     156        write(lunout,*) "init_iophy_new: mpirank=",mpi_rank," nbp_lon=",nbp_lon," nbp_lat=",nbp_lat
     157        write(lunout,*) "init_iophy_new: mpirank=",mpi_rank," data_ibegin=",data_ibegin," data_iend=",data_iend
     158        write(lunout,*) "init_iophy_new: mpirank=",mpi_rank," data_ibegin=",data_ibegin," data_iend=",data_iend
     159        write(lunout,*) "init_iophy_new: mpirank=",mpi_rank," is_south_pole_dyn=",is_south_pole_dyn
     160      ENDIF
     161
     162      ! Initialize the XIOS domain coreesponding to this process:
    154163#endif
    155164!$OMP END MASTER
     165
     166#ifdef CPP_XIOS   
     167        CALL wxios_domain_param("dom_glo")
     168#endif
     169     
     170    ENDIF
    156171     
    157172  END SUBROUTINE init_iophy_new
    … …  
    925940                                 nid_files
    926941  USE print_control_mod, ONLY: prt_level,lunout
    927   USE mod_grid_phy_lmdz, ONLY : nbp_lon, nbp_lat
     942  USE mod_grid_phy_lmdz, ONLY : nbp_lon, nbp_lat, grid_type, unstructured, regular_lonlat
    928943#ifdef CPP_XIOS
    929944  USE xios, ONLY: xios_send_field
    930945#endif
    931 
     946  USE print_control_mod, ONLY: lunout, prt_level
    932947
    933948  IMPLICIT NONE
    … …  
    967982      IF (.not. ok_all_xml) THEN
    968983      IF (prt_level >= 10) THEN
    969       WRITE (lunout,*)"histwrite2d_phy: .not.vars_defined ; time to define ", trim(var%name)
     984      write(lunout,*)"histwrite2d_phy: .not.vars_defined ; time to define ", &
     985                     trim(var%name)
    970986      ENDIF
    971987      DO iff=iff_beg, iff_end
    … …  
    9911007    ENDIF
    9921008!$OMP MASTER
    993     CALL grid1Dto2D_mpi(buffer_omp,Field2d)
     1009    IF (grid_type==regular_lonlat) CALL grid1Dto2D_mpi(buffer_omp,Field2d)
    9941010
    9951011! La boucle sur les fichiers:
    … …  
    10011017             write(lunout,*)'Dans iophy histwrite2D,var%name ', trim(var%name)                       
    10021018          ENDIF
    1003           IF (SIZE(field) == klon) then
     1019         
     1020          IF (grid_type==regular_lonlat) THEN
     1021            IF (SIZE(field) == klon) then
    10041022              CALL xios_send_field(var%name, Field2d)
    1005           ELSE
    1006              CALL xios_send_field(var%name, field)
    1007           ENDIF
     1023            ELSE
     1024               CALL xios_send_field(var%name, field)
     1025            ENDIF
     1026          ELSE IF (grid_type==unstructured) THEN
     1027            CALL xios_send_field(var%name, buffer_omp)
     1028          ENDIF
    10081029          IF (prt_level >= 10) THEN
    1009              WRITE (lunout,*)'Dans iophy histwrite2D,var%name apres xios_send ', trim(var%name)                       
     1030             write(lunout,*)'Dans iophy histwrite2D,var%name apres xios_send ',&
     1031                             trim(var%name)                       
    10101032          ENDIF
    10111033#else
    … …  
    10191041               IF (firstx) THEN
    10201042                  IF (prt_level >= 10) THEN
    1021                      WRITE (lunout,*)'Dans iophy histwrite2D,iff,var%name ', iff,trim(var%name)                       
    1022                      WRITE (lunout,*)"histwrite2d_phy:.NOT.clef_stations(iff)and iff==iff_beg, call xios_send_field"
     1043                     write(lunout,*)'Dans iophy histwrite2D,iff,var%name ',&
     1044                                    iff,trim(var%name)                       
     1045                     write(lunout,*)"histwrite2d_phy:.NOT.clef_stations(iff)and iff==iff_beg, call xios_send_field"
    10231046                  ENDIF
    1024                   IF (SIZE(field) == klon) then
    1025                      CALL xios_send_field(var%name, Field2d)
    1026                   ELSE
    1027                      CALL xios_send_field(var%name, field)
     1047                  IF (grid_type==regular_lonlat) THEN
     1048                    IF (SIZE(field) == klon) then
     1049                       CALL xios_send_field(var%name, Field2d)
     1050                    ELSE
     1051                       CALL xios_send_field(var%name, field)
     1052                    ENDIF
     1053                  ELSE IF (grid_type==unstructured) THEN
     1054                    CALL xios_send_field(var%name, buffer_omp)
    10281055                  ENDIF
     1056
    10291057                  firstx=.false.
    10301058               ENDIF
    … …  
    10391067!#ifdef CPP_XIOS
    10401068!                        IF (iff == iff_beg) THEN
    1041 !                          if (prt_level >= 10) then
     1069!                          IF (prt_level >= 10) THEN
    10421070!                            write(lunout,*)"histwrite2d_phy: .NOT.clef_stations(iff) and iff==iff_beg, call xios_send_field"
    1043 !                          endif
     1071!                          ENDIF
    10441072!                          CALL xios_send_field(var%name, Field2d)
    10451073!                        ENDIF
    … …  
    10631091                       ENDIF ! of IF (is_sequential)
    10641092#ifndef CPP_IOIPSL_NO_OUTPUT
    1065                        IF (prt_level >= 10) THEn
     1093                       IF (prt_level >= 10) THEN
    10661094                         write(lunout,*)"histwrite2d_phy: clef_stations(iff) and iff==iff_beg, call wxios_write_2D"
    10671095                       ENDIF
    … …  
    10911119                                 nfiles, vars_defined, clef_stations, &
    10921120                                 nid_files
    1093   USE mod_grid_phy_lmdz, ONLY : nbp_lon, nbp_lat
     1121  USE mod_grid_phy_lmdz, ONLY : nbp_lon, nbp_lat, grid_type, regular_lonlat, unstructured
    10941122#ifdef CPP_XIOS
    10951123  USE xios, ONLY: xios_send_field
    … …  
    11521180    ENDIF
    11531181!$OMP MASTER
    1154     CALL grid1Dto2D_mpi(buffer_omp,field3d)
     1182    IF (grid_type==regular_lonlat) CALL grid1Dto2D_mpi(buffer_omp,field3d)
    11551183
    11561184
    … …  
    11641192                             trim(var%name)                       
    11651193          ENDIF
    1166           IF (SIZE(field,1) == klon) then
    1167              CALL xios_send_field(var%name, Field3d(:,:,1:nlevx))
    1168           ELSE
    1169              CALL xios_send_field(var%name, field)
     1194          IF (grid_type==regular_lonlat) THEN
     1195            IF (SIZE(field,1) == klon) then
     1196               CALL xios_send_field(var%name, Field3d(:,:,1:nlevx))
     1197            ELSE
     1198               CALL xios_send_field(var%name, field)
     1199            ENDIF
     1200          ELSE IF (grid_type==unstructured) THEN
     1201            CALL xios_send_field(var%name, buffer_omp(:,1:nlevx))
    11701202          ENDIF
     1203
    11711204#else
    11721205        CALL abort_physic ('iophy','cannot have ok_all_xml = .T. without CPP_XIOS defined' ,1)
    … …  
    11791212#ifdef CPP_XIOS
    11801213              IF (firstx) THEN
    1181                 IF (prt_level >= 10) THEn
    1182                   WRITE (lunout,*)'Dans iophy, histwrite3D iff nlev klev firstx', &
     1214                IF (prt_level >= 10) THEN
     1215                  write(lunout,*)'Dans iophy, histwrite3D iff nlev klev firstx', &
    11831216                                  iff,nlev,klev, firstx                       
    1184                   WRITE (lunout,*)'histwrite3d_phy: call xios_send_field for ', &
     1217                  write(lunout,*)'histwrite3d_phy: call xios_send_field for ', &
    11851218                                  trim(var%name), ' with iim jjm nlevx = ', &
    11861219                                  nbp_lon,jj_nb,nlevx
    11871220                ENDIF
    1188                 IF (SIZE(field,1) == klon) then
    1189                     CALL xios_send_field(var%name, Field3d(:,:,1:nlevx))
    1190                 ELSE
    1191                      CALL xios_send_field(var%name, field)
     1221                IF (grid_type==regular_lonlat) THEN
     1222                  IF (SIZE(field,1) == klon) then
     1223                      CALL xios_send_field(var%name, Field3d(:,:,1:nlevx))
     1224                  ELSE
     1225                       CALL xios_send_field(var%name, field)
     1226                  ENDIF
     1227                ELSE IF (grid_type==unstructured) THEN
     1228                  CALL xios_send_field(var%name, buffer_omp(:,1:nlevx))
    11921229                ENDIF
     1230
    11931231                firstx=.false.
    11941232              ENDIF
    … …  
    12511289                                is_sequential, klon_mpi_begin, klon_mpi_end, &
    12521290                                jj_nb, klon_mpi
    1253   USE mod_grid_phy_lmdz, ONLY : nbp_lon, nbp_lat
     1291  USE mod_grid_phy_lmdz, ONLY : nbp_lon, nbp_lat, grid_type, unstructured
    12541292  USE xios, ONLY: xios_send_field
    12551293  USE print_control_mod, ONLY: prt_level,lunout
    … …  
    12791317        CALL Gather_omp(field,buffer_omp)   
    12801318!$OMP MASTER
     1319
     1320      IF (grid_type==unstructured) THEN
     1321 
     1322        CALL xios_send_field(field_name, buffer_omp)
     1323
     1324      ELSE
     1325
    12811326        CALL grid1Dto2D_mpi(buffer_omp,Field2d)
    12821327   
    … …  
    13141359        DEALLOCATE(index2d)
    13151360        DEALLOCATE(fieldok)
     1361      ENDIF
    13161362!$OMP END MASTER   
    13171363    ENDIF
    … …  
    13281374                                jj_nb, klon_mpi
    13291375  USE xios, ONLY: xios_send_field
    1330   USE mod_grid_phy_lmdz, ONLY : nbp_lon, nbp_lat
     1376  USE mod_grid_phy_lmdz, ONLY : nbp_lon, nbp_lat, grid_type, unstructured
    13311377  USE print_control_mod, ONLY: prt_level,lunout
    13321378
    … …  
    13571403        CALL Gather_omp(field,buffer_omp)
    13581404!$OMP MASTER
     1405
     1406    IF (grid_type==unstructured) THEN
     1407
     1408      CALL xios_send_field(field_name, buffer_omp(:,1:nlev))
     1409
     1410    ELSE
    13591411        CALL grid1Dto2D_mpi(buffer_omp,field3d)
    13601412
    … …  
    13921444        DEALLOCATE(index3d)
    13931445        DEALLOCATE(fieldok)
     1446      ENDIF
    13941447!$OMP END MASTER   
    13951448    ENDIF
Note: See TracChangeset for help on using the changeset viewer.