BatchJobSetCoreLevel.h

 
#ifndef WRENCH_BATCHJOBSETCORELEVEL_H
#define WRENCH_BATCHJOBSETCORELEVEL_H
 
#include <wrench/services/compute/batch/BatchJob.h>
 
 
namespace wrench {
 
    /***********************/
    /***********************/
 
    class BatchJobSetCoreLevel {
 
    public:
 
        BatchJobSetCoreLevel() = default;
 
        std::set<std::shared_ptr<BatchJob>> jobs;
 
        std::map<int, unsigned long> core_utilization;
 
        BatchJobSetCoreLevel& operator += (const BatchJobSetCoreLevel& right)
        {
            for (const auto &j : right.jobs) {
                add(j);
            }
            return *this;
        }
 
        BatchJobSetCoreLevel& operator -= (const BatchJobSetCoreLevel& right)
        {
            for (const auto &j : right.jobs) {
                remove(j);
            }
            return *this;
        }
 
        void inline add(std::shared_ptr<BatchJob> job) {
            if (this->jobs.find(job) == this->jobs.end()) {
                for (auto const &i : job->getAllocatedNodeIndices()) {
                    if (this->core_utilization.find(i) == this->core_utilization.end())  {
                        this->core_utilization[i] = job->getRequestedCoresPerNode();
                    } else {
                        this->core_utilization[i] += job->getRequestedCoresPerNode();
                    }
                }
                this->jobs.insert(job);
            }
        }
 
        void inline remove(std::shared_ptr<BatchJob> job) {
            if  (this->jobs.find(job) != this->jobs.end()) {
                for (auto const &i : job->getAllocatedNodeIndices()) {
                    this->core_utilization[i] -= job->getRequestedCoresPerNode();
                }
                this->jobs.erase(job);
            }
        }
 
    };
 
    /***********************/
    /***********************/
 
}
 
#endif //WRENCH_BATCHJOBSETCORELEVEL_H