#!/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 -f "$GENERATED_DATA_PATH"/*
mkdir "$GENERATED_DATA_PATH"
touch "$GENERATED_DATA_PATH"/GeneratedShaderData.h
touch "$GENERATED_DATA_PATH"/GeneratedShaderData.c

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

# 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))
len=${#unique_file_names[@]}
i=0
entriesCountDef="\nconst unsigned int* entriesCount[${len}] = {"
entriesDef="const void* entries[${len}] = {"
entriesCountDecl="extern const unsigned int* entriesCount[${len}];"
entriesDecl="extern const void* entries[${len}];"
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
    rm *.spv
    
    echo -e "#include \"${unique_name}.h\"" >> "$GENERATED_DATA_PATH"/GeneratedShaderData.c
    filename_no_ext_first_upper="${unique_name^}"
    entriesCountDef+="&fileTable${filename_no_ext_first_upper}Size"
    entriesDef+="fileTable${filename_no_ext_first_upper}"
    
    if [ $i -ne $((len-1)) ]; then
        entriesCountDef+=",\n\t\t"
        entriesDef+=",\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

entriesCountDef+=" };"
entriesDef+=" };"
echo -e "${entriesCountDecl}\n${entriesDecl}\\n#endif /*GeneratedShaderData_H_INCLUDED*/" >> "$GENERATED_DATA_PATH"/GeneratedShaderData.h
echo -e "\n${entriesCountDef}\n${entriesDef}" >> "$GENERATED_DATA_PATH"/GeneratedShaderData.c