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
65
66
67
68
69
70
71
72
73
74
75
76
|
// This file is part of the AspectC++ compiler 'ac++'.
// Copyright (C) 1999-2003 The 'ac++' developers (see aspectc.org)
//
// This program 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 2 of
// the License, or (at your option) any later version.
//
// This program 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 for more details.
//
// You should have received a copy of the GNU General Public
// License along with this program; if not, write to the Free
// Software Foundation, Inc., 59 Temple Place, Suite 330, Boston,
// MA 02111-1307 USA
#include "ClangIncludeExpander.h"
#include "LineDirectiveMgr.h"
#include "version.h"
#include "clang/Basic/Version.h"
#include "clang/Frontend/CompilerInstance.h"
#include "ClangInclusionRewriter.h"
#include "llvm/Support/raw_os_ostream.h"
void ClangIncludeExpander::expand(std::ostream &out) {
// Setup the strategy object that decides whether an include should be
// expanded.
struct ProjectStrategy : public clang::FileExpansionStrategy {
ClangIncludeExpander &_ie;
ProjectStrategy(ClangIncludeExpander &ie) : _ie(ie) {}
virtual bool should_expand_file(clang::FileID From, clang::FileID To) {
clang::SourceManager &sm =
_ie._project.get_compiler_instance()->getSourceManager();
// Check if From and To are coming from this project.
#if CLANG_VERSION_NUMBER >= VERSION_NUMBER_4_0_0
const char *from_name = sm.getFileEntryForID(From)->getName().data();
const char *to_name = sm.getFileEntryForID(To)->getName().data();
#else
const char *from_name = sm.getFileEntryForID(From)->getName();
const char *to_name = sm.getFileEntryForID(To)->getName();
#endif
bool from_below = _ie._project.isBelow(from_name);
bool to_below = _ie._project.isBelow(to_name);
// cout << "include '" << to_name
// << "' from '" << from_name << "'" << endl;
if (!from_below && to_below) {
// oops, an external file includes an internal => error!
_ie._err << Puma::sev_error //<< this_token.location ()
<< "internal file '" << to_name
<< "' is included by external '" << from_name << "'"
<< Puma::endMessage;
}
return to_below;
}
};
clang::CompilerInstance *ci = _project.get_compiler_instance();
if (!ci)
return; // No compiler, nothing to do!
// Create a new PP to get rid of old state.
ci->createPreprocessor(clang::TU_Complete);
ci->getDiagnosticClient().BeginSourceFile(ci->getLangOpts(), &ci->getPreprocessor());
// Now call Clang's include rewriter to do the heavy lifting.
llvm::raw_os_ostream osos(out);
ProjectStrategy s(*this);
clang::PreprocessorOutputOptions opts = ci->getPreprocessorOutputOpts();
opts.ShowLineMarkers = (_line_mgr.noline () ? 0 : 1);
clang::RewriteIncludesInInput(ci->getPreprocessor(), &osos, opts, &s);
}
|
