gettc 1.6.2 → 1.7

data/dist/include/go/src/topcoder/writer.go

@@ -0,0 +1,54 @@
+package topcoder
+
+import (
+    "bufio"
+    "fmt"
+    "io"
+    "reflect"
+)
+
+func writeOne(bw *bufio.Writer, x interface{}) error {
+    switch y := x.(type) {
+    case bool:
+        fmt.Fprintf(bw, "%t", y)
+    case byte:
+        fmt.Fprintf(bw, "'%c'", y)
+    case string:
+        fmt.Fprintf(bw, "\"%s\"", y)
+    case int, int64:
+        fmt.Fprintf(bw, "%d", y)
+    case float32, float64:
+        fmt.Fprintf(bw, "%f", y)
+    default:
+        slice := reflect.ValueOf(x)
+        if slice.Kind() != reflect.Slice {
+            return NewErrType(slice.Type())
+        }
+        bw.WriteByte('[')
+        for i := 0; i < slice.Len(); i++ {
+            if i != 0 {
+                bw.WriteString(", ")
+            }
+            elem := slice.Index(i)
+            if err := writeOne(bw, elem.Interface()); err != nil {
+                return err
+            }
+        }
+        bw.WriteByte(']')
+    }
+    return nil
+}
+
+func Write(w io.Writer, xs ...interface{}) error {
+    bw := bufio.NewWriter(w)
+    for i, x := range xs {
+        if i != 0 {
+            bw.WriteString(", ")
+        }
+        if err := writeOne(bw, x); err != nil {
+            return err
+        }
+    }
+    bw.Flush()
+    return nil
+}

data/dist/include/go/src/topcoder/writer_test.go

@@ -0,0 +1,19 @@
+package topcoder
+
+import (
+    "bytes"
+    "testing"
+)
+
+func writer() *bytes.Buffer {
+    return bytes.NewBufferString("")
+}
+
+func TestShow(t *testing.T) {
+    out := writer()
+    Write(out, "Joe", 20, 7.5, false, []string{"Math", "CompSci"}, [][]int{{7, 8}, {}})
+    ret := out.String()
+    if ret != "\"Joe\", 20, 7.500000, false, [\"Math\", \"CompSci\"], [[7, 8], []]" {
+        t.Errorf("Write() produces <" + ret + ">")
+    }
+}

data/dist/include/haskell/engine.rb

@@ -1,11 +1,5 @@
 require "gettc/types" 
 
-class String
-    def uncapitalize
-        return self[0, 1].downcase + self[1..-1]
-    end
-end
-
 module Gettc
     class Type
         def to_haskell
@@ -73,14 +67,21 @@ module Gettc
         end
     end
     class HaskellEngine
-        attr_accessor :func, :vars
+        attr_reader :func, :vars, :declare, :input, :output
         def initialize func, vars
-            @func = Signature.new func.type, func.name.uncapitalize
+            @func = Signature.new func.type, uncapitalize(func.name)
             @vars = vars.map do |var|
-                Signature.new var.type, var.name.uncapitalize
+                Signature.new var.type, uncapitalize(var.name)
             end
+            compute_declare
+            compute_input
+            compute_output
+        end
+    private
+        def uncapitalize str
+            return str[0, 1].downcase + str[1..-1]
         end
-        def declare
+        def compute_declare
             ret = ""
             ret << func.name << " :: "
 
@@ -91,9 +92,9 @@ module Gettc
             temp = vars.map do |var| var.name end
             ret << temp.join(" ") << " = " << func.type.dumb_haskell
 
-            return ret
+            @declare = ret
         end
-        def input
+        def compute_input
             ret = "getVars :: TC.Parser ("
             temp = @vars.map do |var| var.type.to_haskell end
             ret << temp.join(", ") << ")\n"
@@ -111,14 +112,14 @@ module Gettc
                 temp = @vars.map do |var| var.name end
                 ret << temp.join(", ")
             ret << ")"
-            return ret
+            @input = ret
         end
-        def output
+        def compute_output
             ret = ""
             ret << @func.name << " "
             temp = vars.map do |var| var.name end
             ret << temp.join(" ")
-            return ret
+            @output = ret
         end
     end
 end

data/dist/include/java/engine.rb

@@ -80,7 +80,7 @@ module Gettc
             return type unless type.is_a? TArray
             return get_elem_type type.subtype
         end
-        attr_accessor :name, :boxed_name, :type, :boxed_type
+        attr_reader :name, :boxed_name, :type, :boxed_type
         def initialize sig
             @sig = sig
             @name = sig.name
@@ -152,21 +152,25 @@ module Gettc
         end 
     end
     class JavaEngine
-        attr_accessor :func, :vars
+        attr_accessor :declare, :input, :output
         def initialize func, vars
             @func = func
-            @vars = vars            
+            @vars = vars
+            compute_declare
+            compute_input
+            compute_output
         end
-        def declare 
+    private
+        def compute_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
+            @declare = ret
         end
-        def input
+        def compute_input
             temp = @vars.map do |var| 
                 ret = ""
                 if var.type.is_a? TArray
@@ -186,9 +190,9 @@ module Gettc
                 end
                 ret
             end
-            return temp.join "\nreader.next();\n\n"
+            @input = temp.join "\nreader.next();\n\n"
         end
-        def output
+        def compute_output
             ret = ""
             caller = "solver." + @func.name + "("
             temp = @vars.map do |var| var.name end
@@ -201,7 +205,7 @@ module Gettc
             else
                 ret << "writer.write(" << caller << ");"
             end
-            return ret
+            @output = ret
         end
     end
-end
+end

data/dist/include/python/engine.rb

@@ -1,4 +1,5 @@
-require "gettc/types" 
+require "gettc/types"
+require "mkmf"
 
 module Gettc
     class Type
@@ -25,22 +26,44 @@ module Gettc
         end
     end
     class PythonEngine
-        attr_accessor :func, :vars
+        attr_reader :arglist, :input, :shebang
         def initialize func, vars
-            @func = func
-            @vars = vars            
-        end
-        def vars_list
-            temp = @vars.map do |var|
+            temp = vars.map do |var|
                 var.name
             end
-            return temp.join ", "
-        end
-        def input
-            temp = @vars.map do |var| 
+            @arglist = temp.join ", "
+
+            temp = vars.map do |var| 
                 var.name + ' = reader.next("' + var.type.to_s + '")'
             end
-            return temp.join "\nreader.next()\n"
+            @input = temp.join "\nreader.next()\n"
+
+            @shebang = "#! "
+            @shebang << python3_path
+        end
+    private
+        def python3_path
+            default = "/usr/bin/env python"
+
+            version = system_q("python --version").split
+            if version.size != 2 || version[0].upcase != "PYTHON" then
+                return default
+            end
+
+            version = version[1].split(".")[0].to_i
+            if version >= 3 then
+                return default
+            end 
+
+            python3 = system_q "which python3"
+            return python3.empty? ? default : python3
+        rescue
+            return default
+        end
+        def system_q command
+            IO.popen command, err: [:child, :out] do |f|
+                return f.gets
+            end
         end
     end
-end
+end

data/dist/template/solve/cpp/Makefile

@@ -17,10 +17,6 @@ setup :
 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) 
 
@@ -31,4 +27,4 @@ sysv : main
 	$(call run,verbose,sys)
 
 clean :
-	rm -rf $(OUTPUT_D)
+	rm -rf $(OUTPUT_D)

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

@@ -5,7 +5,7 @@ using namespace std;
 class <%= prob.name %> {
 public:
 <%= engine = CppEngine.new func, vars
-    engine.declare %> {
+    engine.declare.gsub /^/, "    " %> {
         return <%= func.type.dumb_cpp %>;
     }
-};
+};

data/dist/template/solve/go/Makefile

@@ -0,0 +1,29 @@
+GETTC_HOME ?= $(HOME)/.gettc
+OUTPUT_D =../../build/go
+DATA_D = ../../data
+RUNNER = sh ../../bin/runner.sh
+
+run = $(RUNNER) $1 $(OUTPUT_D)/<%= prob.name %>Runner $(DATA_D)/$2 $(OUTPUT_D)/$2
+
+default : usage
+
+usage :
+	@cat $(GETTC_HOME)/usage
+
+setup :
+	if [ ! -d $(OUTPUT_D) ]; then mkdir -p $(OUTPUT_D); fi
+
+main : <%= prob.name %>Runner.go <%= prob.name %>/<%= prob.name %>.go setup
+	export GOPATH="$$GOPATH<%= File::PATH_SEPARATOR %>$(GETTC_HOME)/include/go" && go build -o $(OUTPUT_D)/<%= prob.name %>Runner <%= prob.name %>Runner.go 
+
+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/go/{name}/{name}.go

@@ -0,0 +1,7 @@
+package <%= prob.name %>
+<%
+    engine = GoEngine.new func, vars
+%>
+func <%= engine.func_name %>(<%= engine.declare.join ", " %>) <%= func.type.to_go %> {
+    return <%= func.type.dumb_go %>
+}

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

@@ -0,0 +1,34 @@
+package main
+
+import (
+    "./<%= prob.name %>"
+    "fmt"
+    "os"
+    "topcoder"
+)
+<%
+    engine = GoEngine.new func, vars
+%>
+func main() {
+    if len(os.Args) < 3 {
+        fmt.Fprintf(os.Stderr, "Usage: %s <input_file> <output_file>", os.Args[0])
+        os.Exit(-1)
+    }
+
+    var (
+        fileHandle *os.File
+        errIO error
+        <%= engine.declare.join "\n#{' ' * 8}" %>
+    )
+
+    if fileHandle, errIO = os.Open(os.Args[1]); errIO != nil {
+        panic(errIO)
+    }
+    topcoder.Read(fileHandle, <%= engine.input %>)
+    fileHandle.Close()
+
+    if fileHandle, errIO = os.Create(os.Args[2]); errIO != nil {
+        panic(errIO)
+    }
+    topcoder.Write(fileHandle, <%= prob.name %>.<%= engine.func_name %>(<%= engine.output %>))
+}

data/dist/template/solve/haskell/Makefile

@@ -17,10 +17,6 @@ setup :
 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) 
 
@@ -31,4 +27,4 @@ sysv : main
 	$(call run,verbose,sys)
 
 clean :
-	rm -rf $(OUTPUT_D)
+	rm -rf $(OUTPUT_D)

data/dist/template/solve/java/build.xml

@@ -59,22 +59,6 @@ java -cp "`dirname $0`${path.separator}${lib}" <%= prob.name %>Runner $$*
             <arg value="${output.dir}/sys"/>
         </exec>
     </target>
-
-    <target name="compile-test">
-        <mkdir dir="${output.dir}"/>
-        <javac srcdir="." destdir="${output.dir}" 
-               includes="<%= prob.name %>.java,<%= prob.name %>Test.java"
-               includeantruntime="yes">
-            <classpath location="${output.dir}"/>
-        </javac>
-    </target>
-    <target name="test" depends="compile-test">
-        <junit fork="yes" haltonfailure="yes">
-            <classpath location="${output.dir}"/>
-            <formatter type="plain" usefile="false" />
-            <test name="<%= prob.name %>Test"/>
-        </junit>
-    </target>
     
     <target name="clean">
         <delete dir="${output.dir}"/>

data/dist/template/solve/python/Makefile

@@ -24,4 +24,4 @@ sysv : setup
 	$(call run,verbose,sys)
 
 clean :
-	rm -rf $(OUTPUT_D)
+	rm -rf $(OUTPUT_D)

data/dist/template/solve/python/{name}.py

@@ -1,4 +1,4 @@
 <%
     engine = PythonEngine.new func, vars
-%>def <%= func.name %>(<%= engine.vars_list %>):
-    return <%= func.type.dumb_python %>
+%>def <%= func.name %>(<%= engine.arglist %>):
+    return <%= func.type.dumb_python %>

data/dist/template/solve/python/{name}Runner.py

@@ -1,5 +1,5 @@
-#! /usr/bin/env python
-
+<%= engine = PythonEngine.new func, vars
+    engine.shebang %>
 import os, sys
 import <%= prob.name %>
 
@@ -15,10 +15,9 @@ def main():
         input = fi.read()
         reader = tc.Reader(input)
     
-<%= engine = PythonEngine.new func, vars
-    engine.input.gsub(/^/, ' ' * 8) %>
+<%= engine.input.gsub(/^/, ' ' * 8) %>
 
-    result = <%= prob.name %>.<%= func.name %>(<%= engine.vars_list %>)
+    result = <%= prob.name %>.<%= func.name %>(<%= engine.arglist %>)
     with open(sys.argv[2], "w") as fo:
         fo.write(tc.write(result, "<%= func.type %>"))