/*
 * Copyright (c) 2004-2005 The Trustees of Indiana University and Indiana
 *                         University Research and Technology
 *                         Corporation.  All rights reserved.
 * Copyright (c) 2004-2005 The University of Tennessee and The University
 *                         of Tennessee Research Foundation.  All rights
 *                         reserved.
 * Copyright (c) 2004-2005 High Performance Computing Center Stuttgart,
 *                         University of Stuttgart.  All rights reserved.
 * Copyright (c) 2004-2005 The Regents of the University of California.
 *                         All rights reserved.
 * Copyright (c) 2015-2021 Research Organization for Information Science
 *                         and Technology (RIST).  All rights reserved.
 * Copyright (c) 2017      IBM Corporation.  All rights reserved.
 * $COPYRIGHT$
 *
 * Additional copyrights may follow
 *
 * $HEADER$
 */

#include "ompi_config.h"
#include "mpi.h"
#include "ompi/file/file.h"
#include "io_romio341.h"


int
mca_io_romio341_file_write_at (ompi_file_t *fh,
                            MPI_Offset offset,
                            const void *buf,
                            int count,
                            struct ompi_datatype_t *datatype,
                            ompi_status_public_t * status)
{
    int         ret;
    mca_io_romio341_data_t *data;

    data = (mca_io_romio341_data_t *) fh->f_io_selected_data;
    OPAL_THREAD_LOCK (&mca_io_romio341_mutex);
    ret =
        ROMIO_PREFIX(MPI_File_write_at) (data->romio_fh, offset, buf, count,
                                        datatype, status);
    OPAL_THREAD_UNLOCK (&mca_io_romio341_mutex);

    return ret;
}



int
mca_io_romio341_file_write_at_all (ompi_file_t *fh,
                                MPI_Offset offset,
                                const void *buf,
                                int count,
                                struct ompi_datatype_t *datatype,
                                ompi_status_public_t * status)
{
    int         ret;
    mca_io_romio341_data_t *data;

    data = (mca_io_romio341_data_t *) fh->f_io_selected_data;
    OPAL_THREAD_LOCK (&mca_io_romio341_mutex);
    ret =
        ROMIO_PREFIX(MPI_File_write_at_all) (data->romio_fh, offset, buf,
                                             count, datatype, status);
    OPAL_THREAD_UNLOCK (&mca_io_romio341_mutex);

    return ret;
}



int
mca_io_romio341_file_iwrite_at (ompi_file_t *fh,
                             MPI_Offset offset,
                             const void *buf,
                             int count,
                             struct ompi_datatype_t *datatype,
                             ompi_request_t **request)
{
    int ret;
    mca_io_romio341_data_t *data;

    data = (mca_io_romio341_data_t *) fh->f_io_selected_data;
    OPAL_THREAD_LOCK (&mca_io_romio341_mutex);
    ret =
        ROMIO_PREFIX(MPI_File_iwrite_at) (data->romio_fh, offset, buf, count,
                                          datatype, request);
    OPAL_THREAD_UNLOCK (&mca_io_romio341_mutex);

    return ret;
}


int
mca_io_romio341_file_iwrite_at_all (ompi_file_t *fh,
                                    MPI_Offset offset,
                                    const void *buf,
                                    int count,
                                    struct ompi_datatype_t *datatype,
                                    ompi_request_t **request)
{
    int         ret;
    mca_io_romio341_data_t *data;

    data = (mca_io_romio341_data_t *) fh->f_io_selected_data;
    OPAL_THREAD_LOCK (&mca_io_romio341_mutex);
    // ----------------------------------------------------
    // NOTE: If you upgrade ROMIO, replace this with the actual ROMIO call.
    // ----------------------------------------------------
    // No support for non-blocking collective I/O operations.
    // Fake it with individual non-blocking I/O operations.
    // Similar to OMPIO
    ret =
        ROMIO_PREFIX(MPI_File_iwrite_at_all) (data->romio_fh, offset, buf, count,
                                             datatype, request);
    OPAL_THREAD_UNLOCK (&mca_io_romio341_mutex);

    return ret;
}



int
mca_io_romio341_file_write (ompi_file_t *fh,
                         const void *buf,
                         int count,
                         struct ompi_datatype_t *datatype,
                         ompi_status_public_t * status)
{
    int         ret;
    mca_io_romio341_data_t *data;

    data = (mca_io_romio341_data_t *) fh->f_io_selected_data;
    OPAL_THREAD_LOCK (&mca_io_romio341_mutex);
    ret =
        ROMIO_PREFIX(MPI_File_write) (data->romio_fh, buf, count, datatype,
                                     status);
    OPAL_THREAD_UNLOCK (&mca_io_romio341_mutex);

    return ret;
}

int
mca_io_romio341_file_write_all (ompi_file_t *fh,
                             const void *buf,
                             int count,
                             struct ompi_datatype_t *datatype,
                             ompi_status_public_t * status)
{
    int         ret;
    mca_io_romio341_data_t *data;

    data = (mca_io_romio341_data_t *) fh->f_io_selected_data;
    OPAL_THREAD_LOCK (&mca_io_romio341_mutex);
    ret =
        ROMIO_PREFIX(MPI_File_write_all) (data->romio_fh, buf, count, datatype,
                                         status);
    OPAL_THREAD_UNLOCK (&mca_io_romio341_mutex);

    return ret;
}

int
mca_io_romio341_file_iwrite (ompi_file_t *fh,
                          const void *buf,
                          int count,
                          struct ompi_datatype_t *datatype,
                          ompi_request_t **request)
{
    int ret;
    mca_io_romio341_data_t *data;

    data = (mca_io_romio341_data_t *) fh->f_io_selected_data;
    OPAL_THREAD_LOCK (&mca_io_romio341_mutex);
    ret =
        ROMIO_PREFIX(MPI_File_iwrite) (data->romio_fh, buf, count, datatype,
                                       request);
    OPAL_THREAD_UNLOCK (&mca_io_romio341_mutex);

    return ret;
}

int
mca_io_romio341_file_iwrite_all (ompi_file_t *fh,
                                const void *buf,
                                 int count,
                                 struct ompi_datatype_t *datatype,
                                 ompi_request_t **request)
{
    int         ret;
    mca_io_romio341_data_t *data;

    data = (mca_io_romio341_data_t *) fh->f_io_selected_data;
    OPAL_THREAD_LOCK (&mca_io_romio341_mutex);
    // ----------------------------------------------------
    // NOTE: If you upgrade ROMIO, replace this with the actual ROMIO call.
    // ----------------------------------------------------
    // No support for non-blocking collective I/O operations.
    // Fake it with individual non-blocking I/O operations.
    // Similar to OMPIO
    ret =
        ROMIO_PREFIX(MPI_File_iwrite_all) (data->romio_fh, buf, count, datatype,
                                          request);
    OPAL_THREAD_UNLOCK (&mca_io_romio341_mutex);

    return ret;
}


int
mca_io_romio341_file_write_shared (ompi_file_t *fh,
                                const void *buf,
                                int count,
                                struct ompi_datatype_t *datatype,
                                ompi_status_public_t * status)
{
    int         ret;
    mca_io_romio341_data_t *data;

    data = (mca_io_romio341_data_t *) fh->f_io_selected_data;
    OPAL_THREAD_LOCK (&mca_io_romio341_mutex);
    ret =
        ROMIO_PREFIX(MPI_File_write_shared) (data->romio_fh, buf, count,
                                             datatype, status);
    OPAL_THREAD_UNLOCK (&mca_io_romio341_mutex);

    return ret;
}

int
mca_io_romio341_file_iwrite_shared (ompi_file_t *fh,
                                 const void *buf,
                                 int count,
                                 struct ompi_datatype_t *datatype,
                                 ompi_request_t **request)
{
    int ret;
    mca_io_romio341_data_t *data;

    data = (mca_io_romio341_data_t *) fh->f_io_selected_data;
    OPAL_THREAD_LOCK (&mca_io_romio341_mutex);
    ret =
        ROMIO_PREFIX(MPI_File_iwrite_shared) (data->romio_fh, buf, count,
                                              datatype, request);
    OPAL_THREAD_UNLOCK (&mca_io_romio341_mutex);

    return ret;
}

int
mca_io_romio341_file_write_ordered (ompi_file_t *fh,
                                 const void *buf,
                                 int count,
                                 struct ompi_datatype_t *datatype,
                                 ompi_status_public_t * status)
{
    int         ret;
    mca_io_romio341_data_t *data;

    data = (mca_io_romio341_data_t *) fh->f_io_selected_data;
    OPAL_THREAD_LOCK (&mca_io_romio341_mutex);
    ret =
        ROMIO_PREFIX(MPI_File_write_ordered) (data->romio_fh, buf, count,
                                             datatype, status);
    OPAL_THREAD_UNLOCK (&mca_io_romio341_mutex);

    return ret;
}

int
mca_io_romio341_file_write_at_all_begin (ompi_file_t *fh,
                                      MPI_Offset offset,
                                      const void *buf,
                                      int count,
                                      struct ompi_datatype_t *datatype)
{
    int         ret;
    mca_io_romio341_data_t *data;

    data = (mca_io_romio341_data_t *) fh->f_io_selected_data;
    OPAL_THREAD_LOCK (&mca_io_romio341_mutex);
    ret = ROMIO_PREFIX(MPI_File_write_at_all_begin) (data->romio_fh, offset,
                                                     buf, count, datatype);
    OPAL_THREAD_UNLOCK (&mca_io_romio341_mutex);

    return ret;
}

int
mca_io_romio341_file_write_at_all_end (ompi_file_t *fh,
                                    const void *buf,
                                    ompi_status_public_t * status)
{
    int         ret;
    mca_io_romio341_data_t *data;

    data = (mca_io_romio341_data_t *) fh->f_io_selected_data;
    OPAL_THREAD_LOCK (&mca_io_romio341_mutex);
    ret = ROMIO_PREFIX(MPI_File_write_at_all_end) (data->romio_fh, buf,
                                                   status);
    OPAL_THREAD_UNLOCK (&mca_io_romio341_mutex);

    return ret;
}

int
mca_io_romio341_file_write_all_begin (ompi_file_t *fh,
                                   const void *buf,
                                   int count,
                                   struct ompi_datatype_t *datatype)
{
    int         ret;
    mca_io_romio341_data_t *data;

    data = (mca_io_romio341_data_t *) fh->f_io_selected_data;
    OPAL_THREAD_LOCK (&mca_io_romio341_mutex);
    ret = ROMIO_PREFIX(MPI_File_write_all_begin) (data->romio_fh, buf, count,
                                                 datatype);
    OPAL_THREAD_UNLOCK (&mca_io_romio341_mutex);

    return ret;
}

int
mca_io_romio341_file_write_all_end (ompi_file_t *fh,
                                 const void *buf,
                                 ompi_status_public_t * status)
{
    int         ret;
    mca_io_romio341_data_t *data;

    data = (mca_io_romio341_data_t *) fh->f_io_selected_data;
    OPAL_THREAD_LOCK (&mca_io_romio341_mutex);
    ret = ROMIO_PREFIX(MPI_File_write_all_end) (data->romio_fh, buf, status);
    OPAL_THREAD_UNLOCK (&mca_io_romio341_mutex);

    return ret;
}

int
mca_io_romio341_file_write_ordered_begin (ompi_file_t *fh,
                                       const void *buf,
                                       int count,
                                       struct ompi_datatype_t *datatype)
{
    int         ret;
    mca_io_romio341_data_t *data;

    data = (mca_io_romio341_data_t *) fh->f_io_selected_data;
    OPAL_THREAD_LOCK (&mca_io_romio341_mutex);
    ret = ROMIO_PREFIX(MPI_File_write_ordered_begin) (data->romio_fh, buf,
                                                     count, datatype);
    OPAL_THREAD_UNLOCK (&mca_io_romio341_mutex);

    return ret;
}

int
mca_io_romio341_file_write_ordered_end (ompi_file_t *fh,
                                     const void *buf,
                                     ompi_status_public_t * status)
{
    int         ret;
    mca_io_romio341_data_t *data;

    data = (mca_io_romio341_data_t *) fh->f_io_selected_data;
    OPAL_THREAD_LOCK (&mca_io_romio341_mutex);
    ret = ROMIO_PREFIX(MPI_File_write_ordered_end) (data->romio_fh, buf,
                                                    status);
    OPAL_THREAD_UNLOCK (&mca_io_romio341_mutex);

    return ret;
}