UnrealEnginePython/Source/PythonEditor/Private/SPythonEditor.cpp at 64fa655e46bbb631183f3595b9d8b11783f7aaf5 · 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
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
// Copyright 1998-2016 Epic Games, Inc. All Rights Reserved.

#include "SPythonEditor.h"
#include "PythonEditorPrivatePCH.h"
#include "SMultiLineEditableText.h"
#include "PYRichTextSyntaxHighlighterTextLayoutMarshaller.h"
#include "SPythonEditableText.h"


#define LOCTEXT_NAMESPACE "PythonEditor"

void SPythonEditor::Construct(const FArguments& InArgs, UPythonProjectItem* InPythonProjectItem)
{
	bDirty = false;

	check(InPythonProjectItem);
	PythonProjectItem = InPythonProjectItem;

	FString FileText = "File Loading, please wait";
	FFileHelper::LoadFileToString(FileText, *InPythonProjectItem->Path);

	TSharedRef<FPYRichTextSyntaxHighlighterTextLayoutMarshaller> RichTextMarshaller = FPYRichTextSyntaxHighlighterTextLayoutMarshaller::Create(
			FPYRichTextSyntaxHighlighterTextLayoutMarshaller::FSyntaxTextStyle()
			);

	HorizontalScrollbar =
		SNew(SScrollBar)
		.Orientation(Orient_Horizontal)
		.Thickness(FVector2D(10.0f, 10.0f));

	VerticalScrollbar =
		SNew(SScrollBar)
		.Orientation(Orient_Vertical)
		.Thickness(FVector2D(10.0f, 10.0f));

	ChildSlot
	[
		SNew(SBorder)
		.BorderImage(FPythonEditorStyle::Get().GetBrush("TextEditor.Border"))
		[
			SNew(SGridPanel)
			.FillColumn(0, 1.0f)
			.FillRow(0, 2.0f)
			+SGridPanel::Slot(0, 0)
			[
				SAssignNew(PythonEditableText, SPythonEditableText)
				.Text(FText::FromString(FileText))
				.Marshaller(RichTextMarshaller)
				.HScrollBar(HorizontalScrollbar)
				.VScrollBar(VerticalScrollbar)
				.OnTextChanged(this, &SPythonEditor::OnTextChanged)
				.OnExecuted(this,&SPythonEditor::Execute)
			]
			+SGridPanel::Slot(1, 0)
			[
				VerticalScrollbar.ToSharedRef()
			]
			+SGridPanel::Slot(0, 1)
			[
				HorizontalScrollbar.ToSharedRef()
			]
			+ SGridPanel::Slot(0, 2)
			[
				SNew(SBorder).HAlign(EHorizontalAlignment::HAlign_Right)
				[
					SNew(STextBlock)
					.Text(this, &SPythonEditor::GetLineAndColumn)
				]
			]
		]

	];
}

void SPythonEditor::OnTextChanged(const FText& NewText)
{
	bDirty = true;
}

bool SPythonEditor::Save() const
{
	if(bDirty)
	{
		bool bResult = FFileHelper::SaveStringToFile(PythonEditableText->GetText().ToString(), *PythonProjectItem->Path);
		if(bResult)
		{
			bDirty = false;
		}
		return bResult;
	}
	return true;
}

bool SPythonEditor::CanSave() const
{
	return bDirty;
}

void SPythonEditor::Execute() const
{
	Save();
	FUnrealEnginePythonModule &PythonModule = FModuleManager::GetModuleChecked<FUnrealEnginePythonModule>("UnrealEnginePython");

	FString SelectionString = PythonEditableText->GetSelectedText().ToString();
	if (SelectionString.Len() == 0) {
		SelectionString = PythonEditableText->GetText().ToString();
	}
	PythonModule.RunString(TCHAR_TO_UTF8(*SelectionString));
}

void SPythonEditor::ExecuteInSandbox() const
{
	Save();
	FUnrealEnginePythonModule &PythonModule = FModuleManager::GetModuleChecked<FUnrealEnginePythonModule>("UnrealEnginePython");

	FString SelectionString = PythonEditableText->GetSelectedText().ToString();
	if (SelectionString.Len() == 0) {
		SelectionString = PythonEditableText->GetText().ToString();
	}
	PythonModule.RunStringSandboxed(TCHAR_TO_UTF8(*SelectionString));
}

void SPythonEditor::PEP8ize() const
{
	Save();
	FUnrealEnginePythonModule &PythonModule = FModuleManager::GetModuleChecked<FUnrealEnginePythonModule>("UnrealEnginePython");

	FString CleanedCode = PythonModule.Pep8ize(PythonEditableText->GetText().ToString());

	PythonEditableText->SetText(FText::FromString(CleanedCode));
}


void SPythonEditor::GotoLineAndColumn(int32 LineNumber, int32 ColumnNumber)
{
	FTextLocation Location(LineNumber, ColumnNumber);
	PythonEditableText->GoTo(Location);
	PythonEditableText->ScrollTo(Location);
}

FText SPythonEditor::GetLineAndColumn() const
{
	int32 Line;
	int32 Column;
	PythonEditableText->GetLineAndColumn(Line, Column);

	FString LineAndColumn = FString::Printf(TEXT("Line: %d Column: %d"), Line + 1, Column);

	return FText::FromString(LineAndColumn);
}

#undef LOCTEXT_NAMESPACE