#! /usr/bin/env bash
##
## Copyright (C) by Argonne National Laboratory
##     See COPYRIGHT in top-level directory
##

if hash gindent 2>/dev/null; then
	indent=gindent
else
	indent=indent
fi

indent_version=$($indent --version)
if echo "$indent_version" | grep -q "GNU indent 2.2.11"; then
    :
else
    echo "This script requires GNU indent version 2.2.11."
    exit
fi

indent_code()
{
    file=$1

    $indent \
        `# Expansion of Kernighan & Ritchie style` \
        --no-blank-lines-after-declarations \
        `# --no-blank-lines-after-procedures` `# Overwritten below` \
        `# --break-before-boolean-operator` `# Overwritten below` \
        --no-blank-lines-after-commas \
        --braces-on-if-line \
        --braces-on-struct-decl-line \
        `# --comment-indentation33` `# Overwritten below` \
        --declaration-comment-column33 \
        --no-comment-delimiters-on-blank-lines \
        --cuddle-else \
        --continuation-indentation4 \
        `# --case-indentation0` `# Overwritten below` \
        `# --else-endif-column33` `# Overwritten below` \
        --space-after-cast \
        --line-comments-indentation0 \
        --declaration-indentation1 \
        --dont-format-first-column-comments \
        --dont-format-comments \
        --honour-newlines \
        --indent-level4 \
        --parameter-indentation0 \
        `# --line-length75` `# Overwritten below` \
        --continue-at-parentheses \
        --no-space-after-function-call-names \
        --no-space-after-parentheses \
        --dont-break-procedure-type \
        --space-after-for \
        --space-after-if \
        --space-after-while \
        `# --dont-star-comments` `# Overwritten below` \
        --leave-optional-blank-lines \
        --dont-space-special-semicolon \
        `# End of K&R expansion` \
        --line-length100 \
        --else-endif-column1 \
        --start-left-side-of-comments \
        --break-after-boolean-operator \
        --comment-indentation1 \
        --no-tabs \
        --blank-lines-after-procedures \
        --leave-optional-blank-lines \
        --braces-after-func-def-line \
        --brace-indent0 \
        --cuddle-do-while \
        --no-space-after-function-call-names \
        --case-indentation4 \
        ${file}

    rm -f ${file}~
    cp ${file} /tmp/${USER}.__tmp__ && \
	cat ${file} | sed -e 's/ *$//g' -e 's/( */(/g' -e 's/ *)/)/g' \
	-e 's/if(/if (/g' -e 's/while(/while (/g' -e 's/do{/do {/g' -e 's/}while/} while/g' > \
	/tmp/${USER}.__tmp__ && mv /tmp/${USER}.__tmp__ ${file}
}

debug=
run_indent()
{
    if test -n "$debug" ; then
        echo indent_code $1
    else
        indent_code $1
        # indent would sometime toggle code layout. Run it twice to get stability.
        indent_code $1
    fi
}

usage()
{
    echo "Usage: $1 [filename | --all] {--recursive} {--debug} {--ignore ignore_patterns}"
}

# Check usage
if [ -z "$1" ]; then
    usage $0
    exit
fi

filetype_list="\.c$|\.h$|\.c\.in$|\.h\.in$|\.cpp$|\.cpp.in$|\.inc$"

ignore_list="doc/"
ignore_list="$ignore_list|src/mpid/ch3/doc"
ignore_list="$ignore_list|src/mpid/ch3/include|src/mpid/ch3/src"
ignore_list="$ignore_list|src/mpid/ch3/util"
ignore_list="$ignore_list|src/mpid/ch3/channels/nemesis/include"
ignore_list="$ignore_list|src/mpid/ch3/channels/nemesis/src"
ignore_list="$ignore_list|src/mpid/ch3/channels/nemesis/utils"
ignore_list="$ignore_list|src/mpi/romio/include/mpiof.h.in"
ignore_list="$ignore_list|test/mpi/errors/f77/io/addsize.h.in"
ignore_list="$ignore_list|test/mpi/errors/f77/io/iooffset.h.in"
ignore_list="$ignore_list|test/mpi/f77/attr/attraints.h.in"
ignore_list="$ignore_list|test/mpi/f77/datatype/typeaints.h.in"
ignore_list="$ignore_list|test/mpi/f77/ext/add1size.h.in"
ignore_list="$ignore_list|test/mpi/f77/io/ioaint.h.in"
ignore_list="$ignore_list|test/mpi/f77/io/iodisp.h.in"
ignore_list="$ignore_list|test/mpi/f77/io/iooffset.h.in"
ignore_list="$ignore_list|test/mpi/f77/pt2pt/attr1aints.h.in"
ignore_list="$ignore_list|test/mpi/f77/rma/addsize.h.in"
ignore_list="$ignore_list|test/mpi/f77/spawn/type1aint.h.in"
ignore_list="$ignore_list|src/include/mpi.h.in"
ignore_list="$ignore_list|src/mpi/romio/include/mpio.h.in"
ignore_list="$ignore_list|src/mpi/romio/adio/include/adioi_errmsg.h"
ignore_list="$ignore_list|src/pmi/include/pmix.h"
ignore_list="$ignore_list|src/pmi/include/pmix_abi_support.h"
ignore_list="$ignore_list|src/pmi/include/pmix_abi_support_bottom.h"
ignore_list="$ignore_list|src/pmi/include/pmix_fns.h"
ignore_list="$ignore_list|src/pmi/include/pmix_macros.h"
ignore_list="$ignore_list|src/pmi/include/pmix_types.h"
ignore_list="$ignore_list|src/binding/abi/mpi_abi.h"

filelist=""

all=0
recursive=0
ignore=0
got_file=0
for arg in $@; do
    if [ "$ignore" = "1" ] ; then
	ignore_list="$ignore_list|$arg"
	ignore=0
	continue;
    fi

    if [ "$arg" = "--all" ]; then
	all=1
    elif [ "$arg" = "--recursive" ]; then
	recursive=1
    elif [ "$arg" = "--debug" ]; then
	debug="echo"
    elif [ "$arg" = "--ignore" ] ; then
	ignore=1
    else
	got_file=1
        filelist="$filelist $arg"
    fi
done
if [ "$recursive" = "1" -a "$all" = "0" ]; then
    echo "--recursive cannot be used without --all"
    usage $0
    exit
fi
if [ "$got_file" = "1" -a "$all" = "1" ]; then
    echo "--all cannot be used in conjunction with a specific file"
    usage $0
    exit
fi

if [ "$recursive" = "1" ]; then
    for i in `git ls-files | egrep "($filetype_list)" | egrep -v "($ignore_list)"` ; do
	run_indent $i
    done
elif [ "$all" = "1" ]; then
    for i in `git ls-files | cut -d/ -f1 | uniq | egrep "($filetype_list)" | egrep -v "($ignore_list)"` ; do
	run_indent $i
    done
else
    for i in $filelist; do
        filename=`echo $i | egrep "($filetype_list)" | egrep -v "($ignore_list)"`
        if [ "$filename" != "" ] ; then
            run_indent $filename
        fi
    done
fi
