Only lock once per iteration

This commit is contained in:
eriedaberrie 2024-01-08 21:56:16 -08:00
parent d218af9c4a
commit ff65b49da4

View file

@ -182,16 +182,10 @@ int main(int argc, char *argv[]) {
std::cout << "Starting " << wc << " worker threads..." << std::endl;
for (unsigned int w = 1; w <= wc; w++) {
workers.push_back(std::thread([&it, &n_done, &it_lock, &all_fires,
&residentials, &out_dir, w, distance]() {
for (;;) {
it_lock.lock();
if (it == all_fires.end()) {
std::cout << "Worker " << w << " done! (finished " << n_done
<< "/" << all_fires.size() << ")" << std::endl;
it_lock.unlock();
return;
}
&residentials, &out_dir, w, distance]() {
it_lock.lock();
while (it != all_fires.end()) {
const auto& month_data = *(it++);
auto& month = month_data.name;
auto& month_fires = month_data.days;
@ -225,12 +219,13 @@ int main(int argc, char *argv[]) {
outf << '\n';
}
{
std::lock_guard<std::mutex> lk(it_lock);
n_done++;
}
it_lock.lock();
n_done++;
}
std::cout << "Worker " << w << " done! (finished " << n_done
<< "/" << all_fires.size() << ")" << std::endl;
it_lock.unlock();
}));
}