-
Notifications
You must be signed in to change notification settings - Fork 8
/
irsdk_utils.cpp
368 lines (307 loc) · 8.47 KB
/
irsdk_utils.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
/*
Copyright (c) 2013, iRacing.com Motorsport Simulations, LLC.
All rights reserved.
Redistribution and use in source and binary forms, with or without
modification, are permitted provided that the following conditions are met:
* Redistributions of source code must retain the above copyright
notice, this list of conditions and the following disclaimer.
* Redistributions in binary form must reproduce the above copyright
notice, this list of conditions and the following disclaimer in the
documentation and/or other materials provided with the distribution.
* Neither the name of iRacing.com Motorsport Simulations nor the
names of its contributors may be used to endorse or promote products
derived from this software without specific prior written permission.
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR
ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
(INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*/
#define MIN_WIN_VER 0x0501
#ifndef WINVER
# define WINVER MIN_WIN_VER
#endif
#ifndef _WIN32_WINNT
# define _WIN32_WINNT MIN_WIN_VER
#endif
#include <windows.h>
#include <stdio.h>
#include <time.h>
#include <limits.h>
#ifdef _MSC_VER
#include <crtdbg.h>
#endif
#include "irsdk_defines.h"
// for timeBeginPeriod()
#pragma comment(lib, "Winmm")
// for RegisterWindowMessage() and SendMessage()
#pragma comment(lib, "User32")
// Local memory
static HANDLE hDataValidEvent = NULL;
static HANDLE hMemMapFile = NULL;
static const char *pSharedMem = NULL;
static const irsdk_header *pHeader = NULL;
static int lastTickCount = INT_MAX;
static bool isInitialized = false;
static const double timeout = 30.0; // timeout after 30 seconds with no communication
static time_t lastValidTime = 0;
// Function Implementations
bool irsdk_startup()
{
if(!hMemMapFile)
{
hMemMapFile = OpenFileMapping( FILE_MAP_READ, FALSE, IRSDK_MEMMAPFILENAME);
lastTickCount = INT_MAX;
}
if(hMemMapFile)
{
if(!pSharedMem)
{
pSharedMem = (const char *)MapViewOfFile(hMemMapFile, FILE_MAP_READ, 0, 0, 0);
pHeader = (irsdk_header *)pSharedMem;
lastTickCount = INT_MAX;
}
if(pSharedMem)
{
if(!hDataValidEvent)
{
hDataValidEvent = OpenEvent(SYNCHRONIZE, false, IRSDK_DATAVALIDEVENTNAME);
lastTickCount = INT_MAX;
}
if(hDataValidEvent)
{
isInitialized = true;
return isInitialized;
}
//else printf("Error opening event: %d\n", GetLastError());
}
//else printf("Error mapping file: %d\n", GetLastError());
}
//else printf("Error opening file: %d\n", GetLastError());
isInitialized = false;
return isInitialized;
}
void irsdk_shutdown()
{
if(hDataValidEvent)
CloseHandle(hDataValidEvent);
if(pSharedMem)
UnmapViewOfFile(pSharedMem);
if(hMemMapFile)
CloseHandle(hMemMapFile);
hDataValidEvent = NULL;
pSharedMem = NULL;
pHeader = NULL;
hMemMapFile = NULL;
isInitialized = false;
lastTickCount = INT_MAX;
}
bool irsdk_getNewData(char *data)
{
if(isInitialized || irsdk_startup())
{
#ifdef _MSC_VER
_ASSERTE(NULL != pHeader);
#endif
// if sim is not active, then no new data
if(!(pHeader->status & irsdk_stConnected))
{
lastTickCount = INT_MAX;
return false;
}
int latest = 0;
for(int i=1; i<pHeader->numBuf; i++)
if(pHeader->varBuf[latest].tickCount < pHeader->varBuf[i].tickCount)
latest = i;
// if newer than last recieved, than report new data
if(lastTickCount < pHeader->varBuf[latest].tickCount)
{
// if asked to retrieve the data
if(data)
{
// try twice to get the data out
for(int count = 0; count < 2; count++)
{
int curTickCount = pHeader->varBuf[latest].tickCount;
memcpy(data, pSharedMem + pHeader->varBuf[latest].bufOffset, pHeader->bufLen);
if(curTickCount == pHeader->varBuf[latest].tickCount)
{
lastTickCount = curTickCount;
lastValidTime = time(NULL);
return true;
}
}
// if here, the data changed out from under us.
return false;
}
else
{
lastTickCount = pHeader->varBuf[latest].tickCount;
lastValidTime = time(NULL);
return true;
}
}
// if older than last recieved, than reset, we probably disconnected
else if(lastTickCount > pHeader->varBuf[latest].tickCount)
{
lastTickCount = pHeader->varBuf[latest].tickCount;
return false;
}
// else the same, and nothing changed this tick
}
return false;
}
bool irsdk_waitForDataReady(int timeOut, char *data)
{
#ifdef _MSC_VER
_ASSERTE(timeOut >= 0);
#endif
if(isInitialized || irsdk_startup())
{
// just to be sure, check before we sleep
if(irsdk_getNewData(data))
return true;
// sleep till signaled
WaitForSingleObject(hDataValidEvent, timeOut);
// we woke up, so check for data
if(irsdk_getNewData(data))
return true;
else
return false;
}
// sleep if error
if(timeOut > 0)
Sleep(timeOut);
return false;
}
bool irsdk_isConnected()
{
if(isInitialized)
{
int elapsed = (int)difftime(time(NULL), lastValidTime);
return (pHeader->status & irsdk_stConnected) > 0 && elapsed < timeout;
}
return false;
}
const irsdk_header *irsdk_getHeader()
{
if(isInitialized)
{
return pHeader;
}
return NULL;
}
// direct access to the data buffer
// Warnign! This buffer is volitile so read it out fast!
// Use the cached copy from irsdk_waitForDataReady() or irsdk_getNewData() instead
const char *irsdk_getData(int index)
{
if(isInitialized)
{
return pSharedMem + pHeader->varBuf[index].bufOffset;
}
return NULL;
}
const char *irsdk_getSessionInfoStr()
{
if(isInitialized)
{
return pSharedMem + pHeader->sessionInfoOffset;
}
return NULL;
}
const irsdk_varHeader *irsdk_getVarHeaderPtr()
{
if(isInitialized)
{
return ((irsdk_varHeader*)(pSharedMem + pHeader->varHeaderOffset));
}
return NULL;
}
const irsdk_varHeader *irsdk_getVarHeaderEntry(int index)
{
if(isInitialized)
{
if(index >= 0 && index < pHeader->numVars)
{
return &((irsdk_varHeader*)(pSharedMem + pHeader->varHeaderOffset))[index];
}
}
return NULL;
}
// Note: this is a linear search, so cache the results
int irsdk_varNameToIndex(const char *name)
{
const irsdk_varHeader *pVar;
if(name)
{
for(int index=0; index<pHeader->numVars; index++)
{
pVar = irsdk_getVarHeaderEntry(index);
if(pVar && 0 == strncmp(name, pVar->name, IRSDK_MAX_STRING))
{
return index;
}
}
}
return -1;
}
int irsdk_varNameToOffset(const char *name)
{
const irsdk_varHeader *pVar;
if(name)
{
for(int index=0; index<pHeader->numVars; index++)
{
pVar = irsdk_getVarHeaderEntry(index);
if(pVar && 0 == strncmp(name, pVar->name, IRSDK_MAX_STRING))
{
return pVar->offset;
}
}
}
return -1;
}
unsigned int irsdk_getBroadcastMsgID()
{
static unsigned int msgId = RegisterWindowMessage(IRSDK_BROADCASTMSGNAME);
return msgId;
}
void irsdk_broadcastMsg(irsdk_BroadcastMsg msg, int var1, int var2, int var3)
{
irsdk_broadcastMsg(msg, var1, MAKELONG(var2, var3));
}
void irsdk_broadcastMsg(irsdk_BroadcastMsg msg, int var1, float var2)
{
// multiply by 2^16-1 to move fractional part to the integer part
int real = (int)(var2 * 65536.0f);
irsdk_broadcastMsg(msg, var1, real);
}
void irsdk_broadcastMsg(irsdk_BroadcastMsg msg, int var1, int var2)
{
static unsigned int msgId = irsdk_getBroadcastMsgID();
if(msgId && msg >= 0 && msg < irsdk_BroadcastLast)
{
SendNotifyMessage(HWND_BROADCAST, msgId, MAKELONG(msg, var1), var2);
}
}
int irsdk_padCarNum(int num, int zero)
{
int retVal = num;
int numPlace = 1;
if(num > 99)
numPlace = 3;
else if(num > 9)
numPlace = 2;
if(zero)
{
numPlace += zero;
retVal = num + 1000*numPlace;
}
return retVal;
}