[439] | 1 | #include <string.h>
|
---|
| 2 | #include <mruby.h>
|
---|
| 3 | #include <mruby/class.h>
|
---|
| 4 | #include <mruby/data.h>
|
---|
| 5 | #include <mruby/string.h>
|
---|
| 6 |
|
---|
| 7 | /* no argument */
|
---|
| 8 | static mrb_value
|
---|
| 9 | vf_s_format_0(mrb_state *mrb, mrb_value klass)
|
---|
| 10 | {
|
---|
| 11 | mrb_value fmt_str;
|
---|
| 12 | const char *fmt;
|
---|
| 13 |
|
---|
| 14 | mrb_get_args(mrb, "S", &fmt_str);
|
---|
| 15 | fmt = RSTRING_CSTR(mrb, fmt_str);
|
---|
| 16 |
|
---|
| 17 | return mrb_format(mrb, fmt);
|
---|
| 18 | }
|
---|
| 19 |
|
---|
| 20 | /* c char */
|
---|
| 21 | static mrb_value
|
---|
| 22 | vf_s_format_c(mrb_state *mrb, mrb_value klass)
|
---|
| 23 | {
|
---|
| 24 | mrb_value fmt_str, arg_str;
|
---|
| 25 | const char *fmt;
|
---|
| 26 | char c;
|
---|
| 27 |
|
---|
| 28 | mrb_get_args(mrb, "SS", &fmt_str, &arg_str);
|
---|
| 29 | fmt = RSTRING_CSTR(mrb, fmt_str);
|
---|
| 30 | c = RSTRING_CSTR(mrb, arg_str)[0];
|
---|
| 31 |
|
---|
| 32 | return mrb_format(mrb, fmt, c);
|
---|
| 33 | }
|
---|
| 34 |
|
---|
| 35 | /* d int */
|
---|
| 36 | static mrb_value
|
---|
| 37 | vf_s_format_d(mrb_state *mrb, mrb_value klass)
|
---|
| 38 | {
|
---|
| 39 | mrb_value fmt_str;
|
---|
| 40 | const char *fmt;
|
---|
| 41 | mrb_int i;
|
---|
| 42 | int d;
|
---|
| 43 |
|
---|
| 44 | mrb_get_args(mrb, "Si", &fmt_str, &i);
|
---|
| 45 | fmt = RSTRING_CSTR(mrb, fmt_str);
|
---|
| 46 | d = (int)i;
|
---|
| 47 |
|
---|
| 48 | return mrb_format(mrb, fmt, d);
|
---|
| 49 | }
|
---|
| 50 |
|
---|
| 51 | #ifndef MRB_WITHOUT_FLOAT
|
---|
| 52 | /* f float */
|
---|
| 53 | static mrb_value
|
---|
| 54 | vf_s_format_f(mrb_state *mrb, mrb_value klass)
|
---|
| 55 | {
|
---|
| 56 | mrb_value fmt_str;
|
---|
| 57 | const char *fmt;
|
---|
| 58 | mrb_float f;
|
---|
| 59 |
|
---|
| 60 | mrb_get_args(mrb, "Sf", &fmt_str, &f);
|
---|
| 61 | fmt = RSTRING_CSTR(mrb, fmt_str);
|
---|
| 62 |
|
---|
| 63 | return mrb_format(mrb, fmt, f);
|
---|
| 64 | }
|
---|
| 65 | #endif
|
---|
| 66 |
|
---|
| 67 | /* i fixnum */
|
---|
| 68 | static mrb_value
|
---|
| 69 | vf_s_format_i(mrb_state *mrb, mrb_value klass)
|
---|
| 70 | {
|
---|
| 71 | mrb_value fmt_str;
|
---|
| 72 | const char *fmt;
|
---|
| 73 | mrb_int i;
|
---|
| 74 |
|
---|
| 75 | mrb_get_args(mrb, "Si", &fmt_str, &i);
|
---|
| 76 | fmt = RSTRING_CSTR(mrb, fmt_str);
|
---|
| 77 |
|
---|
| 78 | return mrb_format(mrb, fmt, i);
|
---|
| 79 | }
|
---|
| 80 |
|
---|
| 81 | /* l char*, size_t */
|
---|
| 82 | static mrb_value
|
---|
| 83 | vf_s_format_l(mrb_state *mrb, mrb_value klass)
|
---|
| 84 | {
|
---|
| 85 | mrb_value fmt_str, arg_str;
|
---|
| 86 | const char *fmt;
|
---|
| 87 | const char *s;
|
---|
| 88 | mrb_int i;
|
---|
| 89 | size_t len;
|
---|
| 90 |
|
---|
| 91 | mrb_get_args(mrb, "SSi", &fmt_str, &arg_str, &i);
|
---|
| 92 | fmt = RSTRING_CSTR(mrb, fmt_str);
|
---|
| 93 | s = RSTRING_PTR(arg_str);
|
---|
| 94 | len = (size_t)i;
|
---|
| 95 | if (len > (size_t)RSTRING_LEN(arg_str)) len = (size_t)RSTRING_LEN(arg_str);
|
---|
| 96 |
|
---|
| 97 | return mrb_format(mrb, fmt, s, len);
|
---|
| 98 | }
|
---|
| 99 |
|
---|
| 100 | /* n symbol */
|
---|
| 101 | static mrb_value
|
---|
| 102 | vf_s_format_n(mrb_state *mrb, mrb_value klass)
|
---|
| 103 | {
|
---|
| 104 | mrb_value fmt_str;
|
---|
| 105 | const char *fmt;
|
---|
| 106 | mrb_sym sym;
|
---|
| 107 |
|
---|
| 108 | mrb_get_args(mrb, "Sn", &fmt_str, &sym);
|
---|
| 109 | fmt = RSTRING_CSTR(mrb, fmt_str);
|
---|
| 110 |
|
---|
| 111 | return mrb_format(mrb, fmt, sym);
|
---|
| 112 | }
|
---|
| 113 |
|
---|
| 114 | /* s char* */
|
---|
| 115 | static mrb_value
|
---|
| 116 | vf_s_format_s(mrb_state *mrb, mrb_value klass)
|
---|
| 117 | {
|
---|
| 118 | mrb_value fmt_str, arg_str;
|
---|
| 119 | const char *fmt;
|
---|
| 120 | const char *s;
|
---|
| 121 |
|
---|
| 122 | mrb_get_args(mrb, "SS", &fmt_str, &arg_str);
|
---|
| 123 | fmt = RSTRING_CSTR(mrb, fmt_str);
|
---|
| 124 | s = RSTRING_CSTR(mrb, arg_str);
|
---|
| 125 |
|
---|
| 126 | return mrb_format(mrb, fmt, s);
|
---|
| 127 | }
|
---|
| 128 |
|
---|
| 129 | /* C RClass */
|
---|
| 130 | static mrb_value
|
---|
| 131 | vf_s_format_C(mrb_state *mrb, mrb_value klass)
|
---|
| 132 | {
|
---|
| 133 | mrb_value fmt_str, arg_cls;
|
---|
| 134 | const char *fmt;
|
---|
| 135 | struct RClass *c;
|
---|
| 136 |
|
---|
| 137 | mrb_get_args(mrb, "SC", &fmt_str, &arg_cls);
|
---|
| 138 | fmt = RSTRING_CSTR(mrb, fmt_str);
|
---|
| 139 | c = mrb_class_ptr(arg_cls);
|
---|
| 140 |
|
---|
| 141 | return mrb_format(mrb, fmt, c);
|
---|
| 142 | }
|
---|
| 143 |
|
---|
| 144 | /* v value */
|
---|
| 145 | static mrb_value
|
---|
| 146 | vf_s_format_v(mrb_state *mrb, mrb_value klass)
|
---|
| 147 | {
|
---|
| 148 | mrb_value fmt_str, arg_v;
|
---|
| 149 | const char *fmt;
|
---|
| 150 |
|
---|
| 151 | mrb_get_args(mrb, "So", &fmt_str, &arg_v);
|
---|
| 152 | fmt = RSTRING_CSTR(mrb, fmt_str);
|
---|
| 153 |
|
---|
| 154 | return mrb_format(mrb, fmt, arg_v);
|
---|
| 155 | }
|
---|
| 156 |
|
---|
| 157 | void
|
---|
| 158 | mrb_init_test_vformat(mrb_state *mrb)
|
---|
| 159 | {
|
---|
| 160 | struct RClass *vf;
|
---|
| 161 |
|
---|
| 162 | vf = mrb_define_module(mrb, "TestVFormat");
|
---|
| 163 | mrb_define_class_method(mrb, vf, "z", vf_s_format_0, MRB_ARGS_REQ(1));
|
---|
| 164 |
|
---|
| 165 | #define VF_DEFINE_FORMAT_METHOD(t) VF_DEFINE_FORMAT_METHOD_n(t,2)
|
---|
| 166 | #define VF_DEFINE_FORMAT_METHOD_n(t,n) mrb_define_class_method(mrb, vf, #t, vf_s_format_##t, MRB_ARGS_REQ(n));
|
---|
| 167 |
|
---|
| 168 | VF_DEFINE_FORMAT_METHOD(c);
|
---|
| 169 | VF_DEFINE_FORMAT_METHOD(d);
|
---|
| 170 | #ifndef MRB_WITHOUT_FLOAT
|
---|
| 171 | VF_DEFINE_FORMAT_METHOD(f);
|
---|
| 172 | #endif
|
---|
| 173 | VF_DEFINE_FORMAT_METHOD(i);
|
---|
| 174 | VF_DEFINE_FORMAT_METHOD_n(l,3);
|
---|
| 175 | VF_DEFINE_FORMAT_METHOD(n);
|
---|
| 176 | VF_DEFINE_FORMAT_METHOD(s);
|
---|
| 177 | VF_DEFINE_FORMAT_METHOD(C);
|
---|
| 178 | VF_DEFINE_FORMAT_METHOD(v);
|
---|
| 179 | }
|
---|