ResInsight/cafProjectDataModel/cafAppEnum.h
2012-05-18 09:45:23 +02:00

314 lines
10 KiB
C++

//##################################################################################################
//
// Custom Visualization Core library
// Copyright (C) 2011-2012 Ceetron AS
//
// This library is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// This library is distributed in the hope that it will be useful, but WITHOUT ANY
// WARRANTY; without even the implied warranty of MERCHANTABILITY or
// FITNESS FOR A PARTICULAR PURPOSE.
//
// See the GNU General Public License at <<http://www.gnu.org/licenses/gpl.html>>
// for more details.
//
//##################################################################################################
#pragma once
#include <vector>
#include <string>
#include <QtCore/QString>
#include <QTextStream>
namespace caf
{
//==================================================================================================
/// An enum class to make it easier to handle IO and UI based on the enum.
/// Usage:
/// In Header file of SomeClass:
/// enum SomeEnumType
/// {
/// A = 2,
/// B = 7
/// };
/// caf::AppEnum<SomeEnumType> m_enumValue;
///
/// In C++ file :
/// namespace caf {
/// template<>
/// void caf::AppEnum<SomeClass::SomeEnumType>::setUp()
/// {
/// addItem(SomeClass::A, "A", "An A letter");
/// addItem(SomeClass::B, "B", "A B letter");
/// setDefault(SomeClass::B);
/// }
/// }
/// General use:
///
/// m_enumValue = A;
/// if (m_enumValue == A || m_enumValue != B ){}
///
/// switch (m_enumValue)
/// {
/// case A:
/// break;
/// case B:
/// break;
/// }
///
/// cout << m_enumValue.text();
/// m_enumValue.setFromText("A");
///
/// for (size_t i = 0; i < caf::AppEnum<SomeClass::SomeEnumType>::size(); ++i)
/// cout << caf::AppEnum<SomeClass::SomeEnumType>::text(caf::AppEnum<SomeClass::SomeEnumType>::fromIndex(i)) << endl;
///
//==================================================================================================
template <class T>
class AppEnum
{
public:
AppEnum() { m_value = EnumMapper::instance()->defaultValue();}
AppEnum(T value): m_value(value) {}
bool operator== (T value) const { return m_value == value;}
bool operator!= (T value) const { return m_value != value;}
operator T () const { return m_value;}
QString text() const { return EnumMapper::instance()->text(m_value);}
QString uiText() const { return EnumMapper::instance()->uiText(m_value);}
AppEnum& operator= (T value) { m_value = value; return *this; }
bool setFromText(const QString& text){ return EnumMapper::instance()->enumVal(m_value, text);}
bool setFromIndex (size_t index) { return EnumMapper::instance()->enumVal(m_value, index); }
// Static interface to access the properties of the enum definition
static bool isValid(const QString& text) { return EnumMapper::instance()->isValid(text);}
static bool isValid(size_t index) { return index < EnumMapper::instance()->size();}
static size_t size() { return EnumMapper::instance()->size();}
static QStringList uiTexts() { return EnumMapper::instance()->uiTexts(); }
static T fromIndex (size_t idx) { T val; EnumMapper::instance()->enumVal(val, idx); return val;}
static T fromText(const QString& text) { T val; EnumMapper::instance()->enumVal(val, text); return val;}
static size_t index (T enumValue) { return EnumMapper::instance()->index(enumValue);}
static QString text (T enumValue) { return EnumMapper::instance()->text(enumValue);}
static QString textFromIndex (size_t idx) { return text(fromIndex(idx));}
static QString uiText(T enumValue) { return EnumMapper::instance()->uiText(enumValue);}
static QString uiTextFromIndex(size_t idx) { return uiText(fromIndex(idx)); }
private:
//==================================================================================================
/// The setup method is supposed to be specialized for each and every type instantiation of this class,
/// and is supposed to set up the mapping between enum values, text and ui-text using the \m addItem
/// method. It may also set a default value using \m setDefault
//==================================================================================================
static void setUp();
static void addItem(T enumVal, const QString& text, const QString& uiText)
{
EnumMapper::instance()->addItem(enumVal, text, uiText);
}
static void setDefault( T defaultEnumValue)
{
EnumMapper::instance()->setDefault(defaultEnumValue);
}
T m_value;
//==================================================================================================
/// A private class to handle the instance of the mapping vector.
/// all access methods could have been placed directly in the \class AppEnum class,
/// but AppEnum implementation gets nicer this way.
/// The real core of this class is the vector map member and the static instance method
//==================================================================================================
class EnumMapper
{
private:
struct Triplet
{
Triplet(T enumVal, const QString& text, QString uiText) :
m_enumVal(enumVal), m_text(text), m_uiText(uiText)
{}
T m_enumVal;
QString m_text;
QString m_uiText;
};
public:
void addItem(T enumVal, const QString& text, QString uiText)
{
instance()->m_mapping.push_back(Triplet(enumVal, text, uiText));
}
static EnumMapper * instance()
{
static EnumMapper * storedInstance = 0;
if (!storedInstance)
{
storedInstance = new EnumMapper;
AppEnum<T>::setUp();
}
return storedInstance;
}
void setDefault(T defaultEnumValue)
{
m_defaultValue = defaultEnumValue;
m_defaultValueIsSet = true;
}
T defaultValue() const
{
if (m_defaultValueIsSet)
{
return m_defaultValue;
}
else
{
// assert(m_mapping.size());
return m_mapping[0].m_enumVal;
}
}
bool isValid( const QString& text) const
{
size_t idx;
for(idx = 0; idx < m_mapping.size(); ++idx)
{
if (text == m_mapping[idx].m_text) return true;
}
return false;
}
size_t size() const
{
return m_mapping.size();
}
bool enumVal(T& value, const QString& text) const
{
value = defaultValue();
size_t idx;
for(idx = 0; idx < m_mapping.size(); ++idx)
{
if (text == m_mapping[idx].m_text)
{
value = m_mapping[idx].m_enumVal;
return true;
}
}
return false;
}
bool enumVal(T& value, size_t index) const
{
value = defaultValue();
if (index < m_mapping.size())
{
value = m_mapping[index].m_enumVal;
return true;
}
else
return false;
}
size_t index(T enumValue) const
{
size_t idx;
for(idx = 0; idx < m_mapping.size(); ++idx)
{
if (enumValue == m_mapping[idx].m_enumVal)
return idx;
}
return idx;
}
QString uiText(T value) const
{
size_t idx;
for(idx = 0; idx < m_mapping.size(); ++idx)
{
if (value == m_mapping[idx].m_enumVal) return m_mapping[idx].m_uiText;
}
return "";
}
QStringList uiTexts () const
{
QStringList uiTextList;
size_t idx;
for(idx = 0; idx < m_mapping.size(); ++idx)
{
uiTextList.append(m_mapping[idx].m_uiText);
}
return uiTextList;
}
QString text(T value) const
{
size_t idx;
for(idx = 0; idx < m_mapping.size(); ++idx)
{
if (value == m_mapping[idx].m_enumVal) return m_mapping[idx].m_text;
}
return "";
}
private:
EnumMapper() : m_defaultValueIsSet(false) { }
friend class AppEnum<T>;
std::vector<Triplet> m_mapping;
T m_defaultValue;
bool m_defaultValueIsSet;
};
};
}
//==================================================================================================
/// Cant remember why we need those comparison operators...
//==================================================================================================
template<class T>
bool operator == ( T value, const caf::AppEnum<T>& appEnum)
{
return (appEnum == value);
}
template<class T>
bool operator != ( T value, const caf::AppEnum<T>& appEnum)
{
return (appEnum != value);
}
//==================================================================================================
/// Implementation of stream operators to make PdmField<AppEnum<> > work smoothly
/// Assumes that the stream ends at the end of the enum text
//==================================================================================================
template < typename T >
void operator >> (QTextStream& str, caf::AppEnum<T>& appEnum)
{
QString text;
str >> text;
appEnum.setFromText(text);
}
template < typename T >
void operator << (QTextStream& str, const caf::AppEnum<T>& appEnum)
{
str << appEnum.text();
}