67 lines
1.6 KiB
C++
67 lines
1.6 KiB
C++
#include "config.hpp"
|
|
#include "record.hpp"
|
|
#include "sort.cpp"
|
|
#include "writer.hpp"
|
|
#include <fstream>
|
|
#include <iostream>
|
|
#include <sys/stat.h>
|
|
#include <sys/types.h>
|
|
|
|
void generate_file(int N, const std::string& filename)
|
|
{
|
|
std::filesystem::path dirPath = std::filesystem::path(filename).parent_path();
|
|
std::filesystem::create_directories(dirPath);
|
|
|
|
std::ofstream out_stream(filename);
|
|
Writer output(out_stream);
|
|
|
|
for (int i = 0; i < N; i++) {
|
|
Record rec = Record::random();
|
|
output.write(rec);
|
|
}
|
|
}
|
|
|
|
void read_and_evaluate(std::string filename, int n)
|
|
{
|
|
Buffer main_buffer(n);
|
|
auto buff = main_buffer.divide(1)[0];
|
|
|
|
std::ifstream in_stream(filename);
|
|
Reader input_reader(in_stream);
|
|
|
|
while (true) {
|
|
size_t records_read = read_chunk(input_reader, buff);
|
|
|
|
if (records_read == 0) {
|
|
break;
|
|
}
|
|
|
|
for (size_t i = 0; i < records_read; ++i) {
|
|
const Record& rec = buff[i];
|
|
std::cout << "[Evaluate i = " << i << ": " << rec.evaluate() << "] " << rec << "\n";
|
|
}
|
|
}
|
|
}
|
|
|
|
int main(int argc, char** argv)
|
|
{
|
|
try {
|
|
Configuration opts = Configuration::parse_args(argc, argv);
|
|
|
|
if (opts.generate_data) {
|
|
generate_file(opts.N, opts.output_file);
|
|
return 0;
|
|
}
|
|
|
|
if (opts.evaluate_file) {
|
|
read_and_evaluate(opts.input_file, opts.N);
|
|
return 0;
|
|
}
|
|
|
|
sort_file(opts);
|
|
|
|
} catch (const std::exception& e) {
|
|
std::cerr << "Error: " << e.what() << '\n';
|
|
return 1;
|
|
}
|
|
}
|