//------------------------------------------------------------------------------
// GxB_BinaryOp_new_arena: create a new user-defined binary operator
//------------------------------------------------------------------------------

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

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

// Create a new a binary operator: z = f (x,y).  The binary function signature
// must be void f (void *z, const void *x, const void *y), and then it must
// recast its input and output arguments internally as needed.

// If the function pointer is NULL, the function is compiled with the JIT.

#include "GB.h"
#include "binaryop/GB_binop.h"
#include "jitifyer/GB_stringify.h"

#define GB_FREE_ALL GB_Op_free ((GB_Operator *) &op) ;

GrB_Info GxB_BinaryOp_new_arena
(
    GrB_BinaryOp *op_handle,        // handle for the new binary operator
    GxB_binary_function function,   // pointer to the binary function
    GrB_Type ztype,                 // type of output z
    GrB_Type xtype,                 // type of input x
    GrB_Type ytype,                 // type of input y
    const char *binop_name,         // name of the user function
    const char *binop_defn,         // definition of the user function
    const int header_arena
)
{

    //--------------------------------------------------------------------------
    // check inputs
    //--------------------------------------------------------------------------

    GrB_Info info ;
    GB_CHECK_INIT ;
    GB_RETURN_IF_NULL (op_handle) ;
    (*op_handle) = NULL ;
    GrB_BinaryOp op = NULL ;
    GB_RETURN_IF_NULL_OR_FAULTY (ztype) ;
    GB_RETURN_IF_NULL_OR_FAULTY (xtype) ;
    GB_RETURN_IF_NULL_OR_FAULTY (ytype) ;
    GB_OK (GB_check_arena (header_arena)) ;

    //--------------------------------------------------------------------------
    // allocate the binary op
    //--------------------------------------------------------------------------

    uint64_t mem = GB_mem (header_arena, 0) ;
    uint64_t header_mem = mem ;
    op = GB_CALLOC_MEMORY (1, sizeof (struct GB_BinaryOp_opaque), &header_mem) ;
    if (op == NULL)
    { 
        // out of memory
        return (GrB_OUT_OF_MEMORY) ;
    }
    op->header_mem = header_mem ;

    //--------------------------------------------------------------------------
    // create the binary op
    //--------------------------------------------------------------------------

    GB_OK (GB_binop_new (op, function, ztype, xtype, ytype,
        binop_name, binop_defn, GB_USER_binop_code, header_arena)) ;

    //--------------------------------------------------------------------------
    // create the function pointer, if NULL
    //--------------------------------------------------------------------------

    if (function == NULL)
    { 
        GB_BURBLE_START ("GxB_BinaryOp_new") ;
        void *user_function ;
        info = GB_user_op_jit (&user_function, (GB_Operator) op) ;
        if (info != GrB_SUCCESS)
        { 
            // unable to construct the function pointer
            GB_FREE_ALL ;
            // If the JIT fails, it returns GrB_NO_VALUE or GxB_JIT_ERROR.
            // Convert GrB_NO_VALUE to GrB_NULL_POINTER (the function is NULL
            // and cannot be compiled by the JIT).
            return (info == GrB_NO_VALUE ? GrB_NULL_POINTER : info) ;
        }
        #include "include/GB_pedantic_disable.h"
        op->binop_function = (GxB_binary_function) user_function ;
        GB_BURBLE_END ;
    }

    (*op_handle) = op ;
    return (GrB_SUCCESS) ;
}

