From 972bf8dea1001134f575a1c6362e1c12eed579d0 Mon Sep 17 00:00:00 2001
From: Sean Callanan <scallanan@apple.com>
Date: Tue, 11 May 2010 01:27:08 +0000
Subject: [PATCH] Extended the edis "IsBranch" property to call instructions as
 well.  Added support for checking this to the llvm-mc tester as well.

git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@103454 91177308-0d34-0410-b5e6-96231b3b80d8
---
 tools/edis/EDInst.cpp          | 4 +++-
 tools/llvm-mc/Disassembler.cpp | 5 +++++
 2 files changed, 8 insertions(+), 1 deletion(-)

diff --git a/tools/edis/EDInst.cpp b/tools/edis/EDInst.cpp
index af3a54abbcf..c009f0f8687 100644
--- a/tools/edis/EDInst.cpp
+++ b/tools/edis/EDInst.cpp
@@ -81,7 +81,9 @@ unsigned EDInst::instID() {
 
 bool EDInst::isBranch() {
   if (ThisInstInfo)
-    return ThisInstInfo->instructionType == kInstructionTypeBranch;
+    return 
+      ThisInstInfo->instructionType == kInstructionTypeBranch ||
+      ThisInstInfo->instructionType == kInstructionTypeCall;
   else
     return false;
 }
diff --git a/tools/llvm-mc/Disassembler.cpp b/tools/llvm-mc/Disassembler.cpp
index e2120c25d3a..37b2cb80563 100644
--- a/tools/llvm-mc/Disassembler.cpp
+++ b/tools/llvm-mc/Disassembler.cpp
@@ -314,6 +314,11 @@ int Disassembler::disassembleEnhanced(const std::string &TS,
   }
   
   outs() << " ";
+    
+  if (EDInstIsBranch(inst))
+    outs() << "<br> ";
+  if (EDInstIsMove(inst))
+    outs() << "<mov> ";
   
   int numOperands = EDNumOperands(inst);