Add class transformers to convert functions to native

This commit is contained in:
lax1dude 2024-12-02 19:47:22 -08:00
parent bfd8b7b631
commit 530b6a5da5
3 changed files with 54 additions and 0 deletions

View File

@ -115,6 +115,7 @@ import org.teavm.backend.wasm.render.WasmBinaryWriter;
import org.teavm.backend.wasm.render.WasmCRenderer; import org.teavm.backend.wasm.render.WasmCRenderer;
import org.teavm.backend.wasm.render.WasmRenderer; import org.teavm.backend.wasm.render.WasmRenderer;
import org.teavm.backend.wasm.runtime.WasmSupport; import org.teavm.backend.wasm.runtime.WasmSupport;
import org.teavm.backend.wasm.transformation.BaseClassesTransformation;
import org.teavm.backend.wasm.transformation.IndirectCallTraceTransformation; import org.teavm.backend.wasm.transformation.IndirectCallTraceTransformation;
import org.teavm.backend.wasm.transformation.MemoryAccessTraceTransformation; import org.teavm.backend.wasm.transformation.MemoryAccessTraceTransformation;
import org.teavm.backend.wasm.transformation.WasiFileSystemProviderTransformer; import org.teavm.backend.wasm.transformation.WasiFileSystemProviderTransformer;
@ -241,6 +242,7 @@ public class WasmTarget implements TeaVMTarget, TeaVMWasmHost {
public List<ClassHolderTransformer> getTransformers() { public List<ClassHolderTransformer> getTransformers() {
List<ClassHolderTransformer> transformers = new ArrayList<>(); List<ClassHolderTransformer> transformers = new ArrayList<>();
transformers.add(new ClassPatch()); transformers.add(new ClassPatch());
transformers.add(new BaseClassesTransformation());
transformers.add(new WasmDependencyListener()); transformers.add(new WasmDependencyListener());
if (runtimeType == WasmRuntimeType.WASI) { if (runtimeType == WasmRuntimeType.WASI) {
transformers.add(new WasiSupportClassTransformer()); transformers.add(new WasiSupportClassTransformer());

View File

@ -0,0 +1,41 @@
/*
* Copyright 2024 Alexey Andreev.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.teavm.backend.wasm.transformation;
import org.teavm.model.ClassHolder;
import org.teavm.model.ClassHolderTransformer;
import org.teavm.model.ClassHolderTransformerContext;
import org.teavm.model.ElementModifier;
public class BaseClassesTransformation implements ClassHolderTransformer {
@Override
public void transformClass(ClassHolder cls, ClassHolderTransformerContext context) {
if (cls.getName().equals("java.lang.Integer") || cls.getName().equals("java.lang.Long")) {
for (var method : cls.getMethods()) {
switch (method.getName()) {
case "numberOfLeadingZeros":
case "numberOfTrailingZeros":
case "bitCount":
method.setProgram(null);
method.getModifiers().add(ElementModifier.NATIVE);
break;
}
}
}
}
}

View File

@ -121,6 +121,17 @@ public class BaseClassesTransformation implements ClassHolderTransformer {
var clear = cls.getMethod(new MethodDescriptor("clear", void.class)); var clear = cls.getMethod(new MethodDescriptor("clear", void.class));
clear.getModifiers().add(ElementModifier.NATIVE); clear.getModifiers().add(ElementModifier.NATIVE);
clear.setProgram(null); clear.setProgram(null);
} else if (cls.getName().equals("java.lang.Integer") || cls.getName().equals("java.lang.Long")) {
for (var method : cls.getMethods()) {
switch (method.getName()) {
case "numberOfLeadingZeros":
case "numberOfTrailingZeros":
case "bitCount":
method.setProgram(null);
method.getModifiers().add(ElementModifier.NATIVE);
break;
}
}
} }
} }