Merge pull request #26 from online-judge-tools/fix/topcoder

Update about Topcoder

onlinejudge_template/generator/_topcoder.py → onlinejudge_template/generator/topcoder.py

@@ -3,6 +3,11 @@
 from onlinejudge_template.generator._utils import get_analyzed
 
 
+def is_topcoder(data: Dict[str, Any]) -> bool:
+    definition = get_analyzed(data).topcoder_class_definition
+    return definition is not None
+
+
 def class_name(data: Dict[str, Any]) -> str:
     definition = get_analyzed(data).topcoder_class_definition
     if definition is None:

onlinejudge_template_resources/template/main.cpp

@@ -3,6 +3,7 @@
     import platform
 
     import onlinejudge_template.generator.cplusplus as cplusplus
+    import onlinejudge_template.generator.topcoder as topcoder
     import onlinejudge_template.generator.about as about
 %>\
 <%
@@ -27,17 +28,28 @@
 using namespace std;
 
 ${cplusplus.declare_constants(data)}
+% if topcoder.is_topcoder(data):
+<% solve_function = topcoder.class_name(data) + "()." + topcoder.method_name(data) %>\
+class ${topcoder.class_name(data)} {
+public:
+    ${cplusplus.return_type(data)} ${topcoder.method_name(data)}(${cplusplus.formal_arguments(data)}) {
+        // TODO: edit here
+    }
+};
+% else:
+<% solve_function = "solve" %>\
 ${cplusplus.return_type(data)} solve(${cplusplus.formal_arguments(data)}) {
     // TODO: edit here
 }
+% endif
 
 // generated by ${about.title} ${about.version} (${about.url})
 int main() {
     std::ios::sync_with_stdio(false);
     std::cin.tie(nullptr);
     constexpr char endl = '\n';
 ${cplusplus.read_input(data)}
-    auto ${cplusplus.return_value(data)} = solve(${cplusplus.actual_arguments(data)});
+    auto ${cplusplus.return_value(data)} = ${solve_function}(${cplusplus.actual_arguments(data)});
 ${cplusplus.write_output(data)}
     return 0;
 }