From 3330e6ed47c71b87bb79c5e277214a36d3f2ad3b Mon Sep 17 00:00:00 2001 From: Peter Mikkelsen Date: Fri, 14 Jan 2022 13:09:26 +0000 Subject: Add code to simplify arrays of nested scalars, sometimes created by functions or operators --- array.c | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) (limited to 'array.c') diff --git a/array.c b/array.c index 4a14ab9..7675144 100644 --- a/array.c +++ b/array.c @@ -97,4 +97,23 @@ arrayitem(Array *a, int index) exits(nil); } return res; +} + +Array * +simplifyarray(Array *a) +{ + /* simplify an array if possible. */ + if(a->type != AtypeArray || a->size == 0) + return fnSame(a); + int type = a->arraydata[0]->type; + int i; + for(i = 0; i < a->size; i++) + if(a->arraydata[i]->type != type || a->arraydata[i]->rank != 0) + return fnSame(a); + Array *b = allocarray(type, a->rank, a->size); + for(i = 0; i < a->rank; i++) + b->shape[i] = a->shape[i]; + for(i = 0; i < a->size; i++) + memcpy(b->rawdata + i * datasizes[type], a->arraydata[i]->rawdata, datasizes[type]); + return b; } \ No newline at end of file -- cgit v1.2.3