// ...excluded for clarity


void createPolygon(vector<Polygon*> &polygons)
{
	int userInput;
	string polyName;

	while (1)
	{
		cout << "What type of polygon would you like to create?\n"
			 << "0. Go back\n"
			 << "1. Triangle\n"
			 << "2. Rectangle\n";

		cin >> userInput;

		if (!userInput)
			break;

		switch (userInput)
		{
		case 1:
			{
				cout << "Please enter the name of your new triangle: ";
				cin >> polyName;

				polygons.push_back(new Triangle(polyName));
			}
			break;
		case 2:
			break;
		default:
			cout << "Invalid choice.\n";
			break;
		}

		cout << endl;
	}
}

void displayInstructions()
{
	cout << "Please select an option:\n"
		 << "0. Exit\n"
		 << "1. Create a polygon\n"
		 << "2. View your polygons\n"
		 << "3. Delete a polygon\n";
}

void displayPolygons(vector<Polygon*> &polygons)
{
	int temp = 0;
	string tempName;
	for (vector<Polygon*>::iterator iter = polygons.begin(); iter != polygons.end(); ++iter)
	{
		++temp;
		tempName = ((*iter)->GetName());

		cout << temp << ". " << tempName << endl;
	}
}

int main()
{
	vector<Polygon*> polygons;
	int userInput;

	while (1)
	{
		displayInstructions();
		cin >> userInput;

		if (!userInput)
			break;

		cout << endl;

		switch (userInput)
		{
		case 1:
			createPolygon(polygons);
			break;
		case 2:
			displayPolygons(polygons);
			break;
		case 3:
			break;
		default:
			cout << "Invalid choice.\n";
			break;
		}

		cout << endl;
	}

	for (vector<Polygon*>::iterator iter = polygons.begin(); iter != polygons.end(); ++iter)
	{
		delete *iter;
		*iter = NULL;
	}

	polygons.clear();

	_CrtDumpMemoryLeaks();
	system("PAUSE");
	return 0;
}