#!/bin/bash

#####################################################################
#                                                                   #
#    Generates an API for Android or iOS based on a Faust object    #
#               (c) Romain Michon CCRMA and Grame, 2016-2024        #
#                                                                   #
#####################################################################

. faustpath
. faustoptflags
. usage.sh

CXXFLAGS+=" $MYGCCFLAGS"  # So that additional CXXFLAGS can be used

# change if you want to get the log of what's happening
LOG="/dev/null"

# exit if a command fails
set -e

# global option variables

NVOICES="0"
#PACKAGENAME="0"
POLY2="0"
MIDI="0"
SOUNDFILE="0"
OSC="0"
NODOC="0"
NOZIP="0"
APIFOLDER="dsp-faust"
EFFECT=""
BUILD="0"
DYNAMIC_DSP="0"
OPT=""
MISC_OPT=""
TARGETDEFINED="0"

COREAUDIO_DRIVER="0"
ANDROID_DRIVER="0"
IOS_DRIVER="0"
ALSA_DRIVER="0"
JACK_DRIVER="0"
PORTAUDIO_DRIVER="0"
RTAUDIO_DRIVER="0"
MINIAUDIO_DRIVER="0"
COREAUDIO_DRIVER="0"
OPEN_FRAMEWORK_DRIVER="0"
JUCE_DRIVER="0"
DUMMY_DRIVER="0"
TEENSY_DRIVER="0"
ESP32_DRIVER="0"

if [[ $(uname -s) == Darwin ]]; then
    ARCHLIB+=" -framework CoreMIDI -framework CoreFoundation -framework CoreAudio -framework AudioUnit -framework CoreServices "
else
    ARCHLIB=""
fi

echoHelp() 
{
    usage faust2api "[options] [Faust options] <file.dsp>"
    echo "faust2api can be used to generate Faust based dsp objects for various platforms (see https://github.com/grame-cncm/faust/tree/master-dev/architecture/api):"
    echo
    echo "Output options:"
    option -ios "generates an iOS API"
    option -android "generates an Android API"
    option -coreaudio "generates an OSX CoreAudio API"
    option -alsa "generates an ALSA API"
    option -jack "generates a JACK API"
    option -portaudio "generates a PortAudio API"
    option -rtaudio "generates an RTAudio API"
    option -miniaudio "generates a Miniaudio API"
    option -of "generates an openFrameworks API"
    option -juce "generates a JUCE API"
    option -dummy "generates a dummy audio API"
    option -teensy "generates a Teensy API"
    option -esp32 "generates a ESP32 API"
    echo
    echo "Global options:"
    option "-opt native|generic" "activates the best compilation options for the native or generic CPU"
    option "-nvoices <num>" "creates a polyphonic object with <num> voices"
    option "-effect <effect.dsp>" "adds an effect to the polyphonic synth (ignored if -nvoices is not specified)"
    option "-effect auto" "adds an effect (extracted automatically from the dsp file) to the polyphonic synth (ignored if -nvoices is not specified)"
    option -nodoc "prevents documentation from being generated"
    option -nozip "prevents generated files to be put in a zip file"
    option "-target <target>" "sets a name of the target folder or the zip file. Defaults to \"dsp-faust\""
    option "Faust options"
    echo
    echo "Android specific options:"
    option -package "set the JAVA package name (e.g. '-package mypackage' will change the JAVA package name to 'mypackage.DspFaust'). The default package name is 'com.DspFaust.'"
    echo
    echo "Options supported by iOS, CoreAudio, ALSA, JACK, PortAudio, RTAudio, Miniaudio, openFrameworks and JUCE"
    option -midi "add built-in RtMidi support to the API"
    option -osc "add built-in OSC support to the API"
    option -soundfile "add built-in Soundfile support to the API"
    echo
    echo "JACK specific options"
    option -build "build a ready to test binary"
    option -dynamic "add libfaust/LLVM dynamic DSP compilation mode"
    exit
}

if [ "$#" -eq 0 ]; then
    echo 'Please, provide a Faust file to process !'
    echo ''
    echoHelp
fi

createAPI ()
{
    cp -r $FAUSTARCH/api/DspFaust.h $APIFOLDER
    faust $OPTIONS $MISC_OPT -i -a api/DspFaust.cpp "$FILE" -o "$APIFOLDER/DspFaust.cpp" || exit 1
    if [ "$POLY2" = "1" ]; then
        faust $OPTIONS -i -cn effect -a minimal-effect.cpp $EFFECT | cat - "$APIFOLDER/DspFaust.cpp" > temp && mv temp "$APIFOLDER/DspFaust.cpp" || exit 1
        echo "#define POLY2 1" | cat - "$APIFOLDER/DspFaust.cpp" > temp && mv temp "$APIFOLDER/DspFaust.cpp"
    fi
    if [ "$NVOICES" -gt "0" ]; then
        echo "#define NVOICES $NVOICES" | cat - "$APIFOLDER/DspFaust.cpp" > temp && mv temp "$APIFOLDER/DspFaust.cpp"
    fi
    if [ "$MIDI" = "1" ]; then
        echo "#define MIDICTRL 1" | cat - "$APIFOLDER/DspFaust.cpp" > temp && mv temp "$APIFOLDER/DspFaust.cpp"
    fi
    if [ "$OSC" = "1" ]; then
        echo "#define OSCCTRL 1" | cat - "$APIFOLDER/DspFaust.cpp" > temp && mv temp "$APIFOLDER/DspFaust.cpp"
    fi
    if [ "$SOUNDFILE" = "1" ]; then
        echo "#define SOUNDFILE 1" | cat - "$APIFOLDER/DspFaust.cpp" > temp && mv temp "$APIFOLDER/DspFaust.cpp"
    fi
    if [ "$DYNAMIC_DSP" = "1" ]; then
        echo "#define DYNAMIC_DSP 1" | cat - "$APIFOLDER/DspFaust.cpp" > temp && mv temp "$APIFOLDER/DspFaust.cpp"
    fi
    if [ "$BUILD" = "1" ]; then
        echo "#define BUILD 1" | cat - "$APIFOLDER/DspFaust.cpp" > temp && mv temp "$APIFOLDER/DspFaust.cpp"
    fi
}

# dispatch command arguments
while [ $1 ]
do
    p=$1

    if [ $p = "-help" ] || [ $p = "-h" ]; then
        echoHelp
    elif [ $p = "-opt" ]; then
        shift
        OPT=$1
    elif [ "$PACKAGENAME" = "-1" ]; then
        PACKAGENAME="$p"
    elif [[ -f "$p" ]] && [ ${p: -4} == ".dsp" ]; then
        FILE="$p"
    elif [ $p = "-android" ]; then
        ANDROID_DRIVER=1
    elif [ $p = "-ios" ]; then
        IOS_DRIVER="1"
    elif [ $p = "-coreaudio" ]; then
        COREAUDIO_DRIVER=1
    elif [ $p = "-alsa" ]; then
        ALSA_DRIVER=1
    elif [ $p = "-jack" ]; then
        JACK_DRIVER=1
    elif [ $p = "-portaudio" ]; then
        PORTAUDIO_DRIVER=1
    elif [ $p = "-rtaudio" ]; then
        RTAUDIO_DRIVER=1
    elif [ $p = "-miniaudio" ]; then
        MINIAUDIO_DRIVER=1
    elif [ $p = "-of" ]; then
        OPEN_FRAMEWORK_DRIVER=1
    elif [ $p = "-juce" ]; then
        JUCE_DRIVER=1
    elif [ $p = "-dummy" ]; then
        DUMMY_DRIVER=1
    elif [ $p = "-teensy" ]; then
        TEENSY_DRIVER=1
        MISC_OPT="-uim"
    elif [ $p = "-esp32" ]; then
        ESP32_DRIVER=1
        MISC_OPT="-uim"
    elif [ "$p" = "-effect" ]; then
        POLY2="1"
        shift
        EFFECT=$1
    elif [ "$p" = "-midi" ]; then
        MIDI="1"
    elif [ "$p" = "-soundfile" ]; then
        SOUNDFILE="1"
    elif [ "$p" = "-osc" ]; then
        OSC="1"
    elif [ "$p" = "-nodoc" ]; then
        NODOC="1"
    elif [ "$p" = "-nozip" ]; then
        NOZIP="1"
    elif [ "$p" = "-target" ]; then
        TARGETDEFINED="1"
        shift
        APIFOLDER=$1
    elif [ "$p" = "-build" ]; then
        BUILD="1"
    elif [ "$p" = "-dynamic" ]; then
        DYNAMIC_DSP="1"
    elif [ $p = "-nvoices" ]; then
        shift
        NVOICES=$1
    elif [ $p = "-package" ]; then
        PACKAGENAME="-1"
    elif [ ${p:0:1} = "-" ]; then
        OPTIONS="$OPTIONS $p"
    elif [[ -f "$p" ]]; then
        FILES="$FILES $p"
    else
        OPTIONS="$OPTIONS $p"
    fi
    shift
done

###################################
# GENERATING API
###################################

rm -r "$APIFOLDER" 2> /dev/null || true
mkdir "$APIFOLDER"

#---------------------------------------------------------
# discover best compilation options
# discover best compilation options
if [ "$OPT" = "generic" ]; then
    echo "Look for best compilation options in 'generic' mode..."
    OPTIONS="$OPTIONS $(faustbench-llvm -notrace -generic $OPTIONS $f)"
    echo $OPTIONS
elif [ "$OPT" = "native" ]; then
    echo "Look for best compilation options in 'native' mode..."
    OPTIONS="$OPTIONS $(faustbench-llvm -notrace $OPTIONS $f)"
    echo $OPTIONS
fi

#---------------------------------------------------------
# START CHECKING FOR AUTO EFFECT
if [ "$NVOICES" -gt "0" ]; then
if [ "$EFFECT" = "auto" ]; then
    EFFECT="$APIFOLDER/autoeffect.dsp"
    cat > "$EFFECT" << EndOfCode
    adapt(1,1) = _;
    adapt(2,2) = _,_;
    adapt(1,2) = _ <: _,_;
    adapt(2,1) = _,_ :> _;
    adaptor(F,G) = adapt(outputs(F),inputs(G));
    process = adaptor(library("$FILE").process, library("$FILE").effect) : library("$FILE").effect;
EndOfCode
    # We check this is a faust valid code. No error if it is not.
    faust $EFFECT -o /dev/null 2> /dev/null || POLY2="0" EFFECT=""
fi
else
    # No polyphony, make sure no effect
    POLY2="0" EFFECT=""
fi

APILOCATION=""

if [ "$NOZIP" = "0" ]; then
    APILOCATION="$APIFOLDER.zip"
elif [ "$TARGETDEFINED" = "1" ]; then
    APILOCATION="$APIFOLDER/"
fi

# END CHECKING FOR AUTO EFFECT

if [ $ANDROID_DRIVER -eq 1 ]; then
    echo "Your $APILOCATION API package for Android is being created"
    CPPFOLDER="$APIFOLDER/cpp"
    JAVAFOLDER="$APIFOLDER/java"
    mkdir $CPPFOLDER
    mkdir $JAVAFOLDER
    cp -r $FAUSTARCH/api/android/jni/*.java $JAVAFOLDER
    if [ -n "$PACKAGENAME" ]; then
        PACKAGENAME="$PACKAGENAME.DspFaust"
        sed -i "s/com.DspFaust/$PACKAGENAME/g" $JAVAFOLDER/*.java
    fi
    cp -r $FAUSTARCH/api/android/jni/java_interface_wrap.cpp $CPPFOLDER
    createAPI
    mv $APIFOLDER/DspFaust.h $CPPFOLDER
    mv $APIFOLDER/DspFaust.cpp $CPPFOLDER
    echo "#define ANDROID_DRIVER 1" | cat - "$CPPFOLDER/DspFaust.cpp" > temp && mv temp "$CPPFOLDER/DspFaust.cpp"
elif [ $IOS_DRIVER -eq 1 ]; then
    echo "Your $APILOCATION API package for iOS is being created"
    createAPI
    echo "#define IOS_DRIVER 1" | cat - "$APIFOLDER/DspFaust.cpp" > temp && mv temp "$APIFOLDER/DspFaust.cpp"
elif [ $COREAUDIO_DRIVER -eq 1 ]; then
    echo "Your $APILOCATION API package for CoreAudio is being created"
    createAPI
    echo "#define COREAUDIO_DRIVER 1" | cat - "$APIFOLDER/DspFaust.cpp" > temp && mv temp "$APIFOLDER/DspFaust.cpp"
elif [ $ALSA_DRIVER -eq 1 ]; then
    echo "Your $APILOCATION API package for ALSA is being created"
    createAPI
    echo "#define ALSA_DRIVER 1" | cat - "$APIFOLDER/DspFaust.cpp" > temp && mv temp "$APIFOLDER/DspFaust.cpp"
elif [ $JACK_DRIVER -eq 1 ]; then
    echo "Your $APILOCATION API package for JACK is being created"
    createAPI
    echo "#define JACK_DRIVER 1" | cat - "$APIFOLDER/DspFaust.cpp" > temp && mv temp "$APIFOLDER/DspFaust.cpp"
elif [ $PORTAUDIO_DRIVER -eq 1 ]; then
    echo "Your $APILOCATION API package for PortAudio is being created"
    createAPI
    echo "#define PORTAUDIO_DRIVER 1" | cat - "$APIFOLDER/DspFaust.cpp" > temp && mv temp "$APIFOLDER/DspFaust.cpp"
elif [ $RTAUDIO_DRIVER -eq 1 ]; then
    echo "Your $APILOCATION API package for RTAudio is being created"
    createAPI
    echo "#define RTAUDIO_DRIVER 1" | cat - "$APIFOLDER/DspFaust.cpp" > temp && mv temp "$APIFOLDER/DspFaust.cpp"
elif [ $MINIAUDIO_DRIVER -eq 1 ]; then
    echo "Your $APILOCATION API package for Miniaudio is being created"
    createAPI
    echo "#define MINIAUDIO_DRIVER 1" | cat - "$APIFOLDER/DspFaust.cpp" > temp && mv temp "$APIFOLDER/DspFaust.cpp"
elif [ $OPEN_FRAMEWORK_DRIVER -eq 1 ]; then
    echo "Your $APILOCATION API package for OpenFramework is being created"
    createAPI
    echo "#define OPEN_FRAMEWORK_DRIVER 1" | cat - "$APIFOLDER/DspFaust.cpp" > temp && mv temp "$APIFOLDER/DspFaust.cpp"
elif [ $JUCE_DRIVER -eq 1 ]; then
    echo "Your $APILOCATION API package for JUCE is being created"
    createAPI
    echo "#define JUCE_DRIVER 1" | cat - "$APIFOLDER/DspFaust.cpp" > temp && mv temp "$APIFOLDER/DspFaust.cpp"
elif [ $DUMMY_DRIVER -eq 1 ]; then
    echo "Your $APILOCATION API package for dummy audio is being created"
    createAPI
    echo "#define DUMMY_DRIVER 1" | cat - "$APIFOLDER/DspFaust.cpp" > temp && mv temp "$APIFOLDER/DspFaust.cpp"
elif [ $TEENSY_DRIVER -eq 1 ]; then
    echo "Your $APILOCATION API package for teensy audio is being created"
    createAPI
    echo "#define TEENSY_DRIVER 1" | cat - "$APIFOLDER/DspFaust.cpp" > temp && mv temp "$APIFOLDER/DspFaust.cpp"
elif [ $ESP32_DRIVER -eq 1 ]; then
    echo "Your $APILOCATION API package for ESP32 audio is being created"
    createAPI
    echo "#define ESP32_DRIVER 1" | cat - "$APIFOLDER/DspFaust.cpp" > temp && mv temp "$APIFOLDER/DspFaust.cpp"
else
    echoHelp
    exit 1
fi

###################################
# GENERATING BINARY
###################################

if [ "$BUILD" = "1" ]; then
    if [ $JACK_DRIVER -eq 1 ]; then    
        ARCHLIB+=" $(pkg-config --cflags --static --libs jack)"
    fi
    if [ $SOUNDFILE -eq 1 ]; then
        ARCHLIB+=" $(pkg-config --cflags --static --libs sndfile)"
    fi
    if [ $DYNAMIC_DSP -eq 1 ]; then
        ARCHLIB+=" $(llvm-config --link-static --ldflags --libs all --system-libs)"
    fi
    if [ "$DYNAMIC_DSP" = "1" ]; then
        echo "Compile as a binary named 'dynamic-api'"
        # compile c++ to binary liked with libfaust/LLVM
        (
            $CXX $CXXFLAGS $APIFOLDER/DspFaust.cpp /usr/local/lib/libfaust.a -lz -lncurses -lOSCFaust $ARCHLIB -o dynamic-api
        ) > /dev/null || exit 1
        exit 0
   else
        echo "Compile the ${FILE%.dsp} binary"
        # compile c++ to binary
       
        (
            $CXX $CXXFLAGS $APIFOLDER/DspFaust.cpp -lOSCFaust $ARCHLIB -o ${FILE%.dsp}
        ) > /dev/null || exit 1
        exit 0
    fi

fi

###################################
# GENERATING API DOCUMENTATION
###################################

if [ $NODOC -eq 0 ]; then

    APIHEADERFILE="$FAUSTARCH/api/DspFaust.h"

    if [ $ANDROID_DRIVER -eq 1 ]; then
        DOCHEADERFILE="$FAUSTARCH/api/doc/Android.md"
    elif [ $IOS_DRIVER -eq 1 ]; then
        DOCHEADERFILE="$FAUSTARCH/api/doc/iOS.md"
    else
        DOCHEADERFILE="$FAUSTARCH/api/doc/Generic.md"
    fi

    PPFILE=pPrinter.cpp
    PPBINARY=pp
    READMEFILE="$APIFOLDER/README.md"

    faust $OPTIONS $MISC_OPT -i -a path-printer.cpp "$FILE" -o "$PPFILE" || exit 1
    if [ "$POLY2" = "1" ]; then
        faust $OPTIONS -i -cn effect -a minimal-effect.cpp $EFFECT | cat - "$PPFILE" > temp && mv temp "$PPFILE" || exit 1
        echo "#define POLY2 1" | cat - "$PPFILE" > temp && mv temp "$PPFILE"
    fi
    if [ "$NVOICES" -gt "0" ]; then
        echo "#define NVOICES $NVOICES" | cat - "$PPFILE" > temp && mv temp "$PPFILE"
    fi
    if [ "$SOUNDFILE" = "1" ]; then
        echo "#define SOUNDFILE 1" | cat - "$PPFILE"  > temp && mv temp "$PPFILE"
    fi

    # compile c++ to binary
    (
        if [ "$NVOICES" -gt "0" ]; then
            $CXX $CXXFLAGS -DPOLY_VOICES "$PPFILE" `pkg-config --cflags --static --libs sndfile` -pthread -o "$PPBINARY"
        else
            $CXX $CXXFLAGS "$PPFILE" `pkg-config --cflags --static --libs sndfile` -pthread -o "$PPBINARY"
        fi
    ) > /dev/null || exit 1
    rm "$PPFILE" || exit 1

    cp "$DOCHEADERFILE"  "$READMEFILE" || exit 1
    ./"$PPBINARY" | cat "$READMEFILE" - > temp && mv temp "$READMEFILE" || exit 1
    rm "$PPBINARY" || exit 1

    faust2md "$APIHEADERFILE" | cat "$READMEFILE" - > temp && mv temp "$READMEFILE" || exit 1
fi

###################################
# POST PROCESSING
###################################

if [ "$NOZIP" = "0" ]; then
    ZIPOUT="$APIFOLDER.zip"
    rm $ZIPOUT 2> /dev/null || true
    zip -r $ZIPOUT $APIFOLDER > /dev/null || exit 1
    rm -r $APIFOLDER || exit 1
elif [ "$TARGETDEFINED" = "0" ]; then
    mv $APIFOLDER/* . || exit 1
    rm -r $APIFOLDER || exit 1
fi