//------------------------------------------------------------------------------
// GrB_Matrix_extractTuples: extract all tuples from a matrix
//------------------------------------------------------------------------------

// SuiteSparse:GraphBLAS, Timothy A. Davis, (c) 2017-2025, All Rights Reserved.
// SPDX-License-Identifier: Apache-2.0

//------------------------------------------------------------------------------

// Extracts all tuples from a matrix, like [I,J,X] = find (A) in MATLAB.  If
// any parameter I, J and/or X is NULL, then that component is not extracted.
// The size of the I, J, and X arrays (those that are not NULL) is given by
// nvals, which must be at least as large as GrB_nvals (&nvals, A).  The values
// in the matrix are typecasted to the type of X, as needed.

// If any parameter I, J, and/or X is NULL, that component is not extracted.
// For example, to extract just the row and col indices, pass I and J as
// non-NULL, and X as NULL.  This is like [I,J,~] = find (A) in MATLAB.

// If A is iso and X is not NULL, the iso scalar Ax [0] is expanded into X.

#include "GB.h"
#include "extractTuples/GB_extractTuples.h"

#define GB_EXTRACT_TUPLES(function_name,ctype,xtype)                        \
GrB_Info function_name      /* extract tuples from a matrix */              \
(                                                                           \
    uint64_t *I,            /* array for returning row indices of tuples */ \
    uint64_t *J,            /* array for returning col indices of tuples */ \
    ctype *X,               /* array for returning values of tuples      */ \
    uint64_t *p_nvals,      /* I,J,X size on input; # tuples on output   */ \
    const GrB_Matrix A      /* matrix to extract tuples from             */ \
)                                                                           \
{                                                                           \
    GB_WHERE_1 (A, GB_STR(function_name) " (I, J, X, nvals, A)") ;          \
    GB_RETURN_IF_NULL (A) ;                                                 \
    GB_RETURN_IF_NULL (p_nvals) ;                                           \
    GB_BURBLE_START (GB_STR(function_name)) ;                               \
    int data_arena = A->data_arena ;                                        \
    info = GB_extractTuples (I, false, J, false, X, p_nvals, xtype, A,      \
        data_arena, Werk);                                                  \
    GB_BURBLE_END ;                                                         \
    GB_PRAGMA (omp flush)                                                   \
    return (info) ;                                                         \
}

// with 64-bit I and J arrays
GB_EXTRACT_TUPLES (GrB_Matrix_extractTuples_BOOL  , bool       , GrB_BOOL  )
GB_EXTRACT_TUPLES (GrB_Matrix_extractTuples_INT8  , int8_t     , GrB_INT8  )
GB_EXTRACT_TUPLES (GrB_Matrix_extractTuples_INT16 , int16_t    , GrB_INT16 )
GB_EXTRACT_TUPLES (GrB_Matrix_extractTuples_INT32 , int32_t    , GrB_INT32 )
GB_EXTRACT_TUPLES (GrB_Matrix_extractTuples_INT64 , int64_t    , GrB_INT64 )
GB_EXTRACT_TUPLES (GrB_Matrix_extractTuples_UINT8 , uint8_t    , GrB_UINT8 )
GB_EXTRACT_TUPLES (GrB_Matrix_extractTuples_UINT16, uint16_t   , GrB_UINT16)
GB_EXTRACT_TUPLES (GrB_Matrix_extractTuples_UINT32, uint32_t   , GrB_UINT32)
GB_EXTRACT_TUPLES (GrB_Matrix_extractTuples_UINT64, uint64_t   , GrB_UINT64)
GB_EXTRACT_TUPLES (GrB_Matrix_extractTuples_FP32  , float      , GrB_FP32  )
GB_EXTRACT_TUPLES (GrB_Matrix_extractTuples_FP64  , double     , GrB_FP64  )
GB_EXTRACT_TUPLES (GxB_Matrix_extractTuples_FC32  , GxB_FC32_t , GxB_FC32  )
GB_EXTRACT_TUPLES (GxB_Matrix_extractTuples_FC64  , GxB_FC64_t , GxB_FC64  )

//------------------------------------------------------------------------------
// GrB_Matrix_extractTuples_UDT: extract from a matrix with user-defined type
//------------------------------------------------------------------------------

GrB_Info GrB_Matrix_extractTuples_UDT
(
    uint64_t *I,            // array for returning row indices of tuples
    uint64_t *J,            // array for returning col indices of tuples
    void *X,                // array for returning values of tuples
    uint64_t *p_nvals,      // I,J,X size on input; # tuples on output
    const GrB_Matrix A      // matrix to extract tuples from
)
{ 
    GB_WHERE_1 (A, "GrB_Matrix_extractTuples_UDT (I, J, X, nvals, A)") ;
    GB_RETURN_IF_NULL (A) ;
    GB_RETURN_IF_NULL (p_nvals) ;
    GB_BURBLE_START ("GrB_Matrix_extractTuples_UDT") ;

    if (A->type->code != GB_UDT_code)
    { 
        // A must have a user-defined type
        return (GrB_DOMAIN_MISMATCH) ;
    }
    int data_arena = A->data_arena ;
    info = GB_extractTuples (I, false, J, false, X, p_nvals, A->type, A,
        data_arena, Werk) ;
    GB_BURBLE_END ;
    GB_PRAGMA (omp flush)
    return (info) ;
}

