DataMaker/make.cpp at main · ZhongTerry/DataMaker · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
#define _CRT_SECURE_NO_WARNINGS
#include<fstream>
#include<vector>
#include<filesystem>
#include<cstdlib>
#include<random>
#include "includes/json.hpp"
#include "includes/zipfile.hpp"
using namespace std;
using namespace std::filesystem;
using namespace miniz_cpp;
using json = nlohmann::json;

random_device rd;

struct random{
    default_random_engine rdm;
    uniform_int_distribution<> dis;

    random(int low, int up){ rdm=default_random_engine(rd()); dis=uniform_int_distribution(low, up); }
    int operator() (){
        return dis(rdm);
    }
};

string strRand(int length) {			// length: �����ַ����ĳ���
    string buffer;						// buffer: ���淵��ֵ
    string charset="abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ";

    random rdm(0, charset.size() - 1);

    for (int i = 0; i < length; i++) {
        buffer += charset[rdm()];
    }
    return buffer;
}

#ifdef _WIN32
string resolve_name(string name){ return name; }
#else
string resolve_name(string name){ return "./"+name; }
#endif
int _execute(string complete){
    return system(complete.c_str());
}
int execute(string cmd){
    return _execute(resolve_name(cmd));
}
int execute(vector<string> cmd){
	string complete;
	for(string c:cmd)complete+=" ";
    return execute(complete);
}


void generate(string fn, int id, string exename, string std, zip_file& zf, int assumeid){
	string inp = fn + "/" + "in" + to_string(id) + ".in", out = fn + "/" + "out" + to_string(id) + ".out";
	freopen(inp.c_str(), "w", stdout);
	execute({exename, to_string(assumeid)});
	freopen(inp.c_str(), "r", stdin);
	freopen(out.c_str(), "w", stdout);
	execute({std});
	zf.write(inp);
	zf.write(out);
}

int main() {
	ios::sync_with_stdio(false);
	ofstream fout("log.log");
	//ifstream finname("config/Name.txt"), finnum("config/Number.txt"), finbind("config/bind.txt"), finexe("config/exe.txt"), findata("config/data.txt");
	ifstream config_file("config.json");
	json config = json::parse(config_file);
	string fn=config["name"];
	fn="data-"+fn;
	create_directories(fn);
	fout.flush();
	int mode=config["mode"];
	zip_file zf;

	int amount=0;
	if (mode==0) {
		string exename=config["exename"];

		amount=config["test_amount"];
		for (int i = 1; i <= amount; i ++) {
			generate(fn, i, exename, config["std"], zf, i);
			fout << i << ".out/.in Maked\n";
		}
	} else if (mode == 1) {
		int cnt = 0;
		string exename;
		ofstream off(fn + "\\config.yml");
		int T=config["subtasks"].size();
		int lasttask = 100 % T;
		json task_config = config["subtasks"];

		for(int t=0;t<T;t++){
			int begin = task_config[t]["begin"], end = task_config[t]["end"];
			// exename = task_config[t]["exename"];
			exename = config["exename"];
			for (int i=begin; i <= end; i ++){
				string s = fn + "/" + "in" + to_string(i) + ".in", s2 = fn + "/" + "out" + to_string(i) + ".out";
			    generate(fn, i, exename, config["std"], zf, t);
				off << "in"<< i << ".in:\n";
				off << "  subtaskId: " << cnt << "\n";
				if (cnt >= T - lasttask) off << "  score: " << 100 / T + 1 << "\n";
				else off << "  score: " << 100 / T << "\n";
				off << "\n";
				fout << i << ".out/.in Maked\n";
			}
			cnt++;
		}

		zf.write(fn+"\\config.yml");
	} else {
		int cnt = 0;
		string exename;
		ofstream off(fn + "\\config.yml");
		int T=config["subtasks"].size();
		int lasttask = 100 % T;
		json task_config = config["subtasks"];
		int su = task_config[T - 1]["end"];
		int lastsu = 100 % su;
		// fout << lastsu << "\n";
		for(int t=0;t<T;t++){
			int begin = task_config[t]["begin"], end = task_config[t]["end"];
			// exename = task_config[t]["exename"];
			exename = config["exename"];
			for (int i=begin; i <= end; i ++){
				string s = fn + "/" + "in" + to_string(i) + ".in", s2 = fn + "/" + "out" + to_string(i) + ".out";
			    generate(fn, i, exename, config["std"], zf, t);
				off << "in"<< i << ".in:\n";
				off << "  subtaskId: " << cnt << "\n";
				if (cnt >= su - lastsu) off << "  score: " << 100 / su + 1 << "\n";
				else off << "  score: " << 100 / su << "\n";
				off << "\n";
				fout << i << ".out/.in Maked\n";
				cnt ++;
			}

		}

		zf.write(fn+"\\config.yml");
	}
	zf.save(fn+".zip");

	//_execute("zip -r -q -j " + fn + ".zip " + fn);
}