71 lines
1.9 KiB
C++
71 lines
1.9 KiB
C++
#ifndef TOOLS_H
|
||
#define TOOLS_H
|
||
|
||
#include <QHash>
|
||
#include <QStandardItem>
|
||
|
||
template <typename Key, typename Value> //双向map工具类，实现key->value,value->key的映射
|
||
class BiDirectionalMap {
|
||
public:
|
||
// 插入键值对，确保双向唯一性
|
||
void insert(const Key& key, const Value& value) {
|
||
// 删除旧键和旧值的关联（如果存在）
|
||
if (m_keyToValue.contains(key)) {
|
||
Value oldValue = m_keyToValue[key];
|
||
m_valueToKey.remove(oldValue);
|
||
}
|
||
if (m_valueToKey.contains(value)) {
|
||
Key oldKey = m_valueToKey[value];
|
||
m_keyToValue.remove(oldKey);
|
||
}
|
||
// 插入新键值对
|
||
m_keyToValue[key] = value;
|
||
m_valueToKey[value] = key;
|
||
}
|
||
|
||
// 根据键获取值
|
||
Value value(const Key& key) const {
|
||
return m_keyToValue.value(key);
|
||
}
|
||
|
||
// 根据值获取键
|
||
Key key(const Value& value) const {
|
||
return m_valueToKey.value(value);
|
||
}
|
||
|
||
// 检查键是否存在
|
||
bool containsKey(const Key& key) const {
|
||
return m_keyToValue.contains(key);
|
||
}
|
||
|
||
// 检查值是否存在
|
||
bool containsValue(const Value& value) const {
|
||
return m_valueToKey.contains(value);
|
||
}
|
||
|
||
// 通过键删除项
|
||
void removeByKey(const Key& key) {
|
||
if (m_keyToValue.contains(key)) {
|
||
Value value = m_keyToValue[key];
|
||
m_keyToValue.remove(key);
|
||
m_valueToKey.remove(value);
|
||
}
|
||
}
|
||
|
||
// 通过值删除项
|
||
void removeByValue(const Value& value) {
|
||
if (m_valueToKey.contains(value)) {
|
||
Key key = m_valueToKey[value];
|
||
m_valueToKey.remove(value);
|
||
m_keyToValue.remove(key);
|
||
}
|
||
}
|
||
|
||
private:
|
||
QHash<Key, Value> m_keyToValue; // 键 → 值
|
||
QHash<Value, Key> m_valueToKey; // 值 → 键
|
||
};
|
||
|
||
int getLevel(QStandardItem *item);
|
||
#endif // DATABASE_H
|