mirror of
https://github.com/TREX-CoE/trexio.git
synced 2024-12-22 04:14:40 +01:00
Added trexio_cp
This commit is contained in:
parent
cadcf1f62d
commit
fefc01b46b
@ -104,7 +104,7 @@ esac
|
|||||||
## ---------
|
## ---------
|
||||||
|
|
||||||
# Checks for basic header files.
|
# Checks for basic header files.
|
||||||
AC_CHECK_HEADERS([fcntl.h inttypes.h stdint.h stdbool.h stdlib.h string.h unistd.h])
|
AC_CHECK_HEADERS([fcntl.h inttypes.h stdint.h stdbool.h stdlib.h string.h unistd.h sys/types.h sys/wait.h ])
|
||||||
|
|
||||||
|
|
||||||
### HDF5
|
### HDF5
|
||||||
@ -130,11 +130,12 @@ if test "x${with_hdf5}" = xno; then
|
|||||||
recommended to build efficient TREXIO.
|
recommended to build efficient TREXIO.
|
||||||
------------------------------------------])
|
------------------------------------------])
|
||||||
else
|
else
|
||||||
AC_HAVE_LIBRARY([hdf5_hl], [], AC_MSG_ERROR([
|
AC_CHECK_LIB([hdf5_hl],[main],[],[AC_MSG_ERROR(
|
||||||
-----------------------------------
|
-----------------------------------
|
||||||
Error: hdf5_hl library is required
|
Error: hdf5_hl library is required
|
||||||
-----------------------------------
|
-----------------------------------
|
||||||
]), [])
|
)],[])ac_cv_lib_hdf5_hl=ac_cv_lib_hdf5_hl_main
|
||||||
|
|
||||||
fi
|
fi
|
||||||
|
|
||||||
CFLAGS="${HDF5_CFLAGS} ${CFLAGS}"
|
CFLAGS="${HDF5_CFLAGS} ${CFLAGS}"
|
||||||
|
@ -44,6 +44,9 @@ typedef int32_t trexio_exit_code;
|
|||||||
#include <stdbool.h>
|
#include <stdbool.h>
|
||||||
#include <string.h>
|
#include <string.h>
|
||||||
#include <err.h>
|
#include <err.h>
|
||||||
|
#include <sys/types.h>
|
||||||
|
#include <sys/wait.h>
|
||||||
|
|
||||||
|
|
||||||
#include "trexio.h"
|
#include "trexio.h"
|
||||||
#include "trexio_private.h"
|
#include "trexio_private.h"
|
||||||
@ -1585,6 +1588,139 @@ def _inquire(file_name: str) -> bool:
|
|||||||
raise Error(rc)
|
raise Error(rc)
|
||||||
#+end_src
|
#+end_src
|
||||||
|
|
||||||
|
** File copy
|
||||||
|
|
||||||
|
~trexio_cp~ copies a TREXIO file using =/bin/cp=.
|
||||||
|
|
||||||
|
**Input parameters:**
|
||||||
|
1) ~source_file_name~ - string containing the name of the source file
|
||||||
|
2) ~destination_file_name~ - string containing the name of the new file
|
||||||
|
|
||||||
|
**Output:**
|
||||||
|
- ~trexio_exit_code~
|
||||||
|
|
||||||
|
*** C
|
||||||
|
|
||||||
|
#+begin_src c :tangle prefix_front.h :exports none
|
||||||
|
trexio_exit_code trexio_cp(const char* source_file_name, const char* dest_file_name);
|
||||||
|
#+end_src
|
||||||
|
|
||||||
|
#+begin_src c :tangle prefix_front.c
|
||||||
|
|
||||||
|
trexio_exit_code
|
||||||
|
trexio_cp(const char* source, const char* dest)
|
||||||
|
{
|
||||||
|
|
||||||
|
if (source == NULL || source[0] == '\0') {
|
||||||
|
return TREXIO_INVALID_ARG_1;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (dest == NULL || dest[0] == '\0') {
|
||||||
|
return TREXIO_INVALID_ARG_2;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
back_end_t back_end_local = TREXIO_AUTO;
|
||||||
|
/* Try to determine the applicable backend if the back_end argument is TREXIO_AUTO */
|
||||||
|
#ifdef HAVE_HDF5
|
||||||
|
/* Check if the TREXIO file exists and if it is a directory */
|
||||||
|
trexio_exit_code rc_text = trexio_text_inquire(source);
|
||||||
|
if (rc_text == TREXIO_SUCCESS) {
|
||||||
|
back_end_local = TREXIO_TEXT;
|
||||||
|
} else {
|
||||||
|
/* If not, check if it is an HDF5 file */
|
||||||
|
trexio_exit_code rc_hdf5 = trexio_hdf5_inquire(source);
|
||||||
|
if (rc_hdf5 == TREXIO_SUCCESS) {
|
||||||
|
back_end_local = TREXIO_HDF5;
|
||||||
|
} else {
|
||||||
|
/* File is neither a directory nor an HDF5 file -> return an error */
|
||||||
|
return TREXIO_FILE_ERROR;
|
||||||
|
}
|
||||||
|
#else
|
||||||
|
/* In the current implementation if HDF5 back end is not available - then there is only back end left */
|
||||||
|
back_end_local = TREXIO_TEXT;
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
assert (back_end_local != TREXIO_AUTO);
|
||||||
|
|
||||||
|
if (trexio_inquire(dest) == TREXIO_SUCCESS) {
|
||||||
|
/* Destination file already exists */
|
||||||
|
return TREXIO_FILE_ERROR;
|
||||||
|
};
|
||||||
|
|
||||||
|
/* Call cp */
|
||||||
|
|
||||||
|
pid_t pid = fork();
|
||||||
|
if (pid == 0) {
|
||||||
|
switch (back_end_local) {
|
||||||
|
case TREXIO_TEXT:
|
||||||
|
execl("/bin/cp", "/bin/cp", "-r", "-n", source, dest, (char *)0);
|
||||||
|
case TREXIO_HDF5:
|
||||||
|
#ifdef HAVE_HDF5
|
||||||
|
execl("/bin/cp", "/bin/cp", "-n", source, dest, (char *)0);
|
||||||
|
#else
|
||||||
|
return TREXIO_FILE_ERROR;
|
||||||
|
#endif
|
||||||
|
/*
|
||||||
|
case TREXIO_JSON:
|
||||||
|
execl("/bin/cp", "/bin/cp", source, dest, (char *)0);
|
||||||
|
,*/
|
||||||
|
default:
|
||||||
|
return TREXIO_FILE_ERROR;
|
||||||
|
}
|
||||||
|
} else if (pid < 0) {
|
||||||
|
return TREXIO_FILE_ERROR;
|
||||||
|
} else {
|
||||||
|
int wstatus;
|
||||||
|
pid_t ws = waitpid( pid, &wstatus, 0);
|
||||||
|
if (ws != pid || !WIFEXITED(wstatus) )
|
||||||
|
return TREXIO_FILE_ERROR;
|
||||||
|
}
|
||||||
|
|
||||||
|
return TREXIO_SUCCESS;
|
||||||
|
}
|
||||||
|
#+end_src
|
||||||
|
|
||||||
|
*** Fortran
|
||||||
|
|
||||||
|
#+begin_src f90 :tangle prefix_fortran.f90
|
||||||
|
interface
|
||||||
|
integer(trexio_exit_code) function trexio_cp (source, destination) bind(C, name="trexio_cp")
|
||||||
|
use, intrinsic :: iso_c_binding
|
||||||
|
import
|
||||||
|
character(kind=c_char), dimension(*) :: source, destination
|
||||||
|
end function trexio_cp
|
||||||
|
end interface
|
||||||
|
#+end_src
|
||||||
|
|
||||||
|
*** Python
|
||||||
|
|
||||||
|
#+begin_src python :tangle basic_python.py
|
||||||
|
def _cp(source: str, destination: str):
|
||||||
|
"""Copies a TREXIO file
|
||||||
|
|
||||||
|
Parameters:
|
||||||
|
|
||||||
|
source: str
|
||||||
|
Name of the source file
|
||||||
|
|
||||||
|
destination: str
|
||||||
|
Name of the destination file
|
||||||
|
|
||||||
|
Examples:
|
||||||
|
>>> from trexio import cp as tr_cp
|
||||||
|
>>> tr_cp("example.h5", "new.h5")
|
||||||
|
"""
|
||||||
|
|
||||||
|
from shutil import copytree, copyfile
|
||||||
|
|
||||||
|
f = File(filename=source, mode='r', back_end=TREXIO_AUTO)
|
||||||
|
if f.back_end == TREXIO_TEXT:
|
||||||
|
copytree(source, destination)
|
||||||
|
elif f.back_end == TREXIO_HDF5:
|
||||||
|
copyfile(source, destination)
|
||||||
|
#+end_src
|
||||||
|
|
||||||
** File state
|
** File state
|
||||||
|
|
||||||
**Note:** the use of the functions below is discouraged as of version 2.3.0.
|
**Note:** the use of the functions below is discouraged as of version 2.3.0.
|
||||||
@ -1676,7 +1812,7 @@ end interface
|
|||||||
*** Python
|
*** Python
|
||||||
|
|
||||||
See TREXIO File Python class.
|
See TREXIO File Python class.
|
||||||
|
|
||||||
** Tasks to be done before closing
|
** Tasks to be done before closing
|
||||||
|
|
||||||
#+begin_src c :tangle trexio_private.h :exports none
|
#+begin_src c :tangle trexio_private.h :exports none
|
||||||
|
@ -18,19 +18,25 @@ int main() {
|
|||||||
|
|
||||||
bool have_hdf5 = trexio_has_backend(TREXIO_HDF5);
|
bool have_hdf5 = trexio_has_backend(TREXIO_HDF5);
|
||||||
if(have_hdf5) {
|
if(have_hdf5) {
|
||||||
rc = system("rm -f -- test_all.h5");
|
rc = system("rm -f -- test_all.h5 test_all2.h5");
|
||||||
assert (rc == 0);
|
assert (rc == 0);
|
||||||
test_write("test_all.h5", TREXIO_HDF5);
|
test_write("test_all.h5" , TREXIO_HDF5);
|
||||||
test_read ("test_all.h5", TREXIO_HDF5);
|
rc = trexio_cp ("test_all.h5", "test_all2.h5");
|
||||||
rc = system("rm -f -- test_all.h5");
|
assert (rc == TREXIO_SUCCESS);
|
||||||
|
test_read ("test_all2.h5", TREXIO_HDF5);
|
||||||
|
rc = system("rm -f -- test_all.h5 test_all2.h5");
|
||||||
assert (rc == 0);
|
assert (rc == 0);
|
||||||
}
|
}
|
||||||
|
|
||||||
rc = system("rm -f -- test_all.dir/*.txt test_all.dir/*.txt.size test_all.dir/.lock && rm -fd -- test_all.dir");
|
rc = system("rm -f -- test_all.dir/*.txt test_all.dir/*.txt.size test_all.dir/.lock && rm -fd -- test_all.dir && \
|
||||||
|
rm -f -- test_all2.dir/*.txt test_all2.dir/*.txt.size test_all2.dir/.lock && rm -fd -- test_all2.dir");
|
||||||
assert (rc == 0);
|
assert (rc == 0);
|
||||||
test_write("test_all.dir", TREXIO_TEXT);
|
test_write("test_all.dir" , TREXIO_TEXT);
|
||||||
test_read ("test_all.dir", TREXIO_TEXT);
|
rc = trexio_cp ("test_all.dir" , "test_all2.dir");
|
||||||
rc = system("rm -f -- test_all.dir/*.txt test_all.dir/*.txt.size test_all.dir/.lock && rm -fd -- test_all.dir");
|
assert (rc == TREXIO_SUCCESS);
|
||||||
|
test_read ("test_all2.dir", TREXIO_TEXT);
|
||||||
|
rc = system("rm -f -- test_all.dir/*.txt test_all.dir/*.txt.size test_all.dir/.lock && rm -fd -- test_all.dir && \
|
||||||
|
rm -f -- test_all2.dir/*.txt test_all2.dir/*.txt.size test_all2.dir/.lock && rm -fd -- test_all2.dir");
|
||||||
assert (rc == 0);
|
assert (rc == 0);
|
||||||
|
|
||||||
return 0;
|
return 0;
|
||||||
@ -138,6 +144,7 @@ int test_write(const char* file_name, const back_end_t backend) {
|
|||||||
// open file again in 'write' mode
|
// open file again in 'write' mode
|
||||||
file = trexio_open(file_name, 'w', backend, &rc);
|
file = trexio_open(file_name, 'w', backend, &rc);
|
||||||
assert (file != NULL);
|
assert (file != NULL);
|
||||||
|
assert (rc == TREXIO_SUCCESS);
|
||||||
|
|
||||||
// write some missing blocks (e.g. if forgot last time)
|
// write some missing blocks (e.g. if forgot last time)
|
||||||
rc = trexio_write_nucleus_charge(file,charge);
|
rc = trexio_write_nucleus_charge(file,charge);
|
||||||
@ -169,6 +176,7 @@ int test_read(const char* file_name, const back_end_t backend) {
|
|||||||
// open existing file on 'read' mode [created by test_write]
|
// open existing file on 'read' mode [created by test_write]
|
||||||
file = trexio_open(file_name, 'r', TREXIO_AUTO, &rc);
|
file = trexio_open(file_name, 'r', TREXIO_AUTO, &rc);
|
||||||
assert (file != NULL);
|
assert (file != NULL);
|
||||||
|
assert (rc == TREXIO_SUCCESS);
|
||||||
|
|
||||||
// read nucleus_num
|
// read nucleus_num
|
||||||
rc = trexio_read_nucleus_num(file,&num);
|
rc = trexio_read_nucleus_num(file,&num);
|
||||||
|
Loading…
Reference in New Issue
Block a user