make

#!/bin/bash
#----------------------------------------------------------------------------
#
# Builds Ormerod Duet firmware from source
#
# - Downloads Arduino environment from arduino.cc
# - Downloads the selected firmware from github.com
# - Compiles the RepRapFirmware source code
# - Support for RepRapPro original firmware
# - Support for dc42's highly advisable fork
# - Support for zombiepantslol FTP & telnet fork
#
# Update 30/05/2014:
#
# - It only compiles changed files for faster recompilation.
# - It shows less output to keep the compilation result clean.
# - Added command line parameter 'clean' to clean project.
# - Added command line parameter 'verbose' to show output.
# - Removed the (not so interesting) ELF file statistics.
# - Probably supports the MacOS platform for jstck...
#
# Update 31/05/2014:
#
# - Added command line parameter 'install' to upload the firmware.
# - Added download of 'lsusb' utility for MacOS environment.
#
# Update 01/06/2014:
#
# - Added a check to skip the ArduinoCorePatches folder during compilation,
#   these files are only for installing, compiling them twice gives errors
#   with RepRapFirmware-059e-dc42: "multiple definition of emac_phy_read"
#
# Update 05/06/2014:
#
# - Added zombiepantslol's firmware fork with FTP and Telnet.
#   Announcement: http://forums.reprap.org/read.php?340,364934
#
# 3D-ES
#----------------------------------------------------------------------------

# Firmware repository.
FIRMWARE=RepRapFirmware

# Project folders.
LIB=${FIRMWARE}/Libraries
BUILD=${FIRMWARE}/Build
RELEASE=${FIRMWARE}/Release

# Arduino version.
ARDUINO=arduino-1.5.4
ARDUINO_VERSION=154

# Location of the ARM compiler.
GCC_ARM_DIR=${ARDUINO}/hardware/tools/g++_arm_none_eabi

# Cross compiler binary locations.
GCC=${GCC_ARM_DIR}/bin/arm-none-eabi-gcc
GPP=${GCC_ARM_DIR}/bin/arm-none-eabi-g++
COPY=${GCC_ARM_DIR}/bin/arm-none-eabi-objcopy

# Atmel SAM flash programmer utility.
BOSSAC=${ARDUINO}/hardware/tools/bossac

# Flash options.
BOSSAC_OPTIONS=(
    -e # Erase entire flash
    -w # Write file to flash
    -v # Verify write to flash
    -R # Reset CPU after flash
    -b # Boot from flash
    ${RELEASE}/${FIRMWARE}.bin
)

# Which lsusb?
if [ ! -f lsusb ]
    then LSUSB=lsusb
    else LSUSB=./lsusb
fi

# Check if there are script parameters:
#
# - Supports 'clean' to remove the build output.
# - Supports 'verbose' to display script output.
# - Supports 'install' to upload the firmware.
#
if [ $# -gt 0 ]
then
    if [ $1 == "verbose" ]
    then
        set -x
    fi

    if [ $1 == "clean" ]
    then
        echo "Cleaning ${BUILD} and ${RELEASE}"
        echo
        rm -rf ${BUILD} ${RELEASE}
        exit
    fi

    if [ $1 == "install" ]
    then
        # Do we have a firmware binary?
        if [ ! -f ${RELEASE}/${FIRMWARE}.bin ]
        then
            echo "The firmware binary has not been compiled yet."
            echo "Please run '$0' and then '$0 install'."
            echo
            exit
        fi

        # Search for the Arduino device.
        ${LSUSB} -d 2341:003e &> /dev/null

        # Arduino found?
        if [ $? != 1 ]
        then
            # Erase and reset to reconnect as Atmel.
            echo "Erase your board to start the upload."
            echo "Please press MCU_ERASE and then RESET."
            echo
        else
            # Arduino not found! And Atmel?
            ${LSUSB} -d 03eb:6124 &> /dev/null

            # Also missing?
            if [ $? == 1 ]
            then
                # Then the board is not connected..!
                echo "Your board could not be found."
                echo "Please reconnect and try again."
                echo
                exit
            fi
        fi

        while true
        do
            # Search for the Atmel device.
            ${LSUSB} -d 03eb:6124 &> /dev/null

            # Found it?
            if [ $? != 1 ]
            then
                # Give the user some time to press the reset button,
                # otherwise the board won't reset after programming.
                sleep 3

                # We should be able to program the board.
                echo "Found an empty board, starting upload."
                break
            fi
        done

        while true
        do
            # Show why we are waiting, it can take 30 seconds!
            echo "Bossac is searching, this can take a while..."

            # Search for the device.
            ${BOSSAC} -i &> /dev/null

            # Keep trying until bossac finds the device.
            if [ $? == 1 ]; then sleep 1 && continue; fi

            # Write the program to flash!
	    ${BOSSAC} ${BOSSAC_OPTIONS[@]}
            exit
        done
    fi

    # Don't accept unsupported options.
    echo "$0: unsupported option: $1";
    echo
    exit
fi

# Exit immediately if a command exits with a non-zero status.
# Above this point it is normal to have a non-zero exit status.
set -e

# Check if the RepRapFirmware folder is available:
#
# - Ask which firmware version to download.
# - Download the duet branch from the repository.
#
if [ ! -d ${FIRMWARE} ]; then
    while true; do
        echo "Which firmware version do you want to use?"
        echo ""
        echo "[R] = RepRapPro original"
        echo "[D] = dc42's excellent fork"
        echo "[Z] = zombiepantslol's FTP and telnet fork"
        echo ""
        read -p "Please answer R, D or Z: " input
        case $input in
            [Rr]* ) FW_REPO=git://github.com/RepRapPro; break;;
            [Dd]* ) FW_REPO=git://github.com/dc42; break;;
            [Zz]* ) FW_REPO=git://github.com/zombiepantslol; break;;
        esac
    done

    git clone -b duet ${FW_REPO}/${FIRMWARE} ${FIRMWARE}
fi

# Check if the Arduino folder is available.
#
if [ ! -d ${ARDUINO} ]
then
    # Test if we run on linux.
    if [[ $OSTYPE == linux-gnu ]]
    then
        # Is this PC 32 or 64 bits?
        if [ $(uname -m) == x86_64 ]
            then ARCHIVE=${ARDUINO}-linux64.tgz
            else ARCHIVE=${ARDUINO}-linux32.tgz
        fi

        # Download the archive if no cache found.
        if [ ! -f ${ARCHIVE} ]
            then wget http://downloads.arduino.cc/${ARCHIVE}
        fi

        # Extract here.
        tar -xvzf ${ARCHIVE}
    fi

    # Test if we run on MacOS.
    if [[ $OSTYPE == darwin* ]]
    then
        # Only one version available.
        ARCHIVE=${ARDUINO}-macosx.zip

        # Download the archive if no cache found.
        if [ ! -f ${ARCHIVE} ]
            then wget http://downloads.arduino.cc/${ARCHIVE}
        fi

        # Extract here.
        unzip ${ARCHIVE}

        # Make the MacOS Arduino look like a Linux Arduino.
	mv Arduino.app/Contents/Resources/Java/* Arduino.app
	mv Arduino.app ${ARDUINO}

        # Download MacOS 'lsusb' utility script from the repository.
	wget https://raw.githubusercontent.com/jlhonora/lsusb/master/lsusb

	# Energize!
	chmod +x lsusb
    fi
fi

# Check if there are ArduinoCorePatches available:
#
# - These patches are used by dc42 to fix Arduino code.
# - Always overwrite files because of git pull updates.
# - Preserves the timestamps to prevent recompilation.
#
if [ -d ${FIRMWARE}/ArduinoCorePatches ]
then
    cp -r -p ${FIRMWARE}/ArduinoCorePatches/sam ${ARDUINO}/hardware/arduino
fi

# Check if jmgiacalone's libraries are installed:
#
# - RepRapPro keeps the libraries separated from the firmware.
# - dc42's firmware contains a fork of the libraries repository.
#
if [ ! -d ${LIB} ]
then
    git clone git://github.com/jmgiacalone/Arduino-libraries ${LIB}
fi

PLATFORM=(
    -mcpu=cortex-m3
    -DF_CPU=84000000L
    -DARDUINO=${ARDUINO_VERSION}
    -D__SAM3X8E__
    -mthumb
)

USB_OPT=(
    -DUSB_PID=0x003e # Define the USB product ID
    -DUSB_VID=0x2341 # Define the USB vendor ID
    -DUSBCON # TODO: Why?
)

GCC_OPT=(
    -c # Compile the source files, but do not link
    -g # Produce debugging info in OS native format
    -O3 # Optimize for speed, makes binary larger
    -w # Inhibit all warning messages
    -ffunction-sections # Place each function into its own section
    -fdata-sections # Place each data item into its own section
    -nostdlib # Do not use standard system startup files or libraries
    --param max-inline-insns-single=500 # Max. instructions to inline
    -Dprintf=iprintf # Prevent bloat by using integer printf function
    -MMD # Create dependency output file but only of user header files
    -MP # Add phony target for each dependency other than the main file
)

GPP_OPT=(
    -c # Compile the source files, but do not link
    -g # Produce debugging info in OS native format
    -O3 # Optimize for speed, makes binary larger
    -w # Inhibit all warning messages
    -ffunction-sections # Place each function into its own section
    -fdata-sections # Place each data item into its own section
    -nostdlib # Do not use standard system startup files or libraries
    --param max-inline-insns-single=500 # Max. instructions to inline
    -Dprintf=iprintf # Prevent bloat by using integer printf function
    -MMD # Create dependency output file but only of user header files
    -MP # Add phony target for each dependency other than the main file
    -fno-rtti # Don't produce Run-Time Type Information structures
    -fno-exceptions # Disable exception handling to prevent overhead
    -x c++ # Specify the language of the input files
)

COM_OPT=(
    -Os # Optimize for size TODO: use -O3?
    -mcpu=cortex-m3 # Specify target processor
    -T${ARDUINO}/hardware/arduino/sam/variants/arduino_due_x/linker_scripts/gcc/flash.ld
    -L${BUILD} # Specify library path
    -lm # Link with the math library
    -lgcc # Link with the libgcc library
    -mthumb # Use the ARM Thumb instruction set
    -Wl,-Map,${BUILD}/${FIRMWARE}.map # Write a map file
    -Wl,--cref # Output cross reference table
    -Wl,--check-sections # Check section addresses for overlaps
    -Wl,--gc-sections # Remove unused sections
    -Wl,--entry=Reset_Handler # Set start address
    -Wl,--warn-unresolved-symbols # Report warning messages rather than errors
    -Wl,--unresolved-symbols=report-all # Report missing symbols
    -Wl,--warn-common #  Warn about duplicate common symbols
    -Wl,--warn-section-align # Warn if start of section changes due to alignment
    -Wl,--start-group # Start of objects group
    ${BUILD}/*.o # Location of compiled objects
    ${ARDUINO}/hardware/arduino/sam/variants/arduino_due_x/libsam_sam3x8e_gcc_rel.a
    -Wl,--end-group # End of objects group
)

# Folders to include.
INC=(
    -I"${FIRMWARE}/Flash"
    -I"${FIRMWARE}/Libraries/EMAC"
    -I"${FIRMWARE}/Libraries/Lwip"
    -I"${FIRMWARE}/Libraries/MCP4461"
    -I"${FIRMWARE}/Libraries/SamNonDuePin"
    -I"${FIRMWARE}/Libraries/SD_HSMCI"
    -I"${FIRMWARE}/Libraries/SD_HSMCI/utility"
    -I"${FIRMWARE}/network"
    -I"${ARDUINO}/hardware/arduino/sam/cores/arduino"
    -I"${ARDUINO}/hardware/arduino/sam/variants/arduino_due_x"
    -I"${ARDUINO}/hardware/arduino/sam/system/libsam"
    -I"${ARDUINO}/hardware/arduino/sam/system/libsam/include"
    -I"${ARDUINO}/hardware/arduino/sam/system/CMSIS/Device/ATMEL/"
    -I"${ARDUINO}/hardware/arduino/sam/system/CMSIS/CMSIS/Include/"
    -I"${ARDUINO}/hardware/arduino/sam/libraries/Wire"
)

# Create output folders.
mkdir -p ${BUILD} ${RELEASE}

# Locate and compile all the .c files that need to be compiled.
for file in $(find ${FIRMWARE} ${ARDUINO}/hardware/arduino/sam/cores -type f -name "*.c")
do
    # Only compile the patch files that have been installed.
    if [[ $file == *ArduinoCorePatches* ]]; then continue; fi

    # Intermediate build output.
    D=${BUILD}/$(basename $file).d
    O=${BUILD}/$(basename $file).o

    # Skip compile if the object is up-to-date.
    if [ ${O} -nt ${file} ]; then continue; fi

    # Show some progress.
    echo "Compiling ${file}"

    # The C source file is newer than the object output file, we need to compile it.
    ${GCC} ${GCC_OPT[@]} ${PLATFORM[@]} ${USB_OPT[@]} ${INC[@]} ${file} -MF${D} -MT${D} -o${O}
done

# Locate and compile all the .cpp files that need to be compiled.
for file in $(find ${FIRMWARE} ${ARDUINO}/hardware/arduino/sam -type f -name "*.cpp")
do
    # Only compile the patch files that have been installed.
    if [[ $file == *ArduinoCorePatches* ]]; then continue; fi

    # Intermediate build output.
    D=${BUILD}/$(basename $file).d
    O=${BUILD}/$(basename $file).o

    # Skip compile if the object is up-to-date.
    if [ ${O} -nt ${file} ]; then continue; fi

    # Show some progress.
    echo "Compiling ${file}"

    # The CPP source file is newer than the object output file, we need to compile it.
    ${GPP} ${GPP_OPT[@]} ${PLATFORM[@]} ${USB_OPT[@]} ${INC[@]} ${file} -MF${D} -MT${D} -o${O}
done

echo "Converting object files into ELF file."
${GPP} ${COM_OPT[@]} -o ${BUILD}/${FIRMWARE}.elf

echo "Converting ELF file into firmware binary."
${COPY} -O binary ${BUILD}/${FIRMWARE}.elf ${RELEASE}/${FIRMWARE}.bin

echo
echo "Created ${RELEASE}/${FIRMWARE}.bin"
echo "You can run '$0 install' to upload the file."
echo