Skip to content

Commit 2a892fc

Browse files
committed
fix: encapsulate added_traces
1 parent 956b667 commit 2a892fc

3 files changed

Lines changed: 6 additions & 3 deletions

File tree

source/active_learning/algorithms/ldot.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -98,10 +98,10 @@ void ldot_algorithm::proc_counter_record(inputdata& id, const sul_response& rec,
9898

9999
bool ldot_algorithm::add_trace(inputdata& id, const sul_response& r) {
100100
int pk = r.GET_ID();
101-
if (my_sul->added_traces.contains(pk)) {
101+
if (my_sul->contains_uid(pk)) {
102102
return false;
103103
}
104-
my_sul->added_traces.insert(pk);
104+
my_sul->insert_uid(pk);
105105

106106
LOG_S(1) << "pk:" << r.GET_ID() << " " << r.GET_INT() << " " << r.GET_INT_VEC();
107107

source/active_learning/oracle/sqldb_sul_random_oracle.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@ sqldb_sul_random_oracle::equivalence_query(state_merger* merger) {
3131
std::vector<int> possible_ids;
3232

3333
for (int pk = 0; pk <= max_pk; pk++) {
34-
if (!my_sqldb_sul->added_traces.contains(pk))
34+
if (!my_sqldb_sul->contains_uid(pk))
3535
possible_ids.push_back(pk);
3636
}
3737
std::shuffle(possible_ids.begin(), possible_ids.end(), RNG);

source/active_learning/system_under_learning/sqldb_sul.h

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,9 @@ class sqldb_sul : public sul_base {
2323
psql::db& my_sqldb;
2424
std::unordered_set<int> added_traces;
2525

26+
std::pair<std::unordered_set<int>::iterator, bool> insert_uid(int x) { return added_traces.insert(x); };
27+
bool contains_uid(int x) { return added_traces.contains(x); };
28+
2629
explicit sqldb_sul(psql::db& db) : my_sqldb(db){};
2730

2831
void pre(inputdata& id) override;

0 commit comments

Comments
 (0)