/*
 * File:	test.cpp
 * Purpose:	wxGrid test
 * Author:	Julian Smart
 * Created:	1995
 * Updated:	
 * Copyright:   (c) 1995, AIAI, University of Edinburgh
 */

static const char sccsid[] = "%W% %G%";

#ifdef __GNUG__
#pragma implementation
#pragma interface
#endif

// For compilers that support precompilation, includes "wx/wx.h".
#include "wx/wxprec.h"

#ifdef __BORLANDC__
#pragma hdrstop
#endif

#ifndef WX_PRECOMP
#include "wx/wx.h"
#endif

#include "wx/grid.h"
#include "wx/colordlg.h"

// Define a new application type
class MyApp: public wxApp
{ public:
    bool OnInit(void);
};

// Define a new frame type
class MyFrame: public wxFrame
{ public:
    wxGrid *grid;
    MyFrame(wxFrame *frame, const wxString& title, const wxPoint& pos, const wxSize& size);

    bool OnClose(void) { return TRUE; }

    void ToggleEditable(wxCommandEvent& event);
    void ToggleRowLabel(wxCommandEvent& event);
    void ToggleColLabel(wxCommandEvent& event);
    void ToggleDividers(wxCommandEvent& event);
    void LeftCell(wxCommandEvent& event);
    void CentreCell(wxCommandEvent& event);
    void RightCell(wxCommandEvent& event);
    void ColourLabelBackground(wxCommandEvent& event);
    void ColourLabelText(wxCommandEvent& event);
    void NormalLabelColouring(wxCommandEvent& event);
    void ColourCellBackground(wxCommandEvent& event);
    void ColourCellText(wxCommandEvent& event);
    void NormalCellColouring(wxCommandEvent& event);
    void Quit(wxCommandEvent& event);

    void OnActivate(wxActivateEvent& event);

DECLARE_EVENT_TABLE()
};

wxBitmap *cellBitmap1 = NULL;
wxBitmap *cellBitmap2 = NULL;

// ID for the menu quit command
#define GRID_QUIT 1
#define GRID_TOGGLE_EDITABLE 2
#define GRID_LEFT_CELL       3
#define GRID_CENTRE_CELL     4
#define GRID_RIGHT_CELL      5
#define GRID_TOGGLE_ROW_LABEL 6
#define GRID_TOGGLE_COL_LABEL 7
#define GRID_COLOUR_LABEL_BACKGROUND 8
#define GRID_COLOUR_LABEL_TEXT       9
#define GRID_NORMAL_LABEL_COLOURING  10
#define GRID_COLOUR_CELL_BACKGROUND 11
#define GRID_COLOUR_CELL_TEXT       12
#define GRID_NORMAL_CELL_COLOURING  13
#define GRID_TOGGLE_DIVIDERS        14

// Main proc

IMPLEMENT_APP(MyApp)

// `Main program' equivalent, creating windows and returning main app frame
bool MyApp::OnInit(void)
{
#ifdef __WXMSW__
  cellBitmap1 = new wxBitmap("bitmap1");
  cellBitmap2 = new wxBitmap("bitmap2");
#endif

  // Create the main frame window
  MyFrame *frame = new MyFrame(NULL, "wxGrid Sample", wxPoint(50, 50), wxSize(450, 300));
  
  // Give it an icon
#ifdef __WXMSW__
  frame->SetIcon(wxIcon("mondrian"));
#endif
#ifdef __X__
  frame->SetIcon(wxIcon("aiai.xbm"));
#endif

  // Make a menubar
  wxMenu *file_menu = new wxMenu;
  file_menu->Append(GRID_QUIT, "E&xit");

  wxMenu *settings_menu = new wxMenu;
  settings_menu->Append(GRID_TOGGLE_EDITABLE, "&Toggle editable");
  settings_menu->Append(GRID_TOGGLE_ROW_LABEL, "Toggle ro&w label");
  settings_menu->Append(GRID_TOGGLE_COL_LABEL, "Toggle co&l label");
  settings_menu->Append(GRID_TOGGLE_DIVIDERS, "Toggle &dividers");
  settings_menu->AppendSeparator();
  settings_menu->Append(GRID_LEFT_CELL, "&Left cell alignment ");
  settings_menu->Append(GRID_CENTRE_CELL, "&Centre cell alignment ");
  settings_menu->Append(GRID_RIGHT_CELL, "&Right cell alignment ");
  settings_menu->AppendSeparator();
  settings_menu->Append(GRID_COLOUR_LABEL_BACKGROUND, "Choose a label &background colour");
  settings_menu->Append(GRID_COLOUR_LABEL_TEXT, "Choose a label fore&ground colour");
  settings_menu->Append(GRID_NORMAL_LABEL_COLOURING, "&Normal label colouring");
  settings_menu->AppendSeparator();
  settings_menu->Append(GRID_COLOUR_CELL_BACKGROUND, "Choo&se a cell &background colour");
  settings_menu->Append(GRID_COLOUR_CELL_TEXT, "Choose &a cell foreground colour");
  settings_menu->Append(GRID_NORMAL_CELL_COLOURING, "N&ormal cell colouring");

  wxMenuBar *menu_bar = new wxMenuBar;
  menu_bar->Append(file_menu, "&File");
  menu_bar->Append(settings_menu, "&Settings");
  frame->SetMenuBar(menu_bar);

  // Make a grid
  frame->grid = new wxGrid(frame, 0, 0, 400, 400);

  frame->grid->CreateGrid(10, 8);
  frame->grid->SetColumnWidth(3, 200);
  frame->grid->SetRowHeight(4, 45);
  frame->grid->SetCellValue("First cell", 0, 0);
  frame->grid->SetCellValue("Another cell", 1, 1);
  frame->grid->SetCellValue("Yet another cell", 2, 2);
  frame->grid->SetCellTextFont(wxTheFontList->FindOrCreateFont(10, wxROMAN, wxITALIC, wxNORMAL), 0, 0);
  frame->grid->SetCellTextColour(*wxRED, 1, 1);
  frame->grid->SetCellBackgroundColour(*wxCYAN, 2, 2);
  if (cellBitmap1 && cellBitmap2)
  {
    frame->grid->SetCellAlignment(wxCENTRE, 5, 0);
    frame->grid->SetCellAlignment(wxCENTRE, 6, 0);
    frame->grid->SetCellBitmap(cellBitmap1, 5, 0);
    frame->grid->SetCellBitmap(cellBitmap2, 6, 0);
  }
  
  frame->grid->UpdateDimensions();
  
  // Show the frame
  frame->Show(TRUE);

  SetTopWindow(frame);
  return TRUE;
}

// My frame constructor
MyFrame::MyFrame(wxFrame *frame, const wxString& title, const wxPoint& pos, const wxSize& size):
  wxFrame(frame, -1, title, pos, size)
{
  grid = NULL;
}

BEGIN_EVENT_TABLE(MyFrame, wxFrame)
  EVT_MENU(GRID_TOGGLE_EDITABLE, MyFrame::ToggleEditable)
  EVT_MENU(GRID_TOGGLE_ROW_LABEL, MyFrame::ToggleRowLabel)
  EVT_MENU(GRID_TOGGLE_COL_LABEL, MyFrame::ToggleColLabel)
  EVT_MENU(GRID_TOGGLE_DIVIDERS, MyFrame::ToggleDividers)
  EVT_MENU(GRID_LEFT_CELL, MyFrame::LeftCell)
  EVT_MENU(GRID_CENTRE_CELL, MyFrame::CentreCell)
  EVT_MENU(GRID_RIGHT_CELL, MyFrame::RightCell)
  EVT_MENU(GRID_COLOUR_LABEL_BACKGROUND, MyFrame::ColourLabelBackground)
  EVT_MENU(GRID_COLOUR_LABEL_TEXT, MyFrame::ColourLabelText)
  EVT_MENU(GRID_NORMAL_LABEL_COLOURING, MyFrame::NormalLabelColouring)
  EVT_MENU(GRID_COLOUR_CELL_BACKGROUND, MyFrame::ColourCellBackground)
  EVT_MENU(GRID_COLOUR_CELL_TEXT, MyFrame::ColourCellText)
  EVT_MENU(GRID_NORMAL_CELL_COLOURING, MyFrame::NormalCellColouring)
  EVT_MENU(GRID_QUIT, MyFrame::Quit)
END_EVENT_TABLE()

void MyFrame::ToggleEditable(wxCommandEvent& WXUNUSED(event))
{
      grid->SetEditable(!grid->GetEditable());
      grid->Refresh();
}

void MyFrame::ToggleRowLabel(wxCommandEvent& WXUNUSED(event))
{
      if (grid->GetLabelSize(wxVERTICAL) > 0)
        grid->SetLabelSize(wxVERTICAL, 0);
      else
        grid->SetLabelSize(wxVERTICAL, 40);
      grid->Refresh();
}

void MyFrame::ToggleColLabel(wxCommandEvent& WXUNUSED(event))
{
      if (grid->GetLabelSize(wxHORIZONTAL) > 0)
        grid->SetLabelSize(wxHORIZONTAL, 0);
      else
        grid->SetLabelSize(wxHORIZONTAL, 20);
      grid->Refresh();
}

void MyFrame::ToggleDividers(wxCommandEvent& WXUNUSED(event))
{
      if (!grid->GetDividerPen())
        grid->SetDividerPen(wxThePenList->FindOrCreatePen("LIGHT GREY", 1, wxSOLID));
      else
        grid->SetDividerPen(NULL);
      grid->Refresh();
}

void MyFrame::LeftCell(wxCommandEvent& WXUNUSED(event))
{
      grid->SetCellAlignment(wxLEFT);
      grid->Refresh();
}

void MyFrame::CentreCell(wxCommandEvent& WXUNUSED(event))
{
      grid->SetCellAlignment(wxCENTRE);
      grid->Refresh();
}

void MyFrame::RightCell(wxCommandEvent& WXUNUSED(event))
{
      grid->SetCellAlignment(wxRIGHT);
      grid->Refresh();
}

void MyFrame::ColourLabelBackground(wxCommandEvent& WXUNUSED(event))
{
      wxColourData data;
      data.SetChooseFull(TRUE);
      wxColourDialog dialog(this, &data);
      if (dialog.ShowModal() != wxID_CANCEL)
      {
        wxColourData retData = dialog.GetColourData();
        wxColour col = retData.GetColour();
        grid->SetLabelBackgroundColour(col);
        grid->Refresh();
      }
}

void MyFrame::ColourLabelText(wxCommandEvent& WXUNUSED(event))
{
      wxColourData data;
      data.SetChooseFull(TRUE);
      wxColourDialog dialog(this, &data);
      if (dialog.ShowModal() != wxID_CANCEL)
      {
        wxColourData retData = dialog.GetColourData();
        wxColour col = retData.GetColour();
        grid->SetLabelTextColour(col);
        grid->Refresh();
      }
}

void MyFrame::NormalLabelColouring(wxCommandEvent& WXUNUSED(event))
{
      grid->SetLabelBackgroundColour(*wxLIGHT_GREY);
      grid->SetLabelTextColour(*wxBLACK);
      grid->Refresh();
}

void MyFrame::ColourCellBackground(wxCommandEvent& WXUNUSED(event))
{
      wxColourData data;
      data.SetChooseFull(TRUE);
      wxColourDialog dialog(this, &data);
      if (dialog.ShowModal() != wxID_CANCEL)
      {
        wxColourData retData = dialog.GetColourData();
        wxColour col = retData.GetColour();
        grid->SetCellBackgroundColour(col);
        grid->Refresh();
      }
}

void MyFrame::ColourCellText(wxCommandEvent& WXUNUSED(event))
{
      wxColourData data;
      data.SetChooseFull(TRUE);
      wxColourDialog dialog(this, &data);
      if (dialog.ShowModal() != wxID_CANCEL)
      {
        wxColourData retData = dialog.GetColourData();
        wxColour col = retData.GetColour();
        grid->SetCellTextColour(col);
        grid->Refresh();
      }
}

void MyFrame::NormalCellColouring(wxCommandEvent& WXUNUSED(event))
{
      grid->SetCellBackgroundColour(*wxWHITE);
      grid->SetCellTextColour(*wxBLACK);
      grid->Refresh();
}

void MyFrame::Quit(wxCommandEvent& WXUNUSED(event))
{
      this->Close(TRUE);
}

// Ensure that the grid's edit control always has the focus.
void MyFrame::OnActivate(wxActivateEvent& event)
{
  if (grid) grid->OnActivate(event.GetActive());
}