UnrealEnginePython/Source/PythonEditor/Private/PYRichTextSyntaxHighlighterTextLayoutMarshaller.h at 4b5da5bf4ca598e8d0f67eb938749bc381a67d6c · 20tab/UnrealEnginePython · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
// Copyright 1998-2016 Epic Games, Inc. All Rights Reserved.
#pragma once

#include "Runtime/SlateCore/Public/Styling/SlateTypes.h"
#include "PythonSyntaxTokenizer.h"
#include "Runtime/Slate/Public/Framework/Text/PlainTextLayoutMarshaller.h"
#include "Runtime/Slate/Public/Framework/Text/ITextDecorator.h"
#include "PythonEditorStyle.h"


class  FPYRichTextSyntaxHighlighterTextLayoutMarshaller : public FPlainTextLayoutMarshaller
{
public:

	struct FSyntaxTextStyle
	{
		FSyntaxTextStyle()
			: NormalTextStyle(FPythonEditorStyle::Get().GetWidgetStyle<FTextBlockStyle>("SyntaxHighlight.PY.Normal"))
			, OperatorTextStyle(FPythonEditorStyle::Get().GetWidgetStyle<FTextBlockStyle>("SyntaxHighlight.PY.Operator"))
			, KeywordTextStyle(FPythonEditorStyle::Get().GetWidgetStyle<FTextBlockStyle>("SyntaxHighlight.PY.Keyword"))
			, StringTextStyle(FPythonEditorStyle::Get().GetWidgetStyle<FTextBlockStyle>("SyntaxHighlight.PY.String"))
			, NumberTextStyle(FPythonEditorStyle::Get().GetWidgetStyle<FTextBlockStyle>("SyntaxHighlight.PY.Number"))
			, CommentTextStyle(FPythonEditorStyle::Get().GetWidgetStyle<FTextBlockStyle>("SyntaxHighlight.PY.Comment"))
			, BuiltInKeywordTextStyle(FPythonEditorStyle::Get().GetWidgetStyle<FTextBlockStyle>("SyntaxHighlight.PY.BuiltIn"))
			, DefineTextStyle(FPythonEditorStyle::Get().GetWidgetStyle<FTextBlockStyle>("SyntaxHighlight.PY.Define"))

		{
		}

		FSyntaxTextStyle(const FTextBlockStyle& InNormalTextStyle, const FTextBlockStyle& InOperatorTextStyle, const FTextBlockStyle& InKeywordTextStyle, const FTextBlockStyle& InStringTextStyle, const FTextBlockStyle& InNumberTextStyle, const FTextBlockStyle& InCommentTextStyle, const FTextBlockStyle& InBuiltInKeywordTextStyle, const FTextBlockStyle& InDefineTextStyle)
			: NormalTextStyle(InNormalTextStyle)
			, OperatorTextStyle(InOperatorTextStyle)
			, KeywordTextStyle(InKeywordTextStyle)
			, StringTextStyle(InStringTextStyle)
			, NumberTextStyle(InNumberTextStyle)
			, CommentTextStyle(InCommentTextStyle)
			, BuiltInKeywordTextStyle(InBuiltInKeywordTextStyle)
			, DefineTextStyle(InDefineTextStyle)
		{
		}

		FTextBlockStyle NormalTextStyle;
		FTextBlockStyle OperatorTextStyle;
		FTextBlockStyle KeywordTextStyle;
		FTextBlockStyle StringTextStyle;
		FTextBlockStyle NumberTextStyle;
		FTextBlockStyle CommentTextStyle;
		FTextBlockStyle BuiltInKeywordTextStyle;
		FTextBlockStyle DefineTextStyle;
	};

	static TSharedRef< FPYRichTextSyntaxHighlighterTextLayoutMarshaller > Create(const FSyntaxTextStyle& InSyntaxTextStyle);

public:

	virtual ~FPYRichTextSyntaxHighlighterTextLayoutMarshaller();

	// ITextLayoutMarshaller
	virtual void SetText(const FString& SourceString, FTextLayout& TargetTextLayout) override;
	virtual bool RequiresLiveUpdate() const override;

	void EnableSyntaxHighlighting(const bool bEnable);
	bool IsSyntaxHighlightingEnabled() const;

protected:

	void ParseTokens(const FString& SourceString, FTextLayout& TargetTextLayout, TArray<FPythonSyntaxTokenizer::FTokenizedLine> TokenizedLines);


	/** Tokenizer used to style the text */
	TSharedPtr< FPythonSyntaxTokenizer > Tokenizer;

	/** True if syntax highlighting is enabled, false to fallback to plain text */
	bool bSyntaxHighlightingEnabled;

	FPYRichTextSyntaxHighlighterTextLayoutMarshaller(TSharedPtr< FPythonSyntaxTokenizer > InTokenizer, const FSyntaxTextStyle& InSyntaxTextStyle);

	/** Styles used to display the text */
	FSyntaxTextStyle SyntaxTextStyle;

	/** String representing tabs */
	FString TabString;

};