245 lines
6.8 KiB
C++
245 lines
6.8 KiB
C++
#include "sort_controller.hpp"
|
|
#include "basic_sorts.hpp"
|
|
#include "trees.hpp"
|
|
#include <algorithm>
|
|
#include <chrono>
|
|
#include <fstream>
|
|
#include <iostream>
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
// Initialization function for SortController class
|
|
SortController::SortController()
|
|
{
|
|
lineCount = 0;
|
|
currentType = INSERTION;
|
|
defaultFile = 0;
|
|
defaultOnly = 1;
|
|
fileGiven = 0;
|
|
allLists = 0;
|
|
sortGiven = 0;
|
|
}
|
|
|
|
// Checks for command line arguments
|
|
void SortController::CheckArguments(int argc, char* arguments[])
|
|
{
|
|
std::string tempStr;
|
|
for (int i = 0; i < argc; i++)
|
|
{
|
|
tempStr = arguments[i];
|
|
if ((tempStr == "-a") || (tempStr == "--all"))
|
|
{
|
|
defaultOnly = 0;
|
|
allLists = 1;
|
|
}
|
|
if ((tempStr == "-f") || (tempStr == "--filename"))
|
|
{
|
|
filename = arguments[i + 1];
|
|
defaultOnly = 0;
|
|
fileGiven = 1;
|
|
}
|
|
if ((tempStr == "-l") || (tempStr == "--locate"))
|
|
{
|
|
wordToLocate = arguments[i + 1];
|
|
std::transform(wordToLocate.begin(), wordToLocate.end(), wordToLocate.begin(), ::toupper);
|
|
locate = 1;
|
|
}
|
|
if ((tempStr == "-d") || (tempStr == "--default"))
|
|
{
|
|
filename = "test/PERM/perm15K.txt";
|
|
defaultFile = 1;
|
|
}
|
|
if ((tempStr == "-s") || (tempStr == "--sort-type"))
|
|
{
|
|
sortGiven = 1;
|
|
tempStr = arguments[i + 1];
|
|
if (tempStr == "bst")
|
|
currentType = BST;
|
|
if (tempStr == "rbt")
|
|
currentType = RBT;
|
|
if (tempStr == "insertion")
|
|
currentType = INSERTION;
|
|
if (tempStr == "merge")
|
|
currentType = MERGE;
|
|
if (tempStr == "heap")
|
|
currentType = HEAP;
|
|
if (tempStr == "all")
|
|
sortGiven = 0;
|
|
}
|
|
}
|
|
if (defaultOnly)
|
|
filename = "test/PERM/perm15K.txt";
|
|
return;
|
|
}
|
|
|
|
// Sets word list found in file into vector
|
|
void SortController::ReadWordFile(void)
|
|
{
|
|
std::string bufferStr;
|
|
std::ifstream file(this->filename);
|
|
if (!file.is_open())
|
|
{
|
|
std::cout << "Failed opening file: " << this->filename << '\n';
|
|
exit(1);
|
|
}
|
|
while (getline(file, bufferStr))
|
|
{
|
|
originalWordList.push_back(bufferStr);
|
|
lineCount++;
|
|
}
|
|
return;
|
|
}
|
|
|
|
// Main function for calling all sort categories
|
|
void SortController::RunBenchmarks(void)
|
|
{
|
|
if (defaultOnly || fileGiven)
|
|
{
|
|
ReadWordFile();
|
|
Benchmarking();
|
|
}
|
|
if (allLists)
|
|
BenchmarkingAll();
|
|
return;
|
|
}
|
|
|
|
|
|
// Function for starting sort functions
|
|
void SortController::Benchmarking(void)
|
|
{
|
|
if (!sortGiven)
|
|
currentType = INSERTION;
|
|
if (currentType == INSERTION)
|
|
{
|
|
newWordList = originalWordList;
|
|
auto start = std::chrono::system_clock::now();
|
|
basic_sorts::InsertionSort(&newWordList);
|
|
auto end = std::chrono::system_clock::now();
|
|
sortTime = end - start;
|
|
OutputResult();
|
|
}
|
|
if (!sortGiven)
|
|
currentType = MERGE;
|
|
if (currentType == MERGE)
|
|
{
|
|
newWordList = originalWordList;
|
|
auto start = std::chrono::system_clock::now();
|
|
basic_sorts::MergeSort(&newWordList);
|
|
auto end = std::chrono::system_clock::now();
|
|
sortTime = end - start;
|
|
OutputResult();
|
|
}
|
|
if (!sortGiven)
|
|
currentType = HEAP;
|
|
if (currentType == HEAP)
|
|
{
|
|
newWordList = originalWordList;
|
|
auto start = std::chrono::system_clock::now();
|
|
basic_sorts::HeapSort(&newWordList);
|
|
auto end = std::chrono::system_clock::now();
|
|
sortTime = end - start;
|
|
OutputResult();
|
|
}
|
|
if (currentType == BST)
|
|
{
|
|
newWordList = originalWordList;
|
|
tree_implementation::BinarySearchTree newTree;
|
|
auto start = std::chrono::system_clock::now();
|
|
newTree.InsertWordList(&newWordList);
|
|
auto end = std::chrono::system_clock::now();
|
|
sortTime = end - start;
|
|
OutputResult();
|
|
start = std::chrono::system_clock::now();
|
|
newTree.Search(wordToLocate);
|
|
end = std::chrono::system_clock::now();
|
|
sortTime = end - start;
|
|
EchoSortTime("BST");
|
|
newTree.PrintPathToRoot(wordToLocate);
|
|
}
|
|
if (currentType == RBT)
|
|
{
|
|
newWordList = originalWordList;
|
|
tree_implementation::RedBlackTree newTree;
|
|
auto start = std::chrono::system_clock::now();
|
|
newTree.InsertWordList(&newWordList);
|
|
auto end = std::chrono::system_clock::now();
|
|
sortTime = end - start;
|
|
OutputResult();
|
|
}
|
|
}
|
|
|
|
// Sorts all default files if allLists is set
|
|
void SortController::BenchmarkingAll(void)
|
|
{
|
|
int fileCount = 10;
|
|
std::string newFilename;
|
|
for (int i = 1; i <= fileCount; i++)
|
|
{
|
|
lineCount = 0;
|
|
newFilename = "test/PERM/perm";
|
|
newFilename += std::to_string(15*i);
|
|
newFilename += "K.txt";
|
|
filename = newFilename;
|
|
ReadWordFile();
|
|
Benchmarking();
|
|
originalWordList.clear();
|
|
}
|
|
}
|
|
|
|
|
|
// Main function for printing results
|
|
void SortController::OutputResult(void)
|
|
{
|
|
switch(currentType)
|
|
{
|
|
case INSERTION:
|
|
EchoSortTime("IS");
|
|
WriteOutputToFile("IS");
|
|
break;
|
|
case MERGE:
|
|
EchoSortTime("MS");
|
|
WriteOutputToFile("MS");
|
|
break;
|
|
case HEAP:
|
|
EchoSortTime("HS");
|
|
WriteOutputToFile("HS");
|
|
break;
|
|
case BST:
|
|
EchoSortTime("BST");
|
|
WriteOutputToFile("BST");
|
|
break;
|
|
case RBT:
|
|
EchoSortTime("RBT");
|
|
WriteOutputToFile("RBT");
|
|
break;
|
|
default:
|
|
break;
|
|
}
|
|
}
|
|
|
|
// Prints sort time to the screen
|
|
void SortController::EchoSortTime(std::string outputFilename)
|
|
{
|
|
std::string sortSizeString = std::to_string(lineCount / 1000);
|
|
std::cout << outputFilename << sortSizeString;
|
|
std::cout << " took " << sortTime.count() << " s" << std::endl;
|
|
return;
|
|
}
|
|
|
|
// Prints result of sort operation to file
|
|
void SortController::WriteOutputToFile(std::string outputFilename)
|
|
{
|
|
std::string sortSizeString = std::to_string(lineCount / 1000);
|
|
std::string outputPath = "test/OUTPUT/";
|
|
outputPath += outputFilename + sortSizeString + "K.txt";
|
|
std::ofstream file(outputPath);
|
|
if (!file.is_open())
|
|
{
|
|
std::cout << "Failed opening file\n";
|
|
exit(1);
|
|
}
|
|
for (unsigned int i = 0; i < lineCount; i++)
|
|
file << newWordList[i] << '\n';
|
|
file.close();
|
|
}
|