Implement save_npz

Author: minhqdao

src/stdlib_io_np.fypp

@@ -137,7 +137,7 @@ module stdlib_io_np
     interface save_npz
         module subroutine save_npz_from_arrays(filename, arrays, compressed, iostat, iomsg)
             character(len=*), intent(in) :: filename
-            type(t_array_wrapper), intent(in) :: arrays(*)
+            type(t_array_wrapper), intent(in) :: arrays(:)
             !> If true, the file is saved in compressed format. The default is false.
             logical, intent(in), optional :: compressed
             integer, intent(out), optional :: iostat

src/stdlib_io_np_load.fypp

@@ -9,7 +9,7 @@ submodule(stdlib_io_np) stdlib_io_np_load
     use stdlib_array
     use stdlib_error, only: error_stop
     use stdlib_filesystem, only: exists, list_dir, temp_dir
-    use stdlib_io_zip, only: unzip, default_unzip_dir, zip_contents
+    use stdlib_io_zip, only: unzip, default_unzip_dir, zip_contents, zip
     use stdlib_strings, only: to_string, starts_with
     use stdlib_string_type, only: string_type, as_string => char
     implicit none

src/stdlib_io_np_save.fypp

@@ -6,8 +6,12 @@
 
 !> Implementation of saving multidimensional arrays to npy files
 submodule(stdlib_io_np) stdlib_io_np_save
+    use stdlib_array
     use stdlib_error, only: error_stop
+    use stdlib_filesystem, only: run
     use stdlib_strings, only: to_string
+    use stdlib_string_type, only: string_type, char
+    use stdlib_io_zip, only: zip
     implicit none
 
 contains
@@ -137,18 +141,79 @@ contains
     !> ([Specification](../page/specs/stdlib_io.html#save_npz))
     module subroutine save_npz_from_arrays(filename, arrays, compressed, iostat, iomsg)
         character(len=*), intent(in) :: filename
-        type(t_array_wrapper), intent(in) :: arrays(*)
+        type(t_array_wrapper), intent(in) :: arrays(:)
         !> If true, the file is saved in compressed format. The default is false.
         logical, intent(in), optional :: compressed
         integer, intent(out), optional :: iostat
         character(len=:), allocatable, intent(out), optional :: iomsg
 
+        integer :: i, j, stat
         logical :: is_compressed
+        character(len=:), allocatable :: msg
+        type(string_type), allocatable :: files(:)
+
+        if (present(iostat)) iostat = 0
 
         if (present(compressed)) then
             is_compressed = compressed
         else
             is_compressed = .false.
         end if
+
+        allocate(files(size(arrays)))
+        do i = 1, size(arrays)
+            select type (typed_array => arrays(i)%array)
+#:for k1, t1 in KINDS_TYPES
+#:for rank in RANKS
+              class is (t_array_${t1[0]}$${k1}$_${rank}$)
+                do j = 1, size(files)
+                    if (char(files(j)) == typed_array%name) then
+                        if (present(iostat)) iostat = 1
+                        if (present(iomsg)) iomsg = "Error saving array to file '"//filename// &
+                            "': Array with the same name '"//typed_array%name//"' already exists."
+                        call delete_files(files)
+                        return
+                    end if
+                end do
+
+                call save_npy(typed_array%name, typed_array%values, stat, msg)
+                if (stat /= 0) then
+                    if (present(iostat)) iostat = stat
+                    if (present(iomsg)) iomsg = msg
+                    call delete_files(files)
+                    return
+                end if
+
+                files = [files, string_type(typed_array%name)]
+#:endfor
+#:endfor
+              class default
+                if (present(iostat)) iostat = 1
+                if (present(iomsg)) iomsg = "Error saving array to file '"//filename//"': Unsupported array type."
+                call delete_files(files)
+                return
+            end select
+        end do
+
+        call zip(filename, files, stat, msg)
+        if (stat /= 0) then
+            if (present(iostat)) iostat = stat
+            if (present(iomsg)) iomsg = msg
+            call delete_files(files)
+            return
+        end if
+
+        call delete_files(files)
+    end
+
+    subroutine delete_files(files)
+        type(string_type), allocatable, intent(in) :: files(:)
+
+        integer :: i, unit
+
+        do i = 1, size(files)
+            open(newunit=unit, file=char(files(i)))
+            close(unit, status="delete")
+        end do
     end
 end