writer.f90: writer_template

src/CMakeLists.txt

@@ -34,11 +34,40 @@ if (check_error(ier, dname)) then
   error stop
 endif
 ")
-
 configure_file(reader.in.f90 reader.f90)
 
+set(writer_template "
+integer :: varid, dimids(rank(value)), ier
+
+call self%def_dims(dname, dims, shape(value), dimids, ier)
+
+select type (value)
+type is (real(real64))
+  if(ier == NF90_NOERR) ier = nf90_def_var(self%ncid, dname, NF90_DOUBLE, dimids=dimids, varid=varid)
+  if(ier == NF90_NOERR) ier = nf90_put_var(self%ncid, varid, value)
+type is (real(real32))
+  if(ier == NF90_NOERR) ier = nf90_def_var(self%ncid, dname, NF90_FLOAT, dimids=dimids, varid=varid)
+  if(ier == NF90_NOERR) ier = nf90_put_var(self%ncid, varid, value)
+type is (integer(int64))
+  if(ier == NF90_NOERR) ier = nf90_def_var(self%ncid, dname, NF90_INT64, dimids=dimids, varid=varid)
+  if(ier == NF90_NOERR) ier = nf90_put_var(self%ncid, varid, value)
+type is (integer(int32))
+  if(ier == NF90_NOERR) ier = nf90_def_var(self%ncid, dname, NF90_INT, dimids=dimids, varid=varid)
+  if(ier == NF90_NOERR) ier = nf90_put_var(self%ncid, varid, value)
+class default
+  ierr = NF90_EBADTYPE
+end select
+
+if (present(ierr)) ierr = ier
+if (check_error(ier, dname)) then
+  if (present(ierr)) return
+  error stop
+endif
+")
+configure_file(writer.in.f90 writer.f90)
+
 target_sources(nc4fortran PRIVATE
 interface.f90
 read.f90 ${CMAKE_CURRENT_BINARY_DIR}/reader.f90
-write.f90 writer.f90
+write.f90 ${CMAKE_CURRENT_BINARY_DIR}/writer.f90
 string_utils.f90 ${CMAKE_CURRENT_BINARY_DIR}/pathlib.f90)

src/meson.build

@@ -37,15 +37,50 @@ if (check_error(ier, dname)) then
   if (present(ierr)) return
   error stop
 endif
-'''
-)
+''')
 configure_file(
   input : 'reader.in.f90',
   output : 'reader.f90',
   configuration : reader_template
 )
 
+writer_template = configuration_data()
+writer_template.set('writer_template',
+'''
+integer :: varid, dimids(rank(value)), ier
+
+call self%def_dims(dname, dims, shape(value), dimids, ier)
+
+select type (value)
+type is (real(real64))
+  if(ier == NF90_NOERR) ier = nf90_def_var(self%ncid, dname, NF90_DOUBLE, dimids=dimids, varid=varid)
+  if(ier == NF90_NOERR) ier = nf90_put_var(self%ncid, varid, value)
+type is (real(real32))
+  if(ier == NF90_NOERR) ier = nf90_def_var(self%ncid, dname, NF90_FLOAT, dimids=dimids, varid=varid)
+  if(ier == NF90_NOERR) ier = nf90_put_var(self%ncid, varid, value)
+type is (integer(int64))
+  if(ier == NF90_NOERR) ier = nf90_def_var(self%ncid, dname, NF90_INT64, dimids=dimids, varid=varid)
+  if(ier == NF90_NOERR) ier = nf90_put_var(self%ncid, varid, value)
+type is (integer(int32))
+  if(ier == NF90_NOERR) ier = nf90_def_var(self%ncid, dname, NF90_INT, dimids=dimids, varid=varid)
+  if(ier == NF90_NOERR) ier = nf90_put_var(self%ncid, varid, value)
+class default
+  ierr = NF90_EBADTYPE
+end select
+
+if (present(ierr)) ierr = ier
+if (check_error(ier, dname)) then
+  if (present(ierr)) return
+  error stop
+endif
+''')
+configure_file(
+  input : 'writer.in.f90',
+  output : 'writer.f90',
+  configuration : writer_template
+)
+
 nc4_src = files('interface.f90',
   'read.f90', meson.current_build_dir() / 'reader.f90',
-  'write.f90', 'writer.f90',
+  'write.f90', meson.current_build_dir() / 'writer.f90',
   'string_utils.f90', meson.current_build_dir() / 'pathlib.f90')