UnsavedFiles.cpp

#include "UnsavedFiles.hpp"
#include "CompileOptions.hpp"
#include "Styles.hpp"
#include "GlobalVariables.hpp"

UnsavedFiles UnsavedFiles::current;

static unsigned GetStyle(CXTranslationUnit unit, CXToken t, CXCursor c)
{
    CXString spelling_ = clang_getTokenSpelling(unit, t);
    wxString spelling = wxString::FromUTF8(clang_getCString(spelling_));
    clang_disposeString(spelling_);

    if (c.kind >= 500 && clang_getTokenKind(t) != CXToken_Comment)
        return stPREPROCESSOR;

    switch (clang_getTokenKind(t))
    {
    case CXToken_Punctuation:
        return stPUNCTUATION;
    case CXToken_Keyword:
        if (typeKeywords.Index(spelling) != wxNOT_FOUND)
            return stTYPE;
        return stKEYWORD;
    case CXToken_Identifier:
        if (clang_isReference(c.kind) || c.kind == CXCursor_DeclRefExpr || c.kind == CXCursor_MemberRefExpr)
        {
            CXCursor c_ref = clang_getCursorReferenced(c);
            return identifierStyle[c_ref.kind];
        }
//        fprintf(stderr, "%s, %d\n", clang_getCString(spelling_), c.kind);
        return identifierStyle[c.kind];
    case CXToken_Literal:
        return stCONSTANT;
    case CXToken_Comment:
        return stCOMMENT;
    }

    return stINVALID;
}

void UnsavedFiles::PrepareSHData(cbEditor * editor, CXTranslationUnit unit, std::vector<char> & styles, std::vector<Diagnostic> & diagnostics) const
{
    UnsavedFileData const& info = ufMap.find(editor)->second;

    CXFile file = clang_getFile(unit, info.filename.get());

    CXSourceLocation bl = clang_getLocationForOffset(unit, file, 0);
    CXSourceLocation el = clang_getLocationForOffset(unit, file, (unsigned)info.length);

    CXToken * tokens;
    unsigned num_tokens;
    clang_tokenize(unit, clang_getRange(bl, el), &tokens, &num_tokens);

    CXCursor * cursors = new CXCursor[num_tokens];
    clang_annotateTokens(unit, tokens, num_tokens, cursors);

    styles.resize(info.length);
    unsigned j = 0;
    // for each token:
    for (unsigned i = 0; i < num_tokens; ++i)
    {
//        fprintf(stderr, "%s\n", clang_getCString(clang_getTokenSpelling(unit, tokens[i])));
        CXSourceRange range = clang_getTokenExtent(unit, tokens[i]);
        unsigned a, b;
        clang_getSpellingLocation(clang_getRangeStart(range), 0, 0, 0, &a);
        clang_getSpellingLocation(clang_getRangeEnd(range), 0, 0, 0, &b);

        for (; j < a; ++j)
            styles[j] = stDEFAULT;
        unsigned style = GetStyle(unit, tokens[i], cursors[i]);
        for (; j < b; ++j)
            styles[j] = style;
    }

    for (; j < info.length; ++j)
        styles[j] = stDEFAULT;

    diagnostics.clear();
	for (unsigned i = 0, n = clang_getNumDiagnostics(unit); i < n; ++i)
	{
		CXDiagnostic diag = clang_getDiagnostic(unit, i);
		CXDiagnosticSeverity kind = clang_getDiagnosticSeverity(diag);
		if (kind != CXDiagnostic_Fatal && kind != CXDiagnostic_Error && kind != CXDiagnostic_Warning)
            continue;

        Diagnostic d;
        d.kind = kind;
        clang_getSpellingLocation(clang_getDiagnosticLocation(diag), 0, 0, 0, &d.offset);
		CXString diag_cxstr = clang_formatDiagnostic(diag, clang_defaultDiagnosticDisplayOptions());
        d.text = wxString::FromUTF8(clang_getCString(diag_cxstr));

        diagnostics.push_back(d);

		clang_disposeString(diag_cxstr);
		clang_disposeDiagnostic(diag);
	}

    // cleanup
    clang_disposeTokens(unit, tokens, num_tokens);
    delete [] cursors;
}

std::vector<CXUnsavedFile> UnsavedFiles::GetUnsavedFiles() const
{
    std::vector<CXUnsavedFile> data;
    data.reserve(ufMap.size());

    for (auto it = ufMap.begin(); it != ufMap.end(); ++it)
    {
        CXUnsavedFile uf;
        uf.Filename = it->second.filename.get(); // safe
        uf.Contents = it->second.contents.get(); // safe
        uf.Length = it->second.length;
        data.push_back(uf);
    }

    return data;
}

void UnsavedFiles::Print() const
{
    fprintf(stderr, "==== BEGIN UNSAVED FILES ====\n");
    for (auto it = ufMap.begin(); it != ufMap.end(); ++it)
    {
        fprintf(stderr, "EDITOR: %p\n", it->first);
        fprintf(stderr, "\tFILENAME: %s\n", it->second.filename.get());
        fprintf(stderr, "\tLENGTH: %lu\n", it->second.length);
        fprintf(stderr, "\tVERSION: %u\n", it->second.version);
        fprintf(stderr, "\tPROJECT: %p\n", it->second.project);
    }
    fprintf(stderr, "===== END UNSAVED FILES =====\n");
}