program compo_ec
implicit none
integer::i,j,h,imn,imx,jmn,jmx,syear,eyear,iyear,smonth,emonth,imonth,ice1,jce1,xst,xen,yst,yen,month1,n,nto,nto2,nmx,nt,nnt,out,nout,nto3,it,itmax,tt
integer::xst2,xen2,yst2,yen2,xst3,xen3,yst3,yen3,time,time2,time4,itime,t,tmx,tmn,ntmx,nntmx,trid,trid2,xst1,xen1,yst1,yen1,iit
integer::xst4, xen4, yst4, yen4
integer::day1,hour1,hmx,hmn,hhmn,hhmx,hmx2,ob,nob,nfs,undef
parameter(imx=105,jmx=65,hmx=37,syear=1979,eyear=2016,smonth=3,emonth=5,hmn=1,jmn=1,imn=1,tmn=1,hhmn=1,hhmx=40,hmx2=27,itmax=17,nntmx=20000)
parameter(xst=20, xen=20, yst=20, yen=20, xst2=25, xen2=25, yst2=25, yen2=25, xst3=35, xen3=35, yst3=35, yen3=35)
parameter(xst4=40, xen4=40, yst4=40, yen4=40)
real::lat1,lon1,jlat,jlon,pi,es,ssp,slpup,slplow,lon2,lat2,minslp,hoge
real,dimension(nntmx)::trid3,time3
real,dimension(imx,jmx,hmx)::ldiab,cdiab,vdiab
real,dimension(jmx)::co,lat
real,dimension(imx,jmx,hmx2)::spfh
real,dimension(imx,jmx,hhmx)::reh,rehu,rehv,ru,rv,uz,vz
real,dimension(-yst4:yen4,-yst4:yen4,hmx,itmax)::coldiab,cocdiab,covdiab
real,dimension(-yst4:yen4,itmax)::sco3,sco4,sco5
real,dimension(imx,jmx)::cape,sreh,ehi,mdvdz,mlapse,psurf,slp,tmp2m,dep,u10m,v10m,rh2m,sp2m,pt2m,cu,cv,suz,svz,shear_u,shear_v,mshear
real::p(hmx)=(/1000., 975., 950., 925., 900., 875., 850., 825., 800., 775., 750., 700., &
650., 600., 550., 500., 450., 400., 350., 300., 250., 225., 200., 175., 150., 125., 100., &
70., 50., 30., 20., 10.,7., 5., 3., 2., 1./)
real::ber,mdlon,mdlat,bermax
character::jetpos*8,nmem1*1,nmem*2,mmonth*2,year*4,SEA*3,region*3,CAT*2
logical::fast

integer::ii,jj,in,jn,irmax, jrmax
real,dimension(-yst4:yen4)::newy,y_ec,ypos
real,dimension(-xst4:xen4, -yst4:yen4)      :: newi, newj
real,dimension(-yst4:yen4,-yst4:yen4,hmx)::ffmag,front,fcold,fwarm
real,dimension(-yst4:yen4,-yst4:yen4,hmx)::ldiabxy,cdiabxy,vdiabxy
real,dimension(-yst4:yen4,-yst4:yen4,hmx)::ldiab_rot,cdiab_rot,vdiab_rot
real,dimension(-yst4:yen4)::newx
real,dimension(-xst4:xen4,-xst4:xen4)::x_ec,xpos
real::r,dx,dy,defx,defy,ai,aj,dphi
real::dist,dist2,defx2,defy2
real :: dlat, dlon, ang, wang, magwind
real :: hoge3
integer :: hoge1, hoge2

fast=.false. ! a little output variables


nmx=20000
ntmx=20000
pi=acos(-1.)
write(6,*) pi
xst1=10
xen1=10
yst1=10
yen1=10
!slpup=100500.
!slplow=99500.
slpup=110000.
slplow=50000.
ice1=0
jce1=0
out=0
nout=0
sco3=0.
sco4=0.
SEA='DJF'
region='JPN'
CAT='PO'

ob=1
nob=0


write(6,*) 'read namelist'
namelist/namdim/CAT,SEA,jetpos,region
write(6,*) 'category'
write(6,*) CAT,' ',SEA,' ',jetpos,' ',region
read(5,namdim)

r=6378.e3
dphi=1.25*pi/180.

dx=1.25e5
dy=1.25e5

undef=99999


do i=-40,40
 newx(i)=dx*real(i)
enddo

do j=-40,40
 newy(j)=dy*real(j)
enddo

do j=jmn,jmx
lat(j)=80.-1.25*(j-1)
 co(j)=cos(lat(j)/180.*pi)
enddo


open(57,file='compo_diabatic_rotate_DRATE05_100_'//CAT//'_'//SEA//'_'//jetpos//'_190E.data',status='unknown',form='unformatted')


do it = 1,itmax

 if(it.le.9) then
  write(nmem1,'(i1)') it
  nmem = '0'//nmem1
 else
  write(nmem,'(i2)') it
 endif

  open(60+it,file='./'//CAT//'/emem_'//nmem//'_diabatic_rotate_DRATE05_100_'//CAT//'_'//SEA//'_'//jetpos//'_190E.data',status='unknown',form='unformatted')

enddo
!open(57,file='compo_ec_'//CAT//'_'//SEA//'.data',status='unknown',form='unformatted')
!open(58,file='compo_ec_nout_'//SEA//'_50year_oriv1abz0f0_min992.data',status='unknown',form='unformatted')


do iyear=syear,eyear
 write(year,'(i4)') iyear


open(28,file='./EC_MAXDEV/'//year//''//SEA//''//CAT//'_MAXDEV_LONLAT'//region//'_vavelowpass5day_'//jetpos//'_190E_1000_700',status='unknown',form='formatted')
 do nt=1,ntmx
  !write(6,*) n
  !read(22,*,end=120) nto2,trid2,time2
  read(28,*,end=120) mdlon,mdlat,trid2,time2,bermax
 write(6,*) 'KT',iyear,trid2,time2,bermax
!if(trid.eq.trid2.and.minslp.ge.slplow.and.minslp.le.slpup) then 
minslp=100000.
if(minslp.ge.slplow.and.minslp.le.slpup.and.bermax.ge.0.5.and.bermax.le.10.0.and.trid2.ne.undef) then 

it=0
iit=-4
 do itime=time2-itmax/2,time2+itmax/2
write(year,'(i4)') iyear
write(6,*) 'itime is from',time2-itmax/2,'to',time2+itmax/2
write(6,*) 'keytime=',time2
!write(6,*) 'iyear=', iyear,'itime=',itime
if(SEA.eq.'MAM') then

if(itime.le.31*4) then
month1=3
mmonth='03'
tmx=31*4
if(mod(itime,4)==0) then
day1=itime/4
hour1=18
elseif(mod(itime,4)==1) then
day1=itime/4+1
hour1=0
elseif(mod(itime,4)==2) then
day1=itime/4+1
hour1=6
elseif(mod(itime,4)==3) then
day1=itime/4+1
hour1=12
endif

elseif(itime.le.31*4+30*4) then
month1=4
mmonth='04'
tmx=30*4
if(mod(itime,4)==0) then
day1=itime/4-31
hour1=18
elseif(mod(itime,4)==1) then
day1=itime/4+1-31
hour1=0
elseif(mod(itime,4)==2) then
day1=itime/4+1-31
hour1=6
elseif(mod(itime,4)==3) then
day1=itime/4+1-31
hour1=12
endif



write(6,*) 'pass',iyear,trid,minslp,time2


else
month1=5
mmonth='05'
tmx=31*4
if(mod(itime,4)==0) then
day1=itime/4-31-30
hour1=18
elseif(mod(itime,4)==1) then
day1=itime/4+1-31-30
hour1=0
elseif(mod(itime,4)==2) then
day1=itime/4+1-31-30
hour1=6
elseif(mod(itime,4)==3) then
day1=itime/4+1-31-30
hour1=12
endif

endif

endif
!write(6,*) month1




if(SEA.eq.'SON') then

if(itime.le.30*4) then
month1=9
mmonth='09'
tmx=30*4
if(mod(itime,4)==0) then
day1=itime/4
hour1=18
elseif(mod(itime,4)==1) then
day1=itime/4+1
hour1=0
elseif(mod(itime,4)==2) then
day1=itime/4+1
hour1=6
elseif(mod(itime,4)==3) then
day1=itime/4+1
hour1=12
endif

elseif(itime.le.31*4+30*4) then
month1=10
mmonth='10'
tmx=31*4
if(mod(itime,4)==0) then
day1=itime/4-30
hour1=18
elseif(mod(itime,4)==1) then
day1=itime/4+1-30
hour1=0
elseif(mod(itime,4)==2) then
day1=itime/4+1-30
hour1=6
elseif(mod(itime,4)==3) then
day1=itime/4+1-30
hour1=12
endif






write(6,*) 'pass',iyear,trid,minslp,time2


else
month1=11
mmonth='11'
tmx=30*4
if(mod(itime,4)==0) then
day1=itime/4-31-30
hour1=18
elseif(mod(itime,4)==1) then
day1=itime/4+1-31-30
hour1=0
elseif(mod(itime,4)==2) then
day1=itime/4+1-31-30
hour1=6
elseif(mod(itime,4)==3) then
day1=itime/4+1-31-30
hour1=12
endif

endif

endif
!write(6,*) month1

if(SEA.eq.'DJF') then

if(itime.le.31*4) then
month1=0
mmonth='12'
tmx=31*4
if(mod(itime,4)==0) then
day1=itime/4
hour1=18
elseif(mod(itime,4)==1) then
day1=itime/4+1
hour1=0
elseif(mod(itime,4)==2) then
day1=itime/4+1
hour1=6
elseif(mod(itime,4)==3) then
day1=itime/4+1
hour1=12
endif

elseif(itime.le.31*4+31*4) then
month1=1
mmonth='01'
tmx=31*4
if(mod(itime,4)==0) then
day1=itime/4-31
hour1=18
elseif(mod(itime,4)==1) then
day1=itime/4+1-31
hour1=0
elseif(mod(itime,4)==2) then
day1=itime/4+1-31
hour1=6
elseif(mod(itime,4)==3) then
day1=itime/4+1-31
hour1=12
endif

else
month1=2
mmonth='02'

if(mod(iyear,4).eq.0) then
tmx=29*4
else
tmx=28*4
endif

if(mod(itime,4)==0) then
day1=itime/4-31-31
hour1=18
elseif(mod(itime,4)==1) then
day1=itime/4+1-31-31
hour1=0
elseif(mod(itime,4)==2) then
day1=itime/4+1-31-31
hour1=6
elseif(mod(itime,4)==3) then
day1=itime/4+1-31-31
hour1=12
endif
endif

endif

!write(6,*) 'month=', mmonth,'  ',year
iit=iit+1
!write(6,*) month1

open(26,file='./EC_TRACK/'//year//''//SEA//''//CAT//'_track_EC_'//region//'_190E',status='unknown',form='formatted')
open(27,file='./EC_TRACK/'//year//''//SEA//''//CAT//'_movement_EC_'//region//'_190E',status='unknown',form='formatted')

 do nnt=1,nntmx

    read(26,*,end=130) lon1,lat1,time3(nnt),trid3(nnt),ber
    read(27,*) dlon,dlat,hoge1,hoge2,hoge3
    !    write(6,*) 'read track',lon1,lat1,time3(nnt),trid3(nnt),ber
!if(trid2.ne.trid3(nnt).and.time3(nnt).eq.itime.and.month1.ne.3.) then
!it=it+1
!endif
!write(6,*) 'trid2=',trid2,'trid3=',trid3(nnt)
if(trid2.eq.trid3(nnt)) then

!write(6,*) 'pass2',trid2

if(itmax.ge.2) then
do tt=1,itmax/2
!do tt=1,8
if(time3(nnt).eq.time2-itmax/2+tt.and.it.eq.0) then
 it=it+tt
endif
enddo
endif



if(month1.eq.0) then
write(year,'(i4)') iyear-1
else
write(year,'(i4)') iyear
endif

open(50,file='/home/e_tochi/data2/JRA55_DIAB/NEW_LRGHR1000_0_JRA55_'//region//'.'//year//'.'//mmonth//'.data',status='old',form='unformatted',access='sequential', action='READ')
open(51,file='/home/e_tochi/data2/JRA55_DIAB/NEW_CNVHR1000_0_JRA55_'//region//'.'//year//'.'//mmonth//'.data',status='old',form='unformatted',access='sequential', action='READ')
open(53,file='/home/e_tochi/data2/JRA55_DIAB/NEW_VDFHR1000_0_JRA55_'//region//'.'//year//'.'//mmonth//'.data',status='old',form='unformatted',access='sequential', action='READ')


do t=tmn,tmx

if(month1.eq.0) then
time4=t
endif

if(month1.eq.1) then
time4=t+31*4
endif

if(month1.eq.2) then
time4=t+31*4+31*4
endif

if(month1.eq.3) then
time4=t
endif

if(month1.eq.4) then
time4=t+31*4
endif

if(month1.eq.5) then
time4=t+30*4+31*4
endif

if(month1.eq.9) then
time4=t
endif

if(month1.eq.10) then
time4=t+30*4
endif

if(month1.eq.11) then
time4=t+30*4+31*4
endif


do h=hmn,hmx
 read(50) ((ldiab(i,j,h),i=imn,imx),j=jmn,jmx)
enddo

do h=hmn,hmx
 read(51) ((cdiab(i,j,h),i=imn,imx),j=jmn,jmx)
enddo

do h=hmn,hmx
 read(53) ((vdiab(i,j,h),i=imn,imx),j=jmn,jmx)
enddo


do i=imn,imx
 do j=jmn,jmx

 jlat=80.-1.25*real(j-1)
 if(region=='USA') then
 jlon=220.+1.25*real(i-1)
 else
 jlon=80.+1.25*real(i-1)
 endif

 if(lat1.ge.jlat-1.25*0.5.and.lat1.le.jlat+1.25*0.5.and.lon1.ge.jlon-1.25*0.5.and.lon1.le.jlon+1.25*0.5) then
  ice1=i
  jce1=j
  lon2=jlon
  lat2=jlat
 endif

 enddo
enddo
!write(6,*) 'itime==time4?', itime,time4,time3(nnt)
    if(itime.eq.time4.and.itime.eq.time3(nnt).and.minslp.ge.slplow.and.minslp.le.slpup) then
 
 it=it+1
!write(6,*) 'it2= ',it
if(it.eq.itmax/2+1) then
out=out+1
endif


!低気圧中心に相対的なxyを計算
do j=-yst2,yen2
 do i=-xst2,xen2
  x_ec(i,j) = r*co(j+jce1)*dphi*i
  y_ec(j) = r*dphi*j
 enddo
enddo


call ll2xy(ldiab, ldiabxy, ice1, jce1, imx, jmx, hmx, x_ec, y_ec, dx, dy, xst2, xen2, yst2, yen2, xst3, xen3, yst3, yen3, newx, newy)
call ll2xy(cdiab, cdiabxy, ice1, jce1, imx, jmx, hmx, x_ec, y_ec, dx, dy, xst2, xen2, yst2, yen2, xst3, xen3, yst3, yen3, newx, newy)
call ll2xy(vdiab, vdiabxy, ice1, jce1, imx, jmx, hmx, x_ec, y_ec, dx, dy, xst2, xen2, yst2, yen2, xst3, xen3, yst3, yen3, newx, newy)


! rotate

ang=atan2(dlat,dlon)


do i = -xst4,xen4
  do j = -yst4,yen4
   newi(i,j) = real(i) * cos(ang) + real(j) * sin(ang)
   newj(i,j) = real(j) * cos(ang) - real(i) * sin(ang)
  enddo
 enddo

irmax = 0
jrmax = 0

do h = hmn, hmx
 call interp2d(ldiabxy(:, :, h), ldiab_rot(:, :, h), newi, newj, xen2, yen2, irmax, jrmax, xen4, yen4)
enddo

do h = hmn, hmx
 call interp2d(cdiabxy(:, :, h), cdiab_rot(:, :, h), newi, newj, xen2, yen2, irmax, jrmax, xen4, yen4)
enddo

do h = hmn, hmx
 call interp2d(vdiabxy(:, :, h), vdiab_rot(:, :, h), newi, newj, xen2, yen2, irmax, jrmax, xen4, yen4)
enddo

! write each smaple

do h=1,hmx
 write(60+it) ((ldiab_rot(i,j,h),i=-xst2,xen2),j=-yst2,yen2)
enddo

do h=1,hmx
 write(60+it) ((cdiab_rot(i,j,h),i=-xst2,xen2),j=-yst2,yen2)
enddo

do h=1,hmx
 write(60+it) ((vdiab_rot(i,j,h),i=-xst2,xen2),j=-yst2,yen2)
enddo


 write(6,*) it,itime,time3(nnt), iyear,'out'
 write(6,*) lat2,lon2,ice1,jce1

 !write(6,*) it
 do h=hmn,hmx
   do i=-xst2,xen2
    do j=-yst2,yen2

     coldiab(i,j,h,it)=coldiab(i,j,h,it)+ldiab_rot(i,j,h)
     cocdiab(i,j,h,it)=cocdiab(i,j,h,it)+cdiab_rot(i,j,h)
     covdiab(i,j,h,it)=covdiab(i,j,h,it)+vdiab_rot(i,j,h)

    enddo
   enddo
  enddo


 do j=-yst2,yen2
  sco3(j,it)=sco3(j,it)+1
  !write(6,*) it,sco3(j,it),co(j+jce1)
 enddo

    endif ! pick up EC

  

enddo !time loop

endif



close(50)
close(51)
close(52)
close(53)
close(54)

  enddo !nntloop
130 continue
close(26)
close(27)
!write(6,*) n,nt,nnt,iyear
 enddo ! itimeloop
endif

!write(6,*) n,nt
 enddo  !ntloop
  
120 continue
!write(6,*) n,nt
close(28)

enddo !year loop

 do it=1,itmax
  do h=hmn,hmx
   do i=-xst2,xen2
    do j=-yst2,yen2

    coldiab(i,j,h,it)=coldiab(i,j,h,it)/sco3(j,it)
    cocdiab(i,j,h,it)=cocdiab(i,j,h,it)/sco3(j,it)
    covdiab(i,j,h,it)=covdiab(i,j,h,it)/sco3(j,it)

 
   enddo
  enddo
 enddo


enddo

do it=1,itmax



do h=hmn,hmx
 write(57) ((coldiab(i,j,h,it),i=-xst2,xen2),j=-yst2,yen2)
enddo

do h=hmn,hmx
 write(57) ((cocdiab(i,j,h,it),i=-xst2,xen2),j=-yst2,yen2)
enddo

do h=hmn,hmx
 write(57) ((covdiab(i,j,h,it),i=-xst2,xen2),j=-yst2,yen2)
enddo



enddo
!
write(6,*) 'out=',out,'nout=',nout
endprogram compo_ec

subroutine interp2d(var, varrot, newi, newj, ilag, jlag, irmax, jrmax, ilag2, jlag2)

integer             :: ii, jj
integer, intent(in) :: ilag, jlag, ilag2, jlag2, irmax, jrmax
real, intent(in),  dimension(-ilag2 : ilag2, -jlag2 : jlag2)    :: newi, newj
real, intent(in),  dimension(-ilag2 : ilag2, -jlag2 : jlag2)    :: var
real, intent(out), dimension(-ilag2 : ilag2, -jlag2 : jlag2)    :: varrot
real              :: ai, aj

            do j = - jlag, jlag
              do i = - ilag, ilag
                ii = nint(newi(i, j))
                jj = nint(newj(i, j))
                if(ii .ge. -ilag + 1 .and. jj .ge. -jlag + 1 .and. ii .le. ilag - 1 .and. jj .le. jlag - 1) then
                  ai = newi(i, j) - real(ii)
                  aj = newj(i, j) - real(jj)
                   varrot(i, j) = ai * aj * var(ii + irmax + 1, jj + jrmax + 1) &
                           + (1. - ai)*aj*var(ii + irmax, jj + jrmax + 1) + &
                           ai * (1 - aj) * var(ii + irmax + 1, jj + jrmax) + &
                           (1. - ai) * (1. - aj)*var(ii + irmax, jj + jrmax)
                 endif
              enddo
             enddo


end subroutine interp2d



subroutine ll2xy(varll, varxy, ice1, jce1, im, jm, hm, xec, yec, dx, dy, xst2,xen2, yst2, yen2, xst3, xen3, yst3, yen3, newx, newy)
implicit none
integer::in, jn, i, j, h, ii, jj
integer,intent(in)::ice1, jce1, im, jm, hm,xst2, xen2, yst2, yen2, xst3, xen3,yst3, yen3
real,intent(in)::dx, dy
real:: ai, aj
real,intent(in),dimension(-40:40,-40:40)::xec
real,intent(in),dimension(-40:40)::yec
real,intent(in),dimension(-40:40)::newx, newy
real,intent(in),dimension(im,jm,hm)::varll
real,intent(out),dimension(-40:40,-40:40,hm)::varxy
real,dimension(-40:40,-40:40)::xpos
real,dimension(-40:40)::ypos

!write(6,*) 'll2xy',dx,dy,hm
 do h=1,hm
  do j = -yst3,yen3
   do i = -xst3,xen3
    do jj = -yst2,yen2
     do ii = -xst2,xen2
    if(xec(ii,jj).le.newx(i).and.xec(ii+1,jj).ge.newx(i).and.yec(jj).le.newy(j).and.yec(jj+1).ge.newy(j)) then
      xpos(ii,jj) = xec(ii,jj)/dx
      ypos(jj) = yec(jj)/dy
      in = int(xpos(ii,jj))
      jn = int(ypos(jj))
      ai = abs((xec(ii,jj)-newx(i))/(xec(ii+1,jj)-xec(ii,jj)))
      aj = abs((yec(jj)-newy(j))/(yec(jj+1)-yec(jj)))
       !write(6,*) ypos(jj),j,ai,aj
       if(ii+ice1.ge.1.and.jj+jce1.ge.1.and.ii+1+ice1.le.im.and.jj+1+jce1.le.jm) then
         varxy(i,j,h) = ai*aj*varll(ii+1+ice1,jj+1+jce1,h)+(1.-ai)*aj*varll(ii+ice1,jj+1+jce1,h)+ai*(1-aj)*varll(ii+1+ice1,jj+jce1,h)+(1.-ai)*(1.-aj)*varll(ii+ice1,jj+jce1,h)
       endif
     endif
     enddo
    enddo
   enddo
  enddo
 enddo

end subroutine ll2xy