NewGameDialog.cpp

#include "stdafx.h"
#include "NewGameDialog.h"
#include "UCTSearch.h"

NewGameDialog::NewGameDialog( wxWindow* parent )
:
TNewGameDialog( parent )
{
                  
}

void NewGameDialog::doInit( wxInitDialogEvent& event ) {    
    int size = wxConfig::Get()->Read(wxT("DefaultBoardSize"), (long)4);
    m_radioBoxBoardSize->SetSelection(size);
    
    int handicap = wxConfig::Get()->Read(wxT("DefaultHandicap"), (long)0);
    m_spinCtrlHandicap->SetValue(handicap);
    
    int komi = wxConfig::Get()->Read(wxT("DefaultKomi"), (long)7);
    m_spinCtrlKomi->SetValue(komi);
    
    int simulations = wxConfig::Get()->Read(wxT("DefaultSimulations"), (long)6);
    m_radioBoxLevel->SetSelection(simulations);
    
    int minutes = wxConfig::Get()->Read(wxT("DefaultMinutes"), (long)20);
    m_spinCtrlTime->SetValue(minutes);
    
    int color = wxConfig::Get()->Read(wxT("DefaultColor"), (long)0);
    m_radioBoxColor->SetSelection(color);

    bool nets = wxConfig::Get()->Read(wxT("netsEnabled"), true);
    m_checkNeuralNet->SetValue(nets);

    checkNetsEnabled();
    checkHandicapRange();
}

void NewGameDialog::doCancel( wxCommandEvent& event ) {
    event.Skip();
}

void NewGameDialog::doOK( wxCommandEvent& event ) {
    int size = m_radioBoxBoardSize->GetSelection();
    wxConfig::Get()->Write(wxT("DefaultBoardSize"), (long)size);

    int simulations = m_radioBoxLevel->GetSelection();
    wxConfig::Get()->Write(wxT("DefaultSimulations"), (long)simulations);

    int color = m_radioBoxColor->GetSelection();
    wxConfig::Get()->Write(wxT("DefaultColor"), (long)color);

    int handicap = m_spinCtrlHandicap->GetValue();
    wxConfig::Get()->Write(wxT("DefaultHandicap"), (long)handicap);

    int komi = m_spinCtrlKomi->GetValue();
    wxConfig::Get()->Write(wxT("DefaultKomi"), (long)komi);

    int minutes = m_spinCtrlTime->GetValue();
    wxConfig::Get()->Write(wxT("DefaultMinutes"), (long)minutes);

    bool nets = m_checkNeuralNet->GetValue();
    wxConfig::Get()->Write(wxT("netsEnabled"), nets);

    event.Skip();
}

float NewGameDialog::getKomi() {
    float komi = m_spinCtrlKomi->GetValue();
    return komi + 0.5f;
}

int NewGameDialog::getHandicap() {
    int handicap = m_spinCtrlHandicap->GetValue();
    return handicap;
}

int NewGameDialog::getBoardsize() {
    wxString sboardsize = m_radioBoxBoardSize->GetStringSelection();
    if (sboardsize == "7 x 7") {
        return 7;
    } else if (sboardsize == "9 x 9") {
        return 9;
    } else if (sboardsize == "13 x 13") {
        return 13;
    } else if (sboardsize == "17 x 17") {
        return 17;
    } else if (sboardsize == "19 x 19") {
        return 19;
    } else if (sboardsize == "25 x 25") {
        return 25;
    }

    throw std::runtime_error("Invalid board size selection");
}

int NewGameDialog::simulationsToVisitLimit(int simuls) {
    if (simuls == 0) {
        return 100;
    } else if (simuls == 1) {
        return 500;
    } else if (simuls == 2) {
        return 1000;
    } else if (simuls == 3) {
        return 5000;
    } else if (simuls == 4) {
        return 10000;
    } else if (simuls == 5) {
        return 20000;
    } else if (simuls == 6) {
        return 0;
    } else {
        throw std::runtime_error("Invalid simulations level selection");
    }
}

int NewGameDialog::getSimulations() {
    int simuls = m_radioBoxLevel->GetSelection();
    return simulationsToVisitLimit(simuls);
}

int NewGameDialog::getPlayerColor() {
    int color = m_radioBoxColor->GetSelection();
    
    return color;
}

int NewGameDialog::getTimeControl() {
    return m_spinCtrlTime->GetValue();
}

void NewGameDialog::doHandicapUpdate( wxSpinEvent& event ) {    
    m_spinCtrlKomi->SetValue(0);
}

bool NewGameDialog::getNetsEnabled() {
    return m_checkNeuralNet->GetValue();
}

void NewGameDialog::doRadioBox(wxCommandEvent& event) {
    checkNetsEnabled();
    checkHandicapRange();
}

void NewGameDialog::checkNetsEnabled() {
    wxString sboardsize = m_radioBoxBoardSize->GetStringSelection();

    if (sboardsize != "19 x 19") {
        m_checkNeuralNet->Enable(false);
    } else {
        m_checkNeuralNet->Enable(true);
    }
}

void NewGameDialog::checkHandicapRange() {
    wxString sboardsize = m_radioBoxBoardSize->GetStringSelection();

    if (sboardsize == "7 x 7") {
        m_spinCtrlHandicap->SetRange(0, 4);
    } else if (sboardsize == "9 x 9") {
        m_spinCtrlHandicap->SetRange(0, 9);
    } else {
        m_spinCtrlHandicap->SetRange(0, 100);
    }
}