/* kc3
* Copyright 2022,2023,2024 kmx.io <contact@kmx.io>
*
* Permission is hereby granted to use this software granted the above
* copyright notice and this permission paragraph are included in all
* copies and substantial portions of this software.
*
* THIS SOFTWARE IS PROVIDED "AS-IS" WITHOUT ANY GUARANTEE OF
* PURPOSE AND PERFORMANCE. IN NO EVENT WHATSOEVER SHALL THE
* AUTHOR BE CONSIDERED LIABLE FOR THE USE AND PERFORMANCE OF
* THIS SOFTWARE.
*/
#include <float.h>
#include <math.h>
#include <string.h>
#include "../libtommath/tommath.h"
#include "alloc.h"
#include "assert.h"
#include "array.h"
#include "buf.h"
#include "buf_inspect.h"
#include "buf_save.h"
#include "character.h"
#include "cow.h"
#include "data.h"
#include "ident.h"
#include "integer.h"
#include "io.h"
#include "list.h"
#include "operator.h"
#include "special_operator.h"
#include "str.h"
#include "tag.h"
sw buf_inspect_array_data (s_buf *buf, const s_array *array);
sw buf_inspect_array_data_rec (s_buf *buf, const s_array *array,
const u8 **data, const s_tag **tag,
uw *address, uw dimension);
sw buf_inspect_array_data_size (const s_array *array);
sw buf_inspect_array_data_size_rec (const s_array *array,
const u8 **data,
const s_tag **tag,
uw *address, uw dimension);
sw buf_inspect_tag_type (s_buf *buf, e_tag_type type);
sw buf_inspect_array (s_buf *buf, const s_array *array)
{
sw r;
sw result = 0;
assert(buf);
assert(array);
assert(sym_is_array_type(array->array_type));
if ((r = buf_inspect_paren_sym(buf, array->array_type)) <= 0)
goto clean;
result += r;
if ((r = buf_write_1(buf, " ")) < 0)
goto clean;
result += r;
if (! array->dimension) {
if ((r = buf_write_1(buf, "{}")) < 0)
goto clean;
result += r;
r = result;
goto clean;
}
if ((r = buf_inspect_array_data(buf, array)) < 0) {
err_puts("buf_inspect_array: buf_inspect_array_data");
goto clean;
}
result += r;
r = result;
clean:
return r;
}
sw buf_inspect_array_data (s_buf *buf, const s_array *array)
{
uw *address;
u8 *data = NULL;
s_tag *tag = NULL;
sw r;
assert(buf);
assert(array);
// TODO: buf_inspect_array_data_rec_data
// TODO: buf_inspect_array_data_rec_tags
if (array->data)
data = array->data;
else
tag = array->tags;
address = alloc(array->dimension * sizeof(uw));
if (! address)
return -1;
r = buf_inspect_array_data_rec(buf, array, (const u8 **) &data,
(const s_tag **) &tag, address, 0);
free(address);
return r;
}
sw buf_inspect_array_data_rec (s_buf *buf, const s_array *array,
const u8 **data, const s_tag **tag,
uw *address, uw dimension)
{
sw r;
sw result = 0;
if ((r = buf_write_1(buf, "{")) <= 0)
goto clean;
result += r;
address[dimension] = 0;
while (1) {
if (dimension == array->dimension - 1) {
if (*data) {
if ((r = data_buf_inspect(array->element_type,
buf, *data)) <= 0)
goto clean;
result += r;
*data += array->dimensions[dimension].item_size;
}
else if (*tag) {
if ((r = buf_inspect_tag(buf, *tag)) <= 0)
goto clean;
result += r;
(*tag)++;
}
}
else {
if ((r = buf_inspect_array_data_rec(buf, array, data, tag,
address, dimension + 1)) <= 0)
goto clean;
result += r;
}
address[dimension]++;
if (address[dimension] == array->dimensions[dimension].count)
break;
if ((r = buf_write_1(buf, ", ")) <= 0)
goto clean;
result += r;
}
if ((r = buf_write_1(buf, "}")) <= 0)
goto clean;
result += r;
r = result;
clean:
return r;
}
sw buf_inspect_array_data_size (const s_array *array)
{
uw *address;
u8 *data = NULL;
s_tag *tag = NULL;
sw r;
assert(array);
if (array->data)
data = array->data;
else
tag = array->tags;
address = alloc(array->dimension * sizeof(uw));
if (! address)
return -1;
r = buf_inspect_array_data_size_rec(array, (const u8 **) &data,
(const s_tag **) &tag,
address, 0);
free(address);
return r;
}
sw buf_inspect_array_data_size_rec (const s_array *array,
const u8 **data,
const s_tag **tag,
uw *address, uw dimension)
{
sw r;
sw result = 0;
r = strlen("{");
result += r;
address[dimension] = 0;
while (1) {
if (dimension == array->dimension - 1) {
if (*data) {
if ((r = data_buf_inspect_size(array->element_type,
*data)) <= 0)
goto clean;
result += r;
*data += array->dimensions[dimension].item_size;
}
else if (*tag) {
if ((r = buf_inspect_tag_size(*tag)) <= 0)
goto clean;
result += r;
(*tag)++;
}
}
else {
if ((r = buf_inspect_array_data_size_rec(array, data, tag,
address,
dimension + 1)) <= 0)
goto clean;
result += r;
}
address[dimension]++;
if (address[dimension] == array->dimensions[dimension].count)
break;
r = strlen(", ");
result += r;
}
r = strlen("}");
result += r;
r = result;
clean:
return r;
}
sw buf_inspect_array_size (const s_array *array)
{
sw r;
sw result = 0;
assert(array);
if ((r = buf_inspect_paren_sym_size(array->array_type)) <= 0)
goto clean;
result += r;
r = strlen(" ");
result += r;
if ((r = buf_inspect_array_data_size(array)) <= 0) {
err_puts("buf_inspect_array_size: buf_inspect_array_data");
goto clean;
}
result += r;
r = result;
clean:
return r;
}
sw buf_inspect_block (s_buf *buf, const s_block *block)
{
u64 i = 0;
sw r;
sw result = 0;
if (! block->count) {
if (block->short_form)
return buf_write_1(buf, "{}");
else
return buf_write_1(buf, "do end");
}
if (block->short_form) {
if ((r = buf_write_1(buf, "{ ")) < 0)
return r;
}
else {
if ((r = buf_write_1(buf, "do\n ")) < 0)
return r;
}
result += r;
while (i < block->count - 1) {
if ((r = buf_inspect_tag(buf, block->tag + i)) < 0)
return r;
result += r;
if (block->short_form) {
if ((r = buf_write_1(buf, "; ")) < 0)
return r;
}
else {
if ((r = buf_write_1(buf, "\n ")) < 0)
return r;
}
result += r;
i++;
}
if ((r = buf_inspect_tag(buf, block->tag + i)) < 0)
return r;
result += r;
if (block->short_form) {
if ((r = buf_write_1(buf, " }")) < 0)
return r;
}
else {
if ((r = buf_write_1(buf, "\nend")) < 0)
return r;
}
result += r;
return result;
}
sw buf_inspect_block_inner (s_buf *buf, const s_block *block)
{
u64 i = 0;
sw r;
sw result = 0;
if (! block->count)
return 0;
if (block->short_form) {
if ((r = buf_write_1(buf, " ")) < 0)
return r;
}
else {
if ((r = buf_write_1(buf, "\n ")) < 0)
return r;
}
result += r;
while (i < block->count - 1) {
if ((r = buf_inspect_tag(buf, block->tag + i)) < 0)
return r;
result += r;
if (block->short_form) {
if ((r = buf_write_1(buf, "; ")) < 0)
return r;
}
else {
if ((r = buf_write_1(buf, "\n ")) < 0)
return r;
}
result += r;
i++;
}
if ((r = buf_inspect_tag(buf, block->tag + i)) < 0)
return r;
result += r;
return result;
}
sw buf_inspect_block_size (const s_block *block)
{
u64 i = 0;
sw r;
sw result;
if (! block->count)
result = strlen("do end");
result = strlen("do");
while (i < block->count) {
result += strlen("\n ");
if ((r = buf_inspect_tag_size(block->tag + i)) < 0)
return r;
result += r;
i++;
}
result += strlen("\nend");
return result;
}
sw buf_inspect_bool (s_buf *buf, const bool *b)
{
if (*b)
return buf_write_1(buf, "true");
return buf_write_1(buf, "false");
}
sw buf_inspect_bool_size (const bool *b)
{
if (*b)
return strlen("true");
return strlen("false");
}
sw buf_inspect_call (s_buf *buf, const s_call *call)
{
bool b;
bool op;
u8 op_arity;
s8 op_precedence;
sw r;
sw result = 0;
const s_sym *sym;
if (call->ident.module == &g_sym_KC3 &&
call->ident.sym == &g_sym_access)
return buf_inspect_call_access(buf, call);
if (call->ident.module == &g_sym_KC3 &&
call->ident.sym == &g_sym_if_then_else)
return buf_inspect_call_if_then_else(buf, call);
if (! operator_find(&call->ident, &op))
return -1;
if (op) {
if (! operator_symbol(&call->ident, &sym))
return -1;
if (sym == &g_sym__brackets)
return buf_inspect_call_brackets(buf, call);
}
if (call->ident.sym == &g_sym_cast)
return buf_inspect_cast(buf, call);
op_arity = op ? operator_arity(&call->ident) : 0;
if (op_arity == 1)
return buf_inspect_call_op_unary(buf, call);
if (op_arity == 2 &&
(op_precedence = operator_precedence(&call->ident)) > 0)
return buf_inspect_call_op(buf, call, op_precedence);
if (! ident_is_special_operator(&call->ident, &b))
return -1;
if (b)
return buf_inspect_call_special_operator(buf, call);
if ((r = buf_inspect_ident(buf, &call->ident)) < 0)
return r;
result += r;
if ((r = buf_inspect_call_args(buf, call->arguments)) < 0)
return r;
result += r;
return result;
}
sw buf_inspect_call_access (s_buf *buf, const s_call *call)
{
s_list *args;
sw r;
sw result = 0;
s_tag *tag_sym;
assert(buf);
assert(call);
args = call->arguments;
tag_sym = &args->next.data.list->tag;
if ((r = buf_inspect_tag(buf, &args->tag)) < 0)
return r;
result += r;
if ((r = buf_write_1(buf, ".")) < 0)
return r;
result += r;
if ((r = buf_inspect_ident_sym(buf, tag_sym->data.sym)) < 0)
return r;
result += r;
return result;
}
sw buf_inspect_call_args (s_buf *buf, const s_list *args)
{
sw r;
sw result = 0;
if ((r = buf_write_1(buf, "(")) < 0)
return r;
result += r;
while (args) {
if ((r = buf_inspect_tag(buf, &args->tag)) < 0)
return r;
result += r;
if ((args = list_next(args))) {
if ((r = buf_write_1(buf, ", ")) < 0)
return r;
result += r;
}
}
if ((r = buf_write_1(buf, ")")) < 0)
return r;
result += r;
return result;
}
sw buf_inspect_call_args_size (const s_list *args)
{
sw r;
sw result = 0;
result += strlen("(");
while (args) {
if ((r = buf_inspect_tag_size(&args->tag)) < 0)
return r;
result += r;
if ((args = list_next(args))) {
result += strlen(", ");
}
}
result += strlen(")");
return result;
}
sw buf_inspect_call_brackets (s_buf *buf, const s_call *call)
{
s_array *address;
s_tag *array;
uw i = 0;
s_list *next;
sw r;
sw result = 0;
assert(buf);
assert(call);
assert(call->arguments);
next = list_next(call->arguments);
assert(next);
assert(! list_next(next));
assert(next->tag.type == TAG_ARRAY);
address = &next->tag.data.array;
assert(address->dimension == 1);
array = &call->arguments->tag;
if ((r = buf_inspect_tag(buf, array)) < 0)
return r;
result += r;
while (i < address->dimensions[0].count) {
if ((r = buf_write_1(buf, "[")) < 0)
return r;
result += r;
if ((r = buf_inspect_uw(buf, ((uw *) address->data)
+ i)) < 0)
return r;
result += r;
if ((r = buf_write_1(buf, "]")) < 0)
return r;
result += r;
i++;
}
return result;
}
sw buf_inspect_call_brackets_size (const s_call *call)
{
s_array *address;
s_tag *array;
uw i = 0;
s_list *next;
sw r;
sw result = 0;
assert(call);
assert(call->arguments);
next = list_next(call->arguments);
assert(next);
assert(! list_next(next));
assert(next->tag.type == TAG_ARRAY);
address = &next->tag.data.array;
assert(address->dimension == 1);
array = &call->arguments->tag;
if ((r = buf_inspect_tag_size(array)) < 0)
return r;
result += r;
while (i < address->dimensions[0].count) {
result += strlen("[");
if ((r = buf_inspect_uw_size(((uw *) address->data)
+ i)) < 0)
return r;
result += r;
result += strlen("]");
i++;
}
return result;
}
sw buf_inspect_call_if_then_else (s_buf *buf, const s_call *call)
{
s_tag *condition;
s_tag *else_;
sw r;
sw result = 0;
s_tag *then;
if (call->ident.module != &g_sym_KC3 ||
call->ident.sym != &g_sym_if_then_else ||
list_length(call->arguments) != 3)
return -2;
if ((r = buf_write_1(buf, "if ")) < 0)
return r;
result += r;
condition = &call->arguments->tag;
if ((r = buf_inspect_tag(buf, condition)) < 0)
return r;
result += r;
then = &list_next(call->arguments)->tag;
if (then->type == TAG_BLOCK) {
if ((r = buf_inspect_block_inner(buf, &then->data.block)) < 0)
return r;
}
else
if ((r = buf_inspect_tag(buf, then)) < 0)
return r;
result += r;
else_ = &list_next(list_next(call->arguments))->tag;
if (else_->type != TAG_VOID) {
if ((r = buf_write_1(buf, "\nelse")) < 0)
return r;
result += r;
if (then->type == TAG_BLOCK) {
if ((r = buf_inspect_block_inner(buf, &else_->data.block)) < 0)
return r;
}
else
if ((r = buf_inspect_tag(buf, else_)) < 0)
return r;
result += r;
}
if ((r = buf_write_1(buf, "\nend")) < 0)
return r;
result += r;
return result;
}
sw buf_inspect_call_if_then_else_size (const s_call *call)
{
(void) call;
return -1;
}
sw buf_inspect_call_op (s_buf *buf, const s_call *call, s8 op_precedence)
{
s_ident ident;
s_tag *left;
bool op;
bool paren = false;
s8 precedence;
sw r;
sw result = 0;
s_tag *right;
left = &call->arguments->tag;
assert(list_next(call->arguments));
right = &list_next(call->arguments)->tag;
if (left->type == TAG_CALL) {
if (! operator_find(&left->data.call.ident, &op))
return -1;
if (op &&
(precedence = operator_precedence(&left->data.call.ident))
< op_precedence) {
paren = true;
if ((r = buf_write_1(buf, "(")) < 0)
return r;
result += r;
}
}
if ((r = buf_inspect_tag(buf, left)) < 0)
return r;
result += r;
if (paren) {
if ((r = buf_write_1(buf, ")")) < 0)
return r;
result += r;
}
if ((r = buf_write_1(buf, " ")) < 0)
return r;
result += r;
if (! operator_ident(&call->ident, &ident))
return -1;
if ((r = buf_inspect_ident(buf, &ident)) < 0)
return r;
result += r;
if ((r = buf_write_1(buf, " ")) < 0)
return r;
result += r;
if (right->type == TAG_CALL) {
if (! operator_find(&right->data.call.ident, &op))
return -1;
if (op &&
(precedence = operator_precedence(&right->data.call.ident))
< op_precedence) {
paren = true;
if ((r = buf_write_1(buf, "(")) < 0)
return r;
result += r;
}
else
paren = false;
}
if ((r = buf_inspect_tag(buf, right)) < 0)
return r;
result += r;
if (paren) {
if ((r = buf_write_1(buf, ")")) < 0)
return r;
result += r;
}
return result;
}
sw buf_inspect_call_op_size (const s_call *call, s8 op_precedence)
{
(void) call;
(void) op_precedence;
return -1;
}
sw buf_inspect_call_op_unary (s_buf *buf, const s_call *call)
{
sw r;
sw result = 0;
const s_sym *sym;
s_ident tmp;
assert(buf);
assert(call);
if (! operator_symbol(&call->ident, &sym))
return -1;
if (sym == &g_sym__paren)
return buf_inspect_call_paren(buf, call);
if (operator_ident(&call->ident, &tmp) != &tmp)
return -1;
if ((r = buf_inspect_ident(buf, &tmp)) < 0)
return r;
result += r;
if ((r = buf_write_1(buf, " ")) < 0)
return r;
result += r;
if ((r = buf_inspect_tag(buf, &call->arguments->tag)) < 0)
return r;
result += r;
return result;
}
sw buf_inspect_call_op_unary_size (const s_call *call)
{
(void) call;
return -1;
}
sw buf_inspect_call_paren (s_buf *buf, const s_call *call)
{
sw r;
sw result = 0;
assert(buf);
assert(call);
if ((r = buf_write_1(buf, "(")) < 0)
return r;
result += r;
if ((r = buf_inspect_tag(buf, &call->arguments->tag)) < 0)
return r;
result += r;
if ((r = buf_write_1(buf, ")")) < 0)
return r;
result += r;
return result;
}
sw buf_inspect_call_size (const s_call *call)
{
bool b;
bool op;
u8 op_arity;
s8 op_precedence;
sw r;
sw result = 0;
const s_sym *sym;
if (call->ident.module == &g_sym_KC3 &&
call->ident.sym == &g_sym_if_then_else)
return buf_inspect_call_if_then_else_size(call);
if (! operator_find(&call->ident, &op))
return -1;
if (op) {
if (! operator_symbol(&call->ident, &sym))
return -1;
if (sym == &g_sym__brackets)
return buf_inspect_call_brackets_size(call);
}
if (call->ident.sym == &g_sym_cast)
return buf_inspect_cast_size(call);
op_arity = op ? operator_arity(&call->ident) : 0;
if (op_arity == 1)
return buf_inspect_call_op_unary_size(call);
if (op_arity == 2 &&
(op_precedence = operator_precedence(&call->ident)) > 0)
return buf_inspect_call_op_size(call, op_precedence);
if (! ident_is_special_operator(&call->ident, &b))
return -1;
if (b)
return buf_inspect_call_special_operator_size(call);
if ((r = buf_inspect_ident_size(&call->ident)) < 0)
return r;
result += r;
if ((r = buf_inspect_call_args_size(call->arguments)) < 0)
return r;
result += r;
return result;
}
sw buf_inspect_call_special_operator (s_buf *buf, const s_call *call)
{
const s_list *args;
bool b;
sw r;
sw result = 0;
if (! ident_is_special_operator(&call->ident, &b))
return -1;
if (! b) {
err_puts("buf_inspect_call_special_operator:"
" not a special operator");
assert(! "buf_inspect_call_special_operator:"
" not a special operator");
return -1;
}
if (special_operator_arity(&call->ident) !=
list_length(call->arguments)) {
err_puts("buf_inspect_call_special_operator:"
" invalid arity");
assert(! "buf_inspect_call_special_operator:"
" invalid arity");
return -1;
}
args = call->arguments;
if ((r = buf_inspect_ident(buf, &call->ident)) < 0)
return r;
result += r;
while (args) {
if ((r = buf_write_1(buf, " ")) < 0)
return r;
result += r;
if ((r = buf_inspect_tag(buf, &args->tag)) < 0)
return r;
result += r;
args = list_next(args);
}
return result;
}
sw buf_inspect_call_special_operator_size (const s_call *call)
{
const s_list *args;
bool b;
sw r;
sw result = 0;
if (! ident_is_special_operator(&call->ident, &b))
return -1;
if (! b) {
err_puts("buf_inspect_call_special_operator_size:"
" not a special operator");
assert(! "buf_inspect_call_special_operator_size:"
" not a special operator");
return -1;
}
if (special_operator_arity(&call->ident) !=
list_length(call->arguments)) {
err_puts("buf_inspect_call_special_operator_size:"
" invalid arity");
assert(! "buf_inspect_call_special_operator_size:"
" invalid arity");
return -1;
}
args = call->arguments;
if ((r = buf_inspect_ident_size(&call->ident)) < 0)
return r;
result += r;
while (args) {
result += strlen(" ");
if ((r = buf_inspect_tag_size(&args->tag)) < 0)
return r;
result += r;
args = list_next(args);
}
return result;
}
sw buf_inspect_cast (s_buf *buf, const s_call *call)
{
s_tag *arg;
const s_sym *module;
sw r;
sw result = 0;
assert(buf);
assert(call);
assert(call->arguments);
assert(list_next(call->arguments));
assert(! list_next(list_next(call->arguments)));
module = call->ident.module;
if ((r = buf_inspect_paren_sym(buf, module)) < 0)
return r;
result += r;
if ((r = buf_write_1(buf, " ")) < 0)
return r;
result += r;
arg = &list_next(call->arguments)->tag;
if ((r = buf_inspect_tag(buf, arg)) < 0)
return r;
result += r;
return result;
}
sw buf_inspect_cast_size (const s_call *call)
{
s_tag *arg;
const s_sym *module;
sw r;
sw result = 0;
assert(call);
assert(call->arguments);
assert(! list_next(call->arguments));
module = call->ident.module;
if ((r = buf_inspect_paren_sym_size(module)) <= 0)
return r;
result += r;
result += strlen(" ");
arg = &call->arguments->tag;
if ((r = buf_inspect_tag_size(arg)) <= 0)
return r;
result += r;
return result;
}
sw buf_inspect_cfn (s_buf *buf, const s_cfn *cfn)
{
sw r;
sw result = 0;
assert(cfn);
if ((r = buf_write_1(buf, "cfn ")) < 0)
return r;
result += r;
if ((r = buf_inspect_sym(buf, &cfn->result_type)) < 0)
return r;
result += r;
if ((r = buf_write_1(buf, " ")) < 0)
return r;
result += r;
if ((r = buf_inspect_str(buf, &cfn->name->str)) < 0)
return r;
result += r;
if ((r = buf_write_1(buf, " ")) < 0)
return r;
result += r;
if ((r = buf_inspect_list_paren(buf, (const s_list * const *)
&cfn->arg_types)) < 0)
return r;
result += r;
return result;
}
sw buf_inspect_cfn_size (const s_cfn *cfn)
{
s_list *arg_type;
sw r;
sw result = 0;
assert(cfn);
if ((r = buf_inspect_str_size(&cfn->name->str)) < 0)
return r;
result += r;
result += strlen("(");
arg_type = cfn->arg_types;
while (arg_type) {
if ((r = buf_inspect_tag_size(&arg_type->tag)) < 0)
return r;
arg_type = list_next(arg_type);
if (arg_type)
result += strlen(", ");
}
result += strlen(")");
return result;
}
sw buf_inspect_character (s_buf *buf, const character *c)
{
sw r;
sw result = 0;
s_buf_save save;
buf_save_init(buf, &save);
if ((r = buf_write_1(buf, "'")) < 0)
return r;
result += r;
if ((r = buf_inspect_str_character(buf, c)) <= 0)
goto restore;
result += r;
if ((r = buf_write_1(buf, "'")) < 0)
goto restore;
result += r;
r = result;
goto clean;
restore:
buf_save_restore_wpos(buf, &save);
clean:
buf_save_clean(buf, &save);
return result;
}
sw buf_inspect_character_size (const character *c)
{
sw r;
sw result = 0;
result += strlen("'");
if ((r = buf_inspect_str_character_size(c)) <= 0)
return r;
result += r;
result += strlen("'");
return result;
}
sw buf_inspect_complex (s_buf *buf, const s_complex *c)
{
sw r;
sw result = 0;
s_buf_save save;
buf_save_init(buf, &save);
if ((r = buf_inspect_tag(buf, &c->x)) < 0)
goto restore;
result += r;
if ((r = buf_write_1(buf, " +i ")) < 0)
goto restore;
result += r;
if ((r = buf_inspect_tag(buf, &c->y)) < 0)
goto restore;
result += r;
r = result;
goto clean;
restore:
buf_save_restore_wpos(buf, &save);
clean:
buf_save_clean(buf, &save);
return r;
}
sw buf_inspect_complex_size (const s_complex *c)
{
sw r;
sw result = 0;
if ((r = buf_inspect_tag_size(&c->x)) < 0)
return r;
result += r;
result += strlen(" +i ");
if ((r = buf_inspect_tag_size(&c->y)) < 0)
return r;
result += r;
return result;
}
sw buf_inspect_cow (s_buf *buf, const s_cow *cow)
{
sw r;
sw result = 0;
s_buf_save save;
buf_save_init(buf, &save);
if ((r = buf_write_1(buf, "cow ")) < 0)
goto restore;
if ((r = buf_inspect_tag(buf, cow_read_only(cow))) < 0)
goto restore;
result += r;
r = result;
goto clean;
restore:
buf_save_restore_wpos(buf, &save);
clean:
buf_save_clean(buf, &save);
return r;
}
sw buf_inspect_cow_size (const s_cow *cow)
{
sw r;
sw result;
result = strlen("cow ");
if ((r = buf_inspect_tag_size(cow_read_only(cow))) < 0)
return r;
result += r;
return result;
}
sw buf_inspect_f32 (s_buf *buf, const f32 *f)
{
s32 exp;
u8 i;
u8 j;
sw r;
sw result = 0;
f32 x;
assert(buf);
assert(f);
exp = 0;
x = *f;
if (x == 0.0)
return buf_write_1(buf, "0.0f");
if (x < 0) {
if ((r = buf_write_1(buf, "-")) <= 0)
return r;
result += r;
x = -x;
}
if (x >= 1.0)
while (x >= 10.0) {
x /= 10.0;
exp++;
}
else
while (x < 1.0) {
x *= 10.0;
exp--;
}
i = (u8) x;
x -= i;
i += '0';
if ((r = buf_write_u8(buf, i)) <= 0)
return r;
result += r;
if ((r = buf_write_1(buf, ".")) <= 0)
return r;
result += r;
j = 6;
do {
x *= 10;
i = (u8) x;
x -= i;
i += '0';
if ((r = buf_write_u8(buf, i)) <= 0)
return r;
result += r;
j--;
} while (x >= pow(0.1, j) && j);
if (exp) {
if ((r = buf_write_1(buf, "e")) <= 0)
return r;
result += r;
if (exp > 0) {
if ((r = buf_write_1(buf, "+")) <= 0)
return r;
result += r;
}
if ((r = buf_inspect_s32(buf, &exp)) <= 0)
return r;
result += r;
}
if ((r = buf_write_1(buf, "f")) <= 0)
return r;
result += r;
return result;
}
sw buf_inspect_f32_size (const f32 *f)
{
char b[32];
s_buf buf;
assert(f);
buf_init(&buf, false, sizeof(b), b);
return buf_inspect_f32(&buf, f);
}
sw buf_inspect_f64 (s_buf *buf, const f64 *f)
{
s64 exp;
u8 i;
u8 j;
sw r;
sw result = 0;
f64 x;
assert(buf);
assert(f);
exp = 0.0;
x = *f;
if (x == 0.0)
return buf_write_1(buf, "0.0");
if (x < 0) {
if ((r = buf_write_1(buf, "-")) <= 0)
return r;
result += r;
x = -x;
}
if (x >= 1.0)
while (x >= 10.0) {
x /= 10.0;
exp++;
}
else
while (x < 1.0) {
x *= 10.0;
exp--;
}
i = (u8) x;
x -= i;
i += '0';
if ((r = buf_write_u8(buf, i)) <= 0)
return r;
result += r;
if ((r = buf_write_1(buf, ".")) <= 0)
return r;
result += r;
j = 14;
do {
x *= 10;
i = (u8) x;
x -= i;
i += '0';
if ((r = buf_write_u8(buf, i)) <= 0)
return r;
result += r;
j--;
} while (x > pow(0.1, j) && j);
if (exp) {
if ((r = buf_write_1(buf, "e")) <= 0)
return r;
result += r;
if (exp > 0) {
if ((r = buf_write_1(buf, "+")) <= 0)
return r;
result += r;
}
if ((r = buf_inspect_s64(buf, &exp)) <= 0)
return r;
result += r;
}
return result;
}
sw buf_inspect_f64_size (const f64 *x)
{
char b[64];
s_buf buf;
assert(x);
buf_init(&buf, false, sizeof(b), b);
return buf_inspect_f64(&buf, x);
}
sw buf_inspect_f128 (s_buf *buf, const f128 *x)
{
s64 exp;
u8 i;
u8 j;
sw r;
sw result = 0;
f128 y;
assert(buf);
assert(x);
exp = 0.0;
y = *x;
if (y == 0.0)
return buf_write_1(buf, "0.0");
if (y < 0) {
if ((r = buf_write_1(buf, "-")) <= 0)
return r;
result += r;
y = -y;
}
if (y >= 1.0)
while (y >= 10.0) {
y /= 10.0;
exp++;
}
else
while (y < 1.0) {
y *= 10.0;
exp--;
}
i = (u8) y;
y -= i;
i += '0';
if ((r = buf_write_u8(buf, i)) <= 0)
return r;
result += r;
if ((r = buf_write_1(buf, ".")) <= 0)
return r;
result += r;
j = 33;
do {
y *= 10;
i = (u8) y;
y -= i;
i += '0';
if ((r = buf_write_u8(buf, i)) <= 0)
return r;
result += r;
j--;
} while (y > pow(0.1, j) && j);
if (exp) {
if ((r = buf_write_1(buf, "e")) <= 0)
return r;
result += r;
if (exp > 0) {
if ((r = buf_write_1(buf, "+")) <= 0)
return r;
result += r;
}
if ((r = buf_inspect_s64(buf, &exp)) <= 0)
return r;
result += r;
}
return result;
}
sw buf_inspect_f128_size (const f128 *x)
{
char b[128];
s_buf buf;
assert(x);
buf_init(&buf, false, sizeof(b), b);
return buf_inspect_f128(&buf, x);
}
sw buf_inspect_fact (s_buf *buf, const s_fact *fact)
{
sw r;
sw result = 0;
if (fact) {
if ((r = buf_write_1(buf, "{")) < 0)
return r;
result += r;
if ((r = buf_inspect_tag(buf, fact->subject)) < 0)
return r;
result += r;
if ((r = buf_write_1(buf, ", ")) < 0)
return r;
result += r;
if ((r = buf_inspect_tag(buf, fact->predicate)) < 0)
return r;
result += r;
if ((r = buf_write_1(buf, ", ")) < 0)
return r;
result += r;
if ((r = buf_inspect_tag(buf, fact->object)) < 0)
return r;
result += r;
if ((r = buf_write_1(buf, "}")) < 0)
return r;
result += r;
}
return result;
}
sw buf_inspect_fact_size (const s_fact *fact)
{
sw r;
sw result = 0;
if (fact) {
r = strlen("{");
result += r;
r = buf_inspect_tag_size(fact->subject);
result += r;
r = strlen(", ");
result += r;
r = buf_inspect_tag_size(fact->predicate);
result += r;
r = strlen(", ");
result += r;
r = buf_inspect_tag_size(fact->subject);
result += r;
r = strlen("}");
result += r;
}
return result;
}
sw buf_inspect_facts_spec (s_buf *buf, p_facts_spec spec)
{
sw r;
sw result = 0;
if (spec) {
if ((r = buf_write_1(buf, "{")) < 0)
return r;
result += r;
if ((r = buf_inspect_tag(buf, spec[0])) < 0)
return r;
result += r;
while (spec[0] || spec[1]) {
spec++;
if ((r = buf_write_1(buf, ", ")) < 0)
return r;
result += r;
if ((r = buf_inspect_tag(buf, spec[0])) < 0)
return r;
result += r;
}
if ((r = buf_write_1(buf, ", ")) < 0)
return r;
result += r;
if ((r = buf_inspect_tag(buf, spec[1])) < 0)
return r;
result += r;
if ((r = buf_write_1(buf, "}")) < 0)
return r;
result += r;
}
return result;
}
sw buf_inspect_fn (s_buf *buf, const s_fn *fn)
{
const s_fn_clause *clause;
s_ident ident;
sw r;
sw result = 0;
assert(buf);
assert(fn);
if (fn->macro)
ident.sym = &g_sym_macro;
else
ident.sym = &g_sym_fn;
ident.module = fn->module;
if ((r = buf_inspect_ident(buf, &ident)) < 0)
return r;
result += r;
if ((r = buf_write_1(buf, " ")) < 0)
return r;
result += r;
clause = fn->clauses;
assert(clause);
if (clause->next_clause) {
if ((r = buf_write_1(buf, "{\n")) < 0)
return r;
result += r;
while (clause) {
if ((r = buf_write_1(buf, " ")) < 0)
return r;
result += r;
if ((r = buf_inspect_fn_clause(buf, clause)) < 0)
return r;
result += r;
if ((r = buf_write_1(buf, "\n")) < 0)
return r;
result += r;
clause = clause->next_clause;
}
if ((r = buf_write_1(buf, "}")) < 0)
return r;
result += r;
}
else {
if ((r = buf_inspect_fn_clause(buf, clause)) < 0)
return r;
result += r;
}
return result;
}
sw buf_inspect_fn_clause (s_buf *buf, const s_fn_clause *clause)
{
sw r;
sw result = 0;
assert(buf);
assert(clause);
if ((r = buf_inspect_fn_pattern(buf, clause->pattern)) < 0)
return r;
result += r;
if ((r = buf_write_1(buf, " ")) < 0)
return r;
result += r;
if ((r = buf_inspect_block(buf, &clause->algo)) < 0)
return r;
result += r;
return result;
}
sw buf_inspect_fn_clause_size (const s_fn_clause *clause)
{
sw r;
sw result = 0;
assert(clause);
if ((r = buf_inspect_fn_pattern_size(clause->pattern)) < 0)
return r;
result += r;
r = strlen(" ");
result += r;
if ((r = buf_inspect_block_size(&clause->algo)) < 0)
return r;
result += r;
return result;
}
sw buf_inspect_fn_pattern (s_buf *buf, const s_list *pattern)
{
sw r;
sw result = 0;
assert(buf);
if ((r = buf_write_u8(buf, '(')) <= 0)
return r;
result += r;
while (pattern) {
if ((r = buf_inspect_tag(buf, &pattern->tag)) < 0)
return r;
result += r;
pattern = list_next(pattern);
if (pattern) {
if ((r = buf_write_1(buf, ", ")) < 0)
return r;
result += r;
}
}
if ((r = buf_write_u8(buf, ')')) < 0)
return r;
result += r;
return result;
}
sw buf_inspect_fn_pattern_size (const s_list *pattern)
{
sw r;
sw result = 0;
r = strlen("(");
result += r;
while (pattern) {
if ((r = buf_inspect_tag_size(&pattern->tag)) < 0)
return r;
result += r;
pattern = list_next(pattern);
if (pattern) {
r = strlen(", ");
result += r;
}
}
r = strlen(")");
result += r;
return result;
}
sw buf_inspect_fn_size (const s_fn *fn)
{
const s_fn_clause *clause;
sw r;
sw result = 0;
assert(fn);
r = strlen("fn ");
result += r;
clause = fn->clauses;
assert(clause);
if (clause->next_clause) {
r = strlen("{\n");
result += r;
while (fn) {
r = strlen(" ");
result += r;
r = buf_inspect_fn_clause_size(clause);
result += r;
r = strlen("\n");
result += r;
clause = clause->next_clause;
}
r = strlen("}");
result += r;
}
else {
r = buf_inspect_fn_clause_size(clause);
result += r;
}
return result;
}
sw buf_inspect_ident (s_buf *buf, const s_ident *ident)
{
sw r;
sw result = 0;
assert(buf);
assert(ident);
result = 0;
// FIXME
if (ident->module && ident->module != g_kc3_env.current_defmodule) {
if ((r = buf_inspect_sym(buf, &ident->module)) < 0)
return r;
result += r;
if ((r = buf_write_1(buf, ".")) < 0)
return r;
result += r;
}
if ((r = buf_inspect_ident_sym(buf, ident->sym)) < 0)
return r;
result += r;
return result;
}
sw buf_inspect_ident_size (const s_ident *ident)
{
sw r;
sw result = 0;
assert(ident);
if (ident->module) {
if ((r = buf_inspect_sym_size(&ident->module)) < 0)
return r;
result += r;
result += strlen(".");
}
if ((r = buf_inspect_ident_sym_size(ident->sym)) < 0)
return r;
result += r;
return result;
}
sw buf_inspect_ident_sym (s_buf *buf, const s_sym *sym)
{
if (sym->str.size == 0)
return buf_write_1(buf, "_\"\"");
if (sym_has_ident_reserved_characters(sym))
return buf_inspect_ident_sym_reserved(buf, sym);
return buf_write_str(buf, &sym->str);
}
sw buf_inspect_ident_sym_reserved (s_buf *buf, const s_sym *sym)
{
sw size;
assert(buf);
assert(sym);
size = buf_inspect_ident_sym_reserved_size(sym);
if (buf->wpos + size > buf->size) {
assert(! "buffer overflow");
return -1;
}
buf_write_1(buf, "_");
buf_inspect_str(buf, &sym->str);
return size;
}
sw buf_inspect_ident_sym_reserved_size (const s_sym *sym)
{
sw size;
assert(sym);
size = strlen("_");
size += buf_inspect_str_size(&sym->str);
return size;
}
sw buf_inspect_ident_sym_size (const s_sym *sym)
{
if (sym->str.size == 0)
return strlen("_\"\"");
if (sym_has_ident_reserved_characters(sym))
return buf_inspect_ident_sym_reserved_size(sym);
return sym->str.size;
}
sw buf_inspect_integer (s_buf *buf, const s_integer *x)
{
s_buf buf_tmp;
mp_digit d;
sw result = 0;
size_t maxlen;
u8 p;
const mp_digit radix = 10;
s32 size = 0;
mp_int t;
if (mp_radix_size(&x->mp_int, radix, &size) != MP_OKAY)
return -1;
maxlen = size;
if (MP_IS_ZERO(&x->mp_int))
return buf_write_u8(buf, '0');
if (mp_init_copy(&t, &x->mp_int) != MP_OKAY)
return -1;
if (t.sign == MP_NEG) {
t.sign = MP_ZPOS;
maxlen--;
buf_write_u8(buf, '-');
result++;
}
buf_init_alloc(&buf_tmp, maxlen);
while (! MP_IS_ZERO(&t)) {
if (mp_div_d(&t, radix, &t, &d) != MP_OKAY)
goto error;
p = '0' + d;
if (p > '9')
goto error;
buf_write_u8(&buf_tmp, p);
result++;
}
buf_xfer_reverse(&buf_tmp, buf);
mp_clear(&t);
buf_clean(&buf_tmp);
return result;
error:
mp_clear(&t);
buf_clean(&buf_tmp);
return -1;
}
sw buf_inspect_integer_size (const s_integer *x)
{
int size;
if (mp_radix_size(&x->mp_int, 10, &size) != MP_OKAY)
return -1;
return size - 1;
}
sw buf_inspect_list (s_buf *buf, const s_list * const *x)
{
const s_list *i;
sw r;
sw result = 0;
assert(buf);
if ((r = buf_write_1(buf, "[")) <= 0)
return r;
result++;
i = *x;
while (i) {
if ((r = buf_inspect_list_tag(buf, &i->tag)) < 0)
return r;
result += r;
switch (i->next.type) {
case TAG_LIST:
if (i->next.data.list) {
if ((r = buf_write_1(buf, ", ")) < 0)
return r;
result += r;
}
i = i->next.data.list;
continue;
default:
if ((r = buf_write_1(buf, " | ")) < 0)
return r;
result += r;
if ((r = buf_inspect_tag(buf, &i->next)) < 0)
return r;
result += r;
i = NULL;
}
}
if ((r = buf_write_1(buf, "]")) < 0)
return r;
result += r;
return result;
}
sw buf_inspect_list_paren (s_buf *buf, const s_list * const *x)
{
const s_list *i;
sw r;
sw result = 0;
assert(buf);
if ((r = buf_write_1(buf, "(")) <= 0)
return r;
result++;
i = *x;
while (i) {
if ((r = buf_inspect_tag(buf, &i->tag)) < 0)
return r;
result += r;
switch (i->next.type) {
case TAG_LIST:
if (i->next.data.list) {
if ((r = buf_write_1(buf, ", ")) < 0)
return r;
result += r;
}
i = i->next.data.list;
continue;
default:
if ((r = buf_write_1(buf, " | ")) < 0)
return r;
result += r;
if ((r = buf_inspect_tag(buf, &i->next)) < 0)
return r;
result += r;
i = NULL;
}
}
if ((r = buf_write_1(buf, ")")) < 0)
return r;
result += r;
return result;
}
sw buf_inspect_list_size (const s_list * const *list)
{
const s_list *i;
sw r;
sw result = 0;
result += strlen("[");
i = *list;
while (i) {
if ((r = buf_inspect_list_tag_size(&i->tag)) < 0)
return r;
result += r;
switch (i->next.type) {
case TAG_LIST:
if (i->next.data.list)
result += strlen(", ");
i = i->next.data.list;
continue;
default:
result += strlen(" | ");
if ((r = buf_inspect_tag_size(&i->next)) < 0)
return r;
result += r;
break;
}
}
result += strlen("]");
return result;
}
sw buf_inspect_list_tag (s_buf *buf, const s_tag *tag)
{
sw r;
sw result = 0;
const s_sym *sym;
assert(buf);
assert(tag);
if (tag->type == TAG_TUPLE &&
tag->data.tuple.count == 2 &&
tag->data.tuple.tag[0].type == TAG_SYM) {
sym = tag->data.tuple.tag[0].data.sym;
if (sym_has_reserved_characters(sym)) {
if ((r = buf_inspect_str(buf, &sym->str)) < 0)
return r;
}
else
if ((r = buf_write_str(buf, &sym->str)) < 0)
return r;
result += r;
if ((r = buf_write_1(buf, ": ")) < 0)
return r;
result += r;
if ((r = buf_inspect_tag(buf, tag->data.tuple.tag + 1)) < 0)
return r;
result += r;
return result;
}
return buf_inspect_tag(buf, tag);
}
sw buf_inspect_list_tag_size (const s_tag *tag)
{
sw r;
sw result = 0;
const s_sym *sym;
assert(tag);
if (tag->type == TAG_TUPLE &&
tag->data.tuple.count == 2 &&
tag->data.tuple.tag[0].type == TAG_SYM) {
sym = tag->data.tuple.tag[0].data.sym;
if (sym_has_reserved_characters(sym)) {
if ((r = buf_inspect_str_size(&sym->str)) < 0)
return r;
}
else
r = sym->str.size;
result += r;
result += strlen(": ");
if ((r = buf_inspect_tag_size(tag->data.tuple.tag + 1)) < 0)
return r;
result += r;
return result;
}
return buf_inspect_tag_size(tag);
}
sw buf_inspect_map (s_buf *buf, const s_map *map)
{
uw i = 0;
s_tag *k;
sw r;
sw result = 0;
assert(buf);
assert(map);
if ((r = buf_write_1(buf, "%{")) < 0)
return r;
result += r;
while (i < map->count) {
k = map->key + i;
if (k->type == TAG_SYM) {
if (sym_has_reserved_characters(k->data.sym)) {
if ((r = buf_write_str(buf, &k->data.sym->str)) < 0)
return r;
}
else
if ((r = buf_write_1(buf, k->data.sym->str.ptr.pchar)) < 0)
return r;
result += r;
if ((r = buf_write_1(buf, ": ")) < 0)
return r;
result += r;
}
else {
if ((r = buf_inspect_tag(buf, map->key + i)) < 0)
return r;
result += r;
if ((r = buf_write_1(buf, " => ")) < 0)
return r;
result += r;
}
if ((r = buf_inspect_tag(buf, map->value + i)) < 0)
return r;
result += r;
i++;
if (i < map->count) {
if ((r = buf_write_1(buf, ", ")) < 0)
return r;
result += r;
}
}
if ((r = buf_write_1(buf, "}")) < 0)
return r;
result += r;
return result;
}
sw buf_inspect_map_size (const s_map *map)
{
assert(map);
assert(! "buf_inspect_map_size: not implemented");
(void) map;
return -1;
}
sw buf_inspect_paren_sym (s_buf *buf, const s_sym *sym)
{
sw r;
sw result = 0;
assert(buf);
assert(sym);
if ((r = buf_write_1(buf, "(")) <= 0)
goto clean;
result += r;
if (sym_is_array_type(sym)) {
if ((r = buf_write_str(buf, &sym->str)) <= 0)
goto clean;
result += r;
}
else {
if ((r = buf_inspect_sym(buf, &sym)) <= 0)
goto clean;
result += r;
}
if ((r = buf_write_1(buf, ")")) <= 0)
goto clean;
result += r;
r = result;
clean:
return r;
}
sw buf_inspect_paren_sym_size (const s_sym *sym)
{
sw r;
sw result = 0;
assert(sym);
result += strlen("(");
if ((r = buf_inspect_sym_size(&sym)) <= 0)
goto clean;
result += r;
result += strlen(")");
r = result;
clean:
return r;
}
sw buf_inspect_pointer (s_buf *buf, const void *ptr)
{
sw r;
sw result = 0;
assert(buf);
if ((r = buf_write_1(buf, "0x")) < 0)
return r;
result += r;
if ((r = buf_inspect_uw_hexadecimal(buf, (uw *) &ptr)) < 0)
return r;
result += r;
return result;
}
sw buf_inspect_ptag (s_buf *buf, const p_tag *ptag)
{
sw r;
sw result = 0;
assert(buf);
if ((r = buf_write_1(buf, "@0x")) < 0)
return r;
result += r;
if ((r = buf_inspect_uw_hexadecimal(buf, (uw *) ptag)) < 0)
return r;
result += r;
return result;
}
sw buf_inspect_ptag_size (const p_tag *ptag)
{
sw result = 0;
(void) ptag;
result += strlen("@0x");
result += sizeof(uw) / 4;
return result;
}
sw buf_inspect_ptr (s_buf *buf, const u_ptr_w *ptr)
{
sw r;
sw result = 0;
assert(buf);
if ((r = buf_write_1(buf, "(Ptr) 0x")) < 0)
return r;
result += r;
if ((r = buf_inspect_uw_hexadecimal(buf, (uw *) &ptr->p)) < 0)
return r;
result += r;
return result;
}
sw buf_inspect_ptr_free (s_buf *buf, const u_ptr_w *ptr_free)
{
sw r;
sw result = 0;
assert(buf);
if ((r = buf_write_1(buf, "(PtrFree) 0x")) < 0)
return r;
result += r;
r = buf_inspect_uw_hexadecimal(buf, (uw *) &ptr_free->p);
if (r < 0)
return r;
result += r;
return result;
}
sw buf_inspect_ptr_free_size (const u_ptr_w *ptr)
{
sw result;
(void) ptr;
result = strlen("(PtrFree) 0x");
result += sizeof(uw) / 4;
return result;
}
sw buf_inspect_ptr_size (const u_ptr_w *ptr)
{
sw result;
(void) ptr;
result = strlen("(Ptr) 0x");
result += sizeof(uw) / 4;
return result;
}
sw buf_inspect_quote (s_buf *buf, const s_quote *quote)
{
sw r;
sw result = 0;
s_buf_save save;
buf_save_init(buf, &save);
if ((r = buf_write_1(buf, "quote ")) < 0)
goto clean;
result += r;
if ((r = buf_inspect_tag(buf, quote->tag)) < 0)
goto restore;
r = result;
goto clean;
restore:
buf_save_restore_wpos(buf, &save);
clean:
buf_save_clean(buf, &save);
return r;
}
sw buf_inspect_quote_size (const s_quote *quote)
{
sw r;
sw result = 0;
result += strlen("quote ");
if ((r = buf_inspect_tag_size(quote->tag)) < 0)
return r;
result += r;
return result;
}
sw buf_inspect_ratio (s_buf *buf, const s_ratio *ratio)
{
sw r;
sw result = 0;
s_buf_save save;
buf_save_init(buf, &save);
if ((r = buf_inspect_integer(buf, &ratio->numerator)) < 0)
goto restore;
result += r;
if ((r = buf_write_1(buf, "/")) < 0)
goto restore;
result += r;
if ((r = buf_inspect_integer(buf, &ratio->denominator)) < 0)
goto restore;
result += r;
r = result;
goto clean;
restore:
buf_save_restore_wpos(buf, &save);
clean:
buf_save_clean(buf, &save);
return r;
}
sw buf_inspect_ratio_size (const s_ratio *ratio)
{
sw r;
sw result = 0;
if ((r = buf_inspect_integer_size(&ratio->numerator)) < 0)
return r;
result += r;
result += strlen("/");
if ((r = buf_inspect_integer_size(&ratio->denominator)) < 0)
return r;
result += r;
return result;
}
sw buf_inspect_str (s_buf *buf, const s_str *str)
{
sw r;
sw result = 0;
s_buf_save save;
assert(buf);
assert(str);
if (str_has_reserved_characters(str))
return buf_inspect_str_reserved(buf, str);
buf_save_init(buf, &save);
if ((r = buf_write_u8(buf, '"')) <= 0)
goto clean;
result += r;
if ((r = buf_write_str(buf, str)) < 0)
goto restore;
result += r;
if ((r = buf_write_u8(buf, '"')) <= 0)
goto restore;
result += r;
r = result;
goto clean;
restore:
buf_save_restore_wpos(buf, &save);
clean:
buf_save_clean(buf, &save);
return r;
}
sw buf_inspect_str_byte (s_buf *buf, const u8 *byte)
{
sw r;
sw result = 0;
s_buf_save save;
buf_save_init(buf, &save);
if ((r = buf_write_1(buf, "\\")) <= 0) {
r = -1;
goto clean;
}
result += r;
if ((r = buf_write_1(buf, "x")) <= 0)
goto restore;
result += r;
if ((r = buf_u8_to_hex(buf, byte)) != 2)
goto restore;
result += r;
r = result;
goto clean;
restore:
if (r >= 0)
r = -1;
buf_save_restore_wpos(buf, &save);
clean:
buf_save_clean(buf, &save);
return r;
}
sw buf_inspect_str_byte_size (const u8 *byte)
{
sw r;
sw result = 0;
(void) byte;
r = strlen("\\");
result += r;
r = strlen("x");
result += r;
r = 2;
result += r;
return result;
}
sw buf_inspect_str_character (s_buf *buf, const character *c)
{
char b[4];
s_buf char_buf;
int i;
int j;
sw r;
sw result = 0;
sw result1 = 0;
s_buf_save save;
if (! str_character_is_reserved(*c))
return buf_write_character_utf8(buf, *c);
buf_save_init(buf, &save);
if ((r = buf_write_1(buf, "\\")) <= 0)
goto restore;
result += r;
switch (*c) {
case '\0': if ((r = buf_write_u8(buf, '0')) <= 0) goto restore; break;
case '\n': if ((r = buf_write_u8(buf, 'n')) <= 0) goto restore; break;
case '\r': if ((r = buf_write_u8(buf, 'r')) <= 0) goto restore; break;
case '\t': if ((r = buf_write_u8(buf, 't')) <= 0) goto restore; break;
case '\v': if ((r = buf_write_u8(buf, 'v')) <= 0) goto restore; break;
case '\"': if ((r = buf_write_u8(buf, '"')) <= 0) goto restore; break;
case '\'': if ((r = buf_write_u8(buf, '\'')) <= 0) goto restore; break;
case '\\': if ((r = buf_write_u8(buf, '\\')) <= 0) goto restore; break;
default:
buf_init(&char_buf, false, sizeof(b), b);
if ((r = buf_write_character_utf8(&char_buf, *c)) <= 0)
goto restore;
i = r - 1;
j = 0;
if ((r = buf_write_u8(buf, 'x')) != 1)
goto restore;
result1 += r;
if ((r = buf_u8_to_hex(buf, &char_buf.ptr.pu8[j++])) != 2)
goto restore;
result1 += r;
while (i--) {
if ((r = buf_write_1(buf, "\\x")) != 2)
goto restore;
result1 += r;
if ((r = buf_u8_to_hex(buf, &char_buf.ptr.pu8[j++])) != 2)
goto restore;
result1 += r;
}
r = result1;
}
result += r;
r = result;
goto clean;
restore:
if (r >= 0)
r = -1;
buf_save_restore_wpos(buf, &save);
clean:
buf_save_clean(buf, &save);
return r;
}
sw buf_inspect_str_character_size (const character *c)
{
sw csize;
sw size;
if (! str_character_is_reserved(*c))
return character_utf8_size(*c);
size = 0;
switch (*c) {
case '\0':
case '\n':
case '\r':
case '\t':
case '\v':
case '\"':
case '\'':
case '\\':
size += 2;
break;
default:
csize = character_utf8_size(*c);
if (csize <= 0)
return -1;
size += csize * 4;
}
return size;
}
/* keep in sync with buf_inspect_str_reserved_size */
sw buf_inspect_str_reserved (s_buf *buf, const s_str *str)
{
u8 byte;
character c;
sw r;
sw result = 0;
s_str s;
s_buf_save save;
buf_save_init(buf, &save);
if ((r = buf_write_1(buf, "\"")) <= 0) {
if (! r)
r = -1;
goto clean;
}
result += r;
s = *str;
while (r) {
if ((r = str_read_character_utf8(&s, &c)) < 0)
goto restore;
if (r) {
if ((r = buf_inspect_str_character(buf, &c)) <= 0)
goto restore;
result += r;
}
else if ((r = str_read_u8(&s, &byte)) < 0)
goto restore;
else if (r) {
if ((r = buf_inspect_str_byte(buf, &byte)) <= 0)
goto restore;
result += r;
}
}
if ((r = buf_write_u8(buf, '"')) <= 0)
goto restore;
result += r;
r = result;
goto clean;
restore:
if (! r)
r = -1;
buf_save_restore_wpos(buf, &save);
clean:
buf_save_clean(buf, &save);
return r;
}
/* keep in sync with buf_inspect_str_reserved */
sw buf_inspect_str_reserved_size (const s_str *str)
{
u8 byte;
character c;
sw r;
sw result = 0;
s_str s;
r = strlen("\"");
result += r;
s = *str;
while (r) {
if ((r = str_read_character_utf8(&s, &c)) < 0)
goto restore;
if (r) {
if ((r = buf_inspect_str_character_size(&c)) <= 0)
goto restore;
result += r;
}
else if ((r = str_read_u8(&s, &byte)) < 0)
goto restore;
else if (r) {
r = buf_inspect_str_byte_size(&byte);
result += r;
}
}
r = strlen("\"");
result += r;
return result;
restore:
return -1;
}
sw buf_inspect_str_size (const s_str *str)
{
const sw quote_size = strlen("\"");
sw size;
if (str_has_reserved_characters(str))
return buf_inspect_str_reserved_size(str);
size = str->size + 2 * quote_size;
return size;
}
sw buf_inspect_struct (s_buf *buf, const s_struct *s)
{
uw i = 0;
s_tag *k;
sw r;
sw result = 0;
const s_sym *type;
assert(buf);
assert(s);
assert(sym_is_module(s->type->module));
if ((r = buf_write_1(buf, "%")) < 0)
return r;
result += r;
if (! sym_is_module(s->type->module))
return -1;
if (sym_has_reserved_characters(s->type->module)) {
if ((r = buf_write_str(buf, &s->type->module->str)) < 0)
return r;
}
else
if ((r = buf_write_1(buf, s->type->module->str.ptr.pchar)) < 0)
return r;
result += r;
if ((r = buf_write_1(buf, "{")) < 0)
return r;
result += r;
if (s->data || s->tag) {
while (i < s->type->map.count) {
k = s->type->map.key + i;
if (k->type != TAG_SYM) {
err_write_1("buf_inspect_struct: key type is not a symbol: ");
err_inspect_tag(k);
err_write_1(" (");
err_write_1(tag_type_to_string(k->type));
err_puts(")");
assert(k->type == TAG_SYM);
return -1;
}
if (sym_has_reserved_characters(k->data.sym)) {
if ((r = buf_write_str(buf, &k->data.sym->str)) < 0)
return r;
}
else
if ((r = buf_write_1(buf, k->data.sym->str.ptr.pchar)) < 0)
return r;
result += r;
if ((r = buf_write_1(buf, ": ")) < 0)
return r;
result += r;
if (s->data) {
if (! tag_type(s->type->map.value + i, &type))
return -1;
assert(s->type->offset[i] < s->type->size);
if ((r = data_buf_inspect(type, buf, (char *) s->data +
s->type->offset[i])) < 0)
return r;
result += r;
}
else if (s->tag) {
if ((r = buf_inspect_tag(buf, s->tag + i)) < 0)
return r;
result += r;
}
i++;
if (i < s->type->map.count) {
if ((r = buf_write_1(buf, ", ")) < 0)
return r;
result += r;
}
}
}
if ((r = buf_write_1(buf, "}")) < 0)
return r;
result += r;
return result;
}
sw buf_inspect_struct_size (const s_struct *s)
{
assert(s);
assert(! "buf_inspect_struct_size: not implemented");
(void) s;
return -1;
}
sw buf_inspect_struct_type (s_buf *buf, const s_struct_type *st)
{
s_array offset_array;
uw offset_array_dimension;
sw r;
sw result = 0;
assert(buf);
assert(st);
assert(sym_is_module(st->module));
assert(st->offset);
assert(st->size);
if ((r = buf_write_1(buf, "%StructType{module: ")) < 0)
return r;
result += r;
if ((r = buf_inspect_sym(buf, &st->module)) < 0)
return r;
result += r;
if ((r = buf_write_1(buf, ", map: ")) < 0)
return r;
result += r;
if ((r = buf_inspect_map(buf, &st->map)) < 0)
return r;
result += r;
if ((r = buf_write_1(buf, ", offset: ")) < 0)
return r;
result += r;
offset_array_dimension = st->map.count;
array_init(&offset_array, sym_1("Uw[]"), 1, &offset_array_dimension);
offset_array.data = st->offset;
if ((r = buf_inspect_array(buf, &offset_array)) < 0) {
array_clean(&offset_array);
return r;
}
result += r;
array_clean(&offset_array);
if ((r = buf_write_1(buf, ", size: ")) < 0)
return r;
result += r;
if ((r = buf_inspect_uw(buf, &st->size)) < 0)
return r;
result += r;
if ((r = buf_write_1(buf, "}")) < 0)
return r;
result += r;
return result;
}
sw buf_inspect_struct_type_size (const s_struct_type *st)
{
s_array offset_array;
uw offset_array_dimension;
sw r;
sw result;
assert(st);
assert(sym_is_module(st->module));
assert(st->offset);
assert(st->size);
result = strlen("%StructType{module: ");
if ((r = buf_inspect_sym_size(&st->module)) < 0)
return r;
result += r;
result += strlen(", map: ");
if ((r = buf_inspect_map_size(&st->map)) < 0)
return r;
result += r;
result += strlen(", offset: ");
offset_array_dimension = st->map.count;
array_init(&offset_array, &g_sym_Uw, 1, &offset_array_dimension);
offset_array.data = st->offset;
if ((r = buf_inspect_array_size(&offset_array)) < 0) {
array_clean(&offset_array);
return r;
}
result += r;
array_clean(&offset_array);
result += strlen(", size: ");
if ((r = buf_inspect_uw_size(&st->size)) < 0)
return r;
result += r;
result += strlen("}");
return result;
}
sw buf_inspect_sym (s_buf *buf, const s_sym * const *sym)
{
sw r;
sw size;
const s_sym *x;
assert(buf);
assert(sym);
x = *sym;
assert(x);
if (x->str.size == 0)
return buf_write_1(buf, ":\"\"");
if (sym_has_reserved_characters(x))
return buf_inspect_sym_reserved(buf, x);
if (sym_is_module(x) || sym_is_array_type(x))
return buf_write_str(buf, &x->str);
size = x->str.size + 1;
if ((r = buf_write_1(buf, ":")) < 0 ||
(r = buf_write_str(buf, &x->str)) < 0)
return r;
return size;
}
sw buf_inspect_sym_size (const s_sym * const *sym)
{
const sw colon_size = 1;
const s_sym *x;
assert(sym);
x = *sym;
assert(x);
if (x->str.size == 0)
return 3;
if (sym_has_reserved_characters(x))
return buf_inspect_sym_reserved_size(x);
if (sym_is_module(x) || sym_is_array_type(x))
return x->str.size;
return x->str.size + colon_size;
}
sw buf_inspect_sym_reserved (s_buf *buf, const s_sym *x)
{
sw r;
sw result = 0;
if ((r = buf_write_1(buf, ":")) < 0)
return r;
result += r;
if ((r = buf_inspect_str(buf, &x->str)) < 0)
return r;
result += r;
return result;
}
sw buf_inspect_sym_reserved_size (const s_sym *x)
{
sw r;
sw result;
result = strlen(":");
if ((r = buf_inspect_str_size(&x->str)) < 0)
return r;
result += r;
return result;
}
sw buf_inspect_tag (s_buf *buf, const s_tag *tag)
{
if (! tag)
return buf_write_1(buf, "NULL");
switch(tag->type) {
case TAG_ARRAY: return buf_inspect_array(buf, &tag->data.array);
case TAG_BLOCK: return buf_inspect_block(buf, &tag->data.block);
case TAG_BOOL: return buf_inspect_bool(buf, &tag->data.bool);
case TAG_CALL: return buf_inspect_call(buf, &tag->data.call);
case TAG_CFN: return buf_inspect_cfn(buf, &tag->data.cfn);
case TAG_CHARACTER:
return buf_inspect_character(buf, &tag->data.character);
case TAG_COMPLEX: return buf_inspect_complex(buf, tag->data.complex);
case TAG_COW: return buf_inspect_cow(buf, tag->data.cow);
case TAG_F32: return buf_inspect_f32(buf, &tag->data.f32);
case TAG_F64: return buf_inspect_f64(buf, &tag->data.f64);
case TAG_F128: return buf_inspect_f128(buf, &tag->data.f128);
case TAG_FACT: return buf_inspect_fact(buf, &tag->data.fact);
case TAG_FN: return buf_inspect_fn(buf, &tag->data.fn);
case TAG_IDENT: return buf_inspect_ident(buf, &tag->data.ident);
case TAG_INTEGER: return buf_inspect_integer(buf, &tag->data.integer);
case TAG_LIST:
return buf_inspect_list(buf, (const s_list **) &tag->data.list);
case TAG_MAP: return buf_inspect_map(buf, &tag->data.map);
case TAG_PTAG: return buf_inspect_ptag(buf, &tag->data.ptag);
case TAG_PTR: return buf_inspect_ptr(buf, &tag->data.ptr);
case TAG_PTR_FREE:
return buf_inspect_ptr_free(buf, &tag->data.ptr_free);
case TAG_QUOTE: return buf_inspect_quote(buf, &tag->data.quote);
case TAG_RATIO: return buf_inspect_ratio(buf, &tag->data.ratio);
case TAG_S8: return buf_inspect_s8(buf, &tag->data.s8);
case TAG_S16: return buf_inspect_s16(buf, &tag->data.s16);
case TAG_S32: return buf_inspect_s32(buf, &tag->data.s32);
case TAG_S64: return buf_inspect_s64(buf, &tag->data.s64);
case TAG_SW: return buf_inspect_sw(buf, &tag->data.sw);
case TAG_STR: return buf_inspect_str(buf, &tag->data.str);
case TAG_STRUCT: return buf_inspect_struct(buf, &tag->data.struct_);
case TAG_STRUCT_TYPE:
return buf_inspect_struct_type(buf, &tag->data.struct_type);
case TAG_SYM: return buf_inspect_sym(buf, &tag->data.sym);
case TAG_TUPLE: return buf_inspect_tuple(buf, &tag->data.tuple);
case TAG_U8: return buf_inspect_u8(buf, &tag->data.u8);
case TAG_U16: return buf_inspect_u16(buf, &tag->data.u16);
case TAG_U32: return buf_inspect_u32(buf, &tag->data.u32);
case TAG_U64: return buf_inspect_u64(buf, &tag->data.u64);
case TAG_UNQUOTE: return buf_inspect_unquote(buf, &tag->data.unquote);
case TAG_UW: return buf_inspect_uw(buf, &tag->data.uw);
case TAG_VAR: return buf_inspect_var(buf, tag);
case TAG_VOID: return buf_inspect_void(buf, NULL);
}
err_puts("buf_inspect_tag: unknown tag_type");
assert(! "buf_inspect_tag: unknown tag type");
return -1;
}
sw buf_inspect_tag_size (const s_tag *tag)
{
assert(tag);
switch(tag->type) {
case TAG_ARRAY: return buf_inspect_array_size(&tag->data.array);
case TAG_BLOCK: return buf_inspect_block_size(&tag->data.block);
case TAG_BOOL: return buf_inspect_bool_size(&tag->data.bool);
case TAG_CALL: return buf_inspect_call_size(&tag->data.call);
case TAG_CFN: return buf_inspect_cfn_size(&tag->data.cfn);
case TAG_CHARACTER:
return buf_inspect_character_size(&tag->data.character);
case TAG_COMPLEX: return buf_inspect_complex_size(tag->data.complex);
case TAG_COW: return buf_inspect_cow_size(tag->data.cow);
case TAG_F32: return buf_inspect_f32_size(&tag->data.f32);
case TAG_F64: return buf_inspect_f64_size(&tag->data.f64);
case TAG_F128: return buf_inspect_f128_size(&tag->data.f128);
case TAG_FACT: return buf_inspect_fact_size(&tag->data.fact);
case TAG_FN: return buf_inspect_fn_size(&tag->data.fn);
case TAG_IDENT: return buf_inspect_ident_size(&tag->data.ident);
case TAG_INTEGER:
return buf_inspect_integer_size(&tag->data.integer);
case TAG_LIST:
return buf_inspect_list_size((const s_list **) &tag->data.list);
case TAG_MAP: return buf_inspect_map_size(&tag->data.map);
case TAG_PTAG: return buf_inspect_ptag_size(&tag->data.ptag);
case TAG_PTR: return buf_inspect_ptr_size(&tag->data.ptr);
case TAG_PTR_FREE:
return buf_inspect_ptr_free_size(&tag->data.ptr_free);
case TAG_QUOTE: return buf_inspect_quote_size(&tag->data.quote);
case TAG_RATIO: return buf_inspect_ratio_size(&tag->data.ratio);
case TAG_S8: return buf_inspect_s8_size(&tag->data.s8);
case TAG_S16: return buf_inspect_s16_size(&tag->data.s16);
case TAG_S32: return buf_inspect_s32_size(&tag->data.s32);
case TAG_S64: return buf_inspect_s64_size(&tag->data.s64);
case TAG_SW: return buf_inspect_sw_size(&tag->data.sw);
case TAG_STR: return buf_inspect_str_size(&tag->data.str);
case TAG_STRUCT: return buf_inspect_struct_size(&tag->data.struct_);
case TAG_STRUCT_TYPE:
return buf_inspect_struct_type_size(&tag->data.struct_type);
case TAG_SYM: return buf_inspect_sym_size(&tag->data.sym);
case TAG_TUPLE: return buf_inspect_tuple_size(&tag->data.tuple);
case TAG_U8: return buf_inspect_u8_size(&tag->data.u8);
case TAG_U16: return buf_inspect_u16_size(&tag->data.u16);
case TAG_U32: return buf_inspect_u32_size(&tag->data.u32);
case TAG_U64: return buf_inspect_u64_size(&tag->data.u64);
case TAG_UNQUOTE:
return buf_inspect_unquote_size(&tag->data.unquote);
case TAG_UW: return buf_inspect_uw_size(&tag->data.uw);
case TAG_VAR: return buf_inspect_var_size(tag);
case TAG_VOID: return buf_inspect_void_size(NULL);
}
err_puts("buf_inspect_tag_size: unknown tag type");
assert(! "buf_inspect_tag_size: unknown tag type");
return -1;
}
sw buf_inspect_tag_type (s_buf *buf, e_tag_type type)
{
const char *s;
assert(buf);
s = tag_type_to_string(type);
if (! s)
return -1;
return buf_write_1(buf, s);
}
sw buf_inspect_tag_type_size (e_tag_type type)
{
const char *s;
s = tag_type_to_string(type);
if (! s)
return -1;
return strlen(s);
}
sw buf_inspect_tuple (s_buf *buf, const s_tuple *tuple)
{
u64 i = 0;
sw r;
sw result = 0;
if ((r = buf_write_1(buf, "{")) < 0)
return r;
result += r;
if (tuple->count) {
while (i < tuple->count - 1) {
if ((r = buf_inspect_tag(buf, tuple->tag + i)) < 0)
return r;
result += r;
if ((r = buf_write_1(buf, ", ")) < 0)
return r;
result += r;
i++;
}
if ((r = buf_inspect_tag(buf, tuple->tag + i)) < 0)
return r;
result += r;
}
if ((r = buf_write_1(buf, "}")) < 0)
return r;
result += r;
return result;
}
sw buf_inspect_tuple_size (const s_tuple *tuple)
{
u64 i = 0;
sw r;
sw result;
result = strlen("{");
if (tuple->count) {
while (i < tuple->count - 1) {
if ((r = buf_inspect_tag_size(tuple->tag + i)) < 0)
return r;
result += r;
result += strlen(", ");
i++;
}
if ((r = buf_inspect_tag_size(tuple->tag + i)) < 0)
return r;
result += r;
}
result += strlen("}");
return result;
}
sw buf_inspect_unquote (s_buf *buf, const s_unquote *unquote)
{
sw r;
sw result = 0;
s_buf_save save;
buf_save_init(buf, &save);
if ((r = buf_write_1(buf, "unquote(")) < 0)
goto clean;
result += r;
if ((r = buf_inspect_tag(buf, unquote->tag)) < 0)
goto restore;
result += r;
if ((r = buf_write_1(buf, ")")) < 0)
goto clean;
result += r;
r = result;
goto clean;
restore:
buf_save_restore_wpos(buf, &save);
clean:
buf_save_clean(buf, &save);
return r;
}
sw buf_inspect_unquote_size (const s_unquote *unquote)
{
sw r;
sw result = 0;
result += strlen("unquote(");
if ((r = buf_inspect_tag_size(unquote->tag)) < 0)
return r;
result += r;
result += strlen(")");
return result;
}
sw buf_inspect_var (s_buf *buf, const s_tag *tag)
{
sw r;
sw result = 0;
assert(buf);
assert(tag);
assert(tag->type == TAG_VAR);
assert(tag->data.var.type);
if (tag->data.var.type == &g_sym_Tag)
return buf_write_1(buf, "?");
if ((r = buf_inspect_paren_sym(buf, tag->data.var.type)) < 0)
return r;
result += r;
if ((r = buf_write_1(buf, " ?")) < 0)
return r;
result += r;
return result;
}
sw buf_inspect_var_size (const s_tag *tag)
{
sw r;
sw result = 0;
if ((r = buf_inspect_paren_sym_size(tag->data.var.type)) < 0)
return r;
result += r;
result += strlen(" ?");
return result;
}
sw buf_inspect_void (s_buf *buf, const void *_)
{
sw r;
sw result = 0;
assert(buf);
(void) _;
if ((r = buf_write_1(buf, "void")) < 0)
return r;
result += r;
return result;
}
sw buf_inspect_void_size (const void *_)
{
(void) _;
return strlen("void");
}