1 //===- Error.cpp - tblgen error handling helper routines --------*- C++ -*-===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 //
10 // This file contains error handling helper routines to pretty-print diagnostic
11 // messages from tblgen.
12 //
13 //===----------------------------------------------------------------------===//
14 
15 #include "llvm/TableGen/Error.h"
16 #include "llvm/ADT/Twine.h"
17 #include "llvm/Support/Signals.h"
18 #include "llvm/Support/WithColor.h"
19 #include "llvm/Support/raw_ostream.h"
20 #include <cstdlib>
21 
22 namespace llvm {
23 
24 SourceMgr SrcMgr;
25 unsigned ErrorsPrinted = 0;
26 
PrintMessage(ArrayRef<SMLoc> Loc,SourceMgr::DiagKind Kind,const Twine & Msg)27 static void PrintMessage(ArrayRef<SMLoc> Loc, SourceMgr::DiagKind Kind,
28                          const Twine &Msg) {
29   // Count the total number of errors printed.
30   // This is used to exit with an error code if there were any errors.
31   if (Kind == SourceMgr::DK_Error)
32     ++ErrorsPrinted;
33 
34   SMLoc NullLoc;
35   if (Loc.empty())
36     Loc = NullLoc;
37   SrcMgr.PrintMessage(Loc.front(), Kind, Msg);
38   for (unsigned i = 1; i < Loc.size(); ++i)
39     SrcMgr.PrintMessage(Loc[i], SourceMgr::DK_Note,
40                         "instantiated from multiclass");
41 }
42 
PrintNote(ArrayRef<SMLoc> NoteLoc,const Twine & Msg)43 void PrintNote(ArrayRef<SMLoc> NoteLoc, const Twine &Msg) {
44   PrintMessage(NoteLoc, SourceMgr::DK_Note, Msg);
45 }
46 
PrintWarning(ArrayRef<SMLoc> WarningLoc,const Twine & Msg)47 void PrintWarning(ArrayRef<SMLoc> WarningLoc, const Twine &Msg) {
48   PrintMessage(WarningLoc, SourceMgr::DK_Warning, Msg);
49 }
50 
PrintWarning(const char * Loc,const Twine & Msg)51 void PrintWarning(const char *Loc, const Twine &Msg) {
52   SrcMgr.PrintMessage(SMLoc::getFromPointer(Loc), SourceMgr::DK_Warning, Msg);
53 }
54 
PrintWarning(const Twine & Msg)55 void PrintWarning(const Twine &Msg) { WithColor::warning() << Msg << "\n"; }
56 
PrintError(ArrayRef<SMLoc> ErrorLoc,const Twine & Msg)57 void PrintError(ArrayRef<SMLoc> ErrorLoc, const Twine &Msg) {
58   PrintMessage(ErrorLoc, SourceMgr::DK_Error, Msg);
59 }
60 
PrintError(const char * Loc,const Twine & Msg)61 void PrintError(const char *Loc, const Twine &Msg) {
62   SrcMgr.PrintMessage(SMLoc::getFromPointer(Loc), SourceMgr::DK_Error, Msg);
63 }
64 
PrintError(const Twine & Msg)65 void PrintError(const Twine &Msg) { WithColor::error() << Msg << "\n"; }
66 
PrintFatalError(const Twine & Msg)67 void PrintFatalError(const Twine &Msg) {
68   PrintError(Msg);
69   // The following call runs the file cleanup handlers.
70   sys::RunInterruptHandlers();
71   std::exit(1);
72 }
73 
PrintFatalError(ArrayRef<SMLoc> ErrorLoc,const Twine & Msg)74 void PrintFatalError(ArrayRef<SMLoc> ErrorLoc, const Twine &Msg) {
75   PrintError(ErrorLoc, Msg);
76   // The following call runs the file cleanup handlers.
77   sys::RunInterruptHandlers();
78   std::exit(1);
79 }
80 
81 } // end namespace llvm
82