FileLogger.h

Go to the documentation of this file.

#ifndef FILE_LOGGER_H
#define FILE_LOGGER_H
 
#pragma once
 
#include <fstream>
#include <string>
#include <filesystem>
#include <cstddef>
#include "core/Population.h"
 
namespace galib::utils {
    template <typename GeneType>
    class FileLogger {
    private:
        std::string path_m;
        std::size_t interval_m;
        bool header_written_m = false;
 
    public:
        FileLogger(const std::string& path, const std::size_t interval)
            : path_m(path), interval_m(interval) {}
 
        void log(std::size_t gen, const Population<GeneType>& pop) {
            if (interval_m == 0 || (gen % interval_m != 0 && gen != 0)) {
                return;
            }
 
            std::filesystem::path p(path_m);
            if (!header_written_m) {
                if (p.has_parent_path()) {
                    std::filesystem::create_directories(p.parent_path());
                }
            }
 
            std::ofstream file;
            if (!header_written_m) {
                file.open(path_m, std::ios::out);
                if (file.is_open()) {
                    file << "generation,individual_idx,fitness";
                    for (std::size_t j = 0; j < pop.getNumGenes(); ++j) {
                        file << ",gene_" << j;
                    }
                    file << ",genotype\n";
                    header_written_m = true;
                }
            } else {
                file.open(path_m, std::ios::app);
            }
 
            if (!file.is_open()) {
                return;
            }
 
            for (std::size_t i = 0; i < pop.size(); ++i) {
                const auto& genotype = pop[i].getGenotype();
                file << gen << "," << i << "," << pop[i].getFitness();
                
                for (const auto& gene : genotype) {
                    file << "," << gene;
                }
                file << ",";
 
                for (std::size_t j = 0; j < genotype.size(); ++j) {
                    file << genotype[j] << (j == genotype.size() - 1 ? "" : ";");
                }
                file << "\n";
            }
            file.close();
        }
    };
} // namespace galib::utils
 
#endif // FILE_LOGGER_H