dvander@2726: /**
dvander@2726:  * vim: set ts=4 :
dvander@2726:  * =============================================================================
dvander@2726:  * SourceMod
dvander@2726:  * Copyright (C) 2004-2008 AlliedModders LLC.  All rights reserved.
dvander@2726:  * =============================================================================
dvander@2726:  *
dvander@2726:  * This program is free software; you can redistribute it and/or modify it under
dvander@2726:  * the terms of the GNU General Public License, version 3.0, as published by the
dvander@2726:  * Free Software Foundation.
dvander@2726:  * 
dvander@2726:  * This program is distributed in the hope that it will be useful, but WITHOUT
dvander@2726:  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
dvander@2726:  * FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
dvander@2726:  * details.
dvander@2726:  *
dvander@2726:  * You should have received a copy of the GNU General Public License along with
dvander@2726:  * this program.  If not, see <http://www.gnu.org/licenses/>.
dvander@2726:  *
dvander@2726:  * As a special exception, AlliedModders LLC gives you permission to link the
dvander@2726:  * code of this program (as well as its derivative works) to "Half-Life 2," the
dvander@2726:  * "Source Engine," the "SourcePawn JIT," and any Game MODs that run on software
dvander@2726:  * by the Valve Corporation.  You must obey the GNU General Public License in
dvander@2726:  * all respects for all other code used.  Additionally, AlliedModders LLC grants
dvander@2726:  * this exception to all derivative works.  AlliedModders LLC defines further
dvander@2726:  * exceptions, found in LICENSE.txt (as of this writing, version JULY-31-2007),
dvander@2726:  * or <http://www.sourcemod.net/license.php>.
dvander@2726:  *
dvander@2726:  * Version: $Id$
dvander@2726:  */
dvander@2726: 
dvander@2726: #include <stdlib.h>
dvander@2726: #include <string.h>
dvander@2726: #include <time.h>
dvander@2726: #include "common_logic.h"
dvander@2726: #include "CellArray.h"
dvander@2726: 
dvander@2726: /***********************************
dvander@2726:  *   About the double array hack   *
dvander@2726:      ***************************   
dvander@2726: 
dvander@2726:  Double arrays in Pawn are vectors offset by the current offset.  For example:
dvander@2726: 
dvander@2726:  new array[2][2]
dvander@2726: 
dvander@2726:    In this array, index 0 contains the offset from the current offset which
dvander@2726:  results in the final vector [2] (at [0][2]).  Meaning, to dereference [1][2], 
dvander@2726:  it is equivalent to:
dvander@2726: 
dvander@2726:  address = &array[1] + array[1] + 2 * sizeof(cell)
dvander@2726: 
dvander@2726:    The fact that each offset is from the _current_ position rather than the _base_
dvander@2726:  position is very important.  It means that if you to try to swap vector positions,
dvander@2726:  the offsets will no longer match, because their current position has changed.  A 
dvander@2726:  simple and ingenious way around this is to back up the positions in a separate array,
dvander@2726:  then to overwrite each position in the old array with absolute indices.  Pseudo C++ code:
dvander@2726: 
dvander@2726:  cell *array; //assumed to be set to the 2+D array
dvander@2726:  cell *old_offsets = new cell[2];
dvander@2726:  for (int i=0; i<2; i++)
dvander@2726:  {
dvander@2726:     old_offsets = array[i];
dvander@2726: 	array[i] = i;
dvander@2726:  }
dvander@2726: 
dvander@2726:    Now, you can swap the array indices with no problem, and do a reverse-lookup to find the original addresses.
dvander@2726:  After sorting/modification is done, you must relocate the new indices.  For example, if the two vectors in our
dvander@2726:  demo array were swapped, array[0] would be 1 and array[1] would be 0.  This is invalid to the virtual machine.
dvander@2726:  Luckily, this is also simple -- all the information is there.
dvander@2726: 
dvander@2726:  for (int i=0; i<2; i++)
dvander@2726:  {
dvander@2726: 	//get the # of the vector we want to relocate in
dvander@2726: 	cell vector_index = array[i];			
dvander@2726: 	//get the real address of this vector
dvander@2726: 	char *real_address = (char *)array + (vector_index * sizeof(cell)) + old_offsets[vector_index];
dvander@2726: 	//calc and store the new distance offset
dvander@2726: 	array[i] = real_address - ( (char *)array + (vector_index + sizeof(cell)) )
dvander@2726:  }
dvander@2726: 
dvander@2726:  Note that the inner expression can be heavily reduced; it is expanded for readability.
dvander@2726:  **********************************/
dvander@2726: 
dvander@2726: enum SortOrder
dvander@2726: {
dvander@2726: 	Sort_Ascending = 0,
dvander@2726: 	Sort_Descending = 1,
dvander@2726: 	Sort_Random = 2,
dvander@2726: };
dvander@2726: 
dvander@2726: void sort_random(cell_t *array, cell_t size)
dvander@2726: {
dvander@2726: 	srand((unsigned int)time(NULL));
dvander@2726: 
dvander@2726: 	for (int i = size-1; i > 0; i--)
dvander@2726: 	{
januk@3752:         int n = rand() % (i + 1);
dvander@2726: 
dvander@2726: 		if (array[i] != array[n]) 
dvander@2726: 		{
dvander@2726: 			array[i] ^= array[n];
dvander@2726: 			array[n] ^= array[i];
dvander@2726: 			array[i] ^= array[n];
dvander@2726: 		}
dvander@2726: 	}
dvander@2726: }
dvander@2726: 
dvander@2726: int sort_ints_asc(const void *int1, const void *int2)
dvander@2726: {
dvander@2726: 	return (*(int *)int1) - (*(int *)int2);
dvander@2726: }
dvander@2726: 
dvander@2726: int sort_ints_desc(const void *int1, const void *int2)
dvander@2726: {
dvander@2726: 	return (*(int *)int2) - (*(int *)int1);
dvander@2726: }
dvander@2726: 
dvander@2726: static cell_t sm_SortIntegers(IPluginContext *pContext, const cell_t *params)
dvander@2726: {
dvander@2726: 	cell_t *array;
dvander@2726: 	cell_t array_size = params[2];
dvander@2726: 	cell_t type = params[3];
dvander@2726: 
dvander@2726: 	pContext->LocalToPhysAddr(params[1], &array);
dvander@2726: 
dvander@2726: 	if (type == Sort_Ascending)
dvander@2726: 	{
dvander@2726: 		qsort(array, array_size, sizeof(cell_t), sort_ints_asc);
dvander@2726: 	} 
dvander@2726: 	else if (type == Sort_Descending)
dvander@2726: 	{
dvander@2726: 		qsort(array, array_size, sizeof(cell_t), sort_ints_desc);
dvander@2726: 	}
dvander@2726: 	else
dvander@2726: 	{
dvander@2726: 		sort_random(array, array_size);
dvander@2726: 	}
dvander@2726: 
dvander@2726: 	return 1;
dvander@2726: }
dvander@2726: 
dvander@2726: int sort_floats_asc(const void *float1, const void *float2)
dvander@2726: {
dvander@2726: 	float r1 = *(float *)float1;
dvander@2726: 	float r2 = *(float *)float2;
dvander@2726: 	
dvander@2726: 	if (r1 < r2)
dvander@2726: 	{
dvander@2726: 		return -1;
dvander@2726: 	} else if (r2 < r1) {
dvander@2726: 		return 1;
dvander@2726: 	} else {
dvander@2726: 		return 0;
dvander@2726: 	}
dvander@2726: }
dvander@2726: 
dvander@2726: int sort_floats_desc(const void *float1, const void *float2)
dvander@2726: {
dvander@2726: 	float r1 = *(float *)float1;
dvander@2726: 	float r2 = *(float *)float2;
dvander@2726: 
dvander@2726: 	if (r1 < r2)
dvander@2726: 	{
dvander@2726: 		return 1;
dvander@2726: 	} else if (r2 < r1) {
dvander@2726: 		return -1;
dvander@2726: 	} else {
dvander@2726: 		return 0;
dvander@2726: 	}
dvander@2726: }
dvander@2726: 
dvander@2726: static cell_t sm_SortFloats(IPluginContext *pContext, const cell_t *params)
dvander@2726: {
dvander@2726: 	cell_t *array;
dvander@2726: 	cell_t array_size = params[2];
dvander@2726: 	cell_t type = params[3];
dvander@2726: 
dvander@2726: 	pContext->LocalToPhysAddr(params[1], &array);
dvander@2726: 
dvander@2726: 	if (type == Sort_Ascending)
dvander@2726: 	{
dvander@2726: 		qsort(array, array_size, sizeof(cell_t), sort_floats_asc);
dvander@2726: 	}
dvander@2726: 	else if (type == Sort_Descending)
dvander@2726: 	{
dvander@2726: 		qsort(array, array_size, sizeof(cell_t), sort_floats_desc);
dvander@2726: 	}
dvander@2726: 	else
dvander@2726: 	{
dvander@2726: 		sort_random(array, array_size);
dvander@2726: 	}
dvander@2726: 
dvander@2726: 	return 1;
dvander@2726: }
dvander@2726: 
dvander@2726: static cell_t *g_CurStringArray = NULL;
dvander@2726: static cell_t *g_CurRebaseMap = NULL;
dvander@2726: 
dvander@2726: int sort_strings_asc(const void *blk1, const void *blk2)
dvander@2726: {
dvander@2726: 	cell_t reloc1 = *(cell_t *)blk1;
dvander@2726: 	cell_t reloc2 = *(cell_t *)blk2;
dvander@2726: 	
dvander@2726: 	char *str1 = ((char *)(&g_CurStringArray[reloc1]) + g_CurRebaseMap[reloc1]);
dvander@2726: 	char *str2 = ((char *)(&g_CurStringArray[reloc2]) + g_CurRebaseMap[reloc2]);
dvander@2726: 
dvander@2726: 	return strcmp(str1, str2);
dvander@2726: }
dvander@2726: 
dvander@2726: int sort_strings_desc(const void *blk1, const void *blk2)
dvander@2726: {
dvander@2726: 	cell_t reloc1 = *(cell_t *)blk1;
dvander@2726: 	cell_t reloc2 = *(cell_t *)blk2;
dvander@2726: 
dvander@2726: 	char *str1 = ((char *)(&g_CurStringArray[reloc1]) + g_CurRebaseMap[reloc1]);
dvander@2726: 	char *str2 = ((char *)(&g_CurStringArray[reloc2]) + g_CurRebaseMap[reloc2]);
dvander@2726: 
dvander@2726: 	return strcmp(str2, str1);
dvander@2726: }
dvander@2726: 
dvander@2726: static cell_t sm_SortStrings(IPluginContext *pContext, const cell_t *params)
dvander@2726: {
dvander@2726: 	cell_t *array;
dvander@2726: 	cell_t array_size = params[2];
dvander@2726: 	cell_t type = params[3];
dvander@2726: 
dvander@2726: 	pContext->LocalToPhysAddr(params[1], &array);
dvander@2726: 
dvander@2726: 	/** HACKHACK - back up the old indices, replace the indices with something easier */
dvander@2726: 	cell_t amx_addr, *phys_addr;
dvander@2726: 	int err;
dvander@2726: 	if ((err=pContext->HeapAlloc(array_size, &amx_addr, &phys_addr)) != SP_ERROR_NONE)
dvander@2726: 	{
dvander@2726: 		pContext->ThrowNativeErrorEx(err, "Ran out of memory to sort");
dvander@2726: 	}
dvander@2726: 
dvander@2726: 	g_CurStringArray = array;
dvander@2726: 	g_CurRebaseMap = phys_addr;
dvander@2726: 
dvander@2726: 	for (int i=0; i<array_size; i++)
dvander@2726: 	{
dvander@2726: 		phys_addr[i] = array[i];
dvander@2726: 		array[i] = i;
dvander@2726: 	}
dvander@2726: 
dvander@2726: 	if (type == Sort_Ascending)
dvander@2726: 	{
dvander@2726: 		qsort(array, array_size, sizeof(cell_t), sort_strings_asc);
dvander@2726: 	} 
dvander@2726: 	else if (type == Sort_Descending)
dvander@2726: 	{
dvander@2726: 		qsort(array, array_size, sizeof(cell_t), sort_strings_desc);
dvander@2726: 	}
dvander@2726: 	else
dvander@2726: 	{
dvander@2726: 		sort_random(array, array_size);
dvander@2726: 	}
dvander@2726: 
dvander@2726: 	/* END HACKHACK - restore what we damaged so Pawn doesn't throw up.
dvander@2726: 	 * We'll browse through each index of the array and patch up the distance.
dvander@2726: 	 */
dvander@2726: 	for (int i=0; i<array_size; i++)
dvander@2726: 	{
dvander@2726: 		/* Compute the final address of the old array and subtract the new location.
dvander@2726: 		 * This is the fixed up distance.
dvander@2726: 		 */
dvander@2726: 		array[i] = ((char *)&array[array[i]] + phys_addr[array[i]]) - (char *)&array[i];
dvander@2726: 	}
dvander@2726: 
dvander@2726: 	pContext->HeapPop(amx_addr);
dvander@2726: 
dvander@2726: 	g_CurStringArray = NULL;
dvander@2726: 	g_CurRebaseMap = NULL;
dvander@2726: 
dvander@2726: 	return 1;
dvander@2726: }
dvander@2726: 
dvander@2726: struct sort_info
dvander@2726: {
dvander@2726: 	IPluginFunction *pFunc;
dvander@2726: 	Handle_t hndl;
dvander@2726: 	cell_t array_addr;
dvander@2726: 	cell_t *array_base;
dvander@2726: 	cell_t *array_remap;
dvander@2726: };
dvander@2726: 
dvander@2726: sort_info g_SortInfo;
dvander@2726: 
dvander@2726: int sort1d_amx_custom(const void *elem1, const void *elem2)
dvander@2726: {
dvander@2726: 	cell_t c1 = *(cell_t *)elem1;
dvander@2726: 	cell_t c2 = *(cell_t *)elem2;
dvander@2726: 
dvander@2726: 	cell_t result = 0;
dvander@2726: 	IPluginFunction *pf = g_SortInfo.pFunc;
dvander@2726: 	pf->PushCell(c1);
dvander@2726: 	pf->PushCell(c2);
dvander@2726: 	pf->PushCell(g_SortInfo.array_addr);
dvander@2726: 	pf->PushCell(g_SortInfo.hndl);
dvander@2726: 	pf->Execute(&result);
dvander@2726: 
dvander@2726: 	return result;
dvander@2726: }
dvander@2726: 
dvander@2726: static cell_t sm_SortCustom1D(IPluginContext *pContext, const cell_t *params)
dvander@2726: {
dvander@2726: 	cell_t *array;
dvander@2726: 	cell_t array_size = params[2];
dvander@2726: 
dvander@2726: 	IPluginFunction *pFunction = pContext->GetFunctionById(params[3]);
dvander@2726: 	if (!pFunction)
dvander@2726: 	{
dvander@2726: 		return pContext->ThrowNativeError("Function %x is not a valid function", params[3]);
dvander@2726: 	}
dvander@2726: 
dvander@2726: 	pContext->LocalToPhysAddr(params[1], &array);
dvander@2726: 
dvander@2726: 	sort_info oldinfo = g_SortInfo;
dvander@2726: 
dvander@2726: 	
dvander@2726: 	g_SortInfo.hndl = params[4];
dvander@2726: 	g_SortInfo.array_addr = params[1];
dvander@2726: 	g_SortInfo.array_remap = NULL;
dvander@2726: 	g_SortInfo.array_base = NULL;
dvander@2726: 	g_SortInfo.pFunc = pFunction;
dvander@2726: 
dvander@2726: 	qsort(array, array_size, sizeof(cell_t), sort1d_amx_custom);
dvander@2726: 
dvander@2726: 	g_SortInfo = oldinfo;
dvander@2726: 
dvander@2726: 	return 1;
dvander@2726: }
dvander@2726: 
dvander@2726: int sort2d_amx_custom(const void *elem1, const void *elem2)
dvander@2726: {
dvander@2726: 	cell_t c1 = *(cell_t *)elem1;
dvander@2726: 	cell_t c2 = *(cell_t *)elem2;
dvander@2726: 
dvander@2726: 	cell_t c1_addr = g_SortInfo.array_addr + (c1 * sizeof(cell_t)) + g_SortInfo.array_remap[c1];
dvander@2726: 	cell_t c2_addr = g_SortInfo.array_addr + (c2 * sizeof(cell_t)) + g_SortInfo.array_remap[c2];
dvander@2726: 
dvander@2726: 	IPluginContext *pContext = g_SortInfo.pFunc->GetParentContext();
dvander@2726: 	cell_t *c1_r, *c2_r;
dvander@2726: 	pContext->LocalToPhysAddr(c1_addr, &c1_r);
dvander@2726: 	pContext->LocalToPhysAddr(c2_addr, &c2_r);
dvander@2726: 
dvander@2726: 	cell_t result = 0;
dvander@2726: 	g_SortInfo.pFunc->PushCell(c1_addr);
dvander@2726: 	g_SortInfo.pFunc->PushCell(c2_addr);
dvander@2726: 	g_SortInfo.pFunc->PushCell(g_SortInfo.array_addr);
dvander@2726: 	g_SortInfo.pFunc->PushCell(g_SortInfo.hndl);
dvander@2726: 	g_SortInfo.pFunc->Execute(&result);
dvander@2726: 
dvander@2726: 	return result;
dvander@2726: }
dvander@2726: 
dvander@2726: static cell_t sm_SortCustom2D(IPluginContext *pContext, const cell_t *params)
dvander@2726: {
dvander@2726: 	cell_t *array;
dvander@2726: 	cell_t array_size = params[2];
dvander@2726: 	IPluginFunction *pFunction;
dvander@2726: 
dvander@2726: 	pContext->LocalToPhysAddr(params[1], &array);
dvander@2726: 
dvander@2726: 	if ((pFunction=pContext->GetFunctionById(params[3])) == NULL)
dvander@2726: 	{
dvander@2726: 		return pContext->ThrowNativeError("Function %x is not a valid function", params[3]);
dvander@2726: 	}
dvander@2726: 
dvander@2726: 	/** back up the old indices, replace the indices with something easier */
dvander@2726: 	cell_t amx_addr, *phys_addr;
dvander@2726: 	int err;
dvander@2726: 	if ((err=pContext->HeapAlloc(array_size, &amx_addr, &phys_addr)) != SP_ERROR_NONE)
dvander@2726: 	{
dvander@2726: 		pContext->ThrowNativeErrorEx(err, "Ran out of memory to sort");
dvander@2726: 		return 0;
dvander@2726: 	}
dvander@2726: 
dvander@2726: 	sort_info oldinfo = g_SortInfo;
dvander@2726: 
dvander@2726: 	g_SortInfo.pFunc = pFunction;
dvander@2726: 	g_SortInfo.hndl = params[4];
dvander@2726: 	g_SortInfo.array_addr = params[1];
dvander@2726: 	
dvander@2726: 	/** Same process as in strings, back up the old indices for later fixup */
dvander@2726: 	g_SortInfo.array_base = array;
dvander@2726: 	g_SortInfo.array_remap = phys_addr;
dvander@2726: 	
dvander@2726: 	for (int i=0; i<array_size; i++)
dvander@2726: 	{
dvander@2726: 		phys_addr[i] = array[i];
dvander@2726: 		array[i] = i;
dvander@2726: 	}
dvander@2726: 
dvander@2726: 	qsort(array, array_size, sizeof(cell_t), sort2d_amx_custom);
dvander@2726: 
dvander@2726: 	/** Fixup process! */
dvander@2726: 	for (int i=0; i<array_size; i++)
dvander@2726: 	{
dvander@2726: 		/* Compute the final address of the old array and subtract the new location.
dvander@2726: 		 * This is the fixed up distance.
dvander@2726: 		 */
dvander@2726: 		array[i] = ((char *)&array[array[i]] + phys_addr[array[i]]) - (char *)&array[i];
dvander@2726: 	}
dvander@2726: 
dvander@2726: 	pContext->HeapPop(amx_addr);
dvander@2726: 
dvander@2726: 	g_SortInfo = oldinfo;
dvander@2726: 	
dvander@2726: 	return 1;
dvander@2726: }
dvander@2726: 
dvander@2726: enum SortType
dvander@2726: {
dvander@2726: 	Sort_Integer = 0,
dvander@2726: 	Sort_Float,
dvander@2726: 	Sort_String,
dvander@2726: };
dvander@2726: 
dvander@2726: int sort_adtarray_strings_asc(const void *str1, const void *str2)
dvander@2726: {
dvander@2726: 	return strcmp((char *) str1, (char *) str2);
dvander@2726: }
dvander@2726: 
dvander@2726: int sort_adtarray_strings_desc(const void *str1, const void *str2)
dvander@2726: {
dvander@2726: 	return strcmp((char *) str2, (char *) str1);
dvander@2726: }
dvander@2726: 
dvander@2726: void sort_adt_random(CellArray *cArray)
dvander@2726: {
dvander@2726: 	size_t arraysize = cArray->size();
dvander@2726: 
dvander@2726: 	srand((unsigned int)time(NULL));
dvander@2726: 
dvander@2726: 	for (int i = arraysize-1; i > 0; i--)
dvander@2726: 	{
januk@3752:         int n = rand() % (i + 1);
dvander@2726: 
dvander@2726: 		cArray->swap(i, n);
dvander@2726: 	}
dvander@2726: }
dvander@2726: 
dvander@2726: static cell_t sm_SortADTArray(IPluginContext *pContext, const cell_t *params)
dvander@2726: {
dvander@2726: 	CellArray *cArray;
dvander@2726: 	HandleError err;
dvander@2726: 	HandleSecurity sec(pContext->GetIdentity(), g_pCoreIdent);
dvander@2726: 
dvander@2726: 	if ((err = handlesys->ReadHandle(params[1], htCellArray, &sec, (void **)&cArray)) 
dvander@2726: 		!= HandleError_None)
dvander@2726: 	{
dvander@2726: 		return pContext->ThrowNativeError("Invalid Handle %x (error: %d)", params[1], err);
dvander@2726: 	}
dvander@2726: 
dvander@2726: 	cell_t order = params[2];
dvander@2726: 	
dvander@2726: 	if (order == Sort_Random)
dvander@2726: 	{
dvander@2726: 		sort_adt_random(cArray);
dvander@2726: 
dvander@2726: 		return 1;
dvander@2726: 	}
dvander@2726: 
dvander@2726: 	cell_t type = params[3];
dvander@2726: 	size_t arraysize = cArray->size();
dvander@2726: 	size_t blocksize = cArray->blocksize();
dvander@2726: 	cell_t *array = cArray->base();
dvander@2726: 
dvander@2726: 	if (type == Sort_Integer)
dvander@2726: 	{
dvander@2726: 		if (order == Sort_Ascending)
dvander@2726: 		{
dvander@2726: 			qsort(array, arraysize, blocksize * sizeof(cell_t), sort_ints_asc);
dvander@2726: 		}
dvander@2726: 		else
dvander@2726: 		{
dvander@2726: 			qsort(array, arraysize, blocksize * sizeof(cell_t), sort_ints_desc);
dvander@2726: 		}
dvander@2726: 	}
dvander@2726: 	else if (type == Sort_Float)
dvander@2726: 	{
dvander@2726: 		if (order == Sort_Ascending)
dvander@2726: 		{
dvander@2726: 			qsort(array, arraysize, blocksize * sizeof(cell_t), sort_floats_asc);
dvander@2726: 		}
dvander@2726: 		else 
dvander@2726: 		{
dvander@2726: 			qsort(array, arraysize, blocksize * sizeof(cell_t), sort_floats_desc);
dvander@2726: 		}
dvander@2726: 	}
dvander@2726: 	else if (type == Sort_String)
dvander@2726: 	{
dvander@2726: 		if (order == Sort_Ascending)
dvander@2726: 		{
dvander@2726: 			qsort(array, arraysize, blocksize * sizeof(cell_t), sort_adtarray_strings_asc);
dvander@2726: 		}
dvander@2726: 		else 
dvander@2726: 		{
dvander@2726: 			qsort(array, arraysize, blocksize * sizeof(cell_t), sort_adtarray_strings_desc);
dvander@2726: 		}
dvander@2726: 	}
dvander@2726: 
dvander@2726: 	return 1;
dvander@2726: }
dvander@2726: 
dvander@2726: struct sort_infoADT
dvander@2726: {
dvander@2726: 	IPluginFunction *pFunc;
dvander@2726: 	cell_t *array_base;
dvander@2726: 	cell_t array_bsize;
dvander@2726: 	Handle_t array_hndl;
dvander@2726: 	Handle_t hndl;
dvander@2726: };
dvander@2726: 
dvander@2726: sort_infoADT g_SortInfoADT;
dvander@2726: 
dvander@2726: int sort_adtarray_custom(const void *elem1, const void *elem2)
dvander@2726: {
dvander@2726: 	cell_t result = 0;
dvander@2726: 	IPluginFunction *pf = g_SortInfoADT.pFunc;
dvander@2726: 	pf->PushCell(((cell_t) ((cell_t *) elem1 - g_SortInfoADT.array_base)) / g_SortInfoADT.array_bsize);
dvander@2726: 	pf->PushCell(((cell_t) ((cell_t *) elem2 - g_SortInfoADT.array_base)) / g_SortInfoADT.array_bsize);
dvander@2726: 	pf->PushCell(g_SortInfoADT.array_hndl);
dvander@2726: 	pf->PushCell(g_SortInfoADT.hndl);
dvander@2726: 	pf->Execute(&result);
dvander@2726: 
dvander@2726: 	return result;
dvander@2726: }
dvander@2726: 
dvander@2726: static cell_t sm_SortADTArrayCustom(IPluginContext *pContext, const cell_t *params)
dvander@2726: {
dvander@2726: 	CellArray *cArray;
dvander@2726: 	HandleError err;
dvander@2726: 	HandleSecurity sec(pContext->GetIdentity(), g_pCoreIdent);
dvander@2726: 
dvander@2726: 	if ((err = handlesys->ReadHandle(params[1], htCellArray, &sec, (void **)&cArray)) 
dvander@2726: 		!= HandleError_None)
dvander@2726: 	{
dvander@2726: 		return pContext->ThrowNativeError("Invalid Handle %x (error: %d)", params[1], err);
dvander@2726: 	}
dvander@2726: 
dvander@2726: 	IPluginFunction *pFunction = pContext->GetFunctionById(params[2]);
dvander@2726: 	if (!pFunction)
dvander@2726: 	{
dvander@2726: 		return pContext->ThrowNativeError("Function %x is not a valid function", params[2]);
dvander@2726: 	}
dvander@2726: 
dvander@2726: 	size_t arraysize = cArray->size();
dvander@2726: 	size_t blocksize = cArray->blocksize();
dvander@2726: 	cell_t *array = cArray->base();
dvander@2726: 
dvander@2726: 	sort_infoADT oldinfo = g_SortInfoADT;
dvander@2726: 
dvander@2726: 	g_SortInfoADT.pFunc = pFunction;
dvander@2726: 	g_SortInfoADT.array_base = array;
dvander@2726: 	g_SortInfoADT.array_bsize = (cell_t) blocksize;
dvander@2726: 	g_SortInfoADT.array_hndl = params[1];
dvander@2726: 	g_SortInfoADT.hndl = params[3];
dvander@2726: 	
dvander@2726: 	qsort(array, arraysize, blocksize * sizeof(cell_t), sort_adtarray_custom);
dvander@2726: 
dvander@2726: 	g_SortInfoADT = oldinfo;
dvander@2726: 
dvander@2726: 	return 1;
dvander@2726: }
dvander@2726: 
dvander@2726: REGISTER_NATIVES(sortNatives)
dvander@2726: {
dvander@2726: 	{"SortIntegers",			sm_SortIntegers},
dvander@2726: 	{"SortFloats",				sm_SortFloats},
dvander@2726: 	{"SortStrings",				sm_SortStrings},
dvander@2726: 	{"SortCustom1D",			sm_SortCustom1D},
dvander@2726: 	{"SortCustom2D",			sm_SortCustom2D},
dvander@2726: 	{"SortADTArray",			sm_SortADTArray},
dvander@2726: 	{"SortADTArrayCustom",		sm_SortADTArrayCustom},
dvander@2726: 	{NULL,						NULL},
dvander@2726: };