cb-UnitTestPlugin/OutputReader.cpp at main · josephch/cb-UnitTestPlugin · 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
#include "OutputReader.h"
#include "UnitTest.h"
#include "UnitTestProcess.h"

#include <iostream>

using namespace std;

OutputReader::OutputReader(UnitTest *parent) : m_Parent(parent)
{
}

OutputReader::~OutputReader()
{
}

void OutputReader::Start()
{
	outString = _("");
	errString = _("");

	wxTimer::Start(10, false);
}

void OutputReader::Notify()
{
	ReadOutput();
}

void OutputReader::ReadOutput()
{
	UnitTestProcess *process = m_Parent->GetProcess();

	if (process != nullptr)
	{
		wxInputStream *outputStream = process->GetInputStream();
		wxInputStream *errorStream = process->GetErrorStream();

		// read
		if (outputStream != nullptr)
		{
			while (process->IsInputAvailable() == true)
			{
				if (outputStream->Eof() == true)
				{
					SendLine(outString);
					outString = _("");
					break;
				}

				wxChar c = outputStream->GetC();

				if (c == '\n')
				{
					SendLine(outString);
					outString = _("");
				}
				else
				{
					outString += c;
				}
			}
		}

		if (errorStream != nullptr)
		{
			while (process->IsErrorAvailable() == true)
			{
				if (errorStream->Eof() == true)
				{
					SendLine(errString);
					errString = _("");
					break;
				}

				wxChar c = errorStream->GetC();

				if (c == '\n')
				{
					SendLine(errString);
					errString = _("");
				}
				else
				{
					errString += c;
				}
			}
		}
	}
}

void OutputReader::SendLine(const wxString &line)
{
	m_Parent->OnOutput(line);
}