#!/bin/bash

SCRIPT_DIR=$( cd -- "$( dirname -- "${BASH_SOURCE[0]}" )" &> /dev/null && pwd )
GENERATED_DATA_PATH="$(dirname "$SCRIPT_DIR")/openVulkanoCpp/GeneratedShaderData"
SHADER_PATH="$(dirname "$SCRIPT_DIR")/openVulkanoCpp/Shader"

if [[ "$OSTYPE" == "msys" || "$OSTYPE" == "cygwin" ]]; then
  SCRIPT_DIR=$(cygpath -w "$SCRIPT_DIR")
  GENERATED_DATA_PATH=$(cygpath -w "$GENERATED_DATA_PATH")
  SHADER_PATH=$(cygpath -w "$SHADER_PATH")
fi

check_return_code() {
  # return code of last executed command  
  retVal=$?
  if [ $retVal -ne 0 ]; then
    exit 1
  fi
}

rm -f *.spv
rm "$GENERATED_DATA_PATH"/*
touch "$GENERATED_DATA_PATH"/GeneratedShaderData.h

echo "// Auto generated file" >> "$GENERATED_DATA_PATH"/GeneratedShaderData.h
echo -e "#ifndef GeneratedShaderData_H_INCLUDED\n#define GeneratedShaderData_H_INCLUDED\n" >> "$GENERATED_DATA_PATH"/GeneratedShaderData.h
echo -e "extern \"C\" {\n" >> "$GENERATED_DATA_PATH"/GeneratedShaderData.h

# put data from all files with same name into single file
unique_file_names=($(find $SHADER_PATH/ -type f ! -name '*.hpp' ! -name '*.cpp' | awk -F/ '{print $NF}' | awk -F. '{print $1}' | sort | uniq))
entriesCount="\nconst unsigned int entriesCount[] = {"
entries="const TFileTableEntry* entries[] = {"
len=${#unique_file_names[@]}
i=0
for unique_name in ${unique_file_names[@]} ; do
    same_files=$(find $SHADER_PATH/ -type f -name "${unique_name}.*")
    for same_file in ${same_files[@]} ; do
        filename_with_ext_no_path=$(basename -- "$same_file")
        glslangValidator -V $same_file -o $filename_with_ext_no_path.spv
        check_return_code
    done
    
    bin2c -t -d "$GENERATED_DATA_PATH"/$unique_name.h -o "$GENERATED_DATA_PATH"/$unique_name.c *.spv
    check_return_code
    sed -i "$ a #include \"${unique_name}.h\"" "$GENERATED_DATA_PATH"/GeneratedShaderData.h
    rm *.spv
    
    filename_no_ext_first_upper="${unique_name^}"
    entriesCount+="fileTable${filename_no_ext_first_upper}Size"
    entries+="fileTable${filename_no_ext_first_upper}"
    
    if [ $i -ne $((len-1)) ]; then
        entriesCount+=",\n\t\t"
        entries+=",\n\t\t"
    fi
    
    sed -i "3s/.*/#include \"${unique_name}.h\"/" "$GENERATED_DATA_PATH"/$unique_name.c
    sed -i 's/char \*entryName/const char *entryName/' "$GENERATED_DATA_PATH"/$unique_name.h
    sed -i 's/const char \*data/const unsigned char *data/' "$GENERATED_DATA_PATH"/$unique_name.h
    # avoid ODR violation
    sed -i "s/fileTable/fileTable${filename_no_ext_first_upper}/" "$GENERATED_DATA_PATH"/$unique_name.h $GENERATED_DATA_PATH/$unique_name.c
    sed -i "s/typedef struct {/#ifndef TFileTableEntry_H_INCLUDED\n#define TFileTableEntry_H_INCLUDED\ntypedef struct {/" "$GENERATED_DATA_PATH"/$unique_name.h
    sed -i "s/} TFileTableEntry;/} TFileTableEntry;\n#endif \/*TFileTableEntry_H_INCLUDED*\//" "$GENERATED_DATA_PATH"/$unique_name.h
    ((i++))
done

entriesCount+=" };"
entries+=" };"
echo -e $entriesCount >> "$GENERATED_DATA_PATH"/GeneratedShaderData.h
echo -e $entries >> "$GENERATED_DATA_PATH"/GeneratedShaderData.h
echo -e "}\n#endif /*GeneratedShaderData_H_INCLUDED*/" >> "$GENERATED_DATA_PATH"/GeneratedShaderData.h