support go (#235)

* support go

* goのcompilerテストを追加

* タブインデントのロジックを変更

Author: chaemon

atcodertools/codegen/code_generators/go.py

@@ -0,0 +1,16 @@
+from atcodertools.codegen.models.code_gen_args import CodeGenArgs
+from atcodertools.codegen.template_engine import render
+
+from atcodertools.codegen.code_generators.universal_code_generator import UniversalCodeGenerator, get_builtin_code_generator_info_toml_path
+
+
+def main(args: CodeGenArgs) -> str:
+    code_parameters = UniversalCodeGenerator(
+        args.format, args.config, get_builtin_code_generator_info_toml_path("go")).generate_parameters()
+    return render(
+        args.template,
+        mod=args.constants.mod,
+        yes_str=args.constants.yes_str,
+        no_str=args.constants.no_str,
+        **code_parameters
+    )

atcodertools/codegen/code_generators/universal_generator/go.toml

@@ -0,0 +1,59 @@
+base_indent = 1
+insert_space_around_operators = false
+
+# global変数宣言時の接頭辞
+global_prefix = ""
+
+# ループ
+[loop]
+header = "for {loop_var} := int64(0); {loop_var} < {length}; {loop_var}++ {{"
+footer = "}}"
+
+# タイプ
+[type]
+int = "int64"
+float = "float64"
+str = "string"
+
+# デフォルト値
+[default]
+int = "0"
+float = "0.0"
+str = '""'
+
+# 引数
+[arg]
+int = "{name} int64"
+float = "{name} float64"
+str = "{name} string"
+seq = "{name} []{type}"
+2d_seq = "{name} [][]{type}"
+
+# 配列アクセス
+[access]
+seq = "{name}[{index}]"
+2d_seq = "{name}[{index_i}][{index_j}]"
+
+# 宣言
+[declare]
+int = "var {name} int64"
+float = "var {name} float64"
+str = "var {name} string"
+seq = "var {name} []{type}"
+2d_seq = "var {name} [][]{type}"
+
+# 確保
+[allocate]
+seq = "{name} = make([]{type}, {length})"
+2d_seq = "{name} = make([][]{type}, {length_i})\nfor i := int64(0); i < {length_i}; i++ {{\n\t{name}[i] = make([]{type}, {length_j})\n}}"
+
+# 宣言と確保
+[declare_and_allocate]
+seq = "{name} := make([]{type}, {length})"
+2d_seq = "{name} := make([][]{type}, {length_i})\nfor i := int64(0); i < {length_i}; i++ {{\n\t{name}[i] = make([]{type}, {length_j})\n}}"
+
+# 入力
+[input]
+int = "scanner.Scan()\n{name}, _ = strconv.ParseInt(scanner.Text(), 10, 64)"
+float = "scanner.Scan()\n{name}, _ = strconv.ParseFloat(scanner.Text(), 64)"
+str = "scanner.Scan()\n{name} = scanner.Text()"

atcodertools/codegen/code_style_config.py

@@ -19,7 +19,7 @@ class CodeStyleConfigInitError(Exception):
 class CodeStyleConfig:
 
     def __init__(self,
-                 indent_type: str = INDENT_TYPE_SPACE,
+                 indent_type: str = None,
                  indent_width: Optional[int] = None,
                  code_generator_file: Optional[str] = None,
                  template_file: Optional[str] = None,
@@ -37,7 +37,7 @@ def __init__(self,
             raise CodeStyleConfigInitError(
                 "language must be one of {}".format(ALL_LANGUAGE_NAMES))
 
-        if indent_type not in [INDENT_TYPE_SPACE, INDENT_TYPE_TAB]:
+        if indent_type is not None and indent_type not in [INDENT_TYPE_SPACE, INDENT_TYPE_TAB]:
             raise CodeStyleConfigInitError(
                 "indent_type must be 'space' or 'tab'")
 
@@ -54,15 +54,25 @@ def __init__(self,
                 "The specified template file '{}' is not found".format(
                     template_file)
             )
-
-        self.indent_type = indent_type
+        if indent_type is not None:
+            self.indent_type = indent_type
+        elif lang.default_code_style is not None and lang.default_code_style.indent_type is not None:
+            self.indent_type = lang.default_code_style.indent_type
+        else:
+            self.indent_type = INDENT_TYPE_SPACE
 
         if indent_width is not None:
             self.indent_width = indent_width
         elif lang.default_code_style is not None and lang.default_code_style.indent_width is not None:
             self.indent_width = lang.default_code_style.indent_width
         else:
-            self.indent_width = 4
+            if self.indent_type == INDENT_TYPE_SPACE:
+                self.indent_width = 4
+            elif self.indent_type == INDENT_TYPE_TAB:
+                self.indent_width = 1
+            else:
+                raise CodeStyleConfigInitError(
+                    "indent_type must be 'space' or 'tab'")
 
         if code_generator_file is not None:
             try:
@@ -84,4 +94,8 @@ def __init__(self,
     def indent(self, depth):
         if self.indent_type == INDENT_TYPE_SPACE:
             return " " * self.indent_width * depth
-        return "\t" * self.indent_width * depth
+        elif self.indent_type == INDENT_TYPE_TAB:
+            return "\t" * self.indent_width * depth
+        else:
+            raise CodeStyleConfigInitError(
+                "indent_type must be 'space' or 'tab'")

atcodertools/common/language.py

@@ -1,9 +1,10 @@
 import re
 from typing import Pattern, Callable
 
-from atcodertools.codegen.code_generators import cpp, java, rust, python, nim, d, cs, swift
+from atcodertools.codegen.code_generators import cpp, java, rust, python, nim, d, cs, swift, go
 from atcodertools.codegen.models.code_gen_args import CodeGenArgs
 from atcodertools.tools.templates import get_default_template_path
+from atcodertools.codegen.code_style_config import INDENT_TYPE_TAB
 import platform
 
 
@@ -13,9 +14,11 @@ class LanguageNotFoundError(Exception):
 
 class CodeStyle:
     def __init__(self,
-                 indent_width=None
+                 indent_width=None,
+                 indent_type=None
                  ):
         self.indent_width = indent_width
+        self.indent_type = indent_type
 
 
 class Language:
@@ -178,6 +181,19 @@ def from_name(cls, name: str):
     exec_filename="{filename}{exec_extension}"
 )
 
+GO = Language(
+    name="go",
+    display_name="Go",
+    extension="go",
+    submission_lang_pattern=re.compile(".*Go \\(1.*"),
+    default_code_generator=go.main,
+    default_template_path=get_default_template_path('go'),
+    default_code_style=CodeStyle(indent_type=INDENT_TYPE_TAB),
+    compile_command="go build -o {filename} {filename}.go",
+    test_command="{exec_filename}",
+    exec_filename="{filename}{exec_extension}"
+)
+
 
-ALL_LANGUAGES = [CPP, JAVA, RUST, PYTHON, NIM, DLANG, CSHARP, SWIFT]
+ALL_LANGUAGES = [CPP, JAVA, RUST, PYTHON, NIM, DLANG, CSHARP, SWIFT, GO]
 ALL_LANGUAGE_NAMES = [lang.display_name for lang in ALL_LANGUAGES]

atcodertools/tools/templates/default_template.go

@@ -0,0 +1,41 @@
+package main
+{% if prediction_success %}
+
+import (
+	"bufio"
+	"os"
+	"strconv"
+)
+{% endif %}
+{% if mod or yes_str or no_str %}
+
+{% endif %}
+{% if mod %}
+const MOD = {{mod}}
+{% endif %}
+{% if yes_str %}
+const YES = "{{ yes_str }}"
+{% endif %}
+{% if no_str %}
+const NO = "{{ no_str }}"
+{% endif %}
+{% if prediction_success %}
+
+func solve({{ formal_arguments }}) {
+
+}
+{% endif %}
+
+func main() {
+	{% if prediction_success %}
+	scanner := bufio.NewScanner(os.Stdin)
+	const initialBufSize = 4096
+	const maxBufSize = 1000000
+	scanner.Buffer(make([]byte, initialBufSize), maxBufSize)
+	scanner.Split(bufio.ScanWords)
+	{{ input_part }}
+	solve({{ actual_arguments }})
+	{% else %}
+    // Failed to predict input format
+	{% endif %}
+}

tests/resources/test_codegen/template.go

@@ -0,0 +1,21 @@
+package main
+
+import (
+	"bufio"
+	"os"
+	"strconv"
+)
+
+func solve(${formal_arguments}) {
+
+}
+
+func main() {
+	scanner := bufio.NewScanner(os.Stdin)
+	const initialBufSize = 4096
+	const maxBufSize = 1000000
+	scanner.Buffer(make([]byte, initialBufSize), maxBufSize)
+	scanner.Split(bufio.ScanWords)
+	${input_part}
+	solve(${actual_arguments})
+}

tests/resources/test_codegen/template_jinja.go

@@ -0,0 +1,41 @@
+package main
+{% if prediction_success %}
+
+import (
+	"bufio"
+	"os"
+	"strconv"
+)
+{% endif %}
+{% if mod or yes_str or no_str %}
+
+{% endif %}
+{% if mod %}
+const MOD = {{mod}}
+{% endif %}
+{% if yes_str %}
+const YES = "{{ yes_str }}"
+{% endif %}
+{% if no_str %}
+const NO = "{{ no_str }}"
+{% endif %}
+{% if prediction_success %}
+
+func solve({{ formal_arguments }}) {
+
+}
+{% endif %}
+
+func main() {
+	{% if prediction_success %}
+	scanner := bufio.NewScanner(os.Stdin)
+	const initialBufSize = 4096
+	const maxBufSize = 1000000
+	scanner.Buffer(make([]byte, initialBufSize), maxBufSize)
+	scanner.Split(bufio.ScanWords)
+	{{ input_part }}
+	solve({{ actual_arguments }})
+	{% else %}
+	// Failed to predict input format
+	{% endif %}
+}

tests/resources/test_codegen/test_default_code_generators_and_templates/go/echo_template.go

@@ -0,0 +1,81 @@
+package main
+{% if prediction_success %}
+
+import (
+	"bufio"
+	"fmt"
+	"log"
+	"os"
+	"strconv"
+)
+{% endif %}
+{% if mod or yes_str or no_str %}
+
+{% endif %}
+{% if mod %}
+const MOD = {{mod}}
+{% endif %}
+{% if yes_str %}
+const YES = "{{ yes_str }}"
+{% endif %}
+{% if no_str %}
+const NO = "{{ no_str }}"
+{% endif %}
+{% if prediction_success %}
+
+func solve({{ formal_arguments }}) {
+	fmt.Printf("%d %d\n", N, M)
+	if int64(len(H)) != N-1 {
+		log.Fatal()
+	}
+	for i := int64(0); i < N-1; i++ {
+		if int64(len(H[i])) != M-2 {
+			log.Fatal()
+		}
+		for j := int64(0); j < M-2; j++ {
+			if j > 0 {
+				fmt.Printf(" %s", H[i][j])
+			} else {
+				fmt.Printf("%s", H[i][j])
+			}
+		}
+		fmt.Println()
+	}
+
+	if int64(len(A)) != N-1 {
+		log.Fatal()
+	}
+	if int64(len(B)) != N-1 {
+		log.Fatal()
+	}
+	for i := int64(0); i < N-1; i++ {
+		fmt.Printf("%d %.1f\n", A[i], B[i])
+	}
+
+	fmt.Println(Q)
+	if int64(len(X)) != M+Q {
+		log.Fatal()
+	}
+	for i := int64(0); i < M+Q; i++ {
+		fmt.Println(X[i])
+	}
+
+	fmt.Println(YES)
+	fmt.Println(NO)
+	fmt.Println(MOD)
+}
+{% endif %}
+
+func main() {
+	{% if prediction_success %}
+	scanner := bufio.NewScanner(os.Stdin)
+	const initialBufSize = 4096
+	const maxBufSize = 1000000
+	scanner.Buffer(make([]byte, initialBufSize), maxBufSize)
+	scanner.Split(bufio.ScanWords)
+	{{ input_part }}
+	solve({{ actual_arguments }})
+	{% else %}
+    // Failed to predict input format
+	{% endif %}
+}