main.cpp

/**
 * @file main.cpp
 * Contains code to test your recursion, stack, and queue problems.
 */

#include <unistd.h>
#include <iostream>
#include <queue>
#include <sstream>
#include <stack>
#include <string>

#include "quackfun.h"
#include "exercises.h"

using namespace std;

template <typename T>
inline string to_string(const T& value);
template <typename T>
inline string colorize(const T& actual, const T& expected);
string make_red(const string& output);
string make_green(const string& output);

int main()
{
    cout << boolalpha
         << endl; // make true and false print for bools, not 0 and 1

    // Run sumDigits:
    cout << "==== sumDigits ====" << endl;
    cout << "sumDigits(243) = " << colorize(RecursionExercises::sumDigits(243), 9) << endl;
    cout << "expected       = " << 9 << endl;
    cout << "sumDigits(49)  = " << colorize(RecursionExercises::sumDigits(49), 13) << endl;
    cout << "expected       = " << 13 << endl;
    cout << "sumDigits(1)   = " << colorize(RecursionExercises::sumDigits(1), 1) << endl;
    cout << "expected       = " << 1 << endl;
    cout << "sumDigits(0)   = " << colorize(RecursionExercises::sumDigits(0), 0) << endl;
    cout << "expected       = " << 0 << endl << endl;

    // Run triangle:
    cout << "==== triangle ====" << endl;
    cout << "triangle(0) = " << colorize(RecursionExercises::triangle(0), 0) << endl;
    cout << "expected    = " << 0 << endl;
    cout << "triangle(3) = " << colorize(RecursionExercises::triangle(3), 6) << endl;
    cout << "expected    = " << 6 << endl;
    cout << "triangle(6) = " << colorize(RecursionExercises::triangle(6), 21) << endl;
    cout << "expected    = " << 21 << endl << endl;

    // Run sum:
    cout << "==== sum ====" << endl;
    stack<int> s;

    const int n = 16;
    for (int i = 0; i <= n; i++)
        s.push(i);

    stack<int> orig = s;
    cout << "Sum:      " << colorize(QuackFun::sum<int>(s), 136) << endl;
    cout << "Expected: " << 136 << endl;
    if (s == orig)
        cout << "Stack " + make_green("unmodified: :)") << endl;
    else
        cout << "Stack " + make_red("CHANGED: :(") << endl;
    cout << endl;

    // Run scramble:
    cout << "==== scramble ====" << endl;
    queue<int> q;
    for (int i = 0; i <= 16; i++)
        q.push(i);

    QuackFun::scramble<int>(q);

    int expected[] = {0, 2, 1, 3, 4, 5, 9, 8, 7, 6, 10, 11, 12, 13, 14, 16, 15};
    cout << "Scrambled queue: ";
    for (int i = 0; i <= 16; i++) {
        cout << colorize(q.front(), expected[i]) << ' ';
        q.pop();
    }
    cout << endl;
    cout << "Expected:        0 2 1 3 4 5 9 8 7 6 10 11 12 13 14 16 15 " << endl << endl;

    // Run verifySame:
    cout << "==== verifySame ====" << endl;
    stack<int> s1;
    queue<int> q1;
    for (int i = 0; i < 5; i++) {
        s1.push(i);
        q1.push(i);
    }
    q = q1;
    s = s1;
    cout << "verifySame(true): " << colorize(QuackFun::verifySame(s, q), true) << endl;
    if (q == q1 && s == s1)
        cout << "Stack and queue " + make_green("unmodified: :)") << endl;
    else
        cout << "Stack or queue " + make_red("CHANGED: :(") << endl;

    stack<int> s2;
    queue<int> q2;
    for (int i = 0; i < 4; i++) {
        s2.push(i + 1);
        q2.push(i);
    }
    s2.push(5);
    q2.push(5);

    q = q2;
    s = s2;
    cout << "verifySame(false): " << colorize(QuackFun::verifySame(s, q), false) << endl;
    if (q == q2 && s == s2)
        cout << "Stack and queue " + make_green("unmodified: :)") << endl;
    else
        cout << "Stack or queue " + make_red("CHANGED: :(") << endl;
    cout << endl;
}

template <typename T>
inline string to_string(const T& value)
{
    stringstream ss;
    ss << boolalpha << value;
    return ss.str();
}

template <typename T>
inline string colorize(const T& actual, const T& expected)
{
    if (isatty(STDOUT_FILENO)) {
        if (actual == expected)
            return "\033[32m" + to_string(actual) + "\033[0m";
        else
            return "\033[31m" + to_string(actual) + "\033[0m";
    } else
        return to_string(actual);
}

string make_red(const string& output)
{
    if (isatty(STDOUT_FILENO))
        return "\033[31m" + output + "\033[0m";
    else
        return output;
}

string make_green(const string& output)
{
    if (isatty(STDOUT_FILENO))
        return "\033[32m" + output + "\033[0m";
    else
        return output;
}