raw.runtime.truffle.ast.io.binary.BinaryWriterNode Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2023 RAW Labs S.A.
 *
 * Use of this software is governed by the Business Source License
 * included in the file licenses/BSL.txt.
 *
 * As of the Change Date specified in that file, in accordance with
 * the Business Source License, use of this software will be governed
 * by the Apache License, Version 2.0, included in the file
 * licenses/APL.txt.
 */

package raw.runtime.truffle.ast.io.binary;

import com.oracle.truffle.api.frame.VirtualFrame;
import com.oracle.truffle.api.nodes.DirectCallNode;
import com.oracle.truffle.api.nodes.NodeInfo;
import java.io.IOException;
import java.io.OutputStream;
import raw.runtime.truffle.ExpressionNode;
import raw.runtime.truffle.RawContext;
import raw.runtime.truffle.StatementNode;
import raw.runtime.truffle.ast.ProgramStatementNode;
import raw.runtime.truffle.runtime.exceptions.RawTruffleRuntimeException;

@NodeInfo(shortName = "Binary.Write")
public class BinaryWriterNode extends StatementNode {

  @Child private ExpressionNode binaryNode;

  @Child private DirectCallNode innerWriter;

  public BinaryWriterNode(ExpressionNode binaryNode, ProgramStatementNode innerWriter) {
    this.innerWriter = DirectCallNode.create(innerWriter.getCallTarget());
    this.binaryNode = binaryNode;
  }

  @Override
  public void executeVoid(VirtualFrame frame) {
    Object binaryObject = binaryNode.executeGeneric(frame);
    try (OutputStream os = RawContext.get(this).getOutput()) {
      innerWriter.call(binaryObject, os);
    } catch (IOException e) {
      throw new RawTruffleRuntimeException(e.getMessage());
    }
  }
}