diff --git a/.gitignore b/.gitignore index b013f6e36..0118caa80 100644 --- a/.gitignore +++ b/.gitignore @@ -1,4 +1,6 @@ .idea uoj_data/ +confs/ .php-cs-fixer.cache .config.local.php +*.pcap \ No newline at end of file diff --git a/docker-compose.yml b/docker-compose.yml index 7dbae208b..b9487f380 100644 --- a/docker-compose.yml +++ b/docker-compose.yml @@ -8,6 +8,8 @@ services: dockerfile: Dockerfile container_name: uoj-db restart: always + ports: + - "3306:3306" volumes: - ./uoj_data/db/mysql:/var/lib/mysql environment: @@ -15,7 +17,7 @@ services: - MYSQL_ROOT_PASSWORD=root uoj-judger: - image: ghcr.io/universaloj/uoj-judger:latest + image: yanagiorigami/uoj-judger:latest build: context: ./judger/ dockerfile: Dockerfile @@ -27,6 +29,7 @@ services: - SYS_PTRACE volumes: - ./uoj_data/judger/log:/opt/uoj_judger/log + #- ./judger/uoj_judger/include:/opt/uoj_judger/include environment: - UOJ_PROTOCOL=http - UOJ_HOST=uoj-web @@ -36,7 +39,7 @@ services: - SOCKET_PASSWORD=_judger_socket_password_ uoj-web: - image: ghcr.io/universaloj/uoj-web:latest + image: yanagiorigami/uoj-web:latest build: context: ./ dockerfile: web/Dockerfile diff --git a/judger/Dockerfile b/judger/Dockerfile index 2b1d5786a..7d6680d73 100644 --- a/judger/Dockerfile +++ b/judger/Dockerfile @@ -12,7 +12,7 @@ ARG CLONE_ADDFLAG RUN apt-get update && \ apt-get install -y --no-install-recommends gnupg ca-certificates apt-transport-https && \ apt-get update && \ - for pkg in git vim ntp zip unzip curl wget build-essential fp-compiler python python3 python3-requests openjdk-8-jdk openjdk-11-jdk tzdata; do \ + for pkg in git vim ntp zip unzip curl wget build-essential fp-compiler python python3 python3-requests pypy3 openjdk-8-jdk openjdk-11-jdk tzdata; do \ cnt=10 && \ while ! apt-get install -y "$pkg"; do \ if [ $cnt -le 0 ]; then \ @@ -23,6 +23,9 @@ RUN apt-get update && \ done; \ done +ENV PYTHONNOUSERSITE=1 +ENV PYTHONDONTWRITEBYTECODE=1 + ADD . /opt/uoj_judger WORKDIR /opt/uoj_judger diff --git a/judger/install.sh b/judger/install.sh index 5300fc016..6db180202 100644 --- a/judger/install.sh +++ b/judger/install.sh @@ -3,7 +3,22 @@ getAptPackage(){ printf "\n\n==> Getting environment packages\n" export DEBIAN_FRONTEND=noninteractive + + apt-get update && apt-get install -y software-properties-common + + # ✅ 添加新的 GCC 工具链源 + add-apt-repository ppa:ubuntu-toolchain-r/test -y + apt-get update + + # ✅ 安装 g++-13 + apt-get install -y g++-13 gcc-13 + + # ✅ 设置 g++ 默认版本为 g++-13 + update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-13 100 + update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-13 100 + apt-get update && apt-get install -y vim ntp zip unzip curl wget build-essential fp-compiler python2.7 python3.8 python3-requests + apt-get install -y pypy3 } setJudgeConf(){ @@ -26,6 +41,7 @@ cat >uoj_judger/include/uoj_work_path.h < [ [-appes]], * If result file is specified it will contain results. * - * Validator, using testlib running format: + * Validator, using testlib running format: * validator.exe < input.txt, * It will return non-zero exit code and writes message to standard output. * - * Generator, using testlib running format: + * Generator, using testlib running format: * gen.exe [parameter-1] [parameter-2] [... paramerter-n] * You can write generated test(s) into standard output or into the file(s). * - * Interactor, using testlib running format: + * Interactor, using testlib running format: * interactor.exe [ [ [-appes]]], * Reads test from inf (mapped to args[1]), writes result to tout (mapped to argv[2], * can be judged by checker later), reads program output from ouf (mapped to stdin), * writes output to program via stdout (use cout, printf, etc). */ -const char* latestFeatures[] = { - "Removed disable buffers for interactive problems, because it works unexpectedly in wine", - "InStream over string: constructor of InStream from base InStream to inherit policies and std::string", - "Added expectedButFound quit function, examples: expectedButFound(_wa, 10, 20), expectedButFound(_fail, ja, pa, \"[n=%d,m=%d]\", n, m)", - "Fixed incorrect interval parsing in patterns", - "Use registerGen(argc, argv, 1) to develop new generator, use registerGen(argc, argv, 0) to compile old generators (originally created for testlib under 0.8.7)", - "Introduced disableFinalizeGuard() to switch off finalization checkings", - "Use join() functions to format a range of items as a single string (separated by spaces or other separators)", - "Use -DENABLE_UNEXPECTED_EOF to enable special exit code (by default, 8) in case of unexpected eof. It is good idea to use it in interactors", - "Use -DUSE_RND_AS_BEFORE_087 to compile in compatibility mode with random behavior of versions before 0.8.7", - "Fixed bug with nan in stringToDouble", - "Fixed issue around overloads for size_t on x64", - "Added attribute 'points' to the XML output in case of result=_points", - "Exit codes can be customized via macros, e.g. -DPE_EXIT_CODE=14", - "Introduced InStream function readWordTo/readTokenTo/readStringTo/readLineTo for faster reading", - "Introduced global functions: format(), englishEnding(), upperCase(), lowerCase(), compress()", - "Manual buffer in InStreams, some IO speed improvements", - "Introduced quitif(bool, const char* pattern, ...) which delegates to quitf() in case of first argument is true", - "Introduced guard against missed quitf() in checker or readEof() in validators", - "Supported readStrictReal/readStrictDouble - to use in validators to check strictly float numbers", - "Supported registerInteraction(argc, argv)", - "Print checker message to the stderr instead of stdout", - "Supported TResult _points to output calculated score, use quitp(...) functions", - "Fixed to be compilable on Mac", - "PC_BASE_EXIT_CODE=50 in case of defined TESTSYS", - "Fixed issues 19-21, added __attribute__ format printf", - "Some bug fixes", - "ouf.readInt(1, 100) and similar calls return WA", - "Modified random_t to avoid integer overflow", - "Truncated checker output [patch by Stepan Gatilov]", - "Renamed class random -> class random_t", - "Supported name parameter for read-and-validation methods, like readInt(1, 2, \"n\")", - "Fixed bug in readDouble()", - "Improved ensuref(), fixed nextLine to work in case of EOF, added startTest()", - "Supported \"partially correct\", example: quitf(_pc(13), \"result=%d\", result)", - "Added shuffle(begin, end), use it instead of random_shuffle(begin, end)", - "Added readLine(const string& ptrn), fixed the logic of readLine() in the validation mode", - "Package extended with samples of generators and validators", - "Written the documentation for classes and public methods in testlib.h", - "Implemented random routine to support generators, use registerGen() to switch it on", - "Implemented strict mode to validate tests, use registerValidation() to switch it on", - "Now ncmp.cpp and wcmp.cpp are return WA if answer is suffix or prefix of the output", - "Added InStream::readLong() and removed InStream::readLongint()", - "Now no footer added to each report by default (use directive FOOTER to switch on)", - "Now every checker has a name, use setName(const char* format, ...) to set it", - "Now it is compatible with TTS (by Kittens Computing)", - "Added \'ensure(condition, message = \"\")\' feature, it works like assert()", - "Fixed compatibility with MS C++ 7.1", - "Added footer with exit code information", - "Added compatibility with EJUDGE (compile with EJUDGE directive)" - }; +const char *latestFeatures[] = { + "Use setAppesModeEncoding to change xml encoding from windows-1251 to other", + "rnd.any/wany use distance/advance instead of -/+: now they support sets/multisets", + "Use syntax `int t = inf.readInt(1, 3, \"~t\");` to skip the lower bound check. Tildes can be used on either side or both: ~t, t~, ~t~", + "Supported EJUDGE support in registerTestlibCmd", + "Supported '--testMarkupFileName fn' and '--testCase tc/--testCaseFileName fn' for validators", + "Added opt defaults via opt(key/index, default_val); check unused opts when using has_opt or default opt (turn off this check with suppressEnsureNoUnusedOpt()).", + "For checker added --group and --testset command line params (like for validator), use checker.group() or checker.testset() to get values", + "Added quitpi(points_info, message) function to return with _points exit code 7 and given points_info", + "rnd.partition(size, sum[, min_part=1]) returns random (unsorted) partition which is a representation of the given `sum` as a sum of `size` positive integers (or >=min_part if specified)", + "rnd.distinct(size, n) and rnd.distinct(size, from, to)", + "opt(\"some_missing_key\") returns false now", + "has_opt(key)", + "Abort validator on validator.testset()/validator.group() if registered without using command line", + "Print integer range violations in a human readable way like `violates the range [1, 10^9]`", + "Opts supported: use them like n = opt(\"n\"), in a command line you can use an exponential notation", + "Reformatted", + "Use setTestCase(i) or unsetTestCase() to support test cases (you can use it in any type of program: generator, interactor, validator or checker)", + "Fixed issue #87: readStrictDouble accepts \"-0.00\"", + "Fixed issue #83: added InStream::quitif(condition, ...)", + "Fixed issue #79: fixed missed guard against repeated header include", + "Fixed issue #80: fixed UB in case of huge quitf message", + "Fixed issue #84: added readXs(size, indexBase = 1)", + "Fixed stringstream repeated usage issue", + "Fixed compilation in g++ (for std=c++03)", + "Batch of println functions (support collections, iterator ranges)", + "Introduced rnd.perm(size, first = 0) to generate a `first`-indexed permutation", + "Allow any whitespace in readInts-like functions for non-validators", + "Ignore 4+ command line arguments ifdef EJUDGE", + "Speed up of vtos", + "Show line number in validators in case of incorrect format", + "Truncate huge checker/validator/interactor message", + "Fixed issue with readTokenTo of very long tokens, now aborts with _pe/_fail depending of a stream type", + "Introduced InStream::ensure/ensuref checking a condition, returns wa/fail depending of a stream type", + "Fixed compilation in VS 2015+", + "Introduced space-separated read functions: readWords/readTokens, multilines read functions: readStrings/readLines", + "Introduced space-separated read functions: readInts/readIntegers/readLongs/readUnsignedLongs/readDoubles/readReals/readStrictDoubles/readStrictReals", + "Introduced split/tokenize functions to separate string by given char", + "Introduced InStream::readUnsignedLong and InStream::readLong with unsigned long long parameters", + "Supported --testOverviewLogFileName for validator: bounds hits + features", + "Fixed UB (sequence points) in random_t", + "POINTS_EXIT_CODE returned back to 7 (instead of 0)", + "Removed disable buffers for interactive problems, because it works unexpectedly in wine", + "InStream over string: constructor of InStream from base InStream to inherit policies and std::string", + "Added expectedButFound quit function, examples: expectedButFound(_wa, 10, 20), expectedButFound(_fail, ja, pa, \"[n=%d,m=%d]\", n, m)", + "Fixed incorrect interval parsing in patterns", + "Use registerGen(argc, argv, 1) to develop new generator, use registerGen(argc, argv, 0) to compile old generators (originally created for testlib under 0.8.7)", + "Introduced disableFinalizeGuard() to switch off finalization checkings", + "Use join() functions to format a range of items as a single string (separated by spaces or other separators)", + "Use -DENABLE_UNEXPECTED_EOF to enable special exit code (by default, 8) in case of unexpected eof. It is good idea to use it in interactors", + "Use -DUSE_RND_AS_BEFORE_087 to compile in compatibility mode with random behavior of versions before 0.8.7", + "Fixed bug with nan in stringToDouble", + "Fixed issue around overloads for size_t on x64", + "Added attribute 'points' to the XML output in case of result=_points", + "Exit codes can be customized via macros, e.g. -DPE_EXIT_CODE=14", + "Introduced InStream function readWordTo/readTokenTo/readStringTo/readLineTo for faster reading", + "Introduced global functions: format(), englishEnding(), upperCase(), lowerCase(), compress()", + "Manual buffer in InStreams, some IO speed improvements", + "Introduced quitif(bool, const char* pattern, ...) which delegates to quitf() in case of first argument is true", + "Introduced guard against missed quitf() in checker or readEof() in validators", + "Supported readStrictReal/readStrictDouble - to use in validators to check strictly float numbers", + "Supported registerInteraction(argc, argv)", + "Print checker message to the stderr instead of stdout", + "Supported TResult _points to output calculated score, use quitp(...) functions", + "Fixed to be compilable on Mac", + "PC_BASE_EXIT_CODE=50 in case of defined TESTSYS", + "Fixed issues 19-21, added __attribute__ format printf", + "Some bug fixes", + "ouf.readInt(1, 100) and similar calls return WA", + "Modified random_t to avoid integer overflow", + "Truncated checker output [patch by Stepan Gatilov]", + "Renamed class random -> class random_t", + "Supported name parameter for read-and-validation methods, like readInt(1, 2, \"n\")", + "Fixed bug in readDouble()", + "Improved ensuref(), fixed nextLine to work in case of EOF, added startTest()", + "Supported \"partially correct\", example: quitf(_pc(13), \"result=%d\", result)", + "Added shuffle(begin, end), use it instead of random_shuffle(begin, end)", + "Added readLine(const string& ptrn), fixed the logic of readLine() in the validation mode", + "Package extended with samples of generators and validators", + "Written the documentation for classes and public methods in testlib.h", + "Implemented random routine to support generators, use registerGen() to switch it on", + "Implemented strict mode to validate tests, use registerValidation() to switch it on", + "Now ncmp.cpp and wcmp.cpp are return WA if answer is suffix or prefix of the output", + "Added InStream::readLong() and removed InStream::readLongint()", + "Now no footer added to each report by default (use directive FOOTER to switch on)", + "Now every checker has a name, use setName(const char* format, ...) to set it", + "Now it is compatible with TTS (by Kittens Computing)", + "Added \'ensure(condition, message = \"\")\' feature, it works like assert()", + "Fixed compatibility with MS C++ 7.1", + "Added footer with exit code information", + "Added compatibility with EJUDGE (compile with EJUDGE directive)", + "Added compatibility with Contester (compile with CONTESTER directive)" +}; #ifdef _MSC_VER #define _CRT_SECURE_NO_DEPRECATE #define _CRT_SECURE_NO_WARNINGS +#define _CRT_NO_VA_START_VALIDATION #endif /* Overrides random() for Borland C++. */ -#ifdef __EMSCRIPTEN__ -#define srand __srand_deprecated -#define rand __rand_deprecated -#endif #define random __random_deprecated #include #include #include #include #undef random -#ifdef __EMSCRIPTEN__ -#undef rand -#undef srand -#endif #include #include #include #include +#include +#include #include +#include +#include #include #include #include #include #include - #include +#include +#include -#ifdef __EMSCRIPTEN__ -#include +#ifdef TESTLIB_THROW_EXIT_EXCEPTION_INSTEAD_OF_EXIT +# include #endif -#if ( _WIN32 || __WIN32__ || _WIN64 || __WIN64__ ) -# if !defined(_MSC_VER) || _MSC_VER>1400 +#if (_WIN32 || __WIN32__ || __WIN32 || _WIN64 || __WIN64__ || __WIN64 || WINNT || __WINNT || __WINNT__ || __CYGWIN__) +# if !defined(_MSC_VER) || _MSC_VER > 1400 +# define NOMINMAX 1 # include # else # define WORD unsigned short +# include # endif # include # define ON_WINDOWS +# if defined(_MSC_VER) && _MSC_VER > 1400 +# pragma warning( disable : 4127 ) +# pragma warning( disable : 4146 ) +# pragma warning( disable : 4458 ) +# endif #else # define WORD unsigned short +# include #endif -#ifdef linux -#include +#if defined(FOR_WINDOWS) && defined(FOR_LINUX) +#error Only one target system is allowed #endif #ifndef LLONG_MIN #define LLONG_MIN (-9223372036854775807LL - 1) #endif +#ifndef ULLONG_MAX +#define ULLONG_MAX (18446744073709551615) +#endif + #define LF ((char)10) #define CR ((char)13) #define TAB ((char)9) #define SPACE ((char)' ') -#define EOFC ((char)26) +#define EOFC (255) #ifndef OK_EXIT_CODE -# define OK_EXIT_CODE 0 +# ifdef CONTESTER +# define OK_EXIT_CODE 0xAC +# else +# define OK_EXIT_CODE 0 +# endif #endif #ifndef WA_EXIT_CODE # ifdef EJUDGE # define WA_EXIT_CODE 5 +# elif defined(CONTESTER) +# define WA_EXIT_CODE 0xAB # else # define WA_EXIT_CODE 1 # endif @@ -193,6 +250,8 @@ const char* latestFeatures[] = { #ifndef PE_EXIT_CODE # ifdef EJUDGE # define PE_EXIT_CODE 4 +# elif defined(CONTESTER) +# define PE_EXIT_CODE 0xAA # else # define PE_EXIT_CODE 2 # endif @@ -201,6 +260,8 @@ const char* latestFeatures[] = { #ifndef FAIL_EXIT_CODE # ifdef EJUDGE # define FAIL_EXIT_CODE 6 +# elif defined(CONTESTER) +# define FAIL_EXIT_CODE 0xA3 # else # define FAIL_EXIT_CODE 3 # endif @@ -215,7 +276,7 @@ const char* latestFeatures[] = { #endif #ifndef POINTS_EXIT_CODE -# define POINTS_EXIT_CODE 0 +# define POINTS_EXIT_CODE 7 #endif #ifndef UNEXPECTED_EOF_EXIT_CODE @@ -252,7 +313,8 @@ const char* latestFeatures[] = { # define NORETURN #endif -static char __testlib_format_buffer[1024]; +//static char __testlib_format_buffer[16777216]; +static char __testlib_format_buffer[4096]; static int __testlib_format_buffer_usage_count = 0; #define FMT_TO_RESULT(fmt, cstr, result) std::string result; \ @@ -261,35 +323,72 @@ static int __testlib_format_buffer_usage_count = 0; __testlib_format_buffer_usage_count++; \ va_list ap; \ va_start(ap, fmt); \ - std::vsprintf(__testlib_format_buffer, cstr, ap); \ + vsnprintf(__testlib_format_buffer, sizeof(__testlib_format_buffer), cstr, ap); \ va_end(ap); \ + __testlib_format_buffer[sizeof(__testlib_format_buffer) - 1] = 0; \ result = std::string(__testlib_format_buffer); \ __testlib_format_buffer_usage_count--; \ const long long __TESTLIB_LONGLONG_MAX = 9223372036854775807LL; +const int __TESTLIB_MAX_TEST_CASE = 1073741823; + +int __testlib_exitCode; + +bool __testlib_hasTestCase; +int __testlib_testCase = -1; + +void setTestCase(int testCase); + +void unsetTestCase() { + __testlib_hasTestCase = false; + __testlib_testCase = -1; +} -NORETURN static void __testlib_fail(const std::string& message); +NORETURN static void __testlib_fail(const std::string &message); template -static inline T __testlib_abs(const T& x) -{ +#ifdef __GNUC__ +__attribute__((const)) +#endif +static inline T __testlib_abs(const T &x) { return x > 0 ? x : -x; } template -static inline T __testlib_min(const T& a, const T& b) -{ +#ifdef __GNUC__ +__attribute__((const)) +#endif +static inline T __testlib_min(const T &a, const T &b) { return a < b ? a : b; } template -static inline T __testlib_max(const T& a, const T& b) -{ +#ifdef __GNUC__ +__attribute__((const)) +#endif +static inline T __testlib_max(const T &a, const T &b) { return a > b ? a : b; } -static bool __testlib_prelimIsNaN(double r) -{ +template +#ifdef __GNUC__ +__attribute__((const)) +#endif +static inline T __testlib_crop(T value, T a, T b) { + return __testlib_min(__testlib_max(value, a), --b); +} + +#ifdef __GNUC__ +__attribute__((const)) +#endif +static inline double __testlib_crop(double value, double a, double b) { + value = __testlib_min(__testlib_max(value, a), b); + if (value >= b) + value = std::nexttoward(b, a); + return value; +} + +static bool __testlib_prelimIsNaN(double r) { volatile double ra = r; #ifndef __BORLANDC__ return ((ra != ra) == true) && ((ra == ra) == false) && ((1.0 > ra) == false) && ((1.0 < ra) == false); @@ -298,102 +397,295 @@ static bool __testlib_prelimIsNaN(double r) #endif } -static std::string removeDoubleTrailingZeroes(std::string value) -{ +#ifdef __GNUC__ +__attribute__((const)) +#endif +static std::string removeDoubleTrailingZeroes(std::string value) { while (!value.empty() && value[value.length() - 1] == '0' && value.find('.') != std::string::npos) value = value.substr(0, value.length() - 1); - return value + '0'; + if (!value.empty() && value[value.length() - 1] == '.') + return value + '0'; + else + return value; +} + +#ifdef __GNUC__ +__attribute__((const)) +#endif +inline std::string upperCase(std::string s) { + for (size_t i = 0; i < s.length(); i++) + if ('a' <= s[i] && s[i] <= 'z') + s[i] = char(s[i] - 'a' + 'A'); + return s; +} + +#ifdef __GNUC__ +__attribute__((const)) +#endif +inline std::string lowerCase(std::string s) { + for (size_t i = 0; i < s.length(); i++) + if ('A' <= s[i] && s[i] <= 'Z') + s[i] = char(s[i] - 'A' + 'a'); + return s; } #ifdef __GNUC__ __attribute__ ((format (printf, 1, 2))) #endif -std::string format(const char* fmt, ...) -{ +std::string format(const char *fmt, ...) { FMT_TO_RESULT(fmt, fmt, result); return result; } -std::string format(const std::string& fmt, ...) -{ +std::string format(const std::string fmt, ...) { FMT_TO_RESULT(fmt, fmt.c_str(), result); return result; } -static std::string __testlib_part(const std::string& s); +#ifdef __GNUC__ +__attribute__((const)) +#endif +static std::string __testlib_part(const std::string &s); -static bool __testlib_isNaN(double r) -{ +static bool __testlib_isNaN(double r) { __TESTLIB_STATIC_ASSERT(sizeof(double) == sizeof(long long)); volatile double ra = r; long long llr1, llr2; - std::memcpy((void*)&llr1, (void*)&ra, sizeof(double)); + std::memcpy((void *) &llr1, (void *) &ra, sizeof(double)); ra = -ra; - std::memcpy((void*)&llr2, (void*)&ra, sizeof(double)); - long long llnan = 0xFFF8000000000000ll; + std::memcpy((void *) &llr2, (void *) &ra, sizeof(double)); + long long llnan = 0xFFF8000000000000LL; return __testlib_prelimIsNaN(r) || llnan == llr1 || llnan == llr2; } -static double __testlib_nan() -{ +static double __testlib_nan() { __TESTLIB_STATIC_ASSERT(sizeof(double) == sizeof(long long)); - long long llnan = 0xFFF8000000000000ll; +#ifndef NAN + long long llnan = 0xFFF8000000000000LL; double nan; std::memcpy(&nan, &llnan, sizeof(double)); return nan; +#else + return NAN; +#endif } -static bool __testlib_isInfinite(double r) -{ +static bool __testlib_isInfinite(double r) { volatile double ra = r; - return (ra > 1E100 || ra < -1E100); + return (ra > 1E300 || ra < -1E300); } -static void __testlib_set_binary(std::FILE* file) -{ -#ifdef O_BINARY - if (NULL != file) - { -#ifndef __BORLANDC__ - _setmode(_fileno(file), O_BINARY); -#else - setmode(fileno(file), O_BINARY); +#ifdef __GNUC__ +__attribute__((const)) #endif +inline bool doubleCompare(double expected, double result, double MAX_DOUBLE_ERROR) { + MAX_DOUBLE_ERROR += 1E-15; + if (__testlib_isNaN(expected)) { + return __testlib_isNaN(result); + } else if (__testlib_isInfinite(expected)) { + if (expected > 0) { + return result > 0 && __testlib_isInfinite(result); + } else { + return result < 0 && __testlib_isInfinite(result); + } + } else if (__testlib_isNaN(result) || __testlib_isInfinite(result)) { + return false; + } else if (__testlib_abs(result - expected) <= MAX_DOUBLE_ERROR) { + return true; + } else { + double minv = __testlib_min(expected * (1.0 - MAX_DOUBLE_ERROR), + expected * (1.0 + MAX_DOUBLE_ERROR)); + double maxv = __testlib_max(expected * (1.0 - MAX_DOUBLE_ERROR), + expected * (1.0 + MAX_DOUBLE_ERROR)); + return result >= minv && result <= maxv; } +} + +#ifdef __GNUC__ +__attribute__((const)) #endif +inline double doubleDelta(double expected, double result) { + double absolute = __testlib_abs(result - expected); + + if (__testlib_abs(expected) > 1E-9) { + double relative = __testlib_abs(absolute / expected); + return __testlib_min(absolute, relative); + } else + return absolute; } -static FILE *__testlib_fopen(const char *name, const char *mode) -{ -#ifdef __EMSCRIPTEN__ - EM_ASM( - try { - FS.stat('/cwd'); - } catch (e) { - FS.mkdir('/cwd'); - FS.mount(NODEFS, { root: '.' }, '/cwd'); - } - ); - return fopen((std::string("/cwd/") + name).c_str(), mode); +/** It does nothing on non-windows and files differ from stdin/stdout/stderr. */ +static void __testlib_set_binary(std::FILE *file) { + if (NULL != file) { +#ifdef ON_WINDOWS +# ifdef _O_BINARY + if (stdin == file) +# ifdef STDIN_FILENO + return void(_setmode(STDIN_FILENO, _O_BINARY)); +# else + return void(_setmode(_fileno(stdin), _O_BINARY)); +# endif + if (stdout == file) +# ifdef STDOUT_FILENO + return void(_setmode(STDOUT_FILENO, _O_BINARY)); +# else + return void(_setmode(_fileno(stdout), _O_BINARY)); +# endif + if (stderr == file) +# ifdef STDERR_FILENO + return void(_setmode(STDERR_FILENO, _O_BINARY)); +# else + return void(_setmode(_fileno(stderr), _O_BINARY)); +# endif +# elif O_BINARY + if (stdin == file) +# ifdef STDIN_FILENO + return void(setmode(STDIN_FILENO, O_BINARY)); +# else + return void(setmode(fileno(stdin), O_BINARY)); +# endif + if (stdout == file) +# ifdef STDOUT_FILENO + return void(setmode(STDOUT_FILENO, O_BINARY)); +# else + return void(setmode(fileno(stdout), O_BINARY)); +# endif + if (stderr == file) +# ifdef STDERR_FILENO + return void(setmode(STDERR_FILENO, O_BINARY)); +# else + return void(setmode(fileno(stderr), O_BINARY)); +# endif +# endif +#endif + } +} + +#if __cplusplus > 199711L || defined(_MSC_VER) +template +#ifdef __GNUC__ +__attribute__((const)) +#endif +static std::string vtos(const T &t, std::true_type) { + if (t == 0) + return "0"; + else { + T n(t); + bool negative = n < 0; + std::string s; + while (n != 0) { + T digit = n % 10; + if (digit < 0) + digit = -digit; + s += char('0' + digit); + n /= 10; + } + std::reverse(s.begin(), s.end()); + return negative ? "-" + s : s; + } +} + +template +static std::string vtos(const T &t, std::false_type) { + std::string s; + static std::stringstream ss; + ss.str(std::string()); + ss.clear(); + ss << t; + ss >> s; + return s; +} + +template +static std::string vtos(const T &t) { + return vtos(t, std::is_integral()); +} + +/* signed case. */ +template +static std::string toHumanReadableString(const T &n, std::false_type) { + if (n == 0) + return vtos(n); + int trailingZeroCount = 0; + T n_ = n; + while (n_ % 10 == 0) + n_ /= 10, trailingZeroCount++; + if (trailingZeroCount >= 7) { + if (n_ == 1) + return "10^" + vtos(trailingZeroCount); + else if (n_ == -1) + return "-10^" + vtos(trailingZeroCount); + else + return vtos(n_) + "*10^" + vtos(trailingZeroCount); + } else + return vtos(n); +} + +/* unsigned case. */ +template +static std::string toHumanReadableString(const T &n, std::true_type) { + if (n == 0) + return vtos(n); + int trailingZeroCount = 0; + T n_ = n; + while (n_ % 10 == 0) + n_ /= 10, trailingZeroCount++; + if (trailingZeroCount >= 7) { + if (n_ == 1) + return "10^" + vtos(trailingZeroCount); + else + return vtos(n_) + "*10^" + vtos(trailingZeroCount); + } else + return vtos(n); +} + +template +static std::string toHumanReadableString(const T &n) { + return toHumanReadableString(n, std::is_unsigned()); +} #else - return fopen(name, mode); +template +static std::string vtos(const T& t) +{ + std::string s; + static std::stringstream ss; + ss.str(std::string()); + ss.clear(); + ss << t; + ss >> s; + return s; +} + +template +static std::string toHumanReadableString(const T &n) { + return vtos(n); +} #endif + +template +static std::string toString(const T &t) { + return vtos(t); } +#if __cplusplus > 199711L || defined(_MSC_VER) +/* opts */ +void prepareOpts(int argc, char* argv[]); +#endif + /* * Very simple regex-like pattern. * It used for two purposes: validation and generation. - * + * * For example, pattern("[a-z]{1,5}").next(rnd) will return - * random string from lowercase latin letters with length - * from 1 to 5. It is easier to call rnd.next("[a-z]{1,5}") - * for the same effect. - * + * random string from lowercase latin letters with length + * from 1 to 5. It is easier to call rnd.next("[a-z]{1,5}") + * for the same effect. + * * Another samples: * "mike|john" will generate (match) "mike" or "john"; * "-?[1-9][0-9]{0,3}" will generate (match) non-zero integers from -9999 to 9999; * "id-([ac]|b{2})" will generate (match) "id-a", "id-bb", "id-c"; - * "[^0-9]*" will match sequences (empty or non-empty) without digits, you can't + * "[^0-9]*" will match sequences (empty or non-empty) without digits, you can't * use it for generations. * * You can't use pattern for generation if it contains meta-symbol '*'. Also it @@ -401,31 +693,38 @@ static FILE *__testlib_fopen(const char *name, const char *mode) * * For matching very simple greedy algorithm is used. For example, pattern * "[0-9]?1" will not match "1", because of greedy nature of matching. - * Alternations (meta-symbols "|") are processed with brute-force algorithm, so + * Alternations (meta-symbols "|") are processed with brute-force algorithm, so * do not use many alternations in one expression. * * If you want to use one expression many times it is better to compile it into - * a single pattern like "pattern p("[a-z]+")". Later you can use + * a single pattern like "pattern p("[a-z]+")". Later you can use * "p.matches(std::string s)" or "p.next(random_t& rd)" to check matching or generate * new string by pattern. - * + * * Simpler way to read token and check it for pattern matching is "inf.readToken("[a-z]+")". + * + * All spaces are ignored in regex, unless escaped with \. For example, ouf.readLine("NO SOLUTION") + * will expect "NOSOLUTION", the correct call should be ouf.readLine("NO\\ SOLUTION") or + * ouf.readLine(R"(NO\ SOLUTION)") if you prefer raw string literals from C++11. */ class random_t; -class pattern -{ +class pattern { public: /* Create pattern instance by string. */ pattern(std::string s); + /* Generate new string by pattern and given random_t. */ - std::string next(random_t& rnd) const; + std::string next(random_t &rnd) const; + /* Checks if given string match the pattern. */ - bool matches(const std::string& s) const; + bool matches(const std::string &s) const; + /* Returns source string of the pattern. */ std::string src() const; + private: - bool matches(const std::string& s, size_t pos) const; + bool matches(const std::string &s, size_t pos) const; std::string s; std::vector children; @@ -434,9 +733,9 @@ class pattern int to; }; -/* - * Use random_t instances to generate random values. It is preffered - * way to use randoms instead of rand() function or self-written +/* + * Use random_t instances to generate random values. It is preferred + * way to use randoms instead of rand() function or self-written * randoms. * * Testlib defines global variable "rnd" of random_t class. @@ -446,8 +745,7 @@ class pattern * Random generates uniformly distributed values if another strategy is * not specified explicitly. */ -class random_t -{ +class random_t { private: unsigned long long seed; static const unsigned long long multiplier; @@ -455,20 +753,20 @@ class random_t static const unsigned long long mask; static const int lim; - long long nextBits(int bits) - { - if (bits <= 48) - { + long long nextBits(int bits) { + if (bits <= 48) { seed = (seed * multiplier + addend) & mask; - return (long long)(seed >> (48 - bits)); - } - else - { + return (long long) (seed >> (48 - bits)); + } else { if (bits > 63) __testlib_fail("random_t::nextBits(int bits): n must be less than 64"); int lowerBitCount = (random_t::version == 0 ? 31 : 32); - return ((nextBits(31) << 32) ^ nextBits(lowerBitCount)); + + long long left = (nextBits(31) << 32); + long long right = nextBits(lowerBitCount); + + return left ^ right; } } @@ -477,41 +775,38 @@ class random_t /* New random_t with fixed seed. */ random_t() - : seed(3905348978240129619LL) - { + : seed(3905348978240129619LL) { } /* Sets seed by command line. */ - void setSeed(int argc, char* argv[]) - { + void setSeed(int argc, char *argv[]) { random_t p; seed = 3905348978240129619LL; - for (int i = 1; i < argc; i++) - { + for (int i = 1; i < argc; i++) { std::size_t le = std::strlen(argv[i]); for (std::size_t j = 0; j < le; j++) - seed = seed * multiplier + (unsigned int)(argv[i][j]) + addend; + seed = seed * multiplier + (unsigned int) (argv[i][j]) + addend; seed += multiplier / addend; } seed = seed & mask; } - /* Sets seed by given value. */ - void setSeed(long long _seed) - { - _seed = (_seed ^ multiplier) & mask; - seed = _seed; + /* Sets seed by given value. */ + void setSeed(long long _seed) { + seed = (unsigned long long) _seed; + seed = (seed ^ multiplier) & mask; } #ifndef __BORLANDC__ + /* Random string value by given pattern (see pattern documentation). */ - std::string next(const std::string& ptrn) - { + std::string next(const std::string &ptrn) { pattern p(ptrn); return p.next(*this); } + #else /* Random string value by given pattern (see pattern documentation). */ std::string next(std::string ptrn) @@ -522,16 +817,15 @@ class random_t #endif /* Random value in range [0, n-1]. */ - int next(int n) - { + int next(int n) { if (n <= 0) __testlib_fail("random_t::next(int n): n must be positive"); if ((n & -n) == n) // n is a power of 2 - return (int)((n * (long long)nextBits(31)) >> 31); + return (int) ((n * (long long) nextBits(31)) >> 31); const long long limit = INT_MAX / n * n; - + long long bits; do { bits = nextBits(31); @@ -541,21 +835,19 @@ class random_t } /* Random value in range [0, n-1]. */ - unsigned int next(unsigned int n) - { + unsigned int next(unsigned int n) { if (n >= INT_MAX) __testlib_fail("random_t::next(unsigned int n): n must be less INT_MAX"); - return (unsigned int)next(int(n)); + return (unsigned int) next(int(n)); } /* Random value in range [0, n-1]. */ - long long next(long long n) - { + long long next(long long n) { if (n <= 0) __testlib_fail("random_t::next(long long n): n must be positive"); const long long limit = __TESTLIB_LONGLONG_MAX / n * n; - + long long bits; do { bits = nextBits(63); @@ -565,116 +857,111 @@ class random_t } /* Random value in range [0, n-1]. */ - unsigned long long next(unsigned long long n) - { - if (n >= (unsigned long long)(__TESTLIB_LONGLONG_MAX)) + unsigned long long next(unsigned long long n) { + if (n >= (unsigned long long) (__TESTLIB_LONGLONG_MAX)) __testlib_fail("random_t::next(unsigned long long n): n must be less LONGLONG_MAX"); - return (unsigned long long)next((long long)(n)); + return (unsigned long long) next((long long) (n)); } /* Random value in range [0, n-1]. */ - long next(long n) - { - return (long)next((long long)(n)); + long next(long n) { + return (long) next((long long) (n)); } /* Random value in range [0, n-1]. */ - unsigned long next(unsigned long n) - { - if (n >= (unsigned long)(LONG_MAX)) + unsigned long next(unsigned long n) { + if (n >= (unsigned long) (LONG_MAX)) __testlib_fail("random_t::next(unsigned long n): n must be less LONG_MAX"); - return (unsigned long)next((unsigned long long)(n)); + return (unsigned long) next((unsigned long long) (n)); } /* Returns random value in range [from,to]. */ - int next(int from, int to) - { - return int(next((long long)to - from + 1) + from); + int next(int from, int to) { + return int(next((long long) to - from + 1) + from); } /* Returns random value in range [from,to]. */ - unsigned int next(unsigned int from, unsigned int to) - { - return (unsigned int)(next((long long)to - from + 1) + from); + unsigned int next(unsigned int from, unsigned int to) { + return (unsigned int) (next((long long) to - from + 1) + from); } /* Returns random value in range [from,to]. */ - long long next(long long from, long long to) - { + long long next(long long from, long long to) { return next(to - from + 1) + from; } /* Returns random value in range [from,to]. */ - unsigned long long next(unsigned long long from, unsigned long long to) - { + unsigned long long next(unsigned long long from, unsigned long long to) { if (from > to) __testlib_fail("random_t::next(unsigned long long from, unsigned long long to): from can't not exceed to"); return next(to - from + 1) + from; } /* Returns random value in range [from,to]. */ - long next(long from, long to) - { + long next(long from, long to) { return next(to - from + 1) + from; } /* Returns random value in range [from,to]. */ - unsigned long next(unsigned long from, unsigned long to) - { + unsigned long next(unsigned long from, unsigned long to) { if (from > to) __testlib_fail("random_t::next(unsigned long from, unsigned long to): from can't not exceed to"); return next(to - from + 1) + from; } /* Random double value in range [0, 1). */ - double next() - { - return (double)(((long long)(nextBits(26)) << 27) + nextBits(27)) / (double)(1LL << 53); + double next() { + long long left = ((long long) (nextBits(26)) << 27); + long long right = nextBits(27); + return __testlib_crop((double) (left + right) / (double) (1LL << 53), 0.0, 1.0); } /* Random double value in range [0, n). */ - double next(double n) - { - return n * next(); + double next(double n) { + if (n <= 0.0) + __testlib_fail("random_t::next(double): n should be positive"); + return __testlib_crop(n * next(), 0.0, n); } /* Random double value in range [from, to). */ - double next(double from, double to) - { + double next(double from, double to) { + if (from >= to) + __testlib_fail("random_t::next(double from, double to): from should be strictly less than to"); return next(to - from) + from; } /* Returns random element from container. */ - template - typename Container::value_type any(const Container& c) - { - size_t size = c.size(); + template + typename Container::value_type any(const Container &c) { + int size = int(c.size()); if (size <= 0) __testlib_fail("random_t::any(const Container& c): c.size() must be positive"); - return *(c.begin() + next(size)); + typename Container::const_iterator it = c.begin(); + std::advance(it, next(size)); + return *it; } /* Returns random element from iterator range. */ - template - typename Iter::value_type any(const Iter& begin, const Iter& end) - { - int size = int(end - begin); + template + typename Iter::value_type any(const Iter &begin, const Iter &end) { + int size = static_cast(std::distance(begin, end)); if (size <= 0) __testlib_fail("random_t::any(const Iter& begin, const Iter& end): range must have positive length"); - return *(begin + next(size)); + Iter it = begin; + std::advance(it, next(size)); + return *it; } /* Random string value by given pattern (see pattern documentation). */ #ifdef __GNUC__ __attribute__ ((format (printf, 2, 3))) #endif - std::string next(const char* format, ...) - { + std::string next(const char *format, ...) { FMT_TO_RESULT(format, format, ptrn); return next(ptrn); } - /* + /* * Weighted next. If type == 0 than it is usual "next()". * * If type = 1, than it returns "max(next(), next())" @@ -682,222 +969,317 @@ class random_t * * If type < 0, than "max" function replaces with "min". */ - int wnext(int n, int type) - { + int wnext(int n, int type) { if (n <= 0) __testlib_fail("random_t::wnext(int n, int type): n must be positive"); - - if (abs(type) < random_t::lim) - { + + if (abs(type) < random_t::lim) { int result = next(n); for (int i = 0; i < +type; i++) result = __testlib_max(result, next(n)); - + for (int i = 0; i < -type; i++) result = __testlib_min(result, next(n)); return result; - } - else - { + } else { double p; - + if (type > 0) p = std::pow(next() + 0.0, 1.0 / (type + 1)); else p = 1 - std::pow(next() + 0.0, 1.0 / (-type + 1)); - return int(n * p); + return __testlib_crop((int) (double(n) * p), 0, n); } } - + /* See wnext(int, int). It uses the same algorithms. */ - long long wnext(long long n, int type) - { + long long wnext(long long n, int type) { if (n <= 0) __testlib_fail("random_t::wnext(long long n, int type): n must be positive"); - - if (abs(type) < random_t::lim) - { + + if (abs(type) < random_t::lim) { long long result = next(n); for (int i = 0; i < +type; i++) result = __testlib_max(result, next(n)); - + for (int i = 0; i < -type; i++) result = __testlib_min(result, next(n)); return result; - } - else - { + } else { double p; - - if (type > 0) - p = std::pow(next() + 0.0, 1.0 / (type + 1)); - else - p = std::pow(next() + 0.0, - type + 1); - - return __testlib_min(__testlib_max((long long)(double(n) * p), 0LL), n - 1LL); - } - } - - /* See wnext(int, int). It uses the same algorithms. */ - double wnext(int type) - { - if (abs(type) < random_t::lim) - { - double result = next(); - - for (int i = 0; i < +type; i++) - result = __testlib_max(result, next()); - - for (int i = 0; i < -type; i++) - result = __testlib_min(result, next()); - return result; - } - else - { - double p; - if (type > 0) p = std::pow(next() + 0.0, 1.0 / (type + 1)); else - p = std::pow(next() + 0.0, - type + 1); + p = 1 - std::pow(next() + 0.0, 1.0 / (-type + 1)); - return p; + return __testlib_crop((long long) (double(n) * p), 0LL, n); } } - - /* See wnext(int, int). It uses the same algorithms. */ - double wnext(double n, int type) - { + + /* Returns value in [0, n). See wnext(int, int). It uses the same algorithms. */ + double wnext(double n, int type) { if (n <= 0) __testlib_fail("random_t::wnext(double n, int type): n must be positive"); - if (abs(type) < random_t::lim) - { + if (abs(type) < random_t::lim) { double result = next(); for (int i = 0; i < +type; i++) result = __testlib_max(result, next()); - + for (int i = 0; i < -type; i++) result = __testlib_min(result, next()); return n * result; - } - else - { + } else { double p; - + if (type > 0) p = std::pow(next() + 0.0, 1.0 / (type + 1)); else - p = std::pow(next() + 0.0, - type + 1); + p = 1 - std::pow(next() + 0.0, 1.0 / (-type + 1)); - return n * p; + return __testlib_crop(n * p, 0.0, n); } } + /* Returns value in [0, 1). See wnext(int, int). It uses the same algorithms. */ + double wnext(int type) { + return wnext(1.0, type); + } + /* See wnext(int, int). It uses the same algorithms. */ - unsigned int wnext(unsigned int n, int type) - { + unsigned int wnext(unsigned int n, int type) { if (n >= INT_MAX) __testlib_fail("random_t::wnext(unsigned int n, int type): n must be less INT_MAX"); - return (unsigned int)wnext(int(n), type); + return (unsigned int) wnext(int(n), type); } - + /* See wnext(int, int). It uses the same algorithms. */ - unsigned long long wnext(unsigned long long n, int type) - { - if (n >= (unsigned long long)(__TESTLIB_LONGLONG_MAX)) + unsigned long long wnext(unsigned long long n, int type) { + if (n >= (unsigned long long) (__TESTLIB_LONGLONG_MAX)) __testlib_fail("random_t::wnext(unsigned long long n, int type): n must be less LONGLONG_MAX"); - return (unsigned long long)wnext((long long)(n), type); + return (unsigned long long) wnext((long long) (n), type); } /* See wnext(int, int). It uses the same algorithms. */ - long wnext(long n, int type) - { - return (long)wnext((long long)(n), type); + long wnext(long n, int type) { + return (long) wnext((long long) (n), type); } - + /* See wnext(int, int). It uses the same algorithms. */ - unsigned long wnext(unsigned long n, int type) - { - if (n >= (unsigned long)(LONG_MAX)) + unsigned long wnext(unsigned long n, int type) { + if (n >= (unsigned long) (LONG_MAX)) __testlib_fail("random_t::wnext(unsigned long n, int type): n must be less LONG_MAX"); - return (unsigned long)wnext((unsigned long long)(n), type); + return (unsigned long) wnext((unsigned long long) (n), type); } /* Returns weighted random value in range [from, to]. */ - int wnext(int from, int to, int type) - { + int wnext(int from, int to, int type) { + if (from > to) + __testlib_fail("random_t::wnext(int from, int to, int type): from can't not exceed to"); return wnext(to - from + 1, type) + from; } - + /* Returns weighted random value in range [from, to]. */ - int wnext(unsigned int from, unsigned int to, int type) - { + int wnext(unsigned int from, unsigned int to, int type) { + if (from > to) + __testlib_fail("random_t::wnext(unsigned int from, unsigned int to, int type): from can't not exceed to"); return int(wnext(to - from + 1, type) + from); } - + /* Returns weighted random value in range [from, to]. */ - long long wnext(long long from, long long to, int type) - { + long long wnext(long long from, long long to, int type) { + if (from > to) + __testlib_fail("random_t::wnext(long long from, long long to, int type): from can't not exceed to"); return wnext(to - from + 1, type) + from; } - + /* Returns weighted random value in range [from, to]. */ - unsigned long long wnext(unsigned long long from, unsigned long long to, int type) - { + unsigned long long wnext(unsigned long long from, unsigned long long to, int type) { if (from > to) - __testlib_fail("random_t::wnext(unsigned long long from, unsigned long long to, int type): from can't not exceed to"); + __testlib_fail( + "random_t::wnext(unsigned long long from, unsigned long long to, int type): from can't not exceed to"); return wnext(to - from + 1, type) + from; } - + /* Returns weighted random value in range [from, to]. */ - long wnext(long from, long to, int type) - { + long wnext(long from, long to, int type) { + if (from > to) + __testlib_fail("random_t::wnext(long from, long to, int type): from can't not exceed to"); return wnext(to - from + 1, type) + from; } - + /* Returns weighted random value in range [from, to]. */ - unsigned long wnext(unsigned long from, unsigned long to, int type) - { + unsigned long wnext(unsigned long from, unsigned long to, int type) { if (from > to) __testlib_fail("random_t::wnext(unsigned long from, unsigned long to, int type): from can't not exceed to"); return wnext(to - from + 1, type) + from; } - + /* Returns weighted random double value in range [from, to). */ - double wnext(double from, double to, int type) - { + double wnext(double from, double to, int type) { + if (from >= to) + __testlib_fail("random_t::wnext(double from, double to, int type): from should be strictly less than to"); return wnext(to - from, type) + from; } /* Returns weighted random element from container. */ - template - typename Container::value_type wany(const Container& c, int type) - { - size_t size = c.size(); + template + typename Container::value_type wany(const Container &c, int type) { + int size = int(c.size()); if (size <= 0) __testlib_fail("random_t::wany(const Container& c, int type): c.size() must be positive"); - return *(c.begin() + wnext(size, type)); + typename Container::const_iterator it = c.begin(); + std::advance(it, wnext(size, type)); + return *it; } /* Returns weighted random element from iterator range. */ - template - typename Iter::value_type wany(const Iter& begin, const Iter& end, int type) - { - int size = int(end - begin); + template + typename Iter::value_type wany(const Iter &begin, const Iter &end, int type) { + int size = static_cast(std::distance(begin, end)); if (size <= 0) - __testlib_fail("random_t::any(const Iter& begin, const Iter& end, int type): range must have positive length"); - return *(begin + wnext(size, type)); + __testlib_fail( + "random_t::any(const Iter& begin, const Iter& end, int type): range must have positive length"); + Iter it = begin; + std::advance(it, wnext(size, type)); + return *it; + } + + /* Returns random permutation of the given size (values are between `first` and `first`+size-1)*/ + template + std::vector perm(T size, E first) { + if (size < 0) + __testlib_fail("random_t::perm(T size, E first = 0): size must non-negative"); + else if (size == 0) + return std::vector(); + std::vector p(size); + E current = first; + for (T i = 0; i < size; i++) + p[i] = current++; + if (size > 1) + for (T i = 1; i < size; i++) + std::swap(p[i], p[next(i + 1)]); + return p; + } + + /* Returns random permutation of the given size (values are between 0 and size-1)*/ + template + std::vector perm(T size) { + return perm(size, T(0)); + } + + /* Returns `size` unordered (unsorted) distinct numbers between `from` and `to`. */ + template + std::vector distinct(int size, T from, T to) { + std::vector result; + if (size == 0) + return result; + + if (from > to) + __testlib_fail("random_t::distinct expected from <= to"); + + if (size < 0) + __testlib_fail("random_t::distinct expected size >= 0"); + + uint64_t n = to - from + 1; + if (uint64_t(size) > n) + __testlib_fail("random_t::distinct expected size <= to - from + 1"); + + double expected = 0.0; + for (int i = 1; i <= size; i++) + expected += double(n) / double(n - i + 1); + + if (expected < double(n)) { + std::set vals; + while (int(vals.size()) < size) { + T x = T(next(from, to)); + if (vals.insert(x).second) + result.push_back(x); + } + } else { + if (n > 1000000000) + __testlib_fail("random_t::distinct here expected to - from + 1 <= 1000000000"); + std::vector p(perm(int(n), from)); + result.insert(result.end(), p.begin(), p.begin() + size); + } + + return result; + } + + /* Returns `size` unordered (unsorted) distinct numbers between `0` and `upper`-1. */ + template + std::vector distinct(int size, T upper) { + if (size < 0) + __testlib_fail("random_t::distinct expected size >= 0"); + if (size == 0) + return std::vector(); + + if (upper <= 0) + __testlib_fail("random_t::distinct expected upper > 0"); + if (size > upper) + __testlib_fail("random_t::distinct expected size <= upper"); + + return distinct(size, T(0), upper - 1); + } + + /* Returns random (unsorted) partition which is a representation of sum as a sum of integers not less than min_part. */ + template + std::vector partition(int size, T sum, T min_part) { + if (size < 0) + __testlib_fail("random_t::partition: size < 0"); + if (size == 0 && sum != 0) + __testlib_fail("random_t::partition: size == 0 && sum != 0"); + if (min_part * size > sum) + __testlib_fail("random_t::partition: min_part * size > sum"); + if (size == 0 && sum == 0) + return std::vector(); + + T sum_ = sum; + sum -= min_part * size; + + std::vector septums(size); + std::vector d = distinct(size - 1, T(1), T(sum + size - 1)); + for (int i = 0; i + 1 < size; i++) + septums[i + 1] = d[i]; + sort(septums.begin(), septums.end()); + + std::vector result(size); + for (int i = 0; i + 1 < size; i++) + result[i] = septums[i + 1] - septums[i] - 1; + result[size - 1] = sum + size - 1 - septums.back(); + + for (std::size_t i = 0; i < result.size(); i++) + result[i] += min_part; + + T result_sum = 0; + for (std::size_t i = 0; i < result.size(); i++) + result_sum += result[i]; + if (result_sum != sum_) + __testlib_fail("random_t::partition: partition sum is expected to be the given sum"); + + if (*std::min_element(result.begin(), result.end()) < min_part) + __testlib_fail("random_t::partition: partition min is expected to be no less than the given min_part"); + + if (int(result.size()) != size || result.size() != (size_t) size) + __testlib_fail("random_t::partition: partition size is expected to be equal to the given size"); + + return result; + } + + /* Returns random (unsorted) partition which is a representation of sum as a sum of positive integers. */ + template + std::vector partition(int size, T sum) { + return partition(size, sum, T(1)); } }; @@ -908,21 +1290,18 @@ const unsigned long long random_t::mask = (1LL << 48) - 1; int random_t::version = -1; /* Pattern implementation */ -bool pattern::matches(const std::string& s) const -{ +bool pattern::matches(const std::string &s) const { return matches(s, 0); } -static bool __pattern_isSlash(const std::string& s, size_t pos) -{ +static bool __pattern_isSlash(const std::string &s, size_t pos) { return s[pos] == '\\'; } #ifdef __GNUC__ __attribute__((pure)) #endif -static bool __pattern_isCommandChar(const std::string& s, size_t pos, char value) -{ +static bool __pattern_isCommandChar(const std::string &s, size_t pos, char value) { if (pos >= s.length()) return false; @@ -935,8 +1314,7 @@ static bool __pattern_isCommandChar(const std::string& s, size_t pos, char value return slashes % 2 == 0 && s[pos] == value; } -static char __pattern_getChar(const std::string& s, size_t& pos) -{ +static char __pattern_getChar(const std::string &s, size_t &pos) { if (__pattern_isSlash(s, pos)) pos += 2; else @@ -948,12 +1326,10 @@ static char __pattern_getChar(const std::string& s, size_t& pos) #ifdef __GNUC__ __attribute__((pure)) #endif -static int __pattern_greedyMatch(const std::string& s, size_t pos, const std::vector chars) -{ +static int __pattern_greedyMatch(const std::string &s, size_t pos, const std::vector chars) { int result = 0; - while (pos < s.length()) - { + while (pos < s.length()) { char c = s[pos++]; if (!std::binary_search(chars.begin(), chars.end(), c)) break; @@ -964,17 +1340,14 @@ static int __pattern_greedyMatch(const std::string& s, size_t pos, const std::ve return result; } -std::string pattern::src() const -{ +std::string pattern::src() const { return s; } -bool pattern::matches(const std::string& s, size_t pos) const -{ +bool pattern::matches(const std::string &s, size_t pos) const { std::string result; - if (to > 0) - { + if (to > 0) { int size = __pattern_greedyMatch(s, pos, chars); if (size < from) return false; @@ -983,34 +1356,29 @@ bool pattern::matches(const std::string& s, size_t pos) const pos += size; } - if (children.size() > 0) - { + if (children.size() > 0) { for (size_t child = 0; child < children.size(); child++) if (children[child].matches(s, pos)) return true; return false; - } - else + } else return pos == s.length(); } -std::string pattern::next(random_t& rnd) const -{ +std::string pattern::next(random_t &rnd) const { std::string result; result.reserve(20); if (to == INT_MAX) __testlib_fail("pattern::next(random_t& rnd): can't process character '*' for generation"); - if (to > 0) - { + if (to > 0) { int count = rnd.next(to - from + 1) + from; for (int i = 0; i < count; i++) result += chars[rnd.next(int(chars.size()))]; } - if (children.size() > 0) - { + if (children.size() > 0) { int child = rnd.next(int(children.size())); result += children[child].next(rnd); } @@ -1018,23 +1386,19 @@ std::string pattern::next(random_t& rnd) const return result; } -static void __pattern_scanCounts(const std::string& s, size_t& pos, int& from, int& to) -{ - if (pos >= s.length()) - { +static void __pattern_scanCounts(const std::string &s, size_t &pos, int &from, int &to) { + if (pos >= s.length()) { from = to = 1; return; } - - if (__pattern_isCommandChar(s, pos, '{')) - { + + if (__pattern_isCommandChar(s, pos, '{')) { std::vector parts; std::string part; pos++; - while (pos < s.length() && !__pattern_isCommandChar(s, pos, '}')) - { + while (pos < s.length() && !__pattern_isCommandChar(s, pos, '}')) { if (__pattern_isCommandChar(s, pos, ',')) parts.push_back(part), part = "", pos++; else @@ -1054,8 +1418,7 @@ static void __pattern_scanCounts(const std::string& s, size_t& pos, int& from, i std::vector numbers; - for (size_t i = 0; i < parts.size(); i++) - { + for (size_t i = 0; i < parts.size(); i++) { if (parts[i].length() == 0) __testlib_fail("pattern: Illegal pattern (or part) \"" + s + "\""); int number; @@ -1071,53 +1434,45 @@ static void __pattern_scanCounts(const std::string& s, size_t& pos, int& from, i if (from > to) __testlib_fail("pattern: Illegal pattern (or part) \"" + s + "\""); - } - else - { - if (__pattern_isCommandChar(s, pos, '?')) - { + } else { + if (__pattern_isCommandChar(s, pos, '?')) { from = 0, to = 1, pos++; return; } - if (__pattern_isCommandChar(s, pos, '*')) - { + if (__pattern_isCommandChar(s, pos, '*')) { from = 0, to = INT_MAX, pos++; return; } - if (__pattern_isCommandChar(s, pos, '+')) - { + if (__pattern_isCommandChar(s, pos, '+')) { from = 1, to = INT_MAX, pos++; return; } - + from = to = 1; } } -static std::vector __pattern_scanCharSet(const std::string& s, size_t& pos) -{ +static std::vector __pattern_scanCharSet(const std::string &s, size_t &pos) { if (pos >= s.length()) __testlib_fail("pattern: Illegal pattern (or part) \"" + s + "\""); std::vector result; - if (__pattern_isCommandChar(s, pos, '[')) - { + if (__pattern_isCommandChar(s, pos, '[')) { pos++; bool negative = __pattern_isCommandChar(s, pos, '^'); + if (negative) + pos++; char prev = 0; - while (pos < s.length() && !__pattern_isCommandChar(s, pos, ']')) - { - if (__pattern_isCommandChar(s, pos, '-') && prev != 0) - { + while (pos < s.length() && !__pattern_isCommandChar(s, pos, ']')) { + if (__pattern_isCommandChar(s, pos, '-') && prev != 0) { pos++; - if (pos + 1 == s.length() || __pattern_isCommandChar(s, pos, ']')) - { + if (pos + 1 == s.length() || __pattern_isCommandChar(s, pos, ']')) { result.push_back(prev); prev = '-'; continue; @@ -1132,9 +1487,7 @@ static std::vector __pattern_scanCharSet(const std::string& s, size_t& pos result.push_back(next); prev = 0; - } - else - { + } else { if (prev != 0) result.push_back(prev); prev = __pattern_getChar(s, pos); @@ -1149,12 +1502,10 @@ static std::vector __pattern_scanCharSet(const std::string& s, size_t& pos pos++; - if (negative) - { + if (negative) { std::sort(result.begin(), result.end()); std::vector actuals; - for (int code = 0; code < 255; code++) - { + for (int code = 0; code < 255; code++) { char c = char(code); if (!std::binary_search(result.begin(), result.end(), c)) actuals.push_back(c); @@ -1163,15 +1514,13 @@ static std::vector __pattern_scanCharSet(const std::string& s, size_t& pos } std::sort(result.begin(), result.end()); - } - else + } else result.push_back(__pattern_getChar(s, pos)); return result; } -pattern::pattern(std::string s): s(s), from(0), to(0) -{ +pattern::pattern(std::string s) : s(s), from(0), to(0) { std::string t; for (size_t i = 0; i < s.length(); i++) if (!__pattern_isCommandChar(s, i, ' ')) @@ -1182,22 +1531,19 @@ pattern::pattern(std::string s): s(s), from(0), to(0) int firstClose = -1; std::vector seps; - for (size_t i = 0; i < s.length(); i++) - { - if (__pattern_isCommandChar(s, i, '(')) - { + for (size_t i = 0; i < s.length(); i++) { + if (__pattern_isCommandChar(s, i, '(')) { opened++; continue; } - if (__pattern_isCommandChar(s, i, ')')) - { + if (__pattern_isCommandChar(s, i, ')')) { opened--; if (opened == 0 && firstClose == -1) firstClose = int(i); continue; } - + if (opened < 0) __testlib_fail("pattern: Illegal pattern (or part) \"" + s + "\""); @@ -1208,26 +1554,19 @@ pattern::pattern(std::string s): s(s), from(0), to(0) if (opened != 0) __testlib_fail("pattern: Illegal pattern (or part) \"" + s + "\""); - if (seps.size() == 0 && firstClose + 1 == (int)s.length() - && __pattern_isCommandChar(s, 0, '(') && __pattern_isCommandChar(s, s.length() - 1, ')')) - { + if (seps.size() == 0 && firstClose + 1 == (int) s.length() + && __pattern_isCommandChar(s, 0, '(') && __pattern_isCommandChar(s, s.length() - 1, ')')) { children.push_back(pattern(s.substr(1, s.length() - 2))); - } - else - { - if (seps.size() > 0) - { - seps.push_back(s.length()); + } else { + if (seps.size() > 0) { + seps.push_back(int(s.length())); int last = 0; - for (size_t i = 0; i < seps.size(); i++) - { + for (size_t i = 0; i < seps.size(); i++) { children.push_back(pattern(s.substr(last, seps[i] - last))); last = seps[i] + 1; } - } - else - { + } else { size_t pos = 0; chars = __pattern_scanCharSet(s, pos); __pattern_scanCounts(s, pos, from, to); @@ -1236,34 +1575,49 @@ pattern::pattern(std::string s): s(s), from(0), to(0) } } } + /* End of pattern implementation */ -template -inline bool isEof(C c) -{ - return (c == EOF || c == EOFC); +template +inline bool isEof(C c) { + return c == EOFC; } -template -inline bool isEoln(C c) -{ +template +inline bool isEoln(C c) { return (c == LF || c == CR); } template -inline bool isBlanks(C c) -{ +inline bool isBlanks(C c) { return (c == LF || c == CR || c == SPACE || c == TAB); } -enum TMode -{ +inline std::string trim(const std::string &s) { + if (s.empty()) + return s; + + int left = 0; + while (left < int(s.length()) && isBlanks(s[left])) + left++; + if (left >= int(s.length())) + return ""; + + int right = int(s.length()) - 1; + while (right >= 0 && isBlanks(s[right])) + right--; + if (right < 0) + return ""; + + return s.substr(left, right - left + 1); +} + +enum TMode { _input, _output, _answer }; /* Outcomes 6-15 are reserved for future use. */ -enum TResult -{ +enum TResult { _ok = 0, _wa = 1, _pe = 2, @@ -1274,201 +1628,271 @@ enum TResult _partially = 16 }; -enum TTestlibMode -{ - _unknown, _checker, _validator, _generator, _interactor +enum TTestlibMode { + _unknown, _checker, _validator, _generator, _interactor, _scorer }; #define _pc(exitCode) (TResult(_partially + (exitCode))) /* Outcomes 6-15 are reserved for future use. */ const std::string outcomes[] = { - "accepted", - "wrong-answer", - "presentation-error", - "fail", - "fail", + "accepted", + "wrong-answer", + "presentation-error", + "fail", + "fail", #ifndef PCMS2 - "points", + "points", #else - "relative-scoring", -#endif - "reserved", - "reserved", - "unexpected-eof", - "reserved", - "reserved", - "reserved", - "reserved", - "reserved", - "reserved", - "reserved", - "partially-correct" + "relative-scoring", +#endif + "reserved", + "reserved", + "unexpected-eof", + "reserved", + "reserved", + "reserved", + "reserved", + "reserved", + "reserved", + "reserved", + "partially-correct" }; -class InputStreamReader -{ +class InputStreamReader { public: - virtual int curChar() = 0; - virtual int nextChar() = 0; + virtual void setTestCase(int testCase) = 0; + + virtual std::vector getReadChars() = 0; + + virtual int curChar() = 0; + + virtual int nextChar() = 0; + virtual void skipChar() = 0; + virtual void unreadChar(int c) = 0; + virtual std::string getName() = 0; + virtual bool eof() = 0; + + virtual void close() = 0; + + virtual int getLine() = 0; + virtual ~InputStreamReader() = 0; }; -InputStreamReader::~InputStreamReader() -{ +InputStreamReader::~InputStreamReader() { // No operations. } -class StringInputStreamReader: public InputStreamReader -{ +class StringInputStreamReader : public InputStreamReader { private: std::string s; size_t pos; public: - StringInputStreamReader(const std::string& content): s(content), pos(0) - { + StringInputStreamReader(const std::string &content) : s(content), pos(0) { // No operations. } - int curChar() - { + void setTestCase(int) { + __testlib_fail("setTestCase not implemented in StringInputStreamReader"); + } + + std::vector getReadChars() { + __testlib_fail("getReadChars not implemented in StringInputStreamReader"); + } + + int curChar() { if (pos >= s.length()) - return EOF; + return EOFC; else - { return s[pos]; - } } - int nextChar() - { - if (pos >= s.length()) - return EOF; - else + int nextChar() { + if (pos >= s.length()) { + pos++; + return EOFC; + } else return s[pos++]; } - void skipChar() - { + void skipChar() { pos++; } - void unreadChar(int c) - { + void unreadChar(int c) { if (pos == 0) - __testlib_fail("FileFileInputStreamReader::unreadChar(int): pos == 0."); + __testlib_fail("StringInputStreamReader::unreadChar(int): pos == 0."); pos--; if (pos < s.length()) s[pos] = char(c); } - std::string getName() - { + std::string getName() { return __testlib_part(s); } - bool eof() - { + int getLine() { + return -1; + } + + bool eof() { return pos >= s.length(); } + + void close() { + // No operations. + } }; -class FileInputStreamReader: public InputStreamReader -{ +class FileInputStreamReader : public InputStreamReader { private: - std::FILE* file; + std::FILE *file; std::string name; + int line; + std::vector undoChars; + std::vector readChars; + std::vector undoReadChars; + + inline int postprocessGetc(int getcResult) { + if (getcResult != EOF) + return getcResult; + else + return EOFC; + } + + int getc(FILE *file) { + int c; + int rc; + + if (undoChars.empty()) { + c = rc = ::getc(file); + } else { + c = undoChars.back(); + undoChars.pop_back(); + rc = undoReadChars.back(); + undoReadChars.pop_back(); + } + + if (c == LF) + line++; + + readChars.push_back(rc); + return c; + } + + int ungetc(int c/*, FILE* file*/) { + if (!readChars.empty()) { + undoReadChars.push_back(readChars.back()); + readChars.pop_back(); + } + if (c == LF) + line--; + undoChars.push_back(c); + return c; + } public: - FileInputStreamReader(std::FILE* file, const std::string& name): file(file), name(name) - { + FileInputStreamReader(std::FILE *file, const std::string &name) : file(file), name(name), line(1) { // No operations. } - int curChar() - { + void setTestCase(int testCase) { + if (testCase < 0 || testCase > __TESTLIB_MAX_TEST_CASE) + __testlib_fail(format("testCase expected fit in [1,%d], but %d doesn't", __TESTLIB_MAX_TEST_CASE, testCase)); + readChars.push_back(testCase + 256); + } + + std::vector getReadChars() { + return readChars; + } + + int curChar() { if (feof(file)) - return EOF; - else - { + return EOFC; + else { int c = getc(file); - ungetc(c, file); - return c; + ungetc(c/*, file*/); + return postprocessGetc(c); } } - int nextChar() - { + int nextChar() { if (feof(file)) - return EOF; + return EOFC; else - return getc(file); + return postprocessGetc(getc(file)); } - void skipChar() - { + void skipChar() { getc(file); } - void unreadChar(int c) - { - ungetc(c, file); + void unreadChar(int c) { + ungetc(c/*, file*/); } - std::string getName() - { + std::string getName() { return name; } - bool eof() - { - if (feof(file)) + int getLine() { + return line; + } + + bool eof() { + if (NULL == file || feof(file)) return true; - else - { - int c = getc(file); - bool result = (c == EOF); - ungetc(c, file); - return result; + else { + int c = nextChar(); + if (c == EOFC || (c == EOF && feof(file))) + return true; + unreadChar(c); + return false; + } + } + + void close() { + if (NULL != file) { + fclose(file); + file = NULL; } } }; -class BufferedFileInputStreamReader: public InputStreamReader -{ +class BufferedFileInputStreamReader : public InputStreamReader { private: static const size_t BUFFER_SIZE; - static const size_t MAX_UNREAD_COUNT; - - std::FILE* file; - char* buffer; - bool* isEof; - int bufferPos; - size_t bufferSize; + static const size_t MAX_UNREAD_COUNT; + std::FILE *file; std::string name; + int line; - bool refill() - { + char *buffer; + bool *isEof; + int bufferPos; + size_t bufferSize; + + bool refill() { if (NULL == file) __testlib_fail("BufferedFileInputStreamReader: file == NULL (" + getName() + ")"); - if (bufferPos >= int(bufferSize)) - { + if (bufferPos >= int(bufferSize)) { size_t readSize = fread( - buffer + MAX_UNREAD_COUNT, - 1, - BUFFER_SIZE - MAX_UNREAD_COUNT, - file + buffer + MAX_UNREAD_COUNT, + 1, + BUFFER_SIZE - MAX_UNREAD_COUNT, + file ); if (readSize < BUFFER_SIZE - MAX_UNREAD_COUNT - && ferror(file)) + && ferror(file)) __testlib_fail("BufferedFileInputStreamReader: unable to read (" + getName() + ")"); bufferSize = MAX_UNREAD_COUNT + readSize; @@ -1476,95 +1900,112 @@ class BufferedFileInputStreamReader: public InputStreamReader std::memset(isEof + MAX_UNREAD_COUNT, 0, sizeof(isEof[0]) * readSize); return readSize > 0; - } - else + } else return true; } + char increment() { + char c; + if ((c = buffer[bufferPos++]) == LF) + line++; + return c; + } + public: - BufferedFileInputStreamReader(std::FILE* file, const std::string& name): file(file), name(name) - { + BufferedFileInputStreamReader(std::FILE *file, const std::string &name) : file(file), name(name), line(1) { buffer = new char[BUFFER_SIZE]; isEof = new bool[BUFFER_SIZE]; bufferSize = MAX_UNREAD_COUNT; bufferPos = int(MAX_UNREAD_COUNT); } - ~BufferedFileInputStreamReader() - { - if (NULL != buffer) - { + ~BufferedFileInputStreamReader() { + if (NULL != buffer) { delete[] buffer; buffer = NULL; } - if (NULL != isEof) - { + if (NULL != isEof) { delete[] isEof; isEof = NULL; } } - int curChar() - { + void setTestCase(int) { + __testlib_fail("setTestCase not implemented in BufferedFileInputStreamReader"); + } + + std::vector getReadChars() { + __testlib_fail("getReadChars not implemented in BufferedFileInputStreamReader"); + } + + int curChar() { if (!refill()) - return EOF; + return EOFC; - return isEof[bufferPos] ? EOF : buffer[bufferPos]; + return isEof[bufferPos] ? EOFC : buffer[bufferPos]; } - int nextChar() - { + int nextChar() { if (!refill()) - return EOF; + return EOFC; - return isEof[bufferPos] ? EOF : buffer[bufferPos++]; + return isEof[bufferPos] ? EOFC : increment(); } - void skipChar() - { - bufferPos++; + void skipChar() { + increment(); } - void unreadChar(int c) - { + void unreadChar(int c) { bufferPos--; if (bufferPos < 0) __testlib_fail("BufferedFileInputStreamReader::unreadChar(int): bufferPos < 0"); - isEof[bufferPos] = (c == EOF); - buffer[bufferPos] = char(c == EOF ? EOFC : c); + isEof[bufferPos] = (c == EOFC); + buffer[bufferPos] = char(c); + if (c == LF) + line--; } - std::string getName() - { + std::string getName() { return name; } - - bool eof() - { - return !refill() || EOF == curChar(); + + int getLine() { + return line; + } + + bool eof() { + return !refill() || EOFC == curChar(); + } + + void close() { + if (NULL != file) { + fclose(file); + file = NULL; + } } }; -const size_t BufferedFileInputStreamReader::BUFFER_SIZE = 1000000; -const size_t BufferedFileInputStreamReader::MAX_UNREAD_COUNT = 100; +const size_t BufferedFileInputStreamReader::BUFFER_SIZE = 2000000; +const size_t BufferedFileInputStreamReader::MAX_UNREAD_COUNT = BufferedFileInputStreamReader::BUFFER_SIZE / 2; /* * Streams to be used for reading data in checkers or validators. * Each read*() method moves pointer to the next character after the * read value. */ -struct InStream -{ +struct InStream { /* Do not use them. */ InStream(); + ~InStream(); /* Wrap std::string with InStream. */ - InStream(const InStream& baseStream, std::string content); + InStream(const InStream &baseStream, std::string content); - InputStreamReader* reader; + InputStreamReader *reader; + int lastLine; - std::FILE* file; std::string name; TMode mode; bool opened; @@ -1574,237 +2015,724 @@ struct InStream int wordReserveSize; std::string _tmpReadToken; + int readManyIteration; + size_t maxFileSize; + size_t maxTokenLength; + size_t maxMessageLength; + void init(std::string fileName, TMode mode); - void init(std::FILE* f, TMode mode); - /* Moves stream pointer to the first non-white-space character or EOF. */ + void init(std::FILE *f, TMode mode); + + void setTestCase(int testCase); + std::vector getReadChars(); + + /* Moves stream pointer to the first non-white-space character or EOF. */ void skipBlanks(); - + /* Returns current character in the stream. Doesn't remove it from stream. */ char curChar(); + /* Moves stream pointer one character forward. */ void skipChar(); + /* Returns current character and moves pointer one character forward. */ char nextChar(); - + /* Returns current character and moves pointer one character forward. */ char readChar(); + /* As "readChar()" but ensures that the result is equal to given parameter. */ char readChar(char c); + /* As "readChar()" but ensures that the result is equal to the space (code=32). */ char readSpace(); + /* Puts back the character into the stream. */ void unreadChar(char c); /* Reopens stream, you should not use it. */ - void reset(); + void reset(std::FILE *file = NULL); + /* Checks that current position is EOF. If not it doesn't move stream pointer. */ bool eof(); + /* Moves pointer to the first non-white-space character and calls "eof()". */ bool seekEof(); - /* - * Checks that current position contains EOLN. - * If not it doesn't move stream pointer. + /* + * Checks that current position contains EOLN. + * If not it doesn't move stream pointer. * In strict mode expects "#13#10" for windows or "#10" for other platforms. */ bool eoln(); + /* Moves pointer to the first non-space and non-tab character and calls "eoln()". */ bool seekEoln(); /* Moves stream pointer to the first character of the next line (if exists). */ void nextLine(); - /* - * Reads new token. Ignores white-spaces into the non-strict mode - * (strict mode is used in validators usually). + /* + * Reads new token. Ignores white-spaces into the non-strict mode + * (strict mode is used in validators usually). */ std::string readWord(); - /* The same as "readWord()", it is preffered to use "readToken()". */ + + /* The same as "readWord()", it is preferred to use "readToken()". */ std::string readToken(); + /* The same as "readWord()", but ensures that token matches to given pattern. */ - std::string readWord(const std::string& ptrn, const std::string& variableName = ""); - std::string readWord(const pattern& p, const std::string& variableName = ""); - /* The same as "readToken()", but ensures that token matches to given pattern. */ - std::string readToken(const std::string& ptrn, const std::string& variableName = ""); - std::string readToken(const pattern& p, const std::string& variableName = ""); + std::string readWord(const std::string &ptrn, const std::string &variableName = ""); - void readWordTo(std::string& result); - void readWordTo(std::string& result, const pattern& p, const std::string& variableName = ""); - void readWordTo(std::string& result, const std::string& ptrn, const std::string& variableName = ""); + std::string readWord(const pattern &p, const std::string &variableName = ""); - void readTokenTo(std::string& result); - void readTokenTo(std::string& result, const pattern& p, const std::string& variableName = ""); - void readTokenTo(std::string& result, const std::string& ptrn, const std::string& variableName = ""); + std::vector + readWords(int size, const std::string &ptrn, const std::string &variablesName = "", int indexBase = 1); - /* - * Reads new long long value. Ignores white-spaces into the non-strict mode - * (strict mode is used in validators usually). - */ - long long readLong(); - /* - * Reads new int. Ignores white-spaces into the non-strict mode - * (strict mode is used in validators usually). - */ - int readInteger(); - /* - * Reads new int. Ignores white-spaces into the non-strict mode - * (strict mode is used in validators usually). - */ - int readInt(); + std::vector + readWords(int size, const pattern &p, const std::string &variablesName = "", int indexBase = 1); + + std::vector readWords(int size, int indexBase = 1); + + /* The same as "readToken()", but ensures that token matches to given pattern. */ + std::string readToken(const std::string &ptrn, const std::string &variableName = ""); + + std::string readToken(const pattern &p, const std::string &variableName = ""); + + std::vector + readTokens(int size, const std::string &ptrn, const std::string &variablesName = "", int indexBase = 1); + + std::vector + readTokens(int size, const pattern &p, const std::string &variablesName = "", int indexBase = 1); + + std::vector readTokens(int size, int indexBase = 1); + + void readWordTo(std::string &result); + + void readWordTo(std::string &result, const pattern &p, const std::string &variableName = ""); + + void readWordTo(std::string &result, const std::string &ptrn, const std::string &variableName = ""); + + void readTokenTo(std::string &result); + + void readTokenTo(std::string &result, const pattern &p, const std::string &variableName = ""); + + void readTokenTo(std::string &result, const std::string &ptrn, const std::string &variableName = ""); + + /* + * Reads new long long value. Ignores white-spaces into the non-strict mode + * (strict mode is used in validators usually). + */ + long long readLong(); + + unsigned long long readUnsignedLong(); + + /* + * Reads new int. Ignores white-spaces into the non-strict mode + * (strict mode is used in validators usually). + */ + int readInteger(); + + /* + * Reads new int. Ignores white-spaces into the non-strict mode + * (strict mode is used in validators usually). + */ + int readInt(); + + /* As "readLong()" but ensures that value in the range [minv,maxv]. */ + long long readLong(long long minv, long long maxv, const std::string &variableName = ""); + + /* Reads space-separated sequence of long longs. */ + std::vector + readLongs(int size, long long minv, long long maxv, const std::string &variablesName = "", int indexBase = 1); + + /* Reads space-separated sequence of long longs. */ + std::vector readLongs(int size, int indexBase = 1); + + unsigned long long + readUnsignedLong(unsigned long long minv, unsigned long long maxv, const std::string &variableName = ""); + + std::vector + readUnsignedLongs(int size, unsigned long long minv, unsigned long long maxv, const std::string &variablesName = "", + int indexBase = 1); + + std::vector readUnsignedLongs(int size, int indexBase = 1); + + unsigned long long readLong(unsigned long long minv, unsigned long long maxv, const std::string &variableName = ""); + + std::vector + readLongs(int size, unsigned long long minv, unsigned long long maxv, const std::string &variablesName = "", + int indexBase = 1); + + /* As "readInteger()" but ensures that value in the range [minv,maxv]. */ + int readInteger(int minv, int maxv, const std::string &variableName = ""); - /* As "readLong()" but ensures that value in the range [minv,maxv]. */ - long long readLong(long long minv, long long maxv, const std::string& variableName = ""); - /* As "readInteger()" but ensures that value in the range [minv,maxv]. */ - int readInteger(int minv, int maxv, const std::string& variableName = ""); /* As "readInt()" but ensures that value in the range [minv,maxv]. */ - - int readInt(int minv, int maxv, const std::string& variableName = ""); - /* - * Reads new double. Ignores white-spaces into the non-strict mode - * (strict mode is used in validators usually). + int readInt(int minv, int maxv, const std::string &variableName = ""); + + /* Reads space-separated sequence of integers. */ + std::vector + readIntegers(int size, int minv, int maxv, const std::string &variablesName = "", int indexBase = 1); + + /* Reads space-separated sequence of integers. */ + std::vector readIntegers(int size, int indexBase = 1); + + /* Reads space-separated sequence of integers. */ + std::vector readInts(int size, int minv, int maxv, const std::string &variablesName = "", int indexBase = 1); + + /* Reads space-separated sequence of integers. */ + std::vector readInts(int size, int indexBase = 1); + + /* + * Reads new double. Ignores white-spaces into the non-strict mode + * (strict mode is used in validators usually). */ double readReal(); - /* - * Reads new double. Ignores white-spaces into the non-strict mode - * (strict mode is used in validators usually). + + /* + * Reads new double. Ignores white-spaces into the non-strict mode + * (strict mode is used in validators usually). */ double readDouble(); - + /* As "readReal()" but ensures that value in the range [minv,maxv]. */ - double readReal(double minv, double maxv, const std::string& variableName = ""); + double readReal(double minv, double maxv, const std::string &variableName = ""); + + std::vector + readReals(int size, double minv, double maxv, const std::string &variablesName = "", int indexBase = 1); + + std::vector readReals(int size, int indexBase = 1); + /* As "readDouble()" but ensures that value in the range [minv,maxv]. */ - double readDouble(double minv, double maxv, const std::string& variableName = ""); - - /* + double readDouble(double minv, double maxv, const std::string &variableName = ""); + + std::vector + readDoubles(int size, double minv, double maxv, const std::string &variablesName = "", int indexBase = 1); + + std::vector readDoubles(int size, int indexBase = 1); + + /* * As "readReal()" but ensures that value in the range [minv,maxv] and * number of digit after the decimal point is in range [minAfterPointDigitCount,maxAfterPointDigitCount] * and number is in the form "[-]digit(s)[.digit(s)]". */ double readStrictReal(double minv, double maxv, - int minAfterPointDigitCount, int maxAfterPointDigitCount, - const std::string& variableName = ""); - /* + int minAfterPointDigitCount, int maxAfterPointDigitCount, + const std::string &variableName = ""); + + std::vector readStrictReals(int size, double minv, double maxv, + int minAfterPointDigitCount, int maxAfterPointDigitCount, + const std::string &variablesName = "", int indexBase = 1); + + /* * As "readDouble()" but ensures that value in the range [minv,maxv] and * number of digit after the decimal point is in range [minAfterPointDigitCount,maxAfterPointDigitCount] * and number is in the form "[-]digit(s)[.digit(s)]". */ double readStrictDouble(double minv, double maxv, - int minAfterPointDigitCount, int maxAfterPointDigitCount, - const std::string& variableName = ""); - + int minAfterPointDigitCount, int maxAfterPointDigitCount, + const std::string &variableName = ""); + + std::vector readStrictDoubles(int size, double minv, double maxv, + int minAfterPointDigitCount, int maxAfterPointDigitCount, + const std::string &variablesName = "", int indexBase = 1); + /* As readLine(). */ std::string readString(); + + /* Read many lines. */ + std::vector readStrings(int size, int indexBase = 1); + /* See readLine(). */ - void readStringTo(std::string& result); + void readStringTo(std::string &result); + /* The same as "readLine()/readString()", but ensures that line matches to the given pattern. */ - std::string readString(const pattern& p, const std::string& variableName = ""); + std::string readString(const pattern &p, const std::string &variableName = ""); + /* The same as "readLine()/readString()", but ensures that line matches to the given pattern. */ - std::string readString(const std::string& ptrn, const std::string& variableName = ""); + std::string readString(const std::string &ptrn, const std::string &variableName = ""); + + /* Read many lines. */ + std::vector + readStrings(int size, const pattern &p, const std::string &variableName = "", int indexBase = 1); + + /* Read many lines. */ + std::vector + readStrings(int size, const std::string &ptrn, const std::string &variableName = "", int indexBase = 1); + /* The same as "readLine()/readString()", but ensures that line matches to the given pattern. */ - void readStringTo(std::string& result, const pattern& p, const std::string& variableName = ""); + void readStringTo(std::string &result, const pattern &p, const std::string &variableName = ""); + /* The same as "readLine()/readString()", but ensures that line matches to the given pattern. */ - void readStringTo(std::string& result, const std::string& ptrn, const std::string& variableName = ""); + void readStringTo(std::string &result, const std::string &ptrn, const std::string &variableName = ""); - /* - * Reads line from the current position to EOLN or EOF. Moves stream pointer to - * the first character of the new line (if possible). + /* + * Reads line from the current position to EOLN or EOF. Moves stream pointer to + * the first character of the new line (if possible). */ std::string readLine(); + + /* Read many lines. */ + std::vector readLines(int size, int indexBase = 1); + /* See readLine(). */ - void readLineTo(std::string& result); + void readLineTo(std::string &result); + /* The same as "readLine()", but ensures that line matches to the given pattern. */ - std::string readLine(const pattern& p, const std::string& variableName = ""); + std::string readLine(const pattern &p, const std::string &variableName = ""); + /* The same as "readLine()", but ensures that line matches to the given pattern. */ - std::string readLine(const std::string& ptrn, const std::string& variableName = ""); + std::string readLine(const std::string &ptrn, const std::string &variableName = ""); + + /* Read many lines. */ + std::vector + readLines(int size, const pattern &p, const std::string &variableName = "", int indexBase = 1); + + /* Read many lines. */ + std::vector + readLines(int size, const std::string &ptrn, const std::string &variableName = "", int indexBase = 1); + /* The same as "readLine()", but ensures that line matches to the given pattern. */ - void readLineTo(std::string& result, const pattern& p, const std::string& variableName = ""); + void readLineTo(std::string &result, const pattern &p, const std::string &variableName = ""); + /* The same as "readLine()", but ensures that line matches to the given pattern. */ - void readLineTo(std::string& result, const std::string& ptrn, const std::string& variableName = ""); + void readLineTo(std::string &result, const std::string &ptrn, const std::string &variableName = ""); /* Reads EOLN or fails. Use it in validators. Calls "eoln()" method internally. */ void readEoln(); + /* Reads EOF or fails. Use it in validators. Calls "eof()" method internally. */ void readEof(); - /* + /* * Quit-functions aborts program with and : * input/answer streams replace any result to FAIL. */ - NORETURN void quit(TResult result, const char* msg); - /* + NORETURN void quit(TResult result, const char *msg); + /* * Quit-functions aborts program with and : * input/answer streams replace any result to FAIL. */ - NORETURN void quitf(TResult result, const char* msg, ...); - /* + NORETURN void quitf(TResult result, const char *msg, ...); + + /* + * Quit-functions aborts program with and : + * input/answer streams replace any result to FAIL. + */ + void quitif(bool condition, TResult result, const char *msg, ...); + /* * Quit-functions aborts program with and : * input/answer streams replace any result to FAIL. */ NORETURN void quits(TResult result, std::string msg); + /* + * Checks condition and aborts a program if condition is false. + * Returns _wa for ouf and _fail on any other streams. + */ +#ifdef __GNUC__ + __attribute__ ((format (printf, 3, 4))) +#endif + void ensuref(bool cond, const char *format, ...); + + void __testlib_ensure(bool cond, std::string message); + void close(); - const static WORD LightGray = 0x07; - const static WORD LightRed = 0x0c; - const static WORD LightCyan = 0x0b; - const static WORD LightGreen = 0x0a; - const static WORD LightYellow = 0x0e; + const static int NO_INDEX = INT_MAX; + const static char OPEN_BRACKET = char(11); + const static char CLOSE_BRACKET = char(17); + + const static WORD LightGray = 0x07; + const static WORD LightRed = 0x0c; + const static WORD LightCyan = 0x0b; + const static WORD LightGreen = 0x0a; + const static WORD LightYellow = 0x0e; static void textColor(WORD color); - static void quitscr(WORD color, const char* msg); + + static void quitscr(WORD color, const char *msg); + static void quitscrS(WORD color, std::string msg); - void xmlSafeWrite(std::FILE * file, const char* msg); + + void xmlSafeWrite(std::FILE *file, const char *msg); + + + void readSecret(std::string secret); + void readGraderResult(); + + /* Skips UTF-8 Byte Order Mark. */ + void skipBom(); private: - InStream(const InStream&); - InStream& operator =(const InStream&); + InStream(const InStream &); + + InStream &operator=(const InStream &); }; InStream inf; InStream ouf; InStream ans; bool appesMode; +std::string appesModeEncoding = "windows-1251"; std::string resultName; std::string checkerName = "untitled checker"; random_t rnd; TTestlibMode testlibMode = _unknown; double __testlib_points = std::numeric_limits::infinity(); -struct TestlibFinalizeGuard -{ +struct ValidatorBoundsHit { + static const double EPS; + bool minHit; + bool maxHit; + + ValidatorBoundsHit(bool minHit = false, bool maxHit = false) : minHit(minHit), maxHit(maxHit) { + }; + + ValidatorBoundsHit merge(const ValidatorBoundsHit &validatorBoundsHit, bool ignoreMinBound, bool ignoreMaxBound) { + return ValidatorBoundsHit( + __testlib_max(minHit, validatorBoundsHit.minHit) || ignoreMinBound, + __testlib_max(maxHit, validatorBoundsHit.maxHit) || ignoreMaxBound + ); + } +}; + +const double ValidatorBoundsHit::EPS = 1E-12; + +class Validator { +private: + const static std::string TEST_MARKUP_HEADER; + const static std::string TEST_CASE_OPEN_TAG; + const static std::string TEST_CASE_CLOSE_TAG; + + bool _initialized; + std::string _testset; + std::string _group; + + std::string _testOverviewLogFileName; + std::string _testMarkupFileName; + int _testCase = -1; + std::string _testCaseFileName; + + std::map _boundsHitByVariableName; + std::set _features; + std::set _hitFeatures; + + bool isVariableNameBoundsAnalyzable(const std::string &variableName) { + for (size_t i = 0; i < variableName.length(); i++) + if ((variableName[i] >= '0' && variableName[i] <= '9') || variableName[i] < ' ') + return false; + return true; + } + + bool isFeatureNameAnalyzable(const std::string &featureName) { + for (size_t i = 0; i < featureName.length(); i++) + if (featureName[i] < ' ') + return false; + return true; + } + +public: + Validator() : _initialized(false), _testset("tests"), _group() { + } + + void initialize() { + _initialized = true; + } + + std::string testset() const { + if (!_initialized) + __testlib_fail("Validator should be initialized with registerValidation(argc, argv) instead of registerValidation() to support validator.testset()"); + return _testset; + } + + std::string group() const { + if (!_initialized) + __testlib_fail("Validator should be initialized with registerValidation(argc, argv) instead of registerValidation() to support validator.group()"); + return _group; + } + + std::string testOverviewLogFileName() const { + return _testOverviewLogFileName; + } + + std::string testMarkupFileName() const { + return _testMarkupFileName; + } + + int testCase() const { + return _testCase; + } + + std::string testCaseFileName() const { + return _testCaseFileName; + } + + void setTestset(const char *const testset) { + _testset = testset; + } + + void setGroup(const char *const group) { + _group = group; + } + + void setTestOverviewLogFileName(const char *const testOverviewLogFileName) { + _testOverviewLogFileName = testOverviewLogFileName; + } + + void setTestMarkupFileName(const char *const testMarkupFileName) { + _testMarkupFileName = testMarkupFileName; + } + + void setTestCase(int testCase) { + _testCase = testCase; + } + + void setTestCaseFileName(const char *const testCaseFileName) { + _testCaseFileName = testCaseFileName; + } + + std::string prepVariableName(const std::string &variableName) { + if (variableName.length() >= 2 && variableName != "~~") { + if (variableName[0] == '~' && variableName.back() != '~') + return variableName.substr(1); + if (variableName[0] != '~' && variableName.back() == '~') + return variableName.substr(0, variableName.length() - 1); + if (variableName[0] == '~' && variableName.back() == '~') + return variableName.substr(1, variableName.length() - 2); + } + return variableName; + } + + bool ignoreMinBound(const std::string &variableName) { + return variableName.length() >= 2 && variableName != "~~" && variableName[0] == '~'; + } + + bool ignoreMaxBound(const std::string &variableName) { + return variableName.length() >= 2 && variableName != "~~" && variableName.back() == '~'; + } + + void addBoundsHit(const std::string &variableName, ValidatorBoundsHit boundsHit) { + if (isVariableNameBoundsAnalyzable(variableName)) { + std::string preparedVariableName = prepVariableName(variableName); + _boundsHitByVariableName[preparedVariableName] = boundsHit.merge(_boundsHitByVariableName[preparedVariableName], + ignoreMinBound(variableName), ignoreMaxBound(variableName)); + } + } + + std::string getBoundsHitLog() { + std::string result; + for (std::map::iterator i = _boundsHitByVariableName.begin(); + i != _boundsHitByVariableName.end(); + i++) { + result += "\"" + i->first + "\":"; + if (i->second.minHit) + result += " min-value-hit"; + if (i->second.maxHit) + result += " max-value-hit"; + result += "\n"; + } + return result; + } + + std::string getFeaturesLog() { + std::string result; + for (std::set::iterator i = _features.begin(); + i != _features.end(); + i++) { + result += "feature \"" + *i + "\":"; + if (_hitFeatures.count(*i)) + result += " hit"; + result += "\n"; + } + return result; + } + + void writeTestOverviewLog() { + if (!_testOverviewLogFileName.empty()) { + std::string fileName(_testOverviewLogFileName); + _testOverviewLogFileName = ""; + + FILE* f; + bool standard_file = false; + if (fileName == "stdout") + f = stdout, standard_file = true; + else if (fileName == "stderr") + f = stderr, standard_file = true; + else { + f = fopen(fileName.c_str(), "wb"); + if (NULL == f) + __testlib_fail("Validator::writeTestOverviewLog: can't write test overview log to (" + fileName + ")"); + } + fprintf(f, "%s%s", getBoundsHitLog().c_str(), getFeaturesLog().c_str()); + std::fflush(f); + if (!standard_file) + if (std::fclose(f)) + __testlib_fail("Validator::writeTestOverviewLog: can't close test overview log file (" + fileName + ")"); + } + } + + void writeTestMarkup() { + if (!_testMarkupFileName.empty()) { + std::vector readChars = inf.getReadChars(); + if (!readChars.empty()) { + std::string markup(TEST_MARKUP_HEADER); + for (size_t i = 0; i < readChars.size(); i++) { + int c = readChars[i]; + if (i + 1 == readChars.size() && c == -1) + continue; + if (c <= 256) { + char cc = char(c); + if (cc == '\\' || cc == '!') + markup += '\\'; + markup += cc; + } else { + markup += TEST_CASE_OPEN_TAG; + markup += toString(c - 256); + markup += TEST_CASE_CLOSE_TAG; + } + } + FILE* f; + bool standard_file = false; + if (_testMarkupFileName == "stdout") + f = stdout, standard_file = true; + else if (_testMarkupFileName == "stderr") + f = stderr, standard_file = true; + else { + f = fopen(_testMarkupFileName.c_str(), "wb"); + if (NULL == f) + __testlib_fail("Validator::writeTestMarkup: can't write test markup to (" + _testMarkupFileName + ")"); + } + std::fprintf(f, "%s", markup.c_str()); + std::fflush(f); + if (!standard_file) + if (std::fclose(f)) + __testlib_fail("Validator::writeTestMarkup: can't close test markup file (" + _testCaseFileName + ")"); + } + } + } + + void writeTestCase() { + if (_testCase > 0) { + std::vector readChars = inf.getReadChars(); + if (!readChars.empty()) { + std::string content, testCaseContent; + bool matchedTestCase = false; + for (size_t i = 0; i < readChars.size(); i++) { + int c = readChars[i]; + if (i + 1 == readChars.size() && c == -1) + continue; + if (c <= 256) + content += char(c); + else { + if (matchedTestCase) { + testCaseContent = content; + matchedTestCase = false; + } + content = ""; + int testCase = c - 256; + if (testCase == _testCase) + matchedTestCase = true; + } + } + if (matchedTestCase) + testCaseContent = content; + + if (!testCaseContent.empty()) { + FILE* f; + bool standard_file = false; + if (_testCaseFileName.empty() || _testCaseFileName == "stdout") + f = stdout, standard_file = true; + else if (_testCaseFileName == "stderr") + f = stderr, standard_file = true; + else { + f = fopen(_testCaseFileName.c_str(), "wb"); + if (NULL == f) + __testlib_fail("Validator::writeTestCase: can't write test case to (" + _testCaseFileName + ")"); + } + std::fprintf(f, "%s", testCaseContent.c_str()); + std::fflush(f); + if (!standard_file) + if (std::fclose(f)) + __testlib_fail("Validator::writeTestCase: can't close test case file (" + _testCaseFileName + ")"); + } + } + } + } + + void addFeature(const std::string &feature) { + if (_features.count(feature)) + __testlib_fail("Feature " + feature + " registered twice."); + if (!isFeatureNameAnalyzable(feature)) + __testlib_fail("Feature name '" + feature + "' contains restricted characters."); + + _features.insert(feature); + } + + void feature(const std::string &feature) { + if (!isFeatureNameAnalyzable(feature)) + __testlib_fail("Feature name '" + feature + "' contains restricted characters."); + + if (!_features.count(feature)) + __testlib_fail("Feature " + feature + " didn't registered via addFeature(feature)."); + + _hitFeatures.insert(feature); + } +} validator; + +const std::string Validator::TEST_MARKUP_HEADER = "MU\xF3\x01"; +const std::string Validator::TEST_CASE_OPEN_TAG = "!c"; +const std::string Validator::TEST_CASE_CLOSE_TAG = ";"; + +struct TestlibFinalizeGuard { static bool alive; + static bool registered; + int quitCount, readEofCount; - TestlibFinalizeGuard() : quitCount(0), readEofCount(0) - { + TestlibFinalizeGuard() : quitCount(0), readEofCount(0) { // No operations. } - ~TestlibFinalizeGuard() - { + ~TestlibFinalizeGuard() { bool _alive = alive; alive = false; - if (_alive) - { + if (_alive) { if (testlibMode == _checker && quitCount == 0) __testlib_fail("Checker must end with quit or quitf call."); if (testlibMode == _validator && readEofCount == 0 && quitCount == 0) __testlib_fail("Validator must end with readEof call."); + + /* opts */ + autoEnsureNoUnusedOpts(); + + if (!registered) + __testlib_fail("Call register-function in the first line of the main (registerTestlibCmd or other similar)"); + } + + if (__testlib_exitCode == 0) { + validator.writeTestOverviewLog(); + validator.writeTestMarkup(); + validator.writeTestCase(); } } + +private: + /* opts */ + void autoEnsureNoUnusedOpts(); }; bool TestlibFinalizeGuard::alive = true; -TestlibFinalizeGuard testlibFinalizeGuard; +bool TestlibFinalizeGuard::registered = false; +extern TestlibFinalizeGuard testlibFinalizeGuard; /* * Call it to disable checks on finalization. */ -void disableFinalizeGuard() -{ +void disableFinalizeGuard() { TestlibFinalizeGuard::alive = false; } @@ -1815,57 +2743,85 @@ std::fstream tout; /* implementation */ -template -static std::string vtos(const T& t) -{ - std::string s; - std::stringstream ss; - ss << t; - ss >> s; - return s; -} - -template -static std::string toString(const T& t) -{ - return vtos(t); -} - -InStream::InStream() -{ - file = NULL; +InStream::InStream() { + reader = NULL; + lastLine = -1; + opened = false; name = ""; mode = _input; strict = false; stdfile = false; wordReserveSize = 4; + readManyIteration = NO_INDEX; + maxFileSize = 128 * 1024 * 1024; // 128MB. + maxTokenLength = 32 * 1024 * 1024; // 32MB. + maxMessageLength = 32000; } -InStream::InStream(const InStream& baseStream, std::string content) -{ - file = (FILE*)0xbadfeed; - stdfile = false; +InStream::InStream(const InStream &baseStream, std::string content) { + + // UOJ Modified Here + //file = (FILE*)0xbadfeed; + //stdfile = false; + // End of modification + reader = new StringInputStreamReader(content); + lastLine = -1; opened = true; strict = baseStream.strict; + stdfile = false; mode = baseStream.mode; name = "based on " + baseStream.name; + readManyIteration = NO_INDEX; + maxFileSize = 128 * 1024 * 1024; // 128MB. + maxTokenLength = 32 * 1024 * 1024; // 32MB. + maxMessageLength = 32000; } -InStream::~InStream() -{ - if (NULL != reader) - { +InStream::~InStream() { + if (NULL != reader) { + reader->close(); delete reader; reader = NULL; } } +void InStream::setTestCase(int testCase) { + if (testlibMode != _validator || mode != _input || !stdfile || this != &inf) + __testlib_fail("InStream::setTestCase can be used only for inf in validator-mode." + " Actually, prefer setTestCase function instead of InStream member"); + reader->setTestCase(testCase); +} + +std::vector InStream::getReadChars() { + if (testlibMode != _validator || mode != _input || !stdfile || this != &inf) + __testlib_fail("InStream::getReadChars can be used only for inf in validator-mode."); + return reader == NULL ? std::vector() : reader->getReadChars(); +} + +void setTestCase(int testCase) { + static bool first_run = true; + static bool zero_based = false; + + if (first_run && testCase == 0) + zero_based = true; + + if (zero_based) + testCase++; + + __testlib_hasTestCase = true; + __testlib_testCase = testCase; + + if (testlibMode == _validator) + inf.setTestCase(testCase); + + first_run = false; +} + #ifdef __GNUC__ __attribute__((const)) #endif -int resultExitCode(TResult r) -{ +int resultExitCode(TResult r) { if (r == _ok) return OK_EXIT_CODE; if (r == _wa) @@ -1889,162 +2845,257 @@ int resultExitCode(TResult r) return FAIL_EXIT_CODE; } -void InStream::textColor(WORD color) -{ -#if defined(ON_WINDOWS) && (!defined(_MSC_VER) || _MSC_VER>1400) +void InStream::textColor( +#if !(defined(ON_WINDOWS) && (!defined(_MSC_VER) || _MSC_VER > 1400)) && defined(__GNUC__) + __attribute__((unused)) +#endif + WORD color +) { +#if defined(ON_WINDOWS) && (!defined(_MSC_VER) || _MSC_VER > 1400) HANDLE handle = GetStdHandle(STD_OUTPUT_HANDLE); SetConsoleTextAttribute(handle, color); #endif +#if !defined(ON_WINDOWS) && defined(__GNUC__) + if (isatty(2)) + { + switch (color) + { + case LightRed: + fprintf(stderr, "\033[1;31m"); + break; + case LightCyan: + fprintf(stderr, "\033[1;36m"); + break; + case LightGreen: + fprintf(stderr, "\033[1;32m"); + break; + case LightYellow: + fprintf(stderr, "\033[1;33m"); + break; + case LightGray: + default: + fprintf(stderr, "\033[0m"); + } + } +#endif +} -#ifdef linux - char *shell_path = getenv("SHELL"); - if (shell_path && strcmp(shell_path, "/bin/bash") == 0 && isatty(2)) - { - switch (color) - { - case 0: - fprintf(stderr, "\e[0m"); - break; - case LightRed: - fprintf(stderr, "\e[0;31m"); - break; - case LightGreen: - fprintf(stderr, "\e[0;32m"); - break; - case LightYellow: - fprintf(stderr, "\e[0;33m"); - break; - case LightCyan: - fprintf(stderr, "\e[0;36m"); - break; - case LightGray: - fprintf(stderr, "\e[0m"); - break; - } - } -#endif -} - -NORETURN void halt(int exitCode) -{ +#ifdef TESTLIB_THROW_EXIT_EXCEPTION_INSTEAD_OF_EXIT +class exit_exception: public std::exception { +private: + int exitCode; +public: + exit_exception(int exitCode): exitCode(exitCode) {} + int getExitCode() { return exitCode; } +}; +#endif + +NORETURN void halt(int exitCode) { #ifdef FOOTER InStream::textColor(InStream::LightGray); std::fprintf(stderr, "Checker: \"%s\"\n", checkerName.c_str()); std::fprintf(stderr, "Exit code: %d\n", exitCode); InStream::textColor(InStream::LightGray); #endif -#ifdef linux - InStream::textColor(0); + __testlib_exitCode = exitCode; +#ifdef TESTLIB_THROW_EXIT_EXCEPTION_INSTEAD_OF_EXIT + throw exit_exception(exitCode); #endif std::exit(exitCode); } -NORETURN void InStream::quit(TResult result, const char* msg) -{ - if (TestlibFinalizeGuard::alive) - testlibFinalizeGuard.quitCount++; - - if (mode != _output && result != _fail) - quits(_fail, std::string(msg) + " (" + name + ")"); +static bool __testlib_shouldCheckDirt(TResult result) { + return result == _ok || result == _points || result >= _partially; +} - std::FILE * resultFile; - std::string errorName; - - if (result == _ok) - { - if (testlibMode != _interactor && !ouf.seekEof()) - quit(_dirt, "Extra information in the output file"); +static std::string __testlib_appendMessage(const std::string &message, const std::string &extra) { + int openPos = -1, closePos = -1; + for (size_t i = 0; i < message.length(); i++) { + if (message[i] == InStream::OPEN_BRACKET) { + if (openPos == -1) + openPos = int(i); + else + openPos = INT_MAX; + } + if (message[i] == InStream::CLOSE_BRACKET) { + if (closePos == -1) + closePos = int(i); + else + closePos = INT_MAX; + } + } + if (openPos != -1 && openPos != INT_MAX + && closePos != -1 && closePos != INT_MAX + && openPos < closePos) { + size_t index = message.find(extra, openPos); + if (index == std::string::npos || int(index) >= closePos) { + std::string result(message); + result.insert(closePos, ", " + extra); + return result; + } + return message; } - int pctype = result - _partially; - bool isPartial = false; + return message + " " + InStream::OPEN_BRACKET + extra + InStream::CLOSE_BRACKET; +} - switch (result) - { - case _ok: - errorName = "ok "; - quitscrS(LightGreen, errorName); - break; - case _wa: - errorName = "wrong answer "; - quitscrS(LightRed, errorName); - break; - case _pe: - errorName = "wrong output format "; - quitscrS(LightRed, errorName); - break; - case _fail: - errorName = "FAIL "; - quitscrS(LightRed, errorName); - break; - case _dirt: - errorName = "wrong output format "; - quitscrS(LightCyan, errorName); +static std::string __testlib_toPrintableMessage(const std::string &message) { + int openPos = -1, closePos = -1; + for (size_t i = 0; i < message.length(); i++) { + if (message[i] == InStream::OPEN_BRACKET) { + if (openPos == -1) + openPos = int(i); + else + openPos = INT_MAX; + } + if (message[i] == InStream::CLOSE_BRACKET) { + if (closePos == -1) + closePos = int(i); + else + closePos = INT_MAX; + } + } + if (openPos != -1 && openPos != INT_MAX + && closePos != -1 && closePos != INT_MAX + && openPos < closePos) { + std::string result(message); + result[openPos] = '('; + result[closePos] = ')'; + return result; + } + + return message; +} + +NORETURN void InStream::quit(TResult result, const char *msg) { + if (TestlibFinalizeGuard::alive) + testlibFinalizeGuard.quitCount++; + + std::string message(msg); + message = trim(message); + + if (__testlib_hasTestCase) { + if (result != _ok) + message = __testlib_appendMessage(message, "test case " + vtos(__testlib_testCase)); + else { + if (__testlib_testCase == 1) + message = __testlib_appendMessage(message, vtos(__testlib_testCase) + " test case"); + else + message = __testlib_appendMessage(message, vtos(__testlib_testCase) + " test cases"); + } + } + + // You can change maxMessageLength. + // Example: 'inf.maxMessageLength = 1024 * 1024;'. + if (message.length() > maxMessageLength) { + std::string warn = "message length exceeds " + vtos(maxMessageLength) + + ", the message is truncated: "; + message = warn + message.substr(0, maxMessageLength - warn.length()); + } + +#ifndef ENABLE_UNEXPECTED_EOF + if (result == _unexpected_eof) result = _pe; - break; - case _points: - errorName = "points "; - quitscrS(LightYellow, errorName); - break; - case _unexpected_eof: -#ifdef ENABLE_UNEXPECTED_EOF - errorName = "unexpected eof "; -#else - errorName = "wrong output format "; #endif - quitscrS(LightCyan, errorName); - break; - default: - if (result >= _partially) - { - errorName = format("partially correct (%d) ", pctype); - isPartial = true; - quitscrS(LightYellow, errorName); - } + + if (testlibMode == _scorer && result != _fail) + quits(_fail, "Scorer should return points only. Don't use a quit function."); + + if (mode != _output && result != _fail) { + if (mode == _input && testlibMode == _validator && lastLine != -1) + quits(_fail, __testlib_appendMessage(__testlib_appendMessage(message, name), "line " + vtos(lastLine))); else - quit(_fail, "What is the code ??? "); + quits(_fail, __testlib_appendMessage(message, name)); } - if (resultName != "") - { - resultFile = __testlib_fopen(resultName.c_str(), "w"); - if (resultFile == NULL) + std::FILE *resultFile; + std::string errorName; + + if (__testlib_shouldCheckDirt(result)) { + if (testlibMode != _interactor && !ouf.seekEof()) + quit(_dirt, "Extra information in the output file"); + } + + int pctype = result - _partially; + bool isPartial = false; + + switch (result) { + case _ok: + errorName = "ok "; + quitscrS(LightGreen, errorName); + break; + case _wa: + errorName = "wrong answer "; + quitscrS(LightRed, errorName); + break; + case _pe: + errorName = "wrong output format "; + quitscrS(LightRed, errorName); + break; + case _fail: + errorName = "FAIL "; + quitscrS(LightRed, errorName); + break; + case _dirt: + errorName = "wrong output format "; + quitscrS(LightCyan, errorName); + result = _pe; + break; + case _points: + errorName = "points "; + quitscrS(LightYellow, errorName); + break; + case _unexpected_eof: + errorName = "unexpected eof "; + quitscrS(LightCyan, errorName); + break; + default: + if (result >= _partially) { + errorName = format("partially correct (%d) ", pctype); + isPartial = true; + quitscrS(LightYellow, errorName); + } else + quit(_fail, "What is the code ??? "); + } + + if (resultName != "") { + resultFile = std::fopen(resultName.c_str(), "w"); + if (resultFile == NULL) { + resultName = ""; quit(_fail, "Can not write to the result file"); - if (appesMode) - { - std::fprintf(resultFile, ""); + } + if (appesMode) { + std::fprintf(resultFile, "", appesModeEncoding.c_str()); if (isPartial) - std::fprintf(resultFile, "", outcomes[(int)_partially].c_str(), pctype); - else - { + std::fprintf(resultFile, "", + outcomes[(int) _partially].c_str(), pctype); + else { if (result != _points) - std::fprintf(resultFile, "", outcomes[(int)result].c_str()); - else - { + std::fprintf(resultFile, "", outcomes[(int) result].c_str()); + else { if (__testlib_points == std::numeric_limits::infinity()) quit(_fail, "Expected points, but infinity found"); std::string stringPoints = removeDoubleTrailingZeroes(format("%.10f", __testlib_points)); - std::fprintf(resultFile, "", outcomes[(int)result].c_str(), stringPoints.c_str()); + std::fprintf(resultFile, "", + outcomes[(int) result].c_str(), stringPoints.c_str()); } } - xmlSafeWrite(resultFile, msg); + xmlSafeWrite(resultFile, __testlib_toPrintableMessage(message).c_str()); std::fprintf(resultFile, "\n"); - } - else - std::fprintf(resultFile, "%s", msg); - if (NULL == resultFile || fclose(resultFile) != 0) + } else + std::fprintf(resultFile, "%s %s", errorName.c_str(), __testlib_toPrintableMessage(message).c_str()); + if (NULL == resultFile || fclose(resultFile) != 0) { + resultName = ""; quit(_fail, "Can not write to the result file"); + } } - quitscr(LightGray, msg); + quitscr(LightGray, __testlib_toPrintableMessage(message).c_str()); std::fprintf(stderr, "\n"); - if (inf.file) - fclose(inf.file); - if (ouf.file) - fclose(ouf.file); - if (ans.file) - fclose(ans.file); + inf.close(); + ouf.close(); + ans.close(); if (tout.is_open()) tout.close(); @@ -2057,46 +3108,47 @@ NORETURN void InStream::quit(TResult result, const char* msg) } #ifdef __GNUC__ - __attribute__ ((format (printf, 3, 4))) +__attribute__ ((format (printf, 3, 4))) #endif -NORETURN void InStream::quitf(TResult result, const char* msg, ...) -{ +NORETURN void InStream::quitf(TResult result, const char *msg, ...) { FMT_TO_RESULT(msg, msg, message); InStream::quit(result, message.c_str()); } -NORETURN void InStream::quits(TResult result, std::string msg) -{ +#ifdef __GNUC__ +__attribute__ ((format (printf, 4, 5))) +#endif +void InStream::quitif(bool condition, TResult result, const char *msg, ...) { + if (condition) { + FMT_TO_RESULT(msg, msg, message); + InStream::quit(result, message.c_str()); + } +} + +NORETURN void InStream::quits(TResult result, std::string msg) { InStream::quit(result, msg.c_str()); } -void InStream::xmlSafeWrite(std::FILE * file, const char* msg) -{ +void InStream::xmlSafeWrite(std::FILE *file, const char *msg) { size_t lmsg = strlen(msg); - for (size_t i = 0; i < lmsg; i++) - { - if (msg[i] == '&') - { + for (size_t i = 0; i < lmsg; i++) { + if (msg[i] == '&') { std::fprintf(file, "%s", "&"); continue; } - if (msg[i] == '<') - { + if (msg[i] == '<') { std::fprintf(file, "%s", "<"); continue; } - if (msg[i] == '>') - { + if (msg[i] == '>') { std::fprintf(file, "%s", ">"); continue; } - if (msg[i] == '"') - { + if (msg[i] == '"') { std::fprintf(file, "%s", """); continue; } - if (0 <= msg[i] && msg[i] <= 31) - { + if (0 <= msg[i] && msg[i] <= 31) { std::fprintf(file, "%c", '.'); continue; } @@ -2104,150 +3156,168 @@ void InStream::xmlSafeWrite(std::FILE * file, const char* msg) } } -void InStream::quitscrS(WORD color, std::string msg) -{ +void InStream::quitscrS(WORD color, std::string msg) { quitscr(color, msg.c_str()); } -void InStream::quitscr(WORD color, const char* msg) -{ - if (resultName == "") - { +void InStream::quitscr(WORD color, const char *msg) { + if (resultName == "") { textColor(color); std::fprintf(stderr, "%s", msg); textColor(LightGray); } } -void InStream::reset() -{ +void InStream::reset(std::FILE *file) { if (opened && stdfile) quit(_fail, "Can't reset standard handle"); if (opened) close(); - if (!stdfile) - if (NULL == (file = __testlib_fopen(name.c_str(), "rb"))) - { + if (!stdfile && NULL == file) + if (NULL == (file = std::fopen(name.c_str(), "rb"))) { if (mode == _output) - quits(_pe, std::string("File not found: \"") + name + "\""); - } + quits(_pe, std::string("Output file not found: \"") + name + "\""); - opened = true; + if (mode == _answer) + quits(_fail, std::string("Answer file not found: \"") + name + "\""); + } - __testlib_set_binary(file); + if (NULL != file) { + opened = true; + __testlib_set_binary(file); - if (stdfile) - reader = new FileInputStreamReader(file, name); - else - reader = new BufferedFileInputStreamReader(file, name); + if (stdfile) + reader = new FileInputStreamReader(file, name); + else + reader = new BufferedFileInputStreamReader(file, name); + } else { + opened = false; + reader = NULL; + } } -void InStream::init(std::string fileName, TMode mode) -{ +void InStream::init(std::string fileName, TMode mode) { opened = false; - if (fileName == "/dev/stdin") - { - name = "stdin"; - this->file = stdin; - stdfile = true; - } - else - { - name = fileName; - stdfile = false; - } + //UOJ Modified Here + //name = fileName; + //stdfile = false; + if (fileName == "/dev/stdin"){ + name = "stdin"; + //this->file = stdin; + stdfile = true; + }else{ + name = fileName; + stdfile = false; + } + this->mode = mode; + + std::ifstream stream; + stream.open(fileName.c_str(), std::ios::in); + if (stream.is_open()) { + std::streampos start = stream.tellg(); + stream.seekg(0, std::ios::end); + std::streampos end = stream.tellg(); + size_t fileSize = size_t(end - start); + stream.close(); + + // You can change maxFileSize. + // Example: 'inf.maxFileSize = 256 * 1024 * 1024;'. + if (fileSize > maxFileSize) + quitf(_pe, "File size exceeds %d bytes, size is %d", int(maxFileSize), int(fileSize)); + } + reset(); } -void InStream::init(std::FILE* f, TMode mode) -{ +void InStream::init(std::FILE *f, TMode mode) { opened = false; - name = "untitled"; - + this->mode = mode; + if (f == stdin) name = "stdin", stdfile = true; - if (f == stdout) name = "stdout", stdfile = true; - if (f == stderr) name = "stderr", stdfile = true; - this->file = f; - this->mode = mode; - - reset(); + reset(f); } -char InStream::curChar() -{ - int c = reader->curChar(); - return char(c != EOF ? c : EOFC); +void InStream::skipBom() { + const std::string utf8Bom = "\xEF\xBB\xBF"; + size_t index = 0; + while (index < utf8Bom.size() && curChar() == utf8Bom[index]) { + index++; + skipChar(); + } + if (index < utf8Bom.size()) { + while (index != 0) { + unreadChar(utf8Bom[index - 1]); + index--; + } + } } -char InStream::nextChar() -{ - int c = reader->nextChar(); - return char(c != EOF ? c : EOFC); +char InStream::curChar() { + return char(reader->curChar()); } -char InStream::readChar() -{ +char InStream::nextChar() { + return char(reader->nextChar()); +} + +char InStream::readChar() { return nextChar(); } -char InStream::readChar(char c) -{ +char InStream::readChar(char c) { + lastLine = reader->getLine(); char found = readChar(); - if (c != found) - { + if (c != found) { if (!isEoln(found)) - quit(_pe, ("Unexpected character '" + std::string(1, found) + "', but '" + std::string(1, c) + "' expected").c_str()); + quit(_pe, ("Unexpected character '" + std::string(1, found) + "', but '" + std::string(1, c) + + "' expected").c_str()); else - quit(_pe, ("Unexpected character " + ("#" + vtos(int(found))) + ", but '" + std::string(1, c) + "' expected").c_str()); + quit(_pe, ("Unexpected character " + ("#" + vtos(int(found))) + ", but '" + std::string(1, c) + + "' expected").c_str()); } return found; } -char InStream::readSpace() -{ +char InStream::readSpace() { return readChar(' '); } -void InStream::unreadChar(char c) -{ +void InStream::unreadChar(char c) { reader->unreadChar(c); } -void InStream::skipChar() -{ +void InStream::skipChar() { reader->skipChar(); } -void InStream::skipBlanks() -{ +void InStream::skipBlanks() { while (isBlanks(reader->curChar())) reader->skipChar(); } -std::string InStream::readWord() -{ +std::string InStream::readWord() { readWordTo(_tmpReadToken); return _tmpReadToken; } -void InStream::readWordTo(std::string& result) -{ +void InStream::readWordTo(std::string &result) { if (!strict) skipBlanks(); + lastLine = reader->getLine(); int cur = reader->nextChar(); - if (cur == EOF) + if (cur == EOFC) quit(_unexpected_eof, "Unexpected end of file - token expected"); if (isBlanks(cur)) @@ -2255,9 +3325,15 @@ void InStream::readWordTo(std::string& result) result.clear(); - while (!(isBlanks(cur) || cur == EOF)) - { + while (!(isBlanks(cur) || cur == EOFC)) { result += char(cur); + + // You can change maxTokenLength. + // Example: 'inf.maxTokenLength = 128 * 1024 * 1024;'. + if (result.length() > maxTokenLength) + quitf(_pe, "Length of token exceeds %d, token is '%s...'", int(maxTokenLength), + __testlib_part(result).c_str()); + cur = reader->nextChar(); } @@ -2267,84 +3343,146 @@ void InStream::readWordTo(std::string& result) quit(_unexpected_eof, "Unexpected end of file or white-space - token expected"); } -std::string InStream::readToken() -{ +std::string InStream::readToken() { return readWord(); } -void InStream::readTokenTo(std::string& result) -{ +void InStream::readTokenTo(std::string &result) { readWordTo(result); } -static std::string __testlib_part(const std::string& s) -{ - if (s.length() <= 64) - return s; +#ifdef __GNUC__ +__attribute__((const)) +#endif +static std::string __testlib_part(const std::string &s) { + std::string t; + for (size_t i = 0; i < s.length(); i++) + if (s[i] != '\0') + t += s[i]; + else + t += '~'; + if (t.length() <= 64) + return t; else - return s.substr(0, 30) + "..." + s.substr(s.length() - 31, 31); -} - -std::string InStream::readWord(const pattern& p, const std::string& variableName) -{ + return t.substr(0, 30) + "..." + t.substr(s.length() - 31, 31); +} + +#define __testlib_readMany(readMany, readOne, typeName, space) \ + if (size < 0) \ + quit(_fail, #readMany ": size should be non-negative."); \ + if (size > 100000000) \ + quit(_fail, #readMany ": size should be at most 100000000."); \ + \ + std::vector result(size); \ + readManyIteration = indexBase; \ + \ + for (int i = 0; i < size; i++) \ + { \ + result[i] = readOne; \ + readManyIteration++; \ + if (strict && space && i + 1 < size) \ + readSpace(); \ + } \ + \ + readManyIteration = NO_INDEX; \ + return result; \ + + +std::string InStream::readWord(const pattern &p, const std::string &variableName) { readWordTo(_tmpReadToken); - if (!p.matches(_tmpReadToken)) - { - if (variableName.empty()) - quit(_wa, ("Token \"" + __testlib_part(_tmpReadToken) + "\" doesn't correspond to pattern \"" + p.src() + "\"").c_str()); - else - quit(_wa, ("Token parameter [name=" + variableName + "] equals to \"" + __testlib_part(_tmpReadToken) + "\", doesn't correspond to pattern \"" + p.src() + "\"").c_str()); + if (!p.matches(_tmpReadToken)) { + if (readManyIteration == NO_INDEX) { + if (variableName.empty()) + quit(_wa, + ("Token \"" + __testlib_part(_tmpReadToken) + "\" doesn't correspond to pattern \"" + p.src() + + "\"").c_str()); + else + quit(_wa, ("Token parameter [name=" + variableName + "] equals to \"" + __testlib_part(_tmpReadToken) + + "\", doesn't correspond to pattern \"" + p.src() + "\"").c_str()); + } else { + if (variableName.empty()) + quit(_wa, ("Token element [index=" + vtos(readManyIteration) + "] equals to \"" + + __testlib_part(_tmpReadToken) + "\" doesn't correspond to pattern \"" + p.src() + + "\"").c_str()); + else + quit(_wa, ("Token element " + variableName + "[" + vtos(readManyIteration) + "] equals to \"" + + __testlib_part(_tmpReadToken) + "\", doesn't correspond to pattern \"" + p.src() + + "\"").c_str()); + } } return _tmpReadToken; } -std::string InStream::readWord(const std::string& ptrn, const std::string& variableName) -{ +std::vector +InStream::readWords(int size, const pattern &p, const std::string &variablesName, int indexBase) { + __testlib_readMany(readWords, readWord(p, variablesName), std::string, true); +} + +std::vector InStream::readWords(int size, int indexBase) { + __testlib_readMany(readWords, readWord(), std::string, true); +} + +std::string InStream::readWord(const std::string &ptrn, const std::string &variableName) { return readWord(pattern(ptrn), variableName); } -std::string InStream::readToken(const pattern& p, const std::string& variableName) -{ +std::vector +InStream::readWords(int size, const std::string &ptrn, const std::string &variablesName, int indexBase) { + pattern p(ptrn); + __testlib_readMany(readWords, readWord(p, variablesName), std::string, true); +} + +std::string InStream::readToken(const pattern &p, const std::string &variableName) { return readWord(p, variableName); } -std::string InStream::readToken(const std::string& ptrn, const std::string& variableName) -{ +std::vector +InStream::readTokens(int size, const pattern &p, const std::string &variablesName, int indexBase) { + __testlib_readMany(readTokens, readToken(p, variablesName), std::string, true); +} + +std::vector InStream::readTokens(int size, int indexBase) { + __testlib_readMany(readTokens, readToken(), std::string, true); +} + +std::string InStream::readToken(const std::string &ptrn, const std::string &variableName) { return readWord(ptrn, variableName); } -void InStream::readWordTo(std::string& result, const pattern& p, const std::string& variableName) -{ +std::vector +InStream::readTokens(int size, const std::string &ptrn, const std::string &variablesName, int indexBase) { + pattern p(ptrn); + __testlib_readMany(readTokens, readWord(p, variablesName), std::string, true); +} + +void InStream::readWordTo(std::string &result, const pattern &p, const std::string &variableName) { readWordTo(result); - if (!p.matches(result)) - { + if (!p.matches(result)) { if (variableName.empty()) - quit(_wa, ("Token \"" + __testlib_part(result) + "\" doesn't correspond to pattern \"" + p.src() + "\"").c_str()); + quit(_wa, ("Token \"" + __testlib_part(result) + "\" doesn't correspond to pattern \"" + p.src() + + "\"").c_str()); else - quit(_wa, ("Token parameter [name=" + variableName + "] equals to \"" + __testlib_part(result) + "\", doesn't correspond to pattern \"" + p.src() + "\"").c_str()); + quit(_wa, ("Token parameter [name=" + variableName + "] equals to \"" + __testlib_part(result) + + "\", doesn't correspond to pattern \"" + p.src() + "\"").c_str()); } } -void InStream::readWordTo(std::string& result, const std::string& ptrn, const std::string& variableName) -{ +void InStream::readWordTo(std::string &result, const std::string &ptrn, const std::string &variableName) { return readWordTo(result, pattern(ptrn), variableName); } -void InStream::readTokenTo(std::string& result, const pattern& p, const std::string& variableName) -{ +void InStream::readTokenTo(std::string &result, const pattern &p, const std::string &variableName) { return readWordTo(result, p, variableName); } -void InStream::readTokenTo(std::string& result, const std::string& ptrn, const std::string& variableName) -{ +void InStream::readTokenTo(std::string &result, const std::string &ptrn, const std::string &variableName) { return readWordTo(result, ptrn, variableName); } #ifdef __GNUC__ __attribute__((pure)) #endif -static inline bool equals(long long integer, const char* s) -{ +static inline bool equals(long long integer, const char *s) { if (integer == LLONG_MIN) return strcmp(s, "-9223372036854775808") == 0; @@ -2365,8 +3503,7 @@ static inline bool equals(long long integer, const char* s) if (length == 0) return false; - while (integer > 0) - { + while (integer > 0) { int digit = int(integer % 10); if (s[length - 1] != '0' + digit) @@ -2379,9 +3516,36 @@ static inline bool equals(long long integer, const char* s) return length == 0; } -static inline double stringToDouble(InStream& in, const char* buffer) -{ - double retval; +#ifdef __GNUC__ +__attribute__((pure)) +#endif +static inline bool equals(unsigned long long integer, const char *s) { + if (integer == ULLONG_MAX) + return strcmp(s, "18446744073709551615") == 0; + + if (integer == 0ULL) + return strcmp(s, "0") == 0; + + size_t length = strlen(s); + + if (length == 0) + return false; + + while (integer > 0) { + int digit = int(integer % 10); + + if (s[length - 1] != '0' + digit) + return false; + + length--; + integer /= 10; + } + + return length == 0; +} + +static inline double stringToDouble(InStream &in, const char *buffer) { + double result; size_t length = strlen(buffer); @@ -2391,12 +3555,10 @@ static inline double stringToDouble(InStream& in, const char* buffer) int digitCount = 0; int eCount = 0; - for (size_t i = 0; i < length; i++) - { + for (size_t i = 0; i < length; i++) { if (('0' <= buffer[i] && buffer[i] <= '9') || buffer[i] == '.' - || buffer[i] == 'e' || buffer[i] == 'E' - || buffer[i] == '-' || buffer[i] == '+') - { + || buffer[i] == 'e' || buffer[i] == 'E' + || buffer[i] == '-' || buffer[i] == '+') { if ('0' <= buffer[i] && buffer[i] <= '9') digitCount++; if (buffer[i] == 'e' || buffer[i] == 'E') @@ -2407,8 +3569,7 @@ static inline double stringToDouble(InStream& in, const char* buffer) plusCount++; if (buffer[i] == '.') decimalPointCount++; - } - else + } else in.quit(_pe, ("Expected double, but \"" + __testlib_part(buffer) + "\" found").c_str()); } @@ -2416,30 +3577,37 @@ static inline double stringToDouble(InStream& in, const char* buffer) if (digitCount == 0 || minusCount > 2 || plusCount > 2 || decimalPointCount > 1 || eCount > 1) in.quit(_pe, ("Expected double, but \"" + __testlib_part(buffer) + "\" found").c_str()); - char* suffix = new char[length + 1]; - int scanned = std::sscanf(buffer, "%lf%s", &retval, suffix); + char *suffix = new char[length + 1]; + std::memset(suffix, 0, length + 1); + int scanned = std::sscanf(buffer, "%lf%s", &result, suffix); bool empty = strlen(suffix) == 0; delete[] suffix; - if (scanned == 1 || (scanned == 2 && empty)) - { - if (__testlib_isNaN(retval) || __testlib_isInfinite(retval)) + if (scanned == 1 || (scanned == 2 && empty)) { + if (__testlib_isNaN(result)) in.quit(_pe, ("Expected double, but \"" + __testlib_part(buffer) + "\" found").c_str()); - return retval; - } - else + return result; + } else in.quit(_pe, ("Expected double, but \"" + __testlib_part(buffer) + "\" found").c_str()); } -static inline double stringToStrictDouble(InStream& in, const char* buffer, int minAfterPointDigitCount, int maxAfterPointDigitCount) -{ +static inline double stringToDouble(InStream &in, const std::string& buffer) { + for (size_t i = 0; i < buffer.length(); i++) + if (buffer[i] == '\0') + in.quit(_pe, ("Expected double, but \"" + __testlib_part(buffer) + "\" found (it contains \\0)").c_str()); + return stringToDouble(in, buffer.c_str()); +} + +static inline double stringToStrictDouble(InStream &in, const char *buffer, + int minAfterPointDigitCount, int maxAfterPointDigitCount) { if (minAfterPointDigitCount < 0) in.quit(_fail, "stringToStrictDouble: minAfterPointDigitCount should be non-negative."); - + if (minAfterPointDigitCount > maxAfterPointDigitCount) - in.quit(_fail, "stringToStrictDouble: minAfterPointDigitCount should be less or equal to maxAfterPointDigitCount."); + in.quit(_fail, + "stringToStrictDouble: minAfterPointDigitCount should be less or equal to maxAfterPointDigitCount."); - double retval; + double result; size_t length = strlen(buffer); @@ -2449,11 +3617,9 @@ static inline double stringToStrictDouble(InStream& in, const char* buffer, int if (buffer[0] != '-' && (buffer[0] < '0' || buffer[0] > '9')) in.quit(_pe, ("Expected strict double, but \"" + __testlib_part(buffer) + "\" found").c_str()); - int pointPos = -1; - for (size_t i = 1; i + 1 < length; i++) - { - if (buffer[i] == '.') - { + int pointPos = -1; + for (size_t i = 1; i + 1 < length; i++) { + if (buffer[i] == '.') { if (pointPos > -1) in.quit(_pe, ("Expected strict double, but \"" + __testlib_part(buffer) + "\" found").c_str()); pointPos = int(i); @@ -2468,63 +3634,60 @@ static inline double stringToStrictDouble(InStream& in, const char* buffer, int int afterDigitsCount = (pointPos == -1 ? 0 : int(length) - pointPos - 1); if (afterDigitsCount < minAfterPointDigitCount || afterDigitsCount > maxAfterPointDigitCount) in.quit(_pe, ("Expected strict double with number of digits after point in range [" - + vtos(minAfterPointDigitCount) - + "," - + vtos(maxAfterPointDigitCount) - + "], but \"" + __testlib_part(buffer) + "\" found").c_str() + + vtos(minAfterPointDigitCount) + + "," + + vtos(maxAfterPointDigitCount) + + "], but \"" + __testlib_part(buffer) + "\" found").c_str() ); int firstDigitPos = -1; for (size_t i = 0; i < length; i++) - if (buffer[i] >= '0' && buffer[i] <= '9') - { + if (buffer[i] >= '0' && buffer[i] <= '9') { firstDigitPos = int(i); break; } - if (firstDigitPos > 1 || firstDigitPos == -1) + if (firstDigitPos > 1 || firstDigitPos == -1) in.quit(_pe, ("Expected strict double, but \"" + __testlib_part(buffer) + "\" found").c_str()); if (buffer[firstDigitPos] == '0' && firstDigitPos + 1 < int(length) - && buffer[firstDigitPos + 1] >= '0' && buffer[firstDigitPos + 1] <= '9') + && buffer[firstDigitPos + 1] >= '0' && buffer[firstDigitPos + 1] <= '9') in.quit(_pe, ("Expected strict double, but \"" + __testlib_part(buffer) + "\" found").c_str()); - char* suffix = new char[length + 1]; - int scanned = std::sscanf(buffer, "%lf%s", &retval, suffix); + char *suffix = new char[length + 1]; + std::memset(suffix, 0, length + 1); + int scanned = std::sscanf(buffer, "%lf%s", &result, suffix); bool empty = strlen(suffix) == 0; delete[] suffix; - if (scanned == 1 || (scanned == 2 && empty)) - { - if (__testlib_isNaN(retval) || __testlib_isInfinite(retval)) + if (scanned == 1 || (scanned == 2 && empty)) { + if (__testlib_isNaN(result) || __testlib_isInfinite(result)) in.quit(_pe, ("Expected double, but \"" + __testlib_part(buffer) + "\" found").c_str()); - return retval; - } - else + if (buffer[0] == '-' && result >= 0) + in.quit(_pe, ("Redundant minus in \"" + __testlib_part(buffer) + "\" found").c_str()); + return result; + } else in.quit(_pe, ("Expected double, but \"" + __testlib_part(buffer) + "\" found").c_str()); } -static inline long long stringToLongLong(InStream& in, const char* buffer) -{ - if (strcmp(buffer, "-9223372036854775808") == 0) - return LLONG_MIN; +static inline double stringToStrictDouble(InStream &in, const std::string& buffer, + int minAfterPointDigitCount, int maxAfterPointDigitCount) { + for (size_t i = 0; i < buffer.length(); i++) + if (buffer[i] == '\0') + in.quit(_pe, ("Expected double, but \"" + __testlib_part(buffer) + "\" found (it contains \\0)").c_str()); + return stringToStrictDouble(in, buffer.c_str(), minAfterPointDigitCount, maxAfterPointDigitCount); +} - bool minus = false; +static inline long long stringToLongLong(InStream &in, const char *buffer) { size_t length = strlen(buffer); - - if (length > 1 && buffer[0] == '-') - minus = true; - - if (length > 20) + if (length == 0 || length > 20) in.quit(_pe, ("Expected integer, but \"" + __testlib_part(buffer) + "\" found").c_str()); - long long retval = 0LL; - + bool has_minus = (length > 1 && buffer[0] == '-'); int zeroes = 0; - int processingZeroes = true; - - for (int i = (minus ? 1 : 0); i < int(length); i++) - { + bool processingZeroes = true; + + for (int i = (has_minus ? 1 : 0); i < int(length); i++) { if (buffer[i] == '0' && processingZeroes) zeroes++; else @@ -2532,219 +3695,407 @@ static inline long long stringToLongLong(InStream& in, const char* buffer) if (buffer[i] < '0' || buffer[i] > '9') in.quit(_pe, ("Expected integer, but \"" + __testlib_part(buffer) + "\" found").c_str()); - retval = retval * 10 + (buffer[i] - '0'); } - if (retval < 0) + long long int result; + try { + result = std::stoll(buffer); + } catch (const std::exception&) { in.quit(_pe, ("Expected integer, but \"" + __testlib_part(buffer) + "\" found").c_str()); - - if ((zeroes > 0 && (retval != 0 || minus)) || zeroes > 1) + } catch (...) { in.quit(_pe, ("Expected integer, but \"" + __testlib_part(buffer) + "\" found").c_str()); + } - retval = (minus ? -retval : +retval); + if ((zeroes > 0 && (result != 0 || has_minus)) || zeroes > 1) + in.quit(_pe, ("Expected integer, but \"" + __testlib_part(buffer) + "\" found").c_str()); - if (length < 19) - return retval; + return result; +} - if (equals(retval, buffer)) - return retval; - else - in.quit(_pe, ("Expected int64, but \"" + __testlib_part(buffer) + "\" found").c_str()); +static inline long long stringToLongLong(InStream &in, const std::string& buffer) { + for (size_t i = 0; i < buffer.length(); i++) + if (buffer[i] == '\0') + in.quit(_pe, ("Expected integer, but \"" + __testlib_part(buffer) + "\" found (it contains \\0)").c_str()); + return stringToLongLong(in, buffer.c_str()); } -int InStream::readInteger() -{ +static inline unsigned long long stringToUnsignedLongLong(InStream &in, const char *buffer) { + size_t length = strlen(buffer); + + if (length == 0 || length > 20) + in.quit(_pe, ("Expected unsigned integer, but \"" + __testlib_part(buffer) + "\" found").c_str()); + if (length > 1 && buffer[0] == '0') + in.quit(_pe, ("Expected unsigned integer, but \"" + __testlib_part(buffer) + "\" found").c_str()); + + for (int i = 0; i < int(length); i++) { + if (buffer[i] < '0' || buffer[i] > '9') + in.quit(_pe, ("Expected unsigned integer, but \"" + __testlib_part(buffer) + "\" found").c_str()); + } + + unsigned long long result; + try { + result = std::stoull(buffer); + } catch (const std::exception&) { + in.quit(_pe, ("Expected unsigned integer, but \"" + __testlib_part(buffer) + "\" found").c_str()); + } catch (...) { + in.quit(_pe, ("Expected unsigned integer, but \"" + __testlib_part(buffer) + "\" found").c_str()); + } + + return result; +} + +static inline long long stringToUnsignedLongLong(InStream &in, const std::string& buffer) { + for (size_t i = 0; i < buffer.length(); i++) + if (buffer[i] == '\0') + in.quit(_pe, ("Expected unsigned integer, but \"" + __testlib_part(buffer) + "\" found (it contains \\0)").c_str()); + return stringToUnsignedLongLong(in, buffer.c_str()); +} + +int InStream::readInteger() { if (!strict && seekEof()) quit(_unexpected_eof, "Unexpected end of file - int32 expected"); readWordTo(_tmpReadToken); - - long long value = stringToLongLong(*this, _tmpReadToken.c_str()); + + long long value = stringToLongLong(*this, _tmpReadToken); if (value < INT_MIN || value > INT_MAX) quit(_pe, ("Expected int32, but \"" + __testlib_part(_tmpReadToken) + "\" found").c_str()); - + return int(value); } -long long InStream::readLong() -{ +long long InStream::readLong() { if (!strict && seekEof()) quit(_unexpected_eof, "Unexpected end of file - int64 expected"); readWordTo(_tmpReadToken); - return stringToLongLong(*this, _tmpReadToken.c_str()); + + return stringToLongLong(*this, _tmpReadToken); } -long long InStream::readLong(long long minv, long long maxv, const std::string& variableName) -{ - long long result = readLong(); +unsigned long long InStream::readUnsignedLong() { + if (!strict && seekEof()) + quit(_unexpected_eof, "Unexpected end of file - int64 expected"); - if (result < minv || result > maxv) - { - if (variableName.empty()) - quit(_wa, ("Integer " + vtos(result) + " violates the range [" + vtos(minv) + ", " + vtos(maxv) + "]").c_str()); - else - quit(_wa, ("Integer parameter [name=" + variableName + "] equals to " + vtos(result) + ", violates the range [" + vtos(minv) + ", " + vtos(maxv) + "]").c_str()); - } + readWordTo(_tmpReadToken); - return result; + return stringToUnsignedLongLong(*this, _tmpReadToken); } -int InStream::readInt() -{ - return readInteger(); -} +long long InStream::readLong(long long minv, long long maxv, const std::string &variableName) { + long long result = readLong(); -int InStream::readInt(int minv, int maxv, const std::string &variableName) -{ + if (result < minv || result > maxv) { + if (readManyIteration == NO_INDEX) { + if (variableName.empty()) + quit(_wa, ("Integer " + vtos(result) + " violates the range [" + toHumanReadableString(minv) + ", " + toHumanReadableString(maxv) + + "]").c_str()); + else + quit(_wa, ("Integer parameter [name=" + std::string(variableName) + "] equals to " + vtos(result) + + ", violates the range [" + toHumanReadableString(minv) + ", " + toHumanReadableString(maxv) + "]").c_str()); + } else { + if (variableName.empty()) + quit(_wa, ("Integer element [index=" + vtos(readManyIteration) + "] equals to " + vtos(result) + + ", violates the range [" + toHumanReadableString(minv) + ", " + toHumanReadableString(maxv) + "]").c_str()); + else + quit(_wa, + ("Integer element " + std::string(variableName) + "[" + vtos(readManyIteration) + "] equals to " + + vtos(result) + ", violates the range [" + toHumanReadableString(minv) + ", " + toHumanReadableString(maxv) + "]").c_str()); + } + } + + if (strict && !variableName.empty()) + validator.addBoundsHit(variableName, ValidatorBoundsHit(minv == result, maxv == result)); + + return result; +} + +std::vector +InStream::readLongs(int size, long long minv, long long maxv, const std::string &variablesName, int indexBase) { + __testlib_readMany(readLongs, readLong(minv, maxv, variablesName), long long, true) +} + +std::vector InStream::readLongs(int size, int indexBase) { + __testlib_readMany(readLongs, readLong(), long long, true) +} + +unsigned long long +InStream::readUnsignedLong(unsigned long long minv, unsigned long long maxv, const std::string &variableName) { + unsigned long long result = readUnsignedLong(); + + if (result < minv || result > maxv) { + if (readManyIteration == NO_INDEX) { + if (variableName.empty()) + quit(_wa, + ("Unsigned integer " + vtos(result) + " violates the range [" + toHumanReadableString(minv) + ", " + toHumanReadableString(maxv) + + "]").c_str()); + else + quit(_wa, + ("Unsigned integer parameter [name=" + std::string(variableName) + "] equals to " + vtos(result) + + ", violates the range [" + toHumanReadableString(minv) + ", " + toHumanReadableString(maxv) + "]").c_str()); + } else { + if (variableName.empty()) + quit(_wa, + ("Unsigned integer element [index=" + vtos(readManyIteration) + "] equals to " + vtos(result) + + ", violates the range [" + toHumanReadableString(minv) + ", " + toHumanReadableString(maxv) + "]").c_str()); + else + quit(_wa, ("Unsigned integer element " + std::string(variableName) + "[" + vtos(readManyIteration) + + "] equals to " + vtos(result) + ", violates the range [" + toHumanReadableString(minv) + ", " + toHumanReadableString(maxv) + + "]").c_str()); + } + } + + if (strict && !variableName.empty()) + validator.addBoundsHit(variableName, ValidatorBoundsHit(minv == result, maxv == result)); + + return result; +} + +std::vector InStream::readUnsignedLongs(int size, unsigned long long minv, unsigned long long maxv, + const std::string &variablesName, int indexBase) { + __testlib_readMany(readUnsignedLongs, readUnsignedLong(minv, maxv, variablesName), unsigned long long, true) +} + +std::vector InStream::readUnsignedLongs(int size, int indexBase) { + __testlib_readMany(readUnsignedLongs, readUnsignedLong(), unsigned long long, true) +} + +unsigned long long +InStream::readLong(unsigned long long minv, unsigned long long maxv, const std::string &variableName) { + return readUnsignedLong(minv, maxv, variableName); +} + +int InStream::readInt() { + return readInteger(); +} + +int InStream::readInt(int minv, int maxv, const std::string &variableName) { int result = readInt(); - if (result < minv || result > maxv) - { - if (variableName.empty()) - quit(_wa, ("Integer " + vtos(result) + " violates the range [" + vtos(minv) + ", " + vtos(maxv) + "]").c_str()); - else - quit(_wa, ("Integer parameter [name=" + std::string(variableName) + "] equals to " + vtos(result) + ", violates the range [" + vtos(minv) + ", " + vtos(maxv) + "]").c_str()); - } + if (result < minv || result > maxv) { + if (readManyIteration == NO_INDEX) { + if (variableName.empty()) + quit(_wa, ("Integer " + vtos(result) + " violates the range [" + toHumanReadableString(minv) + ", " + toHumanReadableString(maxv) + + "]").c_str()); + else + quit(_wa, ("Integer parameter [name=" + std::string(variableName) + "] equals to " + vtos(result) + + ", violates the range [" + toHumanReadableString(minv) + ", " + toHumanReadableString(maxv) + "]").c_str()); + } else { + if (variableName.empty()) + quit(_wa, ("Integer element [index=" + vtos(readManyIteration) + "] equals to " + vtos(result) + + ", violates the range [" + toHumanReadableString(minv) + ", " + toHumanReadableString(maxv) + "]").c_str()); + else + quit(_wa, + ("Integer element " + std::string(variableName) + "[" + vtos(readManyIteration) + "] equals to " + + vtos(result) + ", violates the range [" + toHumanReadableString(minv) + ", " + toHumanReadableString(maxv) + "]").c_str()); + } + } + + if (strict && !variableName.empty()) + validator.addBoundsHit(variableName, ValidatorBoundsHit(minv == result, maxv == result)); return result; } -int InStream::readInteger(int minv, int maxv, const std::string& variableName) -{ - return readInt(minv, maxv, variableName.c_str()); +int InStream::readInteger(int minv, int maxv, const std::string &variableName) { + return readInt(minv, maxv, variableName); } -double InStream::readReal() -{ +std::vector InStream::readInts(int size, int minv, int maxv, const std::string &variablesName, int indexBase) { + __testlib_readMany(readInts, readInt(minv, maxv, variablesName), int, true) +} + +std::vector InStream::readInts(int size, int indexBase) { + __testlib_readMany(readInts, readInt(), int, true) +} + +std::vector InStream::readIntegers(int size, int minv, int maxv, const std::string &variablesName, int indexBase) { + __testlib_readMany(readIntegers, readInt(minv, maxv, variablesName), int, true) +} + +std::vector InStream::readIntegers(int size, int indexBase) { + __testlib_readMany(readIntegers, readInt(), int, true) +} + +double InStream::readReal() { if (!strict && seekEof()) quit(_unexpected_eof, "Unexpected end of file - double expected"); - return stringToDouble(*this, readWord().c_str()); + return stringToDouble(*this, readWord()); } -double InStream::readDouble() -{ +double InStream::readDouble() { return readReal(); } -double InStream::readReal(double minv, double maxv, const std::string& variableName) -{ +double InStream::readReal(double minv, double maxv, const std::string &variableName) { double result = readReal(); - if (result < minv || result > maxv) - { - if (variableName.empty()) - quit(_wa, ("Double " + vtos(result) + " violates the range [" + vtos(minv) + ", " + vtos(maxv) + "]").c_str()); - else - quit(_wa, ("Double parameter [name=" + variableName + "] equals to " + vtos(result) + ", violates the range [" + vtos(minv) + ", " + vtos(maxv) + "]").c_str()); + if (result < minv || result > maxv) { + if (readManyIteration == NO_INDEX) { + if (variableName.empty()) + quit(_wa, ("Double " + vtos(result) + " violates the range [" + vtos(minv) + ", " + vtos(maxv) + + "]").c_str()); + else + quit(_wa, ("Double parameter [name=" + std::string(variableName) + "] equals to " + vtos(result) + + ", violates the range [" + vtos(minv) + ", " + vtos(maxv) + "]").c_str()); + } else { + if (variableName.empty()) + quit(_wa, ("Double element [index=" + vtos(readManyIteration) + "] equals to " + vtos(result) + + ", violates the range [" + vtos(minv) + ", " + vtos(maxv) + "]").c_str()); + else + quit(_wa, + ("Double element " + std::string(variableName) + "[" + vtos(readManyIteration) + "] equals to " + + vtos(result) + ", violates the range [" + vtos(minv) + ", " + vtos(maxv) + "]").c_str()); + } } + if (strict && !variableName.empty()) + validator.addBoundsHit(variableName, ValidatorBoundsHit( + doubleDelta(minv, result) < ValidatorBoundsHit::EPS, + doubleDelta(maxv, result) < ValidatorBoundsHit::EPS + )); + return result; } -double InStream::readDouble(double minv, double maxv, const std::string& variableName) -{ +std::vector +InStream::readReals(int size, double minv, double maxv, const std::string &variablesName, int indexBase) { + __testlib_readMany(readReals, readReal(minv, maxv, variablesName), double, true) +} + +std::vector InStream::readReals(int size, int indexBase) { + __testlib_readMany(readReals, readReal(), double, true) +} + +double InStream::readDouble(double minv, double maxv, const std::string &variableName) { return readReal(minv, maxv, variableName); -} +} + +std::vector +InStream::readDoubles(int size, double minv, double maxv, const std::string &variablesName, int indexBase) { + __testlib_readMany(readDoubles, readDouble(minv, maxv, variablesName), double, true) +} + +std::vector InStream::readDoubles(int size, int indexBase) { + __testlib_readMany(readDoubles, readDouble(), double, true) +} double InStream::readStrictReal(double minv, double maxv, - int minAfterPointDigitCount, int maxAfterPointDigitCount, - const std::string& variableName) -{ + int minAfterPointDigitCount, int maxAfterPointDigitCount, + const std::string &variableName) { if (!strict && seekEof()) quit(_unexpected_eof, "Unexpected end of file - strict double expected"); - double result = stringToStrictDouble(*this, readWord().c_str(), - minAfterPointDigitCount, maxAfterPointDigitCount); + double result = stringToStrictDouble(*this, readWord(), minAfterPointDigitCount, maxAfterPointDigitCount); - if (result < minv || result > maxv) - { - if (variableName.empty()) - quit(_wa, ("Strict double " + vtos(result) + " violates the range [" + vtos(minv) + ", " + vtos(maxv) + "]").c_str()); - else - quit(_wa, ("Strict double parameter [name=" + variableName + "] equals to " + vtos(result) + ", violates the range [" + vtos(minv) + ", " + vtos(maxv) + "]").c_str()); + if (result < minv || result > maxv) { + if (readManyIteration == NO_INDEX) { + if (variableName.empty()) + quit(_wa, ("Strict double " + vtos(result) + " violates the range [" + vtos(minv) + ", " + vtos(maxv) + + "]").c_str()); + else + quit(_wa, + ("Strict double parameter [name=" + std::string(variableName) + "] equals to " + vtos(result) + + ", violates the range [" + vtos(minv) + ", " + vtos(maxv) + "]").c_str()); + } else { + if (variableName.empty()) + quit(_wa, ("Strict double element [index=" + vtos(readManyIteration) + "] equals to " + vtos(result) + + ", violates the range [" + vtos(minv) + ", " + vtos(maxv) + "]").c_str()); + else + quit(_wa, ("Strict double element " + std::string(variableName) + "[" + vtos(readManyIteration) + + "] equals to " + vtos(result) + ", violates the range [" + vtos(minv) + ", " + vtos(maxv) + + "]").c_str()); + } } + if (strict && !variableName.empty()) + validator.addBoundsHit(variableName, ValidatorBoundsHit( + doubleDelta(minv, result) < ValidatorBoundsHit::EPS, + doubleDelta(maxv, result) < ValidatorBoundsHit::EPS + )); + return result; } +std::vector InStream::readStrictReals(int size, double minv, double maxv, + int minAfterPointDigitCount, int maxAfterPointDigitCount, + const std::string &variablesName, int indexBase) { + __testlib_readMany(readStrictReals, + readStrictReal(minv, maxv, minAfterPointDigitCount, maxAfterPointDigitCount, variablesName), + double, true) +} + double InStream::readStrictDouble(double minv, double maxv, - int minAfterPointDigitCount, int maxAfterPointDigitCount, - const std::string& variableName) -{ + int minAfterPointDigitCount, int maxAfterPointDigitCount, + const std::string &variableName) { return readStrictReal(minv, maxv, - minAfterPointDigitCount, maxAfterPointDigitCount, - variableName); + minAfterPointDigitCount, maxAfterPointDigitCount, + variableName); } -bool InStream::eof() -{ - if (!strict && NULL == file) +std::vector InStream::readStrictDoubles(int size, double minv, double maxv, + int minAfterPointDigitCount, int maxAfterPointDigitCount, + const std::string &variablesName, int indexBase) { + __testlib_readMany(readStrictDoubles, + readStrictDouble(minv, maxv, minAfterPointDigitCount, maxAfterPointDigitCount, variablesName), + double, true) +} + +bool InStream::eof() { + if (!strict && NULL == reader) return true; return reader->eof(); } -bool InStream::seekEof() -{ - if (NULL == file) +bool InStream::seekEof() { + if (!strict && NULL == reader) return true; skipBlanks(); return eof(); } -bool InStream::eoln() -{ - if (!strict && NULL == file) +bool InStream::eoln() { + if (!strict && NULL == reader) return true; int c = reader->nextChar(); - if (!strict) - { - if (c == EOF) + if (!strict) { + if (c == EOFC) return true; - if (c == CR) - { + if (c == CR) { c = reader->nextChar(); - if (c != LF) - { + if (c != LF) { reader->unreadChar(c); reader->unreadChar(CR); return false; - } - else + } else return true; } - + if (c == LF) return true; reader->unreadChar(c); return false; - } - else - { + } else { bool returnCr = false; -#ifdef CR_MUST_IN_EOL - if (c != CR) - { +#if (defined(ON_WINDOWS) && !defined(FOR_LINUX)) || defined(FOR_WINDOWS) + if (c != CR) { reader->unreadChar(c); return false; - } - else - { + } else { if (!returnCr) returnCr = true; c = reader->nextChar(); } -#endif - if (c != LF) - { +#endif + if (c != LF) { reader->unreadChar(c); if (returnCr) reader->unreadChar(CR); @@ -2755,14 +4106,14 @@ bool InStream::eoln() } } -void InStream::readEoln() -{ +void InStream::readEoln() { + lastLine = reader->getLine(); if (!eoln()) quit(_pe, "Expected EOLN"); } -void InStream::readEof() -{ +void InStream::readEof() { + lastLine = reader->getLine(); if (!eof()) quit(_pe, "Expected EOF"); @@ -2770,45 +4121,44 @@ void InStream::readEof() testlibFinalizeGuard.readEofCount++; } -bool InStream::seekEoln() -{ - if (NULL == file) +bool InStream::seekEoln() { + if (!strict && NULL == reader) return true; - + int cur; - do - { + do { cur = reader->nextChar(); - } - while (cur == SPACE || cur == TAB); + } while (cur == SPACE || cur == TAB); reader->unreadChar(cur); return eoln(); } -void InStream::nextLine() -{ +void InStream::nextLine() { readLine(); } -void InStream::readStringTo(std::string& result) -{ - if (NULL == file) +void InStream::readStringTo(std::string &result) { + if (NULL == reader) quit(_pe, "Expected line"); result.clear(); - int cur; - for (;;) - { - cur = reader->curChar(); + for (;;) { + int cur = reader->curChar(); - if (isEoln(cur)) + if (cur == LF || cur == EOFC) break; - if (cur == EOF) - break; + if (cur == CR) { + cur = reader->nextChar(); + if (reader->curChar() == LF) { + reader->unreadChar(cur); + break; + } + } + lastLine = reader->getLine(); result += char(reader->nextChar()); } @@ -2818,97 +4168,136 @@ void InStream::readStringTo(std::string& result) eoln(); } -std::string InStream::readString() -{ +std::string InStream::readString() { readStringTo(_tmpReadToken); return _tmpReadToken; } -void InStream::readStringTo(std::string& result, const pattern& p, const std::string& variableName) -{ +std::vector InStream::readStrings(int size, int indexBase) { + __testlib_readMany(readStrings, readString(), std::string, false) +} + +void InStream::readStringTo(std::string &result, const pattern &p, const std::string &variableName) { readStringTo(result); - if (!p.matches(result)) - { - if (variableName.empty()) - quit(_wa, ("Line \"" + __testlib_part(result) + "\" doesn't correspond to pattern \"" + p.src() + "\"").c_str()); - else - quit(_wa, ("Line [name=" + variableName + "] equals to \"" + __testlib_part(result) + "\", doesn't correspond to pattern \"" + p.src() + "\"").c_str()); + if (!p.matches(result)) { + if (readManyIteration == NO_INDEX) { + if (variableName.empty()) + quit(_wa, ("Line \"" + __testlib_part(result) + "\" doesn't correspond to pattern \"" + p.src() + + "\"").c_str()); + else + quit(_wa, ("Line [name=" + variableName + "] equals to \"" + __testlib_part(result) + + "\", doesn't correspond to pattern \"" + p.src() + "\"").c_str()); + } else { + if (variableName.empty()) + quit(_wa, + ("Line element [index=" + vtos(readManyIteration) + "] equals to \"" + __testlib_part(result) + + "\" doesn't correspond to pattern \"" + p.src() + "\"").c_str()); + else + quit(_wa, + ("Line element " + std::string(variableName) + "[" + vtos(readManyIteration) + "] equals to \"" + + __testlib_part(result) + "\", doesn't correspond to pattern \"" + p.src() + "\"").c_str()); + } } } -void InStream::readStringTo(std::string& result, const std::string& ptrn, const std::string& variableName) -{ +void InStream::readStringTo(std::string &result, const std::string &ptrn, const std::string &variableName) { readStringTo(result, pattern(ptrn), variableName); } -std::string InStream::readString(const pattern& p, const std::string& variableName) -{ +std::string InStream::readString(const pattern &p, const std::string &variableName) { readStringTo(_tmpReadToken, p, variableName); return _tmpReadToken; } -std::string InStream::readString(const std::string& ptrn, const std::string& variableName) -{ +std::vector +InStream::readStrings(int size, const pattern &p, const std::string &variablesName, int indexBase) { + __testlib_readMany(readStrings, readString(p, variablesName), std::string, false) +} + +std::string InStream::readString(const std::string &ptrn, const std::string &variableName) { readStringTo(_tmpReadToken, ptrn, variableName); return _tmpReadToken; } -void InStream::readLineTo(std::string& result) -{ +std::vector +InStream::readStrings(int size, const std::string &ptrn, const std::string &variablesName, int indexBase) { + pattern p(ptrn); + __testlib_readMany(readStrings, readString(p, variablesName), std::string, false) +} + +void InStream::readLineTo(std::string &result) { readStringTo(result); } -std::string InStream::readLine() -{ +std::string InStream::readLine() { return readString(); } -void InStream::readLineTo(std::string& result, const pattern& p, const std::string& variableName) -{ +std::vector InStream::readLines(int size, int indexBase) { + __testlib_readMany(readLines, readString(), std::string, false) +} + +void InStream::readLineTo(std::string &result, const pattern &p, const std::string &variableName) { readStringTo(result, p, variableName); } -void InStream::readLineTo(std::string& result, const std::string& ptrn, const std::string& variableName) -{ +void InStream::readLineTo(std::string &result, const std::string &ptrn, const std::string &variableName) { readStringTo(result, ptrn, variableName); } -std::string InStream::readLine(const pattern& p, const std::string& variableName) -{ +std::string InStream::readLine(const pattern &p, const std::string &variableName) { return readString(p, variableName); } -std::string InStream::readLine(const std::string& ptrn, const std::string& variableName) -{ +std::vector +InStream::readLines(int size, const pattern &p, const std::string &variablesName, int indexBase) { + __testlib_readMany(readLines, readString(p, variablesName), std::string, false) +} + +std::string InStream::readLine(const std::string &ptrn, const std::string &variableName) { return readString(ptrn, variableName); } -void InStream::close() -{ - if (opened) - fclose(file); - - if (NULL != reader) - { +std::vector +InStream::readLines(int size, const std::string &ptrn, const std::string &variablesName, int indexBase) { + pattern p(ptrn); + __testlib_readMany(readLines, readString(p, variablesName), std::string, false) +} + +#ifdef __GNUC__ +__attribute__ ((format (printf, 3, 4))) +#endif +void InStream::ensuref(bool cond, const char *format, ...) { + if (!cond) { + FMT_TO_RESULT(format, format, message); + this->__testlib_ensure(cond, message); + } +} + +void InStream::__testlib_ensure(bool cond, std::string message) { + if (!cond) + this->quit(_wa, message.c_str()); +} + +void InStream::close() { + if (NULL != reader) { + reader->close(); delete reader; reader = NULL; } - + opened = false; } -NORETURN void quit(TResult result, const std::string& msg) -{ +NORETURN void quit(TResult result, const std::string &msg) { ouf.quit(result, msg.c_str()); } -NORETURN void quit(TResult result, const char* msg) -{ +NORETURN void quit(TResult result, const char *msg) { ouf.quit(result, msg); } -NORETURN void __testlib_quitp(double points, const char* message) -{ +NORETURN void __testlib_quitp(double points, const char *message) { __testlib_points = points; std::string stringPoints = removeDoubleTrailingZeroes(format("%.10f", points)); @@ -2921,27 +4310,49 @@ NORETURN void __testlib_quitp(double points, const char* message) quit(_points, quitMessage.c_str()); } -NORETURN void quitp(float points, const std::string& message = "") -{ +NORETURN void __testlib_quitp(int points, const char *message) { + __testlib_points = points; + std::string stringPoints = format("%d", points); + + std::string quitMessage; + if (NULL == message || 0 == strlen(message)) + quitMessage = stringPoints; + else + quitMessage = stringPoints + " " + message; + + quit(_points, quitMessage.c_str()); +} + +NORETURN void quitp(float points, const std::string &message = "") { __testlib_quitp(double(points), message.c_str()); } -NORETURN void quitp(double points, const std::string& message = "") -{ +NORETURN void quitp(double points, const std::string &message = "") { __testlib_quitp(points, message.c_str()); } -NORETURN void quitp(long double points, const std::string& message = "") -{ +NORETURN void quitp(long double points, const std::string &message = "") { __testlib_quitp(double(points), message.c_str()); } +NORETURN void quitp(int points, const std::string &message = "") { + __testlib_quitp(points, message.c_str()); +} + +NORETURN void quitpi(const std::string &points_info, const std::string &message = "") { + if (points_info.find(' ') != std::string::npos) + quit(_fail, "Parameter 'points_info' can't contain spaces"); + if (message.empty()) + quit(_points, ("points_info=" + points_info).c_str()); + else + quit(_points, ("points_info=" + points_info + " " + message).c_str()); +} + template #ifdef __GNUC__ __attribute__ ((format (printf, 2, 3))) #endif -NORETURN void quitp(F points, const char* format, ...) -{ +NORETURN void quitp(F points, const char *format, ...) { FMT_TO_RESULT(format, format, message); quitp(points, message); } @@ -2949,8 +4360,7 @@ NORETURN void quitp(F points, const char* format, ...) #ifdef __GNUC__ __attribute__ ((format (printf, 2, 3))) #endif -NORETURN void quitf(TResult result, const char* format, ...) -{ +NORETURN void quitf(TResult result, const char *format, ...) { FMT_TO_RESULT(format, format, message); quit(result, message); } @@ -2958,39 +4368,35 @@ NORETURN void quitf(TResult result, const char* format, ...) #ifdef __GNUC__ __attribute__ ((format (printf, 3, 4))) #endif -void quitif(bool condition, TResult result, const char* format, ...) -{ - if (condition) - { +void quitif(bool condition, TResult result, const char *format, ...) { + if (condition) { FMT_TO_RESULT(format, format, message); quit(result, message); } } -NORETURN void __testlib_help() -{ +NORETURN void __testlib_help() { InStream::textColor(InStream::LightCyan); - std::fprintf(stderr, "TESTLIB %s, http://code.google.com/p/testlib/ ", VERSION); - std::fprintf(stderr, "by Mike Mirzayanov, copyright(c) 2005-2013\n"); + std::fprintf(stderr, "TESTLIB %s, https://github.com/MikeMirzayanov/testlib/ ", VERSION); + std::fprintf(stderr, "by Mike Mirzayanov, copyright(c) 2005-2020\n"); std::fprintf(stderr, "Checker name: \"%s\"\n", checkerName.c_str()); InStream::textColor(InStream::LightGray); std::fprintf(stderr, "\n"); std::fprintf(stderr, "Latest features: \n"); - for (size_t i = 0; i < sizeof(latestFeatures) / sizeof(char*); i++) - { + for (size_t i = 0; i < sizeof(latestFeatures) / sizeof(char *); i++) { std::fprintf(stderr, "*) %s\n", latestFeatures[i]); } std::fprintf(stderr, "\n"); std::fprintf(stderr, "Program must be run with the following arguments: \n"); - std::fprintf(stderr, " [ [<-appes>]]\n\n"); + std::fprintf(stderr, " [--testset testset] [--group group] [ [<-appes>]]\n\n"); + __testlib_exitCode = FAIL_EXIT_CODE; std::exit(FAIL_EXIT_CODE); } -static void __testlib_ensuresPreconditions() -{ +static void __testlib_ensuresPreconditions() { // testlib assumes: sizeof(int) = 4. __TESTLIB_STATIC_ASSERT(sizeof(int) == 4); @@ -3000,6 +4406,9 @@ static void __testlib_ensuresPreconditions() // testlib assumes: sizeof(long long) = 8. __TESTLIB_STATIC_ASSERT(sizeof(long long) == 8); + // testlib assumes: sizeof(double) = 8. + __TESTLIB_STATIC_ASSERT(sizeof(double) == 8); + // testlib assumes: no -ffast-math. if (!__testlib_isNaN(+__testlib_nan())) quit(_fail, "Function __testlib_isNaN is not working correctly: possible reason is '-ffast-math'"); @@ -3007,17 +4416,49 @@ static void __testlib_ensuresPreconditions() quit(_fail, "Function __testlib_isNaN is not working correctly: possible reason is '-ffast-math'"); } -void registerGen(int argc, char* argv[], int randomGeneratorVersion) -{ +std::string __testlib_testset; + +std::string getTestset() { + return __testlib_testset; +} + +std::string __testlib_group; + +std::string getGroup() { + return __testlib_group; +} + +static void __testlib_set_testset_and_group(int argc, char* argv[]) { + for (int i = 1; i < argc; i++) { + if (!strcmp("--testset", argv[i])) { + if (i + 1 < argc && strlen(argv[i + 1]) > 0) + __testlib_testset = argv[++i]; + else + quit(_fail, std::string("Expected non-empty testset after --testset command line parameter")); + } else if (!strcmp("--group", argv[i])) { + if (i + 1 < argc) + __testlib_group = argv[++i]; + else + quit(_fail, std::string("Expected group after --group command line parameter")); + } + } +} + +void registerGen(int argc, char *argv[], int randomGeneratorVersion) { if (randomGeneratorVersion < 0 || randomGeneratorVersion > 1) quitf(_fail, "Random generator version is expected to be 0 or 1."); random_t::version = randomGeneratorVersion; __testlib_ensuresPreconditions(); + TestlibFinalizeGuard::registered = true; testlibMode = _generator; __testlib_set_binary(stdin); rnd.setSeed(argc, argv); + +#if __cplusplus > 199711L || defined(_MSC_VER) + prepareOpts(argc, argv); +#endif } #ifdef USE_RND_AS_BEFORE_087 @@ -3026,148 +4467,304 @@ void registerGen(int argc, char* argv[]) registerGen(argc, argv, 0); } #else -void registerGen(int argc, char* argv[]) -{ +#ifdef __GNUC__ +#if (__GNUC__ > 4) || ((__GNUC__ == 4) && (__GNUC_MINOR__ > 4)) +__attribute__ ((deprecated("Use registerGen(argc, argv, 0) or registerGen(argc, argv, 1)." +" The third parameter stands for the random generator version." +" If you are trying to compile old generator use macro -DUSE_RND_AS_BEFORE_087 or registerGen(argc, argv, 0)." +" Version 1 has been released on Spring, 2013. Use it to write new generators."))) +#else +__attribute__ ((deprecated)) +#endif +#endif +#ifdef _MSC_VER +__declspec(deprecated("Use registerGen(argc, argv, 0) or registerGen(argc, argv, 1)." + " The third parameter stands for the random generator version." + " If you are trying to compile old generator use macro -DUSE_RND_AS_BEFORE_087 or registerGen(argc, argv, 0)." + " Version 1 has been released on Spring, 2013. Use it to write new generators.")) +#endif +void registerGen(int argc, char *argv[]) { std::fprintf(stderr, "Use registerGen(argc, argv, 0) or registerGen(argc, argv, 1)." - " The third parameter stands for the random generator version." - " If you are trying to compile old generator use macro -DUSE_RND_AS_BEFORE_087 or registerGen(argc, argv, 0)." - " Version 1 has been released on Spring, 2013. Use it to write new generators.\n\n"); + " The third parameter stands for the random generator version." + " If you are trying to compile old generator use macro -DUSE_RND_AS_BEFORE_087 or registerGen(argc, argv, 0)." + " Version 1 has been released on Spring, 2013. Use it to write new generators.\n\n"); registerGen(argc, argv, 0); } #endif -void registerInteraction(int argc, char* argv[]) -{ +void setAppesModeEncoding(std::string appesModeEncoding) { + static const char* const ENCODINGS[] = {"ascii", "utf-7", "utf-8", "utf-16", "utf-16le", "utf-16be", "utf-32", "utf-32le", "utf-32be", "iso-8859-1", +"iso-8859-2", "iso-8859-3", "iso-8859-4", "iso-8859-5", "iso-8859-6", "iso-8859-7", "iso-8859-8", "iso-8859-9", "iso-8859-10", "iso-8859-11", +"iso-8859-13", "iso-8859-14", "iso-8859-15", "iso-8859-16", "windows-1250", "windows-1251", "windows-1252", "windows-1253", "windows-1254", "windows-1255", +"windows-1256", "windows-1257", "windows-1258", "gb2312", "gbk", "gb18030", "big5", "shift-jis", "euc-jp", "euc-kr", +"euc-cn", "euc-tw", "koi8-r", "koi8-u", "tis-620", "ibm437", "ibm850", "ibm852", "ibm855", "ibm857", +"ibm860", "ibm861", "ibm862", "ibm863", "ibm865", "ibm866", "ibm869", "macroman", "maccentraleurope", "maciceland", +"maccroatian", "macromania", "maccyrillic", "macukraine", "macgreek", "macturkish", "machebrew", "macarabic", "macthai", "hz-gb-2312", +"iso-2022-jp", "iso-2022-kr", "iso-2022-cn", "armscii-8", "tscii", "iscii", "viscii", "geostd8", "cp949", "cp874", +"cp1006", "cp775", "cp858", "cp737", "cp853", "cp856", "cp922", "cp1046", "cp1125", "cp1131", +"ptcp154", "koi8-t", "koi8-ru", "mulelao-1", "cp1133", "iso-ir-166", "tcvn", "iso-ir-14", "iso-ir-87", "iso-ir-159"}; + + appesModeEncoding = lowerCase(appesModeEncoding); + bool valid = false; + for (size_t i = 0; i < sizeof(ENCODINGS) / sizeof(ENCODINGS[0]); i++) + if (appesModeEncoding == ENCODINGS[i]) { + valid = true; + break; + } + if (!valid) + quit(_fail, "Unexpected encoding for setAppesModeEncoding(encoding)"); + ::appesModeEncoding = appesModeEncoding; +} + +void registerInteraction(int argc, char *argv[]) { __testlib_ensuresPreconditions(); + __testlib_set_testset_and_group(argc, argv); + TestlibFinalizeGuard::registered = true; testlibMode = _interactor; __testlib_set_binary(stdin); if (argc > 1 && !strcmp("--help", argv[1])) __testlib_help(); - - if (argc < 3 || argc > 6) - { + + if (argc < 3 || argc > 6) { quit(_fail, std::string("Program must be run with the following arguments: ") + - std::string(" [ [ [<-appes>]]]") + - "\nUse \"--help\" to get help information"); + std::string(" [ [ [<-appes>]]]") + + "\nUse \"--help\" to get help information"); } - if (argc <= 4) - { + if (argc <= 4) { resultName = ""; appesMode = false; } - if (argc == 5) - { +#ifndef EJUDGE + if (argc == 5) { resultName = argv[4]; appesMode = false; } - if (argc == 6) - { - if (strcmp("-APPES", argv[5]) && strcmp("-appes", argv[5])) - { + if (argc == 6) { + if (strcmp("-APPES", argv[5]) && strcmp("-appes", argv[5])) { quit(_fail, std::string("Program must be run with the following arguments: ") + " [ [<-appes>]]"); - } - else - { + } else { resultName = argv[4]; appesMode = true; } } +#endif inf.init(argv[1], _input); - if (strcmp(argv[2], "stdout") != 0) - { - tout.open(argv[2], std::ios_base::out); - if (tout.fail() || !tout.is_open()) - quit(_fail, std::string("Can not write to the test-output-file '") + argv[2] + std::string("'")); - } + tout.open(argv[2], std::ios_base::out); + if (tout.fail() || !tout.is_open()) + quit(_fail, std::string("Can not write to the test-output-file '") + argv[2] + std::string("'")); ouf.init(stdin, _output); - + if (argc >= 4) ans.init(argv[3], _answer); else ans.name = "unopened answer stream"; } -void registerValidation() -{ +void registerValidation() { __testlib_ensuresPreconditions(); + TestlibFinalizeGuard::registered = true; testlibMode = _validator; + __testlib_set_binary(stdin); + __testlib_set_binary(stdout); + __testlib_set_binary(stderr); inf.init(stdin, _input); inf.strict = true; } -void registerTestlibCmd(int argc, char* argv[]) -{ - __testlib_ensuresPreconditions(); +void registerValidation(int argc, char *argv[]) { + registerValidation(); + __testlib_set_testset_and_group(argc, argv); - testlibMode = _checker; - __testlib_set_binary(stdin); + validator.initialize(); + TestlibFinalizeGuard::registered = true; - if (argc > 1 && !strcmp("--help", argv[1])) - __testlib_help(); + std::string comment = "Validator must be run with the following arguments:" + " [--testset testset]" + " [--group group]" + " [--testOverviewLogFileName fileName]" + " [--testMarkupFileName fileName]" + " [--testCase testCase]" + " [--testCaseFileName fileName]" + ; - if (argc < 4 || argc > 6) - { - quit(_fail, std::string("Program must be run with the following arguments: ") + - std::string(" [ [<-appes>]]") + - "\nUse \"--help\" to get help information"); + for (int i = 1; i < argc; i++) { + if (!strcmp("--testset", argv[i])) { + if (i + 1 < argc && strlen(argv[i + 1]) > 0) + validator.setTestset(argv[++i]); + else + quit(_fail, comment); + } + if (!strcmp("--group", argv[i])) { + if (i + 1 < argc) + validator.setGroup(argv[++i]); + else + quit(_fail, comment); + } + if (!strcmp("--testOverviewLogFileName", argv[i])) { + if (i + 1 < argc) + validator.setTestOverviewLogFileName(argv[++i]); + else + quit(_fail, comment); + } + if (!strcmp("--testMarkupFileName", argv[i])) { + if (i + 1 < argc) + validator.setTestMarkupFileName(argv[++i]); + else + quit(_fail, comment); + } + if (!strcmp("--testCase", argv[i])) { + if (i + 1 < argc) { + long long testCase = stringToLongLong(inf, argv[++i]); + if (testCase < 1 || testCase >= __TESTLIB_MAX_TEST_CASE) + quit(_fail, format("Argument testCase should be between 1 and %d, but ", __TESTLIB_MAX_TEST_CASE) + + toString(testCase) + " found"); + validator.setTestCase(int(testCase)); + } else + quit(_fail, comment); + } + if (!strcmp("--testCaseFileName", argv[i])) { + if (i + 1 < argc) { + validator.setTestCaseFileName(argv[++i]); + } else + quit(_fail, comment); + } } +} - if (argc == 4) - { - resultName = ""; - appesMode = false; - } +void addFeature(const std::string &feature) { + if (testlibMode != _validator) + quit(_fail, "Features are supported in validators only."); + validator.addFeature(feature); +} - if (argc == 5) - { - resultName = argv[4]; +void feature(const std::string &feature) { + if (testlibMode != _validator) + quit(_fail, "Features are supported in validators only."); + validator.feature(feature); +} + +class Checker { +private: + bool _initialized; + std::string _testset; + std::string _group; + +public: + Checker() : _initialized(false), _testset("tests"), _group() { + } + + void initialize() { + _initialized = true; + } + + std::string testset() const { + if (!_initialized) + __testlib_fail("Checker should be initialized with registerTestlibCmd(argc, argv) instead of registerTestlibCmd() to support checker.testset()"); + return _testset; + } + + std::string group() const { + if (!_initialized) + __testlib_fail("Checker should be initialized with registerTestlibCmd(argc, argv) instead of registerTestlibCmd() to support checker.group()"); + return _group; + } + + void setTestset(const char *const testset) { + _testset = testset; + } + + void setGroup(const char *const group) { + _group = group; + } +} checker; + +void registerTestlibCmd(int argc, char *argv[]) { + __testlib_ensuresPreconditions(); + __testlib_set_testset_and_group(argc, argv); + TestlibFinalizeGuard::registered = true; + + testlibMode = _checker; + __testlib_set_binary(stdin); + + std::vector args(1, argv[0]); + checker.initialize(); + + for (int i = 1; i < argc; i++) { + if (!strcmp("--testset", argv[i])) { + if (i + 1 < argc && strlen(argv[i + 1]) > 0) + checker.setTestset(argv[++i]); + else + quit(_fail, std::string("Expected testset after --testset command line parameter")); + } else if (!strcmp("--group", argv[i])) { + if (i + 1 < argc) + checker.setGroup(argv[++i]); + else + quit(_fail, std::string("Expected group after --group command line parameter")); + } else + args.push_back(argv[i]); + } + + argc = int(args.size()); + if (argc > 1 && "--help" == args[1]) + __testlib_help(); + + if (argc < 4 || argc > 6) { + quit(_fail, std::string("Program must be run with the following arguments: ") + + std::string("[--testset testset] [--group group] [ [<-appes>]]") + + "\nUse \"--help\" to get help information"); + } + + if (argc == 4) { + resultName = ""; appesMode = false; } - if (argc == 6) - { - if (strcmp("-APPES", argv[5]) && strcmp("-appes", argv[5])) - { +#ifndef EJUDGE + if (argc == 5) { + resultName = args[4]; + appesMode = false; + } + + if (argc == 6) { + if ("-APPES" != args[5] && "-appes" != args[5]) { quit(_fail, std::string("Program must be run with the following arguments: ") + " [ [<-appes>]]"); - } - else - { - resultName = argv[4]; + } else { + resultName = args[4]; appesMode = true; } } +#endif - inf.init(argv[1], _input); - ouf.init(argv[2], _output); - ans.init(argv[3], _answer); + inf.init(args[1], _input); + ouf.init(args[2], _output); + ouf.skipBom(); + ans.init(args[3], _answer); } -void registerTestlib(int argc, ...) -{ - if (argc < 3 || argc > 5) +void registerTestlib(int argc, ...) { + if (argc < 3 || argc > 5) quit(_fail, std::string("Program must be run with the following arguments: ") + - " [ [<-appes>]]"); + " [ [<-appes>]]"); + + char **argv = new char *[argc + 1]; - char** argv = new char*[argc + 1]; - va_list ap; va_start(ap, argc); argv[0] = NULL; - for (int i = 0; i < argc; i++) - { + for (int i = 0; i < argc; i++) { argv[i + 1] = va_arg(ap, char*); } va_end(ap); @@ -3176,113 +4773,55 @@ void registerTestlib(int argc, ...) delete[] argv; } -#ifdef __GNUC__ -__attribute__((const)) -#endif -inline bool doubleCompare(double expected, double result, double MAX_DOUBLE_ERROR) -{ - if (__testlib_isNaN(expected)) - { - return __testlib_isNaN(result); - } - else - if (__testlib_isInfinite(expected)) - { - if (expected > 0) - { - return result > 0 && __testlib_isInfinite(result); - } - else - { - return result < 0 && __testlib_isInfinite(result); - } - } - else - if (__testlib_isNaN(result) || __testlib_isInfinite(result)) - { - return false; - } - else - if (__testlib_abs(result - expected) <= MAX_DOUBLE_ERROR + 1E-15) - { - return true; - } - else - { - double minv = __testlib_min(expected * (1.0 - MAX_DOUBLE_ERROR), - expected * (1.0 + MAX_DOUBLE_ERROR)); - double maxv = __testlib_max(expected * (1.0 - MAX_DOUBLE_ERROR), - expected * (1.0 + MAX_DOUBLE_ERROR)); - return result + 1E-15 >= minv && result <= maxv + 1E-15; - } -} - -#ifdef __GNUC__ -__attribute__((const)) -#endif -inline double doubleDelta(double expected, double result) -{ - double absolute = __testlib_abs(result - expected); - - if (__testlib_abs(expected) > 1E-9) - { - double relative = __testlib_abs(absolute / expected); - return __testlib_min(absolute, relative); - } - else - return absolute; -} - -static inline void __testlib_ensure(bool cond, const std::string& msg) -{ +static inline void __testlib_ensure(bool cond, const std::string &msg) { if (!cond) quit(_fail, msg.c_str()); } -static inline void __testlib_ensure(bool cond, const char* msg) -{ +#ifdef __GNUC__ +__attribute__((unused)) +#endif +static inline void __testlib_ensure(bool cond, const char *msg) { if (!cond) quit(_fail, msg); } #define ensure(cond) __testlib_ensure(cond, "Condition failed: \"" #cond "\"") +#define STRINGIZE_DETAIL(x) #x +#define STRINGIZE(x) STRINGIZE_DETAIL(x) +#define ensure_ext(cond) __testlib_ensure(cond, "Line " STRINGIZE(__LINE__) ": Condition failed: \"" #cond "\"") #ifdef __GNUC__ __attribute__ ((format (printf, 2, 3))) #endif -inline void ensuref(bool cond, const char* format, ...) -{ - if (!cond) - { +inline void ensuref(bool cond, const char *format, ...) { + if (!cond) { FMT_TO_RESULT(format, format, message); __testlib_ensure(cond, message); } } -NORETURN static void __testlib_fail(const std::string& message) -{ +NORETURN static void __testlib_fail(const std::string &message) { quitf(_fail, "%s", message.c_str()); } #ifdef __GNUC__ __attribute__ ((format (printf, 1, 2))) #endif -void setName(const char* format, ...) -{ +void setName(const char *format, ...) { FMT_TO_RESULT(format, format, name); checkerName = name; } -/* +/* * Do not use random_shuffle, because it will produce different result * for different C++ compilers. * * This implementation uses testlib random_t to produce random numbers, so * it is stable. - */ + */ template -void shuffle(_RandomAccessIter __first, _RandomAccessIter __last) -{ +void shuffle(_RandomAccessIter __first, _RandomAccessIter __last) { if (__first == __last) return; for (_RandomAccessIter __i = __first + 1; __i != __last; ++__i) std::iter_swap(__i, __first + rnd.next(int(__i - __first) + 1)); @@ -3290,8 +4829,10 @@ void shuffle(_RandomAccessIter __first, _RandomAccessIter __last) template -void random_shuffle(_RandomAccessIter , _RandomAccessIter ) -{ +#if defined(__GNUC__) && !defined(__clang__) +__attribute__ ((error("Don't use random_shuffle(), use shuffle() instead"))) +#endif +void random_shuffle(_RandomAccessIter, _RandomAccessIter) { quitf(_fail, "Don't use random_shuffle(), use shuffle() instead"); } @@ -3301,52 +4842,56 @@ void random_shuffle(_RandomAccessIter , _RandomAccessIter ) # define RAND_THROW_STATEMENT #endif +#if defined(__GNUC__) && !defined(__clang__) + +__attribute__ ((error("Don't use rand(), use rnd.next() instead"))) +#endif +#ifdef _MSC_VER +# pragma warning( disable : 4273 ) +#endif int rand() RAND_THROW_STATEMENT { quitf(_fail, "Don't use rand(), use rnd.next() instead"); - + /* This line never runs. */ //throw "Don't use rand(), use rnd.next() instead"; } +#if defined(__GNUC__) && !defined(__clang__) + +__attribute__ ((error("Don't use srand(), you should use " +"'registerGen(argc, argv, 1);' to initialize generator seed " +"by hash code of the command line params. The third parameter " +"is randomGeneratorVersion (currently the latest is 1)."))) +#endif +#ifdef _MSC_VER +# pragma warning( disable : 4273 ) +#endif void srand(unsigned int seed) RAND_THROW_STATEMENT { - quitf(_fail, "Don't use srand(), you should use " - "'registerGen(argc, argv, 1);' to initialize generator seed " - "by hash code of the command line params. The third parameter " - "is randomGeneratorVersion (currently the latest is 1) [ignored seed=%d].", seed); + quitf(_fail, "Don't use srand(), you should use " + "'registerGen(argc, argv, 1);' to initialize generator seed " + "by hash code of the command line params. The third parameter " + "is randomGeneratorVersion (currently the latest is 1) [ignored seed=%u].", seed); } -void startTest(int test) -{ +void startTest(int test) { const std::string testFileName = vtos(test); if (NULL == freopen(testFileName.c_str(), "wt", stdout)) __testlib_fail("Unable to write file '" + testFileName + "'"); } -inline std::string upperCase(std::string s) -{ - for (size_t i = 0; i < s.length(); i++) - if ('a' <= s[i] && s[i] <= 'z') - s[i] = char(s[i] - 'a' + 'A'); - return s; -} - -inline std::string lowerCase(std::string s) -{ - for (size_t i = 0; i < s.length(); i++) - if ('A' <= s[i] && s[i] <= 'Z') - s[i] = char(s[i] - 'A' + 'a'); - return s; -} - -inline std::string compress(const std::string& s) -{ +#ifdef __GNUC__ +__attribute__((const)) +#endif +inline std::string compress(const std::string &s) { return __testlib_part(s); } -inline std::string englishEnding(int x) -{ +#ifdef __GNUC__ +__attribute__((const)) +#endif +inline std::string englishEnding(int x) { x %= 100; if (x / 10 == 1) return "th"; @@ -3359,33 +4904,14 @@ inline std::string englishEnding(int x) return "th"; } -inline std::string trim(const std::string& s) -{ - if (s.empty()) - return s; - - int left = 0; - while (left < int(s.length()) && isBlanks(s[left])) - left++; - if (left >= int(s.length())) - return ""; - - int right = int(s.length()) - 1; - while (right >= 0 && isBlanks(s[right])) - right--; - if (right < 0) - return ""; - - return s.substr(left, right - left + 1); -} - -template -std::string join(_ForwardIterator first, _ForwardIterator last, _Separator separator) -{ +template +#ifdef __GNUC__ +__attribute__((const)) +#endif +std::string join(_ForwardIterator first, _ForwardIterator last, _Separator separator) { std::stringstream ss; bool repeated = false; - for (_ForwardIterator i = first; i != last; i++) - { + for (_ForwardIterator i = first; i != last; i++) { if (repeated) ss << separator; else @@ -3395,105 +4921,1234 @@ std::string join(_ForwardIterator first, _ForwardIterator last, _Separator separ return ss.str(); } -template -std::string join(_ForwardIterator first, _ForwardIterator last) -{ +template +#ifdef __GNUC__ +__attribute__((const)) +#endif +std::string join(_ForwardIterator first, _ForwardIterator last) { return join(first, last, ' '); } -template -std::string join(const _Collection& collection, _Separator separator) -{ +template +#ifdef __GNUC__ +__attribute__((const)) +#endif +std::string join(const _Collection &collection, _Separator separator) { return join(collection.begin(), collection.end(), separator); } -template -std::string join(const _Collection& collection) -{ +template +#ifdef __GNUC__ +__attribute__((const)) +#endif +std::string join(const _Collection &collection) { return join(collection, ' '); } -NORETURN void __testlib_expectedButFound(TResult result, std::string expected, std::string found, const char* prepend) -{ +/** + * Splits string s by character separator returning exactly k+1 items, + * where k is the number of separator occurrences. + */ +#ifdef __GNUC__ +__attribute__((const)) +#endif +std::vector split(const std::string &s, char separator) { + std::vector result; + std::string item; + for (size_t i = 0; i < s.length(); i++) + if (s[i] == separator) { + result.push_back(item); + item = ""; + } else + item += s[i]; + result.push_back(item); + return result; +} + +/** + * Splits string s by character separators returning exactly k+1 items, + * where k is the number of separator occurrences. + */ +#ifdef __GNUC__ +__attribute__((const)) +#endif +std::vector split(const std::string &s, const std::string &separators) { + if (separators.empty()) + return std::vector(1, s); + + std::vector isSeparator(256); + for (size_t i = 0; i < separators.size(); i++) + isSeparator[(unsigned char) (separators[i])] = true; + + std::vector result; + std::string item; + for (size_t i = 0; i < s.length(); i++) + if (isSeparator[(unsigned char) (s[i])]) { + result.push_back(item); + item = ""; + } else + item += s[i]; + result.push_back(item); + return result; +} + +/** + * Splits string s by character separator returning non-empty items. + */ +#ifdef __GNUC__ +__attribute__((const)) +#endif +std::vector tokenize(const std::string &s, char separator) { + std::vector result; + std::string item; + for (size_t i = 0; i < s.length(); i++) + if (s[i] == separator) { + if (!item.empty()) + result.push_back(item); + item = ""; + } else + item += s[i]; + if (!item.empty()) + result.push_back(item); + return result; +} + +/** + * Splits string s by character separators returning non-empty items. + */ +#ifdef __GNUC__ +__attribute__((const)) +#endif +std::vector tokenize(const std::string &s, const std::string &separators) { + if (separators.empty()) + return std::vector(1, s); + + std::vector isSeparator(256); + for (size_t i = 0; i < separators.size(); i++) + isSeparator[(unsigned char) (separators[i])] = true; + + std::vector result; + std::string item; + for (size_t i = 0; i < s.length(); i++) + if (isSeparator[(unsigned char) (s[i])]) { + if (!item.empty()) + result.push_back(item); + item = ""; + } else + item += s[i]; + + if (!item.empty()) + result.push_back(item); + + return result; +} + +NORETURN void __testlib_expectedButFound(TResult result, std::string expected, std::string found, const char *prepend) { std::string message; if (strlen(prepend) != 0) message = format("%s: expected '%s', but found '%s'", - compress(prepend).c_str(), compress(expected).c_str(), compress(found).c_str()); + compress(prepend).c_str(), compress(expected).c_str(), compress(found).c_str()); else message = format("expected '%s', but found '%s'", - compress(expected).c_str(), compress(found).c_str()); + compress(expected).c_str(), compress(found).c_str()); quit(result, message); } -NORETURN void __testlib_expectedButFound(TResult result, double expected, double found, const char* prepend) -{ +NORETURN void __testlib_expectedButFound(TResult result, double expected, double found, const char *prepend) { std::string expectedString = removeDoubleTrailingZeroes(format("%.12f", expected)); std::string foundString = removeDoubleTrailingZeroes(format("%.12f", found)); __testlib_expectedButFound(result, expectedString, foundString, prepend); } -template +template #ifdef __GNUC__ __attribute__ ((format (printf, 4, 5))) #endif -NORETURN void expectedButFound(TResult result, T expected, T found, const char* prependFormat = "", ...) -{ +NORETURN void expectedButFound(TResult result, T expected, T found, const char *prependFormat = "", ...) { FMT_TO_RESULT(prependFormat, prependFormat, prepend); std::string expectedString = vtos(expected); std::string foundString = vtos(found); __testlib_expectedButFound(result, expectedString, foundString, prepend.c_str()); } -template <> +template<> #ifdef __GNUC__ __attribute__ ((format (printf, 4, 5))) #endif -NORETURN void expectedButFound(TResult result, std::string expected, std::string found, const char* prependFormat, ...) -{ +NORETURN void +expectedButFound(TResult result, std::string expected, std::string found, const char *prependFormat, ...) { FMT_TO_RESULT(prependFormat, prependFormat, prepend); __testlib_expectedButFound(result, expected, found, prepend.c_str()); } -template <> +template<> #ifdef __GNUC__ __attribute__ ((format (printf, 4, 5))) #endif -NORETURN void expectedButFound(TResult result, double expected, double found, const char* prependFormat, ...) -{ +NORETURN void expectedButFound(TResult result, double expected, double found, const char *prependFormat, ...) { FMT_TO_RESULT(prependFormat, prependFormat, prepend); std::string expectedString = removeDoubleTrailingZeroes(format("%.12f", expected)); std::string foundString = removeDoubleTrailingZeroes(format("%.12f", found)); __testlib_expectedButFound(result, expectedString, foundString, prepend.c_str()); } -template <> +template<> #ifdef __GNUC__ __attribute__ ((format (printf, 4, 5))) #endif -NORETURN void expectedButFound(TResult result, const char* expected, const char* found, const char* prependFormat, ...) -{ +NORETURN void +expectedButFound(TResult result, const char *expected, const char *found, const char *prependFormat, + ...) { FMT_TO_RESULT(prependFormat, prependFormat, prepend); __testlib_expectedButFound(result, std::string(expected), std::string(found), prepend.c_str()); } -template <> +template<> #ifdef __GNUC__ __attribute__ ((format (printf, 4, 5))) #endif -NORETURN void expectedButFound(TResult result, float expected, float found, const char* prependFormat, ...) -{ +NORETURN void expectedButFound(TResult result, float expected, float found, const char *prependFormat, ...) { FMT_TO_RESULT(prependFormat, prependFormat, prepend); __testlib_expectedButFound(result, double(expected), double(found), prepend.c_str()); } -template <> +template<> #ifdef __GNUC__ __attribute__ ((format (printf, 4, 5))) #endif -NORETURN void expectedButFound(TResult result, long double expected, long double found, const char* prependFormat, ...) -{ +NORETURN void +expectedButFound(TResult result, long double expected, long double found, const char *prependFormat, ...) { FMT_TO_RESULT(prependFormat, prependFormat, prepend); __testlib_expectedButFound(result, double(expected), double(found), prepend.c_str()); } +#if __cplusplus > 199711L || defined(_MSC_VER) +template +struct is_iterable { + template + static char test(typename U::iterator *x); + + template + static long test(U *x); + + static const bool value = sizeof(test(0)) == 1; +}; + +template +struct __testlib_enable_if { +}; + +template +struct __testlib_enable_if { + typedef T type; +}; + +template +typename __testlib_enable_if::value, void>::type __testlib_print_one(const T &t) { + std::cout << t; +} + +template +typename __testlib_enable_if::value, void>::type __testlib_print_one(const T &t) { + bool first = true; + for (typename T::const_iterator i = t.begin(); i != t.end(); i++) { + if (first) + first = false; + else + std::cout << " "; + std::cout << *i; + } +} + +template<> +typename __testlib_enable_if::value, void>::type +__testlib_print_one(const std::string &t) { + std::cout << t; +} + +template +void __println_range(A begin, B end) { + bool first = true; + for (B i = B(begin); i != end; i++) { + if (first) + first = false; + else + std::cout << " "; + __testlib_print_one(*i); + } + std::cout << std::endl; +} + +template +struct is_iterator { + static T makeT(); + + typedef void *twoptrs[2]; + + static twoptrs &test(...); + + template + static typename R::iterator_category *test(R); + + template + static void *test(R *); + + static const bool value = sizeof(test(makeT())) == sizeof(void *); +}; + +template +struct is_iterator::value>::type> { + static const bool value = false; +}; + +template +typename __testlib_enable_if::value, void>::type println(const A &a, const B &b) { + __testlib_print_one(a); + std::cout << " "; + __testlib_print_one(b); + std::cout << std::endl; +} + +template +typename __testlib_enable_if::value, void>::type println(const A &a, const B &b) { + __println_range(a, b); +} + +template +void println(const A *a, const A *b) { + __println_range(a, b); +} + +template<> +void println(const char *a, const char *b) { + __testlib_print_one(a); + std::cout << " "; + __testlib_print_one(b); + std::cout << std::endl; +} + +template +void println(const T &x) { + __testlib_print_one(x); + std::cout << std::endl; +} + +template +void println(const A &a, const B &b, const C &c) { + __testlib_print_one(a); + std::cout << " "; + __testlib_print_one(b); + std::cout << " "; + __testlib_print_one(c); + std::cout << std::endl; +} + +template +void println(const A &a, const B &b, const C &c, const D &d) { + __testlib_print_one(a); + std::cout << " "; + __testlib_print_one(b); + std::cout << " "; + __testlib_print_one(c); + std::cout << " "; + __testlib_print_one(d); + std::cout << std::endl; +} + +template +void println(const A &a, const B &b, const C &c, const D &d, const E &e) { + __testlib_print_one(a); + std::cout << " "; + __testlib_print_one(b); + std::cout << " "; + __testlib_print_one(c); + std::cout << " "; + __testlib_print_one(d); + std::cout << " "; + __testlib_print_one(e); + std::cout << std::endl; +} + +template +void println(const A &a, const B &b, const C &c, const D &d, const E &e, const F &f) { + __testlib_print_one(a); + std::cout << " "; + __testlib_print_one(b); + std::cout << " "; + __testlib_print_one(c); + std::cout << " "; + __testlib_print_one(d); + std::cout << " "; + __testlib_print_one(e); + std::cout << " "; + __testlib_print_one(f); + std::cout << std::endl; +} + +template +void println(const A &a, const B &b, const C &c, const D &d, const E &e, const F &f, const G &g) { + __testlib_print_one(a); + std::cout << " "; + __testlib_print_one(b); + std::cout << " "; + __testlib_print_one(c); + std::cout << " "; + __testlib_print_one(d); + std::cout << " "; + __testlib_print_one(e); + std::cout << " "; + __testlib_print_one(f); + std::cout << " "; + __testlib_print_one(g); + std::cout << std::endl; +} + +/* opts */ + +/** + * A struct for a singular testlib opt, containing the raw string value, + * and a boolean value for marking whether the opt is used. + */ +struct TestlibOpt { + std::string value; + bool used; + + TestlibOpt() : value(), used(false) {} +}; + +/** + * Get the type of opt based on the number of `-` at the beginning and the + * _validity_ of the key name. + * + * A valid key name must start with an alphabetical character. + * + * Returns: 1 if s has one `-` at the beginning, that is, "-keyName". + * 2 if s has two `-` at the beginning, that is, "--keyName". + * 0 otherwise. That is, if s has no `-` at the beginning, or has more + * than 2 at the beginning ("---keyName", "----keyName", ...), or the + * keyName is invalid (the first character is not an alphabetical + * character). + */ +size_t getOptType(char *s) { + if (!s || strlen(s) <= 1) + return 0; + + if (s[0] == '-') { + if (isalpha(s[1])) + return 1; + else if (s[1] == '-') + return isalpha(s[2]) ? 2 : 0; + } + + return 0; +} + +/** + * Parse the opt at a given index, and put it into the opts maps. + * + * An opt can has the following form: + * 1) -keyName=value or --keyName=value (ex. -n=10 --test-count=20) + * 2) -keyName value or --keyName value (ex. -n 10 --test-count 20) + * 3) -kNumval or --kNumval (ex. -n10 --t20) + * 4) -boolProperty or --boolProperty (ex. -sorted --tree-only) + * + * Only the second form consumes 2 arguments. The other consumes only 1 + * argument. + * + * In the third form, the key is a single character, and after the key is the + * value. The value _should_ be a number. + * + * In the forth form, the value is true. + * + * Params: + * - argc and argv: the number of command line arguments and the command line + * arguments themselves. + * - index: the starting index of the opts. + * - opts: the map containing the resulting opt. + * + * Returns: the number of consumed arguments to parse the opt. + * 0 if there is no arguments to parse. + * + * Algorithm details: + * TODO. Please refer to the implementation to see how the code handles the 3rd and 4th forms separately. + */ +size_t parseOpt(size_t argc, char *argv[], size_t index, std::map &opts) { + if (index >= argc) + return 0; + + size_t type = getOptType(argv[index]), inc = 1; + if (type > 0) { + std::string key(argv[index] + type), val; + size_t sep = key.find('='); + if (sep != std::string::npos) { + val = key.substr(sep + 1); + key = key.substr(0, sep); + } else { + if (index + 1 < argc && getOptType(argv[index + 1]) == 0) { + val = argv[index + 1]; + inc = 2; + } else { + if (key.length() > 1 && isdigit(key[1])) { + val = key.substr(1); + key = key.substr(0, 1); + } else { + val = "true"; + } + } + } + opts[key].value = val; + } else { + return inc; + } + + return inc; +} + +/** + * Global list containing all the arguments in the order given in the command line. + */ +std::vector __testlib_argv; + +/** + * Global dictionary containing all the parsed opts. + */ +std::map __testlib_opts; + +/** + * Whether automatic no unused opts ensurement should be done. This flag will + * be turned on when `has_opt` or `opt(key, default_value)` is called. + * + * The automatic ensurement can be suppressed when + * __testlib_ensureNoUnusedOptsSuppressed is true. + */ +bool __testlib_ensureNoUnusedOptsFlag = false; + +/** + * Suppress no unused opts automatic ensurement. Can be set to true with + * `suppressEnsureNoUnusedOpts()`. + */ +bool __testlib_ensureNoUnusedOptsSuppressed = false; + +/** + * Parse command line arguments into opts. + * The results are stored into __testlib_argv and __testlib_opts. + */ +void prepareOpts(int argc, char *argv[]) { + if (argc <= 0) + __testlib_fail("Opts: expected argc>=0 but found " + toString(argc)); + size_t n = static_cast(argc); // NOLINT(hicpp-use-auto,modernize-use-auto) + __testlib_opts = std::map(); + for (size_t index = 1; index < n; index += parseOpt(n, argv, index, __testlib_opts)); + __testlib_argv = std::vector(n); + for (size_t index = 0; index < n; index++) + __testlib_argv[index] = argv[index]; +} + +/** + * An utility function to get the argument with a given index. This function + * also print a readable message when no arguments are found. + */ +std::string __testlib_indexToArgv(int index) { + if (index < 0 || index >= int(__testlib_argv.size())) + __testlib_fail("Opts: index '" + toString(index) + "' is out of range [0," + + toString(__testlib_argv.size()) + ")"); + return __testlib_argv[size_t(index)]; +} + +/** + * An utility function to get the opt with a given key . This function + * also print a readable message when no opts are found. + */ +std::string __testlib_keyToOpts(const std::string &key) { + auto it = __testlib_opts.find(key); + if (it == __testlib_opts.end()) + __testlib_fail("Opts: unknown key '" + compress(key) + "'"); + it->second.used = true; + return it->second.value; +} + +template +T optValueToIntegral(const std::string &s, bool nonnegative); + +long double optValueToLongDouble(const std::string &s); + +std::string parseExponentialOptValue(const std::string &s) { + size_t pos = std::string::npos; + for (size_t i = 0; i < s.length(); i++) + if (s[i] == 'e' || s[i] == 'E') { + if (pos != std::string::npos) + __testlib_fail("Opts: expected typical exponential notation but '" + compress(s) + "' found"); + pos = i; + } + if (pos == std::string::npos) + return s; + std::string e = s.substr(pos + 1); + if (!e.empty() && e[0] == '+') + e = e.substr(1); + if (e.empty()) + __testlib_fail("Opts: expected typical exponential notation but '" + compress(s) + "' found"); + if (e.length() > 20) + __testlib_fail("Opts: expected typical exponential notation but '" + compress(s) + "' found"); + int ne = optValueToIntegral(e, false); + std::string num = s.substr(0, pos); + if (num.length() > 20) + __testlib_fail("Opts: expected typical exponential notation but '" + compress(s) + "' found"); + if (!num.empty() && num[0] == '+') + num = num.substr(1); + optValueToLongDouble(num); + bool minus = false; + if (num[0] == '-') { + minus = true; + num = num.substr(1); + } + for (int i = 0; i < +ne; i++) { + size_t sep = num.find('.'); + if (sep == std::string::npos) + num += '0'; + else { + if (sep + 1 == num.length()) + num[sep] = '0'; + else + std::swap(num[sep], num[sep + 1]); + } + } + for (int i = 0; i < -ne; i++) { + size_t sep = num.find('.'); + if (sep == std::string::npos) + num.insert(num.begin() + int(num.length()) - 1, '.'); + else { + if (sep == 0) + num.insert(num.begin() + 1, '0'); + else + std::swap(num[sep - 1], num[sep]); + } + } + while (!num.empty() && num[0] == '0') + num = num.substr(1); + while (num.find('.') != std::string::npos && num.back() == '0') + num = num.substr(0, num.length() - 1); + if (!num.empty() && num.back() == '.') + num = num.substr(0, num.length() - 1); + if ((!num.empty() && num[0] == '.') || num.empty()) + num.insert(num.begin(), '0'); + return (minus ? "-" : "") + num; +} + +template +T optValueToIntegral(const std::string &s_, bool nonnegative) { + std::string s(parseExponentialOptValue(s_)); + if (s.empty()) + __testlib_fail("Opts: expected integer but '" + compress(s_) + "' found"); + T value = 0; + long double about = 0.0; + signed char sign = +1; + size_t pos = 0; + if (s[pos] == '-') { + if (nonnegative) + __testlib_fail("Opts: expected non-negative integer but '" + compress(s_) + "' found"); + sign = -1; + pos++; + } + for (size_t i = pos; i < s.length(); i++) { + if (s[i] < '0' || s[i] > '9') + __testlib_fail("Opts: expected integer but '" + compress(s_) + "' found"); + value = T(value * 10 + s[i] - '0'); + about = about * 10 + s[i] - '0'; + } + value *= sign; + about *= sign; + if (fabsl(value - about) > 0.1) + __testlib_fail("Opts: integer overflow: expected integer but '" + compress(s_) + "' found"); + return value; +} + +long double optValueToLongDouble(const std::string &s_) { + std::string s(parseExponentialOptValue(s_)); + if (s.empty()) + __testlib_fail("Opts: expected float number but '" + compress(s_) + "' found"); + long double value = 0.0; + signed char sign = +1; + size_t pos = 0; + if (s[pos] == '-') { + sign = -1; + pos++; + } + bool period = false; + long double mul = 1.0; + for (size_t i = pos; i < s.length(); i++) { + if (s[i] == '.') { + if (period) + __testlib_fail("Opts: expected float number but '" + compress(s_) + "' found"); + else { + period = true; + continue; + } + } + if (period) + mul *= 10.0; + if (s[i] < '0' || s[i] > '9') + __testlib_fail("Opts: expected float number but '" + compress(s_) + "' found"); + if (period) + value += (s[i] - '0') / mul; + else + value = value * 10 + s[i] - '0'; + } + value *= sign; + return value; +} + +/** + * Return true if there is an opt with a given key. + * + * By calling this function, automatic ensurement for no unused opts will be + * done when the program is finalized. Call suppressEnsureNoUnusedOpts() to + * turn it off. + */ +bool has_opt(const std::string &key) { + __testlib_ensureNoUnusedOptsFlag = true; + return __testlib_opts.count(key) != 0; +} + +/* About the following part for opt with 2 and 3 arguments. + * + * To parse the argv/opts correctly for a give type (integer, floating point or + * string), some meta programming must be done to determine the type of + * the type, and use the correct parsing function accordingly. + * + * The pseudo algorithm for determining the type of T and parse it accordingly + * is as follows: + * + * if (T is integral type) { + * if (T is unsigned) { + * parse the argv/opt as an **unsigned integer** of type T. + * } else { + * parse the argv/opt as an **signed integer** of type T. + * } else { + * if (T is floating point type) { + * parse the argv/opt as an **floating point** of type T. + * } else { + * // T should be std::string + * just the raw content of the argv/opts. + * } + * } + * + * To help with meta programming, some `opt` function with 2 or 3 arguments are + * defined. + * + * Opt with 3 arguments: T opt(true/false is_integral, true/false is_unsigned, index/key) + * + * + The first argument is for determining whether the type T is an integral + * type. That is, the result of std::is_integral() should be passed to + * this argument. When false, the type _should_ be either floating point or a + * std::string. + * + * + The second argument is for determining whether the signedness of the type + * T (if it is unsigned or signed). That is, the result of + * std::is_unsigned() should be passed to this argument. This argument can + * be ignored if the first one is false, because it only applies to integer. + * + * Opt with 2 arguments: T opt(true/false is_floating_point, index/key) + * + The first argument is for determining whether the type T is a floating + * point type. That is, the result of std::is_floating_point() should be + * passed to this argument. When false, the type _should_ be a std::string. + */ + +template +T opt(std::false_type is_floating_point, int index); + +template<> +std::string opt(std::false_type /*is_floating_point*/, int index) { + return __testlib_indexToArgv(index); +} + +template +T opt(std::true_type /*is_floating_point*/, int index) { + return T(optValueToLongDouble(__testlib_indexToArgv(index))); +} + +template +T opt(std::false_type /*is_integral*/, U /*is_unsigned*/, int index) { + return opt(std::is_floating_point(), index); +} + +template +T opt(std::true_type /*is_integral*/, std::false_type /*is_unsigned*/, int index) { + return optValueToIntegral(__testlib_indexToArgv(index), false); +} + +template +T opt(std::true_type /*is_integral*/, std::true_type /*is_unsigned*/, int index) { + return optValueToIntegral(__testlib_indexToArgv(index), true); +} + +template<> +bool opt(std::true_type /*is_integral*/, std::true_type /*is_unsigned*/, int index) { + std::string value = __testlib_indexToArgv(index); + if (value == "true" || value == "1") + return true; + if (value == "false" || value == "0") + return false; + __testlib_fail("Opts: opt by index '" + toString(index) + "': expected bool true/false or 0/1 but '" + + compress(value) + "' found"); +} + +/** + * Return the parsed argv by a given index. + */ +template +T opt(int index) { + return opt(std::is_integral(), std::is_unsigned(), index); +} + +/** + * Return the raw string value of an argv by a given index. + */ +std::string opt(int index) { + return opt(index); +} + +/** + * Return the parsed argv by a given index. If the index is bigger than + * the number of argv, return the given default_value. + */ +template +T opt(int index, const T &default_value) { + if (index >= int(__testlib_argv.size())) { + return default_value; + } + return opt(index); +} + +/** + * Return the raw string value of an argv by a given index. If the index is + * bigger than the number of argv, return the given default_value. + */ +std::string opt(int index, const std::string &default_value) { + return opt(index, default_value); +} + +template +T opt(std::false_type is_floating_point, const std::string &key); + +template<> +std::string opt(std::false_type /*is_floating_point*/, const std::string &key) { + return __testlib_keyToOpts(key); +} + +template +T opt(std::true_type /*is_integral*/, const std::string &key) { + return T(optValueToLongDouble(__testlib_keyToOpts(key))); +} + +template +T opt(std::false_type /*is_integral*/, U, const std::string &key) { + return opt(std::is_floating_point(), key); +} + +template +T opt(std::true_type /*is_integral*/, std::false_type /*is_unsigned*/, const std::string &key) { + return optValueToIntegral(__testlib_keyToOpts(key), false); +} + +template +T opt(std::true_type /*is_integral*/, std::true_type /*is_unsigned*/, const std::string &key) { + return optValueToIntegral(__testlib_keyToOpts(key), true); +} + +template<> +bool opt(std::true_type /*is_integral*/, std::true_type /*is_unsigned*/, const std::string &key) { + if (!has_opt(key)) + return false; + std::string value = __testlib_keyToOpts(key); + if (value == "true" || value == "1") + return true; + if (value == "false" || value == "0") + return false; + __testlib_fail("Opts: key '" + compress(key) + "': expected bool true/false or 0/1 but '" + + compress(value) + "' found"); +} + +/** + * Return the parsed opt by a given key. + */ +template +T opt(const std::string &key) { + return opt(std::is_integral(), std::is_unsigned(), key); +} + +/** + * Return the raw string value of an opt by a given key + */ +std::string opt(const std::string &key) { + return opt(key); +} + +/* Scorer started. */ + +enum TestResultVerdict { + SKIPPED, + OK, + WRONG_ANSWER, + RUNTIME_ERROR, + TIME_LIMIT_EXCEEDED, + IDLENESS_LIMIT_EXCEEDED, + MEMORY_LIMIT_EXCEEDED, + COMPILATION_ERROR, + CRASHED, + FAILED +}; + +std::string serializeVerdict(TestResultVerdict verdict) { + switch (verdict) { + case SKIPPED: return "SKIPPED"; + case OK: return "OK"; + case WRONG_ANSWER: return "WRONG_ANSWER"; + case RUNTIME_ERROR: return "RUNTIME_ERROR"; + case TIME_LIMIT_EXCEEDED: return "TIME_LIMIT_EXCEEDED"; + case IDLENESS_LIMIT_EXCEEDED: return "IDLENESS_LIMIT_EXCEEDED"; + case MEMORY_LIMIT_EXCEEDED: return "MEMORY_LIMIT_EXCEEDED"; + case COMPILATION_ERROR: return "COMPILATION_ERROR"; + case CRASHED: return "CRASHED"; + case FAILED: return "FAILED"; + } + throw "Unexpected verdict"; +} + +TestResultVerdict deserializeTestResultVerdict(std::string s) { + if (s == "SKIPPED") + return SKIPPED; + else if (s == "OK") + return OK; + else if (s == "WRONG_ANSWER") + return WRONG_ANSWER; + else if (s == "RUNTIME_ERROR") + return RUNTIME_ERROR; + else if (s == "TIME_LIMIT_EXCEEDED") + return TIME_LIMIT_EXCEEDED; + else if (s == "IDLENESS_LIMIT_EXCEEDED") + return IDLENESS_LIMIT_EXCEEDED; + else if (s == "MEMORY_LIMIT_EXCEEDED") + return MEMORY_LIMIT_EXCEEDED; + else if (s == "COMPILATION_ERROR") + return COMPILATION_ERROR; + else if (s == "CRASHED") + return CRASHED; + else if (s == "FAILED") + return FAILED; + ensuref(false, "Unexpected serialized TestResultVerdict"); + // No return actually. + return FAILED; +} + +struct TestResult { + int testIndex; + std::string testset; + std::string group; + TestResultVerdict verdict; + double points; + long long timeConsumed; + long long memoryConsumed; + std::string input; + std::string output; + std::string answer; + int exitCode; + std::string checkerComment; +}; + +std::string serializePoints(double points) { + if (std::isnan(points)) + return ""; + else { + char c[64]; + snprintf(c, 64, "%.03lf", points); + return c; + } +} + +double deserializePoints(std::string s) { + if (s.empty()) + return std::numeric_limits::quiet_NaN(); + else { + double result; + ensuref(sscanf(s.c_str(), "%lf", &result) == 1, "Invalid serialized points"); + return result; + } +} + +std::string escapeTestResultString(std::string s) { + std::string result; + for (size_t i = 0; i < s.length(); i++) { + if (s[i] == '\r') + continue; + if (s[i] == '\n') { + result += "\\n"; + continue; + } + if (s[i] == '\\' || s[i] == ';') + result += '\\'; + result += s[i]; + } + return result; +} + +std::string unescapeTestResultString(std::string s) { + std::string result; + for (size_t i = 0; i < s.length(); i++) { + if (s[i] == '\\' && i + 1 < s.length()) { + if (s[i + 1] == 'n') { + result += '\n'; + i++; + continue; + } else if (s[i + 1] == ';' || s[i + 1] == '\\') { + result += s[i + 1]; + i++; + continue; + } + } + result += s[i]; + } + return result; +} + +std::string serializeTestResult(TestResult tr) { + std::string result; + result += std::to_string(tr.testIndex); + result += ";"; + result += escapeTestResultString(tr.testset); + result += ";"; + result += escapeTestResultString(tr.group); + result += ";"; + result += serializeVerdict(tr.verdict); + result += ";"; + result += serializePoints(tr.points); + result += ";"; + result += std::to_string(tr.timeConsumed); + result += ";"; + result += std::to_string(tr.memoryConsumed); + result += ";"; + result += escapeTestResultString(tr.input); + result += ";"; + result += escapeTestResultString(tr.output); + result += ";"; + result += escapeTestResultString(tr.answer); + result += ";"; + result += std::to_string(tr.exitCode); + result += ";"; + result += escapeTestResultString(tr.checkerComment); + return result; +} + +TestResult deserializeTestResult(std::string s) { + std::vector items; + std::string t; + for (size_t i = 0; i < s.length(); i++) { + if (s[i] == '\\') { + t += s[i]; + if (i + 1 < s.length()) + t += s[i + 1]; + i++; + continue; + } else { + if (s[i] == ';') { + items.push_back(t); + t = ""; + } else + t += s[i]; + } + } + items.push_back(t); + + ensuref(items.size() == 12, "Invalid TestResult serialization: expected exactly 12 items"); + + TestResult tr; + size_t pos = 0; + tr.testIndex = stoi(items[pos++]); + tr.testset = unescapeTestResultString(items[pos++]); + tr.group = unescapeTestResultString(items[pos++]); + tr.verdict = deserializeTestResultVerdict(items[pos++]); + tr.points = deserializePoints(items[pos++]); + tr.timeConsumed = stoll(items[pos++]); + tr.memoryConsumed = stoll(items[pos++]); + tr.input = unescapeTestResultString(items[pos++]); + tr.output = unescapeTestResultString(items[pos++]); + tr.answer = unescapeTestResultString(items[pos++]); + tr.exitCode = stoi(items[pos++]); + tr.checkerComment = unescapeTestResultString(items[pos++]); + + return tr; +} + +std::vector readTestResults(std::string fileName) { + std::ifstream stream; + stream.open(fileName.c_str(), std::ios::in); + ensuref(stream.is_open(), "Can't read test results file '%s'", fileName.c_str()); + std::vector result; + std::string line; + while (getline(stream, line)) + if (!line.empty()) + result.push_back(deserializeTestResult(line)); + stream.close(); + return result; +} + +std::function)> __testlib_scorer; + +struct TestlibScorerGuard { + ~TestlibScorerGuard() { + if (testlibMode == _scorer) { + std::vector testResults; + while (!inf.eof()) { + std::string line = inf.readLine(); + if (!line.empty()) + testResults.push_back(deserializeTestResult(line)); + } + inf.readEof(); + printf("%.3f\n", __testlib_scorer(testResults)); + } + } +} __testlib_scorer_guard; + +void registerScorer(int argc, char *argv[], std::function)> scorer) { + /* Suppress unused. */ + (void)(argc), (void)(argv); + + __testlib_ensuresPreconditions(); + + testlibMode = _scorer; + __testlib_set_binary(stdin); + + inf.init(stdin, _input); + inf.strict = false; + + __testlib_scorer = scorer; +} + +/* Scorer ended. */ + +/** + * Return the parsed opt by a given key. If no opts with the given key are + * found, return the given default_value. + * + * By calling this function, automatic ensurement for no unused opts will be + * done when the program is finalized. Call suppressEnsureNoUnusedOpts() to + * turn it off. + */ +template +T opt(const std::string &key, const T &default_value) { + if (!has_opt(key)) { + return default_value; + } + return opt(key); +} + +/** + * Return the raw string value of an opt by a given key. If no opts with the + * given key are found, return the given default_value. + * + * By calling this function, automatic ensurement for no unused opts will be + * done when the program is finalized. Call suppressEnsureNoUnusedOpts() to + * turn it off. + */ +std::string opt(const std::string &key, const std::string &default_value) { + return opt(key, default_value); +} + +/** + * Check if all opts are used. If not, __testlib_fail is called. + * Should be used after calling all opt() function calls. + * + * This function is useful when opt() with default_value for checking typos + * in the opt's key. + */ +void ensureNoUnusedOpts() { + for (const auto &opt: __testlib_opts) { + if (!opt.second.used) { + __testlib_fail(format("Opts: unused key '%s'", compress(opt.first).c_str())); + } + } +} + +void suppressEnsureNoUnusedOpts() { + __testlib_ensureNoUnusedOptsSuppressed = true; +} + +void TestlibFinalizeGuard::autoEnsureNoUnusedOpts() { + if (__testlib_ensureNoUnusedOptsFlag && !__testlib_ensureNoUnusedOptsSuppressed) { + ensureNoUnusedOpts(); + } +} + +TestlibFinalizeGuard testlibFinalizeGuard; + #endif +#endif + + +void registerChecker(std::string probName, int argc, char* argv[]) +{ + setName("checker for problem %s", probName.c_str()); + registerTestlibCmd(argc, argv); +} + + + +const std::string _grader_OK = "OK"; +const std::string _grader_SV = "SV"; +const std::string _grader_WA = "WA"; +const std::string _grader_FAIL = "FAIL"; + + +void InStream::readSecret(std::string secret) +{ + if (readWord() != secret) + quitf(_wa, "secret mismatch"); + eoln(); +} + +void readBothSecrets(std::string secret) +{ + ans.readSecret(secret); + ouf.readSecret(secret); +} + +void InStream::readGraderResult() +{ + std::string result = readWord(); + eoln(); + if (result == _grader_OK) + return; + if (result == _grader_SV) + { + if (eof()) + quitf(_wa, "Security violation in grader"); + std::string msg = readLine(); + quitf(_wa, "Security violation in grader: %s", msg.c_str()); + } + if (result == _grader_WA) + { + if (eof()) + quitf(_wa, "WA in grader"); + std::string msg = readLine(); + quitf(_wa, "WA in grader: %s", msg.c_str()); + } + if (result == _grader_FAIL) + { + if (eof()) + quitf(_fail, "FAIL in grader"); + std::string msg = readLine(); + quitf(_fail, "FAIL in grader: %s", msg.c_str()); + } + quitf(_fail, "Unknown grader result"); +} + +void readBothGraderResults() +{ + ans.readGraderResult(); + ouf.readGraderResult(); +} + + +NORETURN void quit(TResult result) +{ + ouf.quit(result, ""); +} \ No newline at end of file diff --git a/judger/uoj_judger/include/testlib_interactor.h b/judger/uoj_judger/include/testlib_interactor.h new file mode 100644 index 000000000..c0be90829 --- /dev/null +++ b/judger/uoj_judger/include/testlib_interactor.h @@ -0,0 +1,3644 @@ +/* + * It is strictly recommended to include "testlib.h" before any other include + * in your code. In this case testlib overrides compiler specific "random()". + * + * If you can't compile your code and compiler outputs something about + * ambiguous call of "random_shuffle", "rand" or "srand" it means that + * you shouldn't use them. Use "shuffle", and "rnd.next()" instead of them + * because these calls produce stable result for any C++ compiler. Read + * sample generator sources for clarification. + * + * Please read the documentation for class "random_t" and use "rnd" instance in + * generators. Probably, these sample calls will be usefull for you: + * rnd.next(); rnd.next(100); rnd.next(1, 2); + * rnd.next(3.14); rnd.next("[a-z]{1,100}"). + * + * Also read about wnext() to generate off-center random distribution. + * + * See http://code.google.com/p/testlib/ to get latest version or bug tracker. + */ + +#ifndef _TESTLIB_H_ +#define _TESTLIB_H_ + +/* + * Copyright (c) 2005-2013 + */ + +#define VERSION "0.9.5-uoj" + +/* + * Mike Mirzayanov + * + * This material is provided "as is", with absolutely no warranty expressed + * or implied. Any use is at your own risk. + * + * Permission to use or copy this software for any purpose is hereby granted + * without fee, provided the above notices are retained on all copies. + * Permission to modify the code and to distribute modified code is granted, + * provided the above notices are retained, and a notice that the code was + * modified is included with the above copyright notice. + * + */ + +/* NOTE: This file contains testlib library for C++. + * + * Check, using testlib running format: + * check.exe [ [-appes]], + * If result file is specified it will contain results. + * + * Validator, using testlib running format: + * validator.exe < input.txt, + * It will return non-zero exit code and writes message to standard output. + * + * Generator, using testlib running format: + * gen.exe [parameter-1] [parameter-2] [... paramerter-n] + * You can write generated test(s) into standard output or into the file(s). + * + * Interactor, using testlib running format: + * interactor.exe [ [ [-appes]]], + * Reads test from inf (mapped to args[1]), writes result to tout (mapped to argv[2], + * can be judged by checker later), reads program output from ouf (mapped to stdin), + * writes output to program via stdout (use cout, printf, etc). + */ + +const char* latestFeatures[] = { + "Removed disable buffers for interactive problems, because it works unexpectedly in wine", + "InStream over string: constructor of InStream from base InStream to inherit policies and std::string", + "Added expectedButFound quit function, examples: expectedButFound(_wa, 10, 20), expectedButFound(_fail, ja, pa, \"[n=%d,m=%d]\", n, m)", + "Fixed incorrect interval parsing in patterns", + "Use registerGen(argc, argv, 1) to develop new generator, use registerGen(argc, argv, 0) to compile old generators (originally created for testlib under 0.8.7)", + "Introduced disableFinalizeGuard() to switch off finalization checkings", + "Use join() functions to format a range of items as a single string (separated by spaces or other separators)", + "Use -DENABLE_UNEXPECTED_EOF to enable special exit code (by default, 8) in case of unexpected eof. It is good idea to use it in interactors", + "Use -DUSE_RND_AS_BEFORE_087 to compile in compatibility mode with random behavior of versions before 0.8.7", + "Fixed bug with nan in stringToDouble", + "Fixed issue around overloads for size_t on x64", + "Added attribute 'points' to the XML output in case of result=_points", + "Exit codes can be customized via macros, e.g. -DPE_EXIT_CODE=14", + "Introduced InStream function readWordTo/readTokenTo/readStringTo/readLineTo for faster reading", + "Introduced global functions: format(), englishEnding(), upperCase(), lowerCase(), compress()", + "Manual buffer in InStreams, some IO speed improvements", + "Introduced quitif(bool, const char* pattern, ...) which delegates to quitf() in case of first argument is true", + "Introduced guard against missed quitf() in checker or readEof() in validators", + "Supported readStrictReal/readStrictDouble - to use in validators to check strictly float numbers", + "Supported registerInteraction(argc, argv)", + "Print checker message to the stderr instead of stdout", + "Supported TResult _points to output calculated score, use quitp(...) functions", + "Fixed to be compilable on Mac", + "PC_BASE_EXIT_CODE=50 in case of defined TESTSYS", + "Fixed issues 19-21, added __attribute__ format printf", + "Some bug fixes", + "ouf.readInt(1, 100) and similar calls return WA", + "Modified random_t to avoid integer overflow", + "Truncated checker output [patch by Stepan Gatilov]", + "Renamed class random -> class random_t", + "Supported name parameter for read-and-validation methods, like readInt(1, 2, \"n\")", + "Fixed bug in readDouble()", + "Improved ensuref(), fixed nextLine to work in case of EOF, added startTest()", + "Supported \"partially correct\", example: quitf(_pc(13), \"result=%d\", result)", + "Added shuffle(begin, end), use it instead of random_shuffle(begin, end)", + "Added readLine(const string& ptrn), fixed the logic of readLine() in the validation mode", + "Package extended with samples of generators and validators", + "Written the documentation for classes and public methods in testlib.h", + "Implemented random routine to support generators, use registerGen() to switch it on", + "Implemented strict mode to validate tests, use registerValidation() to switch it on", + "Now ncmp.cpp and wcmp.cpp are return WA if answer is suffix or prefix of the output", + "Added InStream::readLong() and removed InStream::readLongint()", + "Now no footer added to each report by default (use directive FOOTER to switch on)", + "Now every checker has a name, use setName(const char* format, ...) to set it", + "Now it is compatible with TTS (by Kittens Computing)", + "Added \'ensure(condition, message = \"\")\' feature, it works like assert()", + "Fixed compatibility with MS C++ 7.1", + "Added footer with exit code information", + "Added compatibility with EJUDGE (compile with EJUDGE directive)" + }; + +#ifdef _MSC_VER +#define _CRT_SECURE_NO_DEPRECATE +#define _CRT_SECURE_NO_WARNINGS +#endif + +/* Overrides random() for Borland C++. */ +#ifdef __EMSCRIPTEN__ +#define srand __srand_deprecated +#define rand __rand_deprecated +#endif +#define random __random_deprecated +#include +#include +#include +#include +#undef random +#ifdef __EMSCRIPTEN__ +#undef rand +#undef srand +#endif + +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include + +#include + +#ifdef __EMSCRIPTEN__ +#include +#endif + +#if ( _WIN32 || __WIN32__ || _WIN64 || __WIN64__ ) +# if !defined(_MSC_VER) || _MSC_VER>1400 +# include +# else +# define WORD unsigned short +# endif +# include +# define ON_WINDOWS +#else +# define WORD unsigned short +#endif + +#ifdef linux +#include +#endif + +#ifndef LLONG_MIN +#define LLONG_MIN (-9223372036854775807LL - 1) +#endif + +#define LF ((char)10) +#define CR ((char)13) +#define TAB ((char)9) +#define SPACE ((char)' ') +#define EOFC ((char)26) + +#ifndef OK_EXIT_CODE +# define OK_EXIT_CODE 0 +#endif + +#ifndef WA_EXIT_CODE +# ifdef EJUDGE +# define WA_EXIT_CODE 5 +# else +# define WA_EXIT_CODE 1 +# endif +#endif + +#ifndef PE_EXIT_CODE +# ifdef EJUDGE +# define PE_EXIT_CODE 4 +# else +# define PE_EXIT_CODE 2 +# endif +#endif + +#ifndef FAIL_EXIT_CODE +# ifdef EJUDGE +# define FAIL_EXIT_CODE 6 +# else +# define FAIL_EXIT_CODE 3 +# endif +#endif + +#ifndef DIRT_EXIT_CODE +# ifdef EJUDGE +# define DIRT_EXIT_CODE 6 +# else +# define DIRT_EXIT_CODE 4 +# endif +#endif + +#ifndef POINTS_EXIT_CODE +# define POINTS_EXIT_CODE 0 +#endif + +#ifndef UNEXPECTED_EOF_EXIT_CODE +# define UNEXPECTED_EOF_EXIT_CODE 8 +#endif + +#ifndef PC_BASE_EXIT_CODE +# ifdef TESTSYS +# define PC_BASE_EXIT_CODE 50 +# else +# define PC_BASE_EXIT_CODE 0 +# endif +#endif + +#ifdef __GNUC__ +# define __TESTLIB_STATIC_ASSERT(condition) typedef void* __testlib_static_assert_type[(condition) ? 1 : -1] __attribute__((unused)) +#else +# define __TESTLIB_STATIC_ASSERT(condition) typedef void* __testlib_static_assert_type[(condition) ? 1 : -1] +#endif + +#ifdef ON_WINDOWS +#define I64 "%I64d" +#define U64 "%I64u" +#else +#define I64 "%lld" +#define U64 "%llu" +#endif + +#ifdef _MSC_VER +# define NORETURN __declspec(noreturn) +#elif defined __GNUC__ +# define NORETURN __attribute__ ((noreturn)) +#else +# define NORETURN +#endif + +static char __testlib_format_buffer[1024]; +static int __testlib_format_buffer_usage_count = 0; + +#define FMT_TO_RESULT(fmt, cstr, result) std::string result; \ + if (__testlib_format_buffer_usage_count != 0) \ + __testlib_fail("FMT_TO_RESULT::__testlib_format_buffer_usage_count != 0"); \ + __testlib_format_buffer_usage_count++; \ + va_list ap; \ + va_start(ap, fmt); \ + std::vsprintf(__testlib_format_buffer, cstr, ap); \ + va_end(ap); \ + result = std::string(__testlib_format_buffer); \ + __testlib_format_buffer_usage_count--; \ + +const long long __TESTLIB_LONGLONG_MAX = 9223372036854775807LL; + +NORETURN static void __testlib_fail(const std::string& message); + +template +static inline T __testlib_abs(const T& x) +{ + return x > 0 ? x : -x; +} + +template +static inline T __testlib_min(const T& a, const T& b) +{ + return a < b ? a : b; +} + +template +static inline T __testlib_max(const T& a, const T& b) +{ + return a > b ? a : b; +} + +static bool __testlib_prelimIsNaN(double r) +{ + volatile double ra = r; +#ifndef __BORLANDC__ + return ((ra != ra) == true) && ((ra == ra) == false) && ((1.0 > ra) == false) && ((1.0 < ra) == false); +#else + return std::_isnan(ra); +#endif +} + +static std::string removeDoubleTrailingZeroes(std::string value) +{ + while (!value.empty() && value[value.length() - 1] == '0' && value.find('.') != std::string::npos) + value = value.substr(0, value.length() - 1); + return value + '0'; +} + +#ifdef __GNUC__ +__attribute__ ((format (printf, 1, 2))) +#endif +std::string format(const char* fmt, ...) +{ + FMT_TO_RESULT(fmt, fmt, result); + return result; +} + +std::string format(const std::string& fmt, ...) +{ + FMT_TO_RESULT(fmt, fmt.c_str(), result); + return result; +} + +static std::string __testlib_part(const std::string& s); + +static bool __testlib_isNaN(double r) +{ + __TESTLIB_STATIC_ASSERT(sizeof(double) == sizeof(long long)); + volatile double ra = r; + long long llr1, llr2; + std::memcpy((void*)&llr1, (void*)&ra, sizeof(double)); + ra = -ra; + std::memcpy((void*)&llr2, (void*)&ra, sizeof(double)); + long long llnan = 0xFFF8000000000000ll; + return __testlib_prelimIsNaN(r) || llnan == llr1 || llnan == llr2; +} + +static double __testlib_nan() +{ + __TESTLIB_STATIC_ASSERT(sizeof(double) == sizeof(long long)); + long long llnan = 0xFFF8000000000000ll; + double nan; + std::memcpy(&nan, &llnan, sizeof(double)); + return nan; +} + +static bool __testlib_isInfinite(double r) +{ + volatile double ra = r; + return (ra > 1E100 || ra < -1E100); +} + +static void __testlib_set_binary(std::FILE* file) +{ +#ifdef O_BINARY + if (NULL != file) + { +#ifndef __BORLANDC__ + _setmode(_fileno(file), O_BINARY); +#else + setmode(fileno(file), O_BINARY); +#endif + } +#endif +} + +static FILE *__testlib_fopen(const char *name, const char *mode) +{ +#ifdef __EMSCRIPTEN__ + EM_ASM( + try { + FS.stat('/cwd'); + } catch (e) { + FS.mkdir('/cwd'); + FS.mount(NODEFS, { root: '.' }, '/cwd'); + } + ); + return fopen((std::string("/cwd/") + name).c_str(), mode); +#else + return fopen(name, mode); +#endif +} + +/* + * Very simple regex-like pattern. + * It used for two purposes: validation and generation. + * + * For example, pattern("[a-z]{1,5}").next(rnd) will return + * random string from lowercase latin letters with length + * from 1 to 5. It is easier to call rnd.next("[a-z]{1,5}") + * for the same effect. + * + * Another samples: + * "mike|john" will generate (match) "mike" or "john"; + * "-?[1-9][0-9]{0,3}" will generate (match) non-zero integers from -9999 to 9999; + * "id-([ac]|b{2})" will generate (match) "id-a", "id-bb", "id-c"; + * "[^0-9]*" will match sequences (empty or non-empty) without digits, you can't + * use it for generations. + * + * You can't use pattern for generation if it contains meta-symbol '*'. Also it + * is not recommended to use it for char-sets with meta-symbol '^' like [^a-z]. + * + * For matching very simple greedy algorithm is used. For example, pattern + * "[0-9]?1" will not match "1", because of greedy nature of matching. + * Alternations (meta-symbols "|") are processed with brute-force algorithm, so + * do not use many alternations in one expression. + * + * If you want to use one expression many times it is better to compile it into + * a single pattern like "pattern p("[a-z]+")". Later you can use + * "p.matches(std::string s)" or "p.next(random_t& rd)" to check matching or generate + * new string by pattern. + * + * Simpler way to read token and check it for pattern matching is "inf.readToken("[a-z]+")". + */ +class random_t; + +class pattern +{ +public: + /* Create pattern instance by string. */ + pattern(std::string s); + /* Generate new string by pattern and given random_t. */ + std::string next(random_t& rnd) const; + /* Checks if given string match the pattern. */ + bool matches(const std::string& s) const; + /* Returns source string of the pattern. */ + std::string src() const; +private: + bool matches(const std::string& s, size_t pos) const; + + std::string s; + std::vector children; + std::vector chars; + int from; + int to; +}; + +/* + * Use random_t instances to generate random values. It is preffered + * way to use randoms instead of rand() function or self-written + * randoms. + * + * Testlib defines global variable "rnd" of random_t class. + * Use registerGen(argc, argv, 1) to setup random_t seed be command + * line (to use latest random generator version). + * + * Random generates uniformly distributed values if another strategy is + * not specified explicitly. + */ +class random_t +{ +private: + unsigned long long seed; + static const unsigned long long multiplier; + static const unsigned long long addend; + static const unsigned long long mask; + static const int lim; + + long long nextBits(int bits) + { + if (bits <= 48) + { + seed = (seed * multiplier + addend) & mask; + return (long long)(seed >> (48 - bits)); + } + else + { + if (bits > 63) + __testlib_fail("random_t::nextBits(int bits): n must be less than 64"); + + int lowerBitCount = (random_t::version == 0 ? 31 : 32); + return ((nextBits(31) << 32) ^ nextBits(lowerBitCount)); + } + } + +public: + static int version; + + /* New random_t with fixed seed. */ + random_t() + : seed(3905348978240129619LL) + { + } + + /* Sets seed by command line. */ + void setSeed(int argc, char* argv[]) + { + random_t p; + + seed = 3905348978240129619LL; + for (int i = 1; i < argc; i++) + { + std::size_t le = std::strlen(argv[i]); + for (std::size_t j = 0; j < le; j++) + seed = seed * multiplier + (unsigned int)(argv[i][j]) + addend; + seed += multiplier / addend; + } + + seed = seed & mask; + } + + /* Sets seed by given value. */ + void setSeed(long long _seed) + { + _seed = (_seed ^ multiplier) & mask; + seed = _seed; + } + +#ifndef __BORLANDC__ + /* Random string value by given pattern (see pattern documentation). */ + std::string next(const std::string& ptrn) + { + pattern p(ptrn); + return p.next(*this); + } +#else + /* Random string value by given pattern (see pattern documentation). */ + std::string next(std::string ptrn) + { + pattern p(ptrn); + return p.next(*this); + } +#endif + + /* Random value in range [0, n-1]. */ + int next(int n) + { + if (n <= 0) + __testlib_fail("random_t::next(int n): n must be positive"); + + if ((n & -n) == n) // n is a power of 2 + return (int)((n * (long long)nextBits(31)) >> 31); + + const long long limit = INT_MAX / n * n; + + long long bits; + do { + bits = nextBits(31); + } while (bits >= limit); + + return int(bits % n); + } + + /* Random value in range [0, n-1]. */ + unsigned int next(unsigned int n) + { + if (n >= INT_MAX) + __testlib_fail("random_t::next(unsigned int n): n must be less INT_MAX"); + return (unsigned int)next(int(n)); + } + + /* Random value in range [0, n-1]. */ + long long next(long long n) + { + if (n <= 0) + __testlib_fail("random_t::next(long long n): n must be positive"); + + const long long limit = __TESTLIB_LONGLONG_MAX / n * n; + + long long bits; + do { + bits = nextBits(63); + } while (bits >= limit); + + return bits % n; + } + + /* Random value in range [0, n-1]. */ + unsigned long long next(unsigned long long n) + { + if (n >= (unsigned long long)(__TESTLIB_LONGLONG_MAX)) + __testlib_fail("random_t::next(unsigned long long n): n must be less LONGLONG_MAX"); + return (unsigned long long)next((long long)(n)); + } + + /* Random value in range [0, n-1]. */ + long next(long n) + { + return (long)next((long long)(n)); + } + + /* Random value in range [0, n-1]. */ + unsigned long next(unsigned long n) + { + if (n >= (unsigned long)(LONG_MAX)) + __testlib_fail("random_t::next(unsigned long n): n must be less LONG_MAX"); + return (unsigned long)next((unsigned long long)(n)); + } + + /* Returns random value in range [from,to]. */ + int next(int from, int to) + { + return int(next((long long)to - from + 1) + from); + } + + /* Returns random value in range [from,to]. */ + unsigned int next(unsigned int from, unsigned int to) + { + return (unsigned int)(next((long long)to - from + 1) + from); + } + + /* Returns random value in range [from,to]. */ + long long next(long long from, long long to) + { + return next(to - from + 1) + from; + } + + /* Returns random value in range [from,to]. */ + unsigned long long next(unsigned long long from, unsigned long long to) + { + if (from > to) + __testlib_fail("random_t::next(unsigned long long from, unsigned long long to): from can't not exceed to"); + return next(to - from + 1) + from; + } + + /* Returns random value in range [from,to]. */ + long next(long from, long to) + { + return next(to - from + 1) + from; + } + + /* Returns random value in range [from,to]. */ + unsigned long next(unsigned long from, unsigned long to) + { + if (from > to) + __testlib_fail("random_t::next(unsigned long from, unsigned long to): from can't not exceed to"); + return next(to - from + 1) + from; + } + + /* Random double value in range [0, 1). */ + double next() + { + return (double)(((long long)(nextBits(26)) << 27) + nextBits(27)) / (double)(1LL << 53); + } + + /* Random double value in range [0, n). */ + double next(double n) + { + return n * next(); + } + + /* Random double value in range [from, to). */ + double next(double from, double to) + { + return next(to - from) + from; + } + + /* Returns random element from container. */ + template + typename Container::value_type any(const Container& c) + { + size_t size = c.size(); + if (size <= 0) + __testlib_fail("random_t::any(const Container& c): c.size() must be positive"); + return *(c.begin() + next(size)); + } + + /* Returns random element from iterator range. */ + template + typename Iter::value_type any(const Iter& begin, const Iter& end) + { + int size = int(end - begin); + if (size <= 0) + __testlib_fail("random_t::any(const Iter& begin, const Iter& end): range must have positive length"); + return *(begin + next(size)); + } + + /* Random string value by given pattern (see pattern documentation). */ +#ifdef __GNUC__ + __attribute__ ((format (printf, 2, 3))) +#endif + std::string next(const char* format, ...) + { + FMT_TO_RESULT(format, format, ptrn); + return next(ptrn); + } + + /* + * Weighted next. If type == 0 than it is usual "next()". + * + * If type = 1, than it returns "max(next(), next())" + * (the number of "max" functions equals to "type"). + * + * If type < 0, than "max" function replaces with "min". + */ + int wnext(int n, int type) + { + if (n <= 0) + __testlib_fail("random_t::wnext(int n, int type): n must be positive"); + + if (abs(type) < random_t::lim) + { + int result = next(n); + + for (int i = 0; i < +type; i++) + result = __testlib_max(result, next(n)); + + for (int i = 0; i < -type; i++) + result = __testlib_min(result, next(n)); + + return result; + } + else + { + double p; + + if (type > 0) + p = std::pow(next() + 0.0, 1.0 / (type + 1)); + else + p = 1 - std::pow(next() + 0.0, 1.0 / (-type + 1)); + + return int(n * p); + } + } + + /* See wnext(int, int). It uses the same algorithms. */ + long long wnext(long long n, int type) + { + if (n <= 0) + __testlib_fail("random_t::wnext(long long n, int type): n must be positive"); + + if (abs(type) < random_t::lim) + { + long long result = next(n); + + for (int i = 0; i < +type; i++) + result = __testlib_max(result, next(n)); + + for (int i = 0; i < -type; i++) + result = __testlib_min(result, next(n)); + + return result; + } + else + { + double p; + + if (type > 0) + p = std::pow(next() + 0.0, 1.0 / (type + 1)); + else + p = std::pow(next() + 0.0, - type + 1); + + return __testlib_min(__testlib_max((long long)(double(n) * p), 0LL), n - 1LL); + } + } + + /* See wnext(int, int). It uses the same algorithms. */ + double wnext(int type) + { + if (abs(type) < random_t::lim) + { + double result = next(); + + for (int i = 0; i < +type; i++) + result = __testlib_max(result, next()); + + for (int i = 0; i < -type; i++) + result = __testlib_min(result, next()); + + return result; + } + else + { + double p; + + if (type > 0) + p = std::pow(next() + 0.0, 1.0 / (type + 1)); + else + p = std::pow(next() + 0.0, - type + 1); + + return p; + } + } + + /* See wnext(int, int). It uses the same algorithms. */ + double wnext(double n, int type) + { + if (n <= 0) + __testlib_fail("random_t::wnext(double n, int type): n must be positive"); + + if (abs(type) < random_t::lim) + { + double result = next(); + + for (int i = 0; i < +type; i++) + result = __testlib_max(result, next()); + + for (int i = 0; i < -type; i++) + result = __testlib_min(result, next()); + + return n * result; + } + else + { + double p; + + if (type > 0) + p = std::pow(next() + 0.0, 1.0 / (type + 1)); + else + p = std::pow(next() + 0.0, - type + 1); + + return n * p; + } + } + + /* See wnext(int, int). It uses the same algorithms. */ + unsigned int wnext(unsigned int n, int type) + { + if (n >= INT_MAX) + __testlib_fail("random_t::wnext(unsigned int n, int type): n must be less INT_MAX"); + return (unsigned int)wnext(int(n), type); + } + + /* See wnext(int, int). It uses the same algorithms. */ + unsigned long long wnext(unsigned long long n, int type) + { + if (n >= (unsigned long long)(__TESTLIB_LONGLONG_MAX)) + __testlib_fail("random_t::wnext(unsigned long long n, int type): n must be less LONGLONG_MAX"); + + return (unsigned long long)wnext((long long)(n), type); + } + + /* See wnext(int, int). It uses the same algorithms. */ + long wnext(long n, int type) + { + return (long)wnext((long long)(n), type); + } + + /* See wnext(int, int). It uses the same algorithms. */ + unsigned long wnext(unsigned long n, int type) + { + if (n >= (unsigned long)(LONG_MAX)) + __testlib_fail("random_t::wnext(unsigned long n, int type): n must be less LONG_MAX"); + + return (unsigned long)wnext((unsigned long long)(n), type); + } + + /* Returns weighted random value in range [from, to]. */ + int wnext(int from, int to, int type) + { + return wnext(to - from + 1, type) + from; + } + + /* Returns weighted random value in range [from, to]. */ + int wnext(unsigned int from, unsigned int to, int type) + { + return int(wnext(to - from + 1, type) + from); + } + + /* Returns weighted random value in range [from, to]. */ + long long wnext(long long from, long long to, int type) + { + return wnext(to - from + 1, type) + from; + } + + /* Returns weighted random value in range [from, to]. */ + unsigned long long wnext(unsigned long long from, unsigned long long to, int type) + { + if (from > to) + __testlib_fail("random_t::wnext(unsigned long long from, unsigned long long to, int type): from can't not exceed to"); + return wnext(to - from + 1, type) + from; + } + + /* Returns weighted random value in range [from, to]. */ + long wnext(long from, long to, int type) + { + return wnext(to - from + 1, type) + from; + } + + /* Returns weighted random value in range [from, to]. */ + unsigned long wnext(unsigned long from, unsigned long to, int type) + { + if (from > to) + __testlib_fail("random_t::wnext(unsigned long from, unsigned long to, int type): from can't not exceed to"); + return wnext(to - from + 1, type) + from; + } + + /* Returns weighted random double value in range [from, to). */ + double wnext(double from, double to, int type) + { + return wnext(to - from, type) + from; + } + + /* Returns weighted random element from container. */ + template + typename Container::value_type wany(const Container& c, int type) + { + size_t size = c.size(); + if (size <= 0) + __testlib_fail("random_t::wany(const Container& c, int type): c.size() must be positive"); + return *(c.begin() + wnext(size, type)); + } + + /* Returns weighted random element from iterator range. */ + template + typename Iter::value_type wany(const Iter& begin, const Iter& end, int type) + { + int size = int(end - begin); + if (size <= 0) + __testlib_fail("random_t::any(const Iter& begin, const Iter& end, int type): range must have positive length"); + return *(begin + wnext(size, type)); + } +}; + +const int random_t::lim = 25; +const unsigned long long random_t::multiplier = 0x5DEECE66DLL; +const unsigned long long random_t::addend = 0xBLL; +const unsigned long long random_t::mask = (1LL << 48) - 1; +int random_t::version = -1; + +/* Pattern implementation */ +bool pattern::matches(const std::string& s) const +{ + return matches(s, 0); +} + +static bool __pattern_isSlash(const std::string& s, size_t pos) +{ + return s[pos] == '\\'; +} + +#ifdef __GNUC__ +__attribute__((pure)) +#endif +static bool __pattern_isCommandChar(const std::string& s, size_t pos, char value) +{ + if (pos >= s.length()) + return false; + + int slashes = 0; + + int before = int(pos) - 1; + while (before >= 0 && s[before] == '\\') + before--, slashes++; + + return slashes % 2 == 0 && s[pos] == value; +} + +static char __pattern_getChar(const std::string& s, size_t& pos) +{ + if (__pattern_isSlash(s, pos)) + pos += 2; + else + pos++; + + return s[pos - 1]; +} + +#ifdef __GNUC__ +__attribute__((pure)) +#endif +static int __pattern_greedyMatch(const std::string& s, size_t pos, const std::vector chars) +{ + int result = 0; + + while (pos < s.length()) + { + char c = s[pos++]; + if (!std::binary_search(chars.begin(), chars.end(), c)) + break; + else + result++; + } + + return result; +} + +std::string pattern::src() const +{ + return s; +} + +bool pattern::matches(const std::string& s, size_t pos) const +{ + std::string result; + + if (to > 0) + { + int size = __pattern_greedyMatch(s, pos, chars); + if (size < from) + return false; + if (size > to) + size = to; + pos += size; + } + + if (children.size() > 0) + { + for (size_t child = 0; child < children.size(); child++) + if (children[child].matches(s, pos)) + return true; + return false; + } + else + return pos == s.length(); +} + +std::string pattern::next(random_t& rnd) const +{ + std::string result; + result.reserve(20); + + if (to == INT_MAX) + __testlib_fail("pattern::next(random_t& rnd): can't process character '*' for generation"); + + if (to > 0) + { + int count = rnd.next(to - from + 1) + from; + for (int i = 0; i < count; i++) + result += chars[rnd.next(int(chars.size()))]; + } + + if (children.size() > 0) + { + int child = rnd.next(int(children.size())); + result += children[child].next(rnd); + } + + return result; +} + +static void __pattern_scanCounts(const std::string& s, size_t& pos, int& from, int& to) +{ + if (pos >= s.length()) + { + from = to = 1; + return; + } + + if (__pattern_isCommandChar(s, pos, '{')) + { + std::vector parts; + std::string part; + + pos++; + + while (pos < s.length() && !__pattern_isCommandChar(s, pos, '}')) + { + if (__pattern_isCommandChar(s, pos, ',')) + parts.push_back(part), part = "", pos++; + else + part += __pattern_getChar(s, pos); + } + + if (part != "") + parts.push_back(part); + + if (!__pattern_isCommandChar(s, pos, '}')) + __testlib_fail("pattern: Illegal pattern (or part) \"" + s + "\""); + + pos++; + + if (parts.size() < 1 || parts.size() > 2) + __testlib_fail("pattern: Illegal pattern (or part) \"" + s + "\""); + + std::vector numbers; + + for (size_t i = 0; i < parts.size(); i++) + { + if (parts[i].length() == 0) + __testlib_fail("pattern: Illegal pattern (or part) \"" + s + "\""); + int number; + if (std::sscanf(parts[i].c_str(), "%d", &number) != 1) + __testlib_fail("pattern: Illegal pattern (or part) \"" + s + "\""); + numbers.push_back(number); + } + + if (numbers.size() == 1) + from = to = numbers[0]; + else + from = numbers[0], to = numbers[1]; + + if (from > to) + __testlib_fail("pattern: Illegal pattern (or part) \"" + s + "\""); + } + else + { + if (__pattern_isCommandChar(s, pos, '?')) + { + from = 0, to = 1, pos++; + return; + } + + if (__pattern_isCommandChar(s, pos, '*')) + { + from = 0, to = INT_MAX, pos++; + return; + } + + if (__pattern_isCommandChar(s, pos, '+')) + { + from = 1, to = INT_MAX, pos++; + return; + } + + from = to = 1; + } +} + +static std::vector __pattern_scanCharSet(const std::string& s, size_t& pos) +{ + if (pos >= s.length()) + __testlib_fail("pattern: Illegal pattern (or part) \"" + s + "\""); + + std::vector result; + + if (__pattern_isCommandChar(s, pos, '[')) + { + pos++; + bool negative = __pattern_isCommandChar(s, pos, '^'); + + char prev = 0; + + while (pos < s.length() && !__pattern_isCommandChar(s, pos, ']')) + { + if (__pattern_isCommandChar(s, pos, '-') && prev != 0) + { + pos++; + + if (pos + 1 == s.length() || __pattern_isCommandChar(s, pos, ']')) + { + result.push_back(prev); + prev = '-'; + continue; + } + + char next = __pattern_getChar(s, pos); + if (prev > next) + __testlib_fail("pattern: Illegal pattern (or part) \"" + s + "\""); + + for (char c = prev; c != next; c++) + result.push_back(c); + result.push_back(next); + + prev = 0; + } + else + { + if (prev != 0) + result.push_back(prev); + prev = __pattern_getChar(s, pos); + } + } + + if (prev != 0) + result.push_back(prev); + + if (!__pattern_isCommandChar(s, pos, ']')) + __testlib_fail("pattern: Illegal pattern (or part) \"" + s + "\""); + + pos++; + + if (negative) + { + std::sort(result.begin(), result.end()); + std::vector actuals; + for (int code = 0; code < 255; code++) + { + char c = char(code); + if (!std::binary_search(result.begin(), result.end(), c)) + actuals.push_back(c); + } + result = actuals; + } + + std::sort(result.begin(), result.end()); + } + else + result.push_back(__pattern_getChar(s, pos)); + + return result; +} + +pattern::pattern(std::string s): s(s), from(0), to(0) +{ + std::string t; + for (size_t i = 0; i < s.length(); i++) + if (!__pattern_isCommandChar(s, i, ' ')) + t += s[i]; + s = t; + + int opened = 0; + int firstClose = -1; + std::vector seps; + + for (size_t i = 0; i < s.length(); i++) + { + if (__pattern_isCommandChar(s, i, '(')) + { + opened++; + continue; + } + + if (__pattern_isCommandChar(s, i, ')')) + { + opened--; + if (opened == 0 && firstClose == -1) + firstClose = int(i); + continue; + } + + if (opened < 0) + __testlib_fail("pattern: Illegal pattern (or part) \"" + s + "\""); + + if (__pattern_isCommandChar(s, i, '|') && opened == 0) + seps.push_back(int(i)); + } + + if (opened != 0) + __testlib_fail("pattern: Illegal pattern (or part) \"" + s + "\""); + + if (seps.size() == 0 && firstClose + 1 == (int)s.length() + && __pattern_isCommandChar(s, 0, '(') && __pattern_isCommandChar(s, s.length() - 1, ')')) + { + children.push_back(pattern(s.substr(1, s.length() - 2))); + } + else + { + if (seps.size() > 0) + { + seps.push_back(s.length()); + int last = 0; + + for (size_t i = 0; i < seps.size(); i++) + { + children.push_back(pattern(s.substr(last, seps[i] - last))); + last = seps[i] + 1; + } + } + else + { + size_t pos = 0; + chars = __pattern_scanCharSet(s, pos); + __pattern_scanCounts(s, pos, from, to); + if (pos < s.length()) + children.push_back(pattern(s.substr(pos))); + } + } +} +/* End of pattern implementation */ + +template +inline bool isEof(C c) +{ + return (c == EOF || c == EOFC); +} + +template +inline bool isEoln(C c) +{ + return (c == LF || c == CR); +} + +template +inline bool isBlanks(C c) +{ + return (c == LF || c == CR || c == SPACE || c == TAB); +} + +enum TMode +{ + _input, _output, _answer +}; + +/* Outcomes 6-15 are reserved for future use. */ +enum TResult +{ + _ok = 0, + _wa = 1, + _pe = 2, + _fail = 3, + _dirt = 4, + _points = 5, + _unexpected_eof = 8, + _partially = 16 +}; + +enum TTestlibMode +{ + _unknown, _checker, _validator, _generator, _interactor +}; + +#define _pc(exitCode) (TResult(_partially + (exitCode))) + +/* Outcomes 6-15 are reserved for future use. */ +const std::string outcomes[] = { + "accepted", + "wrong-answer", + "presentation-error", + "fail", + "fail", +#ifndef PCMS2 + "points", +#else + "relative-scoring", +#endif + "reserved", + "reserved", + "unexpected-eof", + "reserved", + "reserved", + "reserved", + "reserved", + "reserved", + "reserved", + "reserved", + "partially-correct" +}; + +class InputStreamReader +{ +public: + virtual int curChar() = 0; + virtual int nextChar() = 0; + virtual void skipChar() = 0; + virtual void unreadChar(int c) = 0; + virtual std::string getName() = 0; + virtual bool eof() = 0; + virtual ~InputStreamReader() = 0; +}; + +InputStreamReader::~InputStreamReader() +{ + // No operations. +} + +class StringInputStreamReader: public InputStreamReader +{ +private: + std::string s; + size_t pos; + +public: + StringInputStreamReader(const std::string& content): s(content), pos(0) + { + // No operations. + } + + int curChar() + { + if (pos >= s.length()) + return EOF; + else + { + return s[pos]; + } + } + + int nextChar() + { + if (pos >= s.length()) + return EOF; + else + return s[pos++]; + } + + void skipChar() + { + pos++; + } + + void unreadChar(int c) + { + if (pos == 0) + __testlib_fail("FileFileInputStreamReader::unreadChar(int): pos == 0."); + pos--; + if (pos < s.length()) + s[pos] = char(c); + } + + std::string getName() + { + return __testlib_part(s); + } + + bool eof() + { + return pos >= s.length(); + } +}; + +class FileInputStreamReader: public InputStreamReader +{ +private: + std::FILE* file; + std::string name; + +public: + FileInputStreamReader(std::FILE* file, const std::string& name): file(file), name(name) + { + // No operations. + } + + int curChar() + { + if (feof(file)) + return EOF; + else + { + int c = getc(file); + ungetc(c, file); + return c; + } + } + + int nextChar() + { + if (feof(file)) + return EOF; + else + return getc(file); + } + + void skipChar() + { + getc(file); + } + + void unreadChar(int c) + { + ungetc(c, file); + } + + std::string getName() + { + return name; + } + + bool eof() + { + if (feof(file)) + return true; + else + { + int c = getc(file); + bool result = (c == EOF); + ungetc(c, file); + return result; + } + } +}; + +class BufferedFileInputStreamReader: public InputStreamReader +{ +private: + static const size_t BUFFER_SIZE; + static const size_t MAX_UNREAD_COUNT; + + std::FILE* file; + char* buffer; + bool* isEof; + int bufferPos; + size_t bufferSize; + + std::string name; + + bool refill() + { + if (NULL == file) + __testlib_fail("BufferedFileInputStreamReader: file == NULL (" + getName() + ")"); + + if (bufferPos >= int(bufferSize)) + { + size_t readSize = fread( + buffer + MAX_UNREAD_COUNT, + 1, + BUFFER_SIZE - MAX_UNREAD_COUNT, + file + ); + + if (readSize < BUFFER_SIZE - MAX_UNREAD_COUNT + && ferror(file)) + __testlib_fail("BufferedFileInputStreamReader: unable to read (" + getName() + ")"); + + bufferSize = MAX_UNREAD_COUNT + readSize; + bufferPos = int(MAX_UNREAD_COUNT); + std::memset(isEof + MAX_UNREAD_COUNT, 0, sizeof(isEof[0]) * readSize); + + return readSize > 0; + } + else + return true; + } + +public: + BufferedFileInputStreamReader(std::FILE* file, const std::string& name): file(file), name(name) + { + buffer = new char[BUFFER_SIZE]; + isEof = new bool[BUFFER_SIZE]; + bufferSize = MAX_UNREAD_COUNT; + bufferPos = int(MAX_UNREAD_COUNT); + } + + ~BufferedFileInputStreamReader() + { + if (NULL != buffer) + { + delete[] buffer; + buffer = NULL; + } + if (NULL != isEof) + { + delete[] isEof; + isEof = NULL; + } + } + + int curChar() + { + if (!refill()) + return EOF; + + return isEof[bufferPos] ? EOF : buffer[bufferPos]; + } + + int nextChar() + { + if (!refill()) + return EOF; + + return isEof[bufferPos] ? EOF : buffer[bufferPos++]; + } + + void skipChar() + { + bufferPos++; + } + + void unreadChar(int c) + { + bufferPos--; + if (bufferPos < 0) + __testlib_fail("BufferedFileInputStreamReader::unreadChar(int): bufferPos < 0"); + isEof[bufferPos] = (c == EOF); + buffer[bufferPos] = char(c == EOF ? EOFC : c); + } + + std::string getName() + { + return name; + } + + bool eof() + { + return !refill() || EOF == curChar(); + } +}; + +const size_t BufferedFileInputStreamReader::BUFFER_SIZE = 1000000; +const size_t BufferedFileInputStreamReader::MAX_UNREAD_COUNT = 100; + +/* + * Streams to be used for reading data in checkers or validators. + * Each read*() method moves pointer to the next character after the + * read value. + */ +struct InStream +{ + /* Do not use them. */ + InStream(); + ~InStream(); + + /* Wrap std::string with InStream. */ + InStream(const InStream& baseStream, std::string content); + + InputStreamReader* reader; + + std::FILE* file; + std::string name; + TMode mode; + bool opened; + bool stdfile; + bool strict; + + int wordReserveSize; + std::string _tmpReadToken; + + int readManyIteration; + + void init(std::string fileName, TMode mode); + void init(std::FILE* f, TMode mode); + + /* Moves stream pointer to the first non-white-space character or EOF. */ + void skipBlanks(); + + /* Returns current character in the stream. Doesn't remove it from stream. */ + char curChar(); + /* Moves stream pointer one character forward. */ + void skipChar(); + /* Returns current character and moves pointer one character forward. */ + char nextChar(); + + /* Returns current character and moves pointer one character forward. */ + char readChar(); + /* As "readChar()" but ensures that the result is equal to given parameter. */ + char readChar(char c); + /* As "readChar()" but ensures that the result is equal to the space (code=32). */ + char readSpace(); + /* Puts back the character into the stream. */ + void unreadChar(char c); + + /* Reopens stream, you should not use it. */ + void reset(); + /* Checks that current position is EOF. If not it doesn't move stream pointer. */ + bool eof(); + /* Moves pointer to the first non-white-space character and calls "eof()". */ + bool seekEof(); + + /* + * Checks that current position contains EOLN. + * If not it doesn't move stream pointer. + * In strict mode expects "#13#10" for windows or "#10" for other platforms. + */ + bool eoln(); + /* Moves pointer to the first non-space and non-tab character and calls "eoln()". */ + bool seekEoln(); + + /* Moves stream pointer to the first character of the next line (if exists). */ + void nextLine(); + + /* + * Reads new token. Ignores white-spaces into the non-strict mode + * (strict mode is used in validators usually). + */ + std::string readWord(); + /* The same as "readWord()", it is preffered to use "readToken()". */ + std::string readToken(); + /* The same as "readWord()", but ensures that token matches to given pattern. */ + std::string readWord(const std::string& ptrn, const std::string& variableName = ""); + std::string readWord(const pattern& p, const std::string& variableName = ""); + /* The same as "readToken()", but ensures that token matches to given pattern. */ + std::string readToken(const std::string& ptrn, const std::string& variableName = ""); + std::string readToken(const pattern& p, const std::string& variableName = ""); + + void readWordTo(std::string& result); + void readWordTo(std::string& result, const pattern& p, const std::string& variableName = ""); + void readWordTo(std::string& result, const std::string& ptrn, const std::string& variableName = ""); + + void readTokenTo(std::string& result); + void readTokenTo(std::string& result, const pattern& p, const std::string& variableName = ""); + void readTokenTo(std::string& result, const std::string& ptrn, const std::string& variableName = ""); + + /* + * Reads new long long value. Ignores white-spaces into the non-strict mode + * (strict mode is used in validators usually). + */ + long long readLong(); + /* + * Reads new int. Ignores white-spaces into the non-strict mode + * (strict mode is used in validators usually). + */ + int readInteger(); + /* + * Reads new int. Ignores white-spaces into the non-strict mode + * (strict mode is used in validators usually). + */ + int readInt(); + + /* As "readLong()" but ensures that value in the range [minv,maxv]. */ + long long readLong(long long minv, long long maxv, const std::string& variableName = ""); + /* As "readInteger()" but ensures that value in the range [minv,maxv]. */ + int readInteger(int minv, int maxv, const std::string& variableName = ""); + /* As "readInt()" but ensures that value in the range [minv,maxv]. */ + + int readInt(int minv, int maxv, const std::string& variableName = ""); + + /* Reads space-separated sequence of integers. */ + std::vector + readIntegers(int size, int minv, int maxv, const std::string &variablesName = "", int indexBase = 1); + + /* Reads space-separated sequence of integers. */ + std::vector readIntegers(int size, int indexBase = 1); + + /* Reads space-separated sequence of integers. */ + std::vector readInts(int size, int minv, int maxv, const std::string &variablesName = "", int indexBase = 1); + + /* Reads space-separated sequence of integers. */ + std::vector readInts(int size, int indexBase = 1); + + + /* + * Reads new double. Ignores white-spaces into the non-strict mode + * (strict mode is used in validators usually). + */ + double readReal(); + /* + * Reads new double. Ignores white-spaces into the non-strict mode + * (strict mode is used in validators usually). + */ + double readDouble(); + + /* As "readReal()" but ensures that value in the range [minv,maxv]. */ + double readReal(double minv, double maxv, const std::string& variableName = ""); + /* As "readDouble()" but ensures that value in the range [minv,maxv]. */ + double readDouble(double minv, double maxv, const std::string& variableName = ""); + + std::vector + readReals(int size, double minv, double maxv, const std::string &variablesName = "", int indexBase = 1); + + std::vector readReals(int size, int indexBase = 1); + + std::vector + readDoubles(int size, double minv, double maxv, const std::string &variablesName = "", int indexBase = 1); + + std::vector readDoubles(int size, int indexBase = 1); + + /* + * As "readReal()" but ensures that value in the range [minv,maxv] and + * number of digit after the decimal point is in range [minAfterPointDigitCount,maxAfterPointDigitCount] + * and number is in the form "[-]digit(s)[.digit(s)]". + */ + double readStrictReal(double minv, double maxv, + int minAfterPointDigitCount, int maxAfterPointDigitCount, + const std::string& variableName = ""); + + std::vector readStrictReals(int size, double minv, double maxv, + int minAfterPointDigitCount, int maxAfterPointDigitCount, + const std::string &variablesName = "", int indexBase = 1); + /* + * As "readDouble()" but ensures that value in the range [minv,maxv] and + * number of digit after the decimal point is in range [minAfterPointDigitCount,maxAfterPointDigitCount] + * and number is in the form "[-]digit(s)[.digit(s)]". + */ + double readStrictDouble(double minv, double maxv, + int minAfterPointDigitCount, int maxAfterPointDigitCount, + const std::string& variableName = ""); + + std::vector readStrictDoubles(int size, double minv, double maxv, + int minAfterPointDigitCount, int maxAfterPointDigitCount, + const std::string &variablesName = "", int indexBase = 1); + + /* As readLine(). */ + std::string readString(); + /* See readLine(). */ + void readStringTo(std::string& result); + /* The same as "readLine()/readString()", but ensures that line matches to the given pattern. */ + std::string readString(const pattern& p, const std::string& variableName = ""); + /* The same as "readLine()/readString()", but ensures that line matches to the given pattern. */ + std::string readString(const std::string& ptrn, const std::string& variableName = ""); + /* The same as "readLine()/readString()", but ensures that line matches to the given pattern. */ + void readStringTo(std::string& result, const pattern& p, const std::string& variableName = ""); + /* The same as "readLine()/readString()", but ensures that line matches to the given pattern. */ + void readStringTo(std::string& result, const std::string& ptrn, const std::string& variableName = ""); + + /* Read many lines. */ + std::vector + readStrings(int size, const pattern &p, const std::string &variableName = "", int indexBase = 1); + + /* Read many lines. */ + std::vector + readStrings(int size, const std::string &ptrn, const std::string &variableName = "", int indexBase = 1); + + /* + * Reads line from the current position to EOLN or EOF. Moves stream pointer to + * the first character of the new line (if possible). + */ + std::string readLine(); + /* See readLine(). */ + void readLineTo(std::string& result); + /* The same as "readLine()", but ensures that line matches to the given pattern. */ + std::string readLine(const pattern& p, const std::string& variableName = ""); + /* The same as "readLine()", but ensures that line matches to the given pattern. */ + std::string readLine(const std::string& ptrn, const std::string& variableName = ""); + /* The same as "readLine()", but ensures that line matches to the given pattern. */ + void readLineTo(std::string& result, const pattern& p, const std::string& variableName = ""); + /* The same as "readLine()", but ensures that line matches to the given pattern. */ + void readLineTo(std::string& result, const std::string& ptrn, const std::string& variableName = ""); + + std::vector + readLines(int size, const pattern &p, const std::string &variableName = "", int indexBase = 1); + + /* Read many lines. */ + std::vector + readLines(int size, const std::string &ptrn, const std::string &variableName = "", int indexBase = 1); + + /* Reads EOLN or fails. Use it in validators. Calls "eoln()" method internally. */ + void readEoln(); + /* Reads EOF or fails. Use it in validators. Calls "eof()" method internally. */ + void readEof(); + + /* + * Quit-functions aborts program with and : + * input/answer streams replace any result to FAIL. + */ + NORETURN void quit(TResult result, const char* msg); + /* + * Quit-functions aborts program with and : + * input/answer streams replace any result to FAIL. + */ + NORETURN void quitf(TResult result, const char* msg, ...); + /* + * Quit-functions aborts program with and : + * input/answer streams replace any result to FAIL. + */ + NORETURN void quits(TResult result, std::string msg); + + void close(); + + const static WORD LightGray = 0x07; + const static WORD LightRed = 0x0c; + const static WORD LightCyan = 0x0b; + const static WORD LightGreen = 0x0a; + const static WORD LightYellow = 0x0e; + + static void textColor(WORD color); + static void quitscr(WORD color, const char* msg); + static void quitscrS(WORD color, std::string msg); + void xmlSafeWrite(std::FILE * file, const char* msg); + +private: + InStream(const InStream&); + InStream& operator =(const InStream&); +}; + +InStream inf; +InStream ouf; +InStream ans; +bool appesMode; +std::string resultName; +std::string checkerName = "untitled checker"; +random_t rnd; +TTestlibMode testlibMode = _unknown; +double __testlib_points = std::numeric_limits::infinity(); + +struct TestlibFinalizeGuard +{ + static bool alive; + int quitCount, readEofCount; + + TestlibFinalizeGuard() : quitCount(0), readEofCount(0) + { + // No operations. + } + + ~TestlibFinalizeGuard() + { + bool _alive = alive; + alive = false; + + if (_alive) + { + if (testlibMode == _checker && quitCount == 0) + __testlib_fail("Checker must end with quit or quitf call."); + + if (testlibMode == _validator && readEofCount == 0 && quitCount == 0) + __testlib_fail("Validator must end with readEof call."); + } + } +}; + +bool TestlibFinalizeGuard::alive = true; +TestlibFinalizeGuard testlibFinalizeGuard; + +/* + * Call it to disable checks on finalization. + */ +void disableFinalizeGuard() +{ + TestlibFinalizeGuard::alive = false; +} + +/* Interactor streams. + */ +std::fstream tout; + +/* implementation + */ + +template +static std::string vtos(const T& t) +{ + std::string s; + std::stringstream ss; + ss << t; + ss >> s; + return s; +} + +template +static std::string toString(const T& t) +{ + return vtos(t); +} + +const static int NO_INDEX = INT_MAX; + +InStream::InStream() +{ + file = NULL; + name = ""; + mode = _input; + strict = false; + stdfile = false; + wordReserveSize = 4; + readManyIteration = NO_INDEX; +} + +InStream::InStream(const InStream& baseStream, std::string content) +{ + file = (FILE*)0xbadfeed; + stdfile = false; + reader = new StringInputStreamReader(content); + opened = true; + strict = baseStream.strict; + mode = baseStream.mode; + name = "based on " + baseStream.name; +} + +InStream::~InStream() +{ + if (NULL != reader) + { + delete reader; + reader = NULL; + } +} + +#ifdef __GNUC__ +__attribute__((const)) +#endif +int resultExitCode(TResult r) +{ + if (r == _ok) + return OK_EXIT_CODE; + if (r == _wa) + return WA_EXIT_CODE; + if (r == _pe) + return PE_EXIT_CODE; + if (r == _fail) + return FAIL_EXIT_CODE; + if (r == _dirt) + return DIRT_EXIT_CODE; + if (r == _points) + return POINTS_EXIT_CODE; + if (r == _unexpected_eof) +#ifdef ENABLE_UNEXPECTED_EOF + return UNEXPECTED_EOF_EXIT_CODE; +#else + return PE_EXIT_CODE; +#endif + if (r >= _partially) + return PC_BASE_EXIT_CODE + (r - _partially); + return FAIL_EXIT_CODE; +} + +void InStream::textColor(WORD color) +{ +#if defined(ON_WINDOWS) && (!defined(_MSC_VER) || _MSC_VER>1400) + HANDLE handle = GetStdHandle(STD_OUTPUT_HANDLE); + SetConsoleTextAttribute(handle, color); +#endif + +#ifdef linux + char *shell_path = getenv("SHELL"); + if (shell_path && strcmp(shell_path, "/bin/bash") == 0 && isatty(2)) + { + switch (color) + { + case 0: + fprintf(stderr, "\e[0m"); + break; + case LightRed: + fprintf(stderr, "\e[0;31m"); + break; + case LightGreen: + fprintf(stderr, "\e[0;32m"); + break; + case LightYellow: + fprintf(stderr, "\e[0;33m"); + break; + case LightCyan: + fprintf(stderr, "\e[0;36m"); + break; + case LightGray: + fprintf(stderr, "\e[0m"); + break; + } + } +#endif +} + +NORETURN void halt(int exitCode) +{ +#ifdef FOOTER + InStream::textColor(InStream::LightGray); + std::fprintf(stderr, "Checker: \"%s\"\n", checkerName.c_str()); + std::fprintf(stderr, "Exit code: %d\n", exitCode); + InStream::textColor(InStream::LightGray); +#endif +#ifdef linux + InStream::textColor(0); +#endif + std::exit(exitCode); +} + +NORETURN void InStream::quit(TResult result, const char* msg) +{ + if (TestlibFinalizeGuard::alive) + testlibFinalizeGuard.quitCount++; + + if (mode != _output && result != _fail) + quits(_fail, std::string(msg) + " (" + name + ")"); + + std::FILE * resultFile; + std::string errorName; + + if (result == _ok) + { + if (testlibMode != _interactor && !ouf.seekEof()) + quit(_dirt, "Extra information in the output file"); + } + + int pctype = result - _partially; + bool isPartial = false; + + switch (result) + { + case _ok: + errorName = "ok "; + quitscrS(LightGreen, errorName); + break; + case _wa: + errorName = "wrong answer "; + quitscrS(LightRed, errorName); + break; + case _pe: + errorName = "wrong output format "; + quitscrS(LightRed, errorName); + break; + case _fail: + errorName = "FAIL "; + quitscrS(LightRed, errorName); + break; + case _dirt: + errorName = "wrong output format "; + quitscrS(LightCyan, errorName); + result = _pe; + break; + case _points: + errorName = "points "; + quitscrS(LightYellow, errorName); + break; + case _unexpected_eof: +#ifdef ENABLE_UNEXPECTED_EOF + errorName = "unexpected eof "; +#else + errorName = "wrong output format "; +#endif + quitscrS(LightCyan, errorName); + break; + default: + if (result >= _partially) + { + errorName = format("partially correct (%d) ", pctype); + isPartial = true; + quitscrS(LightYellow, errorName); + } + else + quit(_fail, "What is the code ??? "); + } + + if (resultName != "") + { + resultFile = __testlib_fopen(resultName.c_str(), "w"); + if (resultFile == NULL) + quit(_fail, "Can not write to the result file"); + if (appesMode) + { + std::fprintf(resultFile, ""); + if (isPartial) + std::fprintf(resultFile, "", outcomes[(int)_partially].c_str(), pctype); + else + { + if (result != _points) + std::fprintf(resultFile, "", outcomes[(int)result].c_str()); + else + { + if (__testlib_points == std::numeric_limits::infinity()) + quit(_fail, "Expected points, but infinity found"); + std::string stringPoints = removeDoubleTrailingZeroes(format("%.10f", __testlib_points)); + std::fprintf(resultFile, "", outcomes[(int)result].c_str(), stringPoints.c_str()); + } + } + xmlSafeWrite(resultFile, msg); + std::fprintf(resultFile, "\n"); + } + else + std::fprintf(resultFile, "%s", msg); + if (NULL == resultFile || fclose(resultFile) != 0) + quit(_fail, "Can not write to the result file"); + } + + quitscr(LightGray, msg); + std::fprintf(stderr, "\n"); + + if (inf.file) + fclose(inf.file); + if (ouf.file) + fclose(ouf.file); + if (ans.file) + fclose(ans.file); + if (tout.is_open()) + tout.close(); + + textColor(LightGray); + + if (resultName != "") + std::fprintf(stderr, "See file to check exit message\n"); + + halt(resultExitCode(result)); +} + +#ifdef __GNUC__ + __attribute__ ((format (printf, 3, 4))) +#endif +NORETURN void InStream::quitf(TResult result, const char* msg, ...) +{ + FMT_TO_RESULT(msg, msg, message); + InStream::quit(result, message.c_str()); +} + +NORETURN void InStream::quits(TResult result, std::string msg) +{ + InStream::quit(result, msg.c_str()); +} + +void InStream::xmlSafeWrite(std::FILE * file, const char* msg) +{ + size_t lmsg = strlen(msg); + for (size_t i = 0; i < lmsg; i++) + { + if (msg[i] == '&') + { + std::fprintf(file, "%s", "&"); + continue; + } + if (msg[i] == '<') + { + std::fprintf(file, "%s", "<"); + continue; + } + if (msg[i] == '>') + { + std::fprintf(file, "%s", ">"); + continue; + } + if (msg[i] == '"') + { + std::fprintf(file, "%s", """); + continue; + } + if (0 <= msg[i] && msg[i] <= 31) + { + std::fprintf(file, "%c", '.'); + continue; + } + std::fprintf(file, "%c", msg[i]); + } +} + +void InStream::quitscrS(WORD color, std::string msg) +{ + quitscr(color, msg.c_str()); +} + +void InStream::quitscr(WORD color, const char* msg) +{ + if (resultName == "") + { + textColor(color); + std::fprintf(stderr, "%s", msg); + textColor(LightGray); + } +} + +void InStream::reset() +{ + if (opened && stdfile) + quit(_fail, "Can't reset standard handle"); + + if (opened) + close(); + + if (!stdfile) + if (NULL == (file = __testlib_fopen(name.c_str(), "rb"))) + { + if (mode == _output) + quits(_pe, std::string("File not found: \"") + name + "\""); + } + + opened = true; + + __testlib_set_binary(file); + + if (stdfile) + reader = new FileInputStreamReader(file, name); + else + reader = new BufferedFileInputStreamReader(file, name); +} + +void InStream::init(std::string fileName, TMode mode) +{ + opened = false; + if (fileName == "/dev/stdin") + { + name = "stdin"; + this->file = stdin; + stdfile = true; + } + else + { + name = fileName; + stdfile = false; + } + this->mode = mode; + reset(); +} + +void InStream::init(std::FILE* f, TMode mode) +{ + opened = false; + + name = "untitled"; + + if (f == stdin) + name = "stdin", stdfile = true; + + if (f == stdout) + name = "stdout", stdfile = true; + + if (f == stderr) + name = "stderr", stdfile = true; + + this->file = f; + this->mode = mode; + + reset(); +} + +char InStream::curChar() +{ + int c = reader->curChar(); + return char(c != EOF ? c : EOFC); +} + +char InStream::nextChar() +{ + int c = reader->nextChar(); + return char(c != EOF ? c : EOFC); +} + +char InStream::readChar() +{ + return nextChar(); +} + +char InStream::readChar(char c) +{ + char found = readChar(); + if (c != found) + { + if (!isEoln(found)) + quit(_pe, ("Unexpected character '" + std::string(1, found) + "', but '" + std::string(1, c) + "' expected").c_str()); + else + quit(_pe, ("Unexpected character " + ("#" + vtos(int(found))) + ", but '" + std::string(1, c) + "' expected").c_str()); + } + return found; +} + +char InStream::readSpace() +{ + return readChar(' '); +} + +void InStream::unreadChar(char c) +{ + reader->unreadChar(c); +} + +void InStream::skipChar() +{ + reader->skipChar(); +} + +void InStream::skipBlanks() +{ + while (isBlanks(reader->curChar())) + reader->skipChar(); +} + +std::string InStream::readWord() +{ + readWordTo(_tmpReadToken); + return _tmpReadToken; +} + +void InStream::readWordTo(std::string& result) +{ + if (!strict) + skipBlanks(); + + int cur = reader->nextChar(); + + if (cur == EOF) + quit(_unexpected_eof, "Unexpected end of file - token expected"); + + if (isBlanks(cur)) + quit(_pe, "Unexpected white-space - token expected"); + + result.clear(); + + while (!(isBlanks(cur) || cur == EOF)) + { + result += char(cur); + cur = reader->nextChar(); + } + + reader->unreadChar(cur); + + if (result.length() == 0) + quit(_unexpected_eof, "Unexpected end of file or white-space - token expected"); +} + +std::string InStream::readToken() +{ + return readWord(); +} + +void InStream::readTokenTo(std::string& result) +{ + readWordTo(result); +} + +static std::string __testlib_part(const std::string& s) +{ + if (s.length() <= 64) + return s; + else + return s.substr(0, 30) + "..." + s.substr(s.length() - 31, 31); +} + +std::string InStream::readWord(const pattern& p, const std::string& variableName) +{ + readWordTo(_tmpReadToken); + if (!p.matches(_tmpReadToken)) + { + if (variableName.empty()) + quit(_wa, ("Token \"" + __testlib_part(_tmpReadToken) + "\" doesn't correspond to pattern \"" + p.src() + "\"").c_str()); + else + quit(_wa, ("Token parameter [name=" + variableName + "] equals to \"" + __testlib_part(_tmpReadToken) + "\", doesn't correspond to pattern \"" + p.src() + "\"").c_str()); + } + return _tmpReadToken; +} + +std::string InStream::readWord(const std::string& ptrn, const std::string& variableName) +{ + return readWord(pattern(ptrn), variableName); +} + +std::string InStream::readToken(const pattern& p, const std::string& variableName) +{ + return readWord(p, variableName); +} + +std::string InStream::readToken(const std::string& ptrn, const std::string& variableName) +{ + return readWord(ptrn, variableName); +} + +void InStream::readWordTo(std::string& result, const pattern& p, const std::string& variableName) +{ + readWordTo(result); + if (!p.matches(result)) + { + if (variableName.empty()) + quit(_wa, ("Token \"" + __testlib_part(result) + "\" doesn't correspond to pattern \"" + p.src() + "\"").c_str()); + else + quit(_wa, ("Token parameter [name=" + variableName + "] equals to \"" + __testlib_part(result) + "\", doesn't correspond to pattern \"" + p.src() + "\"").c_str()); + } +} + +void InStream::readWordTo(std::string& result, const std::string& ptrn, const std::string& variableName) +{ + return readWordTo(result, pattern(ptrn), variableName); +} + +void InStream::readTokenTo(std::string& result, const pattern& p, const std::string& variableName) +{ + return readWordTo(result, p, variableName); +} + +void InStream::readTokenTo(std::string& result, const std::string& ptrn, const std::string& variableName) +{ + return readWordTo(result, ptrn, variableName); +} + +#ifdef __GNUC__ +__attribute__((pure)) +#endif +static inline bool equals(long long integer, const char* s) +{ + if (integer == LLONG_MIN) + return strcmp(s, "-9223372036854775808") == 0; + + if (integer == 0LL) + return strcmp(s, "0") == 0; + + size_t length = strlen(s); + + if (length == 0) + return false; + + if (integer < 0 && s[0] != '-') + return false; + + if (integer < 0) + s++, length--, integer = -integer; + + if (length == 0) + return false; + + while (integer > 0) + { + int digit = int(integer % 10); + + if (s[length - 1] != '0' + digit) + return false; + + length--; + integer /= 10; + } + + return length == 0; +} + +static inline double stringToDouble(InStream& in, const char* buffer) +{ + double retval; + + size_t length = strlen(buffer); + + int minusCount = 0; + int plusCount = 0; + int decimalPointCount = 0; + int digitCount = 0; + int eCount = 0; + + for (size_t i = 0; i < length; i++) + { + if (('0' <= buffer[i] && buffer[i] <= '9') || buffer[i] == '.' + || buffer[i] == 'e' || buffer[i] == 'E' + || buffer[i] == '-' || buffer[i] == '+') + { + if ('0' <= buffer[i] && buffer[i] <= '9') + digitCount++; + if (buffer[i] == 'e' || buffer[i] == 'E') + eCount++; + if (buffer[i] == '-') + minusCount++; + if (buffer[i] == '+') + plusCount++; + if (buffer[i] == '.') + decimalPointCount++; + } + else + in.quit(_pe, ("Expected double, but \"" + __testlib_part(buffer) + "\" found").c_str()); + } + + // If for sure is not a number in standard notation or in e-notation. + if (digitCount == 0 || minusCount > 2 || plusCount > 2 || decimalPointCount > 1 || eCount > 1) + in.quit(_pe, ("Expected double, but \"" + __testlib_part(buffer) + "\" found").c_str()); + + char* suffix = new char[length + 1]; + int scanned = std::sscanf(buffer, "%lf%s", &retval, suffix); + bool empty = strlen(suffix) == 0; + delete[] suffix; + + if (scanned == 1 || (scanned == 2 && empty)) + { + if (__testlib_isNaN(retval) || __testlib_isInfinite(retval)) + in.quit(_pe, ("Expected double, but \"" + __testlib_part(buffer) + "\" found").c_str()); + return retval; + } + else + in.quit(_pe, ("Expected double, but \"" + __testlib_part(buffer) + "\" found").c_str()); +} + +static inline double stringToStrictDouble(InStream& in, const char* buffer, int minAfterPointDigitCount, int maxAfterPointDigitCount) +{ + if (minAfterPointDigitCount < 0) + in.quit(_fail, "stringToStrictDouble: minAfterPointDigitCount should be non-negative."); + + if (minAfterPointDigitCount > maxAfterPointDigitCount) + in.quit(_fail, "stringToStrictDouble: minAfterPointDigitCount should be less or equal to maxAfterPointDigitCount."); + + double retval; + + size_t length = strlen(buffer); + + if (length == 0 || length > 1000) + in.quit(_pe, ("Expected strict double, but \"" + __testlib_part(buffer) + "\" found").c_str()); + + if (buffer[0] != '-' && (buffer[0] < '0' || buffer[0] > '9')) + in.quit(_pe, ("Expected strict double, but \"" + __testlib_part(buffer) + "\" found").c_str()); + + int pointPos = -1; + for (size_t i = 1; i + 1 < length; i++) + { + if (buffer[i] == '.') + { + if (pointPos > -1) + in.quit(_pe, ("Expected strict double, but \"" + __testlib_part(buffer) + "\" found").c_str()); + pointPos = int(i); + } + if (buffer[i] != '.' && (buffer[i] < '0' || buffer[i] > '9')) + in.quit(_pe, ("Expected strict double, but \"" + __testlib_part(buffer) + "\" found").c_str()); + } + + if (buffer[length - 1] < '0' || buffer[length - 1] > '9') + in.quit(_pe, ("Expected strict double, but \"" + __testlib_part(buffer) + "\" found").c_str()); + + int afterDigitsCount = (pointPos == -1 ? 0 : int(length) - pointPos - 1); + if (afterDigitsCount < minAfterPointDigitCount || afterDigitsCount > maxAfterPointDigitCount) + in.quit(_pe, ("Expected strict double with number of digits after point in range [" + + vtos(minAfterPointDigitCount) + + "," + + vtos(maxAfterPointDigitCount) + + "], but \"" + __testlib_part(buffer) + "\" found").c_str() + ); + + int firstDigitPos = -1; + for (size_t i = 0; i < length; i++) + if (buffer[i] >= '0' && buffer[i] <= '9') + { + firstDigitPos = int(i); + break; + } + + if (firstDigitPos > 1 || firstDigitPos == -1) + in.quit(_pe, ("Expected strict double, but \"" + __testlib_part(buffer) + "\" found").c_str()); + + if (buffer[firstDigitPos] == '0' && firstDigitPos + 1 < int(length) + && buffer[firstDigitPos + 1] >= '0' && buffer[firstDigitPos + 1] <= '9') + in.quit(_pe, ("Expected strict double, but \"" + __testlib_part(buffer) + "\" found").c_str()); + + char* suffix = new char[length + 1]; + int scanned = std::sscanf(buffer, "%lf%s", &retval, suffix); + bool empty = strlen(suffix) == 0; + delete[] suffix; + + if (scanned == 1 || (scanned == 2 && empty)) + { + if (__testlib_isNaN(retval) || __testlib_isInfinite(retval)) + in.quit(_pe, ("Expected double, but \"" + __testlib_part(buffer) + "\" found").c_str()); + return retval; + } + else + in.quit(_pe, ("Expected double, but \"" + __testlib_part(buffer) + "\" found").c_str()); +} + +static inline long long stringToLongLong(InStream& in, const char* buffer) +{ + if (strcmp(buffer, "-9223372036854775808") == 0) + return LLONG_MIN; + + bool minus = false; + size_t length = strlen(buffer); + + if (length > 1 && buffer[0] == '-') + minus = true; + + if (length > 20) + in.quit(_pe, ("Expected integer, but \"" + __testlib_part(buffer) + "\" found").c_str()); + + long long retval = 0LL; + + int zeroes = 0; + int processingZeroes = true; + + for (int i = (minus ? 1 : 0); i < int(length); i++) + { + if (buffer[i] == '0' && processingZeroes) + zeroes++; + else + processingZeroes = false; + + if (buffer[i] < '0' || buffer[i] > '9') + in.quit(_pe, ("Expected integer, but \"" + __testlib_part(buffer) + "\" found").c_str()); + retval = retval * 10 + (buffer[i] - '0'); + } + + if (retval < 0) + in.quit(_pe, ("Expected integer, but \"" + __testlib_part(buffer) + "\" found").c_str()); + + if ((zeroes > 0 && (retval != 0 || minus)) || zeroes > 1) + in.quit(_pe, ("Expected integer, but \"" + __testlib_part(buffer) + "\" found").c_str()); + + retval = (minus ? -retval : +retval); + + if (length < 19) + return retval; + + if (equals(retval, buffer)) + return retval; + else + in.quit(_pe, ("Expected int64, but \"" + __testlib_part(buffer) + "\" found").c_str()); +} + +int InStream::readInteger() +{ + if (!strict && seekEof()) + quit(_unexpected_eof, "Unexpected end of file - int32 expected"); + + readWordTo(_tmpReadToken); + + long long value = stringToLongLong(*this, _tmpReadToken.c_str()); + if (value < INT_MIN || value > INT_MAX) + quit(_pe, ("Expected int32, but \"" + __testlib_part(_tmpReadToken) + "\" found").c_str()); + + return int(value); +} + +long long InStream::readLong() +{ + if (!strict && seekEof()) + quit(_unexpected_eof, "Unexpected end of file - int64 expected"); + + readWordTo(_tmpReadToken); + return stringToLongLong(*this, _tmpReadToken.c_str()); +} + +long long InStream::readLong(long long minv, long long maxv, const std::string& variableName) +{ + long long result = readLong(); + + if (result < minv || result > maxv) + { + if (variableName.empty()) + quit(_wa, ("Integer " + vtos(result) + " violates the range [" + vtos(minv) + ", " + vtos(maxv) + "]").c_str()); + else + quit(_wa, ("Integer parameter [name=" + variableName + "] equals to " + vtos(result) + ", violates the range [" + vtos(minv) + ", " + vtos(maxv) + "]").c_str()); + } + + return result; +} + +int InStream::readInt() +{ + return readInteger(); +} + +int InStream::readInt(int minv, int maxv, const std::string &variableName) +{ + int result = readInt(); + + if (result < minv || result > maxv) + { + if (variableName.empty()) + quit(_wa, ("Integer " + vtos(result) + " violates the range [" + vtos(minv) + ", " + vtos(maxv) + "]").c_str()); + else + quit(_wa, ("Integer parameter [name=" + std::string(variableName) + "] equals to " + vtos(result) + ", violates the range [" + vtos(minv) + ", " + vtos(maxv) + "]").c_str()); + } + + return result; +} + +int InStream::readInteger(int minv, int maxv, const std::string& variableName) +{ + return readInt(minv, maxv, variableName.c_str()); +} + +double InStream::readReal() +{ + if (!strict && seekEof()) + quit(_unexpected_eof, "Unexpected end of file - double expected"); + + return stringToDouble(*this, readWord().c_str()); +} + +double InStream::readDouble() +{ + return readReal(); +} + +double InStream::readReal(double minv, double maxv, const std::string& variableName) +{ + double result = readReal(); + + if (result < minv || result > maxv) + { + if (variableName.empty()) + quit(_wa, ("Double " + vtos(result) + " violates the range [" + vtos(minv) + ", " + vtos(maxv) + "]").c_str()); + else + quit(_wa, ("Double parameter [name=" + variableName + "] equals to " + vtos(result) + ", violates the range [" + vtos(minv) + ", " + vtos(maxv) + "]").c_str()); + } + + return result; +} + +double InStream::readDouble(double minv, double maxv, const std::string& variableName) +{ + return readReal(minv, maxv, variableName); +} + +double InStream::readStrictReal(double minv, double maxv, + int minAfterPointDigitCount, int maxAfterPointDigitCount, + const std::string& variableName) +{ + if (!strict && seekEof()) + quit(_unexpected_eof, "Unexpected end of file - strict double expected"); + + double result = stringToStrictDouble(*this, readWord().c_str(), + minAfterPointDigitCount, maxAfterPointDigitCount); + + if (result < minv || result > maxv) + { + if (variableName.empty()) + quit(_wa, ("Strict double " + vtos(result) + " violates the range [" + vtos(minv) + ", " + vtos(maxv) + "]").c_str()); + else + quit(_wa, ("Strict double parameter [name=" + variableName + "] equals to " + vtos(result) + ", violates the range [" + vtos(minv) + ", " + vtos(maxv) + "]").c_str()); + } + + return result; +} + +double InStream::readStrictDouble(double minv, double maxv, + int minAfterPointDigitCount, int maxAfterPointDigitCount, + const std::string& variableName) +{ + return readStrictReal(minv, maxv, + minAfterPointDigitCount, maxAfterPointDigitCount, + variableName); +} + +bool InStream::eof() +{ + if (!strict && NULL == file) + return true; + + return reader->eof(); +} + +bool InStream::seekEof() +{ + if (NULL == file) + return true; + skipBlanks(); + return eof(); +} + +bool InStream::eoln() +{ + if (!strict && NULL == file) + return true; + + int c = reader->nextChar(); + + if (!strict) + { + if (c == EOF) + return true; + + if (c == CR) + { + c = reader->nextChar(); + + if (c != LF) + { + reader->unreadChar(c); + reader->unreadChar(CR); + return false; + } + else + return true; + } + + if (c == LF) + return true; + + reader->unreadChar(c); + return false; + } + else + { + bool returnCr = false; + +#ifdef CR_MUST_IN_EOL + if (c != CR) + { + reader->unreadChar(c); + return false; + } + else + { + if (!returnCr) + returnCr = true; + c = reader->nextChar(); + } +#endif + if (c != LF) + { + reader->unreadChar(c); + if (returnCr) + reader->unreadChar(CR); + return false; + } + + return true; + } +} + +void InStream::readEoln() +{ + if (!eoln()) + quit(_pe, "Expected EOLN"); +} + +void InStream::readEof() +{ + if (!eof()) + quit(_pe, "Expected EOF"); + + if (TestlibFinalizeGuard::alive && this == &inf) + testlibFinalizeGuard.readEofCount++; +} + +bool InStream::seekEoln() +{ + if (NULL == file) + return true; + + int cur; + do + { + cur = reader->nextChar(); + } + while (cur == SPACE || cur == TAB); + + reader->unreadChar(cur); + return eoln(); +} + +void InStream::nextLine() +{ + readLine(); +} + +void InStream::readStringTo(std::string& result) +{ + if (NULL == file) + quit(_pe, "Expected line"); + + result.clear(); + int cur; + + for (;;) + { + cur = reader->curChar(); + + if (isEoln(cur)) + break; + + if (cur == EOF) + break; + + result += char(reader->nextChar()); + } + + if (strict) + readEoln(); + else + eoln(); +} + +std::string InStream::readString() +{ + readStringTo(_tmpReadToken); + return _tmpReadToken; +} + +void InStream::readStringTo(std::string& result, const pattern& p, const std::string& variableName) +{ + readStringTo(result); + if (!p.matches(result)) + { + if (variableName.empty()) + quit(_wa, ("Line \"" + __testlib_part(result) + "\" doesn't correspond to pattern \"" + p.src() + "\"").c_str()); + else + quit(_wa, ("Line [name=" + variableName + "] equals to \"" + __testlib_part(result) + "\", doesn't correspond to pattern \"" + p.src() + "\"").c_str()); + } +} + +void InStream::readStringTo(std::string& result, const std::string& ptrn, const std::string& variableName) +{ + readStringTo(result, pattern(ptrn), variableName); +} + +std::string InStream::readString(const pattern& p, const std::string& variableName) +{ + readStringTo(_tmpReadToken, p, variableName); + return _tmpReadToken; +} + +std::string InStream::readString(const std::string& ptrn, const std::string& variableName) +{ + readStringTo(_tmpReadToken, ptrn, variableName); + return _tmpReadToken; +} + +void InStream::readLineTo(std::string& result) +{ + readStringTo(result); +} + +std::string InStream::readLine() +{ + return readString(); +} + +void InStream::readLineTo(std::string& result, const pattern& p, const std::string& variableName) +{ + readStringTo(result, p, variableName); +} + +void InStream::readLineTo(std::string& result, const std::string& ptrn, const std::string& variableName) +{ + readStringTo(result, ptrn, variableName); +} + +std::string InStream::readLine(const pattern& p, const std::string& variableName) +{ + return readString(p, variableName); +} + +std::string InStream::readLine(const std::string& ptrn, const std::string& variableName) +{ + return readString(ptrn, variableName); +} + +#define __testlib_readMany(readMany, readOne, typeName, space) \ + if (size < 0) \ + quit(_fail, #readMany ": size should be non-negative."); \ + if (size > 100000000) \ + quit(_fail, #readMany ": size should be at most 100000000."); \ + \ + std::vector result(size); \ + readManyIteration = indexBase; \ + \ + for (int i = 0; i < size; i++) \ + { \ + result[i] = readOne; \ + readManyIteration++; \ + if (strict && space && i + 1 < size) \ + readSpace(); \ + } \ + \ + readManyIteration = NO_INDEX; \ + return result; \ + +std::vector InStream::readInts(int size, int minv, int maxv, const std::string &variablesName, int indexBase) { + __testlib_readMany(readInts, readInt(minv, maxv, variablesName), int, true) +} + +std::vector InStream::readInts(int size, int indexBase) { + __testlib_readMany(readInts, readInt(), int, true) +} + +std::vector InStream::readIntegers(int size, int minv, int maxv, const std::string &variablesName, int indexBase) { + __testlib_readMany(readIntegers, readInt(minv, maxv, variablesName), int, true) +} + +std::vector InStream::readIntegers(int size, int indexBase) { + __testlib_readMany(readIntegers, readInt(), int, true) +} + +std::vector +InStream::readReals(int size, double minv, double maxv, const std::string &variablesName, int indexBase) { + __testlib_readMany(readReals, readReal(minv, maxv, variablesName), double, true) +} + +std::vector InStream::readReals(int size, int indexBase) { + __testlib_readMany(readReals, readReal(), double, true) +} + +std::vector +InStream::readDoubles(int size, double minv, double maxv, const std::string &variablesName, int indexBase) { + __testlib_readMany(readDoubles, readDouble(minv, maxv, variablesName), double, true) +} + +std::vector InStream::readDoubles(int size, int indexBase) { + __testlib_readMany(readDoubles, readDouble(), double, true) +} + +std::vector InStream::readStrictReals(int size, double minv, double maxv, + int minAfterPointDigitCount, int maxAfterPointDigitCount, + const std::string &variablesName, int indexBase) { + __testlib_readMany(readStrictReals, + readStrictReal(minv, maxv, minAfterPointDigitCount, maxAfterPointDigitCount, variablesName), + double, true) +} + +std::vector InStream::readStrictDoubles(int size, double minv, double maxv, + int minAfterPointDigitCount, int maxAfterPointDigitCount, + const std::string &variablesName, int indexBase) { + __testlib_readMany(readStrictDoubles, + readStrictDouble(minv, maxv, minAfterPointDigitCount, maxAfterPointDigitCount, variablesName), + double, true) +} + +std::vector +InStream::readStrings(int size, const pattern &p, const std::string &variablesName, int indexBase) { + __testlib_readMany(readStrings, readString(p, variablesName), std::string, false) +} + +std::vector +InStream::readStrings(int size, const std::string &ptrn, const std::string &variablesName, int indexBase) { + pattern p(ptrn); + __testlib_readMany(readStrings, readString(p, variablesName), std::string, false) +} + +std::vector +InStream::readLines(int size, const pattern &p, const std::string &variablesName, int indexBase) { + __testlib_readMany(readLines, readString(p, variablesName), std::string, false) +} + +std::vector +InStream::readLines(int size, const std::string &ptrn, const std::string &variablesName, int indexBase) { + pattern p(ptrn); + __testlib_readMany(readLines, readString(p, variablesName), std::string, false) +} + +void InStream::close() +{ + if (opened) + fclose(file); + + if (NULL != reader) + { + delete reader; + reader = NULL; + } + + opened = false; +} + +NORETURN void quit(TResult result, const std::string& msg) +{ + ouf.quit(result, msg.c_str()); +} + +NORETURN void quit(TResult result, const char* msg) +{ + ouf.quit(result, msg); +} + +NORETURN void __testlib_quitp(double points, const char* message) +{ + __testlib_points = points; + std::string stringPoints = removeDoubleTrailingZeroes(format("%.10f", points)); + + std::string quitMessage; + if (NULL == message || 0 == strlen(message)) + quitMessage = stringPoints; + else + quitMessage = stringPoints + " " + message; + + quit(_points, quitMessage.c_str()); +} + +NORETURN void quitp(float points, const std::string& message = "") +{ + __testlib_quitp(double(points), message.c_str()); +} + +NORETURN void quitp(double points, const std::string& message = "") +{ + __testlib_quitp(points, message.c_str()); +} + +NORETURN void quitp(long double points, const std::string& message = "") +{ + __testlib_quitp(double(points), message.c_str()); +} + +template +#ifdef __GNUC__ +__attribute__ ((format (printf, 2, 3))) +#endif +NORETURN void quitp(F points, const char* format, ...) +{ + FMT_TO_RESULT(format, format, message); + quitp(points, message); +} + +#ifdef __GNUC__ +__attribute__ ((format (printf, 2, 3))) +#endif +NORETURN void quitf(TResult result, const char* format, ...) +{ + FMT_TO_RESULT(format, format, message); + quit(result, message); +} + +#ifdef __GNUC__ +__attribute__ ((format (printf, 3, 4))) +#endif +void quitif(bool condition, TResult result, const char* format, ...) +{ + if (condition) + { + FMT_TO_RESULT(format, format, message); + quit(result, message); + } +} + +NORETURN void __testlib_help() +{ + InStream::textColor(InStream::LightCyan); + std::fprintf(stderr, "TESTLIB %s, http://code.google.com/p/testlib/ ", VERSION); + std::fprintf(stderr, "by Mike Mirzayanov, copyright(c) 2005-2013\n"); + std::fprintf(stderr, "Checker name: \"%s\"\n", checkerName.c_str()); + InStream::textColor(InStream::LightGray); + + std::fprintf(stderr, "\n"); + std::fprintf(stderr, "Latest features: \n"); + for (size_t i = 0; i < sizeof(latestFeatures) / sizeof(char*); i++) + { + std::fprintf(stderr, "*) %s\n", latestFeatures[i]); + } + std::fprintf(stderr, "\n"); + + std::fprintf(stderr, "Program must be run with the following arguments: \n"); + std::fprintf(stderr, " [ [<-appes>]]\n\n"); + + std::exit(FAIL_EXIT_CODE); +} + +static void __testlib_ensuresPreconditions() +{ + // testlib assumes: sizeof(int) = 4. + __TESTLIB_STATIC_ASSERT(sizeof(int) == 4); + + // testlib assumes: INT_MAX == 2147483647. + __TESTLIB_STATIC_ASSERT(INT_MAX == 2147483647); + + // testlib assumes: sizeof(long long) = 8. + __TESTLIB_STATIC_ASSERT(sizeof(long long) == 8); + + // testlib assumes: no -ffast-math. + if (!__testlib_isNaN(+__testlib_nan())) + quit(_fail, "Function __testlib_isNaN is not working correctly: possible reason is '-ffast-math'"); + if (!__testlib_isNaN(-__testlib_nan())) + quit(_fail, "Function __testlib_isNaN is not working correctly: possible reason is '-ffast-math'"); +} + +void registerGen(int argc, char* argv[], int randomGeneratorVersion) +{ + if (randomGeneratorVersion < 0 || randomGeneratorVersion > 1) + quitf(_fail, "Random generator version is expected to be 0 or 1."); + random_t::version = randomGeneratorVersion; + + __testlib_ensuresPreconditions(); + + testlibMode = _generator; + __testlib_set_binary(stdin); + rnd.setSeed(argc, argv); +} + +#ifdef USE_RND_AS_BEFORE_087 +void registerGen(int argc, char* argv[]) +{ + registerGen(argc, argv, 0); +} +#else +void registerGen(int argc, char* argv[]) +{ + std::fprintf(stderr, "Use registerGen(argc, argv, 0) or registerGen(argc, argv, 1)." + " The third parameter stands for the random generator version." + " If you are trying to compile old generator use macro -DUSE_RND_AS_BEFORE_087 or registerGen(argc, argv, 0)." + " Version 1 has been released on Spring, 2013. Use it to write new generators.\n\n"); + registerGen(argc, argv, 0); +} +#endif + +void registerInteraction(int argc, char* argv[]) +{ + __testlib_ensuresPreconditions(); + + testlibMode = _interactor; + __testlib_set_binary(stdin); + + if (argc > 1 && !strcmp("--help", argv[1])) + __testlib_help(); + + if (argc < 3 || argc > 6) + { + quit(_fail, std::string("Program must be run with the following arguments: ") + + std::string(" [ [ [<-appes>]]]") + + "\nUse \"--help\" to get help information"); + } + + if (argc <= 4) + { + resultName = ""; + appesMode = false; + } + + if (argc == 5) + { + resultName = argv[4]; + appesMode = false; + } + + if (argc == 6) + { + if (strcmp("-APPES", argv[5]) && strcmp("-appes", argv[5])) + { + quit(_fail, std::string("Program must be run with the following arguments: ") + + " [ [<-appes>]]"); + } + else + { + resultName = argv[4]; + appesMode = true; + } + } + + inf.init(argv[1], _input); + + if (strcmp(argv[2], "stdout") != 0) + { + tout.open(argv[2], std::ios_base::out); + if (tout.fail() || !tout.is_open()) + quit(_fail, std::string("Can not write to the test-output-file '") + argv[2] + std::string("'")); + } + + ouf.init(stdin, _output); + + if (argc >= 4) + ans.init(argv[3], _answer); + else + ans.name = "unopened answer stream"; +} + +void registerValidation() +{ + __testlib_ensuresPreconditions(); + + testlibMode = _validator; + __testlib_set_binary(stdin); + + inf.init(stdin, _input); + inf.strict = true; +} + +void registerTestlibCmd(int argc, char* argv[]) +{ + __testlib_ensuresPreconditions(); + + testlibMode = _checker; + __testlib_set_binary(stdin); + + if (argc > 1 && !strcmp("--help", argv[1])) + __testlib_help(); + + if (argc < 4 || argc > 6) + { + quit(_fail, std::string("Program must be run with the following arguments: ") + + std::string(" [ [<-appes>]]") + + "\nUse \"--help\" to get help information"); + } + + if (argc == 4) + { + resultName = ""; + appesMode = false; + } + + if (argc == 5) + { + resultName = argv[4]; + appesMode = false; + } + + if (argc == 6) + { + if (strcmp("-APPES", argv[5]) && strcmp("-appes", argv[5])) + { + quit(_fail, std::string("Program must be run with the following arguments: ") + + " [ [<-appes>]]"); + } + else + { + resultName = argv[4]; + appesMode = true; + } + } + + inf.init(argv[1], _input); + ouf.init(argv[2], _output); + ans.init(argv[3], _answer); +} + +void registerTestlib(int argc, ...) +{ + if (argc < 3 || argc > 5) + quit(_fail, std::string("Program must be run with the following arguments: ") + + " [ [<-appes>]]"); + + char** argv = new char*[argc + 1]; + + va_list ap; + va_start(ap, argc); + argv[0] = NULL; + for (int i = 0; i < argc; i++) + { + argv[i + 1] = va_arg(ap, char*); + } + va_end(ap); + + registerTestlibCmd(argc + 1, argv); + delete[] argv; +} + +#ifdef __GNUC__ +__attribute__((const)) +#endif +inline bool doubleCompare(double expected, double result, double MAX_DOUBLE_ERROR) +{ + if (__testlib_isNaN(expected)) + { + return __testlib_isNaN(result); + } + else + if (__testlib_isInfinite(expected)) + { + if (expected > 0) + { + return result > 0 && __testlib_isInfinite(result); + } + else + { + return result < 0 && __testlib_isInfinite(result); + } + } + else + if (__testlib_isNaN(result) || __testlib_isInfinite(result)) + { + return false; + } + else + if (__testlib_abs(result - expected) <= MAX_DOUBLE_ERROR + 1E-15) + { + return true; + } + else + { + double minv = __testlib_min(expected * (1.0 - MAX_DOUBLE_ERROR), + expected * (1.0 + MAX_DOUBLE_ERROR)); + double maxv = __testlib_max(expected * (1.0 - MAX_DOUBLE_ERROR), + expected * (1.0 + MAX_DOUBLE_ERROR)); + return result + 1E-15 >= minv && result <= maxv + 1E-15; + } +} + +#ifdef __GNUC__ +__attribute__((const)) +#endif +inline double doubleDelta(double expected, double result) +{ + double absolute = __testlib_abs(result - expected); + + if (__testlib_abs(expected) > 1E-9) + { + double relative = __testlib_abs(absolute / expected); + return __testlib_min(absolute, relative); + } + else + return absolute; +} + +static inline void __testlib_ensure(bool cond, const std::string& msg) +{ + if (!cond) + quit(_fail, msg.c_str()); +} + +static inline void __testlib_ensure(bool cond, const char* msg) +{ + if (!cond) + quit(_fail, msg); +} + +#define ensure(cond) __testlib_ensure(cond, "Condition failed: \"" #cond "\"") + +#ifdef __GNUC__ +__attribute__ ((format (printf, 2, 3))) +#endif +inline void ensuref(bool cond, const char* format, ...) +{ + if (!cond) + { + FMT_TO_RESULT(format, format, message); + __testlib_ensure(cond, message); + } +} + +NORETURN static void __testlib_fail(const std::string& message) +{ + quitf(_fail, "%s", message.c_str()); +} + +#ifdef __GNUC__ +__attribute__ ((format (printf, 1, 2))) +#endif +void setName(const char* format, ...) +{ + FMT_TO_RESULT(format, format, name); + checkerName = name; +} + +/* + * Do not use random_shuffle, because it will produce different result + * for different C++ compilers. + * + * This implementation uses testlib random_t to produce random numbers, so + * it is stable. + */ +template +void shuffle(_RandomAccessIter __first, _RandomAccessIter __last) +{ + if (__first == __last) return; + for (_RandomAccessIter __i = __first + 1; __i != __last; ++__i) + std::iter_swap(__i, __first + rnd.next(int(__i - __first) + 1)); +} + + +template +void random_shuffle(_RandomAccessIter , _RandomAccessIter ) +{ + quitf(_fail, "Don't use random_shuffle(), use shuffle() instead"); +} + +#ifdef __GLIBC__ +# define RAND_THROW_STATEMENT throw() +#else +# define RAND_THROW_STATEMENT +#endif + +int rand() RAND_THROW_STATEMENT +{ + quitf(_fail, "Don't use rand(), use rnd.next() instead"); + + /* This line never runs. */ + //throw "Don't use rand(), use rnd.next() instead"; +} + +void srand(unsigned int seed) RAND_THROW_STATEMENT +{ + quitf(_fail, "Don't use srand(), you should use " + "'registerGen(argc, argv, 1);' to initialize generator seed " + "by hash code of the command line params. The third parameter " + "is randomGeneratorVersion (currently the latest is 1) [ignored seed=%d].", seed); +} + +void startTest(int test) +{ + const std::string testFileName = vtos(test); + if (NULL == freopen(testFileName.c_str(), "wt", stdout)) + __testlib_fail("Unable to write file '" + testFileName + "'"); +} + +inline std::string upperCase(std::string s) +{ + for (size_t i = 0; i < s.length(); i++) + if ('a' <= s[i] && s[i] <= 'z') + s[i] = char(s[i] - 'a' + 'A'); + return s; +} + +inline std::string lowerCase(std::string s) +{ + for (size_t i = 0; i < s.length(); i++) + if ('A' <= s[i] && s[i] <= 'Z') + s[i] = char(s[i] - 'A' + 'a'); + return s; +} + +inline std::string compress(const std::string& s) +{ + return __testlib_part(s); +} + +inline std::string englishEnding(int x) +{ + x %= 100; + if (x / 10 == 1) + return "th"; + if (x % 10 == 1) + return "st"; + if (x % 10 == 2) + return "nd"; + if (x % 10 == 3) + return "rd"; + return "th"; +} + +inline std::string trim(const std::string& s) +{ + if (s.empty()) + return s; + + int left = 0; + while (left < int(s.length()) && isBlanks(s[left])) + left++; + if (left >= int(s.length())) + return ""; + + int right = int(s.length()) - 1; + while (right >= 0 && isBlanks(s[right])) + right--; + if (right < 0) + return ""; + + return s.substr(left, right - left + 1); +} + +template +std::string join(_ForwardIterator first, _ForwardIterator last, _Separator separator) +{ + std::stringstream ss; + bool repeated = false; + for (_ForwardIterator i = first; i != last; i++) + { + if (repeated) + ss << separator; + else + repeated = true; + ss << *i; + } + return ss.str(); +} + +template +std::string join(_ForwardIterator first, _ForwardIterator last) +{ + return join(first, last, ' '); +} + +template +std::string join(const _Collection& collection, _Separator separator) +{ + return join(collection.begin(), collection.end(), separator); +} + +template +std::string join(const _Collection& collection) +{ + return join(collection, ' '); +} + +NORETURN void __testlib_expectedButFound(TResult result, std::string expected, std::string found, const char* prepend) +{ + std::string message; + if (strlen(prepend) != 0) + message = format("%s: expected '%s', but found '%s'", + compress(prepend).c_str(), compress(expected).c_str(), compress(found).c_str()); + else + message = format("expected '%s', but found '%s'", + compress(expected).c_str(), compress(found).c_str()); + quit(result, message); +} + +NORETURN void __testlib_expectedButFound(TResult result, double expected, double found, const char* prepend) +{ + std::string expectedString = removeDoubleTrailingZeroes(format("%.12f", expected)); + std::string foundString = removeDoubleTrailingZeroes(format("%.12f", found)); + __testlib_expectedButFound(result, expectedString, foundString, prepend); +} + +template +#ifdef __GNUC__ +__attribute__ ((format (printf, 4, 5))) +#endif +NORETURN void expectedButFound(TResult result, T expected, T found, const char* prependFormat = "", ...) +{ + FMT_TO_RESULT(prependFormat, prependFormat, prepend); + std::string expectedString = vtos(expected); + std::string foundString = vtos(found); + __testlib_expectedButFound(result, expectedString, foundString, prepend.c_str()); +} + +template <> +#ifdef __GNUC__ +__attribute__ ((format (printf, 4, 5))) +#endif +NORETURN void expectedButFound(TResult result, std::string expected, std::string found, const char* prependFormat, ...) +{ + FMT_TO_RESULT(prependFormat, prependFormat, prepend); + __testlib_expectedButFound(result, expected, found, prepend.c_str()); +} + +template <> +#ifdef __GNUC__ +__attribute__ ((format (printf, 4, 5))) +#endif +NORETURN void expectedButFound(TResult result, double expected, double found, const char* prependFormat, ...) +{ + FMT_TO_RESULT(prependFormat, prependFormat, prepend); + std::string expectedString = removeDoubleTrailingZeroes(format("%.12f", expected)); + std::string foundString = removeDoubleTrailingZeroes(format("%.12f", found)); + __testlib_expectedButFound(result, expectedString, foundString, prepend.c_str()); +} + +template <> +#ifdef __GNUC__ +__attribute__ ((format (printf, 4, 5))) +#endif +NORETURN void expectedButFound(TResult result, const char* expected, const char* found, const char* prependFormat, ...) +{ + FMT_TO_RESULT(prependFormat, prependFormat, prepend); + __testlib_expectedButFound(result, std::string(expected), std::string(found), prepend.c_str()); +} + +template <> +#ifdef __GNUC__ +__attribute__ ((format (printf, 4, 5))) +#endif +NORETURN void expectedButFound(TResult result, float expected, float found, const char* prependFormat, ...) +{ + FMT_TO_RESULT(prependFormat, prependFormat, prepend); + __testlib_expectedButFound(result, double(expected), double(found), prepend.c_str()); +} + +template <> +#ifdef __GNUC__ +__attribute__ ((format (printf, 4, 5))) +#endif +NORETURN void expectedButFound(TResult result, long double expected, long double found, const char* prependFormat, ...) +{ + FMT_TO_RESULT(prependFormat, prependFormat, prepend); + __testlib_expectedButFound(result, double(expected), double(found), prepend.c_str()); +} + +#endif \ No newline at end of file diff --git a/judger/uoj_judger/include/uoj_judger.h b/judger/uoj_judger/include/uoj_judger.h index ce03fa198..aa49bb7b3 100644 --- a/judger/uoj_judger/include/uoj_judger.h +++ b/judger/uoj_judger/include/uoj_judger.h @@ -597,6 +597,8 @@ struct RunProgramConfig { type = "python2"; } else if (lang == "Python3") { type = "python3"; + } else if (lang == "PyPy3") { + type = "pypy3"; } else if (lang == "Java8") { program_name += "." + conf_str(name + "_main_class"); type = "java8"; @@ -1155,6 +1157,22 @@ RunCompilerResult compile_cpp11(const string &name, const string &path = work_pa return run_compiler(path.c_str(), "/usr/bin/g++", "-o", name.c_str(), "-x", "c++", (name + ".code").c_str(), "-lm", "-O2", "-DONLINE_JUDGE", "-std=c++11", NULL); } +RunCompilerResult compile_cpp14(const string &name, const string &path = work_path) { + return run_compiler(path.c_str(), + "/usr/bin/g++", "-o", name.c_str(), "-x", "c++", (name + ".code").c_str(), "-lm", "-O2", "-DONLINE_JUDGE", "-std=c++14", NULL); +} +RunCompilerResult compile_cpp17(const string &name, const string &path = work_path) { + return run_compiler(path.c_str(), + "/usr/bin/g++", "-o", name.c_str(), "-x", "c++", (name + ".code").c_str(), "-lm", "-O2", "-DONLINE_JUDGE", "-std=c++17", NULL); +} +RunCompilerResult compile_cpp20(const string &name, const string &path = work_path) { + return run_compiler(path.c_str(), + "/usr/bin/g++", "-o", name.c_str(), "-x", "c++", (name + ".code").c_str(), "-lm", "-O2", "-DONLINE_JUDGE", "-std=c++20", NULL); +} +RunCompilerResult compile_cpp23(const string &name, const string &path = work_path) { + return run_compiler(path.c_str(), + "/usr/bin/g++", "-o", name.c_str(), "-x", "c++", (name + ".code").c_str(), "-lm", "-O2", "-DONLINE_JUDGE", "-std=c++23", NULL); +} RunCompilerResult compile_python2(const string &name, const string &path = work_path) { return run_compiler(path.c_str(), "/usr/bin/python2", "-E", "-s", "-B", "-O", "-c", @@ -1164,6 +1182,29 @@ RunCompilerResult compile_python3(const string &name, const string &path = work_ return run_compiler(path.c_str(), "/usr/bin/python3", "-I", "-B", "-O", "-c", ("import py_compile\nimport sys\ntry:\n py_compile.compile('" + name + ".code'" + ", '" + name + "', doraise=True)\n sys.exit(0)\nexcept Exception as e:\n print(e)\n sys.exit(1)").c_str(), NULL); } +// RunCompilerResult compile_pypy3(const string &name, const string &path = work_path) { +// return run_compiler(path.c_str(), +// "/usr/bin/pypy3", "-B", "-O", "-S", "-c", ("import py_compile\nimport sys\ntry:\n py_compile.compile('" + name + ".code'" + ", '" + name + "', doraise=True)\n sys.exit(0)\nexcept Exception as e:\n print(e)\n sys.exit(1)").c_str(), NULL); +// } + +RunCompilerResult compile_pypy3(const string &name, const string &path = work_path) { + return run_compiler( + path.c_str(), + "/usr/bin/pypy3", "-B", "-O", "-S", "-c", + ( + "import py_compile, shutil, sys\n" + "try:\n" + " py_compile.compile('" + name + ".code', doraise=True)\n" + " shutil.copyfile('" + name + ".code', '" + name + "')\n" + " sys.exit(0)\n" + "except Exception as e:\n" + " print(e)\n" + " sys.exit(1)" + ).c_str(), + NULL + ); +} + RunCompilerResult compile_java8(const string &name, const string &path = work_path) { RunCompilerResult ret = prepare_java_source(name, path); if (!ret.succeeded) @@ -1194,7 +1235,7 @@ RunCompilerResult compile_java11(const string &name, const string &path = work_p RunCompilerResult compile(const char *name) { string lang = conf_str(string(name) + "_language"); - if ((lang == "C++" || lang == "C++11" || lang == "C") && has_illegal_keywords_in_file(work_path + "/" + name + ".code")) + if ((lang == "C++" || lang == "C++11" || lang == "C" || lang == "C++14" || lang == "C++17" || lang == "C++20") && has_illegal_keywords_in_file(work_path + "/" + name + ".code")) { RunCompilerResult res; res.type = RS_DGS; @@ -1211,12 +1252,27 @@ RunCompilerResult compile(const char *name) { if (lang == "C++11") { return compile_cpp11(name); } + if (lang == "C++14") { + return compile_cpp14(name); + } + if (lang == "C++17") { + return compile_cpp17(name); + } + if (lang == "C++20") { + return compile_cpp20(name); + } + if (lang == "C++23") { + return compile_cpp23(name); + } if (lang == "Python2") { return compile_python2(name); } if (lang == "Python3") { return compile_python3(name); } + if (lang == "PyPy3") { + return compile_pypy3(name); + } if (lang == "Java8") { return compile_java8(name); } @@ -1231,7 +1287,7 @@ RunCompilerResult compile(const char *name) { } RunCompilerResult res = RunCompilerResult::failed_result(); - res.info = "This language is not supported yet."; + res.info = "This language \"" + lang + "\" is not supported yet."; return res; } @@ -1252,6 +1308,22 @@ RunCompilerResult compile_cpp11_with_implementer(const string &name, const strin return run_compiler(path.c_str(), "/usr/bin/g++", "-o", name.c_str(), "implementer.cpp", "-x", "c++", (name + ".code").c_str(), "-lm", "-O2", "-DONLINE_JUDGE", "-std=c++11", NULL); } +RunCompilerResult compile_cpp14_with_implementer(const string &name, const string &path = work_path) { + return run_compiler(path.c_str(), + "/usr/bin/g++", "-o", name.c_str(), "implementer.cpp", "-x", "c++", (name + ".code").c_str(), "-lm", "-O2", "-DONLINE_JUDGE", "-std=c++14", NULL); +} +RunCompilerResult compile_cpp17_with_implementer(const string &name, const string &path = work_path) { + return run_compiler(path.c_str(), + "/usr/bin/g++", "-o", name.c_str(), "implementer.cpp", "-x", "c++", (name + ".code").c_str(), "-lm", "-O2", "-DONLINE_JUDGE", "-std=c++17", NULL); +} +RunCompilerResult compile_cpp20_with_implementer(const string &name, const string &path = work_path) { + return run_compiler(path.c_str(), + "/usr/bin/g++", "-o", name.c_str(), "implementer.cpp", "-x", "c++", (name + ".code").c_str(), "-lm", "-O2", "-DONLINE_JUDGE", "-std=c++20", NULL); +} +RunCompilerResult compile_cpp23_with_implementer(const string &name, const string &path = work_path) { + return run_compiler(path.c_str(), + "/usr/bin/g++", "-o", name.c_str(), "implementer.cpp", "-x", "c++", (name + ".code").c_str(), "-lm", "-O2", "-DONLINE_JUDGE", "-std=c++23", NULL); +} /* RunCompilerResult compile_python2(const string &name, const string &path = work_path) { return run_compiler(path.c_str(), @@ -1283,6 +1355,18 @@ RunCompilerResult compile_with_implementer(const char *name) { if (lang == "C++11") { return compile_cpp11_with_implementer(name); } + if (lang == "C++14") { + return compile_cpp14_with_implementer(name); + } + if (lang == "C++17") { + return compile_cpp17_with_implementer(name); + } + if (lang == "C++20") { + return compile_cpp20_with_implementer(name); + } + if (lang == "C++23") { + return compile_cpp23_with_implementer(name); + } if (lang == "C") { return compile_c_with_implementer(name); } @@ -1291,7 +1375,7 @@ RunCompilerResult compile_with_implementer(const char *name) { } RunCompilerResult res = RunCompilerResult::failed_result(); - res.info = "This language is not supported yet."; + res.info = "This language \"" + lang + "\" is not supported yet."; return res; } diff --git a/judger/uoj_judger/run/run_program.cpp b/judger/uoj_judger/run/run_program.cpp index f7ebea267..e83c9d499 100644 --- a/judger/uoj_judger/run/run_program.cpp +++ b/judger/uoj_judger/run/run_program.cpp @@ -248,6 +248,10 @@ void parse_args(int argc, char **argv) { } else if (run_program_config.type == "java11") { string pre[3] = {"/usr/lib/jvm/java-11-openjdk-amd64/bin/java", "-Xmx1024m", "-Xss1024m"}; run_program_config.argv.insert(run_program_config.argv.begin(), pre, pre + 3); + } else if (run_program_config.type == "pypy3") { + string pre[4] = {"/usr/bin/pypy3", "-B", "-O", "-S"}; + run_program_config.safe_mode = false; + run_program_config.argv.insert(run_program_config.argv.begin(), pre, pre + 4); } } @@ -330,6 +334,7 @@ void run_child() { exit(16); } if (execv(program_c_argv[0], program_c_argv) == -1) { + //fprintf(stderr, "execv failed: %s\n", program_c_argv[0]); exit(17); } } diff --git a/judger/uoj_judger/run/run_program_conf.h b/judger/uoj_judger/run/run_program_conf.h index bf1e32cac..feecc3b38 100644 --- a/judger/uoj_judger/run/run_program_conf.h +++ b/judger/uoj_judger/run/run_program_conf.h @@ -16,6 +16,9 @@ typedef long int reg_val_t; #define REG_ARG3 esx #endif +#pragma once +#include + const size_t MaxPathLen = 200; set writable_file_name_set; @@ -398,7 +401,27 @@ void init_conf(const RunProgramConfig &config) { # ifdef UOJ_JUDGER_BASESYSTEM_UBUNTU1804 statable_file_name_set.insert("/usr/lib/python38.zip"); # endif - } else if (config.type == "java8") { + } else if (config.type == "pypy3") { + memset(syscall_max_cnt, -1, sizeof(syscall_max_cnt)); + + readable_file_name_set.insert("/usr/bin/pypy3"); + readable_file_name_set.insert("/usr/lib/pypy3/"); + + # ifdef UOJ_JUDGER_PYPY3_VERSION + readable_file_name_set.insert("/usr/bin/pypy3-" UOJ_JUDGER_PYPY3_VERSION); + readable_file_name_set.insert("/usr/lib/pypy3-" UOJ_JUDGER_PYPY3_VERSION "/"); + readable_file_name_set.insert("/usr/bin/lib/pypy3-" UOJ_JUDGER_PYPY3_VERSION "/"); + readable_file_name_set.insert("/usr/local/lib/pypy3-" UOJ_JUDGER_PYPY3_VERSION "/"); + # endif + + statable_file_name_set.insert("/usr"); + statable_file_name_set.insert("/usr/bin"); + statable_file_name_set.insert("/usr/lib"); + + readable_file_name_set.insert("/"); + statable_file_name_set.insert("/"); + + }else if (config.type == "java8") { syscall_max_cnt[__NR_gettid ] = -1; syscall_max_cnt[__NR_set_tid_address] = 1; syscall_max_cnt[__NR_set_robust_list] = 14; diff --git a/python_scripts/__pycache__/compile_checker.cpython-310.pyc b/python_scripts/__pycache__/compile_checker.cpython-310.pyc new file mode 100644 index 000000000..fef12b4e1 Binary files /dev/null and b/python_scripts/__pycache__/compile_checker.cpython-310.pyc differ diff --git a/python_scripts/__pycache__/data_preprocess.cpython-310.pyc b/python_scripts/__pycache__/data_preprocess.cpython-310.pyc new file mode 100644 index 000000000..a10b1ab9e Binary files /dev/null and b/python_scripts/__pycache__/data_preprocess.cpython-310.pyc differ diff --git a/python_scripts/__pycache__/database_interface.cpython-310.pyc b/python_scripts/__pycache__/database_interface.cpython-310.pyc new file mode 100644 index 000000000..954c2b936 Binary files /dev/null and b/python_scripts/__pycache__/database_interface.cpython-310.pyc differ diff --git a/python_scripts/assembly.py b/python_scripts/assembly.py new file mode 100644 index 000000000..8e9231da0 --- /dev/null +++ b/python_scripts/assembly.py @@ -0,0 +1,447 @@ +import json +import pymysql +import zipfile +import os, re +import random +import string +import sys +from pathlib import Path +import argparse + +from database_interface import UOJDatabaseInterface, sync_problem +from compile_checker import compile_checker +from data_preprocess import renamePrefixToA, generate_problem_conf + +def setup_problems(base_folder, db_interface, uoj_problem_base): + """ + 配置题目：建立题号映射表，编译checker，生成problem.conf，同步到UOJ + + Args: + base_folder: 包含所有题目文件夹的基础目录 + db_interface: UOJDatabaseInterface实例 + uoj_problem_base: UOJ题目存储路径，如 ~/UOJ-System/uoj_data/upload/problem/ + """ + base_path = Path(base_folder).expanduser().resolve() + mapping_file = base_path / "problem_mapping.json" + + # 读取或创建题号映射表 + if mapping_file.exists(): + with open(mapping_file, 'r', encoding='utf-8') as f: + problem_mapping = json.load(f) + else: + problem_mapping = {} + + # 扫描所有题目文件夹 + problem_folders = [d for d in base_path.iterdir() if d.is_dir() and not d.name.startswith('.')] + + print(f"找到 {len(problem_folders)} 个题目文件夹") + + for folder in sorted(problem_folders): + folder_name = folder.name + db_interface.reconnect() # 确保数据库连接是最新的 + + # 检查是否已有映射 + if folder_name not in problem_mapping: + # 创建新题目 + problem_id = db_interface.new_problem(title=folder_name) + if problem_id: + problem_mapping[folder_name] = problem_id + print(f"✅ 为文件夹 '{folder_name}' 创建新题目，ID: {problem_id}") + else: + print(f"❌ 无法为文件夹 '{folder_name}' 创建题目") + continue + else: + problem_id = problem_mapping[folder_name] + # 验证题目ID是否存在 + if not db_interface.id_verify(problem_id): + print(f"⚠️ 题目ID {problem_id} 不存在，重新创建") + problem_id = db_interface.new_problem(title=folder_name) + if problem_id: + problem_mapping[folder_name] = problem_id + else: + print(f"❌ 无法为文件夹 '{folder_name}' 创建题目") + continue + + print(f"✅ 题目已经配置完毕: {problem_id}") + continue + + print(f"\n处理题目: {folder_name} (ID: {problem_id})") + + # 1. 编译checker + try: + compile_checker(str(folder)) + print("✓ Checker编译成功") + except Exception as e: + print(f"⚠️ Checker编译失败: {e}") + + # 2. 重命名测试数据并生成problem.conf + try: + max_k = renamePrefixToA(str(folder)) + if max_k > 0: + generate_problem_conf(str(folder), max_k) + print(f"✓ 生成problem.conf，共 {max_k} 组数据") + else: + print("⚠️ 未找到测试数据") + except Exception as e: + print(f"❌ 处理测试数据失败: {e}") + continue + + # 3. 同步到UOJ + try: + sync_problem(str(folder), str(problem_id), uoj_problem_base) + print("✓ 同步到UOJ成功") + except Exception as e: + print(f"❌ 同步失败: {e}") + + # 保存映射表 + with open(mapping_file, 'w', encoding='utf-8') as f: + json.dump(problem_mapping, f, indent=2, ensure_ascii=False) + + print(f"\n✅ 题目配置完成，映射表已保存至: {mapping_file}") + return problem_mapping + +def code_process(code): + lang = 'C++17' if '#include' in code else 'PyPy3'#'Python3' + if lang == 'C++17': + code = code.replace('std::endl', "'\\n'") + code = code.replace('endl', "'\\n'") + return code, lang + +def submit_all_solutions(base_folder, submission_base_folder, db_interface, storage_path): + """ + 提交所有题目的所有解答 + + Args: + base_folder: 包含problem_mapping.json的基础目录（题目配置目录） + submission_base_folder: 独立的提交文件夹，包含与题目同名的子文件夹 + db_interface: UOJDatabaseInterface实例 + storage_path: UOJ提交存储路径 + + Returns: + submission_mapping: 提交映射表 {文件路径: submission_id} + """ + base_path = Path(base_folder).expanduser().resolve() + submission_base_path = Path(submission_base_folder).expanduser().resolve() + + mapping_file = base_path / "problem_mapping.json" + submission_mapping_file = submission_base_path / "submission_mapping.json" + + # 读取题号映射表 + if not mapping_file.exists(): + print("❌ 题号映射表不存在，请先运行 setup_problems") + return {} + + with open(mapping_file, 'r', encoding='utf-8') as f: + problem_mapping = json.load(f) + + # 读取或创建提交映射表 + if submission_mapping_file.exists(): + with open(submission_mapping_file, 'r', encoding='utf-8') as f: + submission_mapping = json.load(f) + else: + submission_mapping = {} + + total_submissions = 0 + + # 遍历题号映射表中的每个题目 + for folder_name, problem_id in problem_mapping.items(): + # 在提交文件夹中查找对应的题目文件夹 + submission_folder = submission_base_path / folder_name + + if not submission_folder.exists(): + print(f"⚠️ 提交文件夹中没有题目 {folder_name} 的文件夹") + continue + + # 查找所有cpp文件 + cpp_files = list(submission_folder.glob("*.cpp")) + if not cpp_files: + print(f"⚠️ 题目 {folder_name} 文件夹中没有cpp文件") + continue + + print(f"\n处理题目 {folder_name} (ID: {problem_id}) 的 {len(cpp_files)} 个提交") + + for cpp_file in cpp_files: + # 使用相对于submission_base_path的路径作为key + file_key = str(cpp_file.relative_to(submission_base_path)) + + # 检查是否已经提交过 + if file_key in submission_mapping: + print(f" ✓ {cpp_file.name} 已提交，ID: {submission_mapping[file_key]}") + continue + + # 读取代码 + try: + with open(cpp_file, 'r', encoding='utf-8') as f: + code = f.read() + except Exception as e: + print(f" ❌ 无法读取 {cpp_file.name}: {e}") + continue + + _code, _lang = code_process(code) + # 提交代码 + submission_id = db_interface.submit_code( + problem_id=problem_id, + username='admin', + code=_code, + language=_lang, + storage_path=storage_path + ) + + if submission_id: + submission_mapping[file_key] = submission_id + total_submissions += 1 + print(f" ✓ {cpp_file.name} 提交成功，ID: {submission_id}") + else: + print(f" ❌ {cpp_file.name} 提交失败") + + # 保存提交映射表到提交文件夹中 + with open(submission_mapping_file, 'w', encoding='utf-8') as f: + json.dump(submission_mapping, f, indent=2, ensure_ascii=False, sort_keys=True) + + print(f"\n✅ 提交完成，共提交 {total_submissions} 份代码") + print(f"提交映射表已保存至: {submission_mapping_file}") + return submission_mapping + +def resubmit_problems(base_folder, submission_base_folder, db_interface, storage_path, problem_names): + """ + 重新提交指定题目的所有解答 + + Args: + base_folder: 包含problem_mapping.json的基础目录（题目配置目录） + submission_base_folder: 独立的提交文件夹，包含与题目同名的子文件夹 + db_interface: UOJDatabaseInterface实例 + storage_path: UOJ提交存储路径 + problem_names: 要重新提交的题目名称列表 + + Returns: + submission_mapping: 更新后的提交映射表 + """ + base_path = Path(base_folder).expanduser().resolve() + submission_base_path = Path(submission_base_folder).expanduser().resolve() + + mapping_file = base_path / "problem_mapping.json" + submission_mapping_file = submission_base_path / "submission_mapping.json" + + # 读取题号映射表 + if not mapping_file.exists(): + print("❌ 题号映射表不存在，请先运行 setup_problems") + return {} + + with open(mapping_file, 'r', encoding='utf-8') as f: + problem_mapping = json.load(f) + + # 读取提交映射表 + if submission_mapping_file.exists(): + with open(submission_mapping_file, 'r', encoding='utf-8') as f: + submission_mapping = json.load(f) + else: + submission_mapping = {} + + total_submissions = 0 + total_resubmissions = 0 + + # 检查指定的题目是否存在 + invalid_problems = [name for name in problem_names if name not in problem_mapping] + if invalid_problems: + print(f"⚠️ 以下题目不存在于题号映射表中: {', '.join(invalid_problems)}") + valid_problems = [name for name in problem_names if name in problem_mapping] + if not valid_problems: + print("❌ 没有有效的题目可以重新提交") + return submission_mapping + problem_names = valid_problems + + print(f"开始重新提交 {len(problem_names)} 个题目: {', '.join(problem_names)}") + + # 只处理指定的题目 + for folder_name in problem_names: + problem_id = problem_mapping[folder_name] + + # 在提交文件夹中查找对应的题目文件夹 + submission_folder = submission_base_path / folder_name + + if not submission_folder.exists(): + print(f"⚠️ 提交文件夹中没有题目 {folder_name} 的文件夹") + continue + + # 查找所有cpp文件 + cpp_files = list(submission_folder.glob("*.cpp")) + if not cpp_files: + print(f"⚠️ 题目 {folder_name} 文件夹中没有cpp文件") + continue + + print(f"\n重新提交题目 {folder_name} (ID: {problem_id}) 的 {len(cpp_files)} 个提交") + + for cpp_file in cpp_files: + # 使用相对于submission_base_path的路径作为key + file_key = str(cpp_file.relative_to(submission_base_path)) + + # 读取代码 + try: + with open(cpp_file, 'r', encoding='utf-8') as f: + code = f.read() + except Exception as e: + print(f" ❌ 无法读取 {cpp_file.name}: {e}") + continue + + _code, _lang = code_process(code) + + # 提交代码（无论是否已经提交过都重新提交） + submission_id = db_interface.submit_code( + problem_id=problem_id, + username='admin', + code=_code, + language=_lang, + storage_path=storage_path + ) + + if submission_id: + # 检查是否是重新提交 + if file_key in submission_mapping: + old_id = submission_mapping[file_key] + print(f" 🔄 {cpp_file.name} 重新提交成功，新ID: {submission_id} (旧ID: {old_id})") + total_resubmissions += 1 + else: + print(f" ✓ {cpp_file.name} 首次提交成功，ID: {submission_id}") + + submission_mapping[file_key] = submission_id + total_submissions += 1 + else: + print(f" ❌ {cpp_file.name} 提交失败") + + # 保存更新后的提交映射表 + with open(submission_mapping_file, 'w', encoding='utf-8') as f: + json.dump(submission_mapping, f, indent=2, ensure_ascii=False, sort_keys=True) + + print(f"\n✅ 重新提交完成") + print(f" - 总提交数: {total_submissions}") + print(f" - 重新提交数: {total_resubmissions}") + print(f" - 首次提交数: {total_submissions - total_resubmissions}") + print(f"提交映射表已保存至: {submission_mapping_file}") + return submission_mapping + + +def fetch_all_results(submission_base_folder, db_interface): + """ + 获取所有提交的评测结果 + + Args: + submission_base_folder: 包含submission_mapping.json的提交文件夹 + db_interface: UOJDatabaseInterface实例 + + Returns: + results: 评测结果字典 + """ + submission_base_path = Path(submission_base_folder).expanduser().resolve() + submission_mapping_file = submission_base_path / "submission_mapping.json" + results_file = submission_base_path / "submission_results.json" + short_results_file = submission_base_path / "submission_short_results.json" + + # 读取提交映射表 + print(submission_mapping_file) + if not submission_mapping_file.exists(): + print("❌ 提交映射表不存在，请先运行 submit_all_solutions") + return {} + + with open(submission_mapping_file, 'r', encoding='utf-8') as f: + submission_mapping = json.load(f) + + if not submission_mapping: + print("⚠️ 没有找到任何提交记录") + return {} + + print(f"开始获取 {len(submission_mapping)} 个提交的评测结果...") + + results = {} + + # 构建查询 + submission_ids = list(submission_mapping.values()) + + for ids in submission_ids: + #db_interface.reconnect() # 确保数据库连接是最新的 + js = db_interface.fetch_result(ids) + results[ids] = js + + #json.dump(results, open(results_file, 'w', encoding='utf-8'), indent=2, ensure_ascii=False) + formal_results = {} + short_results = {} + for key in submission_mapping.keys(): + if submission_mapping[key] in results: + formal_results[key] = results[submission_mapping[key]] + res = results[submission_mapping[key]] + details = res.get("details", "") + matches = re.findall(r'info="([^"]+)"', details) + info = matches[-1] if matches else "N/A" + res_s = {"score": res.get("score", 0), "info": info, "submission_id": submission_mapping[key]} + if "error" in res: + res_s["info"] = res["error"] + short_results[key] = res_s + else: + print(f"⚠️ 提交 {key} 的结果未找到") + + json.dump(formal_results, open(results_file, 'w', encoding='utf-8'), indent=2, ensure_ascii=False) + json.dump(short_results, open(short_results_file, 'w', encoding='utf-8'), indent=2, ensure_ascii=False, sort_keys=True) + print(f"\n✅ 结果已保存至: {results_file}") + return results + + +# 主程序示例 +if __name__ == "__main__": + # 数据库配置 + db_config = { + 'host': 'localhost', + 'port': 3306, + 'user': 'root', + 'password': 'root', + 'database': 'app_uoj233', + } + + # 路径配置 + base_folder = "~/judge2/problems/" # 包含所有题目文件夹的目录 + submission_folder = "~/judge2/submissions/" # 每个题目文件夹内的提交文件夹名 + uoj_problem_base = "~/UOJ-System/uoj_data/web/data" + storage_path = "~/UOJ-System/uoj_data/web/storage/" + + parser = argparse.ArgumentParser(description="UOJ Judger Controller") + parser.add_argument("command", choices=["setup", "submit", "fetch", "resubmit"], help="要执行的命令") + parser.add_argument("--base", default=base_folder, help="题目文件夹目录") + parser.add_argument("--submission", default=submission_folder, help="提交文件夹目录") + parser.add_argument("--problems", nargs='+', help="要重新提交的题目名称列表") + + args = parser.parse_args() + base_folder = Path(args.base).expanduser().resolve() + submission_folder = Path(args.submission).expanduser().resolve() + + # 创建数据库接口 + db = UOJDatabaseInterface(db_config, storage_path) + + # 1. 配置题目 + # python main.py setup + if args.command == "setup": + setup_problems(base_folder, db, uoj_problem_base) + + # 2. 提交评测 + # python main.py submit + elif args.command == "submit": + submit_all_solutions(base_folder, submission_folder, db, storage_path) + + # 3. 获取结果 + # python main.py fetch + elif args.command == "fetch": + fetch_all_results(submission_folder, db) + + elif args.command == "resubmit": + if not args.problems: + print("❌ 请使用 --problems 参数指定要重新提交的题目") + print("示例: python main.py resubmit --problems problem1 problem2") + print("使用 'python main.py list' 查看所有可用题目") + else: + resubmit_problems(base_folder, submission_folder, db, storage_path, args.problems) + + else: + print("用法:") + print(" python main.py setup - 配置题目") + print(" python main.py submit - 提交评测") + print(" python main.py fetch - 获取结果") + print(" python main.py resubmit --problems prob1 prob2 prob3 - 重新提交指定题目") + + db.close() \ No newline at end of file diff --git a/python_scripts/compile_checker.py b/python_scripts/compile_checker.py new file mode 100644 index 000000000..c650e0c1a --- /dev/null +++ b/python_scripts/compile_checker.py @@ -0,0 +1,55 @@ +#!/usr/bin/env python3 +import os, shutil, subprocess, sys, uuid + +def run(cmd: list[str], capture=False, **kw): + print("▶", " ".join(cmd)) + return subprocess.check_output(cmd, text=True, **kw) if capture else subprocess.check_call(cmd, **kw) + +def compile_checker(host_src: str, container: str = "uoj-judger") -> None: + chk_cpp = os.path.join(host_src, "chk.cpp") + checker_cpp = os.path.join(host_src, "checker.cpp") + + # 如果 chk.cpp 不存在但 checker.cpp 存在，就复制一份 + if not os.path.isfile(chk_cpp): + if os.path.isfile(checker_cpp): + shutil.copy(checker_cpp, chk_cpp) + print(f"✅ 已将 checker.cpp 复制为 chk.cpp") + chk_cpp = os.path.join(host_src, "chk.cpp") + else: + sys.exit(f"❌ 找不到 {chk_cpp} 或 {checker_cpp}") + + # 1. 容器内临时目录 + tmp_dir = f"/tmp/chk_build_{uuid.uuid4().hex[:8]}" + run(["docker", "exec", container, "mkdir", "-p", tmp_dir]) + + # 2. 仅复制 chk.cpp → 容器 # ← 修改 + remote_cpp = f"{tmp_dir}/chk.cpp" # ← 修改 + run(["docker", "cp", chk_cpp, f"{container}:{remote_cpp}"]) # ← 修改 + + # 3. 容器内编译（-I 指向临时目录即可） # ← 修改 + include_dir = "/opt/uoj_judger/uoj_judger/include" + compile_cmd = [ + "g++", "-O2", + f"-I{include_dir}", # uoj 头文件 + f"-I{tmp_dir}", # 本地 include（如果 chk.cpp 里有 #include "xxx.h"） + "--std=c++17", "-lm", + "-DONLINE_JUDGE", + "-w", + "-o", f"{tmp_dir}/chk", + remote_cpp, + ] + run(["docker", "exec", container, *compile_cmd]) + + # 4. 把生成的 chk 拉回宿主机 + host_exe = os.path.join(host_src, "chk") + run(["docker", "cp", f"{container}:{tmp_dir}/chk", host_exe]) + os.chmod(host_exe, 0o755) + + # 5. 清理 + run(["docker", "exec", container, "rm", "-rf", tmp_dir]) + print(f"✅ 编译成功 → {host_exe}") + +if __name__ == "__main__": + if not (2 <= len(sys.argv) <= 3): + sys.exit("用法：python3 compile_checker.py /path/to/checker_folder [container_name]") + compile_checker(sys.argv[1], sys.argv[2] if len(sys.argv) == 3 else "uoj-judger") diff --git a/python_scripts/compile_interactor.py b/python_scripts/compile_interactor.py new file mode 100644 index 000000000..afdbdecc2 --- /dev/null +++ b/python_scripts/compile_interactor.py @@ -0,0 +1,46 @@ +#!/usr/bin/env python3 +import os, shutil, subprocess, sys, uuid + +def run(cmd: list[str], capture=False, **kw): + print("▶", " ".join(cmd)) + return subprocess.check_output(cmd, text=True, **kw) if capture else subprocess.check_call(cmd, **kw) + +def compile_checker(host_src: str, container: str = "uoj-judger") -> None: + interactor_cpp = os.path.join(host_src, "interactor.cpp") + if not os.path.isfile(interactor_cpp): + sys.exit(f"❌ 找不到 {interactor_cpp}") + + # 1. 容器内临时目录 + tmp_dir = f"/tmp/interactor_build_{uuid.uuid4().hex[:8]}" + run(["docker", "exec", container, "mkdir", "-p", tmp_dir]) + + # 2. 仅复制 interactor.cpp → 容器 # ← 修改 + remote_cpp = f"{tmp_dir}/interactor.cpp" # ← 修改 + run(["docker", "cp", interactor_cpp, f"{container}:{remote_cpp}"]) # ← 修改 + + # 3. 容器内编译（-I 指向临时目录即可） # ← 修改 + include_dir = "/opt/uoj_judger/uoj_judger/include" + compile_cmd = [ + "g++", "-O2", + f"-I{include_dir}", # uoj 头文件 + f"-I{tmp_dir}", # 本地 include + "--std=c++17", "-lm", + "-DONLINE_JUDGE", + "-o", f"{tmp_dir}/interactor", # 生成的 interactor + remote_cpp, + ] + run(["docker", "exec", container, *compile_cmd]) + + # 4. 把生成的 interactor 拉回宿主机 + host_exe = os.path.join(host_src, "interactor") + run(["docker", "cp", f"{container}:{tmp_dir}/interactor", host_exe]) + os.chmod(host_exe, 0o755) + + # 5. 清理 + run(["docker", "exec", container, "rm", "-rf", tmp_dir]) + print(f"✅ 编译成功 → {host_exe}") + +if __name__ == "__main__": + if not (2 <= len(sys.argv) <= 3): + sys.exit("用法：python3 compile_checker.py /path/to/checker_folder [container_name]") + compile_checker(sys.argv[1], sys.argv[2] if len(sys.argv) == 3 else "uoj-judger") diff --git a/python_scripts/data_preprocess.py b/python_scripts/data_preprocess.py new file mode 100644 index 000000000..b377c1b4b --- /dev/null +++ b/python_scripts/data_preprocess.py @@ -0,0 +1,67 @@ +#!/usr/bin/env python3 +import os +import re +import sys +from typing import Tuple, List, Dict + +def renamePrefixToA(folderPath: str) -> Tuple[str, int]: + try: + all_files = os.listdir(folderPath) + except FileNotFoundError: + print(f"❌ 错误: 文件夹 '{folderPath}' 不存在。") + return 0 + + # 1. 使用集合操作快速找到成对文件的基本名 + in_files = {f[:-3] for f in all_files if f.endswith('.in')} + ans_files = {f[:-4] for f in all_files if f.endswith('.ans')} + + # 2. 获取交集并排序 + paired_basenames = sorted(list(in_files.intersection(ans_files))) + + if not paired_basenames: + print("⚠️ 未找到匹配的文件对。") + return 0 + + # 3. 遍历排序后的列表并重命名 + for i, basename in enumerate(paired_basenames, 1): + new_in_path = os.path.join(folderPath, f"a{i}.in") + new_ans_path = os.path.join(folderPath, f"a{i}.ans") + + # 安全检查，防止覆盖现有文件 + if not (os.path.exists(new_in_path) or os.path.exists(new_ans_path)): + os.rename(os.path.join(folderPath, f"{basename}.in"), new_in_path) + os.rename(os.path.join(folderPath, f"{basename}.ans"), new_ans_path) + print(f"{basename}.in/.ans → a{i}.in/.ans") + + return len(paired_basenames) + +def generate_problem_conf(folder_path: str, n_tests: int): + content = f"""n_tests {n_tests} +n_subtasks 1 +subtask_end_1 {n_tests} +n_ex_tests 0 +n_sample_tests 0 +input_pre a +input_suf in +output_pre a +output_suf ans +time_limit 2 +memory_limit 512 +output_limit 64 +checker_time_limit 2 +use_builtin_judger on +""" + conf_path = os.path.join(folder_path, "problem.conf") + with open(conf_path, "w") as f: + f.write(content) + print(f"✅ 已生成 {conf_path}") + + +if __name__ == "__main__": + if len(sys.argv) != 2: + print(f"用法: {sys.argv[0]} <目标目录>") + sys.exit(1) + + maxK = renamePrefixToA(sys.argv[1]) + print(f"最大 k: {maxK}") + generate_problem_conf(sys.argv[1], maxK) diff --git a/python_scripts/database_interface.py b/python_scripts/database_interface.py new file mode 100644 index 000000000..2e0e5a40c --- /dev/null +++ b/python_scripts/database_interface.py @@ -0,0 +1,296 @@ +import json +import pymysql # 或其他数据库驱动 +import zipfile +import os +import random +import string +import sys, tempfile, shutil +from pathlib import Path + +class UOJDatabaseInterface: + def __init__(self, db_config, submission_storage_path = '~/UOJ-System/uoj_data/web/storage/'): + """ + 初始化数据库连接 + """ + self.db_config = db_config + self.connect(db_config) + + self.submission_storage_path = submission_storage_path + + def connect(self, db_config): + """ + 连接到数据库 + """ + try: + self.connection = pymysql.connect(**db_config, charset='utf8mb4') + self.cursor = self.connection.cursor() + print("✓ 数据库连接成功") + except Exception as e: + print(f"✗ 数据库连接失败: {e}") + self.connection = None + + def reconnect(self): + """ + 重新连接到数据库 + """ + self.connect(self.db_config) + + @staticmethod + def _uoj_rand(l, r): + return random.randint(l, r) + + @staticmethod + def _uoj_rand_string(length, charset=string.ascii_letters + string.digits): + return ''.join(random.choice(charset) for _ in range(length)) + + @staticmethod + def _uoj_rand_avaiable_file_name(directory: Path, storage_path: Path): + while True: + file_name = UOJDatabaseInterface._uoj_rand_string(20) + # 直接在 pathlib.Path 对象上进行拼接 + full_path = storage_path / directory / file_name + if not full_path.exists(): + # 返回相对于 storage_path 的路径 + return directory / file_name + + @staticmethod + def _uoj_rand_avaiable_submission_file_name(storage_path: Path): + num = UOJDatabaseInterface._uoj_rand(1, 10000) + submission_dir = Path("submission") / str(num) + + full_submission_path = storage_path / submission_dir + full_submission_path.mkdir(parents=True, exist_ok=True, mode=0o777) + + # 返回一个 pathlib.Path 对象, 例如: "submission/1234/randomstring..." + return UOJDatabaseInterface._uoj_rand_avaiable_file_name(submission_dir, storage_path) + + def get_problem_is_hidden(self, problem_id): + try: + self.cursor.execute("SELECT is_hidden FROM problems WHERE id = %s", (problem_id,)) + result = self.cursor.fetchone() + return result[0] if result else 0 + except Exception as e: + print(f"查询 is_hidden 失败: {e}") + return 0 + + def submit_code(self, problem_id, username, code, language, storage_path: str): + if not self.connection: + print("✗ 无法提交，数据库未连接。") + return None + + # 将字符串路径转换为 pathlib.Path 对象，以便进行现代化的路径操作 + storage_path_obj = Path(storage_path).expanduser() + + # --- 结合点 1: 使用 UOJ 的目录和文件名生成逻辑 --- + # 不再使用 uuid，而是调用我们上面集成的 UOJ 核心方法 + relative_file_path = self._uoj_rand_avaiable_submission_file_name(storage_path_obj) + zip_file_relative_path = str(relative_file_path) # 数据库中存储的是带 .zip 的字符串 + zip_file_full_path = storage_path_obj / zip_file_relative_path # 实际操作文件用完整路径 + + internal_filename = "answer.code" # UOJ 风格的内部文件名 + + try: + # 确保 zip 文件所在的目录存在 + zip_file_full_path.parent.mkdir(parents=True, exist_ok=True, mode=0o777) + with zipfile.ZipFile(zip_file_full_path, 'w', zipfile.ZIP_DEFLATED) as zf: + zf.writestr(internal_filename, code.encode('utf-8')) + except Exception as e: + print(f"✗ 创建 Zip 文件失败: {e}") + return None + + # --- 结合点 2: `content` 字典现在使用 UOJ 风格的相对路径 --- + content = { + 'file_name': "/" + zip_file_relative_path, # 例如: "submission/1234/xyz.zip" + 'config': [ + ['answer_language', language], + ['problem_id', str(problem_id)] + ] + } + + content_json = json.dumps(content) + # --- 步骤 3: 准备并执行数据库插入 (与原脚本一致) --- + tot_size = len(code.encode('utf-8')) + result_json = json.dumps({'status': 'Waiting'}) + is_hidden = self.get_problem_is_hidden(problem_id) + + sql = """ + INSERT INTO submissions ( + problem_id, submit_time, submitter, content, language, + tot_size, status, result, is_hidden + ) VALUES (%s, NOW(), %s, %s, %s, %s, %s, %s, %s) + """ + + try: + self.cursor.execute(sql, ( + problem_id, username, content_json, language, + tot_size, 'Waiting', result_json, is_hidden + )) + self.connection.commit() + submission_id = self.cursor.lastrowid + print(f"✓ 提交成功！") + print(f" - 提交 ID: {submission_id}") + print(f" - Zip 文件: {zip_file_full_path}") + return submission_id + except Exception as e: + self.connection.rollback() + if os.path.exists(zip_file_full_path): + os.remove(zip_file_full_path) + print(f"✗ 数据库插入失败: {e}") + return None + + def new_problem(self, title, is_hidden=True, submission_requirement=None): + if not self.connection: + print("✗ 无法添加新题目，数据库未连接。") + return None + + try: + with self.cursor as cur: + # 插入 problems 表 + cur.execute("INSERT INTO problems (title, is_hidden, submission_requirement) VALUES (%s, %s, %s)", + (title, is_hidden, json.dumps(submission_requirement or {}))) + problem_id = self.connection.insert_id() + + # 插入 problems_contents 表 + cur.execute("INSERT INTO problems_contents (id, statement, statement_md) VALUES (%s, %s, %s)", + (problem_id, '', '')) + + requirement = [{'name': 'answer', 'type': 'source code', 'file_name': 'answer.code'}] + + cur.execute("UPDATE problems SET submission_requirement = %s WHERE id = %s", + (json.dumps(requirement), problem_id)) + + self.connection.commit() + print(f"成功添加题目 ID: {problem_id}") + return problem_id + except Exception as e: + self.connection.rollback() + print(f"✗ 添加新题目失败: {e}") + return None + + def fetch_result(self, submission_id): + """ + 获取提交结果 + """ + if not self.connection: + print("✗ 无法获取结果: 数据库连接未建立") + return None + + try: + self.cursor.execute("SELECT result FROM submissions WHERE id = %s", (submission_id,)) + result = self.cursor.fetchone() + if result: + json_str = result[0].decode('utf-8') + res = json.loads(json_str) + + return json.loads(json_str) + else: + print(f"✗ 提交 ID {submission_id} 不存在") + return None + except Exception as e: + print(f"✗ 获取提交结果失败: {e}") + return None + + def id_verify(self, problem_id): + if not self.connection: + print("✗ 无法验证题目 ID，数据库未连接。") + return False + + try: + with self.cursor as cur: + cur.execute("SELECT COUNT(*) FROM problems WHERE id = %s", (problem_id,)) + count = cur.fetchone()[0] + return count > 0 + except Exception as e: + print(f"✗ 验证题目 ID 失败: {e}") + return False + + def close(self): + if self.connection: + self.connection.close() + print("✓ 数据库连接已关闭") + + +def create_zip_with_folder(source_dir, zip_path): + """ + 创建包含源文件夹本身的zip文件。 + """ + source_path = Path(source_dir).expanduser() + zip_path = Path(zip_path).expanduser() + + with zipfile.ZipFile(zip_path, 'w', zipfile.ZIP_DEFLATED) as zipf: + folder_name = source_path.name + for file_path in source_path.rglob('*'): + if file_path.is_file(): + arcname = Path(folder_name).expanduser() / file_path.relative_to(source_path) + zipf.write(file_path, arcname) + +def sync_problem(source_folder, problem_id, target_base): + source_folder = Path(source_folder).expanduser().resolve() + target_base = Path(target_base).expanduser().resolve() + + if not source_folder.is_dir(): + print(f"❌ 错误: 源文件夹 {source_folder} 不存在或不是一个目录。") + sys.exit(1) + + if not problem_id.isdigit(): + print(f"❌ 错误: 题目ID '{problem_id}' 必须是数字。") + sys.exit(1) + + target_folder = target_base / problem_id + target_zip = target_base / f"{problem_id}.zip" + + try: + print("--- 开始清理目标位置 ---") + if target_folder.exists(): + shutil.rmtree(target_folder) + print(f"🗑️ 已删除旧的目标文件夹: {target_folder}") + + if target_zip.exists(): + target_zip.unlink() + print(f"🗑️ 已删除旧的目标ZIP: {target_zip}") + + with tempfile.TemporaryDirectory(prefix=f"uoj_{problem_id}_") as temp_dir_str: + temp_dir = Path(temp_dir_str) + print(f"\n--- 在安全的临时目录中操作: {temp_dir} ---") + + work_folder = temp_dir / problem_id + + # 【修改点 1】定义临时zip文件的路径 + temp_zip_path = temp_dir / f"{problem_id}.zip" + + print(f"📂 复制: {source_folder} -> {work_folder}") + shutil.copytree(source_folder, work_folder) + + # 【修改点 2】在临时目录中创建zip文件 + print(f"📦 在临时目录中创建ZIP: {temp_zip_path}") + create_zip_with_folder(work_folder, temp_zip_path) + + # 【修改点 3】将临时zip文件移动到最终位置 + print(f"🚚 移动ZIP文件: {temp_zip_path} -> {target_zip}") + shutil.move(str(temp_zip_path), str(target_zip)) + + # 移动文件夹（这部分逻辑不变） + print(f"🚚 移动文件夹: {work_folder} -> {target_folder}") + shutil.move(str(work_folder), str(target_folder)) + + print("\n--- 操作完成，验证结果 ---") + if not target_folder.exists() or not target_zip.exists(): + raise Exception("验证失败，目标文件或文件夹未成功创建！") + + print(f"\n✅ 成功!") + print(f" -> 文件夹位置: {target_folder}") + print(f" -> ZIP文件位置: {target_zip}") + + print(f"\n📋 ZIP文件内容预览:") + with zipfile.ZipFile(target_zip, 'r') as zipf: + file_list = zipf.namelist() + for filename in file_list[:10]: + print(f" {filename}") + if len(file_list) > 10: + print(f" ... 还有 {len(file_list) - 10} 个文件") + + except Exception as e: + print(f"\n❌ 处理过程中发生严重错误: {e}") + import traceback + traceback.print_exc() + sys.exit(1) \ No newline at end of file diff --git a/python_scripts/ex_req.py b/python_scripts/ex_req.py new file mode 100644 index 000000000..df9cb5b9c --- /dev/null +++ b/python_scripts/ex_req.py @@ -0,0 +1,19 @@ +import pymysql +import json + +conn = pymysql.connect( + host='localhost', # 如果数据库运行在宿主机本地或通过端口映射 + port=3306, + user='root', + password='root', + database='app_uoj233', + charset='utf8mb4' +) + +problem_id = 7 + +requirement = [{'name': 'answer', 'type': 'source code', 'file_name': 'answer.code'}] + +with conn.cursor() as cur: + cur.execute("UPDATE problems SET submission_requirement = %s WHERE id = %s", + (json.dumps(requirement), problem_id)) \ No newline at end of file diff --git a/python_scripts/new_problem.py b/python_scripts/new_problem.py new file mode 100644 index 000000000..6e7fc390e --- /dev/null +++ b/python_scripts/new_problem.py @@ -0,0 +1,34 @@ +import pymysql +import json + +# 如果你映射了端口，比如把容器的3306映射到了宿主机3307，就用 localhost:3307 +conn = pymysql.connect( + host='localhost', # 如果数据库运行在宿主机本地或通过端口映射 + port=3306, + user='root', + password='root', + database='app_uoj233', + charset='utf8mb4' +) + +with conn.cursor() as cur: + # 插入 problems 表 + cur.execute("INSERT INTO problems (title, is_hidden, submission_requirement) VALUES (%s, %s, %s)", + ('New Problem', 1, '{}')) + problem_id = conn.insert_id() + + # 插入 problems_contents 表 + cur.execute("INSERT INTO problems_contents (id, statement, statement_md) VALUES (%s, %s, %s)", + (problem_id, '', '')) + + print(f"成功添加题目 ID: {problem_id}") + + +requirement = [{'name': 'answer', 'type': 'source code', 'file_name': 'answer.code'}] + +with conn.cursor() as cur: + cur.execute("UPDATE problems SET submission_requirement = %s WHERE id = %s", + (json.dumps(requirement), problem_id)) + +conn.commit() +conn.close() \ No newline at end of file diff --git a/python_scripts/result.py b/python_scripts/result.py new file mode 100644 index 000000000..ee0e67963 --- /dev/null +++ b/python_scripts/result.py @@ -0,0 +1,68 @@ +import json +import pymysql # 或其他数据库驱动 +import zipfile +import os +import random +import string +from pathlib import Path + +class UOJResultFetcher: + def __init__(self, db_config): + """ + 初始化数据库连接 + """ + try: + self.connection = pymysql.connect(**db_config, charset='utf8mb4') + self.cursor = self.connection.cursor() + print("✓ 数据库连接成功") + except Exception as e: + print(f"✗ 数据库连接失败: {e}") + self.connection = None + + def fetch_result(self, submission_id): + """ + 获取提交结果 + """ + if not self.connection: + print("✗ 无法获取结果: 数据库连接未建立") + return None + + try: + self.cursor.execute("SELECT result FROM submissions WHERE id = %s", (submission_id,)) + result = self.cursor.fetchone() + if result: + json_str = result[0].decode('utf-8') + res = json.loads(json_str) + + return json.loads(json_str) + else: + print(f"✗ 提交 ID {submission_id} 不存在") + return None + except Exception as e: + print(f"✗ 获取提交结果失败: {e}") + return None + + def close(self): + if self.connection: + self.connection.close() + print("✓ 数据库连接已关闭") + +# --- 使用示例 --- +if __name__ == '__main__': + # 1. 请替换为您的数据库配置 + db_config = { + 'host': 'localhost', + 'port': 3306, + 'user': 'root', + 'password': 'root', + 'database': 'app_uoj233', + } + + submission_id = 1 + + fetch_resulter = UOJResultFetcher(db_config) + result = fetch_resulter.fetch_result(submission_id) + print(result) + + # 6. 关闭连接 + fetch_resulter.close() \ No newline at end of file diff --git a/python_scripts/submit.py b/python_scripts/submit.py new file mode 100644 index 000000000..0a0d30a4d --- /dev/null +++ b/python_scripts/submit.py @@ -0,0 +1,176 @@ +import json +import pymysql # 或其他数据库驱动 +import zipfile +import os +import random +import string +from pathlib import Path + +class UOJSubmitter: + def __init__(self, db_config): + """ + 初始化数据库连接 + """ + try: + self.connection = pymysql.connect(**db_config, charset='utf8mb4') + self.cursor = self.connection.cursor() + print("✓ 数据库连接成功") + except Exception as e: + print(f"✗ 数据库连接失败: {e}") + self.connection = None + + # --- 核心: UOJ 文件名/目录生成逻辑 (来自第一个脚本) --- + # 使用 @staticmethod 表示这些方法是静态的，它们不依赖于类的实例状态 (self) + + @staticmethod + def _uoj_rand(l, r): + return random.randint(l, r) + + @staticmethod + def _uoj_rand_string(length, charset=string.ascii_letters + string.digits): + return ''.join(random.choice(charset) for _ in range(length)) + + @staticmethod + def _uoj_rand_avaiable_file_name(directory: Path, storage_path: Path): + while True: + file_name = UOJSubmitter._uoj_rand_string(20) + # 直接在 pathlib.Path 对象上进行拼接 + full_path = storage_path / directory / file_name + if not full_path.exists(): + # 返回相对于 storage_path 的路径 + return directory / file_name + + @staticmethod + def _uoj_rand_avaiable_submission_file_name(storage_path: Path): + num = UOJSubmitter._uoj_rand(1, 10000) + submission_dir = Path("submission") / str(num) + + full_submission_path = storage_path / submission_dir + full_submission_path.mkdir(parents=True, exist_ok=True) + + # 返回一个 pathlib.Path 对象, 例如: "submission/1234/randomstring..." + return UOJSubmitter._uoj_rand_avaiable_file_name(submission_dir, storage_path) + + # ---------------------------------------------------------------- + def get_problem_is_hidden(self, problem_id): + try: + self.cursor.execute("SELECT is_hidden FROM problems WHERE id = %s", (problem_id,)) + result = self.cursor.fetchone() + return result[0] if result else 0 + except Exception as e: + print(f"查询 is_hidden 失败: {e}") + return 0 + + def submit_code(self, problem_id, username, code, language, storage_path: str): + if not self.connection: + print("✗ 无法提交，数据库未连接。") + return None + + # 将字符串路径转换为 pathlib.Path 对象，以便进行现代化的路径操作 + storage_path_obj = Path(storage_path).expanduser() + + # --- 结合点 1: 使用 UOJ 的目录和文件名生成逻辑 --- + # 不再使用 uuid，而是调用我们上面集成的 UOJ 核心方法 + relative_file_path = self._uoj_rand_avaiable_submission_file_name(storage_path_obj) + zip_file_relative_path = str(relative_file_path) # 数据库中存储的是带 .zip 的字符串 + zip_file_full_path = storage_path_obj / zip_file_relative_path # 实际操作文件用完整路径 + + internal_filename = "answer.code" # UOJ 风格的内部文件名 + + try: + # 确保 zip 文件所在的目录存在 + zip_file_full_path.parent.mkdir(parents=True, exist_ok=True) + with zipfile.ZipFile(zip_file_full_path, 'w', zipfile.ZIP_DEFLATED) as zf: + zf.writestr(internal_filename, code.encode('utf-8')) + except Exception as e: + print(f"✗ 创建 Zip 文件失败: {e}") + return None + + # --- 结合点 2: `content` 字典现在使用 UOJ 风格的相对路径 --- + content = { + 'file_name': "/" + zip_file_relative_path, # 例如: "submission/1234/xyz.zip" + 'config': [ + ['answer_language', language], + ['problem_id', str(problem_id)] + ] + } + + content_json = json.dumps(content) + # --- 步骤 3: 准备并执行数据库插入 (与原脚本一致) --- + tot_size = len(code.encode('utf-8')) + result_json = json.dumps({'status': 'Waiting'}) + is_hidden = self.get_problem_is_hidden(problem_id) + + sql = """ + INSERT INTO submissions ( + problem_id, submit_time, submitter, content, language, + tot_size, status, result, is_hidden + ) VALUES (%s, NOW(), %s, %s, %s, %s, %s, %s, %s) + """ + + try: + self.cursor.execute(sql, ( + problem_id, username, content_json, language, + tot_size, 'Waiting', result_json, is_hidden + )) + self.connection.commit() + submission_id = self.cursor.lastrowid + print(f"✓ 提交成功！") + print(f" - 提交 ID: {submission_id}") + print(f" - Zip 文件: {zip_file_full_path}") + return submission_id + except Exception as e: + self.connection.rollback() + if os.path.exists(zip_file_full_path): + os.remove(zip_file_full_path) + print(f"✗ 数据库插入失败: {e}") + return None + + def close(self): + if self.connection: + self.connection.close() + print("✓ 数据库连接已关闭") + +# --- 使用示例 --- +if __name__ == '__main__': + # 1. 请替换为您的数据库配置 + db_config = { + 'host': 'localhost', + 'port': 3306, + 'user': 'root', + 'password': 'root', + 'database': 'app_uoj233', + } + + # 2. 指定 UOJ 用于存放提交的目录，脚本会自动创建子目录 + # '~/...' 会被自动展开为你的 home 目录 + submission_storage_path = '~/UOJ-System/uoj_data/web/storage/' + + # 3. 创建提交者实例 + submitter = UOJSubmitter(db_config) + + # 4. 准备提交数据 + problem_id = 1 + username = 'admin' + cpp_code = """ +#include +int main() { + int a, b; + std::cin >> a >> b; + std::cout << a + b << std::endl; + return 0; +} +""" + language = 'C++17' + + # 5. 执行提交 + submitter.submit_code( + problem_id=problem_id, + username=username, + code=cpp_code, + language=language, + storage_path=submission_storage_path + ) + + # 6. 关闭连接 + submitter.close() \ No newline at end of file diff --git a/python_scripts/sync.py b/python_scripts/sync.py new file mode 100644 index 000000000..67f9f2c9f --- /dev/null +++ b/python_scripts/sync.py @@ -0,0 +1,108 @@ +#!/usr/bin/env python3 +# -*- coding: utf-8 -*- + +import os +import sys +import shutil +import zipfile +import argparse +import tempfile +from pathlib import Path + +# --- 配置区 --- +TARGET_BASE_DIR = Path("./uoj_data/web/data") +# --- 配置区结束 --- + +def create_zip_with_folder(source_dir, zip_path): + """ + 创建包含源文件夹本身的zip文件。 + """ + source_path = Path(source_dir) + zip_path = Path(zip_path) + + with zipfile.ZipFile(zip_path, 'w', zipfile.ZIP_DEFLATED) as zipf: + folder_name = source_path.name + for file_path in source_path.rglob('*'): + if file_path.is_file(): + arcname = Path(folder_name) / file_path.relative_to(source_path) + zipf.write(file_path, arcname) + +def main(): + parser = argparse.ArgumentParser(description='UOJ题目数据打包工具 (最终版)') + parser.add_argument('source_folder', help='源文件夹路径') + parser.add_argument('problem_id', help='题目ID') + + args = parser.parse_args() + + source_folder = Path(args.source_folder).resolve() + problem_id = args.problem_id + + if not source_folder.is_dir(): + print(f"❌ 错误: 源文件夹 {source_folder} 不存在或不是一个目录。") + sys.exit(1) + + if not problem_id.isdigit(): + print(f"❌ 错误: 题目ID '{problem_id}' 必须是数字。") + sys.exit(1) + + target_folder = TARGET_BASE_DIR / problem_id + target_zip = TARGET_BASE_DIR / f"{problem_id}.zip" + + try: + print("--- 开始清理目标位置 ---") + if target_folder.exists(): + shutil.rmtree(target_folder) + print(f"🗑️ 已删除旧的目标文件夹: {target_folder}") + + if target_zip.exists(): + target_zip.unlink() + print(f"🗑️ 已删除旧的目标ZIP: {target_zip}") + + with tempfile.TemporaryDirectory(prefix=f"uoj_{problem_id}_") as temp_dir_str: + temp_dir = Path(temp_dir_str) + print(f"\n--- 在安全的临时目录中操作: {temp_dir} ---") + + work_folder = temp_dir / problem_id + + # 【修改点 1】定义临时zip文件的路径 + temp_zip_path = temp_dir / f"{problem_id}.zip" + + print(f"📂 复制: {source_folder} -> {work_folder}") + shutil.copytree(source_folder, work_folder) + + # 【修改点 2】在临时目录中创建zip文件 + print(f"📦 在临时目录中创建ZIP: {temp_zip_path}") + create_zip_with_folder(work_folder, temp_zip_path) + + # 【修改点 3】将临时zip文件移动到最终位置 + print(f"🚚 移动ZIP文件: {temp_zip_path} -> {target_zip}") + shutil.move(str(temp_zip_path), str(target_zip)) + + # 移动文件夹（这部分逻辑不变） + print(f"🚚 移动文件夹: {work_folder} -> {target_folder}") + shutil.move(str(work_folder), str(target_folder)) + + print("\n--- 操作完成，验证结果 ---") + if not target_folder.exists() or not target_zip.exists(): + raise Exception("验证失败，目标文件或文件夹未成功创建！") + + print(f"\n✅ 成功!") + print(f" -> 文件夹位置: {target_folder}") + print(f" -> ZIP文件位置: {target_zip}") + + print(f"\n📋 ZIP文件内容预览:") + with zipfile.ZipFile(target_zip, 'r') as zipf: + file_list = zipf.namelist() + for filename in file_list[:10]: + print(f" {filename}") + if len(file_list) > 10: + print(f" ... 还有 {len(file_list) - 10} 个文件") + + except Exception as e: + print(f"\n❌ 处理过程中发生严重错误: {e}") + import traceback + traceback.print_exc() + sys.exit(1) + +if __name__ == "__main__": + main() \ No newline at end of file diff --git a/python_scripts/test.py b/python_scripts/test.py new file mode 100644 index 000000000..cf427a356 --- /dev/null +++ b/python_scripts/test.py @@ -0,0 +1 @@ +print("This is a test to see if python can work successfully.") \ No newline at end of file diff --git a/spawn_judgers.sh b/spawn_judgers.sh new file mode 100644 index 000000000..73c290575 --- /dev/null +++ b/spawn_judgers.sh @@ -0,0 +1,69 @@ +#!/usr/bin/env bash +set -euo pipefail + +# === 可配置 === +IMAGE="yanagiorigami/uoj-judger:latest" +NETWORK="uoj-system_default" +UOJ_HOST_NAME="uoj-web" # 与 UOJ 主容器在同一网络下的容器名；用名字避免写死 IP +MYSQL_CONT="uoj-db" # MySQL 容器名 +MYSQL_DB="app_uoj233" +MYSQL_USER="root" +MYSQL_PASS="root" +JUDGER_PASS_DEFAULT="changeme" # 如果不想每台都不同，就用统一密码 +CPU_LIMIT="" # 例如 "--cpus=1.0" +MEM_LIMIT="" # 例如 "--memory=1g" + +# 用法： ./spawn_judgers.sh 数量 前缀(可选) 密码(可选) +COUNT="${1:-3}" +NAME_PREFIX="${2:-judger}" +JUDGER_PASS="${3:-$JUDGER_PASS_DEFAULT}" + +mkdir -p confs + +# 确保网络存在 +docker network inspect "$NETWORK" >/dev/null 2>&1 || docker network create "$NETWORK" + +for i in $(seq 1 "$COUNT"); do + NAME="${NAME_PREFIX}${i}" + CONF="confs/conf-${NAME}.json" + + # 生成每台评测机的 conf.json + cat > "$CONF" </dev/null + fi + + # 拉起容器：共享同一镜像，外挂只读 conf，限制资源（按需启用） + docker run -d --name "${NAME}" \ + --cap-add SYS_PTRACE \ + --restart=always \ + --network "${NETWORK}" \ + -v "$(pwd)/${CONF}:/opt/uoj_judger/.conf.json:ro" \ + $CPU_LIMIT $MEM_LIMIT \ + --entrypoint /bin/bash \ + "${IMAGE}" -lc '/opt/up; touch /opt/uoj_judger/log/judge.log; tail -F /opt/uoj_judger/log/judge.log /opt/uoj_judger/log/judge_client.log 2>/dev/null || sleep infinity' + + # 取容器 IP（在自定义网络下） + IP=$(docker inspect -f '{{range .NetworkSettings.Networks}}{{.IPAddress}}{{end}}' "${NAME}") + + # 写入 MySQL（入库一次即可） + docker exec -i "${MYSQL_CONT}" \ + mysql -u"${MYSQL_USER}" -p"${MYSQL_PASS}" "${MYSQL_DB}" \ + -e "INSERT INTO judger_info (judger_name, password, ip) VALUES ('${NAME}', '${JUDGER_PASS}', '${IP}') + ON DUPLICATE KEY UPDATE password=VALUES(password), ip=VALUES(ip);" + + echo "Launched ${NAME} at ${IP}" +done + +echo "Done." diff --git a/web/Dockerfile b/web/Dockerfile index 3af78ff19..fb7cbbd0d 100644 --- a/web/Dockerfile +++ b/web/Dockerfile @@ -27,6 +27,9 @@ service ntp start\n\ service apache2 start\n\ exec bash\n" >/opt/up && chmod +x /opt/up +RUN apt-get install -y python3 python3-pip +RUN pip3 install pymysql + ENV LANG=C.UTF-8 TZ=Asia/Shanghai EXPOSE 80 CMD /opt/up diff --git a/web/api/adminsubmit.py b/web/api/adminsubmit.py new file mode 100644 index 000000000..67e75685c --- /dev/null +++ b/web/api/adminsubmit.py @@ -0,0 +1,44 @@ +import json +import pymysql +import zipfile +import os, re +import random +import string +import sys +from pathlib import Path +import argparse + +from database_interface import UOJDatabaseInterface, sync_problem +from compile_checker import compile_checker +from data_preprocess import renamePrefixToA, generate_problem_conf + +def main(): + parser = argparse.ArgumentParser(description="批量配置UOJ题目") + parser.add_argument("problem_id", type=str, help="题目ID") + parser.add_argument("language", type=str, help="编程语言") + parser.add_argument("code", type=str, help="source code") + parser.add_argument("--db_host", type=str, default="uoj-db", help="数据库主机") + parser.add_argument("--db_user", type=str, default="root", help="数据库用户名") + parser.add_argument("--db_password", type=str, default="root", help="数据库密码") + parser.add_argument("--db_name", type=str, default="app_uoj233", help="数据库名称") + parser.add_argument("--uoj_problem_base", type=str, default="/var/uoj_data/", help="UOJ题目存储路径") + parser.add_argument("--storage_path", type=str, default="/opt/uoj/web/app/storage/", help="存储路径") + + args = parser.parse_args() + + db_config = { + 'host': args.db_host, + 'port': 3306, + 'user': args.db_user, + 'password': args.db_password, + 'database': 'app_uoj233', + } + + db_interface = UOJDatabaseInterface(db_config, args.storage_path) + + db_interface.reconnect() + submission_id = db_interface.submit_code(args.problem_id, "admin", args.code, args.language, args.storage_path) + print(submission_id) + +if __name__ == "__main__": + main() \ No newline at end of file diff --git a/web/api/batchsetup.py b/web/api/batchsetup.py new file mode 100644 index 000000000..07017707e --- /dev/null +++ b/web/api/batchsetup.py @@ -0,0 +1,182 @@ +import json +import pymysql +import zipfile +import os, re +import random +import string +import sys +from pathlib import Path +import argparse + +from database_interface import UOJDatabaseInterface, sync_problem +from compile_checker import compile_checker +from data_preprocess import renamePrefixToA, generate_problem_conf + +def setup_problems(base_folder, db_interface, uoj_problem_base): + """ + 配置题目：建立题号映射表，编译checker，生成problem.conf，同步到UOJ + + Args: + base_folder: 包含所有题目文件夹的基础目录 + db_interface: UOJDatabaseInterface实例 + uoj_problem_base: UOJ题目存储路径，如 ~/UOJ-System/uoj_data/upload/problem/ + """ + base_path = Path(base_folder).expanduser().resolve() + mapping_file = base_path / "problem_mapping.json" + + # 读取或创建题号映射表 + if mapping_file.exists(): + with open(mapping_file, 'r', encoding='utf-8') as f: + problem_mapping = json.load(f) + else: + problem_mapping = {} + + # 扫描所有题目文件夹 + problem_folders = [d for d in base_path.iterdir() if d.is_dir() and not d.name.startswith('.')] + + print(f"找到 {len(problem_folders)} 个题目文件夹", file=sys.stderr, flush=True) + + for folder in sorted(problem_folders): + folder_name = folder.name + db_interface.reconnect() # 确保数据库连接是最新的 + + # 检查是否已有映射 + if folder_name not in problem_mapping: + # 创建新题目 + problem_id = db_interface.new_problem(title=folder_name) + if problem_id: + problem_mapping[folder_name] = problem_id + print(f"✅ 为文件夹 '{folder_name}' 创建新题目，ID: {problem_id}", file=sys.stderr, flush=True) + else: + print(f"❌ 无法为文件夹 '{folder_name}' 创建题目") + continue + else: + problem_id = problem_mapping[folder_name] + # 验证题目ID是否存在 + if not db_interface.id_verify(problem_id): + print(f"⚠️ 题目ID {problem_id} 不存在，重新创建", file=sys.stderr, flush=True) + problem_id = db_interface.new_problem(title=folder_name) + if problem_id: + problem_mapping[folder_name] = problem_id + else: + print(f"❌ 无法为文件夹 '{folder_name}' 创建题目") + continue + + print(f"✅ 题目已经配置完毕: {problem_id}", file=sys.stderr, flush=True) + continue + + print(f"\n处理题目: {folder_name} (ID: {problem_id})", file=sys.stderr, flush=True) + + # 1. 编译checker + try: + compile_checker(str(folder)) + print("✓ Checker编译成功", file=sys.stderr, flush=True) + except Exception as e: + print(f"⚠️ Checker编译失败: {e}") + + # 2. 重命名测试数据并生成problem.conf + try: + max_k = renamePrefixToA(str(folder)) + if max_k > 0: + generate_problem_conf(str(folder), max_k) + print(f"✓ 生成problem.conf，共 {max_k} 组数据", file=sys.stderr, flush=True) + else: + print("⚠️ 未找到测试数据", file=sys.stderr, flush=True) + except Exception as e: + print(f"❌ 处理测试数据失败: {e}") + continue + + # 3. 同步到UOJ + try: + sync_problem(str(folder), str(problem_id), uoj_problem_base) + print("✓ 同步到UOJ成功", file=sys.stderr, flush=True) + except Exception as e: + print(f"❌ 同步失败: {e}") + + # 保存映射表 + with open(mapping_file, 'w', encoding='utf-8') as f: + json.dump(problem_mapping, f, indent=2, ensure_ascii=False) + + print(f"\n✅ 题目配置完成，映射表已保存至: {mapping_file}", file=sys.stderr, flush=True) + return problem_mapping + +def batchsetup(base_folder, db_interface, uoj_problem_base): + base_path = Path(base_folder).expanduser().resolve() + mapping_file = base_path / "problem_mapping.json" + problem_mapping = {} + problem_folders = [d for d in base_path.iterdir() if d.is_dir() and not d.name.startswith('.')] + print(f"找到 {len(problem_folders)} 个题目文件夹", file=sys.stderr, flush=True) + #print(f"找到 {len(problem_folders)} 个题目文件夹") + + for folder in sorted(problem_folders): + folder_name = folder.name + db_interface.reconnect() # 确保数据库连接是最新的 + + + problem_id = db_interface.new_problem(title=folder_name) + if problem_id: + problem_mapping[folder_name] = problem_id + print(f"✅ 为文件夹 '{folder_name}' 创建新题目，ID: {problem_id}", file=sys.stderr, flush=True) + else: + print(f"❌ 无法为文件夹 '{folder_name}' 创建题目", file=sys.stderr, flush=True) + continue + + print(f"\n处理题目: {folder_name} (ID: {problem_id})", file=sys.stderr, flush=True) + + # 1. 编译checker + try: + compile_checker(str(folder)) + print("✓ Checker编译成功", file=sys.stderr, flush=True) + except Exception as e: + print(f"⚠️ Checker编译失败: {e}", file=sys.stderr, flush=True) + + # 2. 重命名测试数据并生成problem.conf + try: + max_k = renamePrefixToA(str(folder)) + if max_k > 0: + generate_problem_conf(str(folder), max_k) + print(f"✓ 生成problem.conf，共 {max_k} 组数据", file=sys.stderr, flush=True) + else: + print("⚠️ 未找到测试数据", file=sys.stderr, flush=True) + except Exception as e: + print(f"❌ 处理测试数据失败: {e}", file=sys.stderr, flush=True) + continue + + # 3. 同步到UOJ + try: + sync_problem(str(folder), str(problem_id), uoj_problem_base) + print("✓ 同步到UOJ成功", file=sys.stderr, flush=True) + except Exception as e: + print(f"❌ 同步失败: {e}", file=sys.stderr, flush=True) + + print(json.dumps(problem_mapping, indent=2, ensure_ascii=False), file=sys.stdout, flush=True) + return problem_mapping + +def main(): + parser = argparse.ArgumentParser(description="批量配置UOJ题目") + parser.add_argument("base_folder", type=str, help="包含所有题目文件夹的基础目录") + parser.add_argument("--db_host", type=str, default="uoj-db", help="数据库主机") + parser.add_argument("--db_user", type=str, default="root", help="数据库用户名") + parser.add_argument("--db_password", type=str, default="root", help="数据库密码") + parser.add_argument("--db_name", type=str, default="app_uoj233", help="数据库名称") + parser.add_argument("--uoj_problem_base", type=str, default="/var/uoj_data/", help="UOJ题目存储路径") + parser.add_argument("--storage_path", type=str, default="/opt/uoj/web/app/storage/", help="存储路径") + + args = parser.parse_args() + + db_config = { + 'host': args.db_host, + 'port': 3306, + 'user': args.db_user, + 'password': args.db_password, + 'database': 'app_uoj233', + } + + # 初始化数据库接口 + db_interface = UOJDatabaseInterface(db_config, args.storage_path) + + # 执行题目配置 + batchsetup(args.base_folder, db_interface, args.uoj_problem_base) + +if __name__ == "__main__": + main() \ No newline at end of file diff --git a/web/api/compile_checker.py b/web/api/compile_checker.py new file mode 100644 index 000000000..2de50d0d8 --- /dev/null +++ b/web/api/compile_checker.py @@ -0,0 +1,78 @@ +#!/usr/bin/env python3 +import os, shutil, subprocess, sys +from typing import List, Dict + +def run(cmd: List[str], **kw): + """打印并执行一条命令""" + print("▶", " ".join(cmd), file=sys.stderr, flush=True) + return subprocess.check_call(cmd, **kw) + +def compile_checker(src_dir: str) -> None: + """ + 在本地（容器内）编译 C++ 检查器 (checker)。 + + :param src_dir: 包含 chk.cpp 或 checker.cpp 的目录。 + """ + chk_cpp = os.path.join(src_dir, "chk.cpp") + checker_cpp = os.path.join(src_dir, "checker.cpp") + chk_exec = os.path.join(src_dir, "chk") + checker_exec = os.path.join(src_dir, "checker") + + if os.path.isfile(chk_exec) and os.access(chk_exec, os.X_OK): + print(f"✅ 检查器已存在且可执行: {chk_exec}", file=sys.stderr, flush=True) + return + + if os.path.isfile(checker_exec) and os.access(checker_exec, os.X_OK): + os.rename(checker_exec, chk_exec) + print("✅ 已将 checker 可执行文件重命名为 chk", file=sys.stderr, flush=True) + return + + # 如果 chk.cpp 不存在，则尝试从 checker.cpp 复制 + if not os.path.isfile(chk_cpp): + if os.path.isfile(checker_cpp): + shutil.copy(checker_cpp, chk_cpp) + print(f"✅ 已将 checker.cpp 复制为 chk.cpp", file=sys.stderr, flush=True) + else: + sys.exit(f"❌ 在目录 {src_dir} 中找不到 chk.cpp 或 checker.cpp") + + # 定义输出文件路径 + output_exe = os.path.join(src_dir, "chk") + + # 定义 UOJ 头文件路径 (在标准 UOJ 容器内) + include_dir = "/opt/uoj/judger/uoj_judger/include" + + # 构建 g++ 编译命令 + compile_cmd = [ + "g++", + "-o", output_exe, # 输出文件 + chk_cpp, # 输入源文件 + "-O2", + "--std=c++17", + "-lm", + "-w", # 抑制警告 + "-DONLINE_JUDGE", + f"-I{include_dir}", # UOJ 框架头文件 + f"-I{src_dir}", # 允许 #include "some_local_header.h" + ] + + # 执行编译 + try: + run(compile_cmd) + # 为生成的可执行文件添加执行权限 + os.chmod(output_exe, 0o755) + print(f"✅ 编译成功 → {output_exe}", file=sys.stderr, flush=True) + except subprocess.CalledProcessError as e: + sys.exit(f"❌ 编译失败，返回码: {e.returncode}") + except FileNotFoundError: + sys.exit("❌ 编译失败，找不到 g++。请确保您在正确的容器环境中，并且已安装 g++。") + + +if __name__ == "__main__": + if len(sys.argv) != 2: + sys.exit("用法：python3 compile_checker_local.py /path/to/checker_folder") + + checker_folder = sys.argv[1] + if not os.path.isdir(checker_folder): + sys.exit(f"❌ 错误：提供的路径不是一个有效的目录 -> {checker_folder}") + + compile_checker(checker_folder) \ No newline at end of file diff --git a/web/api/data_preprocess.py b/web/api/data_preprocess.py new file mode 100644 index 000000000..437314f01 --- /dev/null +++ b/web/api/data_preprocess.py @@ -0,0 +1,67 @@ +#!/usr/bin/env python3 +import os +import re +import sys +from typing import Tuple, List, Dict + +def renamePrefixToA(folderPath: str) -> Tuple[str, int]: + try: + all_files = os.listdir(folderPath) + except FileNotFoundError: + print(f"❌ 错误: 文件夹 '{folderPath}' 不存在。") + return 0 + + # 1. 使用集合操作快速找到成对文件的基本名 + in_files = {f[:-3] for f in all_files if f.endswith('.in')} + ans_files = {f[:-4] for f in all_files if f.endswith('.ans')} + + # 2. 获取交集并排序 + paired_basenames = sorted(list(in_files.intersection(ans_files))) + + if not paired_basenames: + print("⚠️ 未找到匹配的文件对。") + return 0 + + # 3. 遍历排序后的列表并重命名 + for i, basename in enumerate(paired_basenames, 1): + new_in_path = os.path.join(folderPath, f"a{i}.in") + new_ans_path = os.path.join(folderPath, f"a{i}.ans") + + # 安全检查，防止覆盖现有文件 + if not (os.path.exists(new_in_path) or os.path.exists(new_ans_path)): + os.rename(os.path.join(folderPath, f"{basename}.in"), new_in_path) + os.rename(os.path.join(folderPath, f"{basename}.ans"), new_ans_path) + print(f"{basename}.in/.ans → a{i}.in/.ans", file=sys.stderr, flush=True) + + return len(paired_basenames) + +def generate_problem_conf(folder_path: str, n_tests: int): + content = f"""n_tests {n_tests} +n_subtasks 1 +subtask_end_1 {n_tests} +n_ex_tests 0 +n_sample_tests 0 +input_pre a +input_suf in +output_pre a +output_suf ans +time_limit 2 +memory_limit 512 +output_limit 64 +checker_time_limit 2 +use_builtin_judger on +""" + conf_path = os.path.join(folder_path, "problem.conf") + with open(conf_path, "w") as f: + f.write(content) + print(f"✅ 已生成 {conf_path}", file=sys.stderr, flush=True) + + +if __name__ == "__main__": + if len(sys.argv) != 2: + print(f"用法: {sys.argv[0]} <目标目录>", file=sys.stderr, flush=True) + sys.exit(1) + + maxK = renamePrefixToA(sys.argv[1]) + print(f"最大 k: {maxK}", file=sys.stderr, flush=True) + generate_problem_conf(sys.argv[1], maxK) diff --git a/web/api/database_interface.py b/web/api/database_interface.py new file mode 100644 index 000000000..de8bcb13b --- /dev/null +++ b/web/api/database_interface.py @@ -0,0 +1,296 @@ +import json +import pymysql # 或其他数据库驱动 +import zipfile +import os +import random +import string +import sys, tempfile, shutil +from pathlib import Path + +class UOJDatabaseInterface: + def __init__(self, db_config, submission_storage_path = '~/UOJ-System/uoj_data/web/storage/'): + """ + 初始化数据库连接 + """ + self.db_config = db_config + self.connect(db_config) + + self.submission_storage_path = submission_storage_path + + def connect(self, db_config): + """ + 连接到数据库 + """ + try: + self.connection = pymysql.connect(**db_config, charset='utf8mb4') + self.cursor = self.connection.cursor() + print("✓ 数据库连接成功", file=sys.stderr, flush=True) + except Exception as e: + print(f"✗ 数据库连接失败: {e}") + self.connection = None + + def reconnect(self): + """ + 重新连接到数据库 + """ + self.connect(self.db_config) + + @staticmethod + def _uoj_rand(l, r): + return random.randint(l, r) + + @staticmethod + def _uoj_rand_string(length, charset=string.ascii_letters + string.digits): + return ''.join(random.choice(charset) for _ in range(length)) + + @staticmethod + def _uoj_rand_avaiable_file_name(directory: Path, storage_path: Path): + while True: + file_name = UOJDatabaseInterface._uoj_rand_string(20) + # 直接在 pathlib.Path 对象上进行拼接 + full_path = storage_path / directory / file_name + if not full_path.exists(): + # 返回相对于 storage_path 的路径 + return directory / file_name + + @staticmethod + def _uoj_rand_avaiable_submission_file_name(storage_path: Path): + num = UOJDatabaseInterface._uoj_rand(1, 10000) + submission_dir = Path("submission") / str(num) + + full_submission_path = storage_path / submission_dir + full_submission_path.mkdir(parents=True, exist_ok=True, mode=0o777) + + # 返回一个 pathlib.Path 对象, 例如: "submission/1234/randomstring..." + return UOJDatabaseInterface._uoj_rand_avaiable_file_name(submission_dir, storage_path) + + def get_problem_is_hidden(self, problem_id): + try: + self.cursor.execute("SELECT is_hidden FROM problems WHERE id = %s", (problem_id,)) + result = self.cursor.fetchone() + return result[0] if result else 0 + except Exception as e: + print(f"查询 is_hidden 失败: {e}") + return 0 + + def submit_code(self, problem_id, username, code, language, storage_path: str): + if not self.connection: + print("✗ 无法提交，数据库未连接。") + return None + + # 将字符串路径转换为 pathlib.Path 对象，以便进行现代化的路径操作 + storage_path_obj = Path(storage_path).expanduser() + + # --- 结合点 1: 使用 UOJ 的目录和文件名生成逻辑 --- + # 不再使用 uuid，而是调用我们上面集成的 UOJ 核心方法 + relative_file_path = self._uoj_rand_avaiable_submission_file_name(storage_path_obj) + zip_file_relative_path = str(relative_file_path) # 数据库中存储的是带 .zip 的字符串 + zip_file_full_path = storage_path_obj / zip_file_relative_path # 实际操作文件用完整路径 + + internal_filename = "answer.code" # UOJ 风格的内部文件名 + + try: + # 确保 zip 文件所在的目录存在 + zip_file_full_path.parent.mkdir(mode=0o777, parents=True, exist_ok=True) + with zipfile.ZipFile(zip_file_full_path, 'w', zipfile.ZIP_DEFLATED) as zf: + zf.writestr(internal_filename, code.encode('utf-8')) + except Exception as e: + print(f"✗ 创建 Zip 文件失败: {e}") + return None + + # --- 结合点 2: `content` 字典现在使用 UOJ 风格的相对路径 --- + content = { + 'file_name': "/" + zip_file_relative_path, # 例如: "submission/1234/xyz.zip" + 'config': [ + ['answer_language', language], + ['problem_id', str(problem_id)] + ] + } + + content_json = json.dumps(content) + # --- 步骤 3: 准备并执行数据库插入 (与原脚本一致) --- + tot_size = len(code.encode('utf-8')) + result_json = json.dumps({'status': 'Waiting'}) + is_hidden = self.get_problem_is_hidden(problem_id) + + sql = """ + INSERT INTO submissions ( + problem_id, submit_time, submitter, content, language, + tot_size, status, result, is_hidden + ) VALUES (%s, NOW(), %s, %s, %s, %s, %s, %s, %s) + """ + + try: + self.cursor.execute(sql, ( + problem_id, username, content_json, language, + tot_size, 'Waiting', result_json, is_hidden + )) + self.connection.commit() + submission_id = self.cursor.lastrowid + print(f"✓ 提交成功！", file=sys.stderr, flush=True) + print(f" - 提交 ID: {submission_id}", file=sys.stderr, flush=True) + print(f" - Zip 文件: {zip_file_full_path}", file=sys.stderr, flush=True) + return submission_id + except Exception as e: + self.connection.rollback() + if os.path.exists(zip_file_full_path): + os.remove(zip_file_full_path) + print(f"✗ 数据库插入失败: {e}") + return None + + def new_problem(self, title, is_hidden=True, submission_requirement=None): + if not self.connection: + print("✗ 无法添加新题目，数据库未连接。") + return None + + try: + with self.cursor as cur: + # 插入 problems 表 + cur.execute("INSERT INTO problems (title, is_hidden, submission_requirement) VALUES (%s, %s, %s)", + (title, is_hidden, json.dumps(submission_requirement or {}))) + problem_id = self.connection.insert_id() + + # 插入 problems_contents 表 + cur.execute("INSERT INTO problems_contents (id, statement, statement_md) VALUES (%s, %s, %s)", + (problem_id, '', '')) + + requirement = [{'name': 'answer', 'type': 'source code', 'file_name': 'answer.code'}] + + cur.execute("UPDATE problems SET submission_requirement = %s WHERE id = %s", + (json.dumps(requirement), problem_id)) + + self.connection.commit() + print(f"成功添加题目 ID: {problem_id}", file=sys.stderr, flush=True) + return problem_id + except Exception as e: + self.connection.rollback() + print(f"✗ 添加新题目失败: {e}") + return None + + def fetch_result(self, submission_id): + """ + 获取提交结果 + """ + if not self.connection: + print("✗ 无法获取结果: 数据库连接未建立") + return None + + try: + self.cursor.execute("SELECT result FROM submissions WHERE id = %s", (submission_id,)) + result = self.cursor.fetchone() + if result: + json_str = result[0].decode('utf-8') + res = json.loads(json_str) + + return json.loads(json_str) + else: + print(f"✗ 提交 ID {submission_id} 不存在") + return None + except Exception as e: + print(f"✗ 获取提交结果失败: {e}") + return None + + def id_verify(self, problem_id): + if not self.connection: + print("✗ 无法验证题目 ID，数据库未连接。") + return False + + try: + with self.cursor as cur: + cur.execute("SELECT COUNT(*) FROM problems WHERE id = %s", (problem_id,)) + count = cur.fetchone()[0] + return count > 0 + except Exception as e: + print(f"✗ 验证题目 ID 失败: {e}") + return False + + def close(self): + if self.connection: + self.connection.close() + print("✓ 数据库连接已关闭", file=sys.stderr, flush=True) + + +def create_zip_with_folder(source_dir, zip_path): + """ + 创建包含源文件夹本身的zip文件。 + """ + source_path = Path(source_dir).expanduser() + zip_path = Path(zip_path).expanduser() + + with zipfile.ZipFile(zip_path, 'w', zipfile.ZIP_DEFLATED) as zipf: + folder_name = source_path.name + for file_path in source_path.rglob('*'): + if file_path.is_file(): + arcname = Path(folder_name).expanduser() / file_path.relative_to(source_path) + zipf.write(file_path, arcname) + +def sync_problem(source_folder, problem_id, target_base): + source_folder = Path(source_folder).expanduser().resolve() + target_base = Path(target_base).expanduser().resolve() + + if not source_folder.is_dir(): + print(f"❌ 错误: 源文件夹 {source_folder} 不存在或不是一个目录。") + sys.exit(1) + + if not problem_id.isdigit(): + print(f"❌ 错误: 题目ID '{problem_id}' 必须是数字。") + sys.exit(1) + + target_folder = target_base / problem_id + target_zip = target_base / f"{problem_id}.zip" + + try: + print("--- 开始清理目标位置 ---", file=sys.stderr, flush=True) + if target_folder.exists(): + shutil.rmtree(target_folder) + print(f"🗑️ 已删除旧的目标文件夹: {target_folder}", file=sys.stderr, flush=True) + + if target_zip.exists(): + target_zip.unlink() + print(f"🗑️ 已删除旧的目标ZIP: {target_zip}", file=sys.stderr, flush=True) + + with tempfile.TemporaryDirectory(prefix=f"uoj_{problem_id}_") as temp_dir_str: + temp_dir = Path(temp_dir_str) + print(f"\n--- 在安全的临时目录中操作: {temp_dir} ---", file=sys.stderr, flush=True) + + work_folder = temp_dir / problem_id + + # 【修改点 1】定义临时zip文件的路径 + temp_zip_path = temp_dir / f"{problem_id}.zip" + + print(f"📂 复制: {source_folder} -> {work_folder}", file=sys.stderr, flush=True) + shutil.copytree(source_folder, work_folder) + + # 【修改点 2】在临时目录中创建zip文件 + print(f"📦 在临时目录中创建ZIP: {temp_zip_path}", file=sys.stderr, flush=True) + create_zip_with_folder(work_folder, temp_zip_path) + + # 【修改点 3】将临时zip文件移动到最终位置 + print(f"🚚 移动ZIP文件: {temp_zip_path} -> {target_zip}", file=sys.stderr, flush=True) + shutil.move(str(temp_zip_path), str(target_zip)) + + # 移动文件夹（这部分逻辑不变） + print(f"🚚 移动文件夹: {work_folder} -> {target_folder}", file=sys.stderr, flush=True) + shutil.move(str(work_folder), str(target_folder)) + + print("\n--- 操作完成，验证结果 ---", file=sys.stderr, flush=True) + if not target_folder.exists() or not target_zip.exists(): + raise Exception("验证失败，目标文件或文件夹未成功创建！") + + print(f"\n✅ 成功!", file=sys.stderr, flush=True) + print(f" -> 文件夹位置: {target_folder}", file=sys.stderr, flush=True) + print(f" -> ZIP文件位置: {target_zip}", file=sys.stderr, flush=True) + + print(f"\n📋 ZIP文件内容预览:", file=sys.stderr, flush=True) + with zipfile.ZipFile(target_zip, 'r') as zipf: + file_list = zipf.namelist() + for filename in file_list[:10]: + print(f" {filename}", file=sys.stderr, flush=True) + if len(file_list) > 10: + print(f" ... 还有 {len(file_list) - 10} 个文件", file=sys.stderr, flush=True) + + except Exception as e: + print(f"\n❌ 处理过程中发生严重错误: {e}") + import traceback + traceback.print_exc() + sys.exit(1) \ No newline at end of file diff --git a/web/api/getresult.py b/web/api/getresult.py new file mode 100644 index 000000000..ee8f5410a --- /dev/null +++ b/web/api/getresult.py @@ -0,0 +1,48 @@ +import json +import pymysql +import zipfile +import os, re +import random +import string +import sys +from pathlib import Path +import argparse + +from database_interface import UOJDatabaseInterface, sync_problem +from compile_checker import compile_checker +from data_preprocess import renamePrefixToA, generate_problem_conf + +def main(): + parser = argparse.ArgumentParser(description="批量配置UOJ题目") + parser.add_argument("submission_id", type=str, help="题目ID") + parser.add_argument("--db_host", type=str, default="uoj-db", help="数据库主机") + parser.add_argument("--db_user", type=str, default="root", help="数据库用户名") + parser.add_argument("--db_password", type=str, default="root", help="数据库密码") + parser.add_argument("--db_name", type=str, default="app_uoj233", help="数据库名称") + parser.add_argument("--uoj_problem_base", type=str, default="/var/uoj_data/", help="UOJ题目存储路径") + parser.add_argument("--storage_path", type=str, default="/opt/uoj/web/app/storage/", help="存储路径") + + args = parser.parse_args() + + db_config = { + 'host': args.db_host, + 'port': 3306, + 'user': args.db_user, + 'password': args.db_password, + 'database': 'app_uoj233', + } + + db_interface = UOJDatabaseInterface(db_config, args.storage_path) + + db_interface.reconnect() + submission_result = db_interface.fetch_result(args.submission_id) + details = submission_result.get("details", "") + matches = re.findall(r'info="([^"]+)"', details) + info = matches[-1] if matches else "N/A" + res_s = {"score": submission_result.get("score", 0), "info": info, "submission_id": args.submission_id} + if "error" in submission_result: + res_s["info"] = submission_result["error"] + print(json.dumps(res_s, ensure_ascii=False, indent=4)) + +if __name__ == "__main__": + main() \ No newline at end of file diff --git a/web/app/controllers/adminsubmit.php b/web/app/controllers/adminsubmit.php new file mode 100644 index 000000000..7b16183ad --- /dev/null +++ b/web/app/controllers/adminsubmit.php @@ -0,0 +1,74 @@ + "error", + "message" => "请使用 POST 方法提交代码。" + ]); + exit; +} + +// 2. Validate incoming parameters. +// Check if 'problem_id', 'language', and 'code' are set and not empty. +if (!isset($_POST['problem_id']) || !isset($_POST['language']) || !isset($_POST['code'])) { + // If any parameter is missing, return a client error. + header('Content-Type: application/json; charset=utf-8'); + http_response_code(400); // Bad Request + echo json_encode([ + "status" => "error", + "message" => "请求缺少必要的参数：problem_id, language, 或 code。" + ]); + exit; +} + +// 3. Retrieve and prepare the parameters for the shell command. +$problemId = escapeshellarg($_POST['problem_id']); +$language = escapeshellarg($_POST['language']); +$code = escapeshellarg($_POST['code']); + +// Define the path to the Python submission script. +$pythonScriptPath = escapeshellarg("/opt/uoj/web/api/adminsubmit.py"); + +// 4. Construct and execute the shell command. +// The command passes problem_id, language, and the code as arguments to the Python script. +$command = "python3 $pythonScriptPath $problemId $language $code"; + +// 'exec' runs the command, populates the $output array with its output lines, +// and sets the $retcode with the script's exit code. +exec($command, $output, $retcode); + +// 5. Handle the response from the Python script. + +// Set the HTTP header to indicate a JSON response. +header('Content-Type: application/json; charset=utf-8'); + +if ($retcode !== 0) { + // If the Python script returned a non-zero exit code, it indicates an error. + http_response_code(500); // Internal Server Error + echo json_encode([ + "status" => "error", + "message" => "代码提交处理失败。", + "details" => implode("\n", $output) // Provide detailed output from the script. + ], JSON_UNESCAPED_UNICODE | JSON_PRETTY_PRINT); +} else { + // If the script executed successfully (exit code 0). + // The Python script's standard output is expected to be the primary result. + echo json_encode([ + "status" => "success", + "output" => implode("\n", $output) // Combine output lines into a single string. + ], JSON_UNESCAPED_UNICODE | JSON_PRETTY_PRINT); +} +?> \ No newline at end of file diff --git a/web/app/controllers/batchsetup.php b/web/app/controllers/batchsetup.php new file mode 100644 index 000000000..78b0edc24 --- /dev/null +++ b/web/app/controllers/batchsetup.php @@ -0,0 +1,60 @@ +open($tmpZipPath) === TRUE) { + $zip->extractTo($tmpExtractDir); + $zip->close(); +} else { + becomeMsgPage("zip 文件解压失败！"); +} + +$pythonScriptPath = escapeshellarg("/opt/uoj/web/api/batchsetup.py"); +$targetDir = escapeshellarg($tmpExtractDir); +$command = "python3 $pythonScriptPath $targetDir"; +$json_output = shell_exec($command); +exec("python3 $pythonScriptPath $targetDir", $output, $retcode); + +if ($retcode !== 0) { + // 如果 Python 脚本出错，返回错误信息 + becomeMsgPage("处理失败：\n" . implode("\n", $output)); +} + +// ==================【核心修改部分】================== + + +if ($retcode !== 0) { + becomeMsgPage("处理失败：\n" . implode("\n", $output)); +} + +// 1. 设置正确的 HTTP 响应头，告诉浏览器返回的是 JSON +header('Content-Type: application/json; charset=utf-8'); + +// 2. 将 Python 输出的行数组合并成一个字符串，并直接输出 +// 这就是 Python 脚本产生的 JSON 内容 +echo implode("\n", $output); + +echo json_encode([ + "status" => "success", + "output" => $output +], JSON_PRETTY_PRINT); +?> diff --git a/web/app/controllers/download.php b/web/app/controllers/download.php index 7d735fe9d..630f23f73 100644 --- a/web/app/controllers/download.php +++ b/web/app/controllers/download.php @@ -26,6 +26,10 @@ $file_name = "/var/uoj_data/$id/download.zip"; $download_name = "problem_$id.zip"; break; + case 'data': + $file_name = "/var/uoj_data/$id.zip"; + $download_name = "data_$id.zip"; + break; case 'testlib.h': $file_name = "/opt/uoj/judger/uoj_judger/include/testlib.h"; $download_name = "testlib.h"; diff --git a/web/app/controllers/getresult.php b/web/app/controllers/getresult.php new file mode 100644 index 000000000..cc0cd42dc --- /dev/null +++ b/web/app/controllers/getresult.php @@ -0,0 +1,65 @@ + "error", + "message" => "请使用 GET 方法获取结果。" + ]); + exit; +} + +// 2. 验证必要的参数 'submission_id' 是否存在 +if (!isset($_GET['submission_id']) || empty($_GET['submission_id'])) { + header('Content-Type: application/json; charset=utf-8'); + http_response_code(400); // 400 Bad Request + echo json_encode([ + "status" => "error", + "message" => "请求缺少必要的参数：submission_id。" + ]); + exit; +} + +// 3. 获取并安全地处理参数 +// 使用 escapeshellarg() 来防止任何命令注入的风险 +$submissionId = escapeshellarg($_GET['submission_id']); + +// 定义后端的 Python 脚本路径 +$pythonScriptPath = escapeshellarg("/opt/uoj/web/api/getresult.py"); + +// 4. 构建并执行 shell 命令 +$command = "python3 $pythonScriptPath $submissionId"; + +// 执行命令，获取输出和返回码 +exec($command, $output, $retcode); + +// 5. 处理并返回结果 + +// 总是返回 JSON 格式 +header('Content-Type: application/json; charset=utf-8'); + +if ($retcode !== 0) { + // 如果 Python 脚本返回非零值，说明发生了错误（例如：提交ID不存在） + // 返回由 PHP 生成的统一错误信息 + http_response_code(404); // 404 Not Found 可能是个合适的错误码 + echo json_encode([ + "status" => "error", + "message" => "获取结果失败。", + "details" => implode("\n", $output) // 从脚本获取具体的错误细节 + ], JSON_UNESCAPED_UNICODE | JSON_PRETTY_PRINT); +} else { + // 如果 Python 脚本成功执行（返回码为0） + // 我们假定 Python 脚本的输出本身就是一个完整的 JSON 字符串。 + // 直接将这个 JSON 字符串作为响应体返回，这是最高效的方式。 + echo implode("\n", $output); +} +?> \ No newline at end of file diff --git a/web/app/libs/uoj-data-lib.php b/web/app/libs/uoj-data-lib.php index b38356bb2..210d1e62d 100644 --- a/web/app/libs/uoj-data-lib.php +++ b/web/app/libs/uoj-data-lib.php @@ -86,7 +86,7 @@ private function compile_at_prepare($name, $config = array()) { $work_path = $this->prepare_dir; } - $cmd_prefix = "$uojMainJudgerWorkPath/run/run_program >{$this->prepare_dir}/run_compiler_result.txt --in=/dev/null --out=stderr --err={$this->prepare_dir}/compiler_result.txt --tl=10 --ml=512 --ol=64 --type=compiler --work-path={$work_path}"; + $cmd_prefix = "$uojMainJudgerWorkPath/run/run_program >{$this->prepare_dir}/run_compiler_result.txt --in=/dev/null --out=stderr --err={$this->prepare_dir}/compiler_result.txt --tl=5000 --ml=512 --ol=64 --type=compiler --work-path={$work_path}"; if (isset($config['need_include_header']) && $config['need_include_header']) { exec("$cmd_prefix --add-readable-raw=$include_path/ /usr/bin/g++ -o $name {$config['src']} -I$include_path -lm -O2 -DONLINE_JUDGE"); } else { diff --git a/web/app/libs/uoj-html-lib.php b/web/app/libs/uoj-html-lib.php index fbf6981ad..f1c80bfd9 100644 --- a/web/app/libs/uoj-html-lib.php +++ b/web/app/libs/uoj-html-lib.php @@ -451,10 +451,14 @@ function echoPasteContent($paste) { switch ($file_language) { case 'C++': case 'C++11': + case 'C++14': + case 'C++17': + case 'C++20': $sh_class = 'sh_cpp'; break; case 'Python2': case 'Python3': + case 'PyPy3': $sh_class = 'sh_python'; break; case 'Java8': @@ -503,10 +507,14 @@ function echoSubmissionContent($submission, $requirement) { switch ($file_language) { case 'C++': case 'C++11': + case 'C++14': + case 'C++17': + case 'C++20': $sh_class = 'sh_cpp'; break; case 'Python2': case 'Python3': + case 'PyPy3': $sh_class = 'sh_python'; break; case 'Java8': diff --git a/web/app/libs/uoj-judger-lib.php b/web/app/libs/uoj-judger-lib.php index c86f26fb7..46883b3f4 100644 --- a/web/app/libs/uoj-judger-lib.php +++ b/web/app/libs/uoj-judger-lib.php @@ -1,6 +1,6 @@