gettc 1.2.2

data/dist/include/java/engine.rb

@@ -0,0 +1,207 @@
+require 'topcoder/types' 
+
+module TopCoder
+    class Type
+        def to_java
+            if self == TInt then
+                return 'int'
+            elsif self == TLong then
+                return 'long'        
+            elsif self == TFloat then
+                return 'float'
+            elsif self == TDouble then
+                return 'double'
+            elsif self == TChar then
+                return 'char'
+            elsif self == TString then
+                return 'String'
+            elsif self == TBoolean then
+                return 'boolean'
+            elsif is_a? TArray then
+                return "#{subtype.to_java}[]"
+            else
+                return 'unknown'
+            end 
+        end
+        def to_java_boxed
+            if self == TInt then
+                return 'Integer'
+            elsif self == TLong then
+                return 'Long'        
+            elsif self == TFloat then
+                return 'Float'
+            elsif self == TDouble then
+                return 'Double'
+            elsif self == TChar then
+                return 'Character'
+            elsif self == TString then
+                return 'String'
+            elsif self == TBoolean then
+                return 'Boolean'
+            elsif self.is_a? TArray then
+                return "List<#{subtype.to_java_boxed}>"
+            else
+                return 'Unknown'
+            end 
+        end
+        def dumb_java
+            if self == TInt then
+                return '0'
+            elsif self == TLong then
+                return '0'          
+            elsif self == TFloat then
+                return '0'
+            elsif self == TDouble then
+                return '0'
+            elsif self == TChar then
+                return "'$'"
+            elsif self == TString then
+                return '"$"'
+            elsif self == TBoolean then
+                return 'true'
+            elsif self.is_a? TArray then
+                return "new #{subtype.to_java}[1]"
+            else
+                return 'Nil'
+            end
+        end
+    end
+    class Signature
+        def to_java
+            return "#{type.to_java} #{name}"
+        end
+    end
+    class JavaArray
+        def self.get_depth type
+            return 0 if not type.is_a? TArray
+            return 1 + get_depth(type.subtype)
+        end
+        def self.get_elem_type type
+            return type if not type.is_a? TArray
+            return get_elem_type type.subtype
+        end
+        attr_accessor :name, :boxed_name, :type, :boxed_type
+        def initialize sig
+            @sig = sig
+            @name = sig.name
+            @boxed_name = "#{@name}Boxed"
+            @type = sig.type.to_java
+            @boxed_type = sig.type.to_java_boxed
+            @depth = self.class.get_depth sig.type
+            @elem_type = self.class.get_elem_type(sig.type).to_java
+        end
+        def unbox_code
+            counters = @depth.times.map do |i| '_' + ('i'[0].ord + i).chr end
+            indents = 0.upto(@depth).map do |i| ' ' * 4 * i end
+            sizes = @depth.times.map do |i|
+                @boxed_name + ('.get(0)' * i) + '.size()'
+            end
+            ret = ''
+
+            ret << @type << ' ' << @name << ' = new ' << @elem_type;
+            @depth.times do |i| ret << '[' << sizes[i] << ']' end
+            ret << ";\n"
+        
+            @depth.times do |i|
+                ret << indents[i] << 'for (int ' << counters[i] << ' = 0; '
+                ret << counters[i] << ' < ' << sizes[i]
+                ret << '; ++' << counters[i] << ")\n"
+            end
+
+            ret << indents[@depth] << @name
+            @depth.times do |i| ret << '[' << counters[i] << ']' end            
+            ret << ' = '
+            ret << @boxed_name
+            @depth.times do |i| ret << '.get(' << counters[i] << ')' end            
+            ret << ";"
+
+            return ret
+        end
+        def box_code_recurse boxed_type, parent, boxed_name, depth, counters
+            ret = ''
+            index = counters.map do |counter| '[' + counter + ']' end
+            index = index.join
+            if depth == 0
+                ret << parent << '.add(' << @name << index << ");\n"
+            else
+                type_str = boxed_type.to_java_boxed
+                ret << type_str << ' ' << boxed_name 
+                ret << ' = new Array' << type_str << '();' << "\n"
+
+                counter = '_' + ('i'[0].ord + counters.size).chr
+                ret << 'for (int ' << counter << ' = 0; '
+                ret << counter << ' < ' << @name << index << '.length; '
+                ret << '++' << counter << ") {\n"
+
+                ret << box_code_recurse(boxed_type.subtype, 
+                                        boxed_name, 
+                                        boxed_name + counter, 
+                                        depth - 1, counters << counter)
+                ret << "}\n"
+
+                if not parent.nil?
+                    ret << parent << '.add(' 
+                    ret << boxed_name  << ");\n"
+                end
+            end
+            ret.gsub! /^/, ' ' * 4 if not parent.nil? 
+            return ret
+        end
+        def box_code
+            box_code_recurse @sig.type, nil, @boxed_name, @depth, []
+        end 
+    end
+    class JavaEngine
+        attr_accessor :func, :vars
+        def initialize func, vars
+            @func = func
+            @vars = vars            
+        end
+        def declare 
+            ret = 'public '
+            ret << @func.to_java << '('
+                indent = ' ' * ret.size
+                temp = @vars.map do |var| var.to_java end
+                ret << temp.join(",\n#{indent}")
+            ret << ')'
+            return ret
+        end
+        def input
+            temp = @vars.map do |var| 
+                ret = ''
+                if var.type.is_a? TArray
+                    arr = JavaArray.new var
+                    ret << arr.boxed_type
+                    ret << ' ' << arr.boxed_name
+                    ret << ' = (' << arr.boxed_type << ') '
+                    ret << 'reader.next(new TypeRef<' << arr.boxed_type << '>'
+                    ret << "(){}.getType());\n"
+                    ret << arr.unbox_code
+                else
+                    ret << var.to_java
+                    ret << ' = (' << var.type.to_java_boxed << ') '
+                    ret << 'reader.next(' 
+                    ret << var.type.to_java_boxed 
+                    ret << '.class);'
+                end
+                ret
+            end
+            return temp.join "\nreader.next();\n\n"
+        end
+        def output
+            ret = ''
+            caller = 'solver.' + @func.name + '('
+            temp = @vars.map do |var| var.name end
+            caller << temp.join(', ') << ')'
+            if @func.type.is_a? TArray
+                ret << @func.type.to_java << ' result = ' << caller << ";\n"
+                arr = JavaArray.new(Signature.new @func.type, 'result')
+                ret << arr.box_code
+                ret << 'writer.write(resultBoxed);'
+            else
+                ret << 'writer.write(' << caller << ');'
+            end
+            return ret
+        end
+    end
+end

data/dist/template/bin/runner.sh

@@ -0,0 +1,113 @@
+mode="$1"
+solver="$2"
+data_d="$3"
+output_d="$4"
+ostream=/dev/null
+
+timeit () {
+    if [ -x /usr/bin/time ]; then
+        /usr/bin/time -o $ostream "$1" "$2" "$3" "$4" "$5" "$6" "$7" "$8" "$9"
+    elif [ $mode = 'verbose' ]; then
+        time "$1" "$2" "$3" "$4" "$5" "$6" "$7" "$8" "$9"
+    else
+        "$1" "$2" "$3" "$4" "$5" "$6" "$7" "$8" "$9"
+    fi
+} 
+
+puts () {
+    echo "$1" > $ostream
+}
+
+write () {
+    echo -n "$1" > $ostream
+}
+
+view_file () {
+    filename="$1"    
+    write '<'
+    cat "$filename" > $ostream
+    puts '>'
+}
+
+report_case () {
+    write '    Input: '
+    view_file "$data_d/$test_case.in"
+    write '    Expected: '
+    view_file "$data_d/$test_case.out"
+    write '    Received: '
+    view_file "$output_d/$test_case.out"
+}
+
+init () {
+    export TIME="Time: %es - Memory: %MK"
+    export TIMEFORMAT=%3lR
+
+    if [ ! -d "$output_d" ]; then
+        mkdir "$output_d"
+    fi 
+
+    if [ "$mode" = 'verbose' ]; then
+        ostream=/dev/stdout
+    elif [ "$mode" = 'file' ]; then
+        ostream=log
+    fi
+
+    dir=`dirname $0`
+    checker="$dir/../build/check" 
+    if [ ! -x "$checker" ]; then
+        undo=`pwd`
+        cd "$dir/../util/check"
+        make
+        cd "$undo"
+    fi
+} 
+
+main () {
+    cases=0
+
+    fails=0
+
+    errors=0
+    failstr=''
+
+    errorstr=''
+    for input in "$data_d"/*.in; do 
+        if [ -f "$input" ]; then
+            cases=$(( cases + 1 ))
+
+            test_case=`basename "$input"` 
+            test_case=${test_case%.in}
+
+            write "Check $test_case ... "
+            timeit "$solver" "$data_d/$test_case.in" "$output_d/$test_case.out"
+            "$checker" "$data_d/$test_case.out" "$output_d/$test_case.out" > $ostream
+            retcode=$?
+
+            if [ $retcode -eq 0 ]; then
+                puts 'Passed'
+
+            elif [ $retcode -eq 1 ]; then
+                fails=$(( fails + 1 ))
+                failstr="$failstr $test_case"
+                puts 'Failed'
+                report_case "$test_case"
+
+            else 
+                errors=$(( errors + 1 ))
+                errorstr="$errorstr $test_case"
+                puts 'Error'
+                report_case "$test_case"
+            fi
+        fi
+    done
+    echo "$cases cases checked, $fails failed, $errors errored"
+    if [ $fails -gt 0 ]; then
+        echo "Failed cases:${failstr}"
+    fi
+    if [ $errors -gt 0 ]; then
+        echo "Errored cases:${errorstr}"
+    fi
+
+}
+init
+main

data/dist/template/prob/{name}.html

@@ -0,0 +1,8 @@
+<html>
+<head>
+    <title><%= @prob.name %></title>
+</head>
+<body>
+<%= @prob.to_html %>
+</body>
+</html>

data/dist/template/prob/{name}.md

@@ -0,0 +1 @@
+<%= @prob.to_md %>

data/dist/template/solve/cpp/Makefile

@@ -0,0 +1,30 @@
+OUTPUT_D=../../build/cpp
+DATA_D=../../data
+COMPILER=g++ -std=gnu++0x -I$$HOME/.gettc/include/cpp
+RUNNER=sh ../../bin/runner.sh
+
+run = $(RUNNER) $1 $(OUTPUT_D)/<%= prob.name %>Runner $(DATA_D)/$2 $(OUTPUT_D)/$2
+
+default : demo
+
+setup :
+	if [ ! -d $(OUTPUT_D) ]; then mkdir -p $(OUTPUT_D); fi
+
+main : <%= prob.name %>Runner.cpp <%= prob.name %>.cpp setup
+	$(COMPILER) -o $(OUTPUT_D)/<%= prob.name %>Runner $<
+
+test : <%= prob.name %>Test.cpp <%= prob.name %>.cpp setup
+	$(COMPILER) -lboost_unit_test_framework -o $(OUTPUT_D)/<%= prob.name %>Test $<
+	$(OUTPUT_D)/<%= prob.name %>Test
+
+demo : main
+	$(call run,verbose,demo) 
+
+sys : main
+	$(call run,quiet,sys)
+
+sysv : main
+	$(call run,verbose,sys)
+
+clean :
+	rm -rf $(OUTPUT_D)

data/dist/template/solve/cpp/{name}.cpp

@@ -0,0 +1,11 @@
+#include <vector>
+#include <string>
+using namespace std;
+<%
+    engine = CppEngine.new func, vars
+%>
+<%=
+    engine.declare
+%> {
+    return <%= func.type.dumb_cpp %>;
+}

data/dist/template/solve/cpp/{name}Runner.cpp

@@ -0,0 +1,26 @@
+#include <topcoder>
+using namespace TopCoder;
+#include "<%= prob.name %>.cpp"
+#include <fstream>
+using namespace std;
+<%
+    engine = CppEngine.new func, vars
+%>
+int main(int argc, char const *argv[]) {
+    try {
+        ifstream ifs(argv[1]);
+<%=
+    engine.input.gsub(/^/, ' ' * 8)
+%>
+        ifs.close();
+
+        ofstream ofs(argv[2]);
+<%=
+    engine.output.gsub(/^/, ' ' * 8)
+%>
+        ofs.close();
+    } catch (exception &e) {
+        cerr << e.what() << endl;
+    }
+    return 0;
+}

data/dist/template/solve/cpp/{name}Test.cpp

@@ -0,0 +1,8 @@
+#define BOOST_TEST_DYN_LINK 
+#define BOOST_TEST_MODULE MyTest
+#include <boost/test/unit_test.hpp> 
+#include "<%= prob.name %>.cpp" 
+
+BOOST_AUTO_TEST_CASE (test_<%= func.name %>) {
+    BOOST_CHECK_EQUAL(0, 0);
+}

data/dist/template/solve/haskell/Makefile

@@ -0,0 +1,30 @@
+OUTPUT_D=../../build/haskell
+DATA_D=../../data
+COMPILER=ghc -i$$HOME/.gettc/include/haskell -outputdir $(OUTPUT_D)
+RUNNER=sh ../../bin/runner.sh
+
+run = $(RUNNER) $1 $(OUTPUT_D)/<%= prob.name %>Runner $(DATA_D)/$2 $(OUTPUT_D)/$2
+
+default : demo
+
+setup :
+	if [ ! -d $(OUTPUT_D) ]; then mkdir -p $(OUTPUT_D); fi
+
+main : <%= prob.name %>Runner.hs <%= prob.name %>.hs setup
+	$(COMPILER) -o $(OUTPUT_D)/<%= prob.name %>Runner $<
+
+test : <%= prob.name %>Test.hs <%= prob.name %>.hs setup
+	$(COMPILER) -o $(OUTPUT_D)/<%= prob.name %>Test $<
+	$(OUTPUT_D)/<%= prob.name %>Test
+
+demo : main
+	$(call run,verbose,demo) 
+
+sys : main
+	$(call run,quiet,sys)
+
+sysv : main
+	$(call run,verbose,sys)
+
+clean :
+	rm -rf $(OUTPUT_D)