From fba50ee14e2c44796b2646c696cd56879063c2f5 Mon Sep 17 00:00:00 2001
From: Ramil <ramil.rgk@gmail.com>
Date: Thu, 6 Jun 2024 16:39:11 +0300
Subject: [PATCH] 2D, 3D & with time recording functionality
---
src/main.f90 | 34 ++++++++++--
src/netcdf_io_module.f90 | 110 +++++++++++++++++++++++++++++++++++++--
2 files changed, 137 insertions(+), 7 deletions(-)
diff --git a/src/main.f90 b/src/main.f90
index 9528569..d9ca9c7 100644
--- a/src/main.f90
+++ b/src/main.f90
@@ -6,9 +6,13 @@ program main
real, dimension(10) :: series_data = (/10.0, 9.0, 8.0, 7.0, 6.0, 5.0, 4.0, 3.0, 2.0, 1.0/)
real, dimension(10, 10) :: timescan_data
real, dimension(10) :: profile_data
- integer :: i, j
+ real, dimension(10, 20, 20) :: d2time_data
+ real, dimension(10, 20, 20, 10) :: d3time_data
+ real, dimension(20, 20) :: d2_data
+ real, dimension(20, 20, 10) :: d3_data
+ integer :: i, j, k, m
- ! Инициализация данных timescan и profile
+ ! Инициализация данных timescan, profile, d2time, d3time, d2, d3
do i = 1, 10
profile_data(i) = real(i)
do j = 1, 10
@@ -16,11 +20,35 @@ program main
end do
end do
- call open_netcdf('series_data.nc', ios, 10, 10)
+ do i = 1, 10
+ do j = 1, 20
+ do k = 1, 20
+ d2time_data(i, j, k) = real(i) + real(j) + real(k)
+ do m = 1, 10
+ d3time_data(i, j, k, m) = real(i) + real(j) + real(k) + real(m)
+ end do
+ end do
+ end do
+ end do
+
+ do i = 1, 20
+ do j = 1, 20
+ d2_data(i, j) = real(i) + real(j)
+ do k = 1, 10
+ d3_data(i, j, k) = real(i) + real(j) + real(k)
+ end do
+ end do
+ end do
+
+ call open_netcdf('series_data.nc', ios, 10, 10, 20, 20)
if (ios%is_open) then
call write_series(ios, time_data, series_data)
call write_timescan(ios, timescan_data)
call write_profile(ios, profile_data)
+ call write_2Dtime(ios, d2time_data)
+ call write_3Dtime(ios, d3time_data)
+ call write_2D(ios, d2_data)
+ call write_3D(ios, d3_data)
call close_netcdf(ios)
else
print *, 'Failed to open NetCDF file.'
diff --git a/src/netcdf_io_module.f90 b/src/netcdf_io_module.f90
index 373d42c..9d18197 100644
--- a/src/netcdf_io_module.f90
+++ b/src/netcdf_io_module.f90
@@ -4,8 +4,9 @@ module netcdf_io_module
type :: io_struct
integer :: ncid
- integer :: time_dimid, height_dimid
+ integer :: time_dimid, height_dimid, lat_dimid, lon_dimid
integer :: varid_time, varid_series, varid_timescan, varid_profile
+ integer :: varid_2Dtime, varid_3Dtime, varid_2D, varid_3D
logical :: is_open = .false.
character(len=128) :: series_name = 'timeseries of variable'
character(len=128) :: series_long_name = 'timeseries of some variable'
@@ -16,13 +17,25 @@ module netcdf_io_module
character(len=128) :: profile_name = 'vertical profile of variable'
character(len=128) :: profile_long_name = 'vertical profile of some variable'
character(len=128) :: profile_units = 'unit'
+ character(len=128) :: d2time_name = '2D variable with time'
+ character(len=128) :: d2time_long_name = '2D variable with time'
+ character(len=128) :: d2time_units = 'unit'
+ character(len=128) :: d3time_name = '3D variable with time'
+ character(len=128) :: d3time_long_name = '3D variable with time'
+ character(len=128) :: d3time_units = 'unit'
+ character(len=128) :: d2_name = '2D variable'
+ character(len=128) :: d2_long_name = '2D variable'
+ character(len=128) :: d2_units = 'unit'
+ character(len=128) :: d3_name = '3D variable'
+ character(len=128) :: d3_long_name = '3D variable'
+ character(len=128) :: d3_units = 'unit'
end type io_struct
contains
- subroutine open_netcdf(filename, ios, time_len, height_len)
+ subroutine open_netcdf(filename, ios, time_len, height_len, lat_len, lon_len)
character(len=*), intent(in) :: filename
type(io_struct), intent(out) :: ios
- integer, intent(in) :: time_len, height_len
+ integer, intent(in) :: time_len, height_len, lat_len, lon_len
integer :: ierr
ierr = nf90_create(filename, nf90_clobber, ios%ncid)
@@ -30,10 +43,22 @@ contains
ios%is_open = .true.
ierr = nf90_def_dim(ios%ncid, 'time', time_len, ios%time_dimid)
ierr = nf90_def_dim(ios%ncid, 'height', height_len, ios%height_dimid)
+ ierr = nf90_def_dim(ios%ncid, 'lat', lat_len, ios%lat_dimid)
+ ierr = nf90_def_dim(ios%ncid, 'lon', lon_len, ios%lon_dimid)
ierr = nf90_def_var(ios%ncid, 'time', nf90_float, ios%time_dimid, ios%varid_time)
ierr = nf90_def_var(ios%ncid, 'series', nf90_float, ios%time_dimid, ios%varid_series)
- ierr = nf90_def_var(ios%ncid, 'timescan', nf90_float, [ios%time_dimid, ios%height_dimid], ios%varid_timescan)
+ ierr = nf90_def_var(ios%ncid, 'timescan', nf90_float, &
+ [ios%time_dimid, ios%height_dimid], ios%varid_timescan)
ierr = nf90_def_var(ios%ncid, 'profile', nf90_float, ios%height_dimid, ios%varid_profile)
+ ierr = nf90_def_var(ios%ncid, '2Dtime', nf90_float, &
+ [ios%time_dimid, ios%lat_dimid, ios%lon_dimid], ios%varid_2Dtime)
+ ierr = nf90_def_var(ios%ncid, '3Dtime', nf90_float, &
+ [ios%time_dimid, ios%lat_dimid, ios%lon_dimid, ios%height_dimid], ios%varid_3Dtime)
+ ierr = nf90_def_var(ios%ncid, '2D', nf90_float, &
+ [ios%lat_dimid, ios%lon_dimid], ios%varid_2D)
+ ierr = nf90_def_var(ios%ncid, '3D', nf90_float, &
+ [ios%lat_dimid, ios%lon_dimid, ios%height_dimid], ios%varid_3D)
+
ierr = nf90_put_att(ios%ncid, ios%varid_series, 'standard_name', ios%series_name)
ierr = nf90_put_att(ios%ncid, ios%varid_series, 'long_name', ios%series_long_name)
ierr = nf90_put_att(ios%ncid, ios%varid_series, 'units', ios%series_units)
@@ -43,6 +68,19 @@ contains
ierr = nf90_put_att(ios%ncid, ios%varid_profile, 'standard_name', ios%profile_name)
ierr = nf90_put_att(ios%ncid, ios%varid_profile, 'long_name', ios%profile_long_name)
ierr = nf90_put_att(ios%ncid, ios%varid_profile, 'units', ios%profile_units)
+ ierr = nf90_put_att(ios%ncid, ios%varid_2Dtime, 'standard_name', ios%d2time_name)
+ ierr = nf90_put_att(ios%ncid, ios%varid_2Dtime, 'long_name', ios%d2time_long_name)
+ ierr = nf90_put_att(ios%ncid, ios%varid_2Dtime, 'units', ios%d2time_units)
+ ierr = nf90_put_att(ios%ncid, ios%varid_3Dtime, 'standard_name', ios%d3time_name)
+ ierr = nf90_put_att(ios%ncid, ios%varid_3Dtime, 'long_name', ios%d3time_long_name)
+ ierr = nf90_put_att(ios%ncid, ios%varid_3Dtime, 'units', ios%d3time_units)
+ ierr = nf90_put_att(ios%ncid, ios%varid_2D, 'standard_name', ios%d2_name)
+ ierr = nf90_put_att(ios%ncid, ios%varid_2D, 'long_name', ios%d2_long_name)
+ ierr = nf90_put_att(ios%ncid, ios%varid_2D, 'units', ios%d2_units)
+ ierr = nf90_put_att(ios%ncid, ios%varid_3D, 'standard_name', ios%d3_name)
+ ierr = nf90_put_att(ios%ncid, ios%varid_3D, 'long_name', ios%d3_long_name)
+ ierr = nf90_put_att(ios%ncid, ios%varid_3D, 'units', ios%d3_units)
+
ierr = nf90_enddef(ios%ncid)
else
print *, 'Error opening file: ', ierr
@@ -99,6 +137,70 @@ contains
endif
end subroutine write_profile
+ subroutine write_2Dtime(ios, d2time_data)
+ type(io_struct), intent(in) :: ios
+ real, dimension(:,:,:) :: d2time_data
+ integer :: ierr
+
+ if (.not. ios%is_open) then
+ print *, "File is not open."
+ return
+ endif
+
+ ierr = nf90_put_var(ios%ncid, ios%varid_2Dtime, d2time_data)
+ if (ierr /= nf90_noerr) then
+ print *, 'Error writing 2Dtime data:', ierr
+ endif
+ end subroutine write_2Dtime
+
+ subroutine write_3Dtime(ios, d3time_data)
+ type(io_struct), intent(in) :: ios
+ real, dimension(:,:,:,:) :: d3time_data
+ integer :: ierr
+
+ if (.not. ios%is_open) then
+ print *, "File is not open."
+ return
+ endif
+
+ ierr = nf90_put_var(ios%ncid, ios%varid_3Dtime, d3time_data)
+ if (ierr /= nf90_noerr) then
+ print *, "Error writing 3Dtime data:", ierr
+ endif
+ end subroutine write_3Dtime
+
+ subroutine write_2D(ios, d2_data)
+ type(io_struct), intent(in) :: ios
+ real, dimension(:,:) :: d2_data
+ integer :: ierr
+
+ if (.not. ios%is_open) then
+ print *, "File is not open."
+ return
+ endif
+
+ ierr = nf90_put_var(ios%ncid, ios%varid_2D, d2_data)
+ if (ierr /= nf90_noerr) then
+ print *, 'Error writing 2D data:', ierr
+ endif
+ end subroutine write_2D
+
+ subroutine write_3D(ios, d3_data)
+ type(io_struct), intent(in) :: ios
+ real, dimension(:,:,:) :: d3_data
+ integer :: ierr
+
+ if (.not. ios%is_open) then
+ print *, "File is not open."
+ return
+ endif
+
+ ierr = nf90_put_var(ios%ncid, ios%varid_3D, d3_data)
+ if (ierr /= nf90_noerr) then
+ print *, 'Error writing 3D data:', ierr
+ endif
+ end subroutine write_3D
+
subroutine close_netcdf(ios)
type(io_struct), intent(inout) :: ios
integer :: ierr
--
GitLab