12345qiupeng
/
gtsam
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

nullptr check and assign default

release/4.3a0
Varun Agrawal 2025-05-16 00:09:59 -04:00
parent 8b6ce55151
commit ab0a4a801e
2 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
gtsam/discrete/DiscreteConditional.cpp
View File

@ -309,6 +309,10 @@ size_t DiscreteConditional::sample(const DiscreteValues& parentsValues,
return value; // shortcut exit return value; // shortcut exit
} }
} }
// Check if rng is nullptr, then assign default
rng = (rng == nullptr) ? &kRandomNumberGenerator : rng;
std::discrete_distribution<size_t> distribution(p.begin(), p.end()); std::discrete_distribution<size_t> distribution(p.begin(), p.end());
return distribution(*rng); return distribution(*rng);
} }

4
gtsam/discrete/TableDistribution.cpp
View File

@ -171,6 +171,10 @@ size_t TableDistribution::sample(const DiscreteValues& parentsValues,
return value; // shortcut exit return value; // shortcut exit
} }
} }
// Check if rng is nullptr, then assign default
rng = (rng == nullptr) ? &kRandomNumberGenerator : rng;
std::discrete_distribution<size_t> distribution(p.begin(), p.end()); std::discrete_distribution<size_t> distribution(p.begin(), p.end());
return distribution(*rng); return distribution(*rng);
} }