C++ Problem::getClassName方法代码示例

/**
 * Generate the actual code from problem object
 */
std::string CppCode::generateCode(Problem& problem) {
    //All header file list
    std::string theCode = this->cppHeaderfiles();
    theCode += std::string("\n#define Zclr(arr) memset(arr, 0, sizeof(arr))\n") +
                "#define Nclr(arr) memset(arr, -1, sizeof(arr))\n" +
                "#define M31 1<<30\n" +
                "#define Pi (2.0*acos(0.0))\n" +
                "#define Eps (1e-9)\n"+
                "#define Rep(_var, _N) for (int _var=0; _var<_N; _var++ )\n"+
                "#define For(_var, _S,  _N) for (int _var=_S; _var<=_N; _var++ )\n\n";

    theCode += "class "+problem.getClassName()+" {\n" +
               "public:\n"+this->methodCode(problem.getMethod()) +
               "};\n\n"; //end of class code

    theCode += "// <editor-fold defaultstate=\"collapsed\" desc=\"Generated Code\">\n";
    theCode += "namespace otocoder {\n" +
               this->generateOutputComparison(problem.getMethod()->getReturnVariable())+
               this->generateRunTest() +
               this->generateFormatResult(problem.getMethod()->getReturnVariable()) +
               this->generateVerifyCase(problem) +
               this->generateRunTestCase(problem) +
               this->generateFullTestReader(problem)+
               "}\n\n";
    theCode += this->generateDefaultMain();
    theCode += "// </editor-fold>\n";

    return theCode;
}

std::string CppCode::generateRunTestCase(Problem& problem) {
    std::string mCode;
    mCode = "    int run_test_case(int casenum) {\n";
    mCode += "        switch (casenum) {\n";
    char arr[200];
    // Generate the individual test cases
    int totalCases = problem.getSampleTestSize();
    for (int i = 0; i < totalCases; ++i) {
        sprintf(arr, "        case %d: {\n", i);
        mCode += arr;
        mCode += this->generateTestCase(i, problem.getClassName(), problem.getMethod(), problem.getSampleTestCaseAt(i));
        mCode += "        }\n";
    }

    // next
    mCode +=std::string("        default:\n") +
            "            return -1;\n" +
            "        }\n" +
            "    }\n\n";
    return mCode;
}

std::string CppCode::generateFullTestReader(Problem& problem) {
    Method* method = problem.getMethod();
    Variable* var;
    std::string code =  
        tabStr + "const std::string currentDateTime() {\n"+
        tabStr + tabStr + "time_t     now = time(0);\n"+
        tabStr + tabStr + "struct tm  tstruct;\n"+
        tabStr + tabStr + "char       buf[80];\n" +
        tabStr + tabStr + "tstruct = *localtime(&now);\n"+
        tabStr + tabStr + "strftime(buf, sizeof(buf), \"%Y-%m-%d %X\", &tstruct);\n"+
        tabStr + tabStr + "return buf;\n"+
        tabStr+"}\n\n"+
        tabStr+"//This method will only accept input files that are formatted in proper way.\n"+
        tabStr+"//Command line example: "+problem.getClassName()+".exe -2 <"+problem.getClassName()+".io\n"+
        tabStr+"bool run_full_test() {\n" +
        tabStr+tabStr+"int testCase, N, caseNo, correct=0;\n";
    //declare all variables
    for (int i=0; i<method->getNumberOfParameters(); i++) {
        var = method->getParameterAt(i);
        code+= tabStr+tabStr+this->generateDatatype(var)+" "+var->getName()+";\n";//vector<int> var;
        //if we have an array then we will create another variable to read single item of that array, redundant but easy to code :|
        if (var->isArray()) {
            code += tabStr+tabStr+this->convertDataType(var->getDataType())+" "+var->getName()+"_sin;\n"; //with _in suffix
        }
    }
    var = method->getReturnVariable();
    code+= tabStr+tabStr+this->generateDatatype(var)+" "+var->getName()+";\n";//vector<int> var;
    if (var->isArray()) {
        code += tabStr+tabStr+this->convertDataType(var->getDataType())+" "+var->getName()+"_sin;\n"; //with _in suffix
    }
    code += "\n";
    code+= tabStr+tabStr+"std::cin>>testCase; std::cin.ignore(10, '\\n');\n" +
        tabStr+tabStr+"for (int i=0; i<testCase; i++) {\n" +
        tabStr+tabStr+tabStr+"std::cin>>caseNo; std::cin.ignore(10, '\\n');\n";
    //code to read all parameters
    for (int i=0; i<method->getNumberOfParameters(); i++) {
        code += this->generateSingleCaseReader(method->getParameterAt(i), tabStr+tabStr);
    }
    //code to read expected result
    code += this->generateSingleCaseReader(method->getReturnVariable(), tabStr+tabStr);

    //code to run class.Method(parameters...)
    code += tabStr+tabStr+tabStr+"clock_t start_           = clock();\n";
    code += tabStr+tabStr+tabStr+this->generateDatatype(method->getReturnVariable())+" received_ = " +
        problem.getClassName()+"()."+method->getName()+"(";

    for (int i=0; i<method->getNumberOfParameters(); i++) {
        if (i) {
            code += ", ";
        }
        code += method->getParameterAt(i)->getName();
    }
    code += ");\n";
    code += tabStr+tabStr+tabStr+"correct += verify_case(caseNo, " + method->getReturnVariable()->getName()  + ", received_, clock()-start_);\n";

    code+=tabStr+tabStr+"}//end of testCase for loop\n\n";
    code += tabStr+tabStr +"if (testCase == 0) {\n" +
            tabStr+tabStr+tabStr+"cerr << \"No test cases run.\" << endl;\n" +
            tabStr+tabStr+"} else if (correct < testCase) {\n"+
            tabStr+tabStr+tabStr+"cerr << \"Some cases FAILED (passed \" << correct << \" of \" << testCase << \").\" << endl;\n"+
            tabStr+tabStr+"} else {\n"+
            tabStr+tabStr+tabStr+"cerr << \"All \" << testCase << \" tests passed!\" << endl;\n"+
            tabStr+tabStr+tabStr+"std::ofstream ofs(\""+problem.getClassName()+".st\", std::ios_base::app);\n"+
            tabStr+tabStr+tabStr+"ofs<<currentDateTime()<<std::endl;\n"+
            tabStr+tabStr+tabStr+"ofs.close();\n"+
            tabStr+tabStr+"}\n"+
            tabStr+tabStr+"return true;\n"+
          tabStr+"}\n\n";
    return code;
}