cppfindreferences.cpp 11.8 KB
Newer Older
Roberto Raggi's avatar
Roberto Raggi committed
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
/**************************************************************************
**
** This file is part of Qt Creator
**
** Copyright (c) 2009 Nokia Corporation and/or its subsidiary(-ies).
**
** Contact: Nokia Corporation (qt-info@nokia.com)
**
** Commercial Usage
**
** Licensees holding valid Qt Commercial licenses may use this file in
** accordance with the Qt Commercial License Agreement provided with the
** Software or, alternatively, in accordance with the terms contained in
** a written agreement between you and Nokia.
**
** GNU Lesser General Public License Usage
**
** Alternatively, this file may be used under the terms of the GNU Lesser
** General Public License version 2.1 as published by the Free Software
** Foundation and appearing in the file LICENSE.LGPL included in the
** packaging of this file.  Please review the following information to
** ensure the GNU Lesser General Public License version 2.1 requirements
** will be met: http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html.
**
** If you are unsure which license is appropriate for your use, please
con's avatar
con committed
26
** contact the sales department at http://qt.nokia.com/contact.
Roberto Raggi's avatar
Roberto Raggi committed
27 28 29 30
**
**************************************************************************/

#include "cppfindreferences.h"
31
#include "cppmodelmanagerinterface.h"
Roberto Raggi's avatar
Roberto Raggi committed
32 33 34 35 36 37 38
#include "cpptoolsconstants.h"

#include <texteditor/basetexteditor.h>
#include <find/searchresultwindow.h>
#include <extensionsystem/pluginmanager.h>
#include <utils/filesearch.h>
#include <coreplugin/progressmanager/progressmanager.h>
39
#include <coreplugin/editormanager/editormanager.h>
Roberto Raggi's avatar
Roberto Raggi committed
40 41 42 43 44 45 46 47
#include <coreplugin/icore.h>

#include <ASTVisitor.h>
#include <AST.h>
#include <Control.h>
#include <Literals.h>
#include <TranslationUnit.h>
#include <Symbols.h>
48 49
#include <Names.h>
#include <Scope.h>
Roberto Raggi's avatar
Roberto Raggi committed
50 51

#include <cplusplus/CppDocument.h>
52
#include <cplusplus/CppBindings.h>
53
#include <cplusplus/Overview.h>
Roberto Raggi's avatar
Roberto Raggi committed
54 55 56 57

#include <QtCore/QTime>
#include <QtCore/QtConcurrentRun>
#include <QtCore/QDir>
58
#include <QtGui/QApplication>
Roberto Raggi's avatar
Roberto Raggi committed
59 60 61 62 63
#include <qtconcurrent/runextensions.h>

using namespace CppTools::Internal;
using namespace CPlusPlus;

64
CppFindReferences::CppFindReferences(CppTools::CppModelManagerInterface *modelManager)
65 66
    : QObject(modelManager),
      _modelManager(modelManager),
Roberto Raggi's avatar
Roberto Raggi committed
67 68 69
      _resultWindow(ExtensionSystem::PluginManager::instance()->getObject<Find::SearchResultWindow>())
{
    m_watcher.setPendingResultsLimit(1);
Roberto Raggi's avatar
Roberto Raggi committed
70
    connect(&m_watcher, SIGNAL(resultsReadyAt(int,int)), this, SLOT(displayResults(int,int)));
Roberto Raggi's avatar
Roberto Raggi committed
71 72 73 74 75 76 77
    connect(&m_watcher, SIGNAL(finished()), this, SLOT(searchFinished()));
}

CppFindReferences::~CppFindReferences()
{
}

78 79 80 81
QList<int> CppFindReferences::references(Symbol *symbol,
                                         Document::Ptr doc,
                                         const Snapshot& snapshot) const
{
Roberto Raggi's avatar
Roberto Raggi committed
82 83
    const Identifier *id = 0;
    if (const Identifier *symbolId = symbol->identifier())
84 85 86 87 88 89 90 91 92 93
        id = doc->control()->findIdentifier(symbolId->chars(), symbolId->size());

    QList<int> references;

    if (! id)
        return references;

    TranslationUnit *translationUnit = doc->translationUnit();
    Q_ASSERT(translationUnit != 0);

Roberto Raggi's avatar
Roberto Raggi committed
94 95 96 97
    FindUsages findUsages(doc, snapshot, /*future = */ 0);
    findUsages.setGlobalNamespaceBinding(bind(doc, snapshot));
    findUsages(symbol, id, translationUnit->ast());
    references = findUsages.references();
98 99 100 101

    return references;
}

102
static void find_helper(QFutureInterface<Usage> &future,
103
                        const QMap<QString, QString> wl,
104 105
                        Snapshot snapshot,
                        Symbol *symbol)
Roberto Raggi's avatar
Roberto Raggi committed
106 107 108
{
    QTime tm;
    tm.start();
109

Roberto Raggi's avatar
Roberto Raggi committed
110
    const Identifier *symbolId = symbol->identifier();
111 112
    Q_ASSERT(symbolId != 0);

Roberto Raggi's avatar
Cleanup  
Roberto Raggi committed
113 114
    const QString sourceFile = QString::fromUtf8(symbol->fileName(), symbol->fileNameLength());
    QStringList files(sourceFile);
115 116 117 118 119 120 121 122 123 124 125 126 127 128

    if (symbol->isClass() || symbol->isForwardClassDeclaration()) {
        foreach (const Document::Ptr &doc, snapshot) {
            if (doc->fileName() == sourceFile)
                continue;

            Control *control = doc->control();

            if (control->findIdentifier(symbolId->chars(), symbolId->size()))
                files.append(doc->fileName());
        }
    } else {
        files += snapshot.dependsOn(sourceFile);
    }
129
    files.removeDuplicates();
130
    //qDebug() << "done in:" << tm.elapsed() << "number of files to parse:" << files.size();
Roberto Raggi's avatar
Roberto Raggi committed
131 132 133 134

    future.setProgressRange(0, files.size());

    for (int i = 0; i < files.size(); ++i) {
135 136 137 138 139 140
        if (future.isPaused())
            future.waitForResume();

        if (future.isCanceled())
            break;

Roberto Raggi's avatar
Cleanup  
Roberto Raggi committed
141 142
        const QString &fileName = files.at(i);
        future.setProgressValueAndText(i, QFileInfo(fileName).fileName());
143

Roberto Raggi's avatar
Roberto Raggi committed
144
        if (Document::Ptr previousDoc = snapshot.value(fileName)) {
145
            Control *control = previousDoc->control();
Roberto Raggi's avatar
Roberto Raggi committed
146
            const Identifier *id = control->findIdentifier(symbolId->chars(), symbolId->size());
147 148 149 150
            if (! id)
                continue; // skip this document, it's not using symbolId.
        }

151 152 153 154 155 156 157 158 159 160 161 162 163 164
        QByteArray source;

        if (wl.contains(fileName))
            source = snapshot.preprocessedCode(wl.value(fileName), fileName);
        else {
            QFile file(fileName);
            if (! file.open(QFile::ReadOnly))
                continue;

            const QString contents = QTextStream(&file).readAll(); // ### FIXME
            source = snapshot.preprocessedCode(contents, fileName);
        }

        Document::Ptr doc = snapshot.documentFromSource(source, fileName);
165
        doc->tokenize();
Roberto Raggi's avatar
Roberto Raggi committed
166 167

        Control *control = doc->control();
Roberto Raggi's avatar
Roberto Raggi committed
168
        if (const Identifier *id = control->findIdentifier(symbolId->chars(), symbolId->size())) {
Roberto Raggi's avatar
Roberto Raggi committed
169 170
            QTime tm;
            tm.start();
171 172
            doc->parse();

Roberto Raggi's avatar
Roberto Raggi committed
173 174 175
            //qDebug() << "***" << unit->fileName() << "parsed in:" << tm.elapsed();

            tm.start();
176
            doc->check();
Roberto Raggi's avatar
Roberto Raggi committed
177 178 179
            //qDebug() << "***" << unit->fileName() << "checked in:" << tm.elapsed();

            tm.start();
180

181
            FindUsages process(doc, snapshot, &future);
182
            process.setGlobalNamespaceBinding(bind(doc, snapshot));
Roberto Raggi's avatar
Roberto Raggi committed
183 184

            TranslationUnit *unit = doc->translationUnit();
185
            process(symbol, id, unit->ast());
Roberto Raggi's avatar
Roberto Raggi committed
186 187

            //qDebug() << "***" << unit->fileName() << "processed in:" << tm.elapsed();
188
        }
Roberto Raggi's avatar
Roberto Raggi committed
189
    }
190

Roberto Raggi's avatar
Roberto Raggi committed
191 192 193
    future.setProgressValue(files.size());
}

194 195
void CppFindReferences::findUsages(Symbol *symbol)
{
Roberto Raggi's avatar
Roberto Raggi committed
196 197 198 199 200
    Find::SearchResult *search = _resultWindow->startNewSearch(Find::SearchResultWindow::SearchOnly);

    connect(search, SIGNAL(activated(Find::SearchResultItem)),
            this, SLOT(openEditor(Find::SearchResultItem)));

201 202 203 204
    findAll_helper(symbol);
}

void CppFindReferences::renameUsages(Symbol *symbol)
Roberto Raggi's avatar
Roberto Raggi committed
205
{
Roberto Raggi's avatar
Roberto Raggi committed
206
    if (const Identifier *id = symbol->identifier()) {
207
        const QString textToReplace = QString::fromUtf8(id->chars(), id->size());
Roberto Raggi's avatar
Roberto Raggi committed
208

209 210
        Find::SearchResult *search = _resultWindow->startNewSearch(Find::SearchResultWindow::SearchAndReplace);
        _resultWindow->setTextToReplace(textToReplace);
211

212 213
        connect(search, SIGNAL(activated(Find::SearchResultItem)),
                this, SLOT(openEditor(Find::SearchResultItem)));
Roberto Raggi's avatar
Roberto Raggi committed
214

215 216 217 218 219
        connect(search, SIGNAL(replaceButtonClicked(QString,QList<Find::SearchResultItem>)),
                SLOT(onReplaceButtonClicked(QString,QList<Find::SearchResultItem>)));

        findAll_helper(symbol);
    }
220 221 222 223
}

void CppFindReferences::findAll_helper(Symbol *symbol)
{
224 225 226
    if (! (symbol && symbol->identifier()))
        return;

Roberto Raggi's avatar
Roberto Raggi committed
227 228
    _resultWindow->popup(true);

229
    const Snapshot snapshot = _modelManager->snapshot();
230
    const QMap<QString, QString> wl = _modelManager->workingCopy();
231

Roberto Raggi's avatar
Roberto Raggi committed
232
    Core::ProgressManager *progressManager = Core::ICore::instance()->progressManager();
Roberto Raggi's avatar
Roberto Raggi committed
233 234 235 236

    QFuture<Usage> result;

    result = QtConcurrent::run(&find_helper, wl, snapshot, symbol);
Roberto Raggi's avatar
Roberto Raggi committed
237 238 239
    m_watcher.setFuture(result);

    Core::FutureProgress *progress = progressManager->addTask(result, tr("Searching..."),
240
                                                              CppTools::Constants::TASK_SEARCH);
Roberto Raggi's avatar
Roberto Raggi committed
241 242 243 244

    connect(progress, SIGNAL(clicked()), _resultWindow, SLOT(popup()));
}

245 246 247 248 249 250 251
static void applyChanges(QTextDocument *doc, const QString &text, const QList<Find::SearchResultItem> &items)
{
    QList<QTextCursor> cursors;

    foreach (const Find::SearchResultItem &item, items) {
        const int blockNumber = item.lineNumber - 1;
        QTextCursor tc(doc->findBlockByNumber(blockNumber));
252 253 254 255 256 257 258 259 260 261 262 263 264 265 266

        const int cursorPosition = tc.position() + item.searchTermStart;

        int cursorIndex = 0;
        for (; cursorIndex < cursors.size(); ++cursorIndex) {
            const QTextCursor &tc = cursors.at(cursorIndex);

            if (tc.position() == cursorPosition)
                break;
        }

        if (cursorIndex != cursors.size())
            continue; // skip this change.

        tc.setPosition(cursorPosition);
267 268 269 270 271 272 273 274 275
        tc.setPosition(tc.position() + item.searchTermLength,
                       QTextCursor::KeepAnchor);
        cursors.append(tc);
    }

    foreach (QTextCursor tc, cursors)
        tc.insertText(text);
}

Roberto Raggi's avatar
Roberto Raggi committed
276 277 278
void CppFindReferences::onReplaceButtonClicked(const QString &text,
                                               const QList<Find::SearchResultItem> &items)
{
279 280
    Core::EditorManager::instance()->hideEditorInfoBar(QLatin1String("CppEditor.Rename"));

Roberto Raggi's avatar
Roberto Raggi committed
281 282 283 284 285 286 287 288
    if (text.isEmpty())
        return;

    QHash<QString, QList<Find::SearchResultItem> > changes;

    foreach (const Find::SearchResultItem &item, items)
        changes[item.fileName].append(item);

289 290
    Core::EditorManager *editorManager = Core::EditorManager::instance();

Roberto Raggi's avatar
Roberto Raggi committed
291 292 293 294 295
    QHashIterator<QString, QList<Find::SearchResultItem> > it(changes);
    while (it.hasNext()) {
        it.next();

        const QString fileName = it.key();
296
        const QList<Find::SearchResultItem> items = it.value();
Roberto Raggi's avatar
Roberto Raggi committed
297

298 299 300 301 302 303 304 305 306 307 308 309 310 311 312
        const QList<Core::IEditor *> editors = editorManager->editorsForFileName(fileName);
        TextEditor::BaseTextEditor *textEditor = 0;
        foreach (Core::IEditor *editor, editors) {
            textEditor = qobject_cast<TextEditor::BaseTextEditor *>(editor->widget());
            if (textEditor != 0)
                break;
        }

        if (textEditor != 0) {
            QTextCursor tc = textEditor->textCursor();
            tc.beginEditBlock();
            applyChanges(textEditor->document(), text, items);
            tc.endEditBlock();
        } else {
            QFile file(fileName);
Roberto Raggi's avatar
Roberto Raggi committed
313

314 315
            if (file.open(QFile::ReadOnly)) {
                QTextStream stream(&file);
Roberto Raggi's avatar
Roberto Raggi committed
316
                // ### set the encoding
317 318 319 320 321 322 323 324 325 326 327 328 329 330
                const QString plainText = stream.readAll();
                file.close();

                QTextDocument doc;
                doc.setPlainText(plainText);

                applyChanges(&doc, text, items);

                QFile newFile(fileName);
                if (newFile.open(QFile::WriteOnly)) {
                    QTextStream stream(&newFile);
                    // ### set the encoding
                    stream << doc.toPlainText();
                }
Roberto Raggi's avatar
Roberto Raggi committed
331 332 333 334 335 336
            }
        }
    }

    const QStringList fileNames = changes.keys();
    _modelManager->updateSourceFiles(fileNames);
337
    _resultWindow->hide();
Roberto Raggi's avatar
Roberto Raggi committed
338 339
}

Roberto Raggi's avatar
Roberto Raggi committed
340
void CppFindReferences::displayResults(int first, int last)
Roberto Raggi's avatar
Roberto Raggi committed
341
{
Roberto Raggi's avatar
Roberto Raggi committed
342 343 344 345 346 347 348 349
    for (int index = first; index != last; ++index) {
        Usage result = m_watcher.future().resultAt(index);
        _resultWindow->addResult(result.path,
                                 result.line,
                                 result.lineText,
                                 result.col,
                                 result.len);
    }
Roberto Raggi's avatar
Roberto Raggi committed
350 351 352 353
}

void CppFindReferences::searchFinished()
{
354
    _resultWindow->finishSearch();
Roberto Raggi's avatar
Roberto Raggi committed
355 356 357
    emit changed();
}

358
void CppFindReferences::openEditor(const Find::SearchResultItem &item)
Roberto Raggi's avatar
Roberto Raggi committed
359
{
360
    TextEditor::BaseTextEditor::openEditorAt(item.fileName, item.lineNumber, item.searchTermStart);
Roberto Raggi's avatar
Roberto Raggi committed
361 362
}