Src/Win32/Dasher.cpp

   1
   2 #include "Common\WinCommon.h"          // must include pch first
   3
   4 #include "../DasherCore/ModuleManager.h"
   5
   6 #include "Dasher.h"
   7 #include "../DasherCore/Event.h"
   8 #include "Common\WinUTF8.h"
   9 #include "Widgets/Canvas.h"
  10 #include "DasherMouseInput.h"
  11
  12 #ifndef _WIN32_WCE
  13 #include "Sockets/SocketInput.h"
  14 #include "BTSocketInput.h"
  15 #endif
  16
  17 #include "Common/WinOptions.h"
  18
  19 #ifndef _WIN32_WCE
  20 #include <sys/stat.h>
  21 #endif
  22
  23 using namespace std;
  24 using namespace Dasher;
  25 using namespace WinUTF8;
  26
  27 // Used to signal our message loop to do our periodic work, the value
  28 // shouldn't collide with anything else in our code.
  29 #define WM_DASHER_TIMER WM_USER + 128
  30
  31 CONST UINT WM_DASHER_EVENT = RegisterWindowMessage(_WM_DASHER_EVENT);
  32 CONST UINT WM_DASHER_FOCUS = RegisterWindowMessage(_WM_DASHER_FOCUS);
  33 CONST UINT WM_DASHER_GAME_MESSAGE = RegisterWindowMessage(_WM_DASHER_GAME_MESSAGE);
  34
  35 CDasher::CDasher(HWND Parent):m_hParent(Parent) {
  36   // This class will be a wrapper for the Dasher 'control' - think ActiveX
  37   m_pEdit = 0;
  38
  39 #ifndef _WIN32_WCE
  40   // Set up COM for the accessibility stuff
  41   CoInitialize(NULL);
  42 #endif
  43
  44   Realize();
  45 }
  46
  47 CDasher::~CDasher(void) {
  48 //  WriteTrainFileFull();
  49   delete m_pCanvas;
  50 }
  51
  52 void CDasher::CreateLocalFactories() {
  53 #ifndef _WIN32_WCE
  54   RegisterModule(new CSocketInput(m_pEventHandler, m_pSettingsStore));
  55   RegisterModule(new CBTSocketInput(m_pEventHandler, m_pSettingsStore));
  56 #endif
  57   RegisterModule(new CDasherMouseInput(m_pEventHandler, m_pSettingsStore, m_pCanvas->getwindow()));
  58 }
  59
  60 void CDasher::Main() {
  61   if(m_pCanvas) {
  62                 m_pCanvas->DoFrame();
  63   }
  64
  65   DWORD dwTicks = GetTickCount();
  66         NewFrame(dwTicks, false);
  67 }
  68
  69 // Handles the work we need to do periodically on a timer event
  70 void CDasher::Log() {
  71
  72   CUserLogBase* pUserLog = GetUserLogPtr();
  73
  74         // We'll use this timer event to periodically log the user's mouse position
  75         if ((pUserLog != NULL) && (m_pCanvas != NULL)) {
  76         // Get the mouse x and y coordinates
  77         POINT sMousePos;
  78         GetCursorPos(&sMousePos);
  79
  80         // Since the everything is in screen relative coordinates, we'll
  81         // make sure we have the right coordinates for our canvas and
  82         // screen since the user may have move the window around.
  83         int     iTop     = 0;
  84         int     iLeft    = 0;
  85         int     iBottom  = 0;
  86         int     iRight   = 0;
  87
  88         if (m_pCanvas->GetCanvasSize(iTop, iLeft, iBottom, iRight))
  89           pUserLog->AddCanvasSize(iTop, iLeft, iBottom, iRight);
  90
  91         // Also update the size of the window in the UserLogTrial object
  92         if (GetWindowSize(&iTop, &iLeft, &iBottom, &iRight))
  93           pUserLog->AddWindowSize(iTop, iLeft, iBottom, iRight);
  94
  95         // We'll store a normalized version so if the user changes the window
  96         // size during a trial, it won't effect our coordinates.  The
  97         // normalization is with respect to the canvas and not the main
  98         // window.
  99         pUserLog->AddMouseLocationNormalized(sMousePos.x,
 100                                              sMousePos.y,
 101                                              true,
 102                                              (float) GetNats());
 103     }
 104
 105 }
 106
 107 void Dasher::CDasher::ExternalEventHandler(CEvent* pEvent) {
 108   SendMessage(m_hParent, WM_DASHER_EVENT, 0, (LPARAM)pEvent);
 109 }
 110
 111 void Dasher::CDasher::GameMessageOut(int message, const void *messagedata)
 112 {
 113   SendMessage(m_hParent, WM_DASHER_GAME_MESSAGE, (WPARAM)message, (LPARAM)messagedata);
 114 }
 115
 116 // Gets the size of the window in screen coordinates.
 117 bool Dasher::CDasher::GetWindowSize(int* pTop, int* pLeft, int* pBottom, int* pRight) {
 118         if ((pTop == NULL) || (pLeft == NULL) || (pBottom == NULL) || (pRight == NULL))
 119                 return false;
 120
 121         RECT sWindowRect;
 122         if (GetWindowRect(m_hParent, &sWindowRect))
 123   {
 124     *pTop    = sWindowRect.top;
 125     *pLeft   = sWindowRect.left;
 126     *pBottom = sWindowRect.bottom;
 127     *pRight  = sWindowRect.right;
 128     return true;
 129   }
 130   else
 131     return false;
 132 }
 133
 134 void Dasher::CDasher::SetEdit(CDashEditbox * pEdit) {
 135   // FIXME - we really need to make sure we have a
 136   // more sensible way of passing messages out here.
 137
 138   m_pEdit = pEdit;
 139 }
 140
 141 void Dasher::CDasher::WriteTrainFile(const std::string &strNewText) {
 142   const std::string TrainFile = GetStringParameter(SP_USER_LOC) + GetStringParameter(SP_TRAIN_FILE);
 143
 144   if(strNewText.size() == 0)
 145     return;
 146
 147   Tstring TTrainFile;
 148   UTF8string_to_wstring(TrainFile, TTrainFile);
 149
 150   HANDLE hFile = CreateFile(TTrainFile.c_str(),
 151                             GENERIC_WRITE, 0, NULL,
 152                             OPEN_ALWAYS,
 153                             FILE_ATTRIBUTE_NORMAL, 0);
 154
 155   if(hFile == INVALID_HANDLE_VALUE) {
 156     OutputDebugString(TEXT("Can not open file\n"));
 157   }
 158   else {
 159     DWORD NumberOfBytesWritten;
 160     SetFilePointer(hFile, 0, NULL, FILE_END);
 161
 162   //// Surely there are better ways to write to files than this??
 163
 164     for(unsigned int i = 0; i < strNewText.size(); i++) {
 165       WriteFile(hFile, &strNewText[i], 1, &NumberOfBytesWritten, NULL);
 166     }
 167
 168     CloseHandle(hFile);
 169   }
 170 }
 171
 172 void CDasher::ScanDirectory(const Tstring &strMask, std::vector<std::string> &vFileList) {
 173   using namespace WinUTF8;
 174
 175   std::string filename;
 176   WIN32_FIND_DATA find;
 177   HANDLE handle;
 178
 179   handle = FindFirstFile(strMask.c_str(), &find);
 180   if(handle != INVALID_HANDLE_VALUE) {
 181     wstring_to_UTF8string(wstring(find.cFileName), filename);
 182     vFileList.push_back(filename);
 183     while(FindNextFile(handle, &find) != false) {
 184       wstring_to_UTF8string(wstring(find.cFileName), filename);
 185       vFileList.push_back(filename);
 186     }
 187     FindClose(handle);
 188   }
 189 }
 190
 191 void CDasher::ScanColourFiles(std::vector<std::string> &vFileList) {
 192   Tstring Colours;
 193
 194   // TODO: Is it okay to have duplicate names in the array?
 195   std::string strAppData2(GetStringParameter(SP_SYSTEM_LOC));
 196   Tstring strAppData;
 197
 198   WinUTF8::UTF8string_to_wstring(strAppData2, strAppData);
 199
 200   Colours = strAppData;
 201   Colours += TEXT("colour*.xml");
 202   ScanDirectory(Colours, vFileList);
 203
 204   std::string strUserData2(GetStringParameter(SP_USER_LOC));
 205   Tstring strUserData;
 206
 207   WinUTF8::UTF8string_to_wstring(strUserData2, strUserData);
 208
 209   Colours = strUserData;
 210   Colours += TEXT("colour*.xml");
 211   ScanDirectory(Colours, vFileList);
 212 }
 213
 214 void CDasher::ScanAlphabetFiles(std::vector<std::string> &vFileList) {
 215   Tstring Alphabets;
 216
 217   // TODO: Is it okay to have duplicate names in the array?
 218   std::string strAppData2(GetStringParameter(SP_SYSTEM_LOC));
 219   Tstring strAppData;
 220
 221   WinUTF8::UTF8string_to_wstring(strAppData2, strAppData);
 222
 223   Alphabets = strAppData;
 224   Alphabets += TEXT("alphabet*.xml");
 225   ScanDirectory(Alphabets, vFileList);
 226
 227   std::string strUserData2(GetStringParameter(SP_USER_LOC));
 228   Tstring strUserData;
 229
 230   WinUTF8::UTF8string_to_wstring(strUserData2, strUserData);
 231
 232   Alphabets = strUserData;
 233   Alphabets += TEXT("alphabet*.xml");
 234   ScanDirectory(Alphabets, vFileList);
 235 }
 236
 237 void CDasher::SetupPaths() {
 238   using namespace WinHelper;
 239   using namespace WinUTF8;
 240
 241   Tstring UserData, AppData;
 242   std::string UserData2, AppData2;
 243   GetUserDirectory(&UserData);
 244   GetAppDirectory(&AppData);
 245   UserData += TEXT("dasher.rc\\");
 246   AppData += TEXT("system.rc\\");
 247   CreateDirectory(UserData.c_str(), NULL);      // Try and create folders. Doesn't seem
 248   CreateDirectory(AppData.c_str(), NULL);       // to do any harm if they already exist.
 249   wstring_to_UTF8string(UserData, UserData2);   // TODO: I don't know if special characters will work.
 250   wstring_to_UTF8string(AppData, AppData2);     // ASCII-only filenames are safest. Being English doesn't help debug this...
 251   SetStringParameter(SP_SYSTEM_LOC, AppData2);
 252   SetStringParameter(SP_USER_LOC, UserData2);
 253 }
 254
 255 void CDasher::SetupUI() {
 256   m_pCanvas = new CCanvas(this, m_pEventHandler, m_pSettingsStore);
 257   m_pCanvas->Create(m_hParent); // TODO - check return
 258
 259   OnUIRealised();
 260 }
 261
 262 int CDasher::GetFileSize(const std::string &strFileName) {
 263 #ifndef _WIN32_WCE
 264   struct _stat sStatInfo;
 265   _stat(strFileName.c_str(), &sStatInfo);
 266   return sStatInfo.st_size;
 267 #else
 268   // TODO: Fix this on Win CE
 269   return 0;
 270 #endif
 271 }
 272
 273 void CDasher::CreateSettingsStore(void) {
 274   m_pSettingsStore = new CWinOptions( "Inference Group", "Dasher3", m_pEventHandler );
 275 }
 276
 277 void CDasher::StartTimer() {
 278   // TODO: See MessageLoop, Main in CDasherWindow - should be brought into this class
 279   // Framerate settings: currently 40fps.
 280   SetTimer(m_pCanvas->getwindow(), 1, 25, NULL);
 281 }
 282
 283 void CDasher::ShutdownTimer() {
 284 }
 285
 286 // TODO: Check that syntax here is sensible
 287 void CDasher::Move(int iX, int iY, int iWidth, int iHeight) {
 288   if(m_pCanvas)
 289     m_pCanvas->Move(iX, iY, iWidth, iHeight);
 290 }
 291
 292 void CDasher::TakeFocus() {
 293   // TODO: Implement me
 294 }