#!/bin/bash

GENERATED_DATA_PATH="../openVulkanoCpp/GeneratedShaderData"
SHADER_PATH="../openVulkanoCpp/Shader"

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

generate_general_shader_header() {
  rm -f $GENERATED_DATA_PATH/GeneratedShaderData.h
  touch $GENERATED_DATA_PATH/GeneratedShaderData.h
}

rm -f *.spv
rm -f $GENERATED_DATA_PATH/*.h $GENERATED_DATA_PATH/*.c

generate_general_shader_header
echo "// Auto generated file" >> $GENERATED_DATA_PATH/GeneratedShaderData.h
sed -i "$ a #ifndef GeneratedShaderData_H_INCLUDED\n#define GeneratedShaderData_H_INCLUDED\n" $GENERATED_DATA_PATH/GeneratedShaderData.h
sed -i "$ a extern \"C\" {" $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 -d))
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")
        glslang -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^}"
    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
done

sed -i "$ a }\n#endif \/*GeneratedShaderData_H_INCLUDED*\/" $GENERATED_DATA_PATH/GeneratedShaderData.h