]>
Commit | Line | Data |
---|---|---|
1d09f67e TL |
1 | /* |
2 | * Licensed to the Apache Software Foundation (ASF) under one or more | |
3 | * contributor license agreements. See the NOTICE file distributed with | |
4 | * this work for additional information regarding copyright ownership. | |
5 | * The ASF licenses this file to You under the Apache License, Version 2.0 | |
6 | * (the "License"); you may not use this file except in compliance with | |
7 | * the License. You may obtain a copy of the License at | |
8 | * | |
9 | * http://www.apache.org/licenses/LICENSE-2.0 | |
10 | * | |
11 | * Unless required by applicable law or agreed to in writing, software | |
12 | * distributed under the License is distributed on an "AS IS" BASIS, | |
13 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |
14 | * See the License for the specific language governing permissions and | |
15 | * limitations under the License. | |
16 | */ | |
17 | ||
18 | package org.apache.arrow.gandiva.evaluator; | |
19 | ||
20 | import java.util.Objects; | |
21 | ||
22 | /** | |
23 | * Used to construct gandiva configuration objects. | |
24 | */ | |
25 | public class ConfigurationBuilder { | |
26 | ||
27 | public long buildConfigInstance(ConfigOptions configOptions) { | |
28 | return buildConfigInstance(configOptions.optimize, configOptions.targetCPU); | |
29 | } | |
30 | ||
31 | private native long buildConfigInstance(boolean optimize, boolean detectHostCPU); | |
32 | ||
33 | public native void releaseConfigInstance(long configId); | |
34 | ||
35 | /** | |
36 | * ConfigOptions contains the configuration parameters to provide to gandiva. | |
37 | */ | |
38 | public static class ConfigOptions { | |
39 | private boolean optimize = true; | |
40 | private boolean targetCPU = true; | |
41 | ||
42 | public static ConfigOptions getDefault() { | |
43 | return new ConfigOptions(); | |
44 | } | |
45 | ||
46 | public ConfigOptions() {} | |
47 | ||
48 | public ConfigOptions withOptimize(boolean optimize) { | |
49 | this.optimize = optimize; | |
50 | return this; | |
51 | } | |
52 | ||
53 | public ConfigOptions withTargetCPU(boolean targetCPU) { | |
54 | this.targetCPU = targetCPU; | |
55 | return this; | |
56 | } | |
57 | ||
58 | @Override | |
59 | public int hashCode() { | |
60 | return Objects.hash(optimize, targetCPU); | |
61 | } | |
62 | ||
63 | @Override | |
64 | public boolean equals(Object obj) { | |
65 | if (!(obj instanceof ConfigOptions)) { | |
66 | return false; | |
67 | } | |
68 | return this.optimize == ((ConfigOptions) obj).optimize && | |
69 | this.targetCPU == ((ConfigOptions) obj).targetCPU; | |
70 | } | |
71 | } | |
72 | } |