clangmt.hpp

Go to the documentation of this file.

 
#ifndef CBUILD_CLANGMT_TOOLCHAIN
#define CBUILD_CLANGMT_TOOLCHAIN
// Project files
#include "../CBuild_defs.hpp"
#include "../hasher/cbuild_hash.hpp"
#include "./Build.hpp"
#include "atomic"
#include "thread"
// Code
namespace CBuild {
template <CBuild::HashImpl hash = CBuild::CBuildHashV2> class CLANGMT : public CBuild::Toolchain {
  public:
    CLANGMT(std::string id) {
        this->id = id;
        this->name = "";
        this->linker = "clang";
        this->compiler = "clang";
        this->packer = "ar cr";
        this->add_link_arg("-Wl,-z,origin");
        this->add_link_arg(" -Wl,-rpath,\"\\$ORIGIN\"");
        this->hasher = new hash(this->id);
    }
    CLANGMT(std::string id, std::string name) {
        this->id = id;
        this->name = name;
        this->linker = "clang";
        this->compiler = "clang";
        this->packer = "ar cr";
        this->add_link_arg("-Wl,-z,origin");
        this->add_link_arg(" -Wl,-rpath,\"\\$ORIGIN\"");
        this->hasher = new hash(this->id);
    }
 
  protected:
    // For docs see g++.hpp
    void build() override {
        // Get all args
        std::string args;
        for (auto elem : this->compiler_args) {
            args += elem;
            args += " ";
        }
        // Get all files
        auto files = this->gen_file_list(this->force);
        // Read pointer
        std::atomic_int64_t read_ptr;
        read_ptr.store(files.size() - 1);
        // Get hw thread count
        unsigned int num_threads = std::thread::hardware_concurrency();
        // Setup threads
        std::thread threads[num_threads];
        for (unsigned int i = 0; i < num_threads; i++) {
            threads[i] = std::thread([&files, this, &args, &read_ptr](void) -> void {
                while (true) {
                    // Fetch read pointer
                    int64_t i = read_ptr.fetch_sub(1);
                    if (i < 0) {
                        return;
                    }
                    // Construct command
                    std::string cmd = this->compiler + " -c ";
                    cmd += files.at(i).key;
                    cmd += " ";
                    cmd += args;
                    cmd += " -o ";
                    cmd += files.at(i).data;
                    // Execute command
                    std::this_thread::get_id();
 
                    this->compile(cmd);
                }
            });
        }
        for (unsigned int i = 0; i < num_threads; i++) {
            threads[i].join();
        }
    }
    void link() override {
        std::string args;
        for (auto elem : this->link_args) {
            args += elem;
            args += " ";
        }
        // Get all files
        auto files = this->gen_file_list(this->force);
        // Read pointer
        std::atomic_int64_t read_ptr;
        read_ptr.store(files.size() - 1);
        // Get hw thread count
        unsigned int num_threads = std::thread::hardware_concurrency();
        // Setup threads
        std::thread threads[num_threads];
        for (unsigned int i = 0; i < num_threads; i++) {
            threads[i] = std::thread([&files, this, &args, &read_ptr](void) -> void {
                while (true) {
                    // Fetch read pointer
                    int64_t i = read_ptr.fetch_sub(1);
                    if (i < 0) {
                        return;
                    }
                    // Construct command
                    std::string cmd = this->compiler + " -c ";
                    cmd += files.at(i).key;
                    cmd += " ";
                    cmd += args;
                    cmd += " -o ";
                    cmd += files.at(i).data;
                    // Execute command
                    std::this_thread::get_id();
 
                    this->compile(cmd);
                }
            });
        }
        for (unsigned int i = 0; i < num_threads; i++) {
            threads[i].join();
        }
    }
    void link_pack() override {
        std::string args;
        for (auto elem : this->link_args) {
            args += elem;
            args += " ";
        }
        this->gen_file_list_for_linking = true;
        auto files = this->gen_file_list(true);
        this->gen_file_list_for_linking = false;
        std::string flist;
        for (unsigned int i = 0; i < files.size(); i++) {
            flist += files.at(i).data;
            flist += " ";
        }
        if (files.size() > 0) {
            std::string cmd = this->packer + " ";
            cmd += this->gen_out_name();
            cmd += " ";
            cmd += flist;
            cmd += " ";
            // CBuild::print(cmd.c_str(), CBuild::BLUE);
            this->compile(cmd);
        }
    }
};
} // namespace CBuild
#endif // CBUILD_CLANGMT_TOOLCHAIN