LimitedStack.h

/*
 * Copyright 2016-2020 Dmitry Ivanov
 *
 * This file is part of libquentier
 *
 * libquentier is free software; you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as published by
 * the Free Software Foundation, version 3 of the License.
 *
 * libquentier 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 Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with libquentier. If not, see <http://www.gnu.org/licenses/>.
 */
 
#ifndef LIB_QUENTIER_UTILITY_LIMITED_STACK_H
#define LIB_QUENTIER_UTILITY_LIMITED_STACK_H
 
#include <QStack>
 
namespace quentier {

template <class T>
class LimitedStack : public QStack<T>
{
public:
    LimitedStack(void (*deleter)(T &) = nullptr) :
        m_limit(-1), m_deleter(deleter)
    {}
 
    LimitedStack(const LimitedStack<T> & other) :
        QStack<T>(other), m_limit(other.m_limit), m_deleter(other.m_deleter)
    {}
 
    LimitedStack(LimitedStack<T> && other) :
        QStack<T>(std::move(other)), m_limit(std::move(other.m_limit)),
        m_deleter(std::move(other.m_deleter))
    {}
 
    LimitedStack<T> & operator=(const LimitedStack<T> & other)
    {
        if (this != &other) {
            QStack<T>::operator=(other);
            m_limit = other.m_limit;
            m_deleter = other.m_deleter;
        }
 
        return *this;
    }
 
    LimitedStack<T> & operator=(LimitedStack<T> && other)
    {
        if (this != &other) {
            QStack<T>::operator=(std::move(other));
            m_limit = std::move(other.m_limit);
            m_deleter = std::move(other.m_deleter);
        }
 
        return *this;
    }
 
    ~LimitedStack()
    {
        if (m_deleter) {
            while (!QStack<T>::isEmpty()) {
                T t = QStack<T>::pop();
                (*m_deleter)(t);
            }
        }
    }
 
    void swap(const LimitedStack<T> & other)
    {
        int limit = other.m_limit;
        other.m_limit = m_limit;
        m_limit = limit;
 
        void (*deleter)(T &) = other.m_deleter;
        other.m_deleter = m_deleter;
        m_deleter = deleter;
 
        QStack<T>::swap(other);
    }
 
    int limit() const
    {
        return m_limit;
    }
    void setLimit(const int limit)
    {
        m_limit = limit;
    }
 
    void push(const T & t)
    {
        if ((m_limit > 0) && (QVector<T>::size() == m_limit - 1)) {
            if (m_deleter) {
                (*m_deleter)(*QVector<T>::begin());
            }
            Q_UNUSED(QVector<T>::erase(QVector<T>::begin()));
        }
 
        QStack<T>::push(t);
    }
 
private:
    int m_limit;
    void (*m_deleter)(T &);
};
 
} // namespace quentier
 
#endif // LIB_QUENTIER_UTILITY_LIMITED_STACK_H