[1897] | 1 | MODULE boxed_domain |
---|
| 2 | |
---|
| 3 | INTEGER,SAVE :: ni_glo |
---|
| 4 | INTEGER,SAVE :: nj_glo |
---|
| 5 | INTEGER,SAVE :: nb_proc |
---|
| 6 | |
---|
| 7 | TYPE local_domain |
---|
| 8 | INTEGER :: ni |
---|
| 9 | INTEGER :: nj |
---|
| 10 | INTEGER :: ibegin |
---|
| 11 | INTEGER :: iend |
---|
| 12 | INTEGER :: jbegin |
---|
| 13 | INTEGER :: jend |
---|
| 14 | REAL,POINTER :: lon(:,:) |
---|
| 15 | REAL,POINTER :: lat(:,:) |
---|
| 16 | END TYPE local_domain |
---|
| 17 | |
---|
| 18 | TYPE(local_domain),ALLOCATABLE,SAVE,TARGET :: domains(:) |
---|
| 19 | |
---|
| 20 | |
---|
| 21 | INTEGER,SAVE :: ni |
---|
| 22 | INTEGER,SAVE :: nj |
---|
| 23 | INTEGER,SAVE :: ibegin |
---|
| 24 | INTEGER,SAVE :: iend |
---|
| 25 | INTEGER,SAVE :: jbegin |
---|
| 26 | INTEGER,SAVE :: jend |
---|
| 27 | |
---|
| 28 | REAL,SAVE,ALLOCATABLE :: lon(:,:) |
---|
| 29 | REAL,SAVE,ALLOCATABLE :: lat(:,:) |
---|
| 30 | |
---|
| 31 | INTEGER,SAVE :: max_vert_size = 100 |
---|
| 32 | REAL,SAVE,ALLOCATABLE :: Field_buffer(:,:,:) |
---|
| 33 | |
---|
| 34 | CONTAINS |
---|
| 35 | |
---|
| 36 | SUBROUTINE Init_global_domain(ni_glo_,nj_glo_,nb_proc_) |
---|
| 37 | IMPLICIT NONE |
---|
| 38 | INTEGER, INTENT(IN) :: ni_glo_ |
---|
| 39 | INTEGER, INTENT(IN) :: nj_glo_ |
---|
| 40 | INTEGER, INTENT(IN) :: nb_proc_ |
---|
| 41 | |
---|
| 42 | ni_glo=ni_glo_ |
---|
| 43 | nj_glo=nj_glo_ |
---|
| 44 | nb_proc=nb_proc_ |
---|
| 45 | |
---|
| 46 | ALLOCATE(domains(nb_proc)) |
---|
| 47 | |
---|
| 48 | END SUBROUTINE Init_global_domain |
---|
| 49 | |
---|
| 50 | |
---|
| 51 | SUBROUTINE Init_local_domain(rank,ni,nj,ibegin,jbegin,lon,lat) |
---|
| 52 | IMPLICIT NONE |
---|
| 53 | INTEGER, INTENT(IN) :: rank |
---|
| 54 | INTEGER, INTENT(IN) :: ni |
---|
| 55 | INTEGER, INTENT(IN) :: nj |
---|
| 56 | INTEGER, INTENT(IN) :: ibegin |
---|
| 57 | INTEGER, INTENT(IN) :: jbegin |
---|
| 58 | REAL, INTENT(IN) :: lon(ni,nj) |
---|
| 59 | REAL, INTENT(IN) :: lat(ni,nj) |
---|
| 60 | |
---|
| 61 | domains(rank)%ni=ni |
---|
| 62 | domains(rank)%nj=nj |
---|
| 63 | domains(rank)%ibegin=ibegin |
---|
| 64 | domains(rank)%jbegin=jbegin |
---|
| 65 | domains(rank)%iend=ibegin+ni-1 |
---|
| 66 | domains(rank)%jend=jbegin+nj-1 |
---|
| 67 | ALLOCATE(domains(rank)%lon(ni,nj)) |
---|
| 68 | ALLOCATE(domains(rank)%lat(ni,nj)) |
---|
| 69 | domains(rank)%lon(:,:)=lon(:,:) |
---|
| 70 | domains(rank)%lat(:,:)=lat(:,:) |
---|
| 71 | |
---|
| 72 | END SUBROUTINE Init_local_domain |
---|
| 73 | |
---|
| 74 | |
---|
| 75 | SUBROUTINE Init_domain |
---|
| 76 | IMPLICIT NONE |
---|
| 77 | INTEGER :: rank |
---|
| 78 | INTEGER :: ib,ie,jb,je |
---|
| 79 | ibegin=domains(1)%ibegin |
---|
| 80 | iend=domains(1)%iend |
---|
| 81 | jbegin=domains(1)%jbegin |
---|
| 82 | jend=domains(1)%jend |
---|
| 83 | |
---|
| 84 | DO rank=1,nb_proc |
---|
| 85 | IF (domains(rank)%ibegin<ibegin) ibegin=domains(rank)%ibegin |
---|
| 86 | IF (domains(rank)%iend>iend) iend=domains(rank)%iend |
---|
| 87 | IF (domains(rank)%jbegin<jbegin) jbegin=domains(rank)%jbegin |
---|
| 88 | IF (domains(rank)%jend>jend) jend=domains(rank)%jend |
---|
| 89 | ENDDO |
---|
| 90 | |
---|
| 91 | ni=iend-ibegin+1 |
---|
| 92 | nj=jend-jbegin+1 |
---|
| 93 | ALLOCATE(lon(ni,nj)) |
---|
| 94 | ALLOCATE(lat(ni,nj)) |
---|
| 95 | |
---|
| 96 | DO rank=1,nb_proc |
---|
| 97 | ib=domains(rank)%ibegin-ibegin+1 |
---|
| 98 | ie=domains(rank)%iend-ibegin+1 |
---|
| 99 | jb=domains(rank)%jbegin-jbegin+1 |
---|
| 100 | je=domains(rank)%jend-jbegin+1 |
---|
| 101 | |
---|
| 102 | domains(rank)%ibegin=ib |
---|
| 103 | domains(rank)%iend=ie |
---|
| 104 | domains(rank)%jbegin=jb |
---|
| 105 | domains(rank)%jend=je |
---|
| 106 | |
---|
| 107 | lon(ib:ie,jb:je)=domains(rank)%lon(:,:) |
---|
| 108 | lat(ib:ie,jb:je)=domains(rank)%lat(:,:) |
---|
| 109 | ENDDO |
---|
| 110 | |
---|
| 111 | ALLOCATE(field_buffer(ni,nj,max_vert_size)) |
---|
| 112 | |
---|
| 113 | END SUBROUTINE |
---|
| 114 | |
---|
| 115 | SUBROUTINE bufferize_field2d(field,rank) |
---|
| 116 | IMPLICIT NONE |
---|
| 117 | REAL,INTENT(IN) :: field(:,:) |
---|
| 118 | INTEGER,INTENT(IN) :: rank |
---|
| 119 | TYPE(local_domain), POINTER :: Pt |
---|
| 120 | |
---|
| 121 | Pt=>domains(rank) |
---|
| 122 | field_buffer(Pt%ibegin:Pt%iend,Pt%jbegin:Pt%jend,1)=field(:,:) |
---|
| 123 | |
---|
| 124 | END SUBROUTINE bufferize_field2d |
---|
| 125 | |
---|
| 126 | |
---|
| 127 | SUBROUTINE bufferize_field3d(field,rank) |
---|
| 128 | IMPLICIT NONE |
---|
| 129 | REAL,INTENT(IN) :: field(:,:,:) |
---|
| 130 | INTEGER,INTENT(IN) :: rank |
---|
| 131 | TYPE(local_domain), POINTER :: Pt |
---|
| 132 | |
---|
| 133 | Pt=>domains(rank) |
---|
| 134 | IF (size(field,3)>max_vert_size) THEN |
---|
| 135 | DEALLOCATE(field_buffer) |
---|
| 136 | max_vert_size=max_vert_size*2 |
---|
| 137 | ALLOCATE(field_buffer(ni,nj,max_vert_size)) |
---|
| 138 | ENDIF |
---|
| 139 | |
---|
| 140 | field_buffer(Pt%ibegin:Pt%iend,Pt%jbegin:Pt%jend,1:size(field,3))=field(:,:,:) |
---|
| 141 | |
---|
| 142 | END SUBROUTINE bufferize_field3d |
---|
| 143 | |
---|
| 144 | |
---|
| 145 | END MODULE boxed_domain |
---|