-
Notifications
You must be signed in to change notification settings - Fork 11
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
74bf673
commit 80b2c3e
Showing
3 changed files
with
58 additions
and
22 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,29 @@ | ||
package io.substrait; | ||
|
||
public class SubstraitPlanLoader { | ||
enum PlanFileFormat { | ||
kBinary = 0, | ||
kJson = 1, | ||
kProtoText = 2, | ||
kText = 3, | ||
}; | ||
|
||
// Load a Substrait plan (in any format) from disk. | ||
// Returns error message on failure, otherwise returns nullptr and | ||
// updates the provided buffer. | ||
public native String loadSubstraitPlan(String filename); | ||
|
||
// Write a Substrait plan to disk in the specified format. | ||
public native String saveSubstraitPlan(Plan plan, String filename, PlanFileFormat format); | ||
|
||
static { | ||
System.load("/path/to/libMYSTUFF.so"); | ||
} | ||
|
||
public static void main(String [] args) { | ||
SubstraitPlanLoader loader = new SubstraitPlanLoader(); | ||
|
||
String result = loader.loadSubstraitPlan("filename.plan"); | ||
System.out.println("Result: " + result); | ||
} | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,34 +1,35 @@ | ||
# SPDX-License-Identifier: Apache-2.0 | ||
|
||
cmake_path(GET CMAKE_CURRENT_BINARY_DIR PARENT_PATH | ||
CMAKE_CURRENT_BINARY_PARENT_DIR) | ||
CMAKE_CURRENT_BINARY_PARENT_DIR) | ||
cmake_path(GET CMAKE_CURRENT_BINARY_PARENT_DIR PARENT_PATH | ||
CMAKE_CURRENT_BINARY_TOPLEVEL_DIR) | ||
CMAKE_CURRENT_BINARY_TOPLEVEL_DIR) | ||
|
||
set(CMAKE_RUNTIME_OUTPUT_DIRECTORY | ||
"${CMAKE_CURRENT_BINARY_TOPLEVEL_DIR}/${BUILD_SUBDIR_NAME}") | ||
"${CMAKE_CURRENT_BINARY_TOPLEVEL_DIR}/${BUILD_SUBDIR_NAME}") | ||
|
||
add_test_case( | ||
planloader_test | ||
SOURCES | ||
PlanLoaderTest.cpp | ||
EXTRA_LINK_LIBS | ||
planloader | ||
gtest | ||
gtest_main) | ||
planloader_test | ||
SOURCES | ||
PlanLoaderTest.cpp | ||
EXTRA_LINK_LIBS | ||
planloader | ||
gtest | ||
gtest_main) | ||
|
||
set(TEXTPLAN_SOURCE_DIR "${CMAKE_SOURCE_DIR}/src/substrait/textplan") | ||
|
||
add_custom_command( | ||
TARGET planloader_test | ||
POST_BUILD | ||
COMMAND ${CMAKE_COMMAND} -E echo "Copying unit test data.." | ||
COMMAND ${CMAKE_COMMAND} -E make_directory | ||
"${CMAKE_RUNTIME_OUTPUT_DIRECTORY}/tests/data" | ||
COMMAND | ||
${CMAKE_COMMAND} -E copy | ||
"${TEXTPLAN_SOURCE_DIR}/converter/data/q6_first_stage.json" | ||
"${CMAKE_RUNTIME_OUTPUT_DIRECTORY}/tests/data/q6_first_stage.json") | ||
TARGET planloader_test | ||
POST_BUILD | ||
COMMAND ${CMAKE_COMMAND} -E echo "Copying unit test data.." | ||
COMMAND ${CMAKE_COMMAND} -E make_directory | ||
"${CMAKE_RUNTIME_OUTPUT_DIRECTORY}/tests/data" | ||
COMMAND | ||
${CMAKE_COMMAND} -E copy | ||
"${TEXTPLAN_SOURCE_DIR}/converter/data/q6_first_stage.json" | ||
"${CMAKE_RUNTIME_OUTPUT_DIRECTORY}/tests/data/q6_first_stage.json") | ||
|
||
message( | ||
STATUS "test data will be here: ${CMAKE_RUNTIME_OUTPUT_DIRECTORY}/tests/data") | ||
STATUS "test data will be here: ${CMAKE_RUNTIME_OUTPUT_DIRECTORY}/tests/data" | ||
) |