contrib/llvm/tools/clang/lib/AST/CommentCommandTraits.cpp

   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"
  11 #include "llvm/ADT/STLExtras.h"
  12
  13 namespace clang {
  14 namespace comments {
  15
  16 #include "clang/AST/CommentCommandInfo.inc"
  17
  18 CommandTraits::CommandTraits(llvm::BumpPtrAllocator &Allocator,
  19                              const CommentOptions &CommentOptions) :
  20     NextID(llvm::array_lengthof(Commands)), Allocator(Allocator) {
  21   registerCommentOptions(CommentOptions);
  22 }
  23
  24 void CommandTraits::registerCommentOptions(
  25     const CommentOptions &CommentOptions) {
  26   for (CommentOptions::BlockCommandNamesTy::const_iterator
  27            I = CommentOptions.BlockCommandNames.begin(),
  28            E = CommentOptions.BlockCommandNames.end();
  29        I != E; I++) {
  30     registerBlockCommand(*I);
  31   }
  32 }
  33
  34 const CommandInfo *CommandTraits::getCommandInfoOrNULL(StringRef Name) const {
  35   if (const CommandInfo *Info = getBuiltinCommandInfo(Name))
  36     return Info;
  37   return getRegisteredCommandInfo(Name);
  38 }
  39
  40 const CommandInfo *CommandTraits::getCommandInfo(unsigned CommandID) const {
  41   if (const CommandInfo *Info = getBuiltinCommandInfo(CommandID))
  42     return Info;
  43   return getRegisteredCommandInfo(CommandID);
  44 }
  45
  46 CommandInfo *CommandTraits::createCommandInfoWithName(StringRef CommandName) {
  47   char *Name = Allocator.Allocate<char>(CommandName.size() + 1);
  48   memcpy(Name, CommandName.data(), CommandName.size());
  49   Name[CommandName.size()] = '\0';
  50
  51   // Value-initialize (=zero-initialize in this case) a new CommandInfo.
  52   CommandInfo *Info = new (Allocator) CommandInfo();
  53   Info->Name = Name;
  54   Info->ID = NextID++;
  55
  56   RegisteredCommands.push_back(Info);
  57
  58   return Info;
  59 }
  60
  61 const CommandInfo *CommandTraits::registerUnknownCommand(
  62                                                   StringRef CommandName) {
  63   CommandInfo *Info = createCommandInfoWithName(CommandName);
  64   Info->IsUnknownCommand = true;
  65   return Info;
  66 }
  67
  68 const CommandInfo *CommandTraits::registerBlockCommand(StringRef CommandName) {
  69   CommandInfo *Info = createCommandInfoWithName(CommandName);
  70   Info->IsBlockCommand = true;
  71   return Info;
  72 }
  73
  74 const CommandInfo *CommandTraits::getBuiltinCommandInfo(
  75                                                   unsigned CommandID) {
  76   if (CommandID < llvm::array_lengthof(Commands))
  77     return &Commands[CommandID];
  78   return NULL;
  79 }
  80
  81 const CommandInfo *CommandTraits::getRegisteredCommandInfo(
  82                                                   StringRef Name) const {
  83   for (unsigned i = 0, e = RegisteredCommands.size(); i != e; ++i) {
  84     if (RegisteredCommands[i]->Name == Name)
  85       return RegisteredCommands[i];
  86   }
  87   return NULL;
  88 }
  89
  90 const CommandInfo *CommandTraits::getRegisteredCommandInfo(
  91                                                   unsigned CommandID) const {
  92   return RegisteredCommands[CommandID - llvm::array_lengthof(Commands)];
  93 }
  94
  95 } // end namespace comments
  96 } // end namespace clang
  97