yadro-task/bin/ftsort/ftsort.cpp

#include "tape_config.h"
#include "tapelib/tape_util.h"
#include <chrono>
#include <cstddef>
#include <fstream>
#include <iostream>
#include <string>

size_t get_mem_peak() {
    const static std::string mem_info_file = "/proc/self/status";
    const static std::string mem_field = "VmPeak:";
    std::ifstream status(mem_info_file);
    size_t mem = 0;
    std::string line;
    while (std::getline(status, line)) {
        if (line.find(mem_field) != std::string::npos) {
            std::stringstream mem_line(line.substr(mem_field.size()));
            mem_line >> mem;
            return mem;
        }
    }
    return mem;
}

int main(int argc, char *argv[]) {
    AppSettings settings = parse_command_line(argc, argv);
    if (settings.version) {
        std::cout << VERSION_MSG << std::endl;
        return 0;
    }
    if (settings.help) {
        std::cerr << HELP_MSG << std::endl;
        return 0;
    }
    if (!settings.config_file_path.empty() &&
        !read_settings_from_file(settings)) {
        return -1;
    }
    tape::FileTape input(settings.input_file_name, settings.ft_settings);
    if (!input.is_open()) {
        std::cerr << "Failed to create INPUT tape - file not found or "
                     "insufficient permissions"
                  << std::endl;
        return -1;
    }
    tape::FileTape output(input, settings.output_file_name,
                          settings.ft_settings);
    if (!output.is_open()) {
        std::cerr << "Failed to create OUTPUT tape - insufficient permissions"
                  << std::endl;
        return -1;
    }
    // tmp tape factory that captures settings.ft_settings from local scope
    tape::TempTapeFactory factory =
        [&](size_t cells) -> std::unique_ptr<tape::Tape> {
        return std::make_unique<tape::FileTape>(
            tape::FileTape(cells, settings.ft_settings));
    };
    auto start = std::chrono::high_resolution_clock::now();
    tape::external_sort(input, output, factory, settings.memory_limit);
    auto end = std::chrono::high_resolution_clock::now();
    auto duration = std::chrono::duration_cast<milliseconds>(end - start);
    std::cout << "Successfully sorted " << settings.input_file_name << " into "
              << settings.output_file_name << std::endl;
    std::cout << "The operation took " << duration.count()
              << " ms to complete and peaked at " << get_mem_peak()
              << " KiB of virtual memory usage" << std::endl;
    return 0;
}