From dbe5f2237cada6a261e297b60857dc305860a88d Mon Sep 17 00:00:00 2001
From: AnnieRuru <jeankof@ymail.com>
Date: Tue, 17 Nov 2015 00:10:26 +0800
Subject: Add *getunittype and update *getmapxy

- unlike rathena, hercules' getunittype use the value from getmapxy
- getmapxy now can search object position using GID

Closes #871 as merged

Signed-off-by: Haru <haru@dotalux.com>
---
 src/map/script.c | 117 ++++++++++++++++++++++++++++++++++++++++++-------------
 1 file changed, 91 insertions(+), 26 deletions(-)

(limited to 'src/map/script.c')

diff --git a/src/map/script.c b/src/map/script.c
index 401b0308a..00985c2a5 100644
--- a/src/map/script.c
+++ b/src/map/script.c
@@ -14132,58 +14132,95 @@ BUILDIN(getmapxy)
 
 	switch (type) {
 		case 0: //Get Character Position
-			if( script_hasdata(st,6) )
-				sd=map->nick2sd(script_getstr(st,6));
-			else
-				sd=script->rid2sd(st);
+			if (script_hasdata(st,6)) {
+				if (script_isstringtype(st,6))
+					sd = map->nick2sd(script_getstr(st,6));
+				else
+					sd = map->id2sd(script_getnum(st,6));
+			} else {
+				sd = script->rid2sd(st);
+			}
 
 			if (sd)
 				bl = &sd->bl;
 			break;
 		case 1: //Get NPC Position
-			if( script_hasdata(st,6) )
-			{
+			if (script_hasdata(st,6)) {
 				struct npc_data *nd;
-				nd=npc->name2id(script_getstr(st,6));
+				if (script_isstringtype(st,6))
+					nd = npc->name2id(script_getstr(st,6));
+				else
+					nd = map->id2nd(script_getnum(st,6));
 				if (nd)
 					bl = &nd->bl;
-			} else //In case the origin is not an npc?
-				bl=map->id2bl(st->oid);
+			} else {
+				//In case the origin is not an npc?
+				bl = map->id2bl(st->oid);
+			}
 			break;
 		case 2: //Get Pet Position
-			if(script_hasdata(st,6))
-				sd=map->nick2sd(script_getstr(st,6));
-			else
-				sd=script->rid2sd(st);
+			if (script_hasdata(st,6)) {
+				if (script_isstringtype(st,6))
+					sd = map->nick2sd(script_getstr(st,6));
+				else {
+					bl = map->id2bl(script_getnum(st,6));
+					break;
+				}
+			} else {
+				sd = script->rid2sd(st);
+			}
 
 			if (sd && sd->pd)
 				bl = &sd->pd->bl;
 			break;
 		case 3: //Get Mob Position
-			break; //Not supported?
+			if (script_hasdata(st,6)) {
+				if (script_isstringtype(st,6))
+					break;
+				bl = map->id2bl(script_getnum(st,6));
+			}
+			break;
 		case 4: //Get Homun Position
-			if(script_hasdata(st,6))
-				sd=map->nick2sd(script_getstr(st,6));
-			else
-				sd=script->rid2sd(st);
+			if (script_hasdata(st,6)) {
+				if (script_isstringtype(st,6)) {
+					sd = map->nick2sd(script_getstr(st,6));
+				} else {
+					bl = map->id2bl(script_getnum(st,6));
+					break;
+				}
+			} else {
+				sd = script->rid2sd(st);
+			}
 
 			if (sd && sd->hd)
 				bl = &sd->hd->bl;
 			break;
 		case 5: //Get Mercenary Position
-			if(script_hasdata(st,6))
-				sd=map->nick2sd(script_getstr(st,6));
-			else
-				sd=script->rid2sd(st);
+			if (script_hasdata(st,6)) {
+				if (script_isstringtype(st,6)) {
+					sd = map->nick2sd(script_getstr(st,6));
+				} else {
+					bl = map->id2bl(script_getnum(st,6));
+					break;
+				}
+			} else {
+				sd = script->rid2sd(st);
+			}
 
 			if (sd && sd->md)
 				bl = &sd->md->bl;
 			break;
 		case 6: //Get Elemental Position
-			if(script_hasdata(st,6))
-				sd=map->nick2sd(script_getstr(st,6));
-			else
-				sd=script->rid2sd(st);
+			if (script_hasdata(st,6)) {
+				if (script_isstringtype(st,6)) {
+					sd = map->nick2sd(script_getstr(st,6));
+				} else {
+					bl = map->id2bl(script_getnum(st,6));
+					break;
+				}
+			} else {
+				sd = script->rid2sd(st);
+			}
 
 			if (sd && sd->ed)
 				bl = &sd->ed->bl;
@@ -16363,6 +16400,33 @@ BUILDIN(pcstopfollow)
 // [zBuffer] List of mob control commands --->
 //## TODO always return if the request/whatever was successfull [FlavioJS]
 
+BUILDIN(getunittype) {
+	struct block_list* bl;
+	int value;
+
+	bl = map->id2bl(script_getnum(st,2));
+
+	if (!bl) {
+		ShowWarning("buildin_getunittype: Error in finding object GID %d!\n", script_getnum(st,2));
+		script_pushint(st,-1);
+		return false;
+	}
+
+	switch (bl->type) {
+		case BL_PC:   value = 0; break;
+		case BL_NPC:  value = 1; break;
+		case BL_PET:  value = 2; break;
+		case BL_MOB:  value = 3; break;
+		case BL_HOM:  value = 4; break;
+		case BL_MER:  value = 5; break;
+		case BL_ELEM: value = 6; break;
+		default:      value = -1; break;
+	}
+
+	script_pushint(st, value);
+	return true;
+}
+
 /// Makes the unit walk to target position or target id
 /// Returns if it was successfull
 ///
@@ -20382,6 +20446,7 @@ void script_parse_builtin(void) {
 		BUILDIN_DEF(pcblockmove,"ii"),
 		// <--- [zBuffer] List of player cont commands
 		// [zBuffer] List of mob control commands --->
+		BUILDIN_DEF(getunittype,"i"),
 		BUILDIN_DEF(unitwalk,"ii?"),
 		BUILDIN_DEF(unitkill,"i"),
 		BUILDIN_DEF(unitwarp,"isii"),
-- 
cgit v1.2.3-70-g09d2