src/functionreplace.cpp


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51

#include "functionreplace.h"

#include <bu/plugger.h>
PluginInterface3( pluginFunctionReplace, replace, FunctionReplace, Function,
		"Mike Buland", 0, 1 );

FunctionReplace::FunctionReplace()
{
}

FunctionReplace::~FunctionReplace()
{
}

Bu::String FunctionReplace::getName() const
{
	return "replace";
}

Variable FunctionReplace::call( Variable &input, VarList lParams )
{
	Bu::String sA, sB;
	sA = lParams.first().getString();
	sB = lParams.last().getString();
	switch( input.getType() )
	{
		case Variable::typeString:
			{
				Variable vOut( input.getString().replace( sA, sB ) );
				return vOut;
			}
			break;

		case Variable::typeList:
			{
				Variable vOut( Variable::typeList );
				for( VarList::iterator i = input.begin(); i; i++ )
				{
					vOut.append( (*i).getString().replace( sA, sB ) );
				}
				return vOut;
			}
			break;

		default:
			break;
	}
	throw Bu::ExceptionBase(
		"replace does not work on non-string or non-list types.");
}