TeraVTK/Line.cpp at main · Ursus14/TeraVTK · 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
#include "Line.h"

Line::Line(vtkSmartPointer<vtkActor> lineActor) {
	points[0] = Point();
	points[1] = Point();
	points[0].SetColor(0, 0, 0);
	points[1].SetColor(0, 0, 0);
	SetBeginPosition(new double[2]{ 0.0, 0.0 });
	SetEndPosition(new double[2]{ 0.0, 0.0 });
	lineActor_ = lineActor;
};

Line::Line(vtkSmartPointer<vtkActor> lineActor, double* coordinateB, double* coordinateE) {
	points[0] = Point();
	points[1] = Point();

	SetBeginPosition(coordinateB);
	SetEndPosition(coordinateE);

	lineActor_ = lineActor;

};

void Line::SetEndPosition(double* coordinate) {
	endPosition[0] = coordinate[0];
	endPosition[1] = coordinate[1];
}

void Line::SetBeginPosition(double* coordinate) {
	beginPosition[0] = coordinate[0];
	beginPosition[1] = coordinate[1];
}

double* Line::GetBeginPosition() {
	return beginPosition;
}

double* Line::GetEndPosition() {
	return endPosition;
}

double Line::GetLengthSquare() {
	double x = beginPosition[0] - endPosition[0];
	double y = beginPosition[1] - endPosition[1];
	return x * x + y * y;
}

void Line::build(double* coordinateE, vtkSmartPointer<vtkRenderer> renderer, double radius) {
	renderer->AddActor(points[0].GetActor());
	renderer->AddActor(points[1].GetActor());
	vtkSmartPointer<vtkLineSource> lineSource =
		vtkSmartPointer<vtkLineSource>::New();

	lineSource->SetPoint1(beginPosition[0], beginPosition[1], 0.0);
	lineSource->SetPoint2(coordinateE[0], coordinateE[1], 0.0);

	endPosition[0] = coordinateE[0];
	endPosition[1] = coordinateE[1];

	vtkSmartPointer<vtkPolyDataMapper> lineMapper = vtkSmartPointer<vtkPolyDataMapper>::New();

	lineMapper->SetInputConnection(lineSource->GetOutputPort());

	lineActor_->SetMapper(lineMapper);
	lineActor_->GetProperty()->SetColor(0.0, 1.0, 0.0);
	points[0].SetRadius(radius);
	points[1].SetRadius(radius);


	points[0].SetPosition(new double[3]{ beginPosition[0], beginPosition[1], 0.0 });
	points[1].SetPosition(new double[3]{ endPosition[0],endPosition[1], 0.0 });

	renderer->AddActor(lineActor_);
}

void Line::rebuild(double* coordinateE, vtkSmartPointer<vtkRenderer> renderer) {
	vtkSmartPointer<vtkLineSource> lineSource =
		vtkSmartPointer<vtkLineSource>::New();

	lineSource->SetPoint1(beginPosition[0], beginPosition[1], 0.0);
	lineSource->SetPoint2(coordinateE[0], coordinateE[1], 0.0);

	endPosition[0] = coordinateE[0];
	endPosition[1] = coordinateE[1];

	points[1].SetPosition(endPosition);

	vtkPolyDataMapper::SafeDownCast(lineActor_->GetMapper())->SetInputConnection(lineSource->GetOutputPort());
}

void Line::Remove(vtkSmartPointer<vtkRenderer> renderer) {
	renderer->RemoveActor(points[0].GetActor());
	renderer->RemoveActor(points[1].GetActor());
	renderer->RemoveActor(lineActor_);
	points[0].SetPosition(new double[3]{ 0.0, 0.0, 0.0 });
	points[1].SetPosition(new double[3]{ 0.0, 0.0, 0.0 });
}