#include "llvm/CompilerDriver/Error.h"
#include "llvm/CompilerDriver/Plugin.h"
+#include "llvm/Support/raw_ostream.h"
#include "llvm/System/Path.h"
-#include <iostream>
#include <stdexcept>
#include <string>
sys::Path getTempDir() {
sys::Path tempDir;
+ // The --temp-dir option.
+ if (!TempDirname.empty()) {
+ tempDir = TempDirname;
+ }
// GCC 4.5-style -save-temps handling.
- if (SaveTemps == SaveTempsEnum::Unset) {
+ else if (SaveTemps == SaveTempsEnum::Unset) {
tempDir = sys::Path::GetTemporaryDirectory();
+ return tempDir;
}
else if (SaveTemps == SaveTempsEnum::Obj && !OutputFilename.empty()) {
tempDir = OutputFilename;
+ tempDir = tempDir.getDirname();
+ }
+ else {
+ // SaveTemps == Cwd --> use current dir (leave tempDir empty).
+ return tempDir;
+ }
- if (!tempDir.exists()) {
- std::string ErrMsg;
- if (tempDir.createDirectoryOnDisk(true, &ErrMsg))
- throw std::runtime_error(ErrMsg);
- }
+ if (!tempDir.exists()) {
+ std::string ErrMsg;
+ if (tempDir.createDirectoryOnDisk(true, &ErrMsg))
+ throw std::runtime_error(ErrMsg);
}
- // else if (SaveTemps == Cwd) -> use current dir (leave tempDir empty)
return tempDir;
}
int BuildTargets(CompilationGraph& graph, const LanguageMap& langMap) {
int ret;
const sys::Path& tempDir = getTempDir();
+ bool toDelete = (SaveTemps == SaveTempsEnum::Unset);
try {
ret = graph.Build(tempDir, langMap);
}
catch(...) {
- if (SaveTemps == SaveTempsEnum::Unset)
+ if (toDelete)
tempDir.eraseFromDisk(true);
throw;
}
- if (SaveTemps == SaveTempsEnum::Unset)
+ if (toDelete)
tempDir.eraseFromDisk(true);
return ret;
}
(argc, argv, "LLVM Compiler Driver (Work In Progress)", true);
PluginLoader Plugins;
- Plugins.PopulateLanguageMap(langMap);
- Plugins.PopulateCompilationGraph(graph);
+ Plugins.RunInitialization(langMap, graph);
if (CheckGraph) {
int ret = graph.Check();
if (!ret)
- std::cerr << "check-graph: no errors found.\n";
+ llvm::errs() << "check-graph: no errors found.\n";
return ret;
}
return ec.code();
}
catch(const std::exception& ex) {
- std::cerr << argv[0] << ": " << ex.what() << '\n';
+ llvm::errs() << argv[0] << ": " << ex.what() << '\n';
}
catch(...) {
- std::cerr << argv[0] << ": unknown error!\n";
+ llvm::errs() << argv[0] << ": unknown error!\n";
}
return 1;
}