Initial version of the ClangComplete class
This commit is contained in:
Strahinja Val Markovic
parent
fb5d25fc81
commit
9a82319c77
233
cpp/ycm/ClangComplete.cpp
Normal file
233
cpp/ycm/ClangComplete.cpp
Normal file
@ -0,0 +1,233 @@
|
|||||||
|
// Copyright (C) 2011, 2012 Strahinja Val Markovic <val@markovic.io>
|
||||||
|
//
|
||||||
|
// This file is part of YouCompleteMe.
|
||||||
|
//
|
||||||
|
// YouCompleteMe is free software: you can redistribute it and/or modify
|
||||||
|
// it under the terms of the GNU General Public License as published by
|
||||||
|
// the Free Software Foundation, either version 3 of the License, or
|
||||||
|
// (at your option) any later version.
|
||||||
|
//
|
||||||
|
// YouCompleteMe is distributed in the hope that it will be useful,
|
||||||
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
// GNU General Public License for more details.
|
||||||
|
//
|
||||||
|
// You should have received a copy of the GNU General Public License
|
||||||
|
// along with YouCompleteMe. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
|
#include "ClangComplete.h"
|
||||||
|
#include "standard.h"
|
||||||
|
#include <clang-c/Index.h>
|
||||||
|
|
||||||
|
namespace YouCompleteMe
|
||||||
|
{
|
||||||
|
|
||||||
|
namespace
|
||||||
|
{
|
||||||
|
|
||||||
|
std::vector< CXUnsavedFile > ToCXUnsavedFiles(
|
||||||
|
const std::vector< UnsavedFile > &unsaved_files )
|
||||||
|
{
|
||||||
|
std::vector< CXUnsavedFile > clang_unsaved_files( unsaved_files.size() );
|
||||||
|
for ( uint i = 0; i < unsaved_files.size(); ++i )
|
||||||
|
{
|
||||||
|
clang_unsaved_files[ i ].Filename = unsaved_files[ i ].filename_->c_str();
|
||||||
|
clang_unsaved_files[ i ].Contents = unsaved_files[ i ].contents_->c_str();
|
||||||
|
clang_unsaved_files[ i ].Length = unsaved_files[ i ].contents_->length();
|
||||||
|
}
|
||||||
|
|
||||||
|
return clang_unsaved_files;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
std::string CXStringToString( CXString text )
|
||||||
|
{
|
||||||
|
std::string final_string( clang_getCString( text ) );
|
||||||
|
clang_disposeString( text );
|
||||||
|
return final_string;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
std::string ChunkToString( CXCompletionString completion_string, int chunk_num )
|
||||||
|
{
|
||||||
|
return CXStringToString(
|
||||||
|
clang_getCompletionChunkText( completion_string, chunk_num ) );
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
std::vector< std::string > ToStringVector( CXCodeCompleteResults *results )
|
||||||
|
{
|
||||||
|
std::vector< std::string > completions;
|
||||||
|
completions.reserve( results->NumResults );
|
||||||
|
|
||||||
|
for ( uint i = 0; i < results->NumResults; ++i )
|
||||||
|
{
|
||||||
|
CXCompletionString completion_string =
|
||||||
|
results->Results[ i ].CompletionString;
|
||||||
|
|
||||||
|
uint num_chunks = clang_getNumCompletionChunks( completion_string );
|
||||||
|
for ( uint j = 0; j < num_chunks; ++j )
|
||||||
|
{
|
||||||
|
CXCompletionChunkKind kind = clang_getCompletionChunkKind(
|
||||||
|
completion_string, j );
|
||||||
|
|
||||||
|
if ( kind == CXCompletionChunk_TypedText )
|
||||||
|
{
|
||||||
|
completions.push_back( ChunkToString( completion_string, j ) );
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return completions;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
} // unnamed namespace
|
||||||
|
|
||||||
|
|
||||||
|
ClangComplete::ClangComplete()
|
||||||
|
{
|
||||||
|
clang_index_ = clang_createIndex( 0, 0 );
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
ClangComplete::~ClangComplete()
|
||||||
|
{
|
||||||
|
foreach ( const TranslationUnitForFilename::value_type &filename_unit,
|
||||||
|
filename_to_translation_unit_ )
|
||||||
|
{
|
||||||
|
clang_disposeTranslationUnit( filename_unit.second );
|
||||||
|
}
|
||||||
|
|
||||||
|
clang_disposeIndex( clang_index_ );
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
void ClangComplete::SetGlobalCompileFlags(
|
||||||
|
const std::vector< std::string > &flags )
|
||||||
|
{
|
||||||
|
global_flags_ = flags;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
void ClangComplete::SetFileCompileFlags(
|
||||||
|
const std::string &filename,
|
||||||
|
const std::vector< std::string > &flags )
|
||||||
|
{
|
||||||
|
flags_for_file_[ filename ] = flags;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
void ClangComplete::UpdateTranslationUnit(
|
||||||
|
const std::string &filename,
|
||||||
|
const std::vector< UnsavedFile > &unsaved_files )
|
||||||
|
{
|
||||||
|
TranslationUnitForFilename::iterator it =
|
||||||
|
filename_to_translation_unit_.find( filename );
|
||||||
|
|
||||||
|
if ( it != filename_to_translation_unit_.end() )
|
||||||
|
{
|
||||||
|
std::vector< CXUnsavedFile > cxunsaved_files = ToCXUnsavedFiles(
|
||||||
|
unsaved_files );
|
||||||
|
|
||||||
|
clang_reparseTranslationUnit(
|
||||||
|
it->second,
|
||||||
|
cxunsaved_files.size(),
|
||||||
|
&cxunsaved_files[ 0 ],
|
||||||
|
clang_defaultEditingTranslationUnitOptions() );
|
||||||
|
}
|
||||||
|
|
||||||
|
else
|
||||||
|
{
|
||||||
|
filename_to_translation_unit_[ filename ] =
|
||||||
|
CreateTranslationUnit( filename, unsaved_files );
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
std::vector< std::string > ClangComplete::CandidatesForLocationInFile(
|
||||||
|
const std::string &filename,
|
||||||
|
int line,
|
||||||
|
int column,
|
||||||
|
const std::vector< UnsavedFile > &unsaved_files )
|
||||||
|
{
|
||||||
|
std::vector< CXUnsavedFile > cxunsaved_files = ToCXUnsavedFiles(
|
||||||
|
unsaved_files );
|
||||||
|
|
||||||
|
// TODO: figure out does codeCompleteAt reparse the TU if the underlying
|
||||||
|
// source file has changed on disk since the last time the TU was updated and
|
||||||
|
// there are no unsaved files.
|
||||||
|
|
||||||
|
CXCodeCompleteResults *results =
|
||||||
|
clang_codeCompleteAt( GetTranslationUnitForFile( filename, unsaved_files ),
|
||||||
|
filename.c_str(),
|
||||||
|
line,
|
||||||
|
column,
|
||||||
|
&cxunsaved_files[ 0 ],
|
||||||
|
cxunsaved_files.size(),
|
||||||
|
clang_defaultCodeCompleteOptions());
|
||||||
|
|
||||||
|
std::vector< std::string > completions = ToStringVector( results );
|
||||||
|
clang_disposeCodeCompleteResults( results );
|
||||||
|
return completions;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
CXTranslationUnit ClangComplete::CreateTranslationUnit(
|
||||||
|
const std::string &filename,
|
||||||
|
const std::vector< UnsavedFile > &unsaved_files )
|
||||||
|
{
|
||||||
|
std::vector< const char* > flags = ClangFlagsForFilename( filename );
|
||||||
|
std::vector< CXUnsavedFile > cxunsaved_files = ToCXUnsavedFiles(
|
||||||
|
unsaved_files );
|
||||||
|
|
||||||
|
return clang_parseTranslationUnit(
|
||||||
|
clang_index_,
|
||||||
|
filename.c_str(),
|
||||||
|
&flags[ 0 ],
|
||||||
|
flags.size(),
|
||||||
|
&cxunsaved_files[ 0 ],
|
||||||
|
cxunsaved_files.size(),
|
||||||
|
clang_defaultEditingTranslationUnitOptions());
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
std::vector< const char* > ClangComplete::ClangFlagsForFilename(
|
||||||
|
const std::string &filename )
|
||||||
|
{
|
||||||
|
std::vector< const char* > flags;
|
||||||
|
|
||||||
|
std::vector< std::string > file_flags = flags_for_file_[ filename ];
|
||||||
|
flags.reserve( file_flags.size() + global_flags_.size() );
|
||||||
|
|
||||||
|
foreach ( const std::string &flag, global_flags_ )
|
||||||
|
{
|
||||||
|
flags.push_back( flag.c_str() );
|
||||||
|
}
|
||||||
|
|
||||||
|
foreach ( const std::string &flag, file_flags )
|
||||||
|
{
|
||||||
|
flags.push_back( flag.c_str() );
|
||||||
|
}
|
||||||
|
|
||||||
|
return flags;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
CXTranslationUnit ClangComplete::GetTranslationUnitForFile(
|
||||||
|
const std::string &filename,
|
||||||
|
const std::vector< UnsavedFile > &unsaved_files )
|
||||||
|
{
|
||||||
|
TranslationUnitForFilename::iterator it =
|
||||||
|
filename_to_translation_unit_.find( filename );
|
||||||
|
|
||||||
|
if ( it != filename_to_translation_unit_.end() )
|
||||||
|
return it->second;
|
||||||
|
|
||||||
|
CXTranslationUnit unit = CreateTranslationUnit( filename, unsaved_files );
|
||||||
|
filename_to_translation_unit_[ filename ] = unit;
|
||||||
|
return unit;
|
||||||
|
}
|
||||||
|
|
||||||
|
} // namespace YouCompleteMe
|
||||||
89
cpp/ycm/ClangComplete.h
Normal file
89
cpp/ycm/ClangComplete.h
Normal file
@ -0,0 +1,89 @@
|
|||||||
|
// Copyright (C) 2011, 2012 Strahinja Val Markovic <val@markovic.io>
|
||||||
|
//
|
||||||
|
// This file is part of YouCompleteMe.
|
||||||
|
//
|
||||||
|
// YouCompleteMe is free software: you can redistribute it and/or modify
|
||||||
|
// it under the terms of the GNU General Public License as published by
|
||||||
|
// the Free Software Foundation, either version 3 of the License, or
|
||||||
|
// (at your option) any later version.
|
||||||
|
//
|
||||||
|
// YouCompleteMe is distributed in the hope that it will be useful,
|
||||||
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
// GNU General Public License for more details.
|
||||||
|
//
|
||||||
|
// You should have received a copy of the GNU General Public License
|
||||||
|
// along with YouCompleteMe. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
|
#ifndef CLANGCOMPLETE_H_WLKDU0ZV
|
||||||
|
#define CLANGCOMPLETE_H_WLKDU0ZV
|
||||||
|
|
||||||
|
#include <boost/utility.hpp>
|
||||||
|
#include <boost/unordered_map.hpp>
|
||||||
|
|
||||||
|
#include <string>
|
||||||
|
|
||||||
|
typedef void *CXIndex;
|
||||||
|
typedef struct CXTranslationUnitImpl *CXTranslationUnit;
|
||||||
|
|
||||||
|
namespace YouCompleteMe
|
||||||
|
{
|
||||||
|
|
||||||
|
struct UnsavedFile
|
||||||
|
{
|
||||||
|
std::string *filename_;
|
||||||
|
std::string *contents_;
|
||||||
|
};
|
||||||
|
|
||||||
|
|
||||||
|
typedef boost::unordered_map< std::string, std::vector< std::string > >
|
||||||
|
FlagsForFile;
|
||||||
|
typedef boost::unordered_map< std::string, CXTranslationUnit >
|
||||||
|
TranslationUnitForFilename;
|
||||||
|
|
||||||
|
|
||||||
|
class ClangComplete : boost::noncopyable
|
||||||
|
{
|
||||||
|
public:
|
||||||
|
ClangComplete();
|
||||||
|
|
||||||
|
~ClangComplete();
|
||||||
|
|
||||||
|
void SetGlobalCompileFlags( const std::vector< std::string > &flags );
|
||||||
|
|
||||||
|
void SetFileCompileFlags( const std::string &filename,
|
||||||
|
const std::vector< std::string > &flags );
|
||||||
|
|
||||||
|
void UpdateTranslationUnit( const std::string &filename,
|
||||||
|
const std::vector< UnsavedFile > &unsaved_files );
|
||||||
|
|
||||||
|
std::vector< std::string > CandidatesForLocationInFile(
|
||||||
|
const std::string &filename,
|
||||||
|
int line,
|
||||||
|
int column,
|
||||||
|
const std::vector< UnsavedFile > &unsaved_files );
|
||||||
|
|
||||||
|
private:
|
||||||
|
|
||||||
|
// caller takes ownership of translation unit
|
||||||
|
CXTranslationUnit CreateTranslationUnit(
|
||||||
|
const std::string &filename,
|
||||||
|
const std::vector< UnsavedFile > &unsaved_files );
|
||||||
|
|
||||||
|
std::vector< const char* > ClangFlagsForFilename(
|
||||||
|
const std::string &filename );
|
||||||
|
|
||||||
|
CXTranslationUnit GetTranslationUnitForFile(
|
||||||
|
const std::string &filename,
|
||||||
|
const std::vector< UnsavedFile > &unsaved_files );
|
||||||
|
|
||||||
|
CXIndex clang_index_;
|
||||||
|
FlagsForFile flags_for_file_;
|
||||||
|
TranslationUnitForFilename filename_to_translation_unit_;
|
||||||
|
std::vector< std::string > global_flags_;
|
||||||
|
|
||||||
|
};
|
||||||
|
|
||||||
|
} // namespace YouCompleteMe
|
||||||
|
|
||||||
|
#endif /* end of include guard: CLANGCOMPLETE_H_WLKDU0ZV */
|
||||||
Loading…
x
Reference in New Issue
Block a user