!-------------------------------------------------------------------------------------------------- ! $Id: DAMASK_spectral_SolverAL.f90 1654 2012-08-03 09:25:48Z MPIE\m.diehl $ !-------------------------------------------------------------------------------------------------- !> @author Pratheek Shanthraj, Max-Planck-Institut für Eisenforschung GmbH !> @author Martin Diehl, Max-Planck-Institut für Eisenforschung GmbH !> @author Philip Eisenlohr, Max-Planck-Institut für Eisenforschung GmbH !> @brief AL scheme solver !-------------------------------------------------------------------------------------------------- module DAMASK_spectral_SolverAL use prec, only: & pInt, & pReal use math, only: & math_I3 use DAMASK_spectral_Utilities, only: & tSolutionState implicit none #include #include #include character (len=*), parameter, public :: & DAMASK_spectral_SolverAL_label = 'al' !-------------------------------------------------------------------------------------------------- ! derived types type tSolutionParams real(pReal), dimension(3,3) :: P_BC, rotation_BC real(pReal) :: timeinc end type tSolutionParams type(tSolutionParams), private :: params !-------------------------------------------------------------------------------------------------- ! PETSc data DM, private :: da SNES, private :: snes Vec, private :: solution_vec !-------------------------------------------------------------------------------------------------- ! common pointwise data real(pReal), private, dimension(:,:,:,:,:), allocatable :: F_lastInc, F_lambda_lastInc, F_lambdaDot, Fdot real(pReal), private, dimension(:,:,:,:), allocatable :: coordinates real(pReal), private, dimension(:,:,:), allocatable :: temperature !-------------------------------------------------------------------------------------------------- ! stress, stiffness and compliance average etc. real(pReal), private, dimension(3,3) :: & F_aimDot, & F_aim = math_I3, & F_aim_lastInc = math_I3, & P_av character(len=1024), private :: incInfo real(pReal), private, dimension(3,3,3,3) :: & C = 0.0_pReal, C_lastInc = 0.0_pReal, & S = 0.0_pReal, & C_scale = 0.0_pReal, & S_scale = 0.0_pReal real(pReal), private :: err_stress, err_f, err_p logical, private :: ForwardData real(pReal), private, dimension(3,3) :: mask_stress = 0.0_pReal integer(pInt) :: reportIter = 0_pInt contains !-------------------------------------------------------------------------------------------------- !> @brief allocates all neccessary fields and fills them with data, potentially from restart info !-------------------------------------------------------------------------------------------------- subroutine AL_init() use, intrinsic :: iso_fortran_env ! to get compiler_version and compiler_options (at least for gfortran >4.6 at the moment) use IO, only: & IO_read_JobBinaryFile, & IO_write_JobBinaryFile use FEsolving, only: & restartInc use DAMASK_interface, only: & getSolverJobName use DAMASK_spectral_Utilities, only: & Utilities_init, & Utilities_constitutiveResponse, & Utilities_updateGamma, & debugRestart use numerics, only: & petsc_options use mesh, only: & res, & geomdim, & mesh_NcpElems use math, only: & math_invSym3333 implicit none #include #include integer(pInt) :: i,j,k real(pReal), dimension(:,:,:,:,:), allocatable :: P PetscErrorCode :: ierr PetscObject :: dummy PetscScalar, pointer, dimension(:,:,:,:) :: xx_psc, F, F_lambda call Utilities_init() write(6,'(/,a)') ' <<<+- DAMASK_spectral_solverAL init -+>>>' write(6,'(a)') ' $Id: DAMASK_spectral_SolverAL.f90 1654 2012-08-03 09:25:48Z MPIE\m.diehl $' #include "compilation_info.f90" write(6,'(a)') '' allocate (F_lastInc (3,3, res(1), res(2),res(3)), source = 0.0_pReal) allocate (Fdot (3,3, res(1), res(2),res(3)), source = 0.0_pReal) ! allocate (Fdot,source = F_lastInc) somethin like that should be possible allocate (F_lambda_lastInc(3,3, res(1), res(2),res(3)), source = 0.0_pReal) allocate (F_lambdaDot(3,3, res(1), res(2),res(3)), source = 0.0_pReal) allocate (P (3,3, res(1), res(2),res(3)), source = 0.0_pReal) allocate (coordinates( res(1), res(2),res(3),3), source = 0.0_pReal) allocate (temperature( res(1), res(2),res(3)), source = 0.0_pReal) !-------------------------------------------------------------------------------------------------- ! PETSc Init call SNESCreate(PETSC_COMM_WORLD,snes,ierr) call DMDACreate3d(PETSC_COMM_WORLD, & DMDA_BOUNDARY_NONE, DMDA_BOUNDARY_NONE, DMDA_BOUNDARY_NONE, & DMDA_STENCIL_BOX,res(1),res(2),res(3),PETSC_DECIDE,PETSC_DECIDE,PETSC_DECIDE, & 18,1,PETSC_NULL_INTEGER,PETSC_NULL_INTEGER,PETSC_NULL_INTEGER,da,ierr) call DMCreateGlobalVector(da,solution_vec,ierr) call DMDASetLocalFunction(da,AL_formResidual,ierr) call SNESSetDM(snes,da,ierr) call SNESSetConvergenceTest(snes,AL_converged,dummy,PETSC_NULL_FUNCTION,ierr) call SNESSetFromOptions(snes,ierr) !-------------------------------------------------------------------------------------------------- ! init fields call DMDAVecGetArrayF90(da,solution_vec,xx_psc,ierr) F => xx_psc(0:8,:,:,:) F_lambda => xx_psc(9:17,:,:,:) if (restartInc == 1_pInt) then ! no deformation (no restart) F_lastInc = spread(spread(spread(math_I3,3,res(1)),4,res(2)),5,res(3)) ! initialize to identity F_lambda_lastInc = F_lastInc F = reshape(F_lastInc,[9,res(1),res(2),res(3)]) F_lambda = F do k = 1_pInt, res(3); do j = 1_pInt, res(2); do i = 1_pInt, res(1) coordinates(i,j,k,1:3) = geomdim/real(res,pReal)*real([i,j,k],pReal) & - geomdim/real(2_pInt*res,pReal) enddo; enddo; enddo elseif (restartInc > 1_pInt) then ! using old values from file if (debugRestart) write(6,'(a,i6,a)') 'Reading values of increment ',& restartInc - 1_pInt,' from file' call IO_read_jobBinaryFile(777,'convergedSpectralDefgrad',& trim(getSolverJobName()),size(F_lastInc)) read (777,rec=1) F close (777) call IO_read_jobBinaryFile(777,'convergedSpectralDefgrad_lastInc',& trim(getSolverJobName()),size(F_lastInc)) read (777,rec=1) F_lastInc close (777) call IO_read_jobBinaryFile(777,'convergedSpectralDefgradLambda',& trim(getSolverJobName()),size(F_lambda_lastInc)) read (777,rec=1) F_lambda close (777) call IO_read_jobBinaryFile(777,'convergedSpectralDefgradLambda_lastInc',& trim(getSolverJobName()),size(F_lambda_lastInc)) read (777,rec=1) F_lambda_lastInc close (777) call IO_read_jobBinaryFile(777,'F_aim',trim(getSolverJobName()),size(F_aim)) read (777,rec=1) F_aim close (777) call IO_read_jobBinaryFile(777,'F_aim_lastInc',trim(getSolverJobName()),size(F_aim_lastInc)) read (777,rec=1) F_aim_lastInc close (777) coordinates = 0.0 ! change it later!!! endif !print*, shape(F) call Utilities_constitutiveResponse(coordinates,F,F,temperature,0.0_pReal,P,C,P_av,.false.,math_I3) call DMDAVecRestoreArrayF90(da,solution_vec,xx_psc,ierr) !-------------------------------------------------------------------------------------------------- ! reference stiffness if (restartInc == 1_pInt) then call IO_write_jobBinaryFile(777,'C_ref',size(C)) write (777,rec=1) C close(777) elseif (restartInc > 1_pInt) then call IO_read_jobBinaryFile(777,'C_ref',trim(getSolverJobName()),size(C)) read (777,rec=1) C close (777) endif call Utilities_updateGamma(C,.True.) C_scale = C S_scale = math_invSym3333(C) end subroutine AL_init !-------------------------------------------------------------------------------------------------- !> @brief solution for the AL scheme with internal iterations !-------------------------------------------------------------------------------------------------- type(tSolutionState) function & AL_solution(incInfoIn,guessmode,timeinc,timeinc_old,P_BC,F_BC,temperature_bc,rotation_BC) use numerics, only: & update_gamma use math, only: & math_mul33x33 ,& math_rotate_backward33 use mesh, only: & res,& geomdim,& deformed_fft use IO, only: & IO_write_JobBinaryFile use DAMASK_spectral_Utilities, only: & tBoundaryCondition, & Utilities_forwardField, & Utilities_calculateRate, & Utilities_maskedCompliance, & Utilities_updateGamma, & cutBack use FEsolving, only: & restartWrite, & terminallyIll implicit none #include #include !-------------------------------------------------------------------------------------------------- ! input data for solution real(pReal), intent(in) :: timeinc, timeinc_old, temperature_bc, guessmode type(tBoundaryCondition), intent(in) :: P_BC,F_BC character(len=*), intent(in) :: incInfoIn real(pReal), dimension(3,3), intent(in) :: rotation_BC real(pReal), dimension(3,3) ,save :: F_aimDot real(pReal), dimension(3,3) :: F_aim_lab !-------------------------------------------------------------------------------------------------- ! loop variables, convergence etc. real(pReal), dimension(3,3) :: temp33_Real !-------------------------------------------------------------------------------------------------- ! PetscScalar, dimension(:,:,:,:), pointer :: xx_psc, F, F_lambda PetscErrorCode :: ierr SNESConvergedReason ::reason !-------------------------------------------------------------------------------------------------- ! restart information for spectral solver incInfo = incInfoIn if (restartWrite) then write(6,'(a)') 'writing converged results for restart' call IO_write_jobBinaryFile(777,'convergedSpectralDefgrad',size(F_lastInc)) write (777,rec=1) F_LastInc close (777) call IO_write_jobBinaryFile(777,'C',size(C)) write (777,rec=1) C close(777) endif AL_solution%converged =.false. call DMDAVecGetArrayF90(da,solution_vec,xx_psc,ierr) F => xx_psc(0:8,:,:,:) F_lambda => xx_psc(9:17,:,:,:) if ( cutBack) then F_aim = F_aim_lastInc F_lambda= reshape(F_lambda_lastInc,[9,res(1),res(2),res(3)]) F = reshape(F_lastInc,[9,res(1),res(2),res(3)]) C = C_lastInc else !-------------------------------------------------------------------------------------------------- C_lastInc = C !-------------------------------------------------------------------------------------------------- ! calculate rate for aim if (F_BC%myType=='l') then ! calculate f_aimDot from given L and current F f_aimDot = F_BC%maskFloat * math_mul33x33(F_BC%values, F_aim) elseif(F_BC%myType=='fdot') then ! f_aimDot is prescribed f_aimDot = F_BC%maskFloat * F_BC%values endif f_aimDot = f_aimDot & + guessmode * P_BC%maskFloat * (F_aim - F_aim_lastInc)/timeinc_old F_aim_lastInc = F_aim !-------------------------------------------------------------------------------------------------- ! update coordinates and rate and forward last inc Fdot = Utilities_calculateRate(math_rotate_backward33(f_aimDot,rotation_BC), & timeinc,timeinc_old,guessmode,F_lastInc,reshape(F,[3,3,res(1),res(2),res(3)])) F_lambdaDot = Utilities_calculateRate(math_rotate_backward33(f_aimDot,rotation_BC), & timeinc,timeinc_old,guessmode,F_lambda_lastInc,reshape(F_lambda,[3,3,res(1),res(2),res(3)])) F_lastInc = reshape(F,[3,3,res(1),res(2),res(3)]) F_lambda_lastInc = reshape(F_lambda,[3,3,res(1),res(2),res(3)]) call deformed_fft(res,geomdim,math_rotate_backward33(F_aim_lastInc,rotation_BC), & 1.0_pReal,F_lastInc,coordinates) endif F_aim = F_aim + f_aimDot * timeinc !-------------------------------------------------------------------------------------------------- ! update local deformation gradient and coordinates ! deltaF_aim = math_rotate_backward33(deltaF_aim,rotation_BC) F = reshape(Utilities_forwardField(timeinc,F_aim,F_lastInc,Fdot),[9,res(1),res(2),res(3)]) F_lambda = reshape(Utilities_forwardField(timeinc,F_aim,F_lambda_lastInc,F_lambdadot),[9,res(1),res(2),res(3)]) call DMDAVecRestoreArrayF90(da,solution_vec,xx_psc,ierr) !-------------------------------------------------------------------------------------------------- ! update stiffness (and gamma operator) S = Utilities_maskedCompliance(rotation_BC,P_BC%maskLogical,C) if (update_gamma) call Utilities_updateGamma(C,restartWrite) ForwardData = .True. mask_stress = P_BC%maskFloat params%P_BC = P_BC%values params%rotation_BC = rotation_BC params%timeinc = timeinc call SNESSolve(snes,PETSC_NULL_OBJECT,solution_vec,ierr) call SNESGetConvergedReason(snes,reason,ierr) AL_solution%termIll = terminallyIll terminallyIll = .false. if (reason > 0 ) then AL_solution%converged = .true. AL_solution%iterationsNeeded = reportIter endif end function AL_solution !-------------------------------------------------------------------------------------------------- !> @brief forms the AL residual vector !-------------------------------------------------------------------------------------------------- subroutine AL_formResidual(in,x_scal,f_scal,dummy,ierr) use numerics, only: & itmax, & itmin use math, only: & math_rotate_backward33, & math_transpose33, & math_mul3333xx33 use mesh, only: & res, & wgt use DAMASK_spectral_Utilities, only: & field_real, & Utilities_FFTforward, & Utilities_fourierConvolution, & Utilities_FFTbackward, & Utilities_constitutiveResponse use IO, only: IO_intOut implicit none integer(pInt) :: i,j,k integer(pInt), save :: callNo = 3_pInt real(pReal), dimension(3,3) :: temp33_Real logical :: report DMDALocalInfo :: in(DMDA_LOCAL_INFO_SIZE) PetscScalar, target :: x_scal(3,3,2,XG_RANGE,YG_RANGE,ZG_RANGE) PetscScalar, target :: f_scal(3,3,2,X_RANGE,Y_RANGE,Z_RANGE) PetscScalar, pointer :: F(:,:,:,:,:), F_lambda(:,:,:,:,:) PetscScalar, pointer :: residual_F(:,:,:,:,:), residual_F_lambda(:,:,:,:,:) PetscInt :: iter, nfuncs PetscObject :: dummy PetscErrorCode :: ierr F => x_scal(:,:,1,:,:,:) F_lambda => x_scal(:,:,2,:,:,:) residual_F => f_scal(:,:,1,:,:,:) residual_F_lambda => f_scal(:,:,2,:,:,:) call SNESGetNumberFunctionEvals(snes,nfuncs,ierr) call SNESGetIterationNumber(snes,iter,ierr) !-------------------------------------------------------------------------------------------------- ! report begin of new iteration if (iter == 0 .and. callNo>2) then callNo = 0_pInt reportIter = 0_pInt endif if (callNo == 0 .or. mod(callNo,2) == 1_pInt) then write(6,'(/,a,3(a,'//IO_intOut(itmax)//'))') trim(incInfo), & ' @ Iter. ', itmin, '<',reportIter, '≤', itmax write(6,'(a,/,3(3(f12.7,1x)/))',advance='no') 'deformation gradient aim =',& math_transpose33(F_aim) reportIter = reportIter + 1_pInt endif callNo = callNo +1_pInt !-------------------------------------------------------------------------------------------------- ! evaluate constitutive response call Utilities_constitutiveResponse(coordinates,F_lastInc,F,temperature,params%timeinc, & residual_F,C,P_av,ForwardData,params%rotation_BC) ForwardData = .False. !-------------------------------------------------------------------------------------------------- ! stress BC handling F_aim = F_aim - math_mul3333xx33(S, ((P_av - params%P_BC))) ! S = 0.0 for no bc err_stress = maxval(abs(mask_stress * (P_av - params%P_BC))) ! mask = 0.0 for no bc !-------------------------------------------------------------------------------------------------- ! doing Fourier transform field_real = 0.0_pReal do k = 1_pInt, res(3); do j = 1_pInt, res(2); do i = 1_pInt, res(1) temp33_Real = math_mul3333xx33(S_scale,residual_F(1:3,1:3,i,j,k)) + math_I3 residual_F(1:3,1:3,i,j,k) = temp33_Real field_real(i,j,k,1:3,1:3) = -math_mul3333xx33(C_scale,F_lambda(1:3,1:3,i,j,k)-F(1:3,1:3,i,j,k)) enddo; enddo; enddo !-------------------------------------------------------------------------------------------------- ! doing Fourier transform call Utilities_FFTforward() call Utilities_fourierConvolution(math_rotate_backward33(F_aim,params%rotation_BC)) call Utilities_FFTbackward() !-------------------------------------------------------------------------------------------------- ! constructing residual residual_F_lambda = F - reshape(field_real(1:res(1),1:res(2),1:res(3),1:3,1:3),& [3,3,res(1),res(2),res(3)],order=[3,4,5,1,2]) residual_F = residual_F - F_lambda + residual_F_lambda !-------------------------------------------------------------------------------------------------- ! calculating errors err_f = wgt*sqrt(sum(residual_F_lambda**2.0_pReal)) err_p = wgt*sqrt(sum((residual_F - residual_F_lambda)**2.0_pReal)) end subroutine AL_formResidual !-------------------------------------------------------------------------------------------------- !> @brief convergence check !-------------------------------------------------------------------------------------------------- subroutine AL_converged(snes_local,it,xnorm,snorm,fnorm,reason,dummy,ierr) use numerics, only: & itmax, & itmin, & err_f_tol, & err_p_tol, & err_stress_tolrel, & err_stress_tolabs implicit none SNES :: snes_local PetscInt :: it PetscReal :: xnorm, snorm, fnorm SNESConvergedReason :: reason PetscObject :: dummy PetscErrorCode ::ierr logical :: Converged Converged = (it > itmin .and. & all([ err_f/sqrt(sum((F_aim-math_I3)*(F_aim-math_I3)))/err_f_tol, & err_p/sqrt(sum((F_aim-math_I3)*(F_aim-math_I3)))/err_p_tol, & err_stress/min(maxval(abs(P_av))*err_stress_tolrel,err_stress_tolabs)] < 1.0_pReal)) if (Converged) then reason = 1 elseif (it > itmax) then reason = -1 else reason = 0 endif write(6,'(a,es14.7)') 'error stress BC = ', err_stress/min(maxval(abs(P_av))*err_stress_tolrel,err_stress_tolabs) write(6,'(a,es14.7)') 'error F = ', err_f/sqrt(sum((F_aim-math_I3)*(F_aim-math_I3)))/err_f_tol write(6,'(a,es14.7)') 'error P = ', err_p/sqrt(sum((F_aim-math_I3)*(F_aim-math_I3)))/err_p_tol write(6,'(/,a)') '==========================================================================' end subroutine AL_converged !-------------------------------------------------------------------------------------------------- !> @brief destroy routine !-------------------------------------------------------------------------------------------------- subroutine AL_destroy() use DAMASK_spectral_Utilities, only: & Utilities_destroy implicit none PetscErrorCode :: ierr call VecDestroy(solution_vec,ierr) call SNESDestroy(snes,ierr) call DMDestroy(da,ierr) call PetscFinalize(ierr) call Utilities_destroy() end subroutine AL_destroy end module DAMASK_spectral_SolverAL