BatchDialog.cpp

Go to the documentation of this file.

/***********************************************************************
 * @file        src/Dialogs/BatchDialog.h
 * @brief       Declares BatchDialog and Related Functions
 * @author      mill-j
 * @since       (C) 2017 - 2023
 * @copyright   GNU GPL v3
 * ********************************************************************/
 
#include <wx/spinctrl.h>
#include <wx/choice.h>
#include <wx/wx.h>
 
#include "DateDialog.cpp"
 
using namespace std;
 
class BatchDialog : public wxDialog
{
    protected:
        wxChoice* birdChoice;
        wxChoice* breedChoice;
        wxChoice* groupChoice;
        wxChoice* IncubatorChoice;
        wxTextCtrl* batchText;
        wxStaticText* batchNameStat;
        wxSpinCtrl* eggNumSpinner;
        wxStaticText* NumberOfEggs;
        wxChoice* candleChoice;
        wxStaticText* showCandleText;
        wxDatePickerCtrl* StartDatePicker;
        wxStaticText* showStartText;
        wxChoice* hatchChoice;
        wxStaticText* showHatchText;
        wxButton* cancelButton;
        wxButton* saveButton;
        bool isCancel = true; //lets other functs know if BatchDialog was saved or closed/canceled
        
        void OnBirdChoice(wxCommandEvent& );
        void OnSaveButton(wxCommandEvent& );
        void OnCancelButton(wxCommandEvent& );
        void OnHatchChoice(wxCommandEvent& );
        void OnBatchName(wxCommandEvent& );
        int GetIncubationDays(int TypeSelect, string BreedName);
    public:
        bool WasCanceled();
        BatchDialog( wxWindow* parent, wxWindowID id = wxID_ANY, const wxString& title = wxT("New Batch"), const wxPoint& pos = wxDefaultPosition, const wxSize& size = wxDefaultSize, long style = wxCLOSE_BOX|wxCAPTION|wxSYSTEM_MENU );
};
 
BatchDialog::BatchDialog( wxWindow* parent, wxWindowID id, const wxString& title, const wxPoint& pos, const wxSize& size, long style ) : wxDialog( parent, id, title, pos, size, style )
{
    this->SetSizeHints( wxDefaultSize, wxDefaultSize );
    
    //Colors
    vector<int> ColorRed = Settings.GetIntArray("ColorRed");
    vector<int> ColorGreen = Settings.GetIntArray("ColorGreen");
    vector<int> ColorBlue = Settings.GetIntArray("ColorBlue");
    this->SetForegroundColour( wxColor(ColorRed[8],ColorGreen[8],ColorBlue[8]) );
    this->SetBackgroundColour( wxColor(ColorRed[7],ColorGreen[7],ColorBlue[7]) );
    wxBoxSizer* bSizer1;
    bSizer1 = new wxBoxSizer( wxVERTICAL );
 
    wxGridSizer* gSizer1;
    gSizer1 = new wxGridSizer( 8, 2, 0, 0 );
 
    wxArrayString birdChoiceChoices;
    birdChoice = new wxChoice( this, wxID_ANY, wxDefaultPosition, wxDefaultSize, birdChoiceChoices, 0 );
    birdChoice->SetSelection( 0 );
    birdChoice->SetToolTip( l("Select Type") );
    gSizer1->Add( birdChoice, 0, wxALL|wxEXPAND, 5 );
 
    wxArrayString breedChoiceChoices;
    breedChoice = new wxChoice( this, wxID_ANY, wxDefaultPosition, wxDefaultSize, breedChoiceChoices, 0 );
    breedChoice->SetSelection( 0 );
    breedChoice->SetToolTip( l("Select Breed") );
    gSizer1->Add( breedChoice, 0, wxALL|wxEXPAND, 5 );
    
    wxArrayString groupChoiceChoices;
    groupChoice = new wxChoice( this, wxID_ANY, wxDefaultPosition, wxDefaultSize, groupChoiceChoices, 0 );
    groupChoice->SetSelection( 0 );
    groupChoice->SetToolTip( l("Select Group") );
    gSizer1->Add( groupChoice, 0, wxALL|wxEXPAND, 5 );
 
    wxArrayString IncubatorChoiceChoices;
    IncubatorChoice = new wxChoice( this, wxID_ANY, wxDefaultPosition, wxDefaultSize, IncubatorChoiceChoices, 0 );
    IncubatorChoice->SetSelection( 0 );
    IncubatorChoice->SetToolTip( l("Select Incubator") );
    gSizer1->Add( IncubatorChoice, 0, wxALL|wxEXPAND, 5 );
    
    
    wxTextValidator textVal(wxFILTER_INCLUDE_CHAR_LIST,NULL);
    textVal.SetCharIncludes(Info.GetValidChars().c_str());
    
    batchText = new wxTextCtrl( this, wxID_ANY, wxEmptyString, wxDefaultPosition, wxDefaultSize, 0,textVal );
    gSizer1->Add( batchText, 0, wxALL|wxEXPAND, 5 );
 
    batchNameStat = new wxStaticText( this, wxID_ANY, l("Batch Name"), wxDefaultPosition, wxDefaultSize, 0 );
    batchNameStat->Wrap( -1 );
    gSizer1->Add( batchNameStat, 0, wxALL|wxALIGN_CENTER_VERTICAL|wxALIGN_CENTER_HORIZONTAL, 5 );
    
    eggNumSpinner = new wxSpinCtrl( this, wxID_ANY, wxEmptyString, wxDefaultPosition, wxDefaultSize, wxSP_ARROW_KEYS, 1, 10000, 1 );
    eggNumSpinner->SetForegroundColour( wxColor(0,0,0) );
    gSizer1->Add( eggNumSpinner, 0, wxALL|wxEXPAND, 5 );
 
    NumberOfEggs = new wxStaticText( this, wxID_ANY, l("Eggs"), wxDefaultPosition, wxDefaultSize, 0 );
    NumberOfEggs->Wrap( -1 );
    gSizer1->Add( NumberOfEggs, 0, wxALL|wxALIGN_CENTER_VERTICAL|wxALIGN_CENTER_HORIZONTAL, 5 );
 
    wxArrayString candleChoiceChoices;
    candleChoice = new wxChoice( this, wxID_ANY, wxDefaultPosition, wxDefaultSize, candleChoiceChoices, 0 );
    candleChoice->SetSelection( 0 );
    gSizer1->Add( candleChoice, 0, wxALL|wxEXPAND, 5 );
 
    showCandleText = new wxStaticText( this, wxID_ANY, l("Candle Date"), wxDefaultPosition, wxDefaultSize, 0 );
    showCandleText->Wrap( -1 );
    gSizer1->Add( showCandleText, 0, wxALL|wxALIGN_CENTER_VERTICAL|wxALIGN_CENTER_HORIZONTAL, 5 );
 
    StartDatePicker = new wxDatePickerCtrl( this, wxID_ANY, wxDefaultDateTime, wxDefaultPosition, wxDefaultSize, wxDP_DEFAULT );
    gSizer1->Add( StartDatePicker, 0, wxALL|wxEXPAND, 5 );
 
    showStartText = new wxStaticText( this, wxID_ANY, l("Start Date"), wxDefaultPosition, wxDefaultSize, 0 );
    showStartText->Wrap( -1 );
    gSizer1->Add( showStartText, 0, wxALL|wxALIGN_CENTER_VERTICAL|wxALIGN_CENTER_HORIZONTAL, 5 );
    
    wxString hatchChoiceChoices[] = { l("Automatic Calculation"), l("Set Hatch Date") };
    int hatchChoiceNChoices = sizeof( hatchChoiceChoices ) / sizeof( wxString );
    hatchChoice = new wxChoice( this, wxID_ANY, wxDefaultPosition, wxDefaultSize,hatchChoiceNChoices, hatchChoiceChoices , 0 );
    hatchChoice->SetSelection( 0 );
    hatchChoice->Enable(false);
    gSizer1->Add( hatchChoice, 0, wxALL|wxEXPAND, 5 );
 
    showHatchText = new wxStaticText( this, wxID_ANY, l("Hatch Date"), wxDefaultPosition, wxDefaultSize, 0 );
    showHatchText->Wrap( -1 );
    gSizer1->Add( showHatchText, 0, wxALL|wxALIGN_CENTER_HORIZONTAL|wxALIGN_CENTER_VERTICAL, 5 );
 
    cancelButton = new wxButton( this, wxID_ANY, l("Cancel"), wxDefaultPosition, wxDefaultSize, 0 );
    gSizer1->Add( cancelButton, 0, wxALL|wxEXPAND, 5 );
 
    saveButton = new wxButton( this, wxID_ANY, l("Save"), wxDefaultPosition,wxDefaultSize, 0 );
    saveButton->Enable(false);
    gSizer1->Add( saveButton, 0, wxALL|wxEXPAND, 5 );
 
 
    bSizer1->Add( gSizer1, 0, wxEXPAND, 5 );
    bSizer1->SetSizeHints(this);
    saveButton->Bind(wxEVT_BUTTON, &BatchDialog::OnSaveButton, this);
    cancelButton->Bind(wxEVT_BUTTON, &BatchDialog::OnCancelButton, this);
    birdChoice->Bind(wxEVT_CHOICE, &BatchDialog::OnBirdChoice, this);
    hatchChoice->Bind(wxEVT_CHOICE, &BatchDialog::OnHatchChoice, this);
    batchText->Bind(wxEVT_TEXT, &BatchDialog::OnBatchName, this);
    
    this->SetSizer( bSizer1 );
    this->Layout();
 
    int counter;
    //Insert Candle Days
    for(counter = 0; counter < 50; counter++)
        candleChoice->Append(l("Candle On Day ") + to_string(counter));
 
    candleChoice->SetSelection( Settings.GetInt("TillCandle") );
    
    //Insert Schedules
    for (counter = 0; counter < Schedules.GetCount();counter++)
        hatchChoice->Append( Schedules.GetName(counter).c_str());
    
    birdChoice->Append(l("Select Type"));
    //Insert Types
    for (counter = 0; counter < Types.GetCount();counter++)
        birdChoice->Append(Types.GetName(counter).c_str());
 
    birdChoice->SetSelection(0);
    
    //Insert Groups
    for (counter = 0; counter < Groups.GetCount();counter++)
        groupChoice->Append(Groups.GetName(counter).c_str());
 
    groupChoice->SetSelection(0);
    
    //Show Incubators
    vector<string> Inc = Settings.GetArray("Incubators");
    for (counter = 0; counter < Inc.size();counter++)
       IncubatorChoice->Append(Inc[counter].c_str());
    IncubatorChoice->SetSelection(0);
}
 
void BatchDialog::OnSaveButton(wxCommandEvent& )
{    
    if(birdChoice->GetSelection() == 0) {
        wxMessageDialog dialog(this,
         l("Please Select A Bird Type First!"));
        dialog.ShowModal();
        return;
    }
    
    if(batchText->GetLineText(0) == "") {
        wxMessageDialog dialog(this,
         l("Please Enter Batch Name!"));
        dialog.ShowModal();
        return;
    }
 
    if(!Batches.IsValidName(batchText->GetLineText(0).ToStdString())) {
        wxMessageDialog dialog(this,
        ("Name Exists! Please Enter A New Name!"));
        dialog.ShowModal();
        return;
    }
 
    wxDateTime WXdate = StartDatePicker->GetValue();
    
    HK_Date HKdate(
        WXdate.GetDay(wxDateTime::Local),
        WXdate.GetMonth(wxDateTime::Local) + 1,
        WXdate.GetYear(wxDateTime::Local)
    );
    
    //Copy Start Date
    HK_Date HKcandle = HKdate;
    HKcandle.AddDays(candleChoice->GetSelection());
 
    //Get Hatch Date
    HK_Date HKhatch = HKdate;
    HKhatch.AddDays(GetIncubationDays(birdChoice->GetSelection() - 1,breedChoice->GetString(breedChoice->GetSelection()).ToStdString()));
 
    //Get Lockdown Date
    HK_Date HKlock = HKhatch;
    HKlock.SubtractDays(Settings.GetInt("BeforeHatch"));
    
    //Create A Batch With Our Data
    HK_Batch Batch(
        Batches.GetValidID(),
        batchText->GetLineText(0).ToStdString(),
        birdChoice->GetString(birdChoice->GetSelection()).ToStdString(),
        breedChoice->GetString(breedChoice->GetSelection()).ToStdString(),
        groupChoice->GetString(groupChoice->GetSelection()).ToStdString(),
        l("Comments:"),
        IncubatorChoice->GetString(IncubatorChoice->GetSelection()).ToStdString(),
        eggNumSpinner->GetValue(),
        0,
        0,
        HKdate,
        HKcandle,
        HKlock,
        HKhatch,
        0,
        0
    );
 
    if(!Database.Execute(Batch.GetSQL())) {
        wxMessageDialog* dialog = new wxMessageDialog(this,
        "Sqlite Error: " + Database.GetErrors());
        dialog->ShowModal();
        dialog->Destroy();
        return;
    }
    isCancel = false;
    EndModal(0);
}
 
void BatchDialog::OnCancelButton(wxCommandEvent& )
{
    isCancel = true;
    EndModal(0);
}
 
void BatchDialog::OnBatchName(wxCommandEvent& ) {
    if(birdChoice->GetSelection() > 0 && batchText->GetLineText(0) != "")
        saveButton->Enable(true);
    else 
        saveButton->Enable(false);
}
 
void BatchDialog::OnBirdChoice(wxCommandEvent& )
{
    if(birdChoice->GetSelection() > 0) {
        hatchChoice->Enable(true);
        
        if(batchText->GetLineText(0) != "")
            saveButton->Enable(true);
    } 
    else {
        hatchChoice->Enable(false);
        saveButton->Enable(false);
    }
    
    string s = birdChoice->GetString(birdChoice->GetSelection()).ToStdString();
    
    breedChoice->Clear();
    
    for(int a = 0;a < Breeds.GetCount(); a++) {
        if(s == Breeds.GetText(a)) 
            breedChoice->Append(Breeds.GetName(a).c_str());
    }
    
    breedChoice->Append(l("Unknown"));
    breedChoice->SetSelection( 0 );
}
 
void BatchDialog::OnHatchChoice(wxCommandEvent & )
{
    int sel = hatchChoice->GetSelection();
    
    if (sel == 0)
        return;
    
    if(birdChoice->GetSelection() == 0) {
        wxMessageDialog dialog(this,
         l("Please Select A Bird Type First!"));
        dialog.ShowModal();
        return;
    }
 
    if (sel == 1) {
        DateDialog  dialog(this,2002,l("Select Hatch Date"));
        dialog.ShowModal();
        HK_Date Date = dialog.GetDate();
    
        Date.SubtractDays(GetIncubationDays(birdChoice->GetSelection() - 1,breedChoice->GetString(breedChoice->GetSelection()).ToStdString()));
        
        wxDateTime startD(
            Date.GetDay(),
            wxDateTime::Month(wxDateTime::Jan + Date.GetMonth() - 1), 
            Date.GetYear(),
            0,0,0,0
        );
            
        StartDatePicker->SetValue(startD);
    }
    if (sel > 1) {
        HK_Date date = Schedules.GetDate(sel-2);
 
        date.SubtractDays(GetIncubationDays(birdChoice->GetSelection() - 1,breedChoice->GetString(breedChoice->GetSelection()).ToStdString()));
        
        wxDateTime startD(date.GetDay(),
            wxDateTime::Month(wxDateTime::Jan + date.GetMonth() - 1), 
            date.GetYear(),0,0,0,0);
            
        StartDatePicker->SetValue(startD);
    }
}
 
int BatchDialog::GetIncubationDays(int TypeSelection, string BreedName) {
    for(int a = 0; a < Breeds.GetCount(); a++) {
        if(Breeds.GetName(a) == BreedName)
            return Breeds.GetNumber(a);
    }
    return Types.GetNumber(TypeSelection);
}
 
 
bool BatchDialog::WasCanceled(){return isCancel;}