subroutine setlats_a(lats_nodes_a,lats_nodes_ext,global_lats_a, & global_lats_ext,iprint,lonsperlat) cc use gfs_dyn_resol_def use gfs_dyn_layout1 use gfs_dyn_mpi_def implicit none cc integer lats_nodes_a(nodes) cc integer global_lats_a(latg) integer lats_nodes_ext(nodes) integer global_lats_ext(latg+2*jintmx+2*nypt*(nodes-1)) cc integer iprint,opt,ifin,nodesio cc integer lonsperlat(latg) cc integer ijk,jcount,jpt,lat,lats_sum,node,i,j integer ILATPE,ngrptg,ngrptl,ipe,irest,idp cc integer,allocatable :: lats_hold(:,:) cc allocate ( lats_hold(latg,nodes) ) cc ! iprint=1 iprint=0 OPT=1 lats_nodes_a=0 !jw if (liope) then !jw if (icolor.eq.2) then !jw nodesio=1 !jw else nodesio=nodes !jw endif !jw else !jw nodesio=nodes !jw endif cc ngrptg=0 do lat=1,latg if (opt.eq.1) then ifin=lonsperlat(lat) elseif(opt.eq.2) then ifin=lonf endif do i=1,ifin ngrptg=ngrptg+1 enddo enddo ! ! distribution of the grid ILATPE=ngrptg/nodesio ngrptl=0 ipe=0 irest=0 idp=1 do lat=1,latg if (opt.eq.1) then ifin=lonsperlat(lat) elseif(opt.eq.2) then ifin=lonf endif ngrptl=ngrptl+ifin if (ngrptl*nodesio.le.ngrptg+irest) then lats_nodes_a(ipe+1) = lats_nodes_a(ipe+1) + 1 lats_hold(idp,ipe+1)=lat idp=idp+1 else ipe=ipe+1 if (ipe.le.nodesio) lats_hold(1,ipe+1) = lat idp=2 irest=irest+ngrptg-(ngrptl-ifin)*nodesio ngrptl=ifin lats_nodes_a(ipe+1) = lats_nodes_a(ipe+1) + 1 endif enddo !! do node=1,nodesio if (nodesio.eq.1) then lats_nodes_ext(node)=lats_nodes_a(node)+2*jintmx else if (node.eq.1.or.node.eq.nodesio) then lats_nodes_ext(node)=lats_nodes_a(node)+jintmx+nypt else lats_nodes_ext(node)=lats_nodes_a(node)+2*nypt endif endif enddo cc........................................................ cc jpt=0 do node=1,nodesio if ( lats_nodes_a(node) .gt. 0 ) then do jcount=1,lats_nodes_a(node) global_lats_a(jpt+jcount) = lats_hold(jcount,node) enddo endif jpt=jpt+lats_nodes_a(node) enddo cc jpt=0 do node=1,nodesio if (nodesio.eq.1) then do i=1,jintmx global_lats_ext(i)=global_lats_a(1) enddo do i=1,jintmx global_lats_ext(jintmx+latg+i)=global_lats_a(latg) enddo do i=1,latg global_lats_ext(i+jintmx)=global_lats_a(i) enddo else do jcount=1,lats_nodes_a(node) global_lats_ext(jpt+jintmx+jcount+2*nypt*(node-1))= & global_lats_a(jpt+jcount) enddo if (node.eq.1) then do i=1,jintmx global_lats_ext(i)=global_lats_a(1) enddo do i=1,nypt global_lats_ext(jintmx+lats_nodes_a(node)+i)= & global_lats_a(lats_nodes_a(node))+i enddo elseif (node.eq.nodesio) then do i=1,jintmx global_lats_ext(latg+jintmx+2*nypt*(nodesio-1)+i)= & global_lats_a(latg) enddo do i=nypt,1,-1 global_lats_ext(jpt+jintmx+2*nypt*(node-1)-i+1)= & global_lats_a(jpt)-i+1 enddo else do i=nypt,1,-1 global_lats_ext(jpt+jintmx+2*nypt*(node-1)-i+1)= & global_lats_a(jpt)-i+1 enddo do i=1,nypt global_lats_ext(jpt+jintmx+2*nypt*(node-1)+ & lats_nodes_a(node)+i)= & global_lats_a(jpt+lats_nodes_a(node))+i enddo endif endif jpt=jpt+lats_nodes_a(node) enddo cc if ( iprint .ne. 1 ) return cc jpt=0 do node=1,nodesio if ( lats_nodes_a(node) .gt. 0 ) then print 600 lats_sum=0 do jcount=1,lats_nodes_a(node) lats_sum=lats_sum + lonsperlat(global_lats_a(jpt+jcount)) print 700, node-1, x node, lats_nodes_a(node), x jpt+jcount, global_lats_a(jpt+jcount), x lonsperlat(global_lats_a(jpt+jcount)), x lats_sum enddo endif jpt=jpt+lats_nodes_a(node) enddo cc print 600 cc 600 format ( ' ' ) cc 700 format ( 'setlats me=', i4, x ' lats_nodes_a(', i4, ' )=', i4, x ' global_lats_a(', i4, ' )=', i4, x ' lonsperlat=', i5, x ' lats_sum=', i6 ) cc deallocate ( lats_hold ) cc return end