#include <afxwin.h>

#define VAREJA 4

void vaihda_varia(int &vari) {
  vari++;
  if (vari == VAREJA) vari=0;
  // vari = (vari+1) % VAREJA;
}

class cLauta {
  int x, y, nx, ny, ruudun_koko;
public:
  cLauta(int x,int y,int nx,int ny,int ruudun_koko);
  void Draw(CDC &dc) const;
};

cLauta::cLauta(int x,int y,int nx,int ny,int ruudun_koko) {
  this->x = x; this->y = y;
  this->nx = nx; this->ny = ny;
  this->ruudun_koko = ruudun_koko;
}

void cLauta::Draw(CDC &dc) const {
  int ix, iy, vari=0, rivin_ekan_vari = 0;
  CBrush br[10];
  CBrush *oldbr;

  br[0].CreateSolidBrush(RGB(200,200,0));
  br[1].CreateSolidBrush(RGB(0,0,0));
  br[2].CreateSolidBrush(RGB(0,0,255));
  br[3].CreateSolidBrush(RGB(255,255,255));

  oldbr = dc.GetCurrentBrush();

  for (iy=0; iy<ny; iy++) {
    vari = rivin_ekan_vari;
    for (ix=0; ix<nx; ix++) {
      dc.SelectObject(&br[vari]);
      dc.Rectangle(x+ix*ruudun_koko,y+iy*ruudun_koko,
        x+(ix+1)*ruudun_koko,y+(iy+1)*ruudun_koko);
      vaihda_varia(vari);
    }
    vaihda_varia(rivin_ekan_vari);
  }

  dc.SelectObject(oldbr);
}

class CMainWindow : public CFrameWnd {
  cLauta lauta;
public:
  CMainWindow() : lauta(50,50,8,8,10) {Create(NULL,"Shakkilauta");}
  afx_msg void OnPaint() { CPaintDC dc(this);  lauta.Draw(dc);  }
  DECLARE_MESSAGE_MAP()
};

BEGIN_MESSAGE_MAP( CMainWindow, CFrameWnd )
  ON_WM_PAINT()
END_MESSAGE_MAP()

class CShakkiApp : public CWinApp {
public:
  virtual BOOL InitInstance() {
    m_pMainWnd = new CMainWindow();
          m_pMainWnd->ShowWindow(m_nCmdShow);
          m_pMainWnd->UpdateWindow();
          return TRUE;
  }
};

CShakkiApp ShakkiApp;