1 /*
 2  * Copyright (c) 2024, Oracle and/or its affiliates. All rights reserved.
 3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
 4  *
 5  * This code is free software; you can redistribute it and/or modify it
 6  * under the terms of the GNU General Public License version 2 only, as
 7  * published by the Free Software Foundation.  Oracle designates this
 8  * particular file as subject to the "Classpath" exception as provided
 9  * by Oracle in the LICENSE file that accompanied this code.
10  *
11  * This code is distributed in the hope that it will be useful, but WITHOUT
12  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
13  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
14  * version 2 for more details (a copy is included in the LICENSE file that
15  * accompanied this code).
16  *
17  * You should have received a copy of the GNU General Public License version
18  * 2 along with this work; if not, write to the Free Software Foundation,
19  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
20  *
21  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
22  * or visit www.oracle.com if you need additional information or have any
23  * questions.
24  */
25 package hat.buffer;
26 
27 import hat.types.Float2;
28 import hat.types.Float4;
29 import optkl.ifacemapper.BoundSchema;
30 import optkl.util.carriers.ArenaAndLookupCarrier;
31 import optkl.ifacemapper.Buffer;
32 import optkl.ifacemapper.MappableIface;
33 import optkl.ifacemapper.Schema;
34 
35 import java.lang.foreign.MemorySegment;
36 
37 import static java.lang.foreign.ValueLayout.JAVA_FLOAT;
38 import static java.lang.foreign.ValueLayout.JAVA_INT;
39 
40 public interface F32ArrayPadded extends Buffer {
41     long PAD_SIZE = 12;
42     default int pad(int pad){return pad;}
43    // Ideally ? @Reflect default void  schema(){pad(12);array(length()+pad(12));}
44     //Schema<F32ArrayPadded> schema = Schema.of(F32ArrayPadded.class);
45     Schema<F32ArrayPadded> schema = Schema.of(F32ArrayPadded.class, $ -> $
46             .arrayLen("length").pad(PAD_SIZE).array("array"));
47     int length();
48     float array(long idx);
49     void array(long idx, float f);
50 
51     long ARRAY_OFFSET = JAVA_INT.byteSize()+PAD_SIZE;
52 
53     static F32ArrayPadded create(ArenaAndLookupCarrier cc, int length){
54         return BoundSchema.of(cc ,schema, length).allocate();
55     }
56 
57     default F32ArrayPadded copyFrom(float[] floats) {
58         MemorySegment.copy(floats, 0, MappableIface.getMemorySegment(this), JAVA_FLOAT, ARRAY_OFFSET, length());
59         return this;
60     }
61 
62     static F32ArrayPadded createFrom(ArenaAndLookupCarrier cc, float[] arr){
63         return create( cc, arr.length).copyFrom(arr);
64     }
65 
66     default F32ArrayPadded copyTo(float[] floats) {
67         MemorySegment.copy(MappableIface.getMemorySegment(this), JAVA_FLOAT, ARRAY_OFFSET, floats, 0, length());
68         return this;
69     }
70 
71     default Float4.MutableImpl[] float4ArrayView() {
72         return null;
73     }
74 
75     default Float4.MutableImpl float4View(int index) {
76         return  null;
77     }
78 
79     default void storeFloat4View(Float4 v, int index) {
80     }
81 
82     default Float2.MutableImpl float2View(int index) {
83         return  null;
84     }
85 
86     default void storeFloat2View(Float2 v, int index) {
87     }
88 
89 }