NumThreadGuard.h

Go to the documentation of this file.

 
#pragma once
 
#include "autopas/utils/WrapOpenMP.h"
 
class NumThreadGuard final {
 public:
  explicit NumThreadGuard(const int newNum) : numThreadsBefore(autopas::autopas_get_max_threads()) {
    autopas::autopas_set_num_threads(newNum);
  }
 
  ~NumThreadGuard() { autopas::autopas_set_num_threads(numThreadsBefore); }
 
  NumThreadGuard(const NumThreadGuard &) = delete;
 
  NumThreadGuard &operator=(const NumThreadGuard &) = delete;
 
 private:
  int numThreadsBefore;
};