Chromium Code Reviews| Index: ppapi/tests/clang/find_affected_interfaces.cc |
| =================================================================== |
| --- ppapi/tests/clang/find_affected_interfaces.cc (revision 0) |
| +++ ppapi/tests/clang/find_affected_interfaces.cc (revision 0) |
| @@ -0,0 +1,154 @@ |
| +// Copyright (c) 2010 The Chromium Authors. All rights reserved. |
| +// Use of this source code is governed by a BSD-style license that can be |
| +// found in the LICENSE file. |
| + |
| +// Clang plugin which finds types that are affected if the types listed in the |
| +// plugin parameters are changed. This is for use in determining what PPAPI |
| +// C-level interfaces are affected if one or more PPAPI C structs are changed. |
|
David Springer
2010/12/13 21:23:08
This comment seems a little terse to me - perhaps
|
| + |
| +#include <algorithm> |
| +#include <cstdio> |
| +#include <map> |
| +#include <sstream> |
| +#include <string> |
| +#include <vector> |
| + |
| +#include "clang/AST/AST.h" |
| +#include "clang/AST/ASTConsumer.h" |
| +#include "clang/AST/CharUnits.h" |
| +#include "clang/Basic/SourceManager.h" |
| +#include "clang/Basic/TargetInfo.h" |
| +#include "clang/Frontend/CompilerInstance.h" |
| +#include "clang/Frontend/FrontendPluginRegistry.h" |
| + |
| +namespace { |
| + |
| +typedef std::vector<std::string> StringVec; |
| + |
| +class FindAffectedInterfacesConsumer : public clang::ASTConsumer { |
| + public: |
| + explicit FindAffectedInterfacesConsumer(const StringVec& changed_classes) { |
| + StringVec::const_iterator iter(changed_classes.begin()); |
| + StringVec::const_iterator the_end(changed_classes.end()); |
| + for (; iter != the_end; ++iter) { |
| + class_is_affected_map_[*iter] = true; |
| + } |
| + } |
| + |
| + private: |
| + typedef std::map<std::string, bool> StringBoolMap; |
| + StringBoolMap class_is_affected_map_; |
| + |
| + bool IsAffected(const clang::Type& type_to_check) { |
|
David Springer
2010/12/13 21:23:08
Please comment this method. It appears to be the
|
| + std::string type_string( |
| + type_to_check.getCanonicalTypeInternal().getAsString()); |
| + std::pair<StringBoolMap::iterator, bool> iter_success_pair = |
| + class_is_affected_map_.insert( |
| + StringBoolMap::value_type(type_string, false)); |
| + StringBoolMap::iterator iter(iter_success_pair.first); |
| + bool successfully_inserted(iter_success_pair.second); |
| + // If we were able to insert, then we haven't yet seen this type. Compute |
| + // IsAffected and put it at the newly inserted location in the map. |
| + if (successfully_inserted) { |
| + if (type_to_check.isPointerType()) { |
|
David Springer
2010/12/13 21:23:08
I am not able to get a clear picture in my head of
|
| + const clang::PointerType* pointer_type = |
| + dyn_cast<clang::PointerType>(&type_to_check); |
|
David Springer
2010/12/13 21:23:08
Maybe a quick comment on dyn_cast vs. dynamic_cast
|
| + // Recurse to the pointee type. |
| + iter->second = IsAffected(*pointer_type->getPointeeType().getTypePtr()); |
| + } else if (type_to_check.isFunctionProtoType()) { |
| + const clang::FunctionProtoType* func_type = |
|
David Springer
2010/12/13 21:23:08
An input stream example that triggers this would b
|
| + dyn_cast<clang::FunctionProtoType>(&type_to_check); |
| + // Recurse to the return type and parameter types. |
| + iter->second = IsAffected(*func_type->getResultType().getTypePtr()); |
| + if (!iter->second) { |
| + clang::FunctionProtoType::arg_type_iterator arg_iter = |
| + func_type->arg_type_begin(); |
| + clang::FunctionProtoType::arg_type_iterator arg_end = |
| + func_type->arg_type_end(); |
| + for (; (arg_iter != arg_end) && (!iter->second); ++arg_iter) { |
| + iter->second = IsAffected(*(arg_iter->getTypePtr())); |
| + } |
| + } |
| + } else if (type_to_check.isRecordType()) { |
| + // For records (unions, structs), recurse to the fields. |
|
David Springer
2010/12/13 21:23:08
Input stream example?
|
| + const clang::RecordType* record = |
| + dyn_cast<clang::RecordType>(&type_to_check); |
| + const clang::RecordDecl* decl = record->getDecl(); |
| + clang::RecordDecl::field_iterator field_iter(decl->field_begin()); |
| + clang::RecordDecl::field_iterator field_end(decl->field_end()); |
| + for (; (field_iter != field_end) && (!iter->second); ++field_iter) { |
| + iter->second = IsAffected(*(field_iter->getType().getTypePtr())); |
| + } |
| + } |
| + } |
| + // By this point, the bool in the map at the location referenced by iter has |
| + // the correct value. Either it was cached, or we computed & inserted it. |
| + return iter->second; |
| + } |
| + |
| + // Virtual function to consume top-level declarations. For each one, we check |
| + // to see if it is a type definition. If it is, we print information about |
| + // it. |
|
David Springer
2010/12/13 21:23:08
But the above IsAffected() code processes function
|
| + virtual void HandleTopLevelDecl(clang::DeclGroupRef decl_group) { |
| + clang::DeclGroupRef::iterator iter(decl_group.begin()); |
| + clang::DeclGroupRef::iterator the_end(decl_group.end()); |
| + for (; iter != the_end; ++iter) { |
| + const clang::Decl *decl = *iter; |
| + if (const clang::TypeDecl* type_decl = dyn_cast<clang::TypeDecl>(decl)) { |
| + std::string name(type_decl->getNameAsString()); |
| + // TagDecl covers structs, enums, unions, and classes. |
| + if (const clang::TagDecl* tag = dyn_cast<clang::TagDecl>(type_decl)) { |
| + // Only print out info when we find the definition; ignore forward |
| + // references. |
| + if (tag->isDefinition()) { |
| + clang::Type* type = type_decl->getTypeForDecl(); |
| + if (IsAffected(*type)) { |
| + std::printf("%s\n", name.c_str()); |
| + } |
| + } |
| + } else if (const clang::TypedefDecl* td = |
| + dyn_cast<clang::TypedefDecl>(type_decl)) { |
| + clang::Type* type = td->getUnderlyingType().getTypePtr(); |
| + if (IsAffected(*type)) { |
| + std::printf("%s\n", name.c_str()); |
| + } |
| + } |
| + } |
| + } |
| + } |
| +}; |
| + |
| +class FindAffectedInterfacesAction : public clang::PluginASTAction { |
| + public: |
| + FindAffectedInterfacesAction() {} |
| + |
| + private: |
| + StringVec types_; |
| + |
| + virtual clang::ASTConsumer *CreateASTConsumer( |
| + clang::CompilerInstance &instance, llvm::StringRef /*input_file*/) { |
| + return new FindAffectedInterfacesConsumer(types_); |
| + } |
| + |
| + virtual bool ParseArgs(const clang::CompilerInstance& /*instance*/, |
| + const std::vector<std::string>& args) { |
| + // Every argument is interpreted as a comma-delimited list of names of types |
| + // that have been changed. |
| + StringVec::const_iterator iter(args.begin()), end(args.end()); |
| + for (; iter != end; ++iter) { |
| + std::stringstream stream(*iter); |
| + std::string type_name; |
| + while (std::getline(stream, type_name, ',')) { |
| + types_.push_back(type_name); |
| + } |
| + } |
| + return true; |
| + } |
| +}; |
| + |
| +} // namespace |
| + |
| +static clang::FrontendPluginRegistry::Add<FindAffectedInterfacesAction> |
| +X("FindAffectedInterfaces", |
| + "Find interfaces affected by changes to the passes classes."); |
| + |
| Property changes on: ppapi/tests/clang/find_affected_interfaces.cc |
| ___________________________________________________________________ |
| Added: svn:eol-style |
| + LF |