eriedaberrie/shapefile-thing
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

Only lock once per iteration

Browse source
This commit is contained in:
eriedaberrie 2024-01-08 21:56:16 -08:00
parent d218af9c4a
commit ff65b49da4
1 changed files with 9 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

23
src/main.cpp
View file

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