StockSelector.cpp

/*
 * StockSelector.cpp
 *
 *  Created on: Dec 20, 2017
 *      Author: ethanyoung
 */

#include "StockSelector.h"
#include <iostream> //For cin and cout
#include "DataIO.h"
#include "Backtest.h"


using namespace std;

/**
 * (Write a succinct description of this class here. You should avoid
 * wordiness and redundancy. If necessary, additional paragraphs should
 * be preceded by <p>, the html tag for a new paragraph.)
 *
 * <p>Bugs: (a list of bugs and other problems)
 *
 * @author E
 */

//Constructor
StockSelector::StockSelector() {
	cout << "Constructor called." << endl;

}

//Destructor
StockSelector::~StockSelector() {
	cout << "Destructor called." << endl;
}

/**
 * (Write a succinct description of this method here.  If necessary,
 * additional paragraphs should be preceded by <p>, the html tag for
 * a new paragraph.)
 *
 * @param (parameter name) (Describe the first parameter here)
 * @param (parameter name) (Do the same for each additional parameter)
 * @return (description of the return value)
 */
int main() {
	DataIO IOStream(stockFilenameDefault, investorFilenameDefault, fundFilenameDefault);
	fund = IOStream.initializeFundLocal();

	//login();
    managerPrompt();

	return 0;
}

//Displays main message and prompts for credentials (Returns 0 on success, -1 on failure)
//Bug: If in while loop and enter a letter, it goes on infinitely (w was tested)
//Bug: Figure out how to make exit be a q

int login() {
	int response;
	bool valid = false; //Input valid or not

	//Prints opening message and gets user response
	cout << openingMessage << endl;


	while (!valid) {
		//Gets user input
		cin >> response;
		valid = true;

		//Calls appropriate prompt depending on response (TODO: Check return)
		switch (response) {
			//Make this exit later instead of return
			case 0:	return -1;
			case 1: investorPrompt();
					break;
			case 2: managerPrompt();
					break;
			case 3: developerPrompt();
					break;
			case 4: newUserPrompt();
					break;
			default: cout << responseInvalidInput << endl;
					valid = false;
					break;
		}
	}
	return 0;
}

//Handle if Investor
int investorPrompt() {
	char option;

	//Gets user name and password
	if (!checkLogin(1)) return -1;
	cout << investorMain << endl;
	cin >> option;

	return 0;
}

//Handle if Manager (Should initialize all investors and stocks)
int managerPrompt() {
	char option;
	//Gets user name and password
	//if (!checkLogin(2)) return -1;

	//Gets the menu option
	cout << managerMain << endl;
	//cin >> option;

    // Run backtest prompt
    //if (option == 'f') {
		//Testing only
		int interval = 30;
		int numDays = 20;
		tm *predictDate = new tm();

		/*
		//This is for the Small stockdate
		 predictDate->tm_mday = 21;
		 predictDate->tm_mon = 9;
		 predictDate->tm_year = 17;

        //Prompt user for possible factors to include
        vector <string> *factorNames = new vector<string>();
        factorNames->push_back("LowPrice");
        factorNames->push_back("HighPrice");
        factorNames->push_back("Volatility");
	*/
		vector<tm *> *dates = fund->getDateList();
		vector<tm *>::iterator date;
		map<int, double> *profits = new map<int, double>();
		predictDate->tm_mday = -1;
		predictDate->tm_mon = 0;
		predictDate->tm_year = 18;
		Stock st("Hey:");
		//Prompt user for possible factors to include
		vector<string> *factorNames = new vector<string>();
		factorNames->push_back("low");
		factorNames->push_back("high");
		factorNames->push_back("Volume");
		bool unknown = true;
		Backtest *bt = new Backtest(fund, interval, factorNames, numDays, predictDate, unknown, 1000, 10, .25, .025);
		double profit = bt->userBacktestMain();
		/*int i = 0;
		for (date = dates->end() - 100; i < 5; date++) {

			//Creates new backtest and runs itFund *fundEx, int interva, vector<string> *factors, int numDay, tm *predictedDate, bool unknown, int budget, int numPur, double upperLim, double lowerLim
			Backtest *bt = new Backtest(fund, interval, factorNames, numDays, *date, unknown, 1000, 10, .25, .025);
			double profit = bt->userBacktestMain();
			cout << "Profit profit: " << profit << endl;
			profits->insert(pair<int, double>(st.convertDate(*date), profit));
			i++;
		}
		map<int, double>::iterator it;
        double avg = 0.0;
        int total = 0;
		for (it = profits->begin(); it != profits->end(); it++) {
            avg = avg + it->second;
			cout << "Profit: " << it->second << endl;
            total++;
		}

        cout << "Average: " << avg/total << endl;
    //}*/




	return 0;
}

//Handle if Developer (Should initialize all investors and stocks)
int developerPrompt() {
	char option;

	//Gets user name and password
	if (!checkLogin(3)) return -1;

	//Gets the menu option
	cout << devMain << endl;
	cin >> option;

	return 0;
}

//Handle if New User
int newUserPrompt() {
	char option;

	//Gets user name and password
	if (!checkLogin(4)) return -1;
	cout << "New User" << endl;

	//Gets the menu option
	//cout << newUserMain << endl;
	cin >> option;

	return 0;
}

//Checks if the user is in the system
//param: prompt - which prompt mode it is in
//Return: Returns true if valid
bool checkLogin(int mode) {
	string username;
	string password;

	while (true) {
		//Gets username
		cout << usernamePrompt << endl;
		cin >> username;

		if (username == "QUIT")
			return false;

		//Gets password
		cout << passwordPrompt << endl;
		cin >> password;

		//Checks if in database
		if (fund->checkCredentials(username, password)) {
			return true;
		}

		cout << invalidUser << endl;
	}
}