Dmitri Gribenko | aa58081 | 2012-08-09 00:03:17 +0000 | [diff] [blame] | 1 | //===--- CommentCommandTraits.cpp - Comment command properties --*- 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 | #include "clang/AST/CommentCommandTraits.h" |
Dmitri Gribenko | e4330a3 | 2012-09-10 20:32:42 +0000 | [diff] [blame] | 11 | #include "llvm/ADT/STLExtras.h" |
Dmitri Gribenko | aa58081 | 2012-08-09 00:03:17 +0000 | [diff] [blame] | 12 | |
| 13 | namespace clang { |
| 14 | namespace comments { |
| 15 | |
Dmitri Gribenko | e4330a3 | 2012-09-10 20:32:42 +0000 | [diff] [blame] | 16 | #include "clang/AST/CommentCommandInfo.inc" |
Dmitri Gribenko | aa58081 | 2012-08-09 00:03:17 +0000 | [diff] [blame] | 17 | |
Dmitri Gribenko | e4330a3 | 2012-09-10 20:32:42 +0000 | [diff] [blame] | 18 | CommandTraits::CommandTraits(llvm::BumpPtrAllocator &Allocator) : |
| 19 | NextID(llvm::array_lengthof(Commands)), Allocator(Allocator) |
| 20 | { } |
Dmitri Gribenko | aa58081 | 2012-08-09 00:03:17 +0000 | [diff] [blame] | 21 | |
Dmitri Gribenko | e4330a3 | 2012-09-10 20:32:42 +0000 | [diff] [blame] | 22 | const CommandInfo *CommandTraits::getCommandInfoOrNULL(StringRef Name) const { |
| 23 | if (const CommandInfo *Info = getBuiltinCommandInfo(Name)) |
| 24 | return Info; |
| 25 | return getRegisteredCommandInfo(Name); |
| 26 | } |
Dmitri Gribenko | aa58081 | 2012-08-09 00:03:17 +0000 | [diff] [blame] | 27 | |
Dmitri Gribenko | e4330a3 | 2012-09-10 20:32:42 +0000 | [diff] [blame] | 28 | const CommandInfo *CommandTraits::getCommandInfo(unsigned CommandID) const { |
| 29 | if (const CommandInfo *Info = getBuiltinCommandInfo(CommandID)) |
| 30 | return Info; |
| 31 | return getRegisteredCommandInfo(CommandID); |
| 32 | } |
Dmitri Gribenko | aa58081 | 2012-08-09 00:03:17 +0000 | [diff] [blame] | 33 | |
Dmitri Gribenko | e4330a3 | 2012-09-10 20:32:42 +0000 | [diff] [blame] | 34 | const CommandInfo *CommandTraits::registerUnknownCommand(StringRef CommandName) { |
Eli Friedman | 116bb09 | 2012-09-11 00:36:26 +0000 | [diff] [blame] | 35 | char *Name = Allocator.Allocate<char>(CommandName.size() + 1); |
Dmitri Gribenko | e4330a3 | 2012-09-10 20:32:42 +0000 | [diff] [blame] | 36 | memcpy(Name, CommandName.data(), CommandName.size()); |
Eli Friedman | 116bb09 | 2012-09-11 00:36:26 +0000 | [diff] [blame] | 37 | Name[CommandName.size()] = '\0'; |
Dmitri Gribenko | aa58081 | 2012-08-09 00:03:17 +0000 | [diff] [blame] | 38 | |
Dmitri Gribenko | e4330a3 | 2012-09-10 20:32:42 +0000 | [diff] [blame] | 39 | // Value-initialize (=zero-initialize in this case) a new CommandInfo. |
| 40 | CommandInfo *Info = new (Allocator) CommandInfo(); |
| 41 | Info->Name = Name; |
| 42 | Info->ID = NextID++; |
Dmitri Gribenko | b0b8a96 | 2012-09-11 19:22:03 +0000 | [diff] [blame] | 43 | Info->IsUnknownCommand = true; |
Dmitri Gribenko | e4330a3 | 2012-09-10 20:32:42 +0000 | [diff] [blame] | 44 | |
| 45 | RegisteredCommands.push_back(Info); |
| 46 | |
| 47 | return Info; |
| 48 | } |
| 49 | |
| 50 | const CommandInfo *CommandTraits::getBuiltinCommandInfo( |
| 51 | unsigned CommandID) { |
| 52 | if (CommandID < llvm::array_lengthof(Commands)) |
| 53 | return &Commands[CommandID]; |
| 54 | return NULL; |
| 55 | } |
| 56 | |
| 57 | const CommandInfo *CommandTraits::getRegisteredCommandInfo( |
| 58 | StringRef Name) const { |
| 59 | for (unsigned i = 0, e = RegisteredCommands.size(); i != e; ++i) { |
| 60 | if (RegisteredCommands[i]->Name == Name) |
| 61 | return RegisteredCommands[i]; |
Dmitri Gribenko | aa58081 | 2012-08-09 00:03:17 +0000 | [diff] [blame] | 62 | } |
Dmitri Gribenko | e4330a3 | 2012-09-10 20:32:42 +0000 | [diff] [blame] | 63 | return NULL; |
Dmitri Gribenko | aa58081 | 2012-08-09 00:03:17 +0000 | [diff] [blame] | 64 | } |
| 65 | |
Dmitri Gribenko | e4330a3 | 2012-09-10 20:32:42 +0000 | [diff] [blame] | 66 | const CommandInfo *CommandTraits::getRegisteredCommandInfo( |
| 67 | unsigned CommandID) const { |
| 68 | return RegisteredCommands[CommandID - llvm::array_lengthof(Commands)]; |
Dmitri Gribenko | aa58081 | 2012-08-09 00:03:17 +0000 | [diff] [blame] | 69 | } |
| 70 | |
| 71 | } // end namespace comments |
| 72 | } // end namespace clang |
| 73 | |