MODULE post
     USE Compact
     USE m_parameters
     USE m_calculate
     USE m_arrays
     USE m_terms

     IMPLICIT NONE
     PRIVATE
     INTEGER :: countnum
     INTEGER :: num_, dummyu_   ! hybrid
     REAL :: tnow

     REAL, DIMENSION(:,:,:,:), ALLOCATABLE :: old_scalar, new_scalar

     PUBLIC :: main
    CONTAINS
 
     SUBROUTINE main
      INTEGER :: fread,i

      CALL READ_INTRO

      CALL ALLOCATE_ARRAYS

      if (iammaster) CALL PRINT_BANNER

      countnum=0


      firstloop: DO fread=startnum,endnum,skipnum

        IF ( to_omit(fread) ) THEN

            WRITE(*,'(a47,i7,a4,i5,a3,i5)') &
                 ' Current fullsavenum = ', fread, ' || ', (fread-startnum+1), ' / ', (endnum-startnum+1)

            WRITE(*,'(a12,i6,a20,i6)') &
                 '      Skip. ', omit_t(i,1), ' <= fullsavenum <= ', omit_t(i,2)

        ELSE
         countnum=countnum+1

         IF (file_dist(fread).eq.myid) THEN

          CALL READ_FILE(fread)

          CALL m_terms_calculate_pass1
         
         END IF

        ENDIF
          
      ENDDO firstloop

      write(*,*) '1st loop finished'


      CALL m_terms_average_pass1(countnum)


      secondloop: DO fread=startnum,endnum,skipnum

       IF ( to_omit(fread) ) THEN

         WRITE(*,'(a40,i7,a4,i4,a3,i4)') &
              ' Current fullsavenum = ', fread, ' || ', (fread-startnum+1), ' / ', (endnum-startnum+1)

         WRITE(*,'(a12,i6,a20,i6)') &
              '     Skip. ', omit_t(i,1), ' <= fullsavenum <= ', omit_t(i,2)

       ELSE

         IF (file_dist(fread).eq.myid) THEN

           CALL READ_FILE(fread)

           CALL m_terms_calculate_pass2

         ENDIF

       ENDIF

      ENDDO secondloop

      CALL m_terms_average_pass2(countnum)
      
      if (iammaster) CALL m_terms_write_result

      CALL DEALLOCATES_CLOSE


      WRITE(*,*) ' Avergaing RAW data is FINISHED'
      WRITE(*,*)  'qEdge_X.dat is generated' 

     END SUBROUTINE main

!========================================================================================
! End of main routine
!========================================================================================

     SUBROUTINE PRINT_BANNER
      WRITE(*,*) ' build date: ', build_date
      WRITE(*,*) ' build base: ', build_base
      WRITE(*,*) ' terms to process '
      WRITE(*,*) term_info

      WRITE(*,*) ' This program, x-edge-cold-bc-5-hybrid, is written by D. Kim, 2018'
      WRITE(*,*) ' It is to study the statistics of the flame parameters at the leading edge'
      WRITE(*,*) ' in turbulent premixed flames.'
      WRITE(*,'(a40,i5,a11,i5,a1)') '  Postprocess will be done from "FORT.',startnum,'" to "FORT.',endnum,'"'
     END SUBROUTINE PRINT_BANNER


     SUBROUTINE READ_FILE(num)
      INTEGER, INTENT(IN) :: num
      REAL, DIMENSION(2) :: tmpr
      INTEGER :: nx, ny, nz
      REAL :: tmp1,tmp2
      REAL :: dt,dummyu
      INTEGER :: ncyc

      OPEN(num,FORM='unformatted',STATUS='unknown')

      READ (num) tnow,nx,ny,nz,tmp1,tmp2

      IF ((nx .ne. nxp) .or. (ny .ne. nyp) .or. (nz .ne. nzp)) THEN
        WRITE(0,*) "Array dimension mismatch", nx, ny, nz, " != ", nxp, nyp, nzp
        STOP -1
      ENDIF

      READ (num) ncyc,dt,dummyu
      READ (num) tmpr(1:2)
      READ (num) tmpr(1:2)
      READ (num) tmpr(1:2)

      WRITE(*,'(a4,i3,a40,f8.3,a2,i7,a2,i5,a4,i5,a3,i5)') ' p ', myid, ': time / NCYC / fullsavenum = ',tnow,&
                ' /',ncyc,' /',num,' || ',(num-startnum+1),' / ',(endnum-startnum+1)
      WRITE(*,*) '    Reading current data file and processing'

      num_=num

      IF(num.le.shiftnum) THEN
        WRITE(*,*) '    with an old fort data from Nueman-0X'
        READ (num) u,v,w,old_scalar
        y(:,:,:) = old_scalar(2,:,:,:)
      ELSE
        WRITE(*,*) '    with a new fort data from Comb-Cluster'
        READ (num) u,v,w,new_scalar
        u = u + dummyu
        y = new_scalar(:,:,:,2)
      ENDIF

      CLOSE (num)

     END SUBROUTINE READ_FILE


     SUBROUTINE ALLOCATE_ARRAYS

      INTEGER :: ierr

       CALL m_arrays_init
       CALL m_calculate_init
       CALL m_terms_init

       ALLOCATE(old_scalar(2,nxp,nyp,nzp),STAT=ierr)  ; old_scalar=0.          ! Main variables
       ALLOCATE(new_scalar(nxp,nyp,nzp,2),STAT=ierr)  ; new_scalar=0.

       WRITE(*,'(a6,i3,a8,i3,a8,i3)') '  NX = ',nxp,' / NY = ',nyp,' / NZ = ',nzp
       WRITE(*,*) ' Preparing memory space for COMPACT SCHEME'
       ! CALL ludcmp(nxp,nyp,nzp,1,0,0)      ! 1,1,0

       WRITE(*,'(a22,i3,a3,i3,a4,i3)') '  Grid number range : ',syp,' ~ ',eyp,' of ',nyp
       WRITE(*,*)

     END SUBROUTINE ALLOCATE_ARRAYS


     SUBROUTINE DEALLOCATES_CLOSE

      CALL m_arrays_finalize
      CALL m_calculate_finalize
      CALL m_terms_finalize

      DEALLOCATE(old_scalar)
      DEALLOCATE(new_scalar)
      
      IF(omitnum.gt.0)   DEALLOCATE(omit_t)

      DEALLOCATE(file_dist)

     END SUBROUTINE DEALLOCATES_CLOSE

    END MODULE post