aboutsummaryrefslogtreecommitdiffstats
path: root/contrib/libs/llvm12/lib/Support/GraphWriter.cpp
diff options
context:
space:
mode:
authororivej <orivej@yandex-team.ru>2022-02-10 16:45:01 +0300
committerDaniil Cherednik <dcherednik@yandex-team.ru>2022-02-10 16:45:01 +0300
commit2d37894b1b037cf24231090eda8589bbb44fb6fc (patch)
treebe835aa92c6248212e705f25388ebafcf84bc7a1 /contrib/libs/llvm12/lib/Support/GraphWriter.cpp
parent718c552901d703c502ccbefdfc3c9028d608b947 (diff)
downloadydb-2d37894b1b037cf24231090eda8589bbb44fb6fc.tar.gz
Restoring authorship annotation for <orivej@yandex-team.ru>. Commit 2 of 2.
Diffstat (limited to 'contrib/libs/llvm12/lib/Support/GraphWriter.cpp')
-rw-r--r--contrib/libs/llvm12/lib/Support/GraphWriter.cpp646
1 files changed, 323 insertions, 323 deletions
diff --git a/contrib/libs/llvm12/lib/Support/GraphWriter.cpp b/contrib/libs/llvm12/lib/Support/GraphWriter.cpp
index 9073b6a906..d8aae92603 100644
--- a/contrib/libs/llvm12/lib/Support/GraphWriter.cpp
+++ b/contrib/libs/llvm12/lib/Support/GraphWriter.cpp
@@ -1,323 +1,323 @@
-//===- GraphWriter.cpp - Implements GraphWriter support routines ----------===//
-//
-// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
-// See https://llvm.org/LICENSE.txt for license information.
-// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
-//
-//===----------------------------------------------------------------------===//
-//
-// This file implements misc. GraphWriter support routines.
-//
-//===----------------------------------------------------------------------===//
-
-#include "llvm/Support/GraphWriter.h"
-#include "llvm/ADT/SmallString.h"
-#include "llvm/ADT/SmallVector.h"
-#include "llvm/ADT/StringRef.h"
-#include "llvm/Config/config.h"
-#include "llvm/Support/CommandLine.h"
-#include "llvm/Support/Compiler.h"
-#include "llvm/Support/ErrorHandling.h"
-#include "llvm/Support/ErrorOr.h"
-#include "llvm/Support/FileSystem.h"
-#include "llvm/Support/Program.h"
-#include "llvm/Support/raw_ostream.h"
-#include <cassert>
-#include <system_error>
-#include <string>
-#include <vector>
-
-using namespace llvm;
-
-static cl::opt<bool> ViewBackground("view-background", cl::Hidden,
- cl::desc("Execute graph viewer in the background. Creates tmp file litter."));
-
-std::string llvm::DOT::EscapeString(const std::string &Label) {
- std::string Str(Label);
- for (unsigned i = 0; i != Str.length(); ++i)
- switch (Str[i]) {
- case '\n':
- Str.insert(Str.begin()+i, '\\'); // Escape character...
- ++i;
- Str[i] = 'n';
- break;
- case '\t':
- Str.insert(Str.begin()+i, ' '); // Convert to two spaces
- ++i;
- Str[i] = ' ';
- break;
- case '\\':
- if (i+1 != Str.length())
- switch (Str[i+1]) {
- case 'l': continue; // don't disturb \l
- case '|': case '{': case '}':
- Str.erase(Str.begin()+i); continue;
- default: break;
- }
- LLVM_FALLTHROUGH;
- case '{': case '}':
- case '<': case '>':
- case '|': case '"':
- Str.insert(Str.begin()+i, '\\'); // Escape character...
- ++i; // don't infinite loop
- break;
- }
- return Str;
-}
-
-/// Get a color string for this node number. Simply round-robin selects
-/// from a reasonable number of colors.
-StringRef llvm::DOT::getColorString(unsigned ColorNumber) {
- static const int NumColors = 20;
- static const char* Colors[NumColors] = {
- "aaaaaa", "aa0000", "00aa00", "aa5500", "0055ff", "aa00aa", "00aaaa",
- "555555", "ff5555", "55ff55", "ffff55", "5555ff", "ff55ff", "55ffff",
- "ffaaaa", "aaffaa", "ffffaa", "aaaaff", "ffaaff", "aaffff"};
- return Colors[ColorNumber % NumColors];
-}
-
-static std::string replaceIllegalFilenameChars(std::string Filename,
- const char ReplacementChar) {
-#ifdef _WIN32
- std::string IllegalChars = "\\/:?\"<>|";
-#else
- std::string IllegalChars = "/";
-#endif
-
- for (char IllegalChar : IllegalChars) {
- std::replace(Filename.begin(), Filename.end(), IllegalChar,
- ReplacementChar);
- }
-
- return Filename;
-}
-
-std::string llvm::createGraphFilename(const Twine &Name, int &FD) {
- FD = -1;
- SmallString<128> Filename;
-
- // Windows can't always handle long paths, so limit the length of the name.
- std::string N = Name.str();
- N = N.substr(0, std::min<std::size_t>(N.size(), 140));
-
- // Replace illegal characters in graph Filename with '_' if needed
- std::string CleansedName = replaceIllegalFilenameChars(N, '_');
-
- std::error_code EC =
- sys::fs::createTemporaryFile(CleansedName, "dot", FD, Filename);
- if (EC) {
- errs() << "Error: " << EC.message() << "\n";
- return "";
- }
-
- errs() << "Writing '" << Filename << "'... ";
- return std::string(Filename.str());
-}
-
-// Execute the graph viewer. Return true if there were errors.
-static bool ExecGraphViewer(StringRef ExecPath, std::vector<StringRef> &args,
- StringRef Filename, bool wait,
- std::string &ErrMsg) {
- if (wait) {
- if (sys::ExecuteAndWait(ExecPath, args, None, {}, 0, 0, &ErrMsg)) {
- errs() << "Error: " << ErrMsg << "\n";
- return true;
- }
- sys::fs::remove(Filename);
- errs() << " done. \n";
- } else {
- sys::ExecuteNoWait(ExecPath, args, None, {}, 0, &ErrMsg);
- errs() << "Remember to erase graph file: " << Filename << "\n";
- }
- return false;
-}
-
-namespace {
-
-struct GraphSession {
- std::string LogBuffer;
-
- bool TryFindProgram(StringRef Names, std::string &ProgramPath) {
- raw_string_ostream Log(LogBuffer);
- SmallVector<StringRef, 8> parts;
- Names.split(parts, '|');
- for (auto Name : parts) {
- if (ErrorOr<std::string> P = sys::findProgramByName(Name)) {
- ProgramPath = *P;
- return true;
- }
- Log << " Tried '" << Name << "'\n";
- }
- return false;
- }
-};
-
-} // end anonymous namespace
-
-static const char *getProgramName(GraphProgram::Name program) {
- switch (program) {
- case GraphProgram::DOT:
- return "dot";
- case GraphProgram::FDP:
- return "fdp";
- case GraphProgram::NEATO:
- return "neato";
- case GraphProgram::TWOPI:
- return "twopi";
- case GraphProgram::CIRCO:
- return "circo";
- }
- llvm_unreachable("bad kind");
-}
-
-bool llvm::DisplayGraph(StringRef FilenameRef, bool wait,
- GraphProgram::Name program) {
- std::string Filename = std::string(FilenameRef);
- std::string ErrMsg;
- std::string ViewerPath;
- GraphSession S;
-
-#ifdef __APPLE__
- wait &= !ViewBackground;
- if (S.TryFindProgram("open", ViewerPath)) {
- std::vector<StringRef> args;
- args.push_back(ViewerPath);
- if (wait)
- args.push_back("-W");
- args.push_back(Filename);
- errs() << "Trying 'open' program... ";
- if (!ExecGraphViewer(ViewerPath, args, Filename, wait, ErrMsg))
- return false;
- }
-#endif
- if (S.TryFindProgram("xdg-open", ViewerPath)) {
- std::vector<StringRef> args;
- args.push_back(ViewerPath);
- args.push_back(Filename);
- errs() << "Trying 'xdg-open' program... ";
- if (!ExecGraphViewer(ViewerPath, args, Filename, wait, ErrMsg))
- return false;
- }
-
- // Graphviz
- if (S.TryFindProgram("Graphviz", ViewerPath)) {
- std::vector<StringRef> args;
- args.push_back(ViewerPath);
- args.push_back(Filename);
-
- errs() << "Running 'Graphviz' program... ";
- return ExecGraphViewer(ViewerPath, args, Filename, wait, ErrMsg);
- }
-
- // xdot
- if (S.TryFindProgram("xdot|xdot.py", ViewerPath)) {
- std::vector<StringRef> args;
- args.push_back(ViewerPath);
- args.push_back(Filename);
-
- args.push_back("-f");
- args.push_back(getProgramName(program));
-
- errs() << "Running 'xdot.py' program... ";
- return ExecGraphViewer(ViewerPath, args, Filename, wait, ErrMsg);
- }
-
- enum ViewerKind {
- VK_None,
- VK_OSXOpen,
- VK_XDGOpen,
- VK_Ghostview,
- VK_CmdStart
- };
- ViewerKind Viewer = VK_None;
-#ifdef __APPLE__
- if (!Viewer && S.TryFindProgram("open", ViewerPath))
- Viewer = VK_OSXOpen;
-#endif
- if (!Viewer && S.TryFindProgram("gv", ViewerPath))
- Viewer = VK_Ghostview;
- if (!Viewer && S.TryFindProgram("xdg-open", ViewerPath))
- Viewer = VK_XDGOpen;
-#ifdef _WIN32
- if (!Viewer && S.TryFindProgram("cmd", ViewerPath)) {
- Viewer = VK_CmdStart;
- }
-#endif
-
- // PostScript or PDF graph generator + PostScript/PDF viewer
- std::string GeneratorPath;
- if (Viewer &&
- (S.TryFindProgram(getProgramName(program), GeneratorPath) ||
- S.TryFindProgram("dot|fdp|neato|twopi|circo", GeneratorPath))) {
- std::string OutputFilename =
- Filename + (Viewer == VK_CmdStart ? ".pdf" : ".ps");
-
- std::vector<StringRef> args;
- args.push_back(GeneratorPath);
- if (Viewer == VK_CmdStart)
- args.push_back("-Tpdf");
- else
- args.push_back("-Tps");
- args.push_back("-Nfontname=Courier");
- args.push_back("-Gsize=7.5,10");
- args.push_back(Filename);
- args.push_back("-o");
- args.push_back(OutputFilename);
-
- errs() << "Running '" << GeneratorPath << "' program... ";
-
- if (ExecGraphViewer(GeneratorPath, args, Filename, true, ErrMsg))
- return true;
-
- // The lifetime of StartArg must include the call of ExecGraphViewer
- // because the args are passed as vector of char*.
- std::string StartArg;
-
- args.clear();
- args.push_back(ViewerPath);
- switch (Viewer) {
- case VK_OSXOpen:
- args.push_back("-W");
- args.push_back(OutputFilename);
- break;
- case VK_XDGOpen:
- wait = false;
- args.push_back(OutputFilename);
- break;
- case VK_Ghostview:
- args.push_back("--spartan");
- args.push_back(OutputFilename);
- break;
- case VK_CmdStart:
- args.push_back("/S");
- args.push_back("/C");
- StartArg =
- (StringRef("start ") + (wait ? "/WAIT " : "") + OutputFilename).str();
- args.push_back(StartArg);
- break;
- case VK_None:
- llvm_unreachable("Invalid viewer");
- }
-
- ErrMsg.clear();
- return ExecGraphViewer(ViewerPath, args, OutputFilename, wait, ErrMsg);
- }
-
- // dotty
- if (S.TryFindProgram("dotty", ViewerPath)) {
- std::vector<StringRef> args;
- args.push_back(ViewerPath);
- args.push_back(Filename);
-
-// Dotty spawns another app and doesn't wait until it returns
-#ifdef _WIN32
- wait = false;
-#endif
- errs() << "Running 'dotty' program... ";
- return ExecGraphViewer(ViewerPath, args, Filename, wait, ErrMsg);
- }
-
- errs() << "Error: Couldn't find a usable graph viewer program:\n";
- errs() << S.LogBuffer << "\n";
- return true;
-}
+//===- GraphWriter.cpp - Implements GraphWriter support routines ----------===//
+//
+// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
+// See https://llvm.org/LICENSE.txt for license information.
+// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+//
+//===----------------------------------------------------------------------===//
+//
+// This file implements misc. GraphWriter support routines.
+//
+//===----------------------------------------------------------------------===//
+
+#include "llvm/Support/GraphWriter.h"
+#include "llvm/ADT/SmallString.h"
+#include "llvm/ADT/SmallVector.h"
+#include "llvm/ADT/StringRef.h"
+#include "llvm/Config/config.h"
+#include "llvm/Support/CommandLine.h"
+#include "llvm/Support/Compiler.h"
+#include "llvm/Support/ErrorHandling.h"
+#include "llvm/Support/ErrorOr.h"
+#include "llvm/Support/FileSystem.h"
+#include "llvm/Support/Program.h"
+#include "llvm/Support/raw_ostream.h"
+#include <cassert>
+#include <system_error>
+#include <string>
+#include <vector>
+
+using namespace llvm;
+
+static cl::opt<bool> ViewBackground("view-background", cl::Hidden,
+ cl::desc("Execute graph viewer in the background. Creates tmp file litter."));
+
+std::string llvm::DOT::EscapeString(const std::string &Label) {
+ std::string Str(Label);
+ for (unsigned i = 0; i != Str.length(); ++i)
+ switch (Str[i]) {
+ case '\n':
+ Str.insert(Str.begin()+i, '\\'); // Escape character...
+ ++i;
+ Str[i] = 'n';
+ break;
+ case '\t':
+ Str.insert(Str.begin()+i, ' '); // Convert to two spaces
+ ++i;
+ Str[i] = ' ';
+ break;
+ case '\\':
+ if (i+1 != Str.length())
+ switch (Str[i+1]) {
+ case 'l': continue; // don't disturb \l
+ case '|': case '{': case '}':
+ Str.erase(Str.begin()+i); continue;
+ default: break;
+ }
+ LLVM_FALLTHROUGH;
+ case '{': case '}':
+ case '<': case '>':
+ case '|': case '"':
+ Str.insert(Str.begin()+i, '\\'); // Escape character...
+ ++i; // don't infinite loop
+ break;
+ }
+ return Str;
+}
+
+/// Get a color string for this node number. Simply round-robin selects
+/// from a reasonable number of colors.
+StringRef llvm::DOT::getColorString(unsigned ColorNumber) {
+ static const int NumColors = 20;
+ static const char* Colors[NumColors] = {
+ "aaaaaa", "aa0000", "00aa00", "aa5500", "0055ff", "aa00aa", "00aaaa",
+ "555555", "ff5555", "55ff55", "ffff55", "5555ff", "ff55ff", "55ffff",
+ "ffaaaa", "aaffaa", "ffffaa", "aaaaff", "ffaaff", "aaffff"};
+ return Colors[ColorNumber % NumColors];
+}
+
+static std::string replaceIllegalFilenameChars(std::string Filename,
+ const char ReplacementChar) {
+#ifdef _WIN32
+ std::string IllegalChars = "\\/:?\"<>|";
+#else
+ std::string IllegalChars = "/";
+#endif
+
+ for (char IllegalChar : IllegalChars) {
+ std::replace(Filename.begin(), Filename.end(), IllegalChar,
+ ReplacementChar);
+ }
+
+ return Filename;
+}
+
+std::string llvm::createGraphFilename(const Twine &Name, int &FD) {
+ FD = -1;
+ SmallString<128> Filename;
+
+ // Windows can't always handle long paths, so limit the length of the name.
+ std::string N = Name.str();
+ N = N.substr(0, std::min<std::size_t>(N.size(), 140));
+
+ // Replace illegal characters in graph Filename with '_' if needed
+ std::string CleansedName = replaceIllegalFilenameChars(N, '_');
+
+ std::error_code EC =
+ sys::fs::createTemporaryFile(CleansedName, "dot", FD, Filename);
+ if (EC) {
+ errs() << "Error: " << EC.message() << "\n";
+ return "";
+ }
+
+ errs() << "Writing '" << Filename << "'... ";
+ return std::string(Filename.str());
+}
+
+// Execute the graph viewer. Return true if there were errors.
+static bool ExecGraphViewer(StringRef ExecPath, std::vector<StringRef> &args,
+ StringRef Filename, bool wait,
+ std::string &ErrMsg) {
+ if (wait) {
+ if (sys::ExecuteAndWait(ExecPath, args, None, {}, 0, 0, &ErrMsg)) {
+ errs() << "Error: " << ErrMsg << "\n";
+ return true;
+ }
+ sys::fs::remove(Filename);
+ errs() << " done. \n";
+ } else {
+ sys::ExecuteNoWait(ExecPath, args, None, {}, 0, &ErrMsg);
+ errs() << "Remember to erase graph file: " << Filename << "\n";
+ }
+ return false;
+}
+
+namespace {
+
+struct GraphSession {
+ std::string LogBuffer;
+
+ bool TryFindProgram(StringRef Names, std::string &ProgramPath) {
+ raw_string_ostream Log(LogBuffer);
+ SmallVector<StringRef, 8> parts;
+ Names.split(parts, '|');
+ for (auto Name : parts) {
+ if (ErrorOr<std::string> P = sys::findProgramByName(Name)) {
+ ProgramPath = *P;
+ return true;
+ }
+ Log << " Tried '" << Name << "'\n";
+ }
+ return false;
+ }
+};
+
+} // end anonymous namespace
+
+static const char *getProgramName(GraphProgram::Name program) {
+ switch (program) {
+ case GraphProgram::DOT:
+ return "dot";
+ case GraphProgram::FDP:
+ return "fdp";
+ case GraphProgram::NEATO:
+ return "neato";
+ case GraphProgram::TWOPI:
+ return "twopi";
+ case GraphProgram::CIRCO:
+ return "circo";
+ }
+ llvm_unreachable("bad kind");
+}
+
+bool llvm::DisplayGraph(StringRef FilenameRef, bool wait,
+ GraphProgram::Name program) {
+ std::string Filename = std::string(FilenameRef);
+ std::string ErrMsg;
+ std::string ViewerPath;
+ GraphSession S;
+
+#ifdef __APPLE__
+ wait &= !ViewBackground;
+ if (S.TryFindProgram("open", ViewerPath)) {
+ std::vector<StringRef> args;
+ args.push_back(ViewerPath);
+ if (wait)
+ args.push_back("-W");
+ args.push_back(Filename);
+ errs() << "Trying 'open' program... ";
+ if (!ExecGraphViewer(ViewerPath, args, Filename, wait, ErrMsg))
+ return false;
+ }
+#endif
+ if (S.TryFindProgram("xdg-open", ViewerPath)) {
+ std::vector<StringRef> args;
+ args.push_back(ViewerPath);
+ args.push_back(Filename);
+ errs() << "Trying 'xdg-open' program... ";
+ if (!ExecGraphViewer(ViewerPath, args, Filename, wait, ErrMsg))
+ return false;
+ }
+
+ // Graphviz
+ if (S.TryFindProgram("Graphviz", ViewerPath)) {
+ std::vector<StringRef> args;
+ args.push_back(ViewerPath);
+ args.push_back(Filename);
+
+ errs() << "Running 'Graphviz' program... ";
+ return ExecGraphViewer(ViewerPath, args, Filename, wait, ErrMsg);
+ }
+
+ // xdot
+ if (S.TryFindProgram("xdot|xdot.py", ViewerPath)) {
+ std::vector<StringRef> args;
+ args.push_back(ViewerPath);
+ args.push_back(Filename);
+
+ args.push_back("-f");
+ args.push_back(getProgramName(program));
+
+ errs() << "Running 'xdot.py' program... ";
+ return ExecGraphViewer(ViewerPath, args, Filename, wait, ErrMsg);
+ }
+
+ enum ViewerKind {
+ VK_None,
+ VK_OSXOpen,
+ VK_XDGOpen,
+ VK_Ghostview,
+ VK_CmdStart
+ };
+ ViewerKind Viewer = VK_None;
+#ifdef __APPLE__
+ if (!Viewer && S.TryFindProgram("open", ViewerPath))
+ Viewer = VK_OSXOpen;
+#endif
+ if (!Viewer && S.TryFindProgram("gv", ViewerPath))
+ Viewer = VK_Ghostview;
+ if (!Viewer && S.TryFindProgram("xdg-open", ViewerPath))
+ Viewer = VK_XDGOpen;
+#ifdef _WIN32
+ if (!Viewer && S.TryFindProgram("cmd", ViewerPath)) {
+ Viewer = VK_CmdStart;
+ }
+#endif
+
+ // PostScript or PDF graph generator + PostScript/PDF viewer
+ std::string GeneratorPath;
+ if (Viewer &&
+ (S.TryFindProgram(getProgramName(program), GeneratorPath) ||
+ S.TryFindProgram("dot|fdp|neato|twopi|circo", GeneratorPath))) {
+ std::string OutputFilename =
+ Filename + (Viewer == VK_CmdStart ? ".pdf" : ".ps");
+
+ std::vector<StringRef> args;
+ args.push_back(GeneratorPath);
+ if (Viewer == VK_CmdStart)
+ args.push_back("-Tpdf");
+ else
+ args.push_back("-Tps");
+ args.push_back("-Nfontname=Courier");
+ args.push_back("-Gsize=7.5,10");
+ args.push_back(Filename);
+ args.push_back("-o");
+ args.push_back(OutputFilename);
+
+ errs() << "Running '" << GeneratorPath << "' program... ";
+
+ if (ExecGraphViewer(GeneratorPath, args, Filename, true, ErrMsg))
+ return true;
+
+ // The lifetime of StartArg must include the call of ExecGraphViewer
+ // because the args are passed as vector of char*.
+ std::string StartArg;
+
+ args.clear();
+ args.push_back(ViewerPath);
+ switch (Viewer) {
+ case VK_OSXOpen:
+ args.push_back("-W");
+ args.push_back(OutputFilename);
+ break;
+ case VK_XDGOpen:
+ wait = false;
+ args.push_back(OutputFilename);
+ break;
+ case VK_Ghostview:
+ args.push_back("--spartan");
+ args.push_back(OutputFilename);
+ break;
+ case VK_CmdStart:
+ args.push_back("/S");
+ args.push_back("/C");
+ StartArg =
+ (StringRef("start ") + (wait ? "/WAIT " : "") + OutputFilename).str();
+ args.push_back(StartArg);
+ break;
+ case VK_None:
+ llvm_unreachable("Invalid viewer");
+ }
+
+ ErrMsg.clear();
+ return ExecGraphViewer(ViewerPath, args, OutputFilename, wait, ErrMsg);
+ }
+
+ // dotty
+ if (S.TryFindProgram("dotty", ViewerPath)) {
+ std::vector<StringRef> args;
+ args.push_back(ViewerPath);
+ args.push_back(Filename);
+
+// Dotty spawns another app and doesn't wait until it returns
+#ifdef _WIN32
+ wait = false;
+#endif
+ errs() << "Running 'dotty' program... ";
+ return ExecGraphViewer(ViewerPath, args, Filename, wait, ErrMsg);
+ }
+
+ errs() << "Error: Couldn't find a usable graph viewer program:\n";
+ errs() << S.LogBuffer << "\n";
+ return true;
+}