tools/c-index-test/c-index-test.c - platform/external/clang - Gitiles

 /* c-index-test.c */

 #include "clang-c/Index.h"
 #include <stdlib.h>
 #include <stdio.h>
 #include <string.h>
 #include <assert.h>

 #ifdef _MSC_VER
 char *basename(const char* path)
 {
     char* base1 = (char*)strrchr(path, '/');
     char* base2 = (char*)strrchr(path, '\\');
     if (base1 && base2)
         return((base1 > base2) ? base1 + 1 : base2 + 1);
     else if (base1)
         return(base1 + 1);
     else if (base2)
         return(base2 + 1);

     return((char*)path);
 }
 #else
 extern char *basename(const char *);
 #endif

 static void PrintCursor(CXCursor Cursor) {
   if (clang_isInvalid(Cursor.kind))
     printf("Invalid Cursor => %s\n", clang_getCursorKindSpelling(Cursor.kind));
   else {
     CXDecl DeclReferenced;
     printf("%s=%s", clang_getCursorKindSpelling(Cursor.kind),
                       clang_getCursorSpelling(Cursor));
     DeclReferenced = clang_getCursorDecl(Cursor);
     if (DeclReferenced)
       printf(":%d:%d", clang_getDeclLine(DeclReferenced),
                        clang_getDeclColumn(DeclReferenced));
   }
 }

 static void DeclVisitor(CXDecl Dcl, CXCursor Cursor, CXClientData Filter)
 {
   if (!Filter || (Cursor.kind == *(enum CXCursorKind *)Filter)) {
     printf("// CHECK: %s:%d:%d: ", basename(clang_getCursorSource(Cursor)),
                                  clang_getCursorLine(Cursor),
                                  clang_getCursorColumn(Cursor));
     PrintCursor(Cursor);
     printf(" [Context=%s]\n", clang_getDeclSpelling(Dcl));
   }
 }
 static void TranslationUnitVisitor(CXTranslationUnit Unit, CXCursor Cursor,
                                    CXClientData Filter)
 {
   if (!Filter || (Cursor.kind == *(enum CXCursorKind *)Filter)) {
     printf("// CHECK: %s:%d:%d: ", basename(clang_getCursorSource(Cursor)),
                                  clang_getCursorLine(Cursor),
                                  clang_getCursorColumn(Cursor));
     PrintCursor(Cursor);
     printf(" [Context=%s]\n", basename(clang_getTranslationUnitSpelling(Unit)));

     clang_loadDeclaration(Cursor.decl, DeclVisitor, 0);

     if (Cursor.kind == CXCursor_FunctionDefn) {
       const char *startBuf, *endBuf;
       unsigned startLine, startColumn, endLine, endColumn;
       clang_getDefinitionSpellingAndExtent(Cursor, &startBuf, &endBuf,
                                            &startLine, &startColumn,
                                            &endLine, &endColumn);
       {
         /* Probe the entire body, looking for both decls and refs. */
         unsigned curLine = startLine, curColumn = startColumn;
         CXCursor Ref;

         while (startBuf < endBuf) {
           if (*startBuf == '\n') {
             startBuf++;
             curLine++;
             curColumn = 1;
           } else if (*startBuf != '\t')
             curColumn++;

           Ref = clang_getCursor(Unit, clang_getCursorSource(Cursor),
                                 curLine, curColumn);
           if (Ref.kind == CXCursor_NoDeclFound) {
             /* Nothing found here; that's fine. */
           } else if (Ref.kind != CXCursor_FunctionDecl) {
             printf("// CHECK: %s:%d:%d: ", basename(clang_getCursorSource(Ref)),
                                              curLine, curColumn);
             PrintCursor(Ref);
 			printf(" [Context:%s]\n", clang_getDeclSpelling(Ref.decl));
           }
           startBuf++;
         }
       }
     }
   }
 }

 /* Parse file:line:column from the input string. Returns 0 on success, non-zero
    on failure. If successful, the pointer *filename will contain newly-allocated
    memory (that will be owned by the caller) to store the file name. */
 int parse_file_line_column(const char *input, char **filename, unsigned *line,
                            unsigned *column) {
   const char *colon = strchr(input, ':');
   char *endptr = 0;
   if (!colon) {
     fprintf(stderr, "could not parse filename:line:column in '%s'\n", input);
     return 1;
   }

   /* Copy the file name. */
   *filename = (char*)malloc(colon - input);
   strncpy(*filename, input, colon - input);
   (*filename)[colon - input] = 0;
   input = colon + 1;

   /* Parse the line number. */
   *line = strtol(input, &endptr, 10);
   if (*endptr != ':') {
     fprintf(stderr, "could not parse line:column in '%s'\n", input);
     free(filename);
     *filename = 0;
     return 1;
   }
   input = endptr + 1;

   /* Parse the column number. */
   *column = strtol(input, &endptr, 10);
   if (*endptr != 0) {
     fprintf(stderr, "could not parse column in '%s'\n", input);
     free(filename);
     *filename = 0;
     return 1;
   }

   return 0;
 }

 const char *
 clang_getCompletionChunkKindSpelling(enum CXCompletionChunkKind Kind) {
   switch (Kind) {
   case CXCompletionChunk_Optional: return "Optional";
   case CXCompletionChunk_TypedText: return "TypedText";
   case CXCompletionChunk_Text: return "Text";
   case CXCompletionChunk_Placeholder: return "Placeholder";
   case CXCompletionChunk_Informative: return "Informative";
   case CXCompletionChunk_CurrentParameter: return "CurrentParameter";
   case CXCompletionChunk_LeftParen: return "LeftParen";
   case CXCompletionChunk_RightParen: return "RightParen";
   case CXCompletionChunk_LeftBracket: return "LeftBracket";
   case CXCompletionChunk_RightBracket: return "RightBracket";
   case CXCompletionChunk_LeftBrace: return "LeftBrace";
   case CXCompletionChunk_RightBrace: return "RightBrace";
   case CXCompletionChunk_LeftAngle: return "LeftAngle";
   case CXCompletionChunk_RightAngle: return "RightAngle";
   case CXCompletionChunk_Comma: return "Comma";
   }

   return "Unknown";
 }

 void print_completion_result(CXCompletionResult *completion_result,
                              CXClientData client_data) {
   FILE *file = (FILE *)client_data;
   int I, N;

   fprintf(file, "%s:",
           clang_getCursorKindSpelling(completion_result->CursorKind));
   N = clang_getNumCompletionChunks(completion_result->CompletionString);
   for (I = 0; I != N; ++I) {
     const char *text
       = clang_getCompletionChunkText(completion_result->CompletionString, I);

     enum CXCompletionChunkKind Kind
       = clang_getCompletionChunkKind(completion_result->CompletionString, I);
     fprintf(file, "{%s %s}",
             clang_getCompletionChunkKindSpelling(Kind),
             text? text : "");
   }
   fprintf(file, "\n");
 }

 void perform_code_completion(int argc, const char **argv) {
   const char *input = argv[1];
   char *filename = 0;
   unsigned line;
   unsigned column;
   CXIndex CIdx;

   input += strlen("-code-completion-at=");
   if (parse_file_line_column(input, &filename, &line, &column))
     return;

   CIdx = clang_createIndex(0, 0);
   clang_codeComplete(CIdx, argv[argc - 1], argc - 3, argv + 2,
                      filename, line, column, &print_completion_result, stdout);
   clang_disposeIndex(CIdx);
   free(filename);
 }

 /*
  * First sign of life:-)
  */
 int main(int argc, char **argv) {
   if (argc > 2 && strstr(argv[1], "-code-completion-at=") == argv[1]) {
     perform_code_completion(argc, (const char **)argv);
     return 0;
   }


   if (argc != 3) {
     printf("Incorrect usage of c-index-test (requires 3 arguments)\n");
     return 0;
   }
   {
   CXIndex Idx;
   CXTranslationUnit TU;
   enum CXCursorKind K = CXCursor_NotImplemented;

   Idx = clang_createIndex(/* excludeDeclsFromPCH */ !strcmp(argv[2], "local") ? 1 : 0,
                           /* displayDiagnostics */ 1);

   TU = clang_createTranslationUnit(Idx, argv[1]);

   if (!TU) {
     fprintf(stderr, "Unable to load translation unit!\n");
     return 1;
   }

   if (!strcmp(argv[2], "all") || !strcmp(argv[2], "local")) {
     clang_loadTranslationUnit(TU, TranslationUnitVisitor, 0);
     clang_disposeTranslationUnit(TU);
     return 1;
   }
   /* Perform some simple filtering. */
   if (!strcmp(argv[2], "category")) K = CXCursor_ObjCCategoryDecl;
   else if (!strcmp(argv[2], "interface")) K = CXCursor_ObjCInterfaceDecl;
   else if (!strcmp(argv[2], "protocol")) K = CXCursor_ObjCProtocolDecl;
   else if (!strcmp(argv[2], "function")) K = CXCursor_FunctionDecl;
   else if (!strcmp(argv[2], "typedef")) K = CXCursor_TypedefDecl;

   clang_loadTranslationUnit(TU, TranslationUnitVisitor, &K);
   clang_disposeTranslationUnit(TU);
   return 1;
   }
 }
	/* c-index-test.c */

	#include "clang-c/Index.h"
	#include <stdlib.h>
	#include <stdio.h>
	#include <string.h>
	#include <assert.h>

	#ifdef _MSC_VER
	char basename(const char path)
	{
	char* base1 = (char*)strrchr(path, '/');
	char* base2 = (char*)strrchr(path, '\\');
	if (base1 && base2)
	return((base1 > base2) ? base1 + 1 : base2 + 1);
	else if (base1)
	return(base1 + 1);
	else if (base2)
	return(base2 + 1);

	return((char*)path);
	}
	#else
	extern char basename(const char );
	#endif

	static void PrintCursor(CXCursor Cursor) {
	if (clang_isInvalid(Cursor.kind))
	printf("Invalid Cursor => %s\n", clang_getCursorKindSpelling(Cursor.kind));
	else {
	CXDecl DeclReferenced;
	printf("%s=%s", clang_getCursorKindSpelling(Cursor.kind),
	clang_getCursorSpelling(Cursor));
	DeclReferenced = clang_getCursorDecl(Cursor);
	if (DeclReferenced)
	printf(":%d:%d", clang_getDeclLine(DeclReferenced),
	clang_getDeclColumn(DeclReferenced));
	}
	}

	static void DeclVisitor(CXDecl Dcl, CXCursor Cursor, CXClientData Filter)
	{
	if (!Filter \|\| (Cursor.kind == (enum CXCursorKind )Filter)) {
	printf("// CHECK: %s:%d:%d: ", basename(clang_getCursorSource(Cursor)),
	clang_getCursorLine(Cursor),
	clang_getCursorColumn(Cursor));
	PrintCursor(Cursor);
	printf(" [Context=%s]\n", clang_getDeclSpelling(Dcl));
	}
	}
	static void TranslationUnitVisitor(CXTranslationUnit Unit, CXCursor Cursor,
	CXClientData Filter)
	{
	if (!Filter \|\| (Cursor.kind == (enum CXCursorKind )Filter)) {
	printf("// CHECK: %s:%d:%d: ", basename(clang_getCursorSource(Cursor)),
	clang_getCursorLine(Cursor),
	clang_getCursorColumn(Cursor));
	PrintCursor(Cursor);
	printf(" [Context=%s]\n", basename(clang_getTranslationUnitSpelling(Unit)));

	clang_loadDeclaration(Cursor.decl, DeclVisitor, 0);

	if (Cursor.kind == CXCursor_FunctionDefn) {
	const char startBuf, endBuf;
	unsigned startLine, startColumn, endLine, endColumn;
	clang_getDefinitionSpellingAndExtent(Cursor, &startBuf, &endBuf,
	&startLine, &startColumn,
	&endLine, &endColumn);
	{
	/* Probe the entire body, looking for both decls and refs. */
	unsigned curLine = startLine, curColumn = startColumn;
	CXCursor Ref;

	while (startBuf < endBuf) {
	if (*startBuf == '\n') {
	startBuf++;
	curLine++;
	curColumn = 1;
	} else if (*startBuf != '\t')
	curColumn++;

	Ref = clang_getCursor(Unit, clang_getCursorSource(Cursor),
	curLine, curColumn);
	if (Ref.kind == CXCursor_NoDeclFound) {
	/* Nothing found here; that's fine. */
	} else if (Ref.kind != CXCursor_FunctionDecl) {
	printf("// CHECK: %s:%d:%d: ", basename(clang_getCursorSource(Ref)),
	curLine, curColumn);
	PrintCursor(Ref);
	printf(" [Context:%s]\n", clang_getDeclSpelling(Ref.decl));
	}
	startBuf++;
	}
	}
	}
	}
	}

	/* Parse file:line:column from the input string. Returns 0 on success, non-zero
	on failure. If successful, the pointer *filename will contain newly-allocated
	memory (that will be owned by the caller) to store the file name. */
	int parse_file_line_column(const char input, char filename, unsigned line,
	unsigned *column) {
	const char *colon = strchr(input, ':');
	char *endptr = 0;
	if (!colon) {
	fprintf(stderr, "could not parse filename:line:column in '%s'\n", input);
	return 1;
	}

	/* Copy the file name. */
	filename = (char)malloc(colon - input);
	strncpy(*filename, input, colon - input);
	(*filename)[colon - input] = 0;
	input = colon + 1;

	/* Parse the line number. */
	*line = strtol(input, &endptr, 10);
	if (*endptr != ':') {
	fprintf(stderr, "could not parse line:column in '%s'\n", input);
	free(filename);
	*filename = 0;
	return 1;
	}
	input = endptr + 1;

	/* Parse the column number. */
	*column = strtol(input, &endptr, 10);
	if (*endptr != 0) {
	fprintf(stderr, "could not parse column in '%s'\n", input);
	free(filename);
	*filename = 0;
	return 1;
	}

	return 0;
	}

	const char *
	clang_getCompletionChunkKindSpelling(enum CXCompletionChunkKind Kind) {
	switch (Kind) {
	case CXCompletionChunk_Optional: return "Optional";
	case CXCompletionChunk_TypedText: return "TypedText";
	case CXCompletionChunk_Text: return "Text";
	case CXCompletionChunk_Placeholder: return "Placeholder";
	case CXCompletionChunk_Informative: return "Informative";
	case CXCompletionChunk_CurrentParameter: return "CurrentParameter";
	case CXCompletionChunk_LeftParen: return "LeftParen";
	case CXCompletionChunk_RightParen: return "RightParen";
	case CXCompletionChunk_LeftBracket: return "LeftBracket";
	case CXCompletionChunk_RightBracket: return "RightBracket";
	case CXCompletionChunk_LeftBrace: return "LeftBrace";
	case CXCompletionChunk_RightBrace: return "RightBrace";
	case CXCompletionChunk_LeftAngle: return "LeftAngle";
	case CXCompletionChunk_RightAngle: return "RightAngle";
	case CXCompletionChunk_Comma: return "Comma";
	}

	return "Unknown";
	}

	void print_completion_result(CXCompletionResult *completion_result,
	CXClientData client_data) {
	FILE file = (FILE )client_data;
	int I, N;

	fprintf(file, "%s:",
	clang_getCursorKindSpelling(completion_result->CursorKind));
	N = clang_getNumCompletionChunks(completion_result->CompletionString);
	for (I = 0; I != N; ++I) {
	const char *text
	= clang_getCompletionChunkText(completion_result->CompletionString, I);

	enum CXCompletionChunkKind Kind
	= clang_getCompletionChunkKind(completion_result->CompletionString, I);
	fprintf(file, "{%s %s}",
	clang_getCompletionChunkKindSpelling(Kind),
	text? text : "");
	}
	fprintf(file, "\n");
	}

	void perform_code_completion(int argc, const char **argv) {
	const char *input = argv[1];
	char *filename = 0;
	unsigned line;
	unsigned column;
	CXIndex CIdx;

	input += strlen("-code-completion-at=");
	if (parse_file_line_column(input, &filename, &line, &column))
	return;

	CIdx = clang_createIndex(0, 0);
	clang_codeComplete(CIdx, argv[argc - 1], argc - 3, argv + 2,
	filename, line, column, &print_completion_result, stdout);
	clang_disposeIndex(CIdx);
	free(filename);
	}

	/*
	* First sign of life:-)
	*/
	int main(int argc, char **argv) {
	if (argc > 2 && strstr(argv[1], "-code-completion-at=") == argv[1]) {
	perform_code_completion(argc, (const char **)argv);
	return 0;
	}


	if (argc != 3) {
	printf("Incorrect usage of c-index-test (requires 3 arguments)\n");
	return 0;
	}
	{
	CXIndex Idx;
	CXTranslationUnit TU;
	enum CXCursorKind K = CXCursor_NotImplemented;

	Idx = clang_createIndex(/* excludeDeclsFromPCH */ !strcmp(argv[2], "local") ? 1 : 0,
	/* displayDiagnostics */ 1);

	TU = clang_createTranslationUnit(Idx, argv[1]);

	if (!TU) {
	fprintf(stderr, "Unable to load translation unit!\n");
	return 1;
	}

	if (!strcmp(argv[2], "all") \|\| !strcmp(argv[2], "local")) {
	clang_loadTranslationUnit(TU, TranslationUnitVisitor, 0);
	clang_disposeTranslationUnit(TU);
	return 1;
	}
	/* Perform some simple filtering. */
	if (!strcmp(argv[2], "category")) K = CXCursor_ObjCCategoryDecl;
	else if (!strcmp(argv[2], "interface")) K = CXCursor_ObjCInterfaceDecl;
	else if (!strcmp(argv[2], "protocol")) K = CXCursor_ObjCProtocolDecl;
	else if (!strcmp(argv[2], "function")) K = CXCursor_FunctionDecl;
	else if (!strcmp(argv[2], "typedef")) K = CXCursor_TypedefDecl;

	clang_loadTranslationUnit(TU, TranslationUnitVisitor, &K);
	clang_disposeTranslationUnit(TU);
	return 1;
	}
	}