167 lines
3.7 KiB
C++
Executable File
167 lines
3.7 KiB
C++
Executable File
/* AMX Assembler
|
|
* Copyright (C)2004 David "BAILOPAN" Anderson
|
|
*
|
|
* This software is provided 'as-is', without any express or implied
|
|
* warranty. In no event will the authors be held liable for any damages
|
|
* arising from the use of this software.
|
|
*
|
|
* Permission is granted to anyone to use this software for any purpose,
|
|
* including commercial applications, and to alter it and redistribute it
|
|
* freely, subject to the following restrictions:
|
|
*
|
|
* 1. The origin of this software must not be misrepresented; you must not
|
|
* claim that you wrote the original software. If you use this software
|
|
* in a product, an acknowledgment in the product documentation would be
|
|
* appreciated but is not required.
|
|
* 2. Altered source versions must be plainly marked as such, and must not be
|
|
* misrepresented as being the original software.
|
|
* 3. This notice may not be removed or altered from any source distribution.
|
|
*
|
|
* Version: $Id$
|
|
*/
|
|
|
|
#include "amxasm.h"
|
|
|
|
MacroList::~MacroList()
|
|
{
|
|
std::vector<MacroList::Macro *>::iterator i;
|
|
|
|
for (i=List.begin(); i!=List.end(); i++)
|
|
{
|
|
if ( (*i) )
|
|
delete (*i);
|
|
}
|
|
|
|
List.clear();
|
|
}
|
|
|
|
MacroList::Macro::~Macro()
|
|
{
|
|
std::vector<std::string *>::iterator i;
|
|
|
|
for (i=argList.begin(); i!=argList.end(); i++)
|
|
{
|
|
if ( (*i) )
|
|
delete (*i);
|
|
}
|
|
|
|
argList.clear();
|
|
}
|
|
|
|
MacroList::MacroList()
|
|
{
|
|
printf("Not instantiated with a compiler\n");
|
|
exit(0);
|
|
}
|
|
|
|
MacroList::MacroList(void *c)
|
|
{
|
|
CError = ((Compiler *)c)->ErrorHandler();
|
|
Cmp = c;
|
|
}
|
|
|
|
std::string *MacroList::BeginReplacement(MacroList::Macro *macro)
|
|
{
|
|
std::string *mstring = new std::string(macro->macro);
|
|
|
|
macro->arg = macro->argList.begin();
|
|
macro->argpos = 0;
|
|
|
|
return mstring;
|
|
}
|
|
|
|
int MacroList::ReplaceArgument(MacroList::Macro *m, std::string *macro, std::string &arg, int pos = 0)
|
|
{
|
|
int i = 0, bPos = 0;
|
|
|
|
bPos = FindSymbol(*macro, *(*m->arg), pos);
|
|
|
|
if (bPos != -1)
|
|
{
|
|
macro->replace(bPos, (*m->arg)->size(), arg);
|
|
bPos += (int)arg.size();
|
|
}
|
|
|
|
m->arg++;
|
|
m->argpos++;
|
|
|
|
return bPos;
|
|
}
|
|
|
|
void MacroList::EndReplacement(MacroList::Macro *m, std::string *macro)
|
|
{
|
|
if (m->arg != m->argList.end())
|
|
{
|
|
CError->ErrorMsg(Err_MacroParamCount, m->symbol.c_str());
|
|
}
|
|
|
|
m->arg = m->argList.begin();
|
|
m->argpos = 0;
|
|
}
|
|
|
|
MacroList::Macro *MacroList::AddMacroBegin(std::string &symbol, std::string &mac)
|
|
{
|
|
Macro *macro = new Macro;
|
|
macro->macro.assign(mac);
|
|
macro->symbol.assign(symbol);
|
|
return macro;
|
|
}
|
|
|
|
void MacroList::AddMacroArgument(MacroList::Macro *m, std::string &arg)
|
|
{
|
|
std::string *sArg = new std::string(arg);
|
|
m->argList.push_back(sArg);
|
|
}
|
|
|
|
void MacroList::AddMacroEnd(MacroList::Macro *m)
|
|
{
|
|
List.push_back(m);
|
|
}
|
|
|
|
MacroList::Macro *MacroList::FindMacro(std::string &sym)
|
|
{
|
|
std::vector<Macro *>::iterator i;
|
|
for (i=List.begin(); i!=List.end(); i++)
|
|
{
|
|
if ((*i)->macro.compare(sym) == 0)
|
|
return (*i);
|
|
}
|
|
|
|
return NULL;
|
|
}
|
|
|
|
void MacroList::SearchAndReplace(std::string &text)
|
|
{
|
|
std::vector<Macro *>::iterator i;
|
|
MacroList::Macro *m = NULL;
|
|
int pos=0, symPos=0, bPos=0, argPos=0;
|
|
|
|
for (i=List.begin(); i!=List.end(); i++)
|
|
{
|
|
m = (*i);
|
|
pos = FindSymbol(text, m->symbol, 0);
|
|
if (pos != -1)
|
|
{
|
|
/* Strip the arguments */
|
|
std::string argstring;
|
|
symPos = pos + (int)m->symbol.size();
|
|
argstring.assign(text.substr(symPos+1, text.size()-symPos));
|
|
std::vector<std::string *> argList;
|
|
((Compiler *)Cmp)->FindArguments(argstring, argList, bPos);
|
|
/* Build the macro */
|
|
std::string *ms;
|
|
ms = BeginReplacement(m);
|
|
std::vector<std::string *>::iterator j;
|
|
for (j=argList.begin(); j!=argList.end(); j++)
|
|
{
|
|
argPos = ReplaceArgument(m, ms, *(*j), argPos);
|
|
}
|
|
EndReplacement(m, ms);
|
|
/* Search and replace */
|
|
text.replace(pos, bPos-pos, *ms);
|
|
/* Cleanup */
|
|
delete ms;
|
|
i = List.begin();
|
|
}
|
|
}
|
|
} |