/*


RTRay 011
---------

  . Added new class-based scene definition

*/



#include <windows.h>
#include <mmsystem.h>
#include <math.h>
#include <stdlib.h>
#include <stdio.h>
#include <conio.h>



//--- CONSTANTS --------------------------------------------------------------//

//- Settings

//#define FULL_WINDOW_DRAG
#define ROUND_IMAGE_SIZE



//---- CLASS DECLARATIONS ----------------------------------------------------//

class Image;
class Scanline;
class Poly;
class Raytrace;

#include "Classes\Image.h"
#include "Classes\Scanline.h"
#include "Classes\Poly.h"
#include "Classes\Raytrace.h"

  class XORPoly;
  class ANDPoly;
  class ORPoly;
  class SUBPoly;

  #include "Classes\Poly\XORPoly.h"
  #include "Classes\Poly\ANDPoly.h"
  #include "Classes\Poly\ORPoly.h"
  #include "Classes\Poly\SUBPoly.h"

  class TranslatePoly;
  class RotatePoly;

  class RegularPoly;
  //class StarPoly;

  #include "Classes\Poly\TranslatePoly.h"
  #include "Classes\Poly\RotatePoly.h"

  #include "Classes\Poly\RegularPoly.h"
  //#include "Classes\Poly\StarPoly.h"



//--- DATA -------------------------------------------------------------------//

//---- Windows data

HINSTANCE hMainInstance;
HWND hMainWindow = NULL;

char szAppName[40] = "RTRay 010";

int mouse_x = 0, mouse_y = 0;
int mouse_xstart = 0, mouse_ystart = 0;

bool mouse_drag = false;

float rx=0, ry=0, rrx=-1, rry=-1;



//---- Image buffer data

int width = 480;
int height = 360;

Image *im;
Raytrace *rt;



//---- Display ---------------------------------------------------------------//

void Display(HDC dc, int *buf, int w, int h)
{
  BITMAPINFO bmi;

  bmi.bmiHeader.biSize = sizeof(BITMAPINFOHEADER);
  bmi.bmiHeader.biWidth = w;
  bmi.bmiHeader.biHeight = -h;
  bmi.bmiHeader.biPlanes = 1;
  bmi.bmiHeader.biBitCount = 32;
  bmi.bmiHeader.biCompression = BI_RGB;
  bmi.bmiHeader.biSizeImage = 0;
  bmi.bmiHeader.biXPelsPerMeter = 1000;
  bmi.bmiHeader.biYPelsPerMeter = 1000;
  bmi.bmiHeader.biClrUsed = 0;
  bmi.bmiHeader.biClrImportant = 0;

  StretchDIBits(
    dc,
    0, 0, w, h,
    0, 0, w, h,
    (LPVOID)buf,
    (LPBITMAPINFO)&bmi,
    DIB_RGB_COLORS, SRCCOPY
  );

}



//---- Startup ---------------------------------------------------------------//

void Startup()
{
  //- Allocate image

  #ifdef ROUND_IMAGE_SIZE

  width = (width+7) & -8;
  height = (height+7) & -8;

  #endif

  im = new Image( width, height );

  //- Set up raytraced scene

  rt = new Raytrace( im );

  int texOrange = 0xFFBB00;
  int texPurple = 0xBB00FF;
  int texWhite  = 0xFFFFFF;
  
  rt->addSphere( 1.0, -1.0, 6.0-6.0, 1.0, texOrange);
  rt->addSphere( 1.0,  3.0, 6.0-6.0, 1.5, texPurple);
  rt->addSphere(-1.0,  2.0, 5.0-6.0, 1.2, texOrange);

  rt->addSphere(-3.0,  1.0, 6.0-6.0, 1.5, texWhite);
  rt->addSphere(-3.0, -1.0, 6.0-6.0, 1.5, texPurple);

  rt->addSphere( 4.0, 3.5, 4.0-6.0, 0.7, texWhite);

  rt->addSphere( 3.0, -1.5,  2.0-6.0, 0.7, texWhite);
  rt->addSphere( 4.0, -1.5,  5.0-6.0, 0.7, texPurple);
  rt->addSphere( 5.0, -1.5,  8.0-6.0, 0.7, texWhite);
  rt->addSphere( 6.0, -1.5, 11.0-6.0, 0.7, texPurple);
  rt->addSphere( 7.0, -1.5, 14.0-6.0, 0.7, texWhite);
  rt->addSphere( 8.0, -1.5, 17.0-6.0, 0.7, texPurple);

  rt->addSphere( 0, -1001.5, 0, 1000, 0x666666);
}




//---- Shutdown --------------------------------------------------------------//

void Shutdown()
{
  delete im;
  delete rt;
}



//---- Redraw ----------------------------------------------------------------//

void Redraw(HDC dc)
{
  static int frame = 0;

  im->clear( 0x000000 );

  //- Render raytraced scene

  //if(rrx == rx && rry == ry) return;

  rrx = rx;
  rry = ry;

  int i, j;

  //- Render

  rt->setLight( 1, 5, 2 );

  rt->setRotation( rx, ry );
  rt->render( );

  //  Frame rate indicator

  frame ++;

  for(j=0; j<8; j++) for(i=0; i<8; i++)
  {
    im->buf[j*width+i] = ((-(((i+j+frame)>>2)&1)) ^ 0xFFFF00 );
  }

  //  FPS based on current frame

  static int lastTime = 0;
  int time = timeGetTime();
  float fps = time > lastTime ? (1000.0 / (time-lastTime)) : 0.0;
  lastTime = time;

  static int times[10] ={0,0,0,0,0,0,0,0,0,0};
  static int timeptr = 0;

  int time10 = time-times[timeptr];

  times[timeptr] = time;
  timeptr = (timeptr+1)%10;

  char s[64];

  sprintf( s, "%d %.02f %.03f %.01f", frame, fps, 0.001 * time10, 10000.0 / time10 );
  im->drawText( 9, 1, s, 0xFFFFFF );

  //- Display

  Display(dc, im->buf, width, height);
}



//---- WinProc ---------------------------------------------------------------//

LRESULT CALLBACK WinProc(HWND hwnd, UINT uiMsg, UINT wParam, LONG lParam)
{
  HDC dc;
  PAINTSTRUCT ps;

  switch(uiMsg) {

  case WM_CREATE:
    Startup();
    SetTimer(hwnd, 1, 10, NULL);
    break;

  case WM_TIMER:
    dc = GetDC(hwnd);
    Redraw(dc);
    ReleaseDC(hwnd, dc);
    return 0L;

  case WM_PAINT:
    dc = BeginPaint(hwnd, &ps);
    Redraw(dc);
    EndPaint(hwnd, &ps);
    return 0L;

  case WM_LBUTTONDOWN:
    mouse_xstart = HIWORD(lParam);
    mouse_ystart = LOWORD(lParam);
    mouse_drag = true;
    SetCapture(hwnd);
    return 0L;

  case WM_MOUSEMOVE:
    mouse_x = HIWORD(lParam);
    mouse_y = LOWORD(lParam);

    if(mouse_drag)
    {
      ry -= (mouse_y-mouse_ystart) * 0.004;
      rx += (mouse_x-mouse_xstart) * 0.004;
      mouse_xstart = mouse_x;
      mouse_ystart = mouse_y;
    }
    return 0L;

  case WM_LBUTTONUP:
    mouse_drag = false;
    ReleaseCapture();
    return 0L;

  #ifdef FULL_WINDOW_DRAG

  case WM_NCHITTEST:
    return HTCAPTION;

  #endif

  case WM_KEYDOWN:
    if( wParam == VK_ESCAPE ) DestroyWindow( hwnd );
    return 0L;

  case WM_DESTROY:
	  Shutdown();
    KillTimer(hwnd, 1);
    PostQuitMessage(0);
    return 0L;

  }

  return DefWindowProc(hwnd, uiMsg, wParam, lParam);
}




//---- WinMain ---------------------------------------------------------------//

int PASCAL WinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance, LPSTR lpCmdLine, int nCmdShow)
{
  WNDCLASS wndclass;
  char cSwitch;
  MSG msg;
  RECT rc;

  hMainInstance = hInstance;

  if(hPrevInstance) return 1;

  //- Create window class

  wndclass.style         = CS_HREDRAW | CS_VREDRAW;
  wndclass.lpfnWndProc   = WinProc;
  wndclass.cbClsExtra    = 0;
  wndclass.cbWndExtra    = 0;
  wndclass.hInstance     = hInstance;
  wndclass.hIcon         = NULL;
  wndclass.hCursor       = LoadCursor(NULL, IDC_ARROW);
  wndclass.hbrBackground = (HBRUSH)GetStockObject(BLACK_BRUSH);
  wndclass.lpszMenuName  = NULL;
  wndclass.lpszClassName = szAppName;

  if(!RegisterClass(&wndclass)) return 1;

  //- Create main window

  GetWindowRect(GetDesktopWindow(), &rc);

  hMainWindow = CreateWindowEx(
    0,
    szAppName,
    szAppName,
    WS_POPUP,
    (rc.right-width)/2,
    (rc.bottom-height)/2,
    width,
    height,
    NULL,
    NULL,
    hInstance,
    NULL
  );

  if(!hMainWindow) return 1;

  ShowWindow(hMainWindow, SW_SHOW);

  //- Message loop

  while(TRUE) {
    while(PeekMessage(&msg, NULL, 0, 0, PM_REMOVE)) {
      if (msg.message == WM_QUIT) return TRUE;
      TranslateMessage(&msg);
      DispatchMessage(&msg);
    }
    HDC dc = GetDC(hMainWindow);
    Redraw(dc);
    ReleaseDC(hMainWindow, dc);
  }

  return 0;
}



//---- END OF FILE -----------------------------------------------------------//