fusion_test.f90

Go to the documentation of this file.

! + + + + + + + + + + + + + + + + + + + + + + + + + + + +  
! + + + + + + + + + + + + + + + + + + + + + + + + + + + +  
program fusion_test
  use itm_types
  use euitm_schemas
  use euitm_routines
  use deallocate_structures
  use write_structures
  use fusion_sources_module
  use fusion_dist_sources_module
  use itm_constants

  implicit none

  real(R8), pointer                :: times(:)
  integer                          :: idx,nslice,islice
  integer                          :: shot, run
  integer                          :: iargc, iarg, i, req_arg=2, narg, ncpo
  character(len=132)               :: arg, arg2
  character(len=132), allocatable  :: cpos(:)
  character(len=132)               :: user, machine, ual_version
  REAL (R8)                        :: time
  type(type_coreprof), pointer     :: coreprof(:)
  type (type_equilibrium), pointer :: equilibrium(:)
  type(type_coresource), pointer   :: coresource(:)
  type (type_distsource), pointer  :: distsource(:)

  call getenv('USER', user)
  call getenv('DATAVERSION', ual_version)
  if(ual_version.eq.'') then
     ual_version = '4.10b'
  endif
  call getenv('TOKAMAKNAME', machine)
  if(machine.eq.'') then
     machine = 'test'
  endif
  iarg=1
  narg=iargc()
10 if(narg.lt.req_arg) then
     write(*,*) 'Need to specify shot and run no'
     write(*,*) 'Optionally preceded by "-u USER -d MACHINE -v UAL_VERSION"'
     stop 'ERROR!'
  endif
  call getarg(iarg,arg)
  if(arg(1:1).eq.'-') then
     call getarg(iarg+1,arg2)
     select case (arg(2:2))
     case ('u')
        user=arg2
     case('d')
        machine=arg2
     case('v')
        ual_version=arg2
     case default
        write(*,*) 'Unrecognized option ', trim(arg), ' ',trim(arg2)
        stop
     end select
     iarg=iarg+2
     req_arg=req_arg+2
     goto 10
  endif
  read(arg,*) shot
  call getarg(iarg+1,arg)
  read(arg,*) run
  iarg=iarg+2

  write(*,*) 'Processing shot/run = ',shot,' / ',run
  call euitm_open_env('euitm',shot,run,idx,trim(user),trim(machine),trim(ual_version))
  time = 1.0e8_r8
  call euitm_get_times(idx,'coreprof',times)
  write(*,*) 'Number of time points ',size(times)
!  write(*,*) times

  allocate(coreprof(1))
  allocate(equilibrium(1))

  call set_write_verbosity(0)  ! no verbose output
  call open_write_file(12, 'coresource_fusion.cpo')
  do i = 1, size(times)
     CALL euitm_get_slice(idx, 'coreprof', coreprof(1), times(i), ual_closest_sample)
     write(*,*) 'Te, Ti(ave) = ', coreprof(1)%te%value(1), &
          sum(coreprof(1)%ti%value(1,:) * coreprof(1)%ni%value(1,:)) / sum(coreprof(1)%ni%value(1,:))
     call fusion_sources(coreprof, coresource)
     call write_cpo(coresource, 'coresource')
  enddo
  call close_write_file

  call set_write_verbosity(0)  ! no verbose output
  call open_write_file(12, 'distsource_fusion.cpo')
  do i = 1, size(times)
     CALL euitm_get_slice(idx, 'coreprof', coreprof(1), times(i), ual_closest_sample)
     CALL euitm_get_slice(idx, 'equilibrium', equilibrium(1), times(i), ual_closest_sample)
     write(*,*) 'Te, Ti(ave) = ', coreprof(1)%te%value(1), &
          sum(coreprof(1)%ti%value(1,:) * coreprof(1)%ni%value(1,:)) / sum(coreprof(1)%ni%value(1,:))
     call fusion_dist_sources(coreprof, equilibrium, distsource)
     call write_cpo(distsource, 'distsource')
  enddo
  call close_write_file

  deallocate(times)
  call euitm_close(idx)
  call deallocate_cpo(coreprof)
  call deallocate_cpo(equilibrium)
  call deallocate_cpo(coresource)
  call deallocate_cpo(distsource)

end program fusion_test