dvereb/Dual-Lookup
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Added the ability to force which value is returned. If you're unsure whether you're passing VALUE or its EQUIVALENT to the function, you can force OPPOSITE, VALUE, or EQUIVALENT as a return value.

Browse Source
This commit is contained in:
David Vereb 2017-11-10 14:57:31 -05:00
parent 50d7342551
commit 757f877a5b
2 changed files with 55 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
DualLookup.h
View File

@ -7,6 +7,16 @@ class DualLookup {
public: public:
DualLookup(); DualLookup();
// NOTE(dev): Used to determine which version of the string you want:
// 1 / OPPOSITE: The string it maps to, opposite of the one you pass in.
// 2 / VALUE: The string passed via the first paramater of 'add.'
// 3 / EQUIVALENT: The string passed via the second parameter of 'add.'
enum Type {
OPPOSITE = 1,
VALUE,
EQUIVALENT,
};
/* Add a mapped pair to the container /* Add a mapped pair to the container
* returns false if it already exists, regardless of direction * returns false if it already exists, regardless of direction
* (i.e. value->equivilent is a duplicate of equivilent->value) * (i.e. value->equivilent is a duplicate of equivilent->value)
@ -16,7 +26,7 @@ public:
/* If found, Set result to the mapped value and true is returned. /* If found, Set result to the mapped value and true is returned.
* If not found, result remains unchanged and false is returned. * If not found, result remains unchanged and false is returned.
*/ */
bool get(T key, T &result); bool get(T key, T &result, const Type &type = Type::OPPOSITE);
/* If found, return true /* If found, return true
* else return false */ * else return false */
@ -27,7 +37,7 @@ private:
std::unordered_map<T, T> mirror; std::unordered_map<T, T> mirror;
// NOTE(dev): These functions actually do the work: // NOTE(dev): These functions actually do the work:
bool get_single(T key, T &result, const std::unordered_map<T, T> &umap); bool get_single(const T &key, T &result, const std::unordered_map<T, T> &umap);
bool contains_single(const T &key, const std::unordered_map<T, T> &umap); bool contains_single(const T &key, const std::unordered_map<T, T> &umap);
}; };
@ -52,12 +62,22 @@ bool DualLookup<T>::add(T value, T equivilent)
} }
template <class T> template <class T>
bool DualLookup<T>::get(T key, T &result) bool DualLookup<T>::get(T key, T &result, const Type &type)
{ {
if(get_single(key, result, owner)) if(get_single(key, result, owner))
{
//if(type == Type::OPPOSITE || type == Type::EQUIVALENT)
if(type == Type::VALUE)
result = key;
return true; return true;
}
if(get_single(key, result, mirror)) if(get_single(key, result, mirror))
{
//if(type == Type::OPPOSITE || type == Type::VALUE)
if(type == Type::EQUIVALENT)
result = key;
return true; return true;
}
return false; // didn't find it return false; // didn't find it
} }
@ -74,7 +94,7 @@ bool DualLookup<T>::contains(const T &key)
} }
template <class T> template <class T>
bool DualLookup<T>::get_single(T key, T &result, const std::unordered_map<T, T> &umap) bool DualLookup<T>::get_single(const T &key, T &result, const std::unordered_map<T, T> &umap)
{ {
auto i = umap.find(key); auto i = umap.find(key);
if(i == umap.end()) if(i == umap.end())

31
main.cpp
View File

@ -56,5 +56,36 @@ TEST_CASE("all tests", "all") {
REQUIRE(string_container_result == "2"); REQUIRE(string_container_result == "2");
} }
} }
SECTION ("choosing result side") {
int container_result;
container.add(1, 2);
container_result = 0;
REQUIRE(container.get(1, container_result));
CHECK(container_result == 2);
container_result = 0;
REQUIRE(container.get(1, container_result, DualLookup<int>::Type::OPPOSITE));
CHECK(container_result == 2);
container_result = 0;
REQUIRE(container.get(1, container_result, DualLookup<int>::Type::EQUIVALENT));
CHECK(container_result == 2);
container_result = 0;
REQUIRE(container.get(1, container_result, DualLookup<int>::Type::VALUE));
CHECK(container_result == 1);
container_result = 0;
REQUIRE(container.get(2, container_result));
CHECK(container_result == 1);
container_result = 0;
REQUIRE(container.get(2, container_result, DualLookup<int>::Type::OPPOSITE));
CHECK(container_result == 1);
container_result = 0;
REQUIRE(container.get(2, container_result, DualLookup<int>::Type::EQUIVALENT));
CHECK(container_result == 2);
container_result = 0;
REQUIRE(container.get(2, container_result, DualLookup<int>::Type::VALUE));
CHECK(container_result == 1);
}
} }
} }