diff --git a/src/CMakeLists.txt b/src/CMakeLists.txt index fbc34173390a1be01be3058772629779558f5c4a..30b342c66e0d308170de6cc6ae89ad644a8f90c2 100644 --- a/src/CMakeLists.txt +++ b/src/CMakeLists.txt @@ -104,6 +104,7 @@ add_executable(SRB2SDL2 MACOSX_BUNDLE WIN32 lua_hudlib_drawlist.c lua_colorlib.c lua_inputlib.c + lua_interceptlib.c ) # This updates the modification time for comptime.c at the diff --git a/src/Sourcefile b/src/Sourcefile index 60ee5db5baf16fe20657c93e5143afe60615bc89..3e926e31850a2055d335326096de6034a87b0a69 100644 --- a/src/Sourcefile +++ b/src/Sourcefile @@ -98,3 +98,4 @@ lua_hudlib.c lua_hudlib_drawlist.c lua_inputlib.c lua_colorlib.c +lua_interceptlib.c diff --git a/src/console.c b/src/console.c index 2eb5b26c3afeae0459a9d4b06aa79274185628f8..29510435ae445396fd9bf3de25bec87d48dc5c09 100644 --- a/src/console.c +++ b/src/console.c @@ -34,6 +34,7 @@ #include "m_menu.h" #include "filesrch.h" #include "m_misc.h" +#include "lua_libs.h" #ifdef _WINDOWS #include "win32/win_main.h" @@ -918,6 +919,22 @@ static void CON_InputDelChar(void) // ---- // +// +// Same as CON_Responder, but is process before everything else, so it cannot be blocked. +// +boolean CON_PreResponder(event_t *ev) +{ + if (ev->type == ev_keydown && shiftdown == 1 && ev->key == KEY_ESCAPE) + { + I_SetTextInputMode(con_destlines == 0 ? true : textinputmodeenabledbylua); // inverse, since this is changed next tic. + consoletoggle = true; + return true; + } + + return false; +} + +// // Handles console key input // boolean CON_Responder(event_t *ev) @@ -957,7 +974,7 @@ boolean CON_Responder(event_t *ev) if (con_destlines == 0 && I_GetTextInputMode()) return false; // some other component is holding keyboard input, don't hijack it! - I_SetTextInputMode(con_destlines == 0); // inverse, since this is changed next tic. + I_SetTextInputMode(con_destlines == 0 ? true : textinputmodeenabledbylua); // inverse, since this is changed next tic. consoletoggle = true; return true; } @@ -977,7 +994,7 @@ boolean CON_Responder(event_t *ev) // escape key toggle off console if (key == KEY_ESCAPE) { - I_SetTextInputMode(false); + I_SetTextInputMode(textinputmodeenabledbylua); consoletoggle = true; return true; } diff --git a/src/console.h b/src/console.h index 2af01f1a3af8bdd50901eabaa5b301107741e213..664b74a223826ee755078f56a046529c553c15b4 100644 --- a/src/console.h +++ b/src/console.h @@ -19,6 +19,7 @@ void CON_Init(void); void CON_StartRefresh(void); void CON_StopRefresh(void); +boolean CON_PreResponder(event_t *ev); boolean CON_Responder(event_t *ev); #ifdef HAVE_THREADS diff --git a/src/d_main.c b/src/d_main.c index d75a4d5013d9a92e7abccad7211721b3fa1f00b5..2b8d4326a0868fe2cbced6bc40274b96d2598c95 100644 --- a/src/d_main.c +++ b/src/d_main.c @@ -227,6 +227,9 @@ void D_ProcessEvents(void) } } + if (CON_PreResponder(ev)) + continue; + // Screenshots over everything so that they can be taken anywhere. if (M_ScreenshotResponder(ev)) continue; // ate the event diff --git a/src/deh_tables.c b/src/deh_tables.c index a96adf8ae6fb432bf04fd472d525eec1b6dcbd2a..57a243f5b737f8ab91c60414af09f0d5eb176ae3 100644 --- a/src/deh_tables.c +++ b/src/deh_tables.c @@ -25,6 +25,7 @@ #include "g_game.h" // Joystick axes (for lua) #include "i_joy.h" #include "g_input.h" // Game controls (for lua) +#include "p_maputl.h" // P_PathTraverse constants (for lua) #include "deh_tables.h" @@ -5829,6 +5830,11 @@ struct int_const_s const INT_CONST[] = { {"MB_BUTTON8",MB_BUTTON8}, {"MB_SCROLLUP",MB_SCROLLUP}, {"MB_SCROLLDOWN",MB_SCROLLDOWN}, + + // P_PathTraverse constants + {"PT_ADDLINES",PT_ADDLINES}, + {"PT_ADDTHINGS",PT_ADDTHINGS}, + {"PT_EARLYOUT",PT_EARLYOUT}, // screen.h constants {"BASEVIDWIDTH",BASEVIDWIDTH}, diff --git a/src/f_finale.c b/src/f_finale.c index c6d957292138915a6def4b975648e81e708cfd5a..49ff820d5a524cff7cf97bb9c1ef03addf002046 100644 --- a/src/f_finale.c +++ b/src/f_finale.c @@ -4589,9 +4589,9 @@ void F_TextPromptDrawer(void) players[j].powers[pw_nocontrol] = 1;\ if (players[j].mo)\ {\ - if (players[j].mo->state == states+S_PLAY_STND && players[j].mo->tics != -1)\ + if (P_IsPlayerInState(&players[j], S_PLAY_STND) && players[j].mo->tics != -1)\ players[j].mo->tics++;\ - else if (players[j].mo->state == states+S_PLAY_WAIT)\ + else if (P_IsPlayerInState(&players[j], S_PLAY_WAIT))\ P_SetMobjState(players[j].mo, S_PLAY_STND);\ }\ } diff --git a/src/hardware/hw_draw.c b/src/hardware/hw_draw.c index ab253d5c80f8dc1b1a958fabbf607c224746c4a0..152501f60e1dcb0ee04eee6e100fdfbe3878eb10 100644 --- a/src/hardware/hw_draw.c +++ b/src/hardware/hw_draw.c @@ -598,43 +598,76 @@ void HWR_DrawCroppedPatch(patch_t *gpatch, fixed_t x, fixed_t y, fixed_t pscale, // -------------------------------------------------------------------------- // Fills a box of pixels using a flat texture as a pattern +// Fixed to properly align like the other draw functions -luigi budd // -------------------------------------------------------------------------- -void HWR_DrawFlatFill (INT32 x, INT32 y, INT32 w, INT32 h, lumpnum_t flatlumpnum) +void HWR_DrawFlatFill(INT32 x, INT32 y, INT32 w, INT32 h, lumpnum_t flatlumpnum) { FOutVector v[4]; const size_t len = W_LumpLength(flatlumpnum); - UINT16 flatflag = R_GetFlatSize(len) - 1; - double dflatsize = (double)(flatflag + 1); + UINT16 flatflag = R_GetFlatSize(len); + double dflatsize = (double)(flatflag); + + // compilers are COOL! + float dup = (float)vid.dup; + float fx = (float)x * dup; + float fy = (float)y * dup; + float fw = (float)w * dup; + float fh = (float)h * dup; + + /* + fx *= dup; + fy *= dup; + fw *= dup; + fh *= dup; + */ + if (fw <= 0 || fh <= 0) + return; + + if (fabsf((float)vid.width - (float)BASEVIDWIDTH * vid.dup) > 1.0E-36f) + { + fx += ((float)vid.width - ((float)BASEVIDWIDTH * vid.dup)) / 2; + } + if (fabsf((float)vid.height - (float)BASEVIDHEIGHT * vid.dup) > 1.0E-36f) + { + fy += ((float)vid.height - ((float)BASEVIDHEIGHT * vid.dup)) / 2; + } + + if (fx >= vid.width || fy >= vid.height) + return; + + fx = -1.0f + (fx / (vid.width / 2.0f)); + fy = 1.0f - (fy / (vid.height / 2.0f)); + fw /= vid.width / 2; + fh /= vid.height / 2; + // 3--2 // | /| // |/ | // 0--1 + + // position vertices + v[0].x = v[3].x = fx; + v[2].x = v[1].x = fx + fw; - v[0].x = v[3].x = (x - 160.0f)/160.0f; - v[2].x = v[1].x = ((x+w) - 160.0f)/160.0f; - v[0].y = v[1].y = -(y - 100.0f)/100.0f; - v[2].y = v[3].y = -((y+h) - 100.0f)/100.0f; + v[0].y = v[1].y = fy; + v[2].y = v[3].y = fy - fh; v[0].z = v[1].z = v[2].z = v[3].z = 1.0f; - v[0].s = v[3].s = (float)((x & flatflag)/dflatsize); - v[2].s = v[1].s = (float)(v[0].s + w/dflatsize); - v[0].t = v[1].t = (float)((y & flatflag)/dflatsize); - v[2].t = v[3].t = (float)(v[0].t + h/dflatsize); + // sides + v[0].s = v[3].s = (float)(flatflag/dflatsize) * 2; + v[2].s = v[1].s = (float)(v[0].s + w/dflatsize) * 2; - HWR_GetRawFlat(flatlumpnum); + // top/bottom + v[0].t = v[1].t = (float)(flatflag/dflatsize) * 2; + v[2].t = v[3].t = (float)(v[0].t + h/dflatsize) * 2; - //Hurdler: Boris, the same comment as above... but maybe for pics - // it not a problem since they don't have any transparent pixel - // if I'm right !? - // BTW, I see we put 0 for PFs, and If I'm right, that - // means we take the previous PFs as default - // how can we be sure they are ok? + // needed to texture the poly + HWR_GetRawFlat(flatlumpnum); HWD.pfnDrawPolygon(NULL, v, 4, PF_NoDepthTest); //PF_Translucent); } - // -------------------------------------------------------------------------- // Fade down the screen so that the menu drawn on top of it looks brighter // -------------------------------------------------------------------------- diff --git a/src/hu_stuff.c b/src/hu_stuff.c index 7f2560bcc4adc2f398d87aa8066119ee9ec3b455..e8796286660188ba70eef269de751a1f0437bb1e 100644 --- a/src/hu_stuff.c +++ b/src/hu_stuff.c @@ -52,6 +52,7 @@ #include "lua_hud.h" #include "lua_hudlib_drawlist.h" #include "lua_hook.h" +#include "lua_libs.h" // coords are scaled #define HU_INPUTX 0 @@ -395,8 +396,11 @@ void HU_AddChatText(const char *text, boolean playsound) if (OLDCHAT) // if we're using oldchat, print directly in console CONS_Printf("%s\n", text); - else // if we aren't, still save the message to log.txt - CON_LogMessage(va("%s\n", text)); + else // if we aren't, still save the message to log.txt + { + CON_LogMessage(text); + CON_LogMessage("\n"); // Add newline. Don't use va for that, since `text` might be refering to va's buffer itself + } } /** Runs a say command, sending an ::XD_SAY message. @@ -1013,7 +1017,7 @@ static void HU_sendChatMessage(void) void HU_clearChatChars(void) { memset(w_chat, '\0', sizeof(w_chat)); - I_SetTextInputMode(false); + I_SetTextInputMode(textinputmodeenabledbylua); chat_on = false; c_input = 0; @@ -1150,7 +1154,7 @@ boolean HU_Responder(event_t *ev) if (!CHAT_MUTE) HU_sendChatMessage(); - I_SetTextInputMode(false); + I_SetTextInputMode(textinputmodeenabledbylua); chat_on = false; c_input = 0; // reset input cursor chat_scrollmedown = true; // you hit enter, so you might wanna autoscroll to see what you just sent. :) @@ -1161,7 +1165,7 @@ boolean HU_Responder(event_t *ev) || c == gamecontrol[GC_TEAMKEY][0] || c == gamecontrol[GC_TEAMKEY][1]) && c >= KEY_MOUSE1)) // If it's not a keyboard key, then the chat button is used as a toggle. { - I_SetTextInputMode(false); + I_SetTextInputMode(textinputmodeenabledbylua); chat_on = false; c_input = 0; // reset input cursor I_UpdateMouseGrab(); diff --git a/src/info.c b/src/info.c index 3998eb88d2b976fc4d718506a4e6ea687c04e49e..9f1224ae2618b3a29adb273075ba3834366d939f 100644 --- a/src/info.c +++ b/src/info.c @@ -4518,7 +4518,7 @@ mobjinfo_t mobjinfo[NUMMOBJTYPES] = sfx_pop, // deathsound 0, // speed 12*FRACUNIT, // radius - 64*FRACUNIT, // height + 32*FRACUNIT, // height 0, // display offset 100, // mass 0, // damage diff --git a/src/lua_baselib.c b/src/lua_baselib.c index 4012083606f73cf1d1965a1e244135a83c9331ce..bb6327eb13c5d32de6072ce98850178f9dee85a1 100644 --- a/src/lua_baselib.c +++ b/src/lua_baselib.c @@ -240,6 +240,8 @@ static const struct { {META_KEYEVENT, "keyevent_t"}, {META_TEXTEVENT, "textevent_t"}, {META_MOUSE, "mouse_t"}, + + {META_INTERCEPT, "intercept_t"}, {NULL, NULL} }; diff --git a/src/lua_hudlib.c b/src/lua_hudlib.c index 1bb4de180eb6521cb1cb428d9c6b090f6d7f21ef..55029828035e99ddb69c22e9c657dc9ccffbb9e3 100644 --- a/src/lua_hudlib.c +++ b/src/lua_hudlib.c @@ -467,7 +467,23 @@ static int libd_patchExists(lua_State *L) static int libd_cachePatch(lua_State *L) { - LUA_PushUserdata(L, W_CachePatchLongName(luaL_checkstring(L, 1), PU_PATCH), META_PATCH); + const char *name = luaL_checkstring(L, 1); + patch_t *patch = W_CachePatchLongName(name, PU_PATCH); + +#ifdef ROTSPRITE + if (lua_isnumber(L, 2)) + { + angle_t rollangle = luaL_checkangle(L, 2); + INT32 rot = R_GetRollAngle(rollangle); + if (rot) { + patch_t *rotpatch = Patch_GetRotated(patch, rot, false); + LUA_PushUserdata(L, rotpatch, META_PATCH); + return 1; + } + } +#endif + + LUA_PushUserdata(L, patch, META_PATCH); return 1; } diff --git a/src/lua_infolib.c b/src/lua_infolib.c index a65ee23ebc913a62711e9960990c5935662b8dfd..f277f2651dcb0d858b50c1b5bb461de4a3658f62 100644 --- a/src/lua_infolib.c +++ b/src/lua_infolib.c @@ -1158,6 +1158,7 @@ enum mobjinfo_e mobjinfo_activesound, mobjinfo_flags, mobjinfo_raisestate, + mobjinfo_name, }; const char *const mobjinfo_opt[] = { @@ -1185,6 +1186,7 @@ const char *const mobjinfo_opt[] = { "activesound", "flags", "raisestate", + "name", NULL, }; @@ -1199,6 +1201,8 @@ static int mobjinfo_get(lua_State *L) I_Assert(info != NULL); I_Assert(info >= mobjinfo); + mobjtype_t id = info-mobjinfo; + switch (field) { case mobjinfo_doomednum: @@ -1273,6 +1277,21 @@ static int mobjinfo_get(lua_State *L) case mobjinfo_raisestate: lua_pushinteger(L, info->raisestate); break; + case mobjinfo_name: + if (id < MT_FIRSTFREESLOT) + { + lua_pushstring(L, MOBJTYPE_LIST[id]+3); + return 1; + } + + id -= MT_FIRSTFREESLOT; + if (id < NUMMOBJFREESLOTS && FREE_MOBJS[id]) + { + lua_pushstring(L, FREE_MOBJS[id]); + return 1; + } + + return 0; default: lua_getfield(L, LUA_REGISTRYINDEX, LREG_EXTVARS); I_Assert(lua_istable(L, -1)); diff --git a/src/lua_inputlib.c b/src/lua_inputlib.c index eb9fe7dc55a5947b6d5594f01fcda962ccfa0178..d4e327c7bba3aae6a5b13bf7265104307613df05 100644 --- a/src/lua_inputlib.c +++ b/src/lua_inputlib.c @@ -10,16 +10,19 @@ /// \brief input library for Lua scripting #include "doomdef.h" +#include "doomstat.h" #include "fastcmp.h" #include "g_input.h" #include "g_game.h" #include "hu_stuff.h" #include "i_system.h" +#include "console.h" #include "lua_script.h" #include "lua_libs.h" boolean mousegrabbedbylua = true; +boolean textinputmodeenabledbylua = false; boolean ignoregameinputs = false; /////////////// @@ -131,13 +134,15 @@ static int lib_getCursorPosition(lua_State *L) static int lib_setTextInputMode(lua_State *L) { - I_SetTextInputMode(luaL_checkboolean(L, 1)); + textinputmodeenabledbylua = luaL_checkboolean(L, 1); + if (!(menuactive || CON_Ready() || chat_on)) + I_SetTextInputMode(textinputmodeenabledbylua); return 0; } static int lib_getTextInputMode(lua_State *L) { - lua_pushinteger(L, I_GetTextInputMode()); + lua_pushinteger(L, textinputmodeenabledbylua); return 1; } diff --git a/src/lua_interceptlib.c b/src/lua_interceptlib.c new file mode 100644 index 0000000000000000000000000000000000000000..5907b2bf5ac909bced02fab2107f013370831e15 --- /dev/null +++ b/src/lua_interceptlib.c @@ -0,0 +1,115 @@ +// SONIC ROBO BLAST 2 +//----------------------------------------------------------------------------- +// Copyright (C) 2024-2024 by Sonic Team Junior. +// +// This program is free software distributed under the +// terms of the GNU General Public License, version 2. +// See the 'LICENSE' file for more details. +//----------------------------------------------------------------------------- +/// \file lua_interceptlib.c +/// \brief intercept library for Lua scripting + +#include "doomdef.h" +#include "fastcmp.h" +#include "p_local.h" +#include "lua_script.h" +#include "lua_libs.h" + +#include "lua_hud.h" // hud_running errors + +#define NOHUD if (hud_running)\ +return luaL_error(L, "HUD rendering code should not call this function!"); + +enum intercept_e { + intercept_valid = 0, + intercept_frac, + intercept_thing, + intercept_line +}; + +static const char *const intercept_opt[] = { + "valid", + "frac", + "thing", + "line", + NULL}; + +static int intercept_fields_ref = LUA_NOREF; + +static boolean Lua_PathTraverser(intercept_t *in) +{ + boolean traverse = false; + I_Assert(in != NULL); + + lua_settop(gL, 6); + lua_pushcfunction(gL, LUA_GetErrorMessage); + + I_Assert(lua_isfunction(gL, -2)); + + lua_pushvalue(gL, -2); + LUA_PushUserdata(gL, in, META_INTERCEPT); + LUA_Call(gL, 1, 1, -3); + + traverse = lua_toboolean(gL, -1); + lua_pop(gL, 1); + + return !traverse; // Stay consistent with the MobjMoveCollide hook +} + +static int intercept_get(lua_State *L) +{ + intercept_t *in = *((intercept_t **)luaL_checkudata(L, 1, META_INTERCEPT)); + enum intercept_e field = Lua_optoption(L, 2, intercept_valid, intercept_fields_ref); + + if (!in) + { + if (field == intercept_valid) { + lua_pushboolean(L, 0); + return 1; + } + return luaL_error(L, "accessed intercept_t doesn't exist anymore."); + } + + switch(field) + { + case intercept_valid: // valid + lua_pushboolean(L, 1); + return 1; + case intercept_frac: + lua_pushfixed(L, in->frac); + return 1; + case intercept_thing: + if (in->isaline) + return 0; + LUA_PushUserdata(L, in->d.thing, META_MOBJ); + return 1; + case intercept_line: + if (!in->isaline) + return 0; + LUA_PushUserdata(L, in->d.line, META_LINE); + return 1; + } + return 0; +} + +static int lib_pPathTraverse(lua_State *L) +{ + fixed_t px1 = luaL_checkfixed(L, 1); + fixed_t py1 = luaL_checkfixed(L, 2); + fixed_t px2 = luaL_checkfixed(L, 3); + fixed_t py2 = luaL_checkfixed(L, 4); + INT32 flags = (INT32)luaL_checkinteger(L, 5); + luaL_checktype(L, 6, LUA_TFUNCTION); + NOHUD + INLEVEL + lua_pushboolean(L, P_PathTraverse(px1, py1, px2, py2, flags, Lua_PathTraverser)); + return 1; +} + +int LUA_InterceptLib(lua_State *L) +{ + LUA_RegisterUserdataMetatable(L, META_INTERCEPT, intercept_get, NULL, NULL); + intercept_fields_ref = Lua_CreateFieldTable(L, intercept_opt); + lua_register(L, "P_PathTraverse", lib_pPathTraverse); + return 0; +} diff --git a/src/lua_libs.h b/src/lua_libs.h index 592f46df560435b085b9fe025a263d6b1317e042..381c3462955312d97b09b6b8529ff91d6e4ee27b 100644 --- a/src/lua_libs.h +++ b/src/lua_libs.h @@ -13,6 +13,7 @@ extern lua_State *gL; extern boolean mousegrabbedbylua; +extern boolean textinputmodeenabledbylua; extern boolean ignoregameinputs; #define MUTABLE_TAGS @@ -97,6 +98,8 @@ extern boolean ignoregameinputs; #define META_KEYEVENT "KEYEVENT_T*" #define META_MOUSE "MOUSE_T*" +#define META_INTERCEPT "INTERCEPT_T*" + boolean luaL_checkboolean(lua_State *L, int narg); int LUA_EnumLib(lua_State *L); @@ -117,3 +120,4 @@ int LUA_BlockmapLib(lua_State *L); int LUA_HudLib(lua_State *L); int LUA_ColorLib(lua_State *L); int LUA_InputLib(lua_State *L); +int LUA_InterceptLib(lua_State *L); diff --git a/src/lua_script.c b/src/lua_script.c index 2234241b6cd73a33643a4abb33935c215a04fd82..c22738ebd5905c35ed97dd530658a43d4be42f29 100644 --- a/src/lua_script.c +++ b/src/lua_script.c @@ -62,6 +62,7 @@ static lua_CFunction liblist[] = { LUA_HudLib, // HUD stuff LUA_ColorLib, // general color functions LUA_InputLib, // inputs + LUA_InterceptLib, // intercept_t NULL }; diff --git a/src/m_menu.c b/src/m_menu.c index e0b5793520c1cada5be10a6b909400ccb4fae4c6..adcd0e953cd38ff6f50a33f7f686ca4518395cf9 100644 --- a/src/m_menu.c +++ b/src/m_menu.c @@ -48,6 +48,7 @@ #include "p_setup.h" #include "f_finale.h" #include "lua_hook.h" +#include "lua_libs.h" #ifdef HWRENDER #include "hardware/hw_main.h" @@ -3803,6 +3804,7 @@ void M_ClearMenus(boolean callexitmenufunc) hidetitlemap = false; I_UpdateMouseGrab(); + I_SetTextInputMode(textinputmodeenabledbylua); } // @@ -11173,37 +11175,11 @@ static INT32 menuRoomIndex = 0; static void M_DrawRoomMenu(void) { - static fixed_t frame = -(12 << FRACBITS); - int dot_frame; - char text[4]; - const char *rmotd; const char *waiting_message; - int dots; - if (m_waiting_mode) - { - dot_frame = (int)(frame >> FRACBITS) / 4; - dots = dot_frame + 3; - - strcpy(text, " "); - - if (dots > 0) - { - if (dot_frame < 0) - dot_frame = 0; - - if (dot_frame != 3) - strncpy(&text[dot_frame], "...", min(dots, 3 - dot_frame)); - } - - frame += renderdeltatics; - while (frame >= (12 << FRACBITS)) - frame -= 12 << FRACBITS; - - currentMenu->menuitems[0].text = text; - } + currentMenu->menuitems[0].text = "..."; // use generic drawer for cursor, items and title M_DrawGenericMenu(); diff --git a/src/m_misc.c b/src/m_misc.c index 24616e9db42e2b27bab8c4d359e5d92b3945f6d7..f8087e8c2b920a44b5642e7ee8d9fb30c72ab62d 100644 --- a/src/m_misc.c +++ b/src/m_misc.c @@ -1720,7 +1720,7 @@ char *va(const char *format, ...) static char string[1024]; va_start(argptr, format); - vsprintf(string, format, argptr); + vsnprintf(string, 1024, format, argptr); va_end(argptr); return string; diff --git a/src/m_perfstats.c b/src/m_perfstats.c index 33a774acfbb4ca9781ff62a177b93f9bffca28f8..f36a1c80949bec45166c57a4630296f8a531c728 100644 --- a/src/m_perfstats.c +++ b/src/m_perfstats.c @@ -127,8 +127,8 @@ perfstatrow_t commoncounter_rows[] = { }; perfstatrow_t interpolation_rows[] = { - {"intpfrc", "Interp frac: ", &ps_interp_frac, PS_TIME}, - {"intplag", "Interp lag: ", &ps_interp_lag, PS_TIME}, + {"intpfrc", "Interp frac: ", &ps_interp_frac, 0}, // PS_TIME is not applicable here, as it is meant for I_GetPreciseTime + {"intplag", "Interp lag: ", &ps_interp_lag, 0}, {0} }; diff --git a/src/m_tokenizer.c b/src/m_tokenizer.c index 09f8915a52b2c476036f560c773118b3d5f823ff..2f89d4dfbb3e40e4f7426a655056f36405d1a0c3 100644 --- a/src/m_tokenizer.c +++ b/src/m_tokenizer.c @@ -1,6 +1,6 @@ // SONIC ROBO BLAST 2 //----------------------------------------------------------------------------- -// Copyright (C) 2013-2024 by Sonic Team Junior. +// Copyright (C) 2013-2025 by Sonic Team Junior. // // This program is free software distributed under the // terms of the GNU General Public License, version 2. @@ -28,6 +28,7 @@ tokenizer_t *Tokenizer_Open(const char *inputString, size_t len, unsigned numTok tokenizer->endPos = 0; tokenizer->inputLength = 0; tokenizer->inComment = 0; + tokenizer->stringNeedsEscaping = false; tokenizer->inString = 0; tokenizer->get = Tokenizer_Read; @@ -92,6 +93,124 @@ static void DetectComment(tokenizer_t *tokenizer, UINT32 *pos) tokenizer->inComment = 2; } +// This function detects escape sequences in a string and attempts to convert them. +static size_t EscapeString(char *output, const char *input, size_t inputLength) +{ + const char *end = input + inputLength; + + size_t i = 0; + + while (input < end) + { + char chr = *input++; + + if (chr == '\\') + { + chr = *input++; + + switch (chr) + { + case 'n': + output[i] = '\n'; + i++; + break; + case 't': + output[i] = '\t'; + i++; + break; + case '\\': + output[i] = '\\'; + i++; + break; + case '"': + output[i] = '\"'; + i++; + break; + case 'x': { + int out = 0, c; + int j = 0; + + chr = *input++; + + for (j = 0; j < 5 && isxdigit(chr); j++) + { + c = ((chr <= '9') ? (chr - '0') : (tolower(chr) - 'a' + 10)); + out = (out << 4) | c; + chr = *input++; + } + + input--; + + switch (j) + { + case 4: + output[i] = (out >> 8) & 0xFF; + i++; + /* FALLTHRU */ + case 2: + output[i] = out & 0xFF; + i++; + break; + default: + // TODO: Displaying parsing errors properly will require + // some refactoring of the tokenizer itself. For now, + // this function will silently return an empty string + // if it encounters a malformed escape sequence. + // This situation cannot happen for i.e. UDMF comments, + // so it's okay to do this right now. + // CONS_Alert(CONS_WARNING, "Escape sequence has wrong size\n"); + i = 0; + goto done; + } + + break; + } + default: + if (isdigit(chr)) + { + int out = 0; + int j = 0; + do + { + out = 10*out + (chr - '0'); + chr = *input++; + } while (++j < 3 && isdigit(chr)); + + input--; + + if (out > 255) + { + // CONS_Alert(CONS_WARNING, "Escape sequence is too large\n"); + i = 0; + goto done; + } + + output[i] = out; + i++; + } + else + { + // CONS_Alert(CONS_WARNING, "Unknown escape sequence '\\%c'\n", chr); + i = 0; + goto done; + } + break; + } + } + else + { + output[i] = chr; + i++; + } + } + +done: + output[i] = '\0'; + i++; + + return i; +} + static void Tokenizer_ReadTokenString(tokenizer_t *tokenizer, UINT32 i) { UINT32 tokenLength = tokenizer->endPos - tokenizer->startPos; @@ -101,10 +220,46 @@ static void Tokenizer_ReadTokenString(tokenizer_t *tokenizer, UINT32 i) // Assign the memory. Don't forget an extra byte for the end of the string! tokenizer->token[i] = (char *)Z_Malloc(tokenizer->capacity[i] * sizeof(char), PU_STATIC, NULL); } + // Copy the string. - M_Memcpy(tokenizer->token[i], tokenizer->input + tokenizer->startPos, (size_t)tokenLength); - // Make the final character NUL. - tokenizer->token[i][tokenLength] = '\0'; + if (tokenizer->stringNeedsEscaping) + { + EscapeString(tokenizer->token[i], tokenizer->input + tokenizer->startPos, (size_t)tokenLength); + } + else + { + M_Memcpy(tokenizer->token[i], tokenizer->input + tokenizer->startPos, (size_t)tokenLength); + + // Make the final character NUL. + tokenizer->token[i][tokenLength] = '\0'; + } +} + +static void ScanString(tokenizer_t *tokenizer) +{ + tokenizer->stringNeedsEscaping = false; + + while (tokenizer->input[tokenizer->endPos] != '"' && tokenizer->endPos < tokenizer->inputLength) + { + if (!DetectLineBreak(tokenizer, tokenizer->endPos)) + { + // Skip one character ahead if this looks like an escape sequence + if (tokenizer->input[tokenizer->endPos] == '\\') + { + tokenizer->stringNeedsEscaping = true; + tokenizer->endPos++; + + // Oh. Naughty. We hit the end of the input. + // Stop scanning, then. + if (tokenizer->endPos == tokenizer->inputLength) + return; + + DetectLineBreak(tokenizer, tokenizer->endPos); + } + } + + tokenizer->endPos++; + } } const char *Tokenizer_Read(tokenizer_t *tokenizer, UINT32 i) @@ -117,11 +272,7 @@ const char *Tokenizer_Read(tokenizer_t *tokenizer, UINT32 i) // If in a string, return the entire string within quotes, except without the quotes. if (tokenizer->inString == 1) { - while (tokenizer->input[tokenizer->endPos] != '"' && tokenizer->endPos < tokenizer->inputLength) - { - DetectLineBreak(tokenizer, tokenizer->endPos); - tokenizer->endPos++; - } + ScanString(tokenizer); Tokenizer_ReadTokenString(tokenizer, i); tokenizer->inString = 2; @@ -134,6 +285,7 @@ const char *Tokenizer_Read(tokenizer_t *tokenizer, UINT32 i) tokenizer->token[i][0] = tokenizer->input[tokenizer->startPos]; tokenizer->token[i][1] = '\0'; tokenizer->inString = 0; + tokenizer->stringNeedsEscaping = false; return tokenizer->token[i]; } @@ -281,11 +433,7 @@ const char *Tokenizer_SRB2Read(tokenizer_t *tokenizer, UINT32 i) else if (tokenizer->input[tokenizer->startPos] == '"') { tokenizer->endPos = ++tokenizer->startPos; - while (tokenizer->input[tokenizer->endPos] != '"' && tokenizer->endPos < tokenizer->inputLength) - { - DetectLineBreak(tokenizer, tokenizer->endPos); - tokenizer->endPos++; - } + ScanString(tokenizer); Tokenizer_ReadTokenString(tokenizer, i); tokenizer->endPos++; diff --git a/src/m_tokenizer.h b/src/m_tokenizer.h index 4f03563a2da69c671edfc196ec478dc4715fa080..bc7090cd82a87ca7065b8355e0f4be5b73675fbe 100644 --- a/src/m_tokenizer.h +++ b/src/m_tokenizer.h @@ -1,6 +1,6 @@ // SONIC ROBO BLAST 2 //----------------------------------------------------------------------------- -// Copyright (C) 2013-2024 by Sonic Team Junior. +// Copyright (C) 2013-2025 by Sonic Team Junior. // // This program is free software distributed under the // terms of the GNU General Public License, version 2. @@ -26,6 +26,7 @@ typedef struct Tokenizer UINT32 inputLength; UINT8 inComment; // 0 = not in comment, 1 = // Single-line, 2 = /* Multi-line */ UINT8 inString; // 0 = not in string, 1 = in string, 2 = just left string + boolean stringNeedsEscaping; int line; const char *(*get)(struct Tokenizer*, UINT32); } tokenizer_t; diff --git a/src/netcode/client_connection.c b/src/netcode/client_connection.c index 8f32dd8b44ceea6655a13404eb02560cb7b8dc4e..7b21a76cd82ee6e887e5bdc88fcf52d1dc4a4eb6 100644 --- a/src/netcode/client_connection.c +++ b/src/netcode/client_connection.c @@ -54,6 +54,8 @@ static boolean IsDownloadingFile(void) static void DrawConnectionStatusBox(void) { M_DrawTextBox(BASEVIDWIDTH/2-128-8, BASEVIDHEIGHT-16-8, 32, 1); + if (cl_mode != CL_DOWNLOADSAVEGAME && filedownload.current != -1) + M_DrawTextBox(BASEVIDWIDTH/2-128-8, BASEVIDHEIGHT-46-8, 32, 1); if (cl_mode == CL_CONFIRMCONNECT || IsDownloadingFile()) return; @@ -84,6 +86,33 @@ static void DrawFileProgress(fileneeded_t *file, int y) V_DrawRightAlignedString(BASEVIDWIDTH/2+128, y, V_20TRANS|V_MONOSPACE, va("%3.1fK/s ", ((double)getbps)/1024)); } +static void DrawOverallProgress(int y) +{ + UINT32 totalsize = filedownload.totalsize; + INT32 downloadedfiles = filedownload.completednum; + INT32 totalfiles = filedownload.remaining + filedownload.completednum; + INT32 downloaded = filedownload.completedsize; + if (fileneeded[filedownload.current].currentsize != fileneeded[filedownload.current].totalsize) + downloaded = filedownload.completedsize + fileneeded[filedownload.current].currentsize; + + INT32 dldlength = (INT32)((downloaded/(double)totalsize) * 256); + if (dldlength > 256) + dldlength = 256; + V_DrawFill(BASEVIDWIDTH/2-128, y, 256, 8, 111); + V_DrawFill(BASEVIDWIDTH/2-128, y, dldlength, 8, 96); + + const char *progress_str; + if (totalsize >= 1024*1024) + progress_str = va(" %.2fMiB/%.2fMiB", (double)downloaded / (1024*1024), (double)totalsize / (1024*1024)); + else if (totalsize < 1024) + progress_str = va(" %4uB/%4uB", downloaded, totalsize); + else + progress_str = va(" %.2fKiB/%.2fKiB", (double)downloaded / 1024, (double)totalsize / 1024); + + V_DrawString(BASEVIDWIDTH/2-128, y, V_20TRANS|V_ALLOWLOWERCASE, progress_str); + V_DrawRightAlignedString(BASEVIDWIDTH/2+128, y, V_20TRANS|V_ALLOWLOWERCASE, va("%2u/%2u Files ", downloadedfiles+1, totalfiles)); +} + // // CL_DrawConnectionStatus // @@ -96,7 +125,7 @@ static void CL_DrawConnectionStatus(void) // Draw background fade V_DrawFadeScreen(0xFF00, 16); // force default - if (cl_mode != CL_DOWNLOADFILES && cl_mode != CL_DOWNLOADHTTPFILES && cl_mode != CL_LOADFILES) + if (cl_mode != CL_DOWNLOADFILES && cl_mode != CL_DOWNLOADHTTPFILES && cl_mode != CL_LOADFILES && cl_mode != CL_CHECKFILES && cl_mode != CL_ASKFULLFILELIST) { INT32 animtime = ((ccstime / 4) & 15) + 16; UINT8 palstart; @@ -179,6 +208,31 @@ static void CL_DrawConnectionStatus(void) V_DrawCenteredString(BASEVIDWIDTH/2, BASEVIDHEIGHT-16, V_20TRANS|V_MONOSPACE, va(" %2u/%2u files",loadcompletednum,fileneedednum)); } + else if ((cl_mode == CL_CHECKFILES) || (cl_mode == CL_ASKFULLFILELIST)) + { + INT32 totalfileslength; + INT32 checkcompletednum = 0; + INT32 i; + + V_DrawCenteredString(BASEVIDWIDTH/2, BASEVIDHEIGHT-16-16, V_YELLOWMAP, "Press ESC to abort"); + + //ima just count files here + if (fileneeded) + { + for (i = 0; i < fileneedednum; i++) + if (fileneeded[i].status != FS_NOTCHECKED) + checkcompletednum++; + } + + // Check progress + V_DrawCenteredString(BASEVIDWIDTH/2, BASEVIDHEIGHT-16-24, V_YELLOWMAP, "Checking server addon list..."); + totalfileslength = (INT32)((checkcompletednum/(double)(fileneedednum)) * 256); + M_DrawTextBox(BASEVIDWIDTH/2-128-8, BASEVIDHEIGHT-16-8, 32, 1); + V_DrawFill(BASEVIDWIDTH/2-128, BASEVIDHEIGHT-16, 256, 8, 111); + V_DrawFill(BASEVIDWIDTH/2-128, BASEVIDHEIGHT-16, totalfileslength, 8, 96); + V_DrawCenteredString(BASEVIDWIDTH/2, BASEVIDHEIGHT-16, V_20TRANS|V_MONOSPACE, + va(" %2u/%2u Files",checkcompletednum,fileneedednum)); + } else if (filedownload.current != -1) { char tempname[28]; @@ -224,7 +278,7 @@ static void CL_DrawConnectionStatus(void) const char *download_str = M_GetText("Downloading \"%s\""); #endif - V_DrawCenteredString(BASEVIDWIDTH/2, BASEVIDHEIGHT-16-24, V_ALLOWLOWERCASE|V_YELLOWMAP, + V_DrawCenteredString(BASEVIDWIDTH/2, BASEVIDHEIGHT-46-24, V_ALLOWLOWERCASE|V_YELLOWMAP, va(download_str, tempname)); // Rusty: actually lets do this instead @@ -244,16 +298,18 @@ static void CL_DrawConnectionStatus(void) strlcpy(tempname, http_source, sizeof(tempname)); } - V_DrawCenteredString(BASEVIDWIDTH/2, BASEVIDHEIGHT-16-16, V_ALLOWLOWERCASE|V_YELLOWMAP, + V_DrawCenteredString(BASEVIDWIDTH/2, BASEVIDHEIGHT-46-16, V_ALLOWLOWERCASE|V_YELLOWMAP, va(M_GetText("from %s"), tempname)); } else { - V_DrawCenteredString(BASEVIDWIDTH/2, BASEVIDHEIGHT-16-16, V_ALLOWLOWERCASE|V_YELLOWMAP, + V_DrawCenteredString(BASEVIDWIDTH/2, BASEVIDHEIGHT-46-16, V_ALLOWLOWERCASE|V_YELLOWMAP, M_GetText("from the server")); } + DrawFileProgress(file, BASEVIDHEIGHT-46); - DrawFileProgress(file, BASEVIDHEIGHT-16); + V_DrawCenteredString(BASEVIDWIDTH/2, BASEVIDHEIGHT-16-14, V_ALLOWLOWERCASE|V_YELLOWMAP, "Total Progress"); + DrawOverallProgress(BASEVIDHEIGHT-16); } else { @@ -657,6 +713,7 @@ static void ShowDownloadConsentMessage(void) if (IsFileDownloadable(&fileneeded[i])) totalsize += fileneeded[i].totalsize; } + filedownload.totalsize = totalsize; const char *downloadsize = GetPrintableFileSize(totalsize); diff --git a/src/netcode/d_clisrv.c b/src/netcode/d_clisrv.c index 1cd26cd4f32ca9e239fd3f26a728e54629cc8ed2..92a4cd784e3cfc409cbdf2d78fd7ab4d090999a5 100644 --- a/src/netcode/d_clisrv.c +++ b/src/netcode/d_clisrv.c @@ -641,6 +641,7 @@ void CL_RemovePlayer(INT32 playernum, kickreason_t reason) void D_QuitNetGame(void) { mousegrabbedbylua = true; + textinputmodeenabledbylua = false; I_UpdateMouseGrab(); if (!netgame || !netbuffer) diff --git a/src/netcode/d_netfil.h b/src/netcode/d_netfil.h index 9f29d18bb79ff39dda69fffd68e2756f8d4f0db9..98456fe13ee003ea71ff672c40374f031d42639c 100644 --- a/src/netcode/d_netfil.h +++ b/src/netcode/d_netfil.h @@ -95,6 +95,7 @@ typedef struct INT32 remaining; INT32 completednum; UINT32 completedsize; + UINT64 totalsize; boolean http_failed; boolean http_running; diff --git a/src/p_enemy.c b/src/p_enemy.c index 7091af5429e834c098ba7d81ace6961cc238e37f..a810176b05a1002113b27f1f4e5d4b62e0ff9265 100644 --- a/src/p_enemy.c +++ b/src/p_enemy.c @@ -6109,7 +6109,7 @@ void A_UnidusBall(mobj_t *actor) else if (locvar1 == 2) { boolean skull = (actor->target->flags2 & MF2_SKULLFLY) == MF2_SKULLFLY; - if (actor->target->state == &states[actor->target->info->painstate]) + if (P_IsMobjInPainState(actor->target)) { P_KillMobj(actor, NULL, NULL, 0); return; diff --git a/src/p_inter.c b/src/p_inter.c index 27e612154dc87c1b88fec2e42e890af0d27c382e..c9d0d39647914cc16c2c2a893e9e8a0e1c980812 100644 --- a/src/p_inter.c +++ b/src/p_inter.c @@ -412,7 +412,7 @@ void P_TouchSpecialThing(mobj_t *special, mobj_t *toucher, boolean heightcheck) { if (special->type == MT_STEAM) { - if (player && player->mo->state == &states[player->mo->info->painstate]) // can't use gas jets when player is in pain! + if (player && P_IsPlayerInState(player, S_PLAY_PAIN)) // can't use gas jets when player is in pain! return; fixed_t speed = special->info->mass; // gas jets use this for the vertical thrust @@ -1820,7 +1820,7 @@ void P_TouchSpecialThing(mobj_t *special, mobj_t *toucher, boolean heightcheck) if (!player->climbing) { - if (player->bot && player->bot != BOT_MPAI && toucher->state-states != S_PLAY_GASP) + if (player->bot && player->bot != BOT_MPAI && !P_IsPlayerInState(player, S_PLAY_GASP)) S_StartSound(toucher, special->info->deathsound); // Force it to play a sound for bots P_SetMobjState(toucher, S_PLAY_GASP); P_ResetPlayer(player); @@ -1839,8 +1839,8 @@ void P_TouchSpecialThing(mobj_t *special, mobj_t *toucher, boolean heightcheck) special->z = toucher->z+toucher->height-FixedMul(8*FRACUNIT, special->scale); special->momz = 0; special->flags |= MF_NOGRAVITY; - P_SetMobjState (special, special->info->deathstate); - S_StartSound (special, special->info->deathsound+(P_RandomKey(special->info->mass))); + P_SetMobjState(special, special->info->deathstate); + S_StartSound(special, special->info->deathsound+(P_RandomKey(special->info->mass))); } return; @@ -3226,8 +3226,8 @@ static boolean P_TagDamage(mobj_t *target, mobj_t *inflictor, mobj_t *source, IN return false; // Ignore IT players shooting each other, unless friendlyfire is on. - if ((player->pflags & PF_TAGIT && !((cv_friendlyfire.value || (gametyperules & GTR_FRIENDLYFIRE) || (damagetype & DMG_CANHURTSELF)) && - source && source->player && source->player->pflags & PF_TAGIT))) + if ((player->pflags & PF_TAGIT && source && source->player && !(((cv_friendlyfire.value || (gametyperules & GTR_FRIENDLYFIRE)) || ((damagetype & DMG_CANHURTSELF) && source->player == player)) && + source->player->pflags & PF_TAGIT))) { if (inflictor->type == MT_LHRT && !(player->powers[pw_shield] & SH_NOSTACK)) { @@ -3242,7 +3242,8 @@ static boolean P_TagDamage(mobj_t *target, mobj_t *inflictor, mobj_t *source, IN // Don't allow players on the same team to hurt one another, // unless cv_friendlyfire is on. - if (!(cv_friendlyfire.value || (gametyperules & GTR_FRIENDLYFIRE) || (damagetype & DMG_CANHURTSELF)) && (player->pflags & PF_TAGIT) == (source->player->pflags & PF_TAGIT)) + if (source && source->player && !((cv_friendlyfire.value || (gametyperules & GTR_FRIENDLYFIRE)) || ((damagetype & DMG_CANHURTSELF) && source->player == player)) && + (player->pflags & PF_TAGIT) == (source->player->pflags & PF_TAGIT)) { if (inflictor->type == MT_LHRT && !(player->powers[pw_shield] & SH_NOSTACK)) { @@ -3344,7 +3345,7 @@ static boolean P_PlayerHitsPlayer(mobj_t *target, mobj_t *inflictor, mobj_t *sou // Tag handling if (G_TagGametype()) return P_TagDamage(target, inflictor, source, damage, damagetype); - else if (damagetype & DMG_CANHURTSELF) + else if ((damagetype & DMG_CANHURTSELF) && source && source->player && source->player == player) return true; else if (G_GametypeHasTeams()) // CTF + Team Match { @@ -3413,18 +3414,20 @@ static void P_KillPlayer(player_t *player, mobj_t *source, INT32 damage) if ((gametyperules & GTR_TEAMFLAGS) && (player->gotflag & (GF_REDFLAG|GF_BLUEFLAG))) { P_PlayerFlagBurst(player, false); - if (source && source->player) + if (source && source->player && source->player != player) // Don't score points against yourself { // Award no points when players shoot each other when cv_friendlyfire is on. if (!G_GametypeHasTeams() || !(source->player->ctfteam == player->ctfteam && source != player->mo)) P_AddPlayerScore(source->player, 25); } } - if (source && source->player && !player->powers[pw_super]) //don't score points against super players + if (source && source->player && source->player != player && !player->powers[pw_super]) //don't score points against super players or yourself { // Award no points when players shoot each other when cv_friendlyfire is on. if (!G_GametypeHasTeams() || !(source->player->ctfteam == player->ctfteam && source != player->mo)) + { P_AddPlayerScore(source->player, 100); + } } // If the player was super, tell them he/she ain't so super nomore. @@ -3538,14 +3541,14 @@ static void P_ShieldDamage(player_t *player, mobj_t *inflictor, mobj_t *source, if ((gametyperules & GTR_TEAMFLAGS) && (player->gotflag & (GF_REDFLAG|GF_BLUEFLAG))) { P_PlayerFlagBurst(player, false); - if (source && source->player) + if (source && source->player && source->player != player) // Don't score points against yourself { // Award no points when players shoot each other when cv_friendlyfire is on. if (!G_GametypeHasTeams() || !(source->player->ctfteam == player->ctfteam && source != player->mo)) P_AddPlayerScore(source->player, 25); } } - if (source && source->player && !player->powers[pw_super]) //don't score points against super players + if (source && source->player && source->player != player && !player->powers[pw_super]) //don't score points against super players or yourself { // Award no points when players shoot each other when cv_friendlyfire is on. if (!G_GametypeHasTeams() || !(source->player->ctfteam == player->ctfteam && source != player->mo)) @@ -3562,7 +3565,7 @@ static void P_RingDamage(player_t *player, mobj_t *inflictor, mobj_t *source, IN if (damagetype == DMG_SPIKE) // spikes S_StartSound(player->mo, sfx_spkdth); - if (source && source->player && !player->powers[pw_super]) //don't score points against super players + if (source && source->player && source->player != player && !player->powers[pw_super]) //don't score points against super players { // Award no points when players shoot each other when cv_friendlyfire is on. if (!G_GametypeHasTeams() || !(source->player->ctfteam == player->ctfteam && source != player->mo)) @@ -3572,7 +3575,7 @@ static void P_RingDamage(player_t *player, mobj_t *inflictor, mobj_t *source, IN if ((gametyperules & GTR_TEAMFLAGS) && (player->gotflag & (GF_REDFLAG|GF_BLUEFLAG))) { P_PlayerFlagBurst(player, false); - if (source && source->player) + if (source && source->player && source->player != player) // Don't score points against yourself { // Award no points when players shoot each other when cv_friendlyfire is on. if (!G_GametypeHasTeams() || !(source->player->ctfteam == player->ctfteam && source != player->mo)) @@ -3885,7 +3888,7 @@ boolean P_DamageMobj(mobj_t *target, mobj_t *inflictor, mobj_t *source, INT32 da // To reduce griefing potential, don't allow players to be killed // by friendly fire. Spilling their rings and other items is enough. else if (!force && G_GametypeHasTeams() - && source && source->player && (source->player->ctfteam == player->ctfteam) + && source && source->player && source->player != player && (source->player->ctfteam == player->ctfteam) && (cv_friendlyfire.value || (gametyperules & GTR_FRIENDLYFIRE))) { damage = 0; diff --git a/src/p_local.h b/src/p_local.h index 85a31cf8982114068b0da41c7a59971d011929ba..814f5b1a3d6a618c2c05c69a0855ef5e444a7aa1 100644 --- a/src/p_local.h +++ b/src/p_local.h @@ -130,6 +130,10 @@ boolean P_TryCameraMove(fixed_t x, fixed_t y, camera_t *thiscam); void P_SlideCameraMove(camera_t *thiscam); boolean P_MoveChaseCamera(player_t *player, camera_t *thiscam, boolean resetcalled); pflags_t P_GetJumpFlags(player_t *player); +statenum_t P_GetCanonicalPlayerState(player_t *player, statenum_t state); +boolean P_IsPlayerInState(player_t *player, statenum_t state); +boolean P_IsPlayerInSuperTransformationState(player_t *player); +boolean P_IsPlayerInNightsTransformationState(player_t *player); boolean P_PlayerInPain(player_t *player); void P_DoPlayerPain(player_t *player, mobj_t *source, mobj_t *inflictor); void P_ResetPlayer(player_t *player); @@ -553,5 +557,6 @@ void P_DoSuperDetransformation(player_t *player); void P_ExplodeMissile(mobj_t *mo); void P_CheckGravity(mobj_t *mo, boolean affect); void P_SetPitchRollFromSlope(mobj_t *mo, pslope_t *slope); +boolean P_IsMobjInPainState(mobj_t *mobj); #endif // __P_LOCAL__ diff --git a/src/p_map.c b/src/p_map.c index 337344b2fec2c966d0774513c02d9397b2974382..4a4dffc695ac97f3c70ebb68e22d1bea7b238409 100644 --- a/src/p_map.c +++ b/src/p_map.c @@ -508,7 +508,7 @@ static void P_DoFan(mobj_t *fan, mobj_t *object) fixed_t speed = fan->info->mass; // fans use this for the vertical thrust SINT8 flipval = P_MobjFlip(fan); // virtually everything here centers around the thruster's gravity, not the object's! - if (p && object->state == &states[object->info->painstate]) // can't use fans when player is in pain! + if (p && P_IsPlayerInState(p, S_PLAY_PAIN)) // can't use fans when player is in pain! return; // is object's top below thruster's position? if not, calculate distance between their bottoms @@ -3033,7 +3033,7 @@ static boolean P_ThingHeightClip(mobj_t *thing) if (tmfloorz > oldfloorz+thing->height) return true; - bouncing = thing->player && thing->state-states == S_PLAY_BOUNCE_LANDING && P_IsObjectOnGround(thing); + bouncing = thing->player && P_IsPlayerInState(thing->player, S_PLAY_BOUNCE_LANDING) && P_IsObjectOnGround(thing); if ((onfloor || bouncing) && !(thing->flags & MF_NOGRAVITY) && floormoved) { diff --git a/src/p_maputl.c b/src/p_maputl.c index 4c7b959270993a7117c58d83f71c86348206937e..de239056a51524813b4466519035c631f3975a02 100644 --- a/src/p_maputl.c +++ b/src/p_maputl.c @@ -1056,7 +1056,7 @@ boolean P_BlockThingsIterator(INT32 x, INT32 y, boolean (*func)(mobj_t *), mobj_ boolean checkthing = false; if (thing) - checkthing = true; + checkthing = true; if (x < 0 || y < 0 || x >= bmapwidth || y >= bmapheight) return true; @@ -1434,7 +1434,7 @@ static boolean P_TraverseIntercepts(traverser_t func, fixed_t maxfrac) if (dist > maxfrac) return true; // Checked everything in range. - + if (!func(in)) return false; // Don't bother going farther. @@ -1462,7 +1462,7 @@ boolean P_PathTraverse(fixed_t px1, fixed_t py1, fixed_t px2, fixed_t py2, validcount++; intercept_p = intercepts; - + if (((px1 - bmaporgx) & (MAPBLOCKSIZE-1)) == 0) px1 += FRACUNIT; // Don't side exactly on a line. @@ -1475,7 +1475,7 @@ boolean P_PathTraverse(fixed_t px1, fixed_t py1, fixed_t px2, fixed_t py2, trace.dy = py2 - py1; xt1 = px1>>MAPBLOCKSHIFT; - yt1 = py2>>MAPBLOCKSHIFT; + yt1 = py1>>MAPBLOCKSHIFT; px1 = (unsigned)(px1 - bmaporgx); py1 = (unsigned)(py1 - bmaporgy); @@ -1624,6 +1624,7 @@ boolean P_PathTraverse(fixed_t px1, fixed_t py1, fixed_t px2, fixed_t py2, break; } } + // Go through the sorted list return P_TraverseIntercepts(trav, FRACUNIT); } diff --git a/src/p_mobj.c b/src/p_mobj.c index e758563f4180f06782d71569e66da540073a997f..f81c725c84dc7f58d04b762c5cbcfd6322a7951c 100644 --- a/src/p_mobj.c +++ b/src/p_mobj.c @@ -36,6 +36,7 @@ #include "p_slopes.h" #include "f_finale.h" #include "m_cond.h" +#include "simple_hashmap.h" #include "netcode/net_command.h" static CV_PossibleValue_t CV_BobSpeed[] = {{0, "MIN"}, {4*FRACUNIT, "MAX"}, {0, NULL}}; @@ -178,6 +179,61 @@ static void P_CycleMobjState(mobj_t *mobj) } } +static panim_t GetPlayerAnimationFromState(player_t *player, statenum_t state) +{ + switch(P_GetCanonicalPlayerState(player, state)) + { + case S_PLAY_STND: + case S_PLAY_WAIT: + case S_PLAY_NIGHTS_STAND: + return PA_IDLE; + case S_PLAY_EDGE: + return PA_EDGE; + case S_PLAY_WALK: + case S_PLAY_SKID: + case S_PLAY_FLOAT: + return PA_WALK; + case S_PLAY_RUN: + case S_PLAY_FLOAT_RUN: + return PA_RUN; + case S_PLAY_DASH: + return PA_DASH; + case S_PLAY_PAIN: + case S_PLAY_STUN: + return PA_PAIN; + case S_PLAY_ROLL: + //case S_PLAY_SPINDASH: -- everyone can ROLL thanks to zoom tubes... + case S_PLAY_NIGHTS_ATTACK: + return PA_ROLL; + case S_PLAY_JUMP: + return PA_JUMP; + case S_PLAY_SPRING: + return PA_SPRING; + case S_PLAY_FALL: + case S_PLAY_NIGHTS_FLOAT: + return PA_FALL; + case S_PLAY_FLY: + case S_PLAY_FLY_TIRED: + case S_PLAY_SWIM: + case S_PLAY_GLIDE: + case S_PLAY_BOUNCE: + case S_PLAY_BOUNCE_LANDING: + case S_PLAY_TWINSPIN: + return PA_ABILITY; + case S_PLAY_SPINDASH: // ...but the act of SPINDASHING is charability2 specific. + case S_PLAY_FIRE: + case S_PLAY_FIRE_FINISH: + case S_PLAY_MELEE: + case S_PLAY_MELEE_FINISH: + case S_PLAY_MELEE_LANDING: + return PA_ABILITY2; + case S_PLAY_RIDE: + return PA_RIDE; + default: + return PA_ETC; + } +} + // // P_SetPlayerMobjState // Returns true if the mobj is still present. @@ -201,6 +257,12 @@ static boolean P_SetPlayerMobjState(mobj_t *mobj, statenum_t state) I_Error("P_SetPlayerMobjState used for non-player mobj. Use P_SetMobjState instead!\n(Mobj type: %d, State: %d)", mobj->type, state); #endif + // If the state has been overriden for this skin, use the replacement instead + statenum_t customskinstate; + SIMPLEHASH_FIND_INT(skins[player->skin]->defaulttocustomstate, hashentry_int32_int32_t, state, S_NULL, customskinstate) + if (customskinstate) + state = customskinstate; + // Catch falling for nojumpspin if ((state == S_PLAY_JUMP) && (player->charflags & SF_NOJUMPSPIN) && (P_MobjFlip(mobj)*mobj->momz < 0)) return P_SetPlayerMobjState(mobj, S_PLAY_FALL); @@ -222,88 +284,24 @@ static boolean P_SetPlayerMobjState(mobj_t *mobj, statenum_t state) { if (state == S_PLAY_JUMP) { - if (player->mo->state-states == S_PLAY_WALK) + if (P_IsPlayerInState(player, S_PLAY_WALK)) return P_SetPlayerMobjState(mobj, S_PLAY_FLOAT); return true; } - else if (player->mo->state-states == S_PLAY_FLOAT && state == S_PLAY_STND) + else if (P_IsPlayerInState(player, S_PLAY_FLOAT) && state == S_PLAY_STND) return true; } // You were in pain state after taking a hit, and you're moving out of pain state now? - else if (mobj->state == &states[mobj->info->painstate] && player->powers[pw_flashing] == flashingtics && state != mobj->info->painstate) + else if (P_IsPlayerInState(player, S_PLAY_PAIN) + && player->powers[pw_flashing] == flashingtics + && P_GetCanonicalPlayerState(player, state) != S_PLAY_PAIN) { // Start flashing, since you've landed. player->powers[pw_flashing] = flashingtics-1; P_DoPityCheck(player); } - // Set animation state - // The pflags version of this was just as convoluted. - switch(state) - { - case S_PLAY_STND: - case S_PLAY_WAIT: - case S_PLAY_NIGHTS_STAND: - player->panim = PA_IDLE; - break; - case S_PLAY_EDGE: - player->panim = PA_EDGE; - break; - case S_PLAY_WALK: - case S_PLAY_SKID: - case S_PLAY_FLOAT: - player->panim = PA_WALK; - break; - case S_PLAY_RUN: - case S_PLAY_FLOAT_RUN: - player->panim = PA_RUN; - break; - case S_PLAY_DASH: - player->panim = PA_DASH; - break; - case S_PLAY_PAIN: - case S_PLAY_STUN: - player->panim = PA_PAIN; - break; - case S_PLAY_ROLL: - //case S_PLAY_SPINDASH: -- everyone can ROLL thanks to zoom tubes... - case S_PLAY_NIGHTS_ATTACK: - player->panim = PA_ROLL; - break; - case S_PLAY_JUMP: - player->panim = PA_JUMP; - break; - case S_PLAY_SPRING: - player->panim = PA_SPRING; - break; - case S_PLAY_FALL: - case S_PLAY_NIGHTS_FLOAT: - player->panim = PA_FALL; - break; - case S_PLAY_FLY: - case S_PLAY_FLY_TIRED: - case S_PLAY_SWIM: - case S_PLAY_GLIDE: - case S_PLAY_BOUNCE: - case S_PLAY_BOUNCE_LANDING: - case S_PLAY_TWINSPIN: - player->panim = PA_ABILITY; - break; - case S_PLAY_SPINDASH: // ...but the act of SPINDASHING is charability2 specific. - case S_PLAY_FIRE: - case S_PLAY_FIRE_FINISH: - case S_PLAY_MELEE: - case S_PLAY_MELEE_FINISH: - case S_PLAY_MELEE_LANDING: - player->panim = PA_ABILITY2; - break; - case S_PLAY_RIDE: - player->panim = PA_RIDE; - break; - default: - player->panim = PA_ETC; - break; - } + player->panim = GetPlayerAnimationFromState(player, state); if (recursion++) // if recursion detected, memset(seenstate = tempstate, 0, sizeof tempstate); // clear state table @@ -3717,7 +3715,7 @@ static void P_PlayerMobjThinker(mobj_t *mobj) // momentum movement mobj->eflags &= ~MFE_JUSTSTEPPEDDOWN; - if (mobj->state-states == S_PLAY_BOUNCE_LANDING) + if (P_IsPlayerInState(mobj->player, S_PLAY_BOUNCE_LANDING)) goto animonly; // no need for checkposition - doesn't move at ALL // Zoom tube @@ -14365,3 +14363,11 @@ mobj_t *P_SpawnMobjFromMobj(mobj_t *mobj, fixed_t xofs, fixed_t yofs, fixed_t zo return newmobj; } + +boolean P_IsMobjInPainState(mobj_t *mobj) +{ + if (mobj->player) + return P_IsPlayerInState(mobj->player, S_PLAY_PAIN); + else + return (mobj->state == &states[mobj->info->painstate]); +} diff --git a/src/p_spec.c b/src/p_spec.c index 649f63518eed2017cbfd06817eb322d27280d52e..394de0167de8b517e396752ea8a374d63a6143cc 100644 --- a/src/p_spec.c +++ b/src/p_spec.c @@ -4677,7 +4677,7 @@ static void P_ProcessZoomTube(player_t *player, mtag_t sectag, boolean end) player->pflags &= ~(PF_JUMPED|PF_NOJUMPDAMAGE|PF_GLIDING|PF_BOUNCING|PF_SLIDING|PF_CANCARRY); player->climbing = 0; - if (player->mo->state-states != S_PLAY_ROLL) + if (!P_IsPlayerInState(player, S_PLAY_ROLL)) { P_SetMobjState(player->mo, S_PLAY_ROLL); S_StartSound(player->mo, sfx_spin); @@ -4758,7 +4758,7 @@ static void P_ProcessRopeHang(player_t *player, mtag_t sectag) if (player->cmd.buttons & BT_SPIN) return; - if (!(player->pflags & PF_SLIDING) && player->mo->state == &states[player->mo->info->painstate]) + if (!(player->pflags & PF_SLIDING) && P_IsPlayerInState(player, S_PLAY_PAIN)) return; if (player->exiting) @@ -8805,7 +8805,7 @@ void T_Pusher(pusher_t *p) if (thing->player && thing->player->powers[pw_carry] == CR_ROPEHANG) continue; - if (thing->player && (thing->state == &states[thing->info->painstate]) && (thing->player->powers[pw_flashing] > (flashingtics/4)*3 && thing->player->powers[pw_flashing] <= flashingtics)) + if (thing->player && P_IsPlayerInState(thing->player, S_PLAY_PAIN) && (thing->player->powers[pw_flashing] > (flashingtics/4)*3 && thing->player->powers[pw_flashing] <= flashingtics)) continue; inFOF = touching = moved = false; diff --git a/src/p_user.c b/src/p_user.c index ba22808eade2a389dc6c7bf56383fcd6149c033d..ad757d1e56edb9a2e4d1e593658c28d73c26c862 100644 --- a/src/p_user.c +++ b/src/p_user.c @@ -47,6 +47,7 @@ #include "m_cheat.h" // Thok camera snap (ctrl-f "chalupa") #include "g_input.h" +#include "simple_hashmap.h" #ifdef HW3SOUND #include "hardware/hw3sound.h" @@ -968,6 +969,33 @@ pflags_t P_GetJumpFlags(player_t *player) return PF_JUMPED; } +// If the state is a custom state for the player's skin, retrieve its "canonical" state +// e.g. S_SKIN_BIGTHECAT_WALK => S_PLAY_WALK +statenum_t P_GetCanonicalPlayerState(player_t *player, statenum_t state) +{ + skin_t *skin = skins[player->skin]; + statenum_t mappedstate; + SIMPLEHASH_FIND_INT(skin->customtodefaultstate, hashentry_int32_int32_t, state, state, mappedstate) + return mappedstate; +} + +boolean P_IsPlayerInState(player_t *player, statenum_t state) +{ + return (P_GetCanonicalPlayerState(player, player->mo->state - states) == state); +} + +boolean P_IsPlayerInSuperTransformationState(player_t *player) +{ + statenum_t state = player->mo->state - states; + return (state >= S_PLAY_SUPER_TRANS1 && state <= S_PLAY_SUPER_TRANS6); +} + +boolean P_IsPlayerInNightsTransformationState(player_t *player) +{ + statenum_t state = player->mo->state - states; + return (state >= S_PLAY_NIGHTS_TRANS1 && state <= S_PLAY_NIGHTS_TRANS6); +} + // // P_PlayerInPain // @@ -983,7 +1011,7 @@ boolean P_PlayerInPain(player_t *player) if (!(player->pflags & PF_SLIDING) && player->mo->state == &states[player->mo->info->painstate] && player->powers[pw_flashing]) return true; - if (player->mo->state == &states[S_PLAY_STUN]) + if (P_IsPlayerInState(player, S_PLAY_STUN)) return true; return false; @@ -1007,7 +1035,7 @@ void P_DoPlayerPain(player_t *player, mobj_t *source, mobj_t *inflictor) fixed_t fallbackspeed; P_ResetPlayer(player); - P_SetMobjState(player->mo, player->mo->info->painstate); + P_SetMobjState(player->mo, S_PLAY_PAIN); if (player->mo->eflags & MFE_VERTICALFLIP) player->mo->z--; @@ -2390,7 +2418,7 @@ boolean P_PlayerHitFloor(player_t *player, boolean dorollstuff) if (player->pflags & PF_BOUNCING) { - if (dorollstuff && player->mo->state-states != S_PLAY_BOUNCE_LANDING) + if (dorollstuff && !P_IsPlayerInState(player, S_PLAY_BOUNCE_LANDING)) { P_MobjCheckWater(player->mo); player->mo->momz *= -1; @@ -2427,9 +2455,9 @@ boolean P_PlayerHitFloor(player_t *player, boolean dorollstuff) player->pflags &= ~PF_GLIDING; } else if (player->charability == CA_GLIDEANDCLIMB && player->pflags & PF_THOKKED && !(player->pflags & (PF_JUMPED|PF_SHIELDABILITY)) - && (player->mo->floorz != player->mo->watertop) && player->mo->state-states == S_PLAY_FALL) + && (player->mo->floorz != player->mo->watertop) && P_IsPlayerInState(player, S_PLAY_FALL)) { - if (player->mo->state-states != S_PLAY_GLIDE_LANDING) + if (!P_IsPlayerInState(player, S_PLAY_GLIDE_LANDING)) { P_ResetPlayer(player); P_SetMobjState(player->mo, S_PLAY_GLIDE_LANDING); @@ -2450,7 +2478,7 @@ boolean P_PlayerHitFloor(player_t *player, boolean dorollstuff) else if (player->charability2 == CA2_MELEE && ((player->panim == PA_ABILITY2) || (player->charability == CA_TWINSPIN && player->panim == PA_ABILITY && player->cmd.buttons & (BT_JUMP|BT_SPIN)))) { - if (player->mo->state-states != S_PLAY_MELEE_LANDING) + if (!P_IsPlayerInState(player, S_PLAY_MELEE_LANDING)) { mobjtype_t type = player->revitem; P_SetMobjState(player->mo, S_PLAY_MELEE_LANDING); @@ -2498,7 +2526,7 @@ boolean P_PlayerHitFloor(player_t *player, boolean dorollstuff) } } } - else if (player->charability == CA_GLIDEANDCLIMB && (player->mo->state-states == S_PLAY_GLIDE_LANDING)) + else if (player->charability == CA_GLIDEANDCLIMB && (P_IsPlayerInState(player, S_PLAY_GLIDE_LANDING))) ; else if (player->charability2 == CA2_GUNSLINGER && player->panim == PA_ABILITY2) ; @@ -2521,10 +2549,10 @@ boolean P_PlayerHitFloor(player_t *player, boolean dorollstuff) if (player->charflags & SF_DASHMODE && player->dashmode >= DASHMODE_THRESHOLD && player->panim != PA_DASH) P_SetMobjState(player->mo, S_PLAY_DASH); else if (player->speed >= runspd - && (player->panim != PA_RUN || player->mo->state-states == S_PLAY_FLOAT_RUN)) + && (player->panim != PA_RUN || P_IsPlayerInState(player, S_PLAY_FLOAT_RUN))) P_SetMobjState(player->mo, S_PLAY_RUN); else if ((player->rmomx || player->rmomy) - && (player->panim != PA_WALK || player->mo->state-states == S_PLAY_FLOAT)) + && (player->panim != PA_WALK || P_IsPlayerInState(player, S_PLAY_FLOAT))) P_SetMobjState(player->mo, S_PLAY_WALK); else if (!player->rmomx && !player->rmomy && player->panim != PA_IDLE) P_SetMobjState(player->mo, S_PLAY_STND); @@ -2534,10 +2562,10 @@ boolean P_PlayerHitFloor(player_t *player, boolean dorollstuff) if (player->charflags & SF_DASHMODE && player->dashmode >= DASHMODE_THRESHOLD && player->panim != PA_DASH) P_SetMobjState(player->mo, S_PLAY_DASH); else if (player->speed >= runspd - && (player->panim != PA_RUN || player->mo->state-states == S_PLAY_FLOAT_RUN)) + && (player->panim != PA_RUN || P_IsPlayerInState(player, S_PLAY_FLOAT_RUN))) P_SetMobjState(player->mo, S_PLAY_RUN); else if ((player->mo->momx || player->mo->momy) - && (player->panim != PA_WALK || player->mo->state-states == S_PLAY_FLOAT)) + && (player->panim != PA_WALK || P_IsPlayerInState(player, S_PLAY_FLOAT))) P_SetMobjState(player->mo, S_PLAY_WALK); else if (!player->mo->momx && !player->mo->momy && player->panim != PA_IDLE) P_SetMobjState(player->mo, S_PLAY_STND); @@ -3711,9 +3739,9 @@ static void P_DoClimbing(player_t *player) climb = false; if (player->climbing && climb && (player->mo->momx || player->mo->momy || player->mo->momz) - && player->mo->state-states != S_PLAY_CLIMB) + && !P_IsPlayerInState(player, S_PLAY_CLIMB)) P_SetMobjState(player->mo, S_PLAY_CLIMB); - else if ((!(player->mo->momx || player->mo->momy || player->mo->momz) || !climb) && player->mo->state-states != S_PLAY_CLING) + else if ((!(player->mo->momx || player->mo->momy || player->mo->momz) || !climb) && !P_IsPlayerInState(player, S_PLAY_CLING)) P_SetMobjState(player->mo, S_PLAY_CLING); if (!floorclimb) @@ -3746,9 +3774,9 @@ static void P_DoClimbing(player_t *player) climb = false; if (player->climbing && climb && (player->mo->momx || player->mo->momy || player->mo->momz) - && player->mo->state-states != S_PLAY_CLIMB) + && !P_IsPlayerInState(player, S_PLAY_CLIMB)) P_SetMobjState(player->mo, S_PLAY_CLIMB); - else if ((!(player->mo->momx || player->mo->momy || player->mo->momz) || !climb) && player->mo->state-states != S_PLAY_CLING) + else if ((!(player->mo->momx || player->mo->momy || player->mo->momz) || !climb) && !P_IsPlayerInState(player, S_PLAY_CLING)) P_SetMobjState(player->mo, S_PLAY_CLING); if (cmd->buttons & BT_SPIN && !(player->pflags & PF_JUMPSTASIS)) @@ -4436,7 +4464,7 @@ static void P_DoSuperStuff(player_t *player) G_GhostAddColor(GHC_SUPER); - if (player->mo->state == &states[S_PLAY_SUPER_TRANS6]) // stop here for now + if (P_IsPlayerInState(player, S_PLAY_SUPER_TRANS6)) // stop here for now return; // Deplete one ring every second while super @@ -4719,7 +4747,7 @@ static void P_DoSpinAbility(player_t *player, ticcmd_t *cmd) { boolean canstand = true; // can we stand on the ground? (mostly relevant for slopes) if (player->pflags & PF_STASIS - && (player->pflags & PF_JUMPSTASIS || player->mo->state-states != S_PLAY_GLIDE_LANDING)) + && (player->pflags & PF_JUMPSTASIS || !P_IsPlayerInState(player, S_PLAY_GLIDE_LANDING))) return; if (cmd->buttons & BT_SPIN) @@ -4739,7 +4767,7 @@ static void P_DoSpinAbility(player_t *player, ticcmd_t *cmd) { case CA2_SPINDASH: // Spinning and Spindashing // Start revving - if ((cmd->buttons & BT_SPIN) && (player->speed < FixedMul(5<<FRACBITS, player->mo->scale) || player->mo->state - states == S_PLAY_GLIDE_LANDING) + if ((cmd->buttons & BT_SPIN) && (player->speed < FixedMul(5<<FRACBITS, player->mo->scale) || P_IsPlayerInState(player, S_PLAY_GLIDE_LANDING)) && !player->mo->momz && onground && !(player->pflags & (PF_SPINDOWN|PF_SPINNING)) && canstand) { @@ -5028,7 +5056,7 @@ void P_DoBubbleBounce(player_t *player) // void P_DoAbilityBounce(player_t *player, boolean changemomz) { - if (player->mo->state-states == S_PLAY_BOUNCE_LANDING) + if (P_IsPlayerInState(player, S_PLAY_BOUNCE_LANDING)) return; if (changemomz) @@ -5202,7 +5230,7 @@ static boolean P_PlayerShieldThink(player_t *player, ticcmd_t *cmd, mobj_t *lock if (!P_MobjWasRemoved(visual)) { P_SetTarget(&visual->target, lockonshield); - visual->drawonlyforplayer = player; // Hide it from the other player in splitscreen, and yourself when spectating + visual->drawonlyforplayer = player; // Hide it from the other player in splitscreen, and yourself when spectating P_SetMobjStateNF(visual, visual->info->spawnstate+1); } } @@ -6109,7 +6137,7 @@ static void P_3dMovement(player_t *player) // When sliding, don't allow forward/back if (player->pflags & PF_SLIDING) cmd->forwardmove = 0; - else if (onground && player->mo->state == states+S_PLAY_PAIN) + else if (onground && P_IsPlayerInState(player, S_PLAY_PAIN)) P_SetMobjState(player->mo, S_PLAY_WALK); player->aiming = cmd->aiming<<FRACBITS; @@ -6162,7 +6190,7 @@ static void P_3dMovement(player_t *player) { if (player->pflags & PF_BOUNCING) { - if (player->mo->state-states == S_PLAY_BOUNCE_LANDING) + if (P_IsPlayerInState(player, S_PLAY_BOUNCE_LANDING)) { thrustfactor = player->thrustfactor*8; acceleration = player->accelstart/8 + (FixedDiv(player->speed, player->mo->scale)>>FRACBITS) * player->acceleration/8; @@ -6915,10 +6943,10 @@ static void P_DoNiGHTSCapsule(player_t *player) { if (player->mo->momx || player->mo->momy || player->mo->momz) { - if (player->mo->state != &states[S_PLAY_NIGHTS_PULL]) + if (!P_IsPlayerInState(player, S_PLAY_NIGHTS_PULL)) P_SetMobjState(player->mo, S_PLAY_NIGHTS_PULL); } - else if (player->mo->state != &states[S_PLAY_NIGHTS_ATTACK]) + else if (!P_IsPlayerInState(player, S_PLAY_NIGHTS_ATTACK)) { S_StartSound(player->mo, sfx_spin); P_SetMobjState(player->mo, S_PLAY_NIGHTS_ATTACK); @@ -6934,7 +6962,7 @@ static void P_DoNiGHTSCapsule(player_t *player) if (!(player->charflags & SF_NONIGHTSROTATION)) { - if ((player->mo->state == &states[S_PLAY_NIGHTS_PULL]) + if ((P_IsPlayerInState(player, S_PLAY_NIGHTS_PULL)) && (player->mo->sprite2 == SPR2_NPUL)) player->mo->spriteroll -= ANG30; else @@ -7249,15 +7277,11 @@ static void P_NiGHTSMovement(player_t *player) if (playeringame[i] /*&& players[i].powers[pw_carry] == CR_NIGHTSMODE*/ && (players[i].capsule && players[i].capsule->reactiontime)) capsule = true; - if (!capsule - && !(player->mo->state >= &states[S_PLAY_NIGHTS_TRANS1] - && player->mo->state <= &states[S_PLAY_NIGHTS_TRANS6]) - && !player->exiting) + if (!capsule && !P_IsPlayerInNightsTransformationState(player) && !player->exiting) player->nightstime--; } else if (!(gametyperules & GTR_RACE) - && !(player->mo->state >= &states[S_PLAY_NIGHTS_TRANS1] - && player->mo->state <= &states[S_PLAY_NIGHTS_TRANS6]) + && !P_IsPlayerInNightsTransformationState(player) && !(player->capsule && player->capsule->reactiontime) && !player->exiting) player->nightstime--; @@ -7397,8 +7421,7 @@ static void P_NiGHTSMovement(player_t *player) return; } - if (player->mo->state >= &states[S_PLAY_NIGHTS_TRANS1] - && player->mo->state <= &states[S_PLAY_NIGHTS_TRANS6]) + if (P_IsPlayerInNightsTransformationState(player)) { player->mo->momx = player->mo->momy = player->mo->momz = 0; player->mo->spriteroll = 0; @@ -7417,14 +7440,14 @@ static void P_NiGHTSMovement(player_t *player) #if 0//def ROTSPRITE if (!(player->charflags & SF_NONIGHTSROTATION) && player->mo->momz) { - if (player->mo->state != &states[S_PLAY_NIGHTS_DRILL]) + if (!P_IsPlayerInState(player, S_PLAY_NIGHTS_DRILL)) P_SetMobjState(player->mo, S_PLAY_NIGHTS_DRILL); player->mo->spriteroll = ANGLE_90; } else #endif { - if (player->mo->state != &states[S_PLAY_NIGHTS_FLOAT]) + if (!P_IsPlayerInState(player, S_PLAY_NIGHTS_FLOAT)) P_SetMobjState(player->mo, S_PLAY_NIGHTS_FLOAT); player->drawangle += ANGLE_22h; } @@ -8056,7 +8079,7 @@ static void P_SkidStuff(player_t *player) // Spawn a particle every 3 tics. if (!(player->skidtime % 3)) { - if (player->mo->state-states == S_PLAY_GLIDE_LANDING) + if (P_IsPlayerInState(player, S_PLAY_GLIDE_LANDING)) P_SpawnSkidDust(player, player->mo->radius, true); else P_SpawnSkidDust(player, 0, false); @@ -8076,7 +8099,7 @@ static void P_SkidStuff(player_t *player) // If your push angle is more than this close to a full 180 degrees, trigger a skid. if (dang > ANGLE_157h) { - if (player->mo->state-states != S_PLAY_SKID) + if (!P_IsPlayerInState(player, S_PLAY_SKID)) P_SetMobjState(player->mo, S_PLAY_SKID); player->mo->tics = player->skidtime = (player->mo->movefactor == FRACUNIT) ? TICRATE/2 : (FixedDiv(35<<(FRACBITS-1), FixedSqrt(player->mo->movefactor)))>>FRACBITS; S_StartSound(player->mo, sfx_skid); @@ -8100,7 +8123,7 @@ void P_MovePlayer(player_t *player) fixed_t runspd; - if (player->mo->state >= &states[S_PLAY_SUPER_TRANS1] && player->mo->state <= &states[S_PLAY_SUPER_TRANS6]) + if (P_IsPlayerInSuperTransformationState(player)) { player->mo->momx = player->mo->momy = player->mo->momz = 0; return; @@ -8121,7 +8144,7 @@ void P_MovePlayer(player_t *player) if ((player->powers[pw_carry] == CR_BRAKGOOP) || (player->pflags & PF_GLIDING && player->skidtime) || (player->charability2 == CA2_GUNSLINGER && player->panim == PA_ABILITY2) - || (player->charability2 == CA2_MELEE && player->mo->state-states == S_PLAY_MELEE_LANDING)) + || (player->charability2 == CA2_MELEE && P_IsPlayerInState(player, S_PLAY_MELEE_LANDING))) player->pflags |= PF_FULLSTASIS; else if (player->powers[pw_nocontrol]) { @@ -8130,7 +8153,7 @@ void P_MovePlayer(player_t *player) player->pflags |= PF_JUMPSTASIS; } - if (player->charability == CA_GLIDEANDCLIMB && player->mo->state-states == S_PLAY_GLIDE_LANDING) + if (player->charability == CA_GLIDEANDCLIMB && P_IsPlayerInState(player, S_PLAY_GLIDE_LANDING)) { player->pflags |= PF_STASIS; } @@ -8309,7 +8332,7 @@ void P_MovePlayer(player_t *player) { // If the player is in dashmode, here's their peelout. if (player->charflags & SF_DASHMODE && player->dashmode >= DASHMODE_THRESHOLD && player->panim == PA_RUN && !player->skidtime && (onground || ((player->charability == CA_FLOAT || player->charability == CA_SLOWFALL) && player->secondjump == 1) || player->powers[pw_super])) - P_SetMobjState (player->mo, S_PLAY_DASH); + P_SetMobjState(player->mo, S_PLAY_DASH); // If the player is moving fast enough, // break into a run! else if (player->speed >= runspd && player->panim == PA_WALK && !player->skidtime @@ -8323,11 +8346,11 @@ void P_MovePlayer(player_t *player) // Floating at slow speeds has its own special animation. else if ((((player->charability == CA_FLOAT || player->charability == CA_SLOWFALL) && player->secondjump == 1) || player->powers[pw_super]) && player->panim == PA_IDLE && !onground) - P_SetMobjState (player->mo, S_PLAY_FLOAT); + P_SetMobjState(player->mo, S_PLAY_FLOAT); // Otherwise, just walk. else if ((player->rmomx || player->rmomy) && player->panim == PA_IDLE) - P_SetMobjState (player->mo, S_PLAY_WALK); + P_SetMobjState(player->mo, S_PLAY_WALK); } // If your peelout animation is playing, and you're @@ -8348,9 +8371,9 @@ void P_MovePlayer(player_t *player) // Correct floating when ending up on the ground. if (onground) { - if (player->mo->state-states == S_PLAY_FLOAT) + if (P_IsPlayerInState(player, S_PLAY_FLOAT)) P_SetMobjState(player->mo, S_PLAY_WALK); - else if (player->mo->state-states == S_PLAY_FLOAT_RUN) + else if (P_IsPlayerInState(player, S_PLAY_FLOAT_RUN)) P_SetMobjState(player->mo, S_PLAY_RUN); } @@ -8416,7 +8439,7 @@ void P_MovePlayer(player_t *player) fixed_t glidespeed = player->actionspd; fixed_t momx = mo->momx - player->cmomx, momy = mo->momy - player->cmomy; angle_t angle, moveangle = R_PointToAngle2(0, 0, momx, momy); - boolean swimming = mo->state - states == S_PLAY_SWIM; + boolean swimming = P_IsPlayerInState(player, S_PLAY_SWIM); boolean in2d = mo->flags2 & MF2_TWOD || twodlevel; if (player->powers[pw_super] || player->powers[pw_sneakers]) @@ -8563,7 +8586,7 @@ void P_MovePlayer(player_t *player) } } } - else if (player->mo->state-states == S_PLAY_BOUNCE) + else if (P_IsPlayerInState(player, S_PLAY_BOUNCE)) P_SetMobjState(player->mo, S_PLAY_FALL); // If you're running fast enough, you can create splashes as you run in shallow water. @@ -8607,7 +8630,7 @@ void P_MovePlayer(player_t *player) if (!(player->charability == CA_FLY || player->charability == CA_SWIM)) // why are you flying when you cannot fly?! { if (player->powers[pw_tailsfly] - || player->mo->state-states == S_PLAY_FLY_TIRED) + || P_IsPlayerInState(player, S_PLAY_FLY_TIRED)) { if (onground) P_SetMobjState(player->mo, S_PLAY_WALK); @@ -8675,11 +8698,11 @@ void P_MovePlayer(player_t *player) else { // Tails-gets-tired Stuff - if (player->panim == PA_ABILITY && player->mo->state-states != S_PLAY_FLY_TIRED) + if (player->panim == PA_ABILITY && !P_IsPlayerInState(player, S_PLAY_FLY_TIRED)) P_SetMobjState(player->mo, S_PLAY_FLY_TIRED); if (player->charability == CA_FLY && (leveltime % 10 == 0) - && player->mo->state-states == S_PLAY_FLY_TIRED + && P_IsPlayerInState(player, S_PLAY_FLY_TIRED) && !(player->mo->eflags & MFE_UNDERWATER) && !player->spectator) S_StartSound(player->mo, sfx_pudpud); @@ -8774,7 +8797,7 @@ void P_MovePlayer(player_t *player) } // Otherwise, face the direction you're travelling. else if (player->panim == PA_WALK || player->panim == PA_RUN || player->panim == PA_DASH || player->panim == PA_ROLL || player->panim == PA_JUMP - || (player->panim == PA_ABILITY && player->mo->state-states == S_PLAY_GLIDE)) + || (player->panim == PA_ABILITY && P_IsPlayerInState(player, S_PLAY_GLIDE))) player->mo->angle = R_PointToAngle2(0, 0, player->rmomx, player->rmomy); // Update the local angle control. @@ -9035,7 +9058,7 @@ static void P_DoRopeHang(player_t *player) return; } - if (player->mo->state-states != S_PLAY_RIDE) + if (!P_IsPlayerInState(player, S_PLAY_RIDE)) P_SetMobjState(player->mo, S_PLAY_RIDE); // If not allowed to move, we're done here. @@ -9972,9 +9995,7 @@ boolean P_MoveChaseCamera(player_t *player, camera_t *thiscam, boolean resetcall && !((gametyperules & GTR_FRIENDLY) && (netgame || multiplayer) && cv_exitmove.value) && !(twodlevel || (mo->flags2 & MF2_TWOD))) sign = mo->target; - else if ((player->powers[pw_carry] == CR_NIGHTSMODE) - && !(player->mo->state >= &states[S_PLAY_NIGHTS_TRANS1] - && player->mo->state <= &states[S_PLAY_NIGHTS_TRANS6])) + else if (player->powers[pw_carry] == CR_NIGHTSMODE && !P_IsPlayerInNightsTransformationState(player)) { P_CalcChasePostImg(player, thiscam); return true; @@ -11291,7 +11312,7 @@ static void P_MinecartThink(player_t *player) } } - if (player->mo->state-states != S_PLAY_STND) + if (!P_IsPlayerInState(player, S_PLAY_STND)) { P_SetMobjState(player->mo, S_PLAY_STND); player->mo->tics = -1; @@ -11369,12 +11390,12 @@ void P_DoTailsOverlay(player_t *player, mobj_t *tails) } else if (player->panim == PA_PAIN) backwards /= 16; - else if (player->mo->state-states == S_PLAY_GASP) + else if (P_IsPlayerInState(player, S_PLAY_GASP)) { backwards /= 16; zoffs += 12*FRACUNIT; } - else if (player->mo->state-states == S_PLAY_EDGE) + else if (P_IsPlayerInState(player, S_PLAY_EDGE)) { backwards /= 16; zoffs = 3*FRACUNIT; @@ -11403,13 +11424,13 @@ void P_DoTailsOverlay(player_t *player, mobj_t *tails) } else if (player->panim == PA_SPRING || player->panim == PA_JUMP) chosenstate = S_TAILSOVERLAY_MINUS60DEGREES; - else if (player->panim == PA_FALL || player->mo->state-states == S_PLAY_RIDE) + else if (player->panim == PA_FALL || P_IsPlayerInState(player, S_PLAY_RIDE)) chosenstate = S_TAILSOVERLAY_PLUS60DEGREES; else if (player->panim == PA_PAIN) chosenstate = S_TAILSOVERLAY_PAIN; - else if (player->mo->state-states == S_PLAY_GASP) + else if (P_IsPlayerInState(player, S_PLAY_GASP)) chosenstate = S_TAILSOVERLAY_GASP; - else if (player->mo->state-states == S_PLAY_EDGE) + else if (P_IsPlayerInState(player, S_PLAY_EDGE)) chosenstate = S_TAILSOVERLAY_EDGE; else if (player->panim == PA_DASH) chosenstate = S_TAILSOVERLAY_DASH; @@ -11417,7 +11438,7 @@ void P_DoTailsOverlay(player_t *player, mobj_t *tails) chosenstate = S_TAILSOVERLAY_RUN; else if (player->panim == PA_WALK) { - if (!smilesonground || player->mo->state-states == S_PLAY_SKID) + if (!smilesonground || P_IsPlayerInState(player, S_PLAY_SKID)) chosenstate = S_TAILSOVERLAY_PLUS30DEGREES; else if (player->speed >= FixedMul(player->runspeed/2, player->mo->scale)) chosenstate = S_TAILSOVERLAY_0DEGREES; @@ -11459,7 +11480,7 @@ void P_DoTailsOverlay(player_t *player, mobj_t *tails) #endif // animation... - if (player->panim == PA_SPRING || player->panim == PA_FALL || player->mo->state-states == S_PLAY_RIDE) + if (player->panim == PA_SPRING || player->panim == PA_FALL || P_IsPlayerInState(player, S_PLAY_RIDE)) { if (FixedDiv(abs(player->mo->momz), player->mo->scale) < 20<<FRACBITS) ticnum = 2; @@ -11468,7 +11489,7 @@ void P_DoTailsOverlay(player_t *player, mobj_t *tails) } else if (player->panim == PA_PAIN) ticnum = 2; - else if (player->mo->state-states == S_PLAY_GASP) + else if (P_IsPlayerInState(player, S_PLAY_GASP)) tails->tics = -1; else if (player->mo->sprite2 == SPR2_TIRE) ticnum = (doswim ? 2 : 4); @@ -12291,10 +12312,9 @@ void P_PlayerThink(player_t *player) diff = InvAngle(diff); if (diff > ANG10/2) { - statenum_t stat = player->mo->state-states; - if (stat == S_PLAY_WAIT) + if (P_IsPlayerInState(player, S_PLAY_WAIT)) P_SetMobjState(player->mo, S_PLAY_STND); - else if (stat == S_PLAY_STND && player->mo->tics != -1) + else if (P_IsPlayerInState(player, S_PLAY_STND) && player->mo->tics != -1) player->mo->tics++; } } @@ -12319,7 +12339,7 @@ void P_PlayerThink(player_t *player) // fake skidding! see P_SkidStuff for reference on conditionals else if (!player->skidtime && !(player->mo->eflags & MFE_GOOWATER) && !(player->pflags & (PF_JUMPED|PF_SPINNING|PF_SLIDING)) && !(player->charflags & SF_NOSKID) && P_AproxDistance(player->mo->momx, player->mo->momy) >= FixedMul(player->runspeed, player->mo->scale)) // modified from player->runspeed/2 'cuz the skid was just TOO frequent ngl { - if (player->mo->state-states != S_PLAY_SKID) + if (!P_IsPlayerInState(player, S_PLAY_SKID)) P_SetMobjState(player->mo, S_PLAY_SKID); player->mo->tics = player->skidtime = (player->mo->movefactor == FRACUNIT) ? TICRATE/2 : (FixedDiv(35<<(FRACBITS-1), FixedSqrt(player->mo->movefactor)))>>FRACBITS; @@ -12490,16 +12510,14 @@ void P_PlayerThink(player_t *player) player->stronganim = 0; //pw_super acts as a timer now - if (player->powers[pw_super] - && (player->mo->state < &states[S_PLAY_SUPER_TRANS1] - || player->mo->state > &states[S_PLAY_SUPER_TRANS6])) + if (player->powers[pw_super] && !P_IsPlayerInSuperTransformationState(player)) player->powers[pw_super]++; if (player->powers[pw_carry] == CR_BRAKGOOP) { if (!player->powers[pw_flashing]) { - if (player->mo->state != &states[S_PLAY_STND]) + if (!P_IsPlayerInState(player, S_PLAY_STND)) P_SetMobjState(player->mo, S_PLAY_STND); else player->mo->tics = 2; @@ -12864,7 +12882,7 @@ void P_PlayerAfterThink(player_t *player) S_StartSound(NULL, sfx_wepchg); if ((player->pflags & PF_SLIDING) && ((player->pflags & (PF_JUMPED|PF_NOJUMPDAMAGE)) != PF_JUMPED)) - P_SetMobjState(player->mo, player->mo->info->painstate); + P_SetMobjState(player->mo, S_PLAY_PAIN); /* if (player->powers[pw_carry] == CR_NONE && player->mo->tracer && !player->homing) P_SetTarget(&player->mo->tracer, NULL); @@ -12924,7 +12942,7 @@ void P_PlayerAfterThink(player_t *player) if (player->powers[pw_carry] == CR_PLAYER) { - if (player->mo->state-states != S_PLAY_RIDE) + if (!P_IsPlayerInState(player, S_PLAY_RIDE)) P_SetMobjState(player->mo, S_PLAY_RIDE); if (tails->player && (tails->skin && ((skin_t *)(tails->skin))->sprites[SPR2_SWIM].numframes) && (tails->eflags & MFE_UNDERWATER)) tails->player->powers[pw_tailsfly] = 0; @@ -12949,7 +12967,7 @@ void P_PlayerAfterThink(player_t *player) player->mo->z = item->z - FixedDiv(player->mo->height, 3*FRACUNIT/2); player->mo->momx = player->mo->momy = player->mo->momz = 0; P_SetThingPosition(player->mo); - if (player->mo->state-states != S_PLAY_RIDE) + if (!P_IsPlayerInState(player, S_PLAY_RIDE)) P_SetMobjState(player->mo, S_PLAY_RIDE); // Controllable missile @@ -13147,7 +13165,7 @@ void P_PlayerAfterThink(player_t *player) ptera->waterbottom >>= 1; ptera->cvmem >>= 1; - if (player->mo->state-states != S_PLAY_FALL) + if (!P_IsPlayerInState(player, S_PLAY_FALL)) P_SetMobjState(player->mo, S_PLAY_FALL); break; @@ -13294,9 +13312,9 @@ boolean P_PlayerCanEnterSpinGaps(player_t *player) return false; return ((player->pflags & (PF_SPINNING|PF_SLIDING|PF_GLIDING)) // players who are spinning, sliding, or gliding - || (player->charability == CA_GLIDEANDCLIMB && player->mo->state-states == S_PLAY_GLIDE_LANDING) // players who are landing from a glide + || (player->charability == CA_GLIDEANDCLIMB && P_IsPlayerInState(player, S_PLAY_GLIDE_LANDING)) // players who are landing from a glide || ((player->charflags & (SF_DASHMODE|SF_MACHINE)) == (SF_DASHMODE|SF_MACHINE) - && player->dashmode >= DASHMODE_THRESHOLD && player->mo->state-states == S_PLAY_DASH) // machine players in dashmode + && player->dashmode >= DASHMODE_THRESHOLD && P_IsPlayerInState(player, S_PLAY_DASH)) // machine players in dashmode || JUMPCURLED(player)); // players who are jumpcurled, but only if they would normally jump that way } @@ -13304,13 +13322,13 @@ boolean P_PlayerCanEnterSpinGaps(player_t *player) boolean P_PlayerShouldUseSpinHeight(player_t *player) { return ((player->pflags & (PF_SPINNING|PF_SLIDING|PF_GLIDING)) - || (player->mo->state == &states[player->mo->info->painstate]) + || P_IsPlayerInState(player, S_PLAY_PAIN) || (player->panim == PA_ROLL) - || ((player->powers[pw_tailsfly] || (player->charability == CA_FLY && player->mo->state-states == S_PLAY_FLY_TIRED)) + || ((player->powers[pw_tailsfly] || (player->charability == CA_FLY && P_IsPlayerInState(player, S_PLAY_FLY_TIRED))) && !(player->charflags & SF_NOJUMPSPIN)) - || (player->charability == CA_GLIDEANDCLIMB && player->mo->state-states == S_PLAY_GLIDE_LANDING) + || (player->charability == CA_GLIDEANDCLIMB && P_IsPlayerInState(player, S_PLAY_GLIDE_LANDING)) || ((player->charflags & (SF_DASHMODE|SF_MACHINE)) == (SF_DASHMODE|SF_MACHINE) - && player->dashmode >= DASHMODE_THRESHOLD && player->mo->state-states == S_PLAY_DASH) + && player->dashmode >= DASHMODE_THRESHOLD && P_IsPlayerInState(player, S_PLAY_DASH)) || JUMPCURLED(player)); } diff --git a/src/r_patchrotation.c b/src/r_patchrotation.c index 989665d74179854fcfac552814132671c3d00ed2..7c4f6257eeb029ccb3a0fa30b9a44222fe1a7958 100644 --- a/src/r_patchrotation.c +++ b/src/r_patchrotation.c @@ -54,12 +54,26 @@ INT32 R_GetRollAngle(angle_t rollangle) patch_t *Patch_GetRotated(patch_t *patch, INT32 angle, boolean flip) { rotsprite_t *rotsprite = patch->rotated; - if (rotsprite == NULL || angle < 1 || angle >= ROTANGLES) + if (angle < 1 || angle >= ROTANGLES) return NULL; + if (rotsprite == NULL) + { + rotsprite = RotatedPatch_Create(ROTANGLES); + patch->rotated = rotsprite; + } + if (flip) angle += rotsprite->angles; + if (rotsprite->patches[angle] == NULL) + { + INT32 xpivot = 0, ypivot = 0; + xpivot = patch->leftoffset; + ypivot = patch->topoffset; + RotatedPatch_DoRotation(rotsprite, patch, angle, xpivot, ypivot, flip); + } + return rotsprite->patches[angle]; } diff --git a/src/r_skins.c b/src/r_skins.c index 7b5e547148a6ba975775dfc933602614be33921d..47ead32637c869af7ea44cb668a43dfc6bac8631 100644 --- a/src/r_skins.c +++ b/src/r_skins.c @@ -28,6 +28,7 @@ #include "p_local.h" #include "dehacked.h" // get_number (for thok) #include "m_cond.h" +#include "deh_tables.h" #ifdef HWRENDER #include "hardware/hw_md2.h" #endif @@ -791,6 +792,88 @@ static boolean R_ProcessPatchableFields(skin_t *skin, char *stoken, char *value) return true; } +// e.g. "RUN" => S_PLAY_RUN, "RUN1" => S_PLAY_RUN, "RUN8" => S_PLAY_RUN +static statenum_t GetCanonicalPlayerStateNumByName(const char *name) +{ + if (startswith(name, "STAND")) return S_PLAY_STND; + else if (startswith(name, "STND")) return S_PLAY_STND; + else if (startswith(name, "WAIT")) return S_PLAY_WAIT; + else if (startswith(name, "WALK")) return S_PLAY_WALK; + else if (startswith(name, "SKID")) return S_PLAY_SKID; + else if (startswith(name, "RUN")) return S_PLAY_RUN; + else if (startswith(name, "DASH")) return S_PLAY_DASH; + else if (startswith(name, "PAIN")) return S_PLAY_PAIN; + else if (startswith(name, "STUN")) return S_PLAY_STUN; + else if (startswith(name, "DEAD")) return S_PLAY_DEAD; + else if (startswith(name, "DRWN")) return S_PLAY_DRWN; + else if (startswith(name, "ROLL")) return S_PLAY_ROLL; + else if (startswith(name, "GASP")) return S_PLAY_GASP; + else if (startswith(name, "JUMP")) return S_PLAY_JUMP; + else if (startswith(name, "SPRING")) return S_PLAY_SPRING; + else if (startswith(name, "FALL")) return S_PLAY_FALL; + else if (startswith(name, "EDGE")) return S_PLAY_EDGE; + else if (startswith(name, "RIDE")) return S_PLAY_RIDE; + else if (startswith(name, "SPINDASH")) return S_PLAY_SPINDASH; + else if (startswith(name, "FLY")) return S_PLAY_FLY; + else if (startswith(name, "SWIM")) return S_PLAY_SWIM; + else if (startswith(name, "FLY_TIRED")) return S_PLAY_FLY_TIRED; + else if (startswith(name, "GLIDE")) return S_PLAY_GLIDE; + else if (startswith(name, "GLIDE_LANDING")) return S_PLAY_GLIDE_LANDING; + else if (startswith(name, "CLING")) return S_PLAY_CLING; + else if (startswith(name, "CLIMB")) return S_PLAY_CLIMB; + else if (startswith(name, "FLOAT")) return S_PLAY_FLOAT; + else if (startswith(name, "FLOAT_RUN")) return S_PLAY_FLOAT_RUN; + else if (startswith(name, "BOUNCE")) return S_PLAY_BOUNCE; + else if (startswith(name, "BOUNCE_LANDING")) return S_PLAY_BOUNCE_LANDING; + else if (startswith(name, "FIRE")) return S_PLAY_FIRE; + else if (startswith(name, "FIRE_FINISH")) return S_PLAY_FIRE_FINISH; + else if (startswith(name, "TWINSPIN")) return S_PLAY_TWINSPIN; + else if (startswith(name, "MELEE")) return S_PLAY_MELEE; + else if (startswith(name, "MELEE_FINISH")) return S_PLAY_MELEE_FINISH; + else if (startswith(name, "MELEE_LANDING")) return S_PLAY_MELEE_LANDING; + else if (startswith(name, "NIGHTS_STAND")) return S_PLAY_NIGHTS_STAND; + else if (startswith(name, "NIGHTS_FLOAT")) return S_PLAY_NIGHTS_FLOAT; + else if (startswith(name, "NIGHTS_FLY")) return S_PLAY_NIGHTS_FLY; + else if (startswith(name, "NIGHTS_DRILL")) return S_PLAY_NIGHTS_DRILL; + else if (startswith(name, "NIGHTS_STUN")) return S_PLAY_NIGHTS_STUN; + else if (startswith(name, "NIGHTS_PULL")) return S_PLAY_NIGHTS_PULL; + else if (startswith(name, "NIGHTS_ATTACK")) return S_PLAY_NIGHTS_ATTACK; + else return S_NULL; +} + +static void CacheCustomSkinStates(skin_t *skin) +{ + SIMPLEHASH_CLEAR(skin->defaulttocustomstate, hashentry_int32_int32_t) + SIMPLEHASH_CLEAR(skin->customtodefaultstate, hashentry_int32_int32_t) + + char *skinstateprefix = va("SKIN_%s_", skin->name); + strupr(skinstateprefix); + size_t skinstateprefixlen = strlen(skinstateprefix); + + for (INT32 state = S_FIRSTFREESLOT; state <= S_LASTFREESLOT; state++) + { + const char *statename = FREE_STATES[state - S_FIRSTFREESLOT]; + + if (!statename) + continue; + if (strncmp(statename, skinstateprefix, skinstateprefixlen)) + continue; + + statenum_t defaultstate = GetCanonicalPlayerStateNumByName(&statename[skinstateprefixlen]); + if (defaultstate) + { + // If a default state is overriden by multiple custom states, use the first one as reference + // e.g. WALK+WALK2+WALK3+WALK4 instead of just WALK + statenum_t alreadyoverriden; + SIMPLEHASH_FIND_INT(skin->defaulttocustomstate, hashentry_int32_int32_t, defaultstate, S_NULL, alreadyoverriden) + if (!alreadyoverriden) + SIMPLEHASH_REPLACE_INT(skin->defaulttocustomstate, hashentry_int32_int32_t, defaultstate, state) + + SIMPLEHASH_REPLACE_INT(skin->customtodefaultstate, hashentry_int32_int32_t, state, defaultstate) + } + } +} + // // Find skin sprites, sounds & optional status bar face, & add them // @@ -945,6 +1028,8 @@ next_token: R_FlushTranslationColormapCache(); + CacheCustomSkinStates(skin); + if (mainfile == false) CONS_Printf(M_GetText("Added skin '%s'\n"), skin->name); @@ -1082,6 +1167,8 @@ next_token: R_FlushTranslationColormapCache(); + CacheCustomSkinStates(skin); + if (mainfile == false) CONS_Printf(M_GetText("Patched skin '%s'\n"), skin->name); } diff --git a/src/r_skins.h b/src/r_skins.h index 645fea540c53dd26e61cc7f3a2e71ecc3c365d1a..7df0d584bd88a09b25a6e8a31b2da351a1a06db5 100644 --- a/src/r_skins.h +++ b/src/r_skins.h @@ -20,6 +20,7 @@ #include "r_patch.h" #include "r_picformats.h" // spriteinfo_t #include "r_defs.h" // spritedef_t +#include "simple_hashmap.h" /// Defaults #define SKINNAMESIZE 16 @@ -89,6 +90,9 @@ typedef struct spriteinfo_t sprinfo[NUMPLAYERSPRITES]; } super; + hashentry_int32_int32_t *defaulttocustomstate; // e.g. S_PLAY_WALK => S_SKIN_BIGTHECAT_WALK + hashentry_int32_int32_t *customtodefaultstate; // e.g. S_SKIN_BIGTHECAT_WALK => S_PLAY_WALK + // TODO: 2.3: Delete spritedef_t sprites_compat[NUMPLAYERSPRITES * 2]; } skin_t; diff --git a/src/sdl/Srb2SDL-vc10.vcxproj b/src/sdl/Srb2SDL-vc10.vcxproj index 8026cd3acbdd4adb6652269dbf308c8ec6fa4e12..2a25f33031adee58358a423e0251937995e5471c 100644 --- a/src/sdl/Srb2SDL-vc10.vcxproj +++ b/src/sdl/Srb2SDL-vc10.vcxproj @@ -513,6 +513,7 @@ <ClCompile Include="..\lua_hudlib_drawlist.c" /> <ClCompile Include="..\lua_infolib.c" /> <ClCompile Include="..\lua_inputlib.c" /> + <ClCompile Include="..\lua_interceptlib.c" /> <ClCompile Include="..\lua_maplib.c" /> <ClCompile Include="..\lua_mathlib.c" /> <ClCompile Include="..\lua_mobjlib.c" /> diff --git a/src/sdl/Srb2SDL-vc10.vcxproj.filters b/src/sdl/Srb2SDL-vc10.vcxproj.filters index bbfca04d4c1132bc5241e02e612aa6fc2e48c7e5..8daaa7773f091d5dfbf6bac965b58160380bb50c 100644 --- a/src/sdl/Srb2SDL-vc10.vcxproj.filters +++ b/src/sdl/Srb2SDL-vc10.vcxproj.filters @@ -783,6 +783,9 @@ </ClCompile> <ClCompile Include="..\lua_inputlib.c"> <Filter>LUA</Filter> + </ClCompile> + <ClCompile Include="..\lua_interceptlib.c"> + <Filter>LUA</Filter> </ClCompile> <ClCompile Include="..\lua_maplib.c"> <Filter>LUA</Filter> diff --git a/src/sdl/i_video.c b/src/sdl/i_video.c index 2d1029ed09e50dd3b1834e6f421868aef8d752c8..a1ff94480ecbfbfe37fa5337831c8245790cf556 100644 --- a/src/sdl/i_video.c +++ b/src/sdl/i_video.c @@ -1858,7 +1858,7 @@ void I_StartupGraphics(void) SDLdoGrabMouse(); // disable text input right off the bat, since we don't need it at the start. - I_SetTextInputMode(false); + I_SetTextInputMode(textinputmodeenabledbylua); graphics_started = true; } diff --git a/src/simple_hashmap.h b/src/simple_hashmap.h new file mode 100644 index 0000000000000000000000000000000000000000..0e322455b057446b13f552bcc3716194589fe14e --- /dev/null +++ b/src/simple_hashmap.h @@ -0,0 +1,58 @@ +// SONIC ROBO BLAST 2 +//----------------------------------------------------------------------------- +// Copyright (C) 2024 by Sonic Team Junior. +// +// This program is free software distributed under the +// terms of the GNU General Public License, version 2. +// See the 'LICENSE' file for more details. +//----------------------------------------------------------------------------- +/// \file simplehash.h +/// \brief Macros for handling basic hashmap types + +#ifndef __SIMPLEHASH__ +#define __SIMPLEHASH__ + +#include "uthash.h" + +typedef struct +{ + INT32 k; + INT32 v; + UT_hash_handle hh; +} hashentry_int32_int32_t; + +// hashmap<type>[key] = value; +#define SIMPLEHASH_REPLACE_INT(hashmap, type, key, value) \ +{ \ + type *entry = malloc(sizeof(type)); \ + if (!entry) \ + I_Error("%s:%d: Out of memory\n", __func__, __LINE__); \ + entry->k = (key); \ + entry->v = (value); \ + \ + type *oldentry; \ + HASH_REPLACE_INT((hashmap), k, entry, oldentry); \ + if (oldentry) \ + free(oldentry); \ +} + +#define SIMPLEHASH_CLEAR(hashmap, type) \ +{ \ + type *entry, *tmpentry; \ + HASH_ITER(hh, (hashmap), entry, tmpentry) \ + { \ + HASH_DEL((hashmap), entry); \ + free(entry); \ + } \ +} + +// value = hashmap<type>[key] or fallback; +#define SIMPLEHASH_FIND_INT(hashmap, type, key, fallback, value) \ +{ \ + int tmpkey = (key); \ + type *entry; \ + HASH_FIND_INT((hashmap), &tmpkey, entry); \ + (value) = entry ? entry->v : (int)(fallback); \ +} + +#endif //__SIMPLEHASH__ diff --git a/src/snake.c b/src/snake.c index 4219d5b8fa54ee69ce1b99167e697e935cd3483b..1276daafd0ae83f738a9cea0becefbc252097f20 100644 --- a/src/snake.c +++ b/src/snake.c @@ -23,7 +23,7 @@ #define SPEED 5 #define NUM_BLOCKS_X 20 -#define NUM_BLOCKS_Y 10 +#define NUM_BLOCKS_Y 8 #define BLOCK_SIZE 12 #define BORDER_SIZE 12 @@ -32,7 +32,7 @@ #define LEFT_X ((BASEVIDWIDTH - MAP_WIDTH) / 2 - BORDER_SIZE) #define RIGHT_X (LEFT_X + MAP_WIDTH + BORDER_SIZE * 2 - 1) -#define BOTTOM_Y (BASEVIDHEIGHT - 48) +#define BOTTOM_Y (BASEVIDHEIGHT - 76) #define TOP_Y (BOTTOM_Y - MAP_HEIGHT - BORDER_SIZE * 2 + 1) enum bonustype_s { diff --git a/src/uthash.h b/src/uthash.h new file mode 100644 index 0000000000000000000000000000000000000000..68693bf39603d91a529ca8f8c46ad720aa4f094a --- /dev/null +++ b/src/uthash.h @@ -0,0 +1,1140 @@ +/* +Copyright (c) 2003-2022, Troy D. Hanson https://troydhanson.github.io/uthash/ +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. + +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. +*/ + +#ifndef UTHASH_H +#define UTHASH_H + +#define UTHASH_VERSION 2.3.0 + +#include <string.h> /* memcmp, memset, strlen */ +#include <stddef.h> /* ptrdiff_t */ +#include <stdlib.h> /* exit */ + +#if defined(HASH_DEFINE_OWN_STDINT) && HASH_DEFINE_OWN_STDINT +/* This codepath is provided for backward compatibility, but I plan to remove it. */ +#warning "HASH_DEFINE_OWN_STDINT is deprecated; please use HASH_NO_STDINT instead" +typedef unsigned int uint32_t; +typedef unsigned char uint8_t; +#elif defined(HASH_NO_STDINT) && HASH_NO_STDINT +#else +#include <stdint.h> /* uint8_t, uint32_t */ +#endif + +/* These macros use decltype or the earlier __typeof GNU extension. + As decltype is only available in newer compilers (VS2010 or gcc 4.3+ + when compiling c++ source) this code uses whatever method is needed + or, for VS2008 where neither is available, uses casting workarounds. */ +#if !defined(DECLTYPE) && !defined(NO_DECLTYPE) +#if defined(_MSC_VER) /* MS compiler */ +#if _MSC_VER >= 1600 && defined(__cplusplus) /* VS2010 or newer in C++ mode */ +#define DECLTYPE(x) (decltype(x)) +#else /* VS2008 or older (or VS2010 in C mode) */ +#define NO_DECLTYPE +#endif +#elif defined(__MCST__) /* Elbrus C Compiler */ +#define DECLTYPE(x) (__typeof(x)) +#elif defined(__BORLANDC__) || defined(__ICCARM__) || defined(__LCC__) || defined(__WATCOMC__) +#define NO_DECLTYPE +#else /* GNU, Sun and other compilers */ +#define DECLTYPE(x) (__typeof(x)) +#endif +#endif + +#ifdef NO_DECLTYPE +#define DECLTYPE(x) +#define DECLTYPE_ASSIGN(dst,src) \ +do { \ + char **_da_dst = (char**)(&(dst)); \ + *_da_dst = (char*)(src); \ +} while (0) +#else +#define DECLTYPE_ASSIGN(dst,src) \ +do { \ + (dst) = DECLTYPE(dst)(src); \ +} while (0) +#endif + +#ifndef uthash_malloc +#define uthash_malloc(sz) malloc(sz) /* malloc fcn */ +#endif +#ifndef uthash_free +#define uthash_free(ptr,sz) free(ptr) /* free fcn */ +#endif +#ifndef uthash_bzero +#define uthash_bzero(a,n) memset(a,'\0',n) +#endif +#ifndef uthash_strlen +#define uthash_strlen(s) strlen(s) +#endif + +#ifndef HASH_FUNCTION +#define HASH_FUNCTION(keyptr,keylen,hashv) HASH_JEN(keyptr, keylen, hashv) +#endif + +#ifndef HASH_KEYCMP +#define HASH_KEYCMP(a,b,n) memcmp(a,b,n) +#endif + +#ifndef uthash_noexpand_fyi +#define uthash_noexpand_fyi(tbl) /* can be defined to log noexpand */ +#endif +#ifndef uthash_expand_fyi +#define uthash_expand_fyi(tbl) /* can be defined to log expands */ +#endif + +#ifndef HASH_NONFATAL_OOM +#define HASH_NONFATAL_OOM 0 +#endif + +#if HASH_NONFATAL_OOM +/* malloc failures can be recovered from */ + +#ifndef uthash_nonfatal_oom +#define uthash_nonfatal_oom(obj) do {} while (0) /* non-fatal OOM error */ +#endif + +#define HASH_RECORD_OOM(oomed) do { (oomed) = 1; } while (0) +#define IF_HASH_NONFATAL_OOM(x) x + +#else +/* malloc failures result in lost memory, hash tables are unusable */ + +#ifndef uthash_fatal +#define uthash_fatal(msg) exit(-1) /* fatal OOM error */ +#endif + +#define HASH_RECORD_OOM(oomed) uthash_fatal("out of memory") +#define IF_HASH_NONFATAL_OOM(x) + +#endif + +/* initial number of buckets */ +#define HASH_INITIAL_NUM_BUCKETS 32U /* initial number of buckets */ +#define HASH_INITIAL_NUM_BUCKETS_LOG2 5U /* lg2 of initial number of buckets */ +#define HASH_BKT_CAPACITY_THRESH 10U /* expand when bucket count reaches */ + +/* calculate the element whose hash handle address is hhp */ +#define ELMT_FROM_HH(tbl,hhp) ((void*)(((char*)(hhp)) - ((tbl)->hho))) +/* calculate the hash handle from element address elp */ +#define HH_FROM_ELMT(tbl,elp) ((UT_hash_handle*)(void*)(((char*)(elp)) + ((tbl)->hho))) + +#define HASH_ROLLBACK_BKT(hh, head, itemptrhh) \ +do { \ + struct UT_hash_handle *_hd_hh_item = (itemptrhh); \ + unsigned _hd_bkt; \ + HASH_TO_BKT(_hd_hh_item->hashv, (head)->hh.tbl->num_buckets, _hd_bkt); \ + (head)->hh.tbl->buckets[_hd_bkt].count++; \ + _hd_hh_item->hh_next = NULL; \ + _hd_hh_item->hh_prev = NULL; \ +} while (0) + +#define HASH_VALUE(keyptr,keylen,hashv) \ +do { \ + HASH_FUNCTION(keyptr, keylen, hashv); \ +} while (0) + +#define HASH_FIND_BYHASHVALUE(hh,head,keyptr,keylen,hashval,out) \ +do { \ + (out) = NULL; \ + if (head) { \ + unsigned _hf_bkt; \ + HASH_TO_BKT(hashval, (head)->hh.tbl->num_buckets, _hf_bkt); \ + if (HASH_BLOOM_TEST((head)->hh.tbl, hashval) != 0) { \ + HASH_FIND_IN_BKT((head)->hh.tbl, hh, (head)->hh.tbl->buckets[ _hf_bkt ], keyptr, keylen, hashval, out); \ + } \ + } \ +} while (0) + +#define HASH_FIND(hh,head,keyptr,keylen,out) \ +do { \ + (out) = NULL; \ + if (head) { \ + unsigned _hf_hashv; \ + HASH_VALUE(keyptr, keylen, _hf_hashv); \ + HASH_FIND_BYHASHVALUE(hh, head, keyptr, keylen, _hf_hashv, out); \ + } \ +} while (0) + +#ifdef HASH_BLOOM +#define HASH_BLOOM_BITLEN (1UL << HASH_BLOOM) +#define HASH_BLOOM_BYTELEN (HASH_BLOOM_BITLEN/8UL) + (((HASH_BLOOM_BITLEN%8UL)!=0UL) ? 1UL : 0UL) +#define HASH_BLOOM_MAKE(tbl,oomed) \ +do { \ + (tbl)->bloom_nbits = HASH_BLOOM; \ + (tbl)->bloom_bv = (uint8_t*)uthash_malloc(HASH_BLOOM_BYTELEN); \ + if (!(tbl)->bloom_bv) { \ + HASH_RECORD_OOM(oomed); \ + } else { \ + uthash_bzero((tbl)->bloom_bv, HASH_BLOOM_BYTELEN); \ + (tbl)->bloom_sig = HASH_BLOOM_SIGNATURE; \ + } \ +} while (0) + +#define HASH_BLOOM_FREE(tbl) \ +do { \ + uthash_free((tbl)->bloom_bv, HASH_BLOOM_BYTELEN); \ +} while (0) + +#define HASH_BLOOM_BITSET(bv,idx) (bv[(idx)/8U] |= (1U << ((idx)%8U))) +#define HASH_BLOOM_BITTEST(bv,idx) (bv[(idx)/8U] & (1U << ((idx)%8U))) + +#define HASH_BLOOM_ADD(tbl,hashv) \ + HASH_BLOOM_BITSET((tbl)->bloom_bv, ((hashv) & (uint32_t)((1UL << (tbl)->bloom_nbits) - 1U))) + +#define HASH_BLOOM_TEST(tbl,hashv) \ + HASH_BLOOM_BITTEST((tbl)->bloom_bv, ((hashv) & (uint32_t)((1UL << (tbl)->bloom_nbits) - 1U))) + +#else +#define HASH_BLOOM_MAKE(tbl,oomed) +#define HASH_BLOOM_FREE(tbl) +#define HASH_BLOOM_ADD(tbl,hashv) +#define HASH_BLOOM_TEST(tbl,hashv) (1) +#define HASH_BLOOM_BYTELEN 0U +#endif + +#define HASH_MAKE_TABLE(hh,head,oomed) \ +do { \ + (head)->hh.tbl = (UT_hash_table*)uthash_malloc(sizeof(UT_hash_table)); \ + if (!(head)->hh.tbl) { \ + HASH_RECORD_OOM(oomed); \ + } else { \ + uthash_bzero((head)->hh.tbl, sizeof(UT_hash_table)); \ + (head)->hh.tbl->tail = &((head)->hh); \ + (head)->hh.tbl->num_buckets = HASH_INITIAL_NUM_BUCKETS; \ + (head)->hh.tbl->log2_num_buckets = HASH_INITIAL_NUM_BUCKETS_LOG2; \ + (head)->hh.tbl->hho = (char*)(&(head)->hh) - (char*)(head); \ + (head)->hh.tbl->buckets = (UT_hash_bucket*)uthash_malloc( \ + HASH_INITIAL_NUM_BUCKETS * sizeof(struct UT_hash_bucket)); \ + (head)->hh.tbl->signature = HASH_SIGNATURE; \ + if (!(head)->hh.tbl->buckets) { \ + HASH_RECORD_OOM(oomed); \ + uthash_free((head)->hh.tbl, sizeof(UT_hash_table)); \ + } else { \ + uthash_bzero((head)->hh.tbl->buckets, \ + HASH_INITIAL_NUM_BUCKETS * sizeof(struct UT_hash_bucket)); \ + HASH_BLOOM_MAKE((head)->hh.tbl, oomed); \ + IF_HASH_NONFATAL_OOM( \ + if (oomed) { \ + uthash_free((head)->hh.tbl->buckets, \ + HASH_INITIAL_NUM_BUCKETS*sizeof(struct UT_hash_bucket)); \ + uthash_free((head)->hh.tbl, sizeof(UT_hash_table)); \ + } \ + ) \ + } \ + } \ +} while (0) + +#define HASH_REPLACE_BYHASHVALUE_INORDER(hh,head,fieldname,keylen_in,hashval,add,replaced,cmpfcn) \ +do { \ + (replaced) = NULL; \ + HASH_FIND_BYHASHVALUE(hh, head, &((add)->fieldname), keylen_in, hashval, replaced); \ + if (replaced) { \ + HASH_DELETE(hh, head, replaced); \ + } \ + HASH_ADD_KEYPTR_BYHASHVALUE_INORDER(hh, head, &((add)->fieldname), keylen_in, hashval, add, cmpfcn); \ +} while (0) + +#define HASH_REPLACE_BYHASHVALUE(hh,head,fieldname,keylen_in,hashval,add,replaced) \ +do { \ + (replaced) = NULL; \ + HASH_FIND_BYHASHVALUE(hh, head, &((add)->fieldname), keylen_in, hashval, replaced); \ + if (replaced) { \ + HASH_DELETE(hh, head, replaced); \ + } \ + HASH_ADD_KEYPTR_BYHASHVALUE(hh, head, &((add)->fieldname), keylen_in, hashval, add); \ +} while (0) + +#define HASH_REPLACE(hh,head,fieldname,keylen_in,add,replaced) \ +do { \ + unsigned _hr_hashv; \ + HASH_VALUE(&((add)->fieldname), keylen_in, _hr_hashv); \ + HASH_REPLACE_BYHASHVALUE(hh, head, fieldname, keylen_in, _hr_hashv, add, replaced); \ +} while (0) + +#define HASH_REPLACE_INORDER(hh,head,fieldname,keylen_in,add,replaced,cmpfcn) \ +do { \ + unsigned _hr_hashv; \ + HASH_VALUE(&((add)->fieldname), keylen_in, _hr_hashv); \ + HASH_REPLACE_BYHASHVALUE_INORDER(hh, head, fieldname, keylen_in, _hr_hashv, add, replaced, cmpfcn); \ +} while (0) + +#define HASH_APPEND_LIST(hh, head, add) \ +do { \ + (add)->hh.next = NULL; \ + (add)->hh.prev = ELMT_FROM_HH((head)->hh.tbl, (head)->hh.tbl->tail); \ + (head)->hh.tbl->tail->next = (add); \ + (head)->hh.tbl->tail = &((add)->hh); \ +} while (0) + +#define HASH_AKBI_INNER_LOOP(hh,head,add,cmpfcn) \ +do { \ + do { \ + if (cmpfcn(DECLTYPE(head)(_hs_iter), add) > 0) { \ + break; \ + } \ + } while ((_hs_iter = HH_FROM_ELMT((head)->hh.tbl, _hs_iter)->next)); \ +} while (0) + +#ifdef NO_DECLTYPE +#undef HASH_AKBI_INNER_LOOP +#define HASH_AKBI_INNER_LOOP(hh,head,add,cmpfcn) \ +do { \ + char *_hs_saved_head = (char*)(head); \ + do { \ + DECLTYPE_ASSIGN(head, _hs_iter); \ + if (cmpfcn(head, add) > 0) { \ + DECLTYPE_ASSIGN(head, _hs_saved_head); \ + break; \ + } \ + DECLTYPE_ASSIGN(head, _hs_saved_head); \ + } while ((_hs_iter = HH_FROM_ELMT((head)->hh.tbl, _hs_iter)->next)); \ +} while (0) +#endif + +#if HASH_NONFATAL_OOM + +#define HASH_ADD_TO_TABLE(hh,head,keyptr,keylen_in,hashval,add,oomed) \ +do { \ + if (!(oomed)) { \ + unsigned _ha_bkt; \ + (head)->hh.tbl->num_items++; \ + HASH_TO_BKT(hashval, (head)->hh.tbl->num_buckets, _ha_bkt); \ + HASH_ADD_TO_BKT((head)->hh.tbl->buckets[_ha_bkt], hh, &(add)->hh, oomed); \ + if (oomed) { \ + HASH_ROLLBACK_BKT(hh, head, &(add)->hh); \ + HASH_DELETE_HH(hh, head, &(add)->hh); \ + (add)->hh.tbl = NULL; \ + uthash_nonfatal_oom(add); \ + } else { \ + HASH_BLOOM_ADD((head)->hh.tbl, hashval); \ + HASH_EMIT_KEY(hh, head, keyptr, keylen_in); \ + } \ + } else { \ + (add)->hh.tbl = NULL; \ + uthash_nonfatal_oom(add); \ + } \ +} while (0) + +#else + +#define HASH_ADD_TO_TABLE(hh,head,keyptr,keylen_in,hashval,add,oomed) \ +do { \ + unsigned _ha_bkt; \ + (head)->hh.tbl->num_items++; \ + HASH_TO_BKT(hashval, (head)->hh.tbl->num_buckets, _ha_bkt); \ + HASH_ADD_TO_BKT((head)->hh.tbl->buckets[_ha_bkt], hh, &(add)->hh, oomed); \ + HASH_BLOOM_ADD((head)->hh.tbl, hashval); \ + HASH_EMIT_KEY(hh, head, keyptr, keylen_in); \ +} while (0) + +#endif + + +#define HASH_ADD_KEYPTR_BYHASHVALUE_INORDER(hh,head,keyptr,keylen_in,hashval,add,cmpfcn) \ +do { \ + IF_HASH_NONFATAL_OOM( int _ha_oomed = 0; ) \ + (add)->hh.hashv = (hashval); \ + (add)->hh.key = (char*) (keyptr); \ + (add)->hh.keylen = (unsigned) (keylen_in); \ + if (!(head)) { \ + (add)->hh.next = NULL; \ + (add)->hh.prev = NULL; \ + HASH_MAKE_TABLE(hh, add, _ha_oomed); \ + IF_HASH_NONFATAL_OOM( if (!_ha_oomed) { ) \ + (head) = (add); \ + IF_HASH_NONFATAL_OOM( } ) \ + } else { \ + void *_hs_iter = (head); \ + (add)->hh.tbl = (head)->hh.tbl; \ + HASH_AKBI_INNER_LOOP(hh, head, add, cmpfcn); \ + if (_hs_iter) { \ + (add)->hh.next = _hs_iter; \ + if (((add)->hh.prev = HH_FROM_ELMT((head)->hh.tbl, _hs_iter)->prev)) { \ + HH_FROM_ELMT((head)->hh.tbl, (add)->hh.prev)->next = (add); \ + } else { \ + (head) = (add); \ + } \ + HH_FROM_ELMT((head)->hh.tbl, _hs_iter)->prev = (add); \ + } else { \ + HASH_APPEND_LIST(hh, head, add); \ + } \ + } \ + HASH_ADD_TO_TABLE(hh, head, keyptr, keylen_in, hashval, add, _ha_oomed); \ + HASH_FSCK(hh, head, "HASH_ADD_KEYPTR_BYHASHVALUE_INORDER"); \ +} while (0) + +#define HASH_ADD_KEYPTR_INORDER(hh,head,keyptr,keylen_in,add,cmpfcn) \ +do { \ + unsigned _hs_hashv; \ + HASH_VALUE(keyptr, keylen_in, _hs_hashv); \ + HASH_ADD_KEYPTR_BYHASHVALUE_INORDER(hh, head, keyptr, keylen_in, _hs_hashv, add, cmpfcn); \ +} while (0) + +#define HASH_ADD_BYHASHVALUE_INORDER(hh,head,fieldname,keylen_in,hashval,add,cmpfcn) \ + HASH_ADD_KEYPTR_BYHASHVALUE_INORDER(hh, head, &((add)->fieldname), keylen_in, hashval, add, cmpfcn) + +#define HASH_ADD_INORDER(hh,head,fieldname,keylen_in,add,cmpfcn) \ + HASH_ADD_KEYPTR_INORDER(hh, head, &((add)->fieldname), keylen_in, add, cmpfcn) + +#define HASH_ADD_KEYPTR_BYHASHVALUE(hh,head,keyptr,keylen_in,hashval,add) \ +do { \ + IF_HASH_NONFATAL_OOM( int _ha_oomed = 0; ) \ + (add)->hh.hashv = (hashval); \ + (add)->hh.key = (const void*) (keyptr); \ + (add)->hh.keylen = (unsigned) (keylen_in); \ + if (!(head)) { \ + (add)->hh.next = NULL; \ + (add)->hh.prev = NULL; \ + HASH_MAKE_TABLE(hh, add, _ha_oomed); \ + IF_HASH_NONFATAL_OOM( if (!_ha_oomed) { ) \ + (head) = (add); \ + IF_HASH_NONFATAL_OOM( } ) \ + } else { \ + (add)->hh.tbl = (head)->hh.tbl; \ + HASH_APPEND_LIST(hh, head, add); \ + } \ + HASH_ADD_TO_TABLE(hh, head, keyptr, keylen_in, hashval, add, _ha_oomed); \ + HASH_FSCK(hh, head, "HASH_ADD_KEYPTR_BYHASHVALUE"); \ +} while (0) + +#define HASH_ADD_KEYPTR(hh,head,keyptr,keylen_in,add) \ +do { \ + unsigned _ha_hashv; \ + HASH_VALUE(keyptr, keylen_in, _ha_hashv); \ + HASH_ADD_KEYPTR_BYHASHVALUE(hh, head, keyptr, keylen_in, _ha_hashv, add); \ +} while (0) + +#define HASH_ADD_BYHASHVALUE(hh,head,fieldname,keylen_in,hashval,add) \ + HASH_ADD_KEYPTR_BYHASHVALUE(hh, head, &((add)->fieldname), keylen_in, hashval, add) + +#define HASH_ADD(hh,head,fieldname,keylen_in,add) \ + HASH_ADD_KEYPTR(hh, head, &((add)->fieldname), keylen_in, add) + +#define HASH_TO_BKT(hashv,num_bkts,bkt) \ +do { \ + bkt = ((hashv) & ((num_bkts) - 1U)); \ +} while (0) + +/* delete "delptr" from the hash table. + * "the usual" patch-up process for the app-order doubly-linked-list. + * The use of _hd_hh_del below deserves special explanation. + * These used to be expressed using (delptr) but that led to a bug + * if someone used the same symbol for the head and deletee, like + * HASH_DELETE(hh,users,users); + * We want that to work, but by changing the head (users) below + * we were forfeiting our ability to further refer to the deletee (users) + * in the patch-up process. Solution: use scratch space to + * copy the deletee pointer, then the latter references are via that + * scratch pointer rather than through the repointed (users) symbol. + */ +#define HASH_DELETE(hh,head,delptr) \ + HASH_DELETE_HH(hh, head, &(delptr)->hh) + +#define HASH_DELETE_HH(hh,head,delptrhh) \ +do { \ + const struct UT_hash_handle *_hd_hh_del = (delptrhh); \ + if ((_hd_hh_del->prev == NULL) && (_hd_hh_del->next == NULL)) { \ + HASH_BLOOM_FREE((head)->hh.tbl); \ + uthash_free((head)->hh.tbl->buckets, \ + (head)->hh.tbl->num_buckets * sizeof(struct UT_hash_bucket)); \ + uthash_free((head)->hh.tbl, sizeof(UT_hash_table)); \ + (head) = NULL; \ + } else { \ + unsigned _hd_bkt; \ + if (_hd_hh_del == (head)->hh.tbl->tail) { \ + (head)->hh.tbl->tail = HH_FROM_ELMT((head)->hh.tbl, _hd_hh_del->prev); \ + } \ + if (_hd_hh_del->prev != NULL) { \ + HH_FROM_ELMT((head)->hh.tbl, _hd_hh_del->prev)->next = _hd_hh_del->next; \ + } else { \ + DECLTYPE_ASSIGN(head, _hd_hh_del->next); \ + } \ + if (_hd_hh_del->next != NULL) { \ + HH_FROM_ELMT((head)->hh.tbl, _hd_hh_del->next)->prev = _hd_hh_del->prev; \ + } \ + HASH_TO_BKT(_hd_hh_del->hashv, (head)->hh.tbl->num_buckets, _hd_bkt); \ + HASH_DEL_IN_BKT((head)->hh.tbl->buckets[_hd_bkt], _hd_hh_del); \ + (head)->hh.tbl->num_items--; \ + } \ + HASH_FSCK(hh, head, "HASH_DELETE_HH"); \ +} while (0) + +/* convenience forms of HASH_FIND/HASH_ADD/HASH_DEL */ +#define HASH_FIND_STR(head,findstr,out) \ +do { \ + unsigned _uthash_hfstr_keylen = (unsigned)uthash_strlen(findstr); \ + HASH_FIND(hh, head, findstr, _uthash_hfstr_keylen, out); \ +} while (0) +#define HASH_ADD_STR(head,strfield,add) \ +do { \ + unsigned _uthash_hastr_keylen = (unsigned)uthash_strlen((add)->strfield); \ + HASH_ADD(hh, head, strfield[0], _uthash_hastr_keylen, add); \ +} while (0) +#define HASH_REPLACE_STR(head,strfield,add,replaced) \ +do { \ + unsigned _uthash_hrstr_keylen = (unsigned)uthash_strlen((add)->strfield); \ + HASH_REPLACE(hh, head, strfield[0], _uthash_hrstr_keylen, add, replaced); \ +} while (0) +#define HASH_FIND_INT(head,findint,out) \ + HASH_FIND(hh,head,findint,sizeof(int),out) +#define HASH_ADD_INT(head,intfield,add) \ + HASH_ADD(hh,head,intfield,sizeof(int),add) +#define HASH_REPLACE_INT(head,intfield,add,replaced) \ + HASH_REPLACE(hh,head,intfield,sizeof(int),add,replaced) +#define HASH_FIND_PTR(head,findptr,out) \ + HASH_FIND(hh,head,findptr,sizeof(void *),out) +#define HASH_ADD_PTR(head,ptrfield,add) \ + HASH_ADD(hh,head,ptrfield,sizeof(void *),add) +#define HASH_REPLACE_PTR(head,ptrfield,add,replaced) \ + HASH_REPLACE(hh,head,ptrfield,sizeof(void *),add,replaced) +#define HASH_DEL(head,delptr) \ + HASH_DELETE(hh,head,delptr) + +/* HASH_FSCK checks hash integrity on every add/delete when HASH_DEBUG is defined. + * This is for uthash developer only; it compiles away if HASH_DEBUG isn't defined. + */ +#ifdef HASH_DEBUG +#include <stdio.h> /* fprintf, stderr */ +#define HASH_OOPS(...) do { fprintf(stderr, __VA_ARGS__); exit(-1); } while (0) +#define HASH_FSCK(hh,head,where) \ +do { \ + struct UT_hash_handle *_thh; \ + if (head) { \ + unsigned _bkt_i; \ + unsigned _count = 0; \ + char *_prev; \ + for (_bkt_i = 0; _bkt_i < (head)->hh.tbl->num_buckets; ++_bkt_i) { \ + unsigned _bkt_count = 0; \ + _thh = (head)->hh.tbl->buckets[_bkt_i].hh_head; \ + _prev = NULL; \ + while (_thh) { \ + if (_prev != (char*)(_thh->hh_prev)) { \ + HASH_OOPS("%s: invalid hh_prev %p, actual %p\n", \ + (where), (void*)_thh->hh_prev, (void*)_prev); \ + } \ + _bkt_count++; \ + _prev = (char*)(_thh); \ + _thh = _thh->hh_next; \ + } \ + _count += _bkt_count; \ + if ((head)->hh.tbl->buckets[_bkt_i].count != _bkt_count) { \ + HASH_OOPS("%s: invalid bucket count %u, actual %u\n", \ + (where), (head)->hh.tbl->buckets[_bkt_i].count, _bkt_count); \ + } \ + } \ + if (_count != (head)->hh.tbl->num_items) { \ + HASH_OOPS("%s: invalid hh item count %u, actual %u\n", \ + (where), (head)->hh.tbl->num_items, _count); \ + } \ + _count = 0; \ + _prev = NULL; \ + _thh = &(head)->hh; \ + while (_thh) { \ + _count++; \ + if (_prev != (char*)_thh->prev) { \ + HASH_OOPS("%s: invalid prev %p, actual %p\n", \ + (where), (void*)_thh->prev, (void*)_prev); \ + } \ + _prev = (char*)ELMT_FROM_HH((head)->hh.tbl, _thh); \ + _thh = (_thh->next ? HH_FROM_ELMT((head)->hh.tbl, _thh->next) : NULL); \ + } \ + if (_count != (head)->hh.tbl->num_items) { \ + HASH_OOPS("%s: invalid app item count %u, actual %u\n", \ + (where), (head)->hh.tbl->num_items, _count); \ + } \ + } \ +} while (0) +#else +#define HASH_FSCK(hh,head,where) +#endif + +/* When compiled with -DHASH_EMIT_KEYS, length-prefixed keys are emitted to + * the descriptor to which this macro is defined for tuning the hash function. + * The app can #include <unistd.h> to get the prototype for write(2). */ +#ifdef HASH_EMIT_KEYS +#define HASH_EMIT_KEY(hh,head,keyptr,fieldlen) \ +do { \ + unsigned _klen = fieldlen; \ + write(HASH_EMIT_KEYS, &_klen, sizeof(_klen)); \ + write(HASH_EMIT_KEYS, keyptr, (unsigned long)fieldlen); \ +} while (0) +#else +#define HASH_EMIT_KEY(hh,head,keyptr,fieldlen) +#endif + +/* The Bernstein hash function, used in Perl prior to v5.6. Note (x<<5+x)=x*33. */ +#define HASH_BER(key,keylen,hashv) \ +do { \ + unsigned _hb_keylen = (unsigned)keylen; \ + const unsigned char *_hb_key = (const unsigned char*)(key); \ + (hashv) = 0; \ + while (_hb_keylen-- != 0U) { \ + (hashv) = (((hashv) << 5) + (hashv)) + *_hb_key++; \ + } \ +} while (0) + + +/* SAX/FNV/OAT/JEN hash functions are macro variants of those listed at + * http://eternallyconfuzzled.com/tuts/algorithms/jsw_tut_hashing.aspx + * (archive link: https://archive.is/Ivcan ) + */ +#define HASH_SAX(key,keylen,hashv) \ +do { \ + unsigned _sx_i; \ + const unsigned char *_hs_key = (const unsigned char*)(key); \ + hashv = 0; \ + for (_sx_i=0; _sx_i < keylen; _sx_i++) { \ + hashv ^= (hashv << 5) + (hashv >> 2) + _hs_key[_sx_i]; \ + } \ +} while (0) +/* FNV-1a variation */ +#define HASH_FNV(key,keylen,hashv) \ +do { \ + unsigned _fn_i; \ + const unsigned char *_hf_key = (const unsigned char*)(key); \ + (hashv) = 2166136261U; \ + for (_fn_i=0; _fn_i < keylen; _fn_i++) { \ + hashv = hashv ^ _hf_key[_fn_i]; \ + hashv = hashv * 16777619U; \ + } \ +} while (0) + +#define HASH_OAT(key,keylen,hashv) \ +do { \ + unsigned _ho_i; \ + const unsigned char *_ho_key=(const unsigned char*)(key); \ + hashv = 0; \ + for(_ho_i=0; _ho_i < keylen; _ho_i++) { \ + hashv += _ho_key[_ho_i]; \ + hashv += (hashv << 10); \ + hashv ^= (hashv >> 6); \ + } \ + hashv += (hashv << 3); \ + hashv ^= (hashv >> 11); \ + hashv += (hashv << 15); \ +} while (0) + +#define HASH_JEN_MIX(a,b,c) \ +do { \ + a -= b; a -= c; a ^= ( c >> 13 ); \ + b -= c; b -= a; b ^= ( a << 8 ); \ + c -= a; c -= b; c ^= ( b >> 13 ); \ + a -= b; a -= c; a ^= ( c >> 12 ); \ + b -= c; b -= a; b ^= ( a << 16 ); \ + c -= a; c -= b; c ^= ( b >> 5 ); \ + a -= b; a -= c; a ^= ( c >> 3 ); \ + b -= c; b -= a; b ^= ( a << 10 ); \ + c -= a; c -= b; c ^= ( b >> 15 ); \ +} while (0) + +#define HASH_JEN(key,keylen,hashv) \ +do { \ + unsigned _hj_i,_hj_j,_hj_k; \ + unsigned const char *_hj_key=(unsigned const char*)(key); \ + hashv = 0xfeedbeefu; \ + _hj_i = _hj_j = 0x9e3779b9u; \ + _hj_k = (unsigned)(keylen); \ + while (_hj_k >= 12U) { \ + _hj_i += (_hj_key[0] + ( (unsigned)_hj_key[1] << 8 ) \ + + ( (unsigned)_hj_key[2] << 16 ) \ + + ( (unsigned)_hj_key[3] << 24 ) ); \ + _hj_j += (_hj_key[4] + ( (unsigned)_hj_key[5] << 8 ) \ + + ( (unsigned)_hj_key[6] << 16 ) \ + + ( (unsigned)_hj_key[7] << 24 ) ); \ + hashv += (_hj_key[8] + ( (unsigned)_hj_key[9] << 8 ) \ + + ( (unsigned)_hj_key[10] << 16 ) \ + + ( (unsigned)_hj_key[11] << 24 ) ); \ + \ + HASH_JEN_MIX(_hj_i, _hj_j, hashv); \ + \ + _hj_key += 12; \ + _hj_k -= 12U; \ + } \ + hashv += (unsigned)(keylen); \ + switch ( _hj_k ) { \ + case 11: hashv += ( (unsigned)_hj_key[10] << 24 ); /* FALLTHROUGH */ \ + case 10: hashv += ( (unsigned)_hj_key[9] << 16 ); /* FALLTHROUGH */ \ + case 9: hashv += ( (unsigned)_hj_key[8] << 8 ); /* FALLTHROUGH */ \ + case 8: _hj_j += ( (unsigned)_hj_key[7] << 24 ); /* FALLTHROUGH */ \ + case 7: _hj_j += ( (unsigned)_hj_key[6] << 16 ); /* FALLTHROUGH */ \ + case 6: _hj_j += ( (unsigned)_hj_key[5] << 8 ); /* FALLTHROUGH */ \ + case 5: _hj_j += _hj_key[4]; /* FALLTHROUGH */ \ + case 4: _hj_i += ( (unsigned)_hj_key[3] << 24 ); /* FALLTHROUGH */ \ + case 3: _hj_i += ( (unsigned)_hj_key[2] << 16 ); /* FALLTHROUGH */ \ + case 2: _hj_i += ( (unsigned)_hj_key[1] << 8 ); /* FALLTHROUGH */ \ + case 1: _hj_i += _hj_key[0]; /* FALLTHROUGH */ \ + default: ; \ + } \ + HASH_JEN_MIX(_hj_i, _hj_j, hashv); \ +} while (0) + +/* The Paul Hsieh hash function */ +#undef get16bits +#if (defined(__GNUC__) && defined(__i386__)) || defined(__WATCOMC__) \ + || defined(_MSC_VER) || defined (__BORLANDC__) || defined (__TURBOC__) +#define get16bits(d) (*((const uint16_t *) (d))) +#endif + +#if !defined (get16bits) +#define get16bits(d) ((((uint32_t)(((const uint8_t *)(d))[1])) << 8) \ + +(uint32_t)(((const uint8_t *)(d))[0]) ) +#endif +#define HASH_SFH(key,keylen,hashv) \ +do { \ + unsigned const char *_sfh_key=(unsigned const char*)(key); \ + uint32_t _sfh_tmp, _sfh_len = (uint32_t)keylen; \ + \ + unsigned _sfh_rem = _sfh_len & 3U; \ + _sfh_len >>= 2; \ + hashv = 0xcafebabeu; \ + \ + /* Main loop */ \ + for (;_sfh_len > 0U; _sfh_len--) { \ + hashv += get16bits (_sfh_key); \ + _sfh_tmp = ((uint32_t)(get16bits (_sfh_key+2)) << 11) ^ hashv; \ + hashv = (hashv << 16) ^ _sfh_tmp; \ + _sfh_key += 2U*sizeof (uint16_t); \ + hashv += hashv >> 11; \ + } \ + \ + /* Handle end cases */ \ + switch (_sfh_rem) { \ + case 3: hashv += get16bits (_sfh_key); \ + hashv ^= hashv << 16; \ + hashv ^= (uint32_t)(_sfh_key[sizeof (uint16_t)]) << 18; \ + hashv += hashv >> 11; \ + break; \ + case 2: hashv += get16bits (_sfh_key); \ + hashv ^= hashv << 11; \ + hashv += hashv >> 17; \ + break; \ + case 1: hashv += *_sfh_key; \ + hashv ^= hashv << 10; \ + hashv += hashv >> 1; \ + break; \ + default: ; \ + } \ + \ + /* Force "avalanching" of final 127 bits */ \ + hashv ^= hashv << 3; \ + hashv += hashv >> 5; \ + hashv ^= hashv << 4; \ + hashv += hashv >> 17; \ + hashv ^= hashv << 25; \ + hashv += hashv >> 6; \ +} while (0) + +/* iterate over items in a known bucket to find desired item */ +#define HASH_FIND_IN_BKT(tbl,hh,head,keyptr,keylen_in,hashval,out) \ +do { \ + if ((head).hh_head != NULL) { \ + DECLTYPE_ASSIGN(out, ELMT_FROM_HH(tbl, (head).hh_head)); \ + } else { \ + (out) = NULL; \ + } \ + while ((out) != NULL) { \ + if ((out)->hh.hashv == (hashval) && (out)->hh.keylen == (keylen_in)) { \ + if (HASH_KEYCMP((out)->hh.key, keyptr, keylen_in) == 0) { \ + break; \ + } \ + } \ + if ((out)->hh.hh_next != NULL) { \ + DECLTYPE_ASSIGN(out, ELMT_FROM_HH(tbl, (out)->hh.hh_next)); \ + } else { \ + (out) = NULL; \ + } \ + } \ +} while (0) + +/* add an item to a bucket */ +#define HASH_ADD_TO_BKT(head,hh,addhh,oomed) \ +do { \ + UT_hash_bucket *_ha_head = &(head); \ + _ha_head->count++; \ + (addhh)->hh_next = _ha_head->hh_head; \ + (addhh)->hh_prev = NULL; \ + if (_ha_head->hh_head != NULL) { \ + _ha_head->hh_head->hh_prev = (addhh); \ + } \ + _ha_head->hh_head = (addhh); \ + if ((_ha_head->count >= ((_ha_head->expand_mult + 1U) * HASH_BKT_CAPACITY_THRESH)) \ + && !(addhh)->tbl->noexpand) { \ + HASH_EXPAND_BUCKETS(addhh,(addhh)->tbl, oomed); \ + IF_HASH_NONFATAL_OOM( \ + if (oomed) { \ + HASH_DEL_IN_BKT(head,addhh); \ + } \ + ) \ + } \ +} while (0) + +/* remove an item from a given bucket */ +#define HASH_DEL_IN_BKT(head,delhh) \ +do { \ + UT_hash_bucket *_hd_head = &(head); \ + _hd_head->count--; \ + if (_hd_head->hh_head == (delhh)) { \ + _hd_head->hh_head = (delhh)->hh_next; \ + } \ + if ((delhh)->hh_prev) { \ + (delhh)->hh_prev->hh_next = (delhh)->hh_next; \ + } \ + if ((delhh)->hh_next) { \ + (delhh)->hh_next->hh_prev = (delhh)->hh_prev; \ + } \ +} while (0) + +/* Bucket expansion has the effect of doubling the number of buckets + * and redistributing the items into the new buckets. Ideally the + * items will distribute more or less evenly into the new buckets + * (the extent to which this is true is a measure of the quality of + * the hash function as it applies to the key domain). + * + * With the items distributed into more buckets, the chain length + * (item count) in each bucket is reduced. Thus by expanding buckets + * the hash keeps a bound on the chain length. This bounded chain + * length is the essence of how a hash provides constant time lookup. + * + * The calculation of tbl->ideal_chain_maxlen below deserves some + * explanation. First, keep in mind that we're calculating the ideal + * maximum chain length based on the *new* (doubled) bucket count. + * In fractions this is just n/b (n=number of items,b=new num buckets). + * Since the ideal chain length is an integer, we want to calculate + * ceil(n/b). We don't depend on floating point arithmetic in this + * hash, so to calculate ceil(n/b) with integers we could write + * + * ceil(n/b) = (n/b) + ((n%b)?1:0) + * + * and in fact a previous version of this hash did just that. + * But now we have improved things a bit by recognizing that b is + * always a power of two. We keep its base 2 log handy (call it lb), + * so now we can write this with a bit shift and logical AND: + * + * ceil(n/b) = (n>>lb) + ( (n & (b-1)) ? 1:0) + * + */ +#define HASH_EXPAND_BUCKETS(hh,tbl,oomed) \ +do { \ + unsigned _he_bkt; \ + unsigned _he_bkt_i; \ + struct UT_hash_handle *_he_thh, *_he_hh_nxt; \ + UT_hash_bucket *_he_new_buckets, *_he_newbkt; \ + _he_new_buckets = (UT_hash_bucket*)uthash_malloc( \ + sizeof(struct UT_hash_bucket) * (tbl)->num_buckets * 2U); \ + if (!_he_new_buckets) { \ + HASH_RECORD_OOM(oomed); \ + } else { \ + uthash_bzero(_he_new_buckets, \ + sizeof(struct UT_hash_bucket) * (tbl)->num_buckets * 2U); \ + (tbl)->ideal_chain_maxlen = \ + ((tbl)->num_items >> ((tbl)->log2_num_buckets+1U)) + \ + ((((tbl)->num_items & (((tbl)->num_buckets*2U)-1U)) != 0U) ? 1U : 0U); \ + (tbl)->nonideal_items = 0; \ + for (_he_bkt_i = 0; _he_bkt_i < (tbl)->num_buckets; _he_bkt_i++) { \ + _he_thh = (tbl)->buckets[ _he_bkt_i ].hh_head; \ + while (_he_thh != NULL) { \ + _he_hh_nxt = _he_thh->hh_next; \ + HASH_TO_BKT(_he_thh->hashv, (tbl)->num_buckets * 2U, _he_bkt); \ + _he_newbkt = &(_he_new_buckets[_he_bkt]); \ + if (++(_he_newbkt->count) > (tbl)->ideal_chain_maxlen) { \ + (tbl)->nonideal_items++; \ + if (_he_newbkt->count > _he_newbkt->expand_mult * (tbl)->ideal_chain_maxlen) { \ + _he_newbkt->expand_mult++; \ + } \ + } \ + _he_thh->hh_prev = NULL; \ + _he_thh->hh_next = _he_newbkt->hh_head; \ + if (_he_newbkt->hh_head != NULL) { \ + _he_newbkt->hh_head->hh_prev = _he_thh; \ + } \ + _he_newbkt->hh_head = _he_thh; \ + _he_thh = _he_hh_nxt; \ + } \ + } \ + uthash_free((tbl)->buckets, (tbl)->num_buckets * sizeof(struct UT_hash_bucket)); \ + (tbl)->num_buckets *= 2U; \ + (tbl)->log2_num_buckets++; \ + (tbl)->buckets = _he_new_buckets; \ + (tbl)->ineff_expands = ((tbl)->nonideal_items > ((tbl)->num_items >> 1)) ? \ + ((tbl)->ineff_expands+1U) : 0U; \ + if ((tbl)->ineff_expands > 1U) { \ + (tbl)->noexpand = 1; \ + uthash_noexpand_fyi(tbl); \ + } \ + uthash_expand_fyi(tbl); \ + } \ +} while (0) + + +/* This is an adaptation of Simon Tatham's O(n log(n)) mergesort */ +/* Note that HASH_SORT assumes the hash handle name to be hh. + * HASH_SRT was added to allow the hash handle name to be passed in. */ +#define HASH_SORT(head,cmpfcn) HASH_SRT(hh,head,cmpfcn) +#define HASH_SRT(hh,head,cmpfcn) \ +do { \ + unsigned _hs_i; \ + unsigned _hs_looping,_hs_nmerges,_hs_insize,_hs_psize,_hs_qsize; \ + struct UT_hash_handle *_hs_p, *_hs_q, *_hs_e, *_hs_list, *_hs_tail; \ + if (head != NULL) { \ + _hs_insize = 1; \ + _hs_looping = 1; \ + _hs_list = &((head)->hh); \ + while (_hs_looping != 0U) { \ + _hs_p = _hs_list; \ + _hs_list = NULL; \ + _hs_tail = NULL; \ + _hs_nmerges = 0; \ + while (_hs_p != NULL) { \ + _hs_nmerges++; \ + _hs_q = _hs_p; \ + _hs_psize = 0; \ + for (_hs_i = 0; _hs_i < _hs_insize; ++_hs_i) { \ + _hs_psize++; \ + _hs_q = ((_hs_q->next != NULL) ? \ + HH_FROM_ELMT((head)->hh.tbl, _hs_q->next) : NULL); \ + if (_hs_q == NULL) { \ + break; \ + } \ + } \ + _hs_qsize = _hs_insize; \ + while ((_hs_psize != 0U) || ((_hs_qsize != 0U) && (_hs_q != NULL))) { \ + if (_hs_psize == 0U) { \ + _hs_e = _hs_q; \ + _hs_q = ((_hs_q->next != NULL) ? \ + HH_FROM_ELMT((head)->hh.tbl, _hs_q->next) : NULL); \ + _hs_qsize--; \ + } else if ((_hs_qsize == 0U) || (_hs_q == NULL)) { \ + _hs_e = _hs_p; \ + if (_hs_p != NULL) { \ + _hs_p = ((_hs_p->next != NULL) ? \ + HH_FROM_ELMT((head)->hh.tbl, _hs_p->next) : NULL); \ + } \ + _hs_psize--; \ + } else if ((cmpfcn( \ + DECLTYPE(head)(ELMT_FROM_HH((head)->hh.tbl, _hs_p)), \ + DECLTYPE(head)(ELMT_FROM_HH((head)->hh.tbl, _hs_q)) \ + )) <= 0) { \ + _hs_e = _hs_p; \ + if (_hs_p != NULL) { \ + _hs_p = ((_hs_p->next != NULL) ? \ + HH_FROM_ELMT((head)->hh.tbl, _hs_p->next) : NULL); \ + } \ + _hs_psize--; \ + } else { \ + _hs_e = _hs_q; \ + _hs_q = ((_hs_q->next != NULL) ? \ + HH_FROM_ELMT((head)->hh.tbl, _hs_q->next) : NULL); \ + _hs_qsize--; \ + } \ + if ( _hs_tail != NULL ) { \ + _hs_tail->next = ((_hs_e != NULL) ? \ + ELMT_FROM_HH((head)->hh.tbl, _hs_e) : NULL); \ + } else { \ + _hs_list = _hs_e; \ + } \ + if (_hs_e != NULL) { \ + _hs_e->prev = ((_hs_tail != NULL) ? \ + ELMT_FROM_HH((head)->hh.tbl, _hs_tail) : NULL); \ + } \ + _hs_tail = _hs_e; \ + } \ + _hs_p = _hs_q; \ + } \ + if (_hs_tail != NULL) { \ + _hs_tail->next = NULL; \ + } \ + if (_hs_nmerges <= 1U) { \ + _hs_looping = 0; \ + (head)->hh.tbl->tail = _hs_tail; \ + DECLTYPE_ASSIGN(head, ELMT_FROM_HH((head)->hh.tbl, _hs_list)); \ + } \ + _hs_insize *= 2U; \ + } \ + HASH_FSCK(hh, head, "HASH_SRT"); \ + } \ +} while (0) + +/* This function selects items from one hash into another hash. + * The end result is that the selected items have dual presence + * in both hashes. There is no copy of the items made; rather + * they are added into the new hash through a secondary hash + * hash handle that must be present in the structure. */ +#define HASH_SELECT(hh_dst, dst, hh_src, src, cond) \ +do { \ + unsigned _src_bkt, _dst_bkt; \ + void *_last_elt = NULL, *_elt; \ + UT_hash_handle *_src_hh, *_dst_hh, *_last_elt_hh=NULL; \ + ptrdiff_t _dst_hho = ((char*)(&(dst)->hh_dst) - (char*)(dst)); \ + if ((src) != NULL) { \ + for (_src_bkt=0; _src_bkt < (src)->hh_src.tbl->num_buckets; _src_bkt++) { \ + for (_src_hh = (src)->hh_src.tbl->buckets[_src_bkt].hh_head; \ + _src_hh != NULL; \ + _src_hh = _src_hh->hh_next) { \ + _elt = ELMT_FROM_HH((src)->hh_src.tbl, _src_hh); \ + if (cond(_elt)) { \ + IF_HASH_NONFATAL_OOM( int _hs_oomed = 0; ) \ + _dst_hh = (UT_hash_handle*)(void*)(((char*)_elt) + _dst_hho); \ + _dst_hh->key = _src_hh->key; \ + _dst_hh->keylen = _src_hh->keylen; \ + _dst_hh->hashv = _src_hh->hashv; \ + _dst_hh->prev = _last_elt; \ + _dst_hh->next = NULL; \ + if (_last_elt_hh != NULL) { \ + _last_elt_hh->next = _elt; \ + } \ + if ((dst) == NULL) { \ + DECLTYPE_ASSIGN(dst, _elt); \ + HASH_MAKE_TABLE(hh_dst, dst, _hs_oomed); \ + IF_HASH_NONFATAL_OOM( \ + if (_hs_oomed) { \ + uthash_nonfatal_oom(_elt); \ + (dst) = NULL; \ + continue; \ + } \ + ) \ + } else { \ + _dst_hh->tbl = (dst)->hh_dst.tbl; \ + } \ + HASH_TO_BKT(_dst_hh->hashv, _dst_hh->tbl->num_buckets, _dst_bkt); \ + HASH_ADD_TO_BKT(_dst_hh->tbl->buckets[_dst_bkt], hh_dst, _dst_hh, _hs_oomed); \ + (dst)->hh_dst.tbl->num_items++; \ + IF_HASH_NONFATAL_OOM( \ + if (_hs_oomed) { \ + HASH_ROLLBACK_BKT(hh_dst, dst, _dst_hh); \ + HASH_DELETE_HH(hh_dst, dst, _dst_hh); \ + _dst_hh->tbl = NULL; \ + uthash_nonfatal_oom(_elt); \ + continue; \ + } \ + ) \ + HASH_BLOOM_ADD(_dst_hh->tbl, _dst_hh->hashv); \ + _last_elt = _elt; \ + _last_elt_hh = _dst_hh; \ + } \ + } \ + } \ + } \ + HASH_FSCK(hh_dst, dst, "HASH_SELECT"); \ +} while (0) + +#define HASH_CLEAR(hh,head) \ +do { \ + if ((head) != NULL) { \ + HASH_BLOOM_FREE((head)->hh.tbl); \ + uthash_free((head)->hh.tbl->buckets, \ + (head)->hh.tbl->num_buckets*sizeof(struct UT_hash_bucket)); \ + uthash_free((head)->hh.tbl, sizeof(UT_hash_table)); \ + (head) = NULL; \ + } \ +} while (0) + +#define HASH_OVERHEAD(hh,head) \ + (((head) != NULL) ? ( \ + (size_t)(((head)->hh.tbl->num_items * sizeof(UT_hash_handle)) + \ + ((head)->hh.tbl->num_buckets * sizeof(UT_hash_bucket)) + \ + sizeof(UT_hash_table) + \ + (HASH_BLOOM_BYTELEN))) : 0U) + +#ifdef NO_DECLTYPE +#define HASH_ITER(hh,head,el,tmp) \ +for(((el)=(head)), ((*(char**)(&(tmp)))=(char*)((head!=NULL)?(head)->hh.next:NULL)); \ + (el) != NULL; ((el)=(tmp)), ((*(char**)(&(tmp)))=(char*)((tmp!=NULL)?(tmp)->hh.next:NULL))) +#else +#define HASH_ITER(hh,head,el,tmp) \ +for(((el)=(head)), ((tmp)=DECLTYPE(el)((head!=NULL)?(head)->hh.next:NULL)); \ + (el) != NULL; ((el)=(tmp)), ((tmp)=DECLTYPE(el)((tmp!=NULL)?(tmp)->hh.next:NULL))) +#endif + +/* obtain a count of items in the hash */ +#define HASH_COUNT(head) HASH_CNT(hh,head) +#define HASH_CNT(hh,head) ((head != NULL)?((head)->hh.tbl->num_items):0U) + +typedef struct UT_hash_bucket { + struct UT_hash_handle *hh_head; + unsigned count; + + /* expand_mult is normally set to 0. In this situation, the max chain length + * threshold is enforced at its default value, HASH_BKT_CAPACITY_THRESH. (If + * the bucket's chain exceeds this length, bucket expansion is triggered). + * However, setting expand_mult to a non-zero value delays bucket expansion + * (that would be triggered by additions to this particular bucket) + * until its chain length reaches a *multiple* of HASH_BKT_CAPACITY_THRESH. + * (The multiplier is simply expand_mult+1). The whole idea of this + * multiplier is to reduce bucket expansions, since they are expensive, in + * situations where we know that a particular bucket tends to be overused. + * It is better to let its chain length grow to a longer yet-still-bounded + * value, than to do an O(n) bucket expansion too often. + */ + unsigned expand_mult; + +} UT_hash_bucket; + +/* random signature used only to find hash tables in external analysis */ +#define HASH_SIGNATURE 0xa0111fe1u +#define HASH_BLOOM_SIGNATURE 0xb12220f2u + +typedef struct UT_hash_table { + UT_hash_bucket *buckets; + unsigned num_buckets, log2_num_buckets; + unsigned num_items; + struct UT_hash_handle *tail; /* tail hh in app order, for fast append */ + ptrdiff_t hho; /* hash handle offset (byte pos of hash handle in element */ + + /* in an ideal situation (all buckets used equally), no bucket would have + * more than ceil(#items/#buckets) items. that's the ideal chain length. */ + unsigned ideal_chain_maxlen; + + /* nonideal_items is the number of items in the hash whose chain position + * exceeds the ideal chain maxlen. these items pay the penalty for an uneven + * hash distribution; reaching them in a chain traversal takes >ideal steps */ + unsigned nonideal_items; + + /* ineffective expands occur when a bucket doubling was performed, but + * afterward, more than half the items in the hash had nonideal chain + * positions. If this happens on two consecutive expansions we inhibit any + * further expansion, as it's not helping; this happens when the hash + * function isn't a good fit for the key domain. When expansion is inhibited + * the hash will still work, albeit no longer in constant time. */ + unsigned ineff_expands, noexpand; + + uint32_t signature; /* used only to find hash tables in external analysis */ +#ifdef HASH_BLOOM + uint32_t bloom_sig; /* used only to test bloom exists in external analysis */ + uint8_t *bloom_bv; + uint8_t bloom_nbits; +#endif + +} UT_hash_table; + +typedef struct UT_hash_handle { + struct UT_hash_table *tbl; + void *prev; /* prev element in app order */ + void *next; /* next element in app order */ + struct UT_hash_handle *hh_prev; /* previous hh in bucket order */ + struct UT_hash_handle *hh_next; /* next hh in bucket order */ + const void *key; /* ptr to enclosing struct's key */ + unsigned keylen; /* enclosing struct's key len */ + unsigned hashv; /* result of hash-fcn(key) */ +} UT_hash_handle; + +#endif /* UTHASH_H */