DS-Project/MainBranchClass.cpp at main · A-hmy/DS-Project · 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
#include "MainBranchClass.h"

MainBranchClass::MainBranchClass()
{
	name = "";
	coordinate.setX(0);
	coordinate.setY(0);
	NumberOfBranch = 0;
}

MainBranchClass::MainBranchClass(string Name, float x, float y)
{
	name = Name;
	coordinate.setX(x);
	coordinate.setY(y);
	NumberOfBranch = 0;
}

void MainBranchClass::setBranch(const Branch& branch)
{
	Branches.push_back(branch);
}

void MainBranchClass::removeBranch(point branch)
{
	for (auto it = Branches.begin(); it != Branches.end(); ++it) {
		if ((*it).getCoordinate() == branch) {
			Branches.erase(it);
			break;
		}
	}
}

string MainBranchClass::getName()
{
	return name;
}

point MainBranchClass::getCordinate()
{
	return coordinate;
}

void MainBranchClass::printBranch()
{
	cout << name << "\t\t\t\033[38;5;28mbranches:";
	for (auto it : Branches) {
		cout <<"\t\t\t\033[38;5;28mName:" << it.getName() << "Coordinate:" << it.getCoordinate();
	}
	cout << "\033[0m";
}

void MainBranchClass::setName(string Name)
{
	name = Name;
}

void MainBranchClass::setCordinate(int x, int y)
{
	coordinate.setX(x);
	coordinate.setY(y);
}

void MainBranchClass::AddNumberOfBranch()
{
	NumberOfBranch++;
}

void MainBranchClass::ReduceNumberOfBeranch()
{
	NumberOfBranch--;
}

int MainBranchClass::getNumberOfBranch()
{
	return NumberOfBranch;
}

vector<Branch> MainBranchClass::getBranches()//==================
{
	return Branches;
}

bool MainBranchClass::operator==(const MainBranchClass& Mainbranch) const
{
	if (name == Mainbranch.name && coordinate == Mainbranch.coordinate)
		return 1;
	else
		return 0;
}