qmanga/ztextdocumentcontroller.cpp at master · kernel1024/qmanga · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
#include "ztextdocumentcontroller.h"

ZTextDocumentController::ZTextDocumentController(QObject *parent)
    : QObject(parent)
{
}

ZTextDocumentController::~ZTextDocumentController()
{
    QMutexLocker locker(&m_mutex);

    m_refs.clear();

    for (auto& doc : m_docs)
        doc->deleteLater();
    m_docs.clear();
}

void ZTextDocumentController::addDocument(const QString &filename, ZAbstractTextDocument *doc)
{
    QMutexLocker locker(&m_mutex);

    if (m_docs.contains(filename)) {
        auto *oldDoc = m_docs.take(filename);
        oldDoc->deleteLater();
    }
    m_docs.insert(filename,doc);
    m_refs.insert(filename,0);
}

QTextDocument *ZTextDocumentController::getDocument(const QString& filename)
{
    QMutexLocker locker(&m_mutex);

    if (!m_docs.contains(filename))
        return nullptr;

    ZAbstractTextDocument *doc = m_docs.value(filename);
    QTextDocument *res = doc->clone(nullptr);

    m_refs[filename]++;

    connect(res,&QTextDocument::destroyed,this,[this,filename](){
        QMutexLocker locker(&m_mutex);
        if (m_refs.contains(filename)) {
            m_refs[filename]--;
            if (m_refs.value(filename) == 0)
                m_refs.remove(filename);
        }
        if (!m_refs.contains(filename)) {
            if (m_docs.contains(filename)) {
                auto *doc = m_docs.take(filename);
                doc->deleteLater();
            }
        }
    });

    return res;
}

bool ZTextDocumentController::contains(const QString &filename)
{
    return m_docs.contains(filename);
}