#!/bin/bash

# Update the Rust bindings to GPU code (via a header). This script must be run
# whenever the GPU code changes.

# This script requires bindgen. This can be provided by a package manager or
# installed with "cargo install bindgen-cli".

# https://stackoverflow.com/questions/4774054/reliable-way-for-a-bash-script-to-get-the-full-path-to-itself
SCRIPTPATH="$(cd -- "$(dirname "$0")" >/dev/null 2>&1 ; pwd -P)"

for PRECISION in SINGLE DOUBLE; do
    LOWER_CASE=$(echo "${PRECISION}" | tr '[:upper:]' '[:lower:]')
    echo "Generating bindings for ${LOWER_CASE}-precision GPU code"

    bindgen "${SCRIPTPATH}"/fee/gpu/fee.h \
        --allowlist-function "gpu_fee_calc_jones.*" \
        --allowlist-type "FEECoeffs" \
        -- -D BINDGEN -D "${PRECISION}" \
        -I "${SCRIPTPATH}"/gpu_common/ \
        > "${SCRIPTPATH}/fee/gpu/${LOWER_CASE}.rs"

    bindgen "${SCRIPTPATH}"/analytic/gpu/analytic.h \
        --allowlist-function "gpu_analytic_calc_jones.*" \
        --allowlist-type "AnalyticType" \
        -- -D BINDGEN -D "${PRECISION}" \
        -I "${SCRIPTPATH}"/gpu_common/ \
        > "${SCRIPTPATH}/analytic/gpu/${LOWER_CASE}.rs"
done