1 // Add (std::string) like methods to Nasal strings
3 // Copyright (C) 2013 Thomas Geymayer <tomgey@gmail.com>
5 // This program is free software; you can redistribute it and/or
6 // modify it under the terms of the GNU General Public License as
7 // published by the Free Software Foundation; either version 2 of the
8 // License, or (at your option) any later version.
10 // This program is distributed in the hope that it will be useful, but
11 // WITHOUT ANY WARRANTY; without even the implied warranty of
12 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
13 // General Public License for more details.
15 // You should have received a copy of the GNU General Public License
16 // along with this program; if not, write to the Free Software
17 // Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
23 #include "NasalString.hxx"
25 #include <simgear/nasal/cppbind/from_nasal.hxx>
26 #include <simgear/nasal/cppbind/Ghost.hxx>
27 #include <simgear/nasal/cppbind/NasalHash.hxx>
28 #include <simgear/nasal/cppbind/NasalString.hxx>
31 * Compare (sub)string with other string
34 * compare(pos, len, s)
36 static naRef f_compare(const nasal::CallContext& ctx)
38 nasal::String str = ctx.from_nasal<nasal::String>(ctx.me),
39 rhs = ctx.requireArg<nasal::String>(ctx.argc > 1 ? 2 : 0);
40 size_t pos = ctx.argc > 1 ? ctx.requireArg<int>(1) : 0;
41 size_t len = ctx.argc > 1 ? ctx.requireArg<int>(2) : 0;
44 len = nasal::String::npos;
46 return naNum( str.compare(pos, len, rhs) );
50 * Check whether string starts with other string
52 static naRef f_starts_with(const nasal::CallContext& ctx)
54 nasal::String str = ctx.from_nasal<nasal::String>(ctx.me),
55 rhs = ctx.requireArg<nasal::String>(0);
57 return naNum( str.starts_with(rhs) );
61 * Check whether string ends with other string
63 static naRef f_ends_with(const nasal::CallContext& ctx)
65 nasal::String str = ctx.from_nasal<nasal::String>(ctx.me),
66 rhs = ctx.requireArg<nasal::String>(0);
68 return naNum( str.ends_with(rhs) );
72 * Helper to convert size_t position/npos to Nasal conventions (-1 == npos)
74 naRef pos_to_nasal(size_t pos)
76 if( pos == nasal::String::npos )
83 * Find first occurrence of single character
87 static naRef f_find(const nasal::CallContext& ctx)
89 nasal::String str = ctx.from_nasal<nasal::String>(ctx.me),
90 find = ctx.requireArg<nasal::String>(0);
91 size_t pos = ctx.getArg<int>(1, 0);
93 if( find.size() != 1 )
94 naRuntimeError(ctx.c, "string::find: single character expected");
96 return pos_to_nasal( str.find(*find.c_str(), pos) );
100 * Find first character of a string occurring in this string
102 * find_first_of(search, pos = 0)
104 static naRef f_find_first_of(const nasal::CallContext& ctx)
106 nasal::String str = ctx.from_nasal<nasal::String>(ctx.me),
107 find = ctx.requireArg<nasal::String>(0);
108 size_t pos = ctx.getArg<int>(1, 0);
110 return pos_to_nasal( str.find_first_of(find, pos) );
114 * Find first character of this string not occurring in the other string
116 * find_first_not_of(search, pos = 0)
118 static naRef f_find_first_not_of(const nasal::CallContext& ctx)
120 nasal::String str = ctx.from_nasal<nasal::String>(ctx.me),
121 find = ctx.requireArg<nasal::String>(0);
122 size_t pos = ctx.getArg<int>(1, 0);
124 return pos_to_nasal( str.find_first_not_of(find, pos) );
127 //------------------------------------------------------------------------------
128 naRef initNasalString(naRef globals, naRef string, naContext c)
130 nasal::Hash string_module(string, c);
132 string_module.set("compare", f_compare);
133 string_module.set("starts_with", f_starts_with);
134 string_module.set("ends_with", f_ends_with);
135 string_module.set("find", f_find);
136 string_module.set("find_first_of", f_find_first_of);
137 string_module.set("find_first_not_of", f_find_first_not_of);