annotate truffle/com.oracle.truffle.tck/src/com/oracle/truffle/tck/Schema.java @ 22495:aeba89e1d8da

Add ComplexNumber sequence tests
author Matthias Grimmer <grimmer@ssw.jku.at>
date Fri, 11 Dec 2015 15:20:27 +0100
parents
children b3569a53c24c
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
22495
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
1 /*
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
2 * Copyright (c) 2015, 2015, Oracle and/or its affiliates. All rights reserved.
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
4 *
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
5 * This code is free software; you can redistribute it and/or modify it
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
6 * under the terms of the GNU General Public License version 2 only, as
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
7 * published by the Free Software Foundation. Oracle designates this
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
8 * particular file as subject to the "Classpath" exception as provided
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
9 * by Oracle in the LICENSE file that accompanied this code.
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
10 *
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
11 * This code is distributed in the hope that it will be useful, but WITHOUT
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
12 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
13 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
14 * version 2 for more details (a copy is included in the LICENSE file that
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
15 * accompanied this code).
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
16 *
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
17 * You should have received a copy of the GNU General Public License version
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
18 * 2 along with this work; if not, write to the Free Software Foundation,
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
19 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
20 *
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
21 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
22 * or visit www.oracle.com if you need additional information or have any
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
23 * questions.
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
24 */
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
25 package com.oracle.truffle.tck;
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
26
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
27 import java.nio.ByteBuffer;
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
28 import java.util.HashMap;
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
29 import java.util.List;
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
30 import java.util.Map;
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
31
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
32 public final class Schema {
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
33
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
34 public enum Type {
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
35 DOUBLE(Double.SIZE / Byte.SIZE),
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
36 INT(Integer.SIZE / Byte.SIZE);
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
37
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
38 private final int size;
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
39
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
40 private Type(int size) {
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
41 this.size = size;
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
42 }
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
43 }
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
44
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
45 private final int size;
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
46 private final boolean rowBased;
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
47 private final List<String> names;
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
48 private final List<Type> types;
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
49
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
50 public Schema(int size, boolean rowBased, List<String> names, List<Type> types) {
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
51 this.size = size;
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
52 this.rowBased = rowBased;
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
53 this.names = names;
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
54 this.types = types;
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
55 }
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
56
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
57 public int length() {
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
58 return size;
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
59 }
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
60
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
61 // for simplicity: structured data is read-only
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
62
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
63 public Map<String, Object> getEntry(byte[] buffer, int index) {
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
64 Map<String, Object> entry = new HashMap<>();
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
65 for (int i = 0; i < names.size(); i++) {
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
66 String name = names.get(i);
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
67 Object value = get(buffer, index, name);
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
68 entry.put(name, value);
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
69 }
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
70 return entry;
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
71 }
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
72
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
73 private Object get(byte[] buffer, int index, String name) {
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
74 assert names.contains(name);
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
75 int offset = rowBased ? getRowOffset(name, index) : getColumnOffset(name, index);
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
76 if (types.get(names.indexOf(name)) == Type.DOUBLE) {
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
77 byte[] b = new byte[Type.DOUBLE.size];
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
78 for (int i = 0; i < Type.DOUBLE.size; i++) {
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
79 b[i] = buffer[offset + i];
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
80 }
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
81 return ByteBuffer.wrap(b).getDouble();
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
82 } else if (types.get(names.indexOf(name)) == Type.INT) {
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
83 byte[] b = new byte[Type.INT.size];
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
84 for (int i = 0; i < Type.INT.size; i++) {
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
85 b[i] = buffer[offset + i];
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
86 }
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
87 return ByteBuffer.wrap(b).getInt();
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
88 }
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
89 throw new IllegalStateException();
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
90 }
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
91
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
92 private int getRowSize() {
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
93 assert rowBased;
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
94 int rowSize = 0;
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
95 for (Type t : types) {
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
96 rowSize += t.size;
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
97 }
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
98 return rowSize;
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
99 }
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
100
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
101 private int getRowOffset(String name, int index) {
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
102 assert rowBased;
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
103 if (names.contains(name)) {
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
104 int offset = 0;
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
105 for (int i = 0; i < names.size(); i++) {
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
106 if (names.get(i).equals(name)) {
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
107 return index * getRowSize() + offset;
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
108 } else {
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
109 offset += types.get(i).size;
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
110 }
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
111 }
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
112 } else {
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
113 throw new IllegalArgumentException();
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
114 }
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
115 throw new IllegalStateException();
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
116 }
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
117
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
118 private int getColumnOffset(String name, int index) {
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
119 assert !rowBased;
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
120 if (names.contains(name)) {
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
121 int offset = 0;
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
122 for (int i = 0; i < names.size(); i++) {
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
123 if (names.get(i).equals(name)) {
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
124 return offset + index * types.get(i).size;
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
125 } else {
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
126 offset += types.get(i).size * size;
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
127 }
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
128 }
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
129 } else {
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
130 throw new IllegalArgumentException();
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
131 }
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
132 throw new IllegalStateException();
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
133 }
aeba89e1d8da Add ComplexNumber sequence tests
Matthias Grimmer <grimmer@ssw.jku.at>
parents:
diff changeset
134 }