gtdata_internal_map.f90

Go to the documentation of this file.

 
 
module gtdata_internal_map
 
  use dc_types, only: string
  use gtdata_internal_vartable, only: vid_invalid, &
    & vtb_class_unused, vtb_class_netcdf
    ! これらは, 外部から gtdata_internal_vartable を直接呼ばないようにするため,
    ! gtdata_internal_map を介して公開する.
 
  implicit none
 
  type gt_dimmap
    integer:: dimno
          ! 正ならば実体変数の次元番号, 他変数参照時は非正値
    character(len=STRING):: url
          ! 次元変数の url
    integer:: offset
          !=== 実体と gtool ユーザの格子番号対応
          !
          ! ユーザからみて 1..allcount が実体の
          ! (1..allcount) * step + offset に写像される。
          ! これらの値の変更の際は実体変数の許容添字範囲および
          ! 成長可能性を確認する必要あり。
          ! start 値に対するオフセット。
    integer:: step
          ! 1 か -1 の値をとることが期待される。
    integer:: allcount
          ! 見掛けの格子番号上限: start + count * stride <= allcount
    integer:: start
          !=== イテレータ本体
          ! 入出力範囲は (start:start+count*stride:stride) である。
          ! イテレータ start 値
    integer:: count
          ! イテレータ count 値
    integer:: stride
          ! イテレータ stride 値
    logical:: scalar
          ! スカラー変数である場合, .true.
  end type gt_dimmap
 
  type map_table_entry
    integer:: vid
    integer:: ndims
    type(gt_dimmap), pointer:: map(:)
  end type map_table_entry
 
  type(map_table_entry), save, target, allocatable:: maptab(:)
  integer, parameter:: maptab_init_size = 16
 
  public:: maptabadd, maptabdelete, map_lookup
  public:: var_class, map_create, map_to_internal_specs, dimrange
  public:: vid_invalid, vtb_class_unused, vtb_class_netcdf
  private:: maptab, maptab_init_size
 
  interface dimrange
    module procedure dimrange_by_dimno
  end interface
 
contains
 
  subroutine dimrange_by_dimno(var, dimno, dimlo, dimhi)
    use gtdata_types, only: gt_variable
    use gtdata_generic, only: open, close
    use gtdata_internal_vartable, only: dimrange
    type(gt_variable), intent(in):: var
    integer, intent(in):: dimno
    integer, intent(out):: dimlo, dimhi
    type(gt_variable):: dimvar
    integer:: vid
    call open(dimvar, var, dimno, count_compact=.true., &
      & inherit_slice=.false.)
    call map_lookup(dimvar, vid=vid)
    call dimrange(vid, dimlo, dimhi)
    call close(dimvar)
  end subroutine dimrange_by_dimno
 
  subroutine map_dup(var, source_var)
    use gtdata_types, only: gt_variable
    use gtdata_internal_vartable, only: vartableadd, vartablelookup
    use dc_trace, only: dbgmessage
    type(gt_variable), intent(out):: var
    type(gt_variable), intent(in):: source_var
    integer:: vid, mid1, mid2, vid2, nd, class, cid
    call map_lookup(source_var, vid=vid)
    if (vid < 0) then
      var = gt_variable(-1)
      return
    endif
    if (vid == 0) then
      vid2 = 0
    else
      call vartablelookup(vid, class=class, cid=cid)
      call vartableadd(vid2, class, cid)
    endif
    call maptabadd(var%mapid, vid2)
    mid1 = source_var%mapid
    mid2 = var%mapid
    maptab(mid2)%ndims = maptab(mid1)%ndims
    if (associated(maptab(mid1)%map)) then
      nd = size(maptab(mid1)%map)
      allocate(maptab(mid2)%map(nd))
      maptab(mid2)%map(1:nd) = maptab(mid1)%map(1:nd)
    else
      nullify(maptab(mid2)%map)
    endif
    call dbgmessage('map_dup mapid(%d from %d) vid(%d from %d)', &
      & i=(/mid2, mid1, maptab(mid2)%vid, maptab(mid1)%vid/))
  end subroutine map_dup
 
  subroutine map_create(var, class, cid, ndims, allcount, stat)
    use gtdata_types, only: gt_variable
    use gtdata_internal_vartable, only: vartableadd
    use dc_error, only: gt_enomoredims, dc_noerr
    type(gt_variable), intent(out):: var
    integer, intent(in):: class, cid, ndims, allcount(:)
    integer, intent(out):: stat
    type(gt_dimmap), pointer:: map(:)
    integer:: vid, i
  continue
 
    stat = dc_noerr
    if ( ndims < 0 ) then
      stat = gt_enomoredims
      goto 999
    end if
    call vartableadd(vid, class, cid)
    call maptabadd(var%mapid, vid)
    if (ndims > 0) then
      call map_allocate(map, ndims)
      maptab(var%mapid)%ndims = ndims
      maptab(var%mapid)%map => map
 
      do, i = 1, ndims
        map(i)%dimno = i
        map(i)%allcount = allcount(i)
        map(i)%count = allcount(i)
        map(i)%offset = 0
        map(i)%start = 1
        map(i)%step = 1
        map(i)%stride = 1
        map(i)%scalar = .false.
      enddo
    else
      ! スカラー変数 (ndims = 0) の場合
      call map_allocate(map, 1)
      maptab(var%mapid)%ndims = 0
      maptab(var%mapid)%map => map
      map(1)%dimno = 1
      map(1)%allcount = 1
      map(1)%count = 1
      map(1)%offset = 0
      map(1)%start = 1
      map(1)%step = 1
      map(1)%stride = 1
      map(1)%scalar = .true.
    end if
 
999 continue
    return
  end subroutine map_create
 
  subroutine maptabadd(mapid, vid)
    integer, intent(out):: mapid
    integer, intent(in):: vid
    type(map_table_entry), allocatable:: tmp_maptab(:)
    integer:: i, n
    ! 必要なら初期確保
    if (.not. allocated(maptab)) then
      allocate(maptab(maptab_init_size))
      maptab(:)%vid = vid_invalid
      do, n = 1, maptab_init_size
        nullify(maptab(n)%map)
      enddo
    endif
    ! 空き地があればそこに割り当て
    do, i = 1, size(maptab)
      if (maptab(i)%vid == vid_invalid) then
        mapid = i
        maptab(mapid)%vid = vid
        return
      endif
    enddo
    ! 空き地はなかったのだから倍幅確保
    n = size(maptab)
    allocate(tmp_maptab(n))
    tmp_maptab(:) = maptab(:)
    deallocate(maptab)
    allocate(maptab(n * 2))
    ! 確保したところはクリア
    maptab(1:n) = tmp_maptab(1:n)
    do, i = n + 1, (2 * size(tmp_maptab))
      maptab(i)%vid = vid_invalid
      nullify(maptab(i)%map)
    enddo
    deallocate(tmp_maptab)
    mapid = n + 1
    maptab(mapid)%vid = vid
  end subroutine maptabadd
 
  subroutine maptabdelete(var, err)
    use dc_error, only: nf90_enotvar, storeerror, dc_noerr
    use gtdata_types, only: gt_variable
    use dc_trace, only: dbgmessage
    implicit none
    type(gt_variable), intent(in):: var
    logical, intent(out), optional:: err
    integer:: mapid
    mapid = var%mapid
    if (.not. allocated(maptab)) goto 999
    if (mapid <= 0 .or. mapid > size(maptab)) goto 999
    if (maptab(mapid)%vid == vid_invalid) goto 999
    maptab(mapid)%vid = vid_invalid
    if (associated(maptab(mapid)%map)) deallocate(maptab(mapid)%map)
    call storeerror(dc_noerr, 'maptabdelete', err)
    call dbgmessage('gtdata_internal_map table %d deleted', i=(/mapid/))
    return
999 continue
    call storeerror(nf90_enotvar, 'maptabdelete', err)
  end subroutine maptabdelete
 
  subroutine map_lookup(var, vid, map, ndims)
    use gtdata_types, only: gt_variable
    type(gt_variable), intent(in):: var
    integer, intent(out), optional:: vid
    type(gt_dimmap), intent(out), optional:: map(:)
    integer, intent(out), optional:: ndims
    if (.not. allocated(maptab)) goto 999
    if (var%mapid <= 0 .or. var%mapid > size(maptab)) goto 999
    if (maptab(var%mapid)%vid == vid_invalid) goto 999
    if (present(vid)) vid = maptab(var%mapid)%vid
    if (present(map)) map(:) = maptab(var%mapid)%map(1:size(map))
    if (present(ndims)) ndims = maptab(var%mapid)%ndims
    return
999 continue
    if (present(vid)) vid = vid_invalid
    if (present(map)) then
      map(:)%dimno = -1
      map(:)%url = " "
    endif
    if (present(ndims)) ndims = 0
  end subroutine map_lookup
 
  subroutine map_set(var, map, stat)
    use gtdata_types, only: gt_variable
    use dc_error, only: nf90_enotvar, gt_enomoredims, dc_noerr
    type(gt_variable), intent(in):: var
    type(gt_dimmap), intent(in):: map(:)
    integer, intent(out):: stat
    if (.not. allocated(maptab)) goto 999
    if (var%mapid <= 0 .or. var%mapid > size(maptab)) goto 999
    if (maptab(var%mapid)%vid == vid_invalid) goto 999
    if (size(map) > size(maptab(var%mapid)%map)) then
      stat = gt_enomoredims
      return
    endif
    maptab(var%mapid)%map(1:size(map)) = map(:)
    stat = dc_noerr
    return
999 continue
    stat = nf90_enotvar
  end subroutine map_set
 
  subroutine var_class(var, class, cid)
    use gtdata_types, only: gt_variable
    use gtdata_internal_vartable, only: vartablelookup
    type(gt_variable), intent(in):: var
    integer, intent(out), optional:: class, cid
    integer:: vid
    call map_lookup(var, vid=vid)
    call vartablelookup(vid, class=class, cid=cid)
  end subroutine var_class
 
  subroutine map_set_ndims(var, ndims, stat)
    use gtdata_types, only: gt_variable
    use dc_error, only: nf90_enotvar, gt_enomoredims, dc_noerr
    type(gt_variable), intent(in):: var
    integer, intent(in):: ndims
    integer, intent(out):: stat
    integer:: vid
    call map_lookup(var, vid=vid)
    if (vid == vid_invalid) then
      stat = nf90_enotvar
      return
    endif
    if (.not. associated(maptab(var%mapid)%map)) then
      if (ndims == 0) then
        stat = dc_noerr
        maptab(var%mapid)%ndims = 0
      else
        stat = gt_enomoredims
      endif
    else
      if (ndims > size(maptab(var%mapid)%map)) then
        stat = gt_enomoredims
      else
        stat = dc_noerr
        maptab(var%mapid)%ndims = ndims
      endif
    endif
  end subroutine map_set_ndims
 
  subroutine map_set_rank(var, rank, stat)
    use gtdata_types, only: gt_variable
    use dc_error, only: nf90_enotvar, gt_enomoredims, dc_noerr
    type(gt_variable), intent(in):: var
    integer, intent(in):: rank
    integer, intent(out):: stat
    type(gt_dimmap), pointer:: tmpmap(:)
    integer:: ndims
    integer:: vid, nd
    call map_lookup(var, vid, ndims=ndims)
    if (vid == vid_invalid) then
      stat = nf90_enotvar
      return
    endif
    if (ndims < rank) then
      stat = gt_enomoredims
      return
    endif
    tmpmap => maptab(var%mapid)%map
    do, nd = ndims, 1, -1
      if (count(tmpmap(1:ndims)%count > 1) <= rank) exit
      tmpmap(nd)%count = 1
    enddo
    stat = dc_noerr
  end subroutine map_set_rank
 
  subroutine map_to_internal_specs(var, specs, ndims)
    use gtdata_types, only: gt_variable
    use gtdata_internal_vartable, only: num_dimensions => ndims
    type(gt_variable), intent(in):: var
    integer, pointer:: specs(:, :)
    integer, intent(out), optional:: ndims
    type(gt_dimmap), pointer:: it
    integer:: vid, i, j, imap, internal_ndims
    integer:: external_ndims
  continue
    call map_lookup(var, vid, ndims=external_ndims)
    internal_ndims = num_dimensions(vid)
    if (present(ndims)) ndims = internal_ndims
    allocate(specs(max(1, internal_ndims), 4))
    specs(:, 1) = 1
    specs(:, 2) = 1
    specs(:, 3) = 1
    specs(:, 4) = 0
    imap = 1
    do, i = 1, size(maptab(var%mapid)%map)
      it => maptab(var%mapid)%map(i)
      j = it%dimno
      if (j > 0 .and. j <= internal_ndims) then
        specs(j, 1) = it%start + it%offset
        specs(j, 2) = it%count
        if (i > external_ndims) specs(j, 2) = 1
        specs(j, 3) = it%stride * it%step
        specs(j, 4) = imap
      endif
      imap = imap * it%count
    enddo
  end subroutine map_to_internal_specs
 
  subroutine map_allocate(map, ndims)
    type(gt_dimmap), pointer:: map(:)
    integer, intent(in):: ndims
    if (ndims <= 0) then
      nullify(map)
      return
    endif
    allocate(map(1:ndims))
    map(1:ndims)%dimno = -1
    map(1:ndims)%url = ' '
    map(1:ndims)%allcount = 0
    map(1:ndims)%offset = 0
    map(1:ndims)%step = 1
    map(1:ndims)%start = 1
    map(1:ndims)%count = 0
    map(1:ndims)%stride = 1
    map(1:ndims)%scalar = .false.
  end subroutine map_allocate
 
  subroutine map_apply(var, map)
    use gtdata_types, only: gt_variable
    type(gt_variable), intent(inout):: var
    type(gt_dimmap), pointer:: map(:)
    type(gt_dimmap), pointer:: tmpmap(:), varmap
    integer:: i, nd
    nd = size(map)
    allocate(tmpmap(nd))
    do, i = 1, nd
      tmpmap(i)%allcount = map(i)%allcount
      tmpmap(i)%count = map(i)%count
      if (map(i)%dimno > 0) then
        varmap => maptab(var%mapid)%map(map(i)%dimno)
        tmpmap(i)%url = varmap%url
        tmpmap(i)%dimno = varmap%dimno
        tmpmap(i)%offset = varmap%offset + map(i)%offset
        tmpmap(i)%step = varmap%step * map(i)%step
      else
        tmpmap(i)%url = map(i)%url
        tmpmap(i)%dimno = 0
        tmpmap(i)%offset = map(i)%offset
        tmpmap(i)%step = map(i)%step
      endif
    enddo
    deallocate(map)
    map => tmpmap
  end subroutine map_apply
 
  subroutine map_resize(var, ndims)
    use gtdata_types, only: gt_variable
    type(gt_variable), intent(in):: var
    integer, intent(in):: ndims
    type(gt_dimmap), pointer:: newmap(:)
    type(gt_dimmap), pointer:: tmpmap(:)
    integer:: n
    if (associated(maptab(var%mapid)%map)) then
      tmpmap => maptab(var%mapid)%map
      call map_allocate(newmap, ndims)
      n = min(size(tmpmap), ndims)
      newmap(1:n) = tmpmap(1:n)
      deallocate(tmpmap)
      maptab(var%mapid)%map => newmap
      newmap(n+1:ndims)%dimno = -1
      newmap(n+1:ndims)%url = ' '
      newmap(n+1:ndims)%allcount = 0
      newmap(n+1:ndims)%offset = 0
      newmap(n+1:ndims)%step = 1
      newmap(n+1:ndims)%start = 1
      newmap(n+1:ndims)%count = 0
      newmap(n+1:ndims)%stride = 1
    else
      call map_allocate(maptab(var%mapid)%map, ndims)
      n = 1
    endif
  end subroutine map_resize
 
  subroutine gtvar_dump(var)
    use gtdata_types, only: gt_variable
    use gtdata_internal_vartable, only: vartable_dump
    use dc_trace, only: debug, dbgmessage
    type(gt_variable), intent(in):: var
    integer:: idim, imap
    logical:: dbg_mode
  continue
    call debug( dbg_mode )
    if (.not. dbg_mode) return
    imap = var%mapid
    if (imap < 1 .or. imap > size(maptab)) then
      call dbgmessage('[gt_variable %d: invalid id]', i=(/imap/))
      return
    endif
    if (associated(maptab(imap)%map)) then
      call dbgmessage('[gt_variable %d: ndims=%d, map.size=%d]', &
        & i=(/imap, maptab(imap)%ndims, size(maptab(imap)%map)/))
      do, idim = 1, size(maptab(imap)%map)
        call dbgmessage('[dim%d dimno=%d ofs=%d step=%d' &
          &// ' all=%d start=%d count=%d stride=%d url=%c]', &
          & c1=trim(maptab(imap)%map(idim)%url), &
          & i=(/idim, maptab(imap)%map(idim)%dimno, &
          & maptab(imap)%map(idim)%offset, &
          & maptab(imap)%map(idim)%step, &
          & maptab(imap)%map(idim)%allcount, &
          & maptab(imap)%map(idim)%start, &
          & maptab(imap)%map(idim)%count, &
          & maptab(imap)%map(idim)%stride/))
      enddo
    else
      call dbgmessage('[gt_variable %d: ndims=%d, map=null]', &
        & i=(/imap, maptab(imap)%ndims/))
    endif
    call vartable_dump(maptab(imap)%vid)
  end subroutine gtvar_dump
 
  integer function dimord_skip_compact(dimord, map) result(result)
    use dc_trace, only: dbgmessage
    integer, intent(in):: dimord
    type(gt_dimmap), intent(in):: map(:)
    integer:: nd, id
    result = -1
    nd = 0
    do, id = 1, size(map)
      if (map(id)%count < 2) cycle
      nd = nd + 1
      if (nd < dimord) cycle
      result = id
      call dbgmessage('compact dim skip: %d <= %d', i=(/result, dimord/))
      exit
    enddo
  end function dimord_skip_compact
 
end module gtdata_internal_map