cppfindreferences.cpp 24.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
26
27
28
29
30
/**************************************************************************
**
** 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
** contact the sales department at http://www.qtsoftware.com/contact.
**
**************************************************************************/

#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>
Roberto Raggi's avatar
Roberto Raggi committed
53
54
#include <cplusplus/ExpressionUnderCursor.h>
#include <cplusplus/ResolveExpression.h>
55
#include <cplusplus/Overview.h>
Roberto Raggi's avatar
Roberto Raggi committed
56
#include <cplusplus/TypeOfExpression.h>
57
#include <cplusplus/FastPreprocessor.h>
Roberto Raggi's avatar
Roberto Raggi committed
58
59
60
61

#include <QtCore/QTime>
#include <QtCore/QtConcurrentRun>
#include <QtCore/QDir>
62
#include <QtGui/QApplication>
Roberto Raggi's avatar
Roberto Raggi committed
63
64
65
66
67
68
69
70
71
72
#include <qtconcurrent/runextensions.h>

using namespace CppTools::Internal;
using namespace CPlusPlus;

namespace {

struct Process: protected ASTVisitor
{
public:
73
    Process(Document::Ptr doc, const Snapshot &snapshot,
74
            QFutureInterface<Utils::FileSearchResult> *future)
Roberto Raggi's avatar
Roberto Raggi committed
75
76
77
78
            : ASTVisitor(doc->control()),
              _future(future),
              _doc(doc),
              _snapshot(snapshot),
79
              _source(_doc->source()),
80
81
              _sem(doc->control()),
              _inSimpleDeclaration(0)
82
83
84
    {
        _snapshot.insert(_doc);
    }
Roberto Raggi's avatar
Roberto Raggi committed
85

86
87
88
89
90
    void setGlobalNamespaceBinding(NamespaceBindingPtr globalNamespaceBinding)
    {
        _globalNamespaceBinding = globalNamespaceBinding;
    }

91
    QList<int> operator()(Symbol *symbol, Identifier *id, AST *ast)
Roberto Raggi's avatar
Roberto Raggi committed
92
    {
93
        _references.clear();
94
        _declSymbol = symbol;
Roberto Raggi's avatar
Roberto Raggi committed
95
96
97
        _id = id;
        _exprDoc = Document::create("<references>");
        accept(ast);
98
        return _references;
Roberto Raggi's avatar
Roberto Raggi committed
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
    }

protected:
    using ASTVisitor::visit;

    QString matchingLine(const Token &tk) const
    {
        const char *beg = _source.constData();
        const char *cp = beg + tk.offset;
        for (; cp != beg - 1; --cp) {
            if (*cp == '\n')
                break;
        }

        ++cp;

        const char *lineEnd = cp + 1;
        for (; *lineEnd; ++lineEnd) {
            if (*lineEnd == '\n')
                break;
        }

        const QString matchingLine = QString::fromUtf8(cp, lineEnd - cp);
        return matchingLine;

    }

126
127
128
129
130
131
132
133
    void reportResult(unsigned tokenIndex, const QList<Symbol *> &candidates)
    {
        const bool isStrongResult = checkCandidates(candidates);

        if (isStrongResult)
            reportResult(tokenIndex);
    }

Roberto Raggi's avatar
Roberto Raggi committed
134
135
136
137
138
139
140
141
142
143
144
    void reportResult(unsigned tokenIndex)
    {
        const Token &tk = tokenAt(tokenIndex);
        const QString lineText = matchingLine(tk);

        unsigned line, col;
        getTokenStartPosition(tokenIndex, &line, &col);

        if (col)
            --col;  // adjust the column position.

145
        const int len = tk.f.length;
Roberto Raggi's avatar
Roberto Raggi committed
146

147
        if (_future)
148
            _future->reportResult(Utils::FileSearchResult(QDir::toNativeSeparators(_doc->fileName()),
149
                                                          line, lineText, col, len));
150
151

        _references.append(tokenIndex);
Roberto Raggi's avatar
Roberto Raggi committed
152
153
    }

154
    bool checkCandidates(const QList<Symbol *> &candidates) const
Roberto Raggi's avatar
Roberto Raggi committed
155
    {
156
157
        if (Symbol *canonicalSymbol = LookupContext::canonicalSymbol(candidates, _globalNamespaceBinding.data())) {

Roberto Raggi's avatar
Roberto Raggi committed
158
159
160
161
162
163
164
165
166
167
#if 0
            qDebug() << "*** canonical symbol:" << canonicalSymbol->fileName()
                    << canonicalSymbol->line() << canonicalSymbol->column()
                    << "candidates:" << candidates.size();
#endif

            return isDeclSymbol(canonicalSymbol);
        }

        return false;
Roberto Raggi's avatar
Roberto Raggi committed
168
169
    }

170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
    bool checkScope(Symbol *symbol, Symbol *otherSymbol) const
    {
        if (! (symbol && otherSymbol))
            return false;

        else if (symbol->scope() == otherSymbol->scope())
            return true;

        else if (symbol->name() && otherSymbol->name()) {

            if (! symbol->name()->isEqualTo(otherSymbol->name()))
                return false;

        } else if (symbol->name() != otherSymbol->name()) {
            return false;
        }

        return checkScope(symbol->enclosingSymbol(), otherSymbol->enclosingSymbol());
    }

190
    bool isDeclSymbol(Symbol *symbol) const
Roberto Raggi's avatar
Roberto Raggi committed
191
    {
Roberto Raggi's avatar
Roberto Raggi committed
192
        if (! symbol) {
193
            return false;
Roberto Raggi's avatar
Roberto Raggi committed
194

Roberto Raggi's avatar
Roberto Raggi committed
195
        } else if (symbol == _declSymbol) {
196
197
            return true;

198
        } else if (symbol->line() == _declSymbol->line() && symbol->column() == _declSymbol->column()) {
199
200
            if (! qstrcmp(symbol->fileName(), _declSymbol->fileName()))
                return true;
201
202
203

        } else if (symbol->isForwardClassDeclaration() && (_declSymbol->isClass() ||
                                                           _declSymbol->isForwardClassDeclaration())) {
204
            return checkScope(symbol, _declSymbol);
205
206
207

        } else if (_declSymbol->isForwardClassDeclaration() && (symbol->isClass() ||
                                                                symbol->isForwardClassDeclaration())) {
208
            return checkScope(symbol, _declSymbol);
209
210
211
        }

        return false;
Roberto Raggi's avatar
Roberto Raggi committed
212
213
    }

Roberto Raggi's avatar
Roberto Raggi committed
214
215
216
    LookupContext _previousContext;

    LookupContext currentContext(AST *ast)
Roberto Raggi's avatar
Roberto Raggi committed
217
    {
218
219
220
        unsigned line, column;
        getTokenStartPosition(ast->firstToken(), &line, &column);
        Symbol *lastVisibleSymbol = _doc->findSymbolAt(line, column);
Roberto Raggi's avatar
Roberto Raggi committed
221
222
223
224

        if (lastVisibleSymbol && lastVisibleSymbol == _previousContext.symbol())
            return _previousContext;

Roberto Raggi's avatar
Roberto Raggi committed
225
        LookupContext ctx(lastVisibleSymbol, _exprDoc, _doc, _snapshot);
Roberto Raggi's avatar
Roberto Raggi committed
226
        _previousContext = ctx;
Roberto Raggi's avatar
Roberto Raggi committed
227
        return ctx;
Roberto Raggi's avatar
Roberto Raggi committed
228
229
    }

Roberto Raggi's avatar
Roberto Raggi committed
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
    void ensureNameIsValid(NameAST *ast)
    {
        if (ast && ! ast->name)
            ast->name = _sem.check(ast, /*scope = */ 0);
    }

    virtual bool visit(MemInitializerAST *ast)
    {
        if (ast->name && ast->name->asSimpleName() != 0) {
            ensureNameIsValid(ast->name);

            SimpleNameAST *simple = ast->name->asSimpleName();
            if (identifier(simple->identifier_token) == _id) {
                LookupContext context = currentContext(ast);
                const QList<Symbol *> candidates = context.resolve(simple->name);
245
                reportResult(simple->identifier_token, candidates);
Roberto Raggi's avatar
Roberto Raggi committed
246
247
248
249
250
251
            }
        }
        accept(ast->expression);
        return false;
    }

Roberto Raggi's avatar
Roberto Raggi committed
252
253
254
255
256
257
    virtual bool visit(PostfixExpressionAST *ast)
    {
        _postfixExpressionStack.append(ast);
        return true;
    }

Roberto Raggi's avatar
Roberto Raggi committed
258
    virtual void endVisit(PostfixExpressionAST *)
Roberto Raggi's avatar
Roberto Raggi committed
259
260
261
262
263
264
    {
        _postfixExpressionStack.removeLast();
    }

    virtual bool visit(MemberAccessAST *ast)
    {
Roberto Raggi's avatar
Roberto Raggi committed
265
266
267
268
        if (ast->member_name) {
            if (SimpleNameAST *simple = ast->member_name->asSimpleName()) {
                if (identifier(simple->identifier_token) == _id) {
                    Q_ASSERT(! _postfixExpressionStack.isEmpty());
Roberto Raggi's avatar
Roberto Raggi committed
269

Roberto Raggi's avatar
Roberto Raggi committed
270
271
                    checkExpression(_postfixExpressionStack.last()->firstToken(),
                                    simple->identifier_token);
Roberto Raggi's avatar
Roberto Raggi committed
272

Roberto Raggi's avatar
Roberto Raggi committed
273
274
275
276
                    return false;
                }
            }
        }
Roberto Raggi's avatar
Roberto Raggi committed
277

Roberto Raggi's avatar
Roberto Raggi committed
278
279
        return true;
    }
Roberto Raggi's avatar
Roberto Raggi committed
280

Roberto Raggi's avatar
Roberto Raggi committed
281
282
283
284
    void checkExpression(unsigned startToken, unsigned endToken)
    {
        const unsigned begin = tokenAt(startToken).begin();
        const unsigned end = tokenAt(endToken).end();
Roberto Raggi's avatar
Roberto Raggi committed
285

Roberto Raggi's avatar
Roberto Raggi committed
286
        const QString expression = _source.mid(begin, end - begin);
Roberto Raggi's avatar
Roberto Raggi committed
287
        // qDebug() << "*** check expression:" << expression;
Roberto Raggi's avatar
Roberto Raggi committed
288

Roberto Raggi's avatar
Roberto Raggi committed
289
290
        TypeOfExpression typeofExpression;
        typeofExpression.setSnapshot(_snapshot);
Roberto Raggi's avatar
Roberto Raggi committed
291

Roberto Raggi's avatar
Roberto Raggi committed
292
293
294
        unsigned line, column;
        getTokenStartPosition(startToken, &line, &column);
        Symbol *lastVisibleSymbol = _doc->findSymbolAt(line, column);
Roberto Raggi's avatar
Roberto Raggi committed
295

Roberto Raggi's avatar
Roberto Raggi committed
296
297
        const QList<TypeOfExpression::Result> results =
                typeofExpression(expression, _doc, lastVisibleSymbol,
298
                                 TypeOfExpression::Preprocess);
Roberto Raggi's avatar
Roberto Raggi committed
299

Roberto Raggi's avatar
Roberto Raggi committed
300
        QList<Symbol *> candidates;
Roberto Raggi's avatar
Roberto Raggi committed
301

Roberto Raggi's avatar
Roberto Raggi committed
302
303
304
        foreach (TypeOfExpression::Result r, results) {
            FullySpecifiedType ty = r.first;
            Symbol *lastVisibleSymbol = r.second;
Roberto Raggi's avatar
Roberto Raggi committed
305

Roberto Raggi's avatar
Roberto Raggi committed
306
            candidates.append(lastVisibleSymbol);
Roberto Raggi's avatar
Roberto Raggi committed
307
308
        }

309
        reportResult(endToken, candidates);
Roberto Raggi's avatar
Roberto Raggi committed
310
311
    }

Roberto Raggi's avatar
Roberto Raggi committed
312
313
    virtual bool visit(QualifiedNameAST *ast)
    {
Roberto Raggi's avatar
Roberto Raggi committed
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
        for (NestedNameSpecifierAST *nested_name_specifier = ast->nested_name_specifier;
             nested_name_specifier; nested_name_specifier = nested_name_specifier->next) {

            if (NameAST *class_or_namespace_name = nested_name_specifier->class_or_namespace_name) {
                SimpleNameAST *simple_name = class_or_namespace_name->asSimpleName();

                TemplateIdAST *template_id = 0;
                if (! simple_name) {
                    template_id = class_or_namespace_name->asTemplateId();

                    if (template_id) {
                        for (TemplateArgumentListAST *template_arguments = template_id->template_arguments;
                             template_arguments; template_arguments = template_arguments->next) {
                            accept(template_arguments->template_argument);
                        }
                    }
                }

                if (simple_name || template_id) {
                    const unsigned identifier_token = simple_name
                               ? simple_name->identifier_token
                               : template_id->identifier_token;

                    if (identifier(identifier_token) == _id)
                        checkExpression(ast->firstToken(), identifier_token);
                }
            }
        }

343
344
345
346
347
348
349
350
        if (NameAST *unqualified_name = ast->unqualified_name) {
            unsigned identifier_token = 0;

            if (SimpleNameAST *simple_name = unqualified_name->asSimpleName())
                identifier_token = simple_name->identifier_token;

            else if (DestructorNameAST *dtor_name = unqualified_name->asDestructorName())
                identifier_token = dtor_name->identifier_token;
Roberto Raggi's avatar
Roberto Raggi committed
351
352

            TemplateIdAST *template_id = 0;
353
354
            if (! identifier_token) {
                template_id = unqualified_name->asTemplateId();
Roberto Raggi's avatar
Roberto Raggi committed
355
356

                if (template_id) {
357
358
                    identifier_token = template_id->identifier_token;

Roberto Raggi's avatar
Roberto Raggi committed
359
360
361
362
363
364
365
                    for (TemplateArgumentListAST *template_arguments = template_id->template_arguments;
                         template_arguments; template_arguments = template_arguments->next) {
                        accept(template_arguments->template_argument);
                    }
                }
            }

366
367
            if (identifier_token && identifier(identifier_token) == _id)
                checkExpression(ast->firstToken(), identifier_token);
368
369
        }

Roberto Raggi's avatar
Roberto Raggi committed
370
371
372
        return false;
    }

Roberto Raggi's avatar
Roberto Raggi committed
373
374
375
376
377
378
379
380
381
382
383
384
385
386
    virtual bool visit(EnumeratorAST *ast)
    {
        Identifier *id = identifier(ast->identifier_token);
        if (id == _id) {
            LookupContext context = currentContext(ast);
            const QList<Symbol *> candidates = context.resolve(control()->nameId(id));
            reportResult(ast->identifier_token, candidates);
        }

        accept(ast->expression);

        return false;
    }

Roberto Raggi's avatar
Roberto Raggi committed
387
388
389
390
    virtual bool visit(SimpleNameAST *ast)
    {
        Identifier *id = identifier(ast->identifier_token);
        if (id == _id) {
391
392
            LookupContext context = currentContext(ast);
            const QList<Symbol *> candidates = context.resolve(ast->name);
393
            reportResult(ast->identifier_token, candidates);
394
395
396
        }

        return false;
Roberto Raggi's avatar
Roberto Raggi committed
397
398
    }

Roberto Raggi's avatar
Roberto Raggi committed
399
    virtual bool visit(DestructorNameAST *ast)
Roberto Raggi's avatar
Roberto Raggi committed
400
401
402
    {
        Identifier *id = identifier(ast->identifier_token);
        if (id == _id) {
403
404
            LookupContext context = currentContext(ast);
            const QList<Symbol *> candidates = context.resolve(ast->name);
405
            reportResult(ast->identifier_token, candidates);
Roberto Raggi's avatar
Roberto Raggi committed
406
407
408
409
410
411
412
        }

        return false;
    }

    virtual bool visit(TemplateIdAST *ast)
    {
413
        if (_id == identifier(ast->identifier_token)) {
414
415
            LookupContext context = currentContext(ast);
            const QList<Symbol *> candidates = context.resolve(ast->name);
416
            reportResult(ast->identifier_token, candidates);
417
        }
Roberto Raggi's avatar
Roberto Raggi committed
418

419
420
421
        for (TemplateArgumentListAST *template_arguments = ast->template_arguments;
             template_arguments; template_arguments = template_arguments->next) {
            accept(template_arguments->template_argument);
422
        }
Roberto Raggi's avatar
Roberto Raggi committed
423

424
        return false;
Roberto Raggi's avatar
Roberto Raggi committed
425
426
    }

427
428
429
430
431
432
433
434
435
436
437
438
    virtual bool visit(ParameterDeclarationAST *ast)
    {
        for (SpecifierAST *spec = ast->type_specifier; spec; spec = spec->next)
            accept(spec);

        if (DeclaratorAST *declarator = ast->declarator) {
            for (SpecifierAST *attr = declarator->attributes; attr; attr = attr->next)
                accept(attr);

            for (PtrOperatorAST *ptr_op = declarator->ptr_operators; ptr_op; ptr_op = ptr_op->next)
                accept(ptr_op);

439
440
            if (! _inSimpleDeclaration) // visit the core declarator only if we are not in simple-declaration.
                accept(declarator->core_declarator);
441
442
443
444
445
446
447
448
449
450
451
452
453
454

            for (PostfixDeclaratorAST *fx_op = declarator->postfix_declarators; fx_op; fx_op = fx_op->next)
                accept(fx_op);

            for (SpecifierAST *spec = declarator->post_attributes; spec; spec = spec->next)
                accept(spec);

            accept(declarator->initializer);
        }

        accept(ast->expression);
        return false;
    }

Roberto Raggi's avatar
Roberto Raggi committed
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
    virtual bool visit(ExpressionOrDeclarationStatementAST *ast)
    {
        accept(ast->declaration);
        return false;
    }

    virtual bool visit(FunctionDeclaratorAST *ast)
    {
        accept(ast->parameters);

        for (SpecifierAST *spec = ast->cv_qualifier_seq; spec; spec = spec->next)
            accept(spec);

        accept(ast->exception_specification);

        return false;
    }

473
474
475
476
477
478
479
480
481
    virtual bool visit(SimpleDeclarationAST *)
    {
        ++_inSimpleDeclaration;
        return true;
    }

    virtual void endVisit(SimpleDeclarationAST *)
    { --_inSimpleDeclaration; }

Roberto Raggi's avatar
Roberto Raggi committed
482
private:
483
    QFutureInterface<Utils::FileSearchResult> *_future;
484
485
    Identifier *_id; // ### remove me
    Symbol *_declSymbol;
Roberto Raggi's avatar
Roberto Raggi committed
486
487
488
489
    Document::Ptr _doc;
    Snapshot _snapshot;
    QByteArray _source;
    Document::Ptr _exprDoc;
490
    Semantic _sem;
491
    NamespaceBindingPtr _globalNamespaceBinding;
Roberto Raggi's avatar
Roberto Raggi committed
492
    QList<PostfixExpressionAST *> _postfixExpressionStack;
Roberto Raggi's avatar
Roberto Raggi committed
493
    QList<QualifiedNameAST *> _qualifiedNameStack;
494
    QList<int> _references;
495
    int _inSimpleDeclaration;
Roberto Raggi's avatar
Roberto Raggi committed
496
497
498
499
};

} // end of anonymous namespace

500
CppFindReferences::CppFindReferences(CppTools::CppModelManagerInterface *modelManager)
Roberto Raggi's avatar
Roberto Raggi committed
501
502
503
504
505
506
507
508
509
510
511
512
    : _modelManager(modelManager),
      _resultWindow(ExtensionSystem::PluginManager::instance()->getObject<Find::SearchResultWindow>())
{
    m_watcher.setPendingResultsLimit(1);
    connect(&m_watcher, SIGNAL(resultReadyAt(int)), this, SLOT(displayResult(int)));
    connect(&m_watcher, SIGNAL(finished()), this, SLOT(searchFinished()));
}

CppFindReferences::~CppFindReferences()
{
}

513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
QList<int> CppFindReferences::references(Symbol *symbol,
                                         Document::Ptr doc,
                                         const Snapshot& snapshot) const
{
    Identifier *id = 0;
    if (Identifier *symbolId = symbol->identifier())
        id = doc->control()->findIdentifier(symbolId->chars(), symbolId->size());

    QList<int> references;

    if (! id)
        return references;

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

    Process process(doc, snapshot, /*future = */ 0);
530
    process.setGlobalNamespaceBinding(bind(doc, snapshot));
531
532
533
534
535
    references = process(symbol, id, translationUnit->ast());

    return references;
}

536
static void find_helper(QFutureInterface<Utils::FileSearchResult> &future,
537
                        const QMap<QString, QString> wl,
538
539
                        Snapshot snapshot,
                        Symbol *symbol)
Roberto Raggi's avatar
Roberto Raggi committed
540
541
542
{
    QTime tm;
    tm.start();
543

544
545
546
    Identifier *symbolId = symbol->identifier();
    Q_ASSERT(symbolId != 0);

Roberto Raggi's avatar
Cleanup    
Roberto Raggi committed
547
    const QString sourceFile = QString::fromUtf8(symbol->fileName(), symbol->fileNameLength());
548

Roberto Raggi's avatar
Cleanup    
Roberto Raggi committed
549
    QStringList files(sourceFile);
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564

    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);
    }

565
    //qDebug() << "done in:" << tm.elapsed() << "number of files to parse:" << files.size();
Roberto Raggi's avatar
Roberto Raggi committed
566
567
568
569

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

    for (int i = 0; i < files.size(); ++i) {
570
571
572
573
574
575
        if (future.isPaused())
            future.waitForResume();

        if (future.isCanceled())
            break;

Roberto Raggi's avatar
Cleanup    
Roberto Raggi committed
576
577
        const QString &fileName = files.at(i);
        future.setProgressValueAndText(i, QFileInfo(fileName).fileName());
578

Roberto Raggi's avatar
Roberto Raggi committed
579
        if (Document::Ptr previousDoc = snapshot.value(fileName)) {
580
581
582
583
584
585
            Control *control = previousDoc->control();
            Identifier *id = control->findIdentifier(symbolId->chars(), symbolId->size());
            if (! id)
                continue; // skip this document, it's not using symbolId.
        }

586
587
588
589
590
591
592
593
594
595
596
597
598
599
        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);
600
        doc->tokenize();
Roberto Raggi's avatar
Roberto Raggi committed
601
602

        Control *control = doc->control();
603
        if (Identifier *id = control->findIdentifier(symbolId->chars(), symbolId->size())) {
Roberto Raggi's avatar
Roberto Raggi committed
604
605
            QTime tm;
            tm.start();
606
607
            doc->parse();

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

            tm.start();
611
            doc->check();
Roberto Raggi's avatar
Roberto Raggi committed
612
613
614
            //qDebug() << "***" << unit->fileName() << "checked in:" << tm.elapsed();

            tm.start();
615

616
            Process process(doc, snapshot, &future);
617
            process.setGlobalNamespaceBinding(bind(doc, snapshot));
Roberto Raggi's avatar
Roberto Raggi committed
618
619

            TranslationUnit *unit = doc->translationUnit();
620
            process(symbol, id, unit->ast());
Roberto Raggi's avatar
Roberto Raggi committed
621
622

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

Roberto Raggi's avatar
Roberto Raggi committed
626
627
628
    future.setProgressValue(files.size());
}

629
630
void CppFindReferences::findUsages(Symbol *symbol)
{
Roberto Raggi's avatar
Roberto Raggi committed
631
632
633
634
635
    Find::SearchResult *search = _resultWindow->startNewSearch(Find::SearchResultWindow::SearchOnly);

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

636
637
638
639
    findAll_helper(symbol);
}

void CppFindReferences::renameUsages(Symbol *symbol)
Roberto Raggi's avatar
Roberto Raggi committed
640
{
641
642
    if (Identifier *id = symbol->identifier()) {
        const QString textToReplace = QString::fromUtf8(id->chars(), id->size());
Roberto Raggi's avatar
Roberto Raggi committed
643

644
645
        Find::SearchResult *search = _resultWindow->startNewSearch(Find::SearchResultWindow::SearchAndReplace);
        _resultWindow->setTextToReplace(textToReplace);
646

647
648
        connect(search, SIGNAL(activated(Find::SearchResultItem)),
                this, SLOT(openEditor(Find::SearchResultItem)));
Roberto Raggi's avatar
Roberto Raggi committed
649

650
651
652
653
654
        connect(search, SIGNAL(replaceButtonClicked(QString,QList<Find::SearchResultItem>)),
                SLOT(onReplaceButtonClicked(QString,QList<Find::SearchResultItem>)));

        findAll_helper(symbol);
    }
655
656
657
658
}

void CppFindReferences::findAll_helper(Symbol *symbol)
{
Roberto Raggi's avatar
Roberto Raggi committed
659
660
    _resultWindow->popup(true);

661
    const Snapshot snapshot = _modelManager->snapshot();
662
    const QMap<QString, QString> wl = _modelManager->workingCopy();
663

Roberto Raggi's avatar
Roberto Raggi committed
664
665
    Core::ProgressManager *progressManager = Core::ICore::instance()->progressManager();

666
    QFuture<Utils::FileSearchResult> result = QtConcurrent::run(&find_helper, wl, snapshot, symbol);
Roberto Raggi's avatar
Roberto Raggi committed
667
668
669
    m_watcher.setFuture(result);

    Core::FutureProgress *progress = progressManager->addTask(result, tr("Searching..."),
670
                                                              CppTools::Constants::TASK_SEARCH,
Roberto Raggi's avatar
Roberto Raggi committed
671
672
673
674
675
                                                              Core::ProgressManager::CloseOnSuccess);

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

676
677
678
679
680
681
682
683
684
685
686
687
688
689
690
691
692
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));
        tc.setPosition(tc.position() + item.searchTermStart);
        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
693
694
695
void CppFindReferences::onReplaceButtonClicked(const QString &text,
                                               const QList<Find::SearchResultItem> &items)
{
696
697
    Core::EditorManager::instance()->hideEditorInfoBar(QLatin1String("CppEditor.Rename"));

Roberto Raggi's avatar
Roberto Raggi committed
698
699
700
701
702
703
704
705
    if (text.isEmpty())
        return;

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

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

706
707
    Core::EditorManager *editorManager = Core::EditorManager::instance();

Roberto Raggi's avatar
Roberto Raggi committed
708
709
710
711
712
    QHashIterator<QString, QList<Find::SearchResultItem> > it(changes);
    while (it.hasNext()) {
        it.next();

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

715
716
717
718
719
720
721
722
723
724
725
726
727
728
729
        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
730

731
732
            if (file.open(QFile::ReadOnly)) {
                QTextStream stream(&file);
Roberto Raggi's avatar
Roberto Raggi committed
733
                // ### set the encoding
734
735
736
737
738
739
740
741
742
743
744
745
746
747
                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
748
749
750
751
752
753
            }
        }
    }

    const QStringList fileNames = changes.keys();
    _modelManager->updateSourceFiles(fileNames);
754
    _resultWindow->hide();
Roberto Raggi's avatar
Roberto Raggi committed
755
756
}

Roberto Raggi's avatar
Roberto Raggi committed
757
758
void CppFindReferences::displayResult(int index)
{
759
    Utils::FileSearchResult result = m_watcher.future().resultAt(index);
760
761
762
763
764
    _resultWindow->addResult(result.fileName,
                             result.lineNumber,
                             result.matchingLine,
                             result.matchStart,
                             result.matchLength);
Roberto Raggi's avatar
Roberto Raggi committed
765
766
767
768
769
770
771
}

void CppFindReferences::searchFinished()
{
    emit changed();
}

772
void CppFindReferences::openEditor(const Find::SearchResultItem &item)
Roberto Raggi's avatar
Roberto Raggi committed
773
{
774
    TextEditor::BaseTextEditor::openEditorAt(item.fileName, item.lineNumber, item.searchTermStart);
Roberto Raggi's avatar
Roberto Raggi committed
775
776
}