!-------------------------------------------------------------------------------

!                      Code_Saturne version 5.0.7
!                      --------------------------
! This file is part of Code_Saturne, a general-purpose CFD tool.
!
! Copyright (C) 1998-2018 EDF S.A.
!
! This program is free software; you can redistribute it and/or modify it under
! the terms of the GNU General Public License as published by the Free Software
! Foundation; either version 2 of the License, or (at your option) any later
! version.
!
! This program is distributed in the hope that it will be useful, but WITHOUT
! ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
! FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
! details.
!
! You should have received a copy of the GNU General Public License along with
! this program; if not, write to the Free Software Foundation, Inc., 51 Franklin
! Street, Fifth Floor, Boston, MA 02110-1301, USA.

!-------------------------------------------------------------------------------

!===============================================================================
! Purpose:
! -------

!> \file cs_user_extra_operations.f90
!>
!> \brief This function is called at the end of each time step, and has a very
!>  general purpose
!>  (i.e. anything that does not have another dedicated user subroutine)
!>
!> See \subpage cs_user_extra_operations_examples and
!> \subpage cs_user_extra_operations-nusselt_calculation for examples.
!>
!-------------------------------------------------------------------------------

!-------------------------------------------------------------------------------
! Arguments
!______________________________________________________________________________.
!  mode           name          role                                           !
!______________________________________________________________________________!
!> \param[in]     nvar          total number of variables
!> \param[in]     nscal         total number of scalars
!> \param[in]     dt            time step (per cell)
!_______________________________________________________________________________

subroutine cs_f_user_extra_operations &
 ( nvar   , nscal  ,                                              &
   dt     )

!===============================================================================

!===============================================================================
! Module files
!===============================================================================

use paramx
use dimens, only: ndimfb
use pointe
use numvar
use optcal
use cstphy
use cstnum
use entsor
use lagran
use parall
use period
use ppppar
use ppthch
use ppincl
use mesh
use field
use field_operator ! Give access to cell gradient
use turbomachinery
use cs_c_bindings

!===============================================================================

implicit none

! Arguments

integer          nvar   , nscal

double precision dt(ncelet)

! Local variables

! INSERT_VARIABLE_DEFINITIONS_HERE

integer, allocatable, dimension(:) :: lstelt


!integer impout

integer iel
integer use_previous_t, imrgra, inc, recompute_cocg ! For gradient calculation

double precision, pointer, dimension(:) :: spg

double precision, allocatable, dimension(:,:) :: grad_p

!===============================================================================


!===============================================================================
! Initialization
!===============================================================================

allocate(grad_p(3,ncelet)) ! Allocate array for pressure gradient

!===============================================================================
! Assign boundary conditions to boundary faces here

! For each subset:
! - use selection criteria to filter boundary faces of a given subset
! - loop on faces from a subset
!   - set the boundary condition for each face
!===============================================================================

!-------------------------------------------------------------------------------

!	Calculate the pressure gradient
	!> \brief  Compute cell gradient of scalar field or component of vector or
	!>         tensor field.
  !> \subroutine  field_gradient_scalar (f_id, use_previous_t, 
  !                                      imrgra, inc, recompute_cocg, grad)
	!> \param[in]   f_id             field id
	!> \param[in]   use_previous_t   1 if values at previous time step should
	!>                               be used, 0 otherwise
	!> \param[in]   imrgra           gradient computation mode: 
	!									0 iterative gradient; 
	!									1 least square gradient
	!> \param[in]   inc              0: increment; 1: do not increment
	!> \param[in]   recompute_cocg   1 or 0: recompute COCG or not
	!> \param[out]  grad             gradient
  
  use_previous_t = 1 ! Values at previous time step are not used
  imrgra = 1 ! Least square gradient
  inc = 1 ! Do not increment
  recompute_cocg = 0 ! Do not recompute COCG
  
  call field_gradient_scalar(ivarfl(ipr), use_previous_t,            &
                             imrgra, inc, recompute_cocg, grad_p)
!-------------------------------------------------------------------------------

call field_get_val_s_by_name("SPG", spg)

do iel = 1, ncelet

      spg(iel) = sqrt(grad_p(1,iel)**2.d0                    &
                     +grad_p(2,iel)**2.d0                    &
                     +grad_p(3,iel)**2.d0)
enddo

!-------------------------------------------------------------------------------


!--------
! Formats
!--------

!----
! End
!----

deallocate(grad_p)  ! Deallocate array for pressure gradient


return
end subroutine cs_f_user_extra_operations
