array.c


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
52
53
54
55

#include <u.h>
#include <libc.h>
#include <bio.h>

#include "apl9.h"

int datasizes[] = {
	[AtypeInt] = sizeof(vlong),
	[AtypeArray] = sizeof(Array *)
};

Array *
mkarray(arrayDataType t, int rank, int size)
{
	Array *a = malloc(sizeof(Array));
	a->rank = rank;
	a->type = t;
	a->size = size;
	a->stranded = 0;
	a->shape = malloc(sizeof(int) * rank);
	a->rawdata = malloc(datasizes[t] * size);
	a->type = t;
	return a;
}

Array *
mkscalarint(vlong i)
{
	Array *a = mkarray(AtypeInt, 0, 1);
	a->intdata[0] = i;

	return a;
}

Array *
duparray(Array *a)
{
	Array *b = mkarray(a->type, a->rank, a->size);
	memcpy(b->shape, a->shape, sizeof(int) * a->rank);
	memcpy(b->rawdata, a->rawdata, datasizes[a->type]*a->size);
	/* TODO duplicate recursivley */
	return b;
}

int
simplearray(Array *a)
{
	return a->type != AtypeArray;
}

int
simplescalar(Array *a)
{
	return simplearray(a) && a->rank == 0;
}