Jason Sams | 69f0d31 | 2009-08-03 18:11:17 -0700 | [diff] [blame] | 1 | /* |
Stephen Hines | adeb809 | 2012-04-20 14:26:06 -0700 | [diff] [blame] | 2 | * Copyright (C) 2008-2012 The Android Open Source Project |
Jason Sams | 69f0d31 | 2009-08-03 18:11:17 -0700 | [diff] [blame] | 3 | * |
| 4 | * Licensed under the Apache License, Version 2.0 (the "License"); |
| 5 | * you may not use this file except in compliance with the License. |
| 6 | * You may obtain a copy of the License at |
| 7 | * |
| 8 | * http://www.apache.org/licenses/LICENSE-2.0 |
| 9 | * |
| 10 | * Unless required by applicable law or agreed to in writing, software |
| 11 | * distributed under the License is distributed on an "AS IS" BASIS, |
| 12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 13 | * See the License for the specific language governing permissions and |
| 14 | * limitations under the License. |
| 15 | */ |
| 16 | |
| 17 | package android.renderscript; |
| 18 | |
Jason Sams | 08a8158 | 2012-09-18 12:32:10 -0700 | [diff] [blame] | 19 | import android.util.SparseArray; |
| 20 | |
Stephen Hines | 9c9ad3f8c2 | 2012-05-07 15:34:29 -0700 | [diff] [blame] | 21 | /** |
Jason Sams | a23d4e7 | 2011-01-04 18:59:12 -0800 | [diff] [blame] | 22 | * |
Jason Sams | 69f0d31 | 2009-08-03 18:11:17 -0700 | [diff] [blame] | 23 | **/ |
| 24 | public class Script extends BaseObj { |
Jason Sams | 08a8158 | 2012-09-18 12:32:10 -0700 | [diff] [blame] | 25 | |
| 26 | /** |
| 27 | * KernelID is an identifier for a Script + root function pair. It is used |
| 28 | * as an identifier for ScriptGroup creation. |
| 29 | * |
| 30 | * This class should not be directly created. Instead use the method in the |
| 31 | * reflected or intrinsic code "getKernelID_funcname()". |
| 32 | * |
| 33 | */ |
| 34 | public static final class KernelID extends BaseObj { |
| 35 | Script mScript; |
| 36 | int mSlot; |
| 37 | int mSig; |
| 38 | KernelID(int id, RenderScript rs, Script s, int slot, int sig) { |
| 39 | super(id, rs); |
| 40 | mScript = s; |
| 41 | mSlot = slot; |
| 42 | mSig = sig; |
| 43 | } |
| 44 | } |
| 45 | |
| 46 | private final SparseArray<KernelID> mKIDs = new SparseArray<KernelID>(); |
| 47 | /** |
| 48 | * Only to be used by generated reflected classes. |
| 49 | * |
| 50 | * |
| 51 | * @param slot |
| 52 | * @param sig |
| 53 | * @param ein |
| 54 | * @param eout |
| 55 | * |
| 56 | * @return KernelID |
| 57 | */ |
| 58 | protected KernelID createKernelID(int slot, int sig, Element ein, Element eout) { |
| 59 | KernelID k = mKIDs.get(slot); |
| 60 | if (k != null) { |
| 61 | return k; |
| 62 | } |
| 63 | |
| 64 | int id = mRS.nScriptKernelIDCreate(getID(mRS), slot, sig); |
| 65 | if (id == 0) { |
| 66 | throw new RSDriverException("Failed to create KernelID"); |
| 67 | } |
| 68 | |
| 69 | k = new KernelID(id, mRS, this, slot, sig); |
| 70 | mKIDs.put(slot, k); |
| 71 | return k; |
| 72 | } |
| 73 | |
| 74 | /** |
| 75 | * FieldID is an identifier for a Script + exported field pair. It is used |
| 76 | * as an identifier for ScriptGroup creation. |
| 77 | * |
| 78 | * This class should not be directly created. Instead use the method in the |
| 79 | * reflected or intrinsic code "getFieldID_funcname()". |
| 80 | * |
| 81 | */ |
| 82 | public static final class FieldID extends BaseObj { |
| 83 | Script mScript; |
| 84 | int mSlot; |
| 85 | FieldID(int id, RenderScript rs, Script s, int slot) { |
| 86 | super(id, rs); |
| 87 | mScript = s; |
| 88 | mSlot = slot; |
| 89 | } |
| 90 | } |
| 91 | |
| 92 | private final SparseArray<FieldID> mFIDs = new SparseArray(); |
| 93 | /** |
| 94 | * Only to be used by generated reflected classes. |
| 95 | * |
| 96 | * @param slot |
| 97 | * @param e |
| 98 | * |
| 99 | * @return FieldID |
| 100 | */ |
| 101 | protected FieldID createFieldID(int slot, Element e) { |
| 102 | FieldID f = mFIDs.get(slot); |
| 103 | if (f != null) { |
| 104 | return f; |
| 105 | } |
| 106 | |
| 107 | int id = mRS.nScriptFieldIDCreate(getID(mRS), slot); |
| 108 | if (id == 0) { |
| 109 | throw new RSDriverException("Failed to create FieldID"); |
| 110 | } |
| 111 | |
| 112 | f = new FieldID(id, mRS, this, slot); |
| 113 | mFIDs.put(slot, f); |
| 114 | return f; |
| 115 | } |
| 116 | |
| 117 | |
Stephen Hines | 9c9ad3f8c2 | 2012-05-07 15:34:29 -0700 | [diff] [blame] | 118 | /** |
Jason Sams | 67e3d20 | 2011-01-09 13:49:01 -0800 | [diff] [blame] | 119 | * Only intended for use by generated reflected code. |
| 120 | * |
| 121 | * @param slot |
| 122 | */ |
Jason Sams | 2d71bc7 | 2010-03-26 16:06:43 -0700 | [diff] [blame] | 123 | protected void invoke(int slot) { |
Jason Sams | e07694b | 2012-04-03 15:36:36 -0700 | [diff] [blame] | 124 | mRS.nScriptInvoke(getID(mRS), slot); |
Jason Sams | 2d71bc7 | 2010-03-26 16:06:43 -0700 | [diff] [blame] | 125 | } |
| 126 | |
Stephen Hines | 9c9ad3f8c2 | 2012-05-07 15:34:29 -0700 | [diff] [blame] | 127 | /** |
Jason Sams | 67e3d20 | 2011-01-09 13:49:01 -0800 | [diff] [blame] | 128 | * Only intended for use by generated reflected code. |
| 129 | * |
| 130 | * @param slot |
| 131 | * @param v |
| 132 | */ |
Jason Sams | 96ed4cf | 2010-06-15 12:15:57 -0700 | [diff] [blame] | 133 | protected void invoke(int slot, FieldPacker v) { |
| 134 | if (v != null) { |
Jason Sams | e07694b | 2012-04-03 15:36:36 -0700 | [diff] [blame] | 135 | mRS.nScriptInvokeV(getID(mRS), slot, v.getData()); |
Jason Sams | 96ed4cf | 2010-06-15 12:15:57 -0700 | [diff] [blame] | 136 | } else { |
Jason Sams | e07694b | 2012-04-03 15:36:36 -0700 | [diff] [blame] | 137 | mRS.nScriptInvoke(getID(mRS), slot); |
Jason Sams | 96ed4cf | 2010-06-15 12:15:57 -0700 | [diff] [blame] | 138 | } |
Jason Sams | 4d33993 | 2010-05-11 14:03:58 -0700 | [diff] [blame] | 139 | } |
| 140 | |
Stephen Hines | 9c9ad3f8c2 | 2012-05-07 15:34:29 -0700 | [diff] [blame] | 141 | /** |
Jason Sams | 6e494d3 | 2011-04-27 16:33:11 -0700 | [diff] [blame] | 142 | * Only intended for use by generated reflected code. |
| 143 | * |
| 144 | * @param slot |
| 145 | * @param ain |
| 146 | * @param aout |
| 147 | * @param v |
| 148 | */ |
| 149 | protected void forEach(int slot, Allocation ain, Allocation aout, FieldPacker v) { |
| 150 | if (ain == null && aout == null) { |
| 151 | throw new RSIllegalArgumentException( |
| 152 | "At least one of ain or aout is required to be non-null."); |
| 153 | } |
| 154 | int in_id = 0; |
| 155 | if (ain != null) { |
Jason Sams | e07694b | 2012-04-03 15:36:36 -0700 | [diff] [blame] | 156 | in_id = ain.getID(mRS); |
Jason Sams | 6e494d3 | 2011-04-27 16:33:11 -0700 | [diff] [blame] | 157 | } |
| 158 | int out_id = 0; |
| 159 | if (aout != null) { |
Jason Sams | e07694b | 2012-04-03 15:36:36 -0700 | [diff] [blame] | 160 | out_id = aout.getID(mRS); |
Jason Sams | 6e494d3 | 2011-04-27 16:33:11 -0700 | [diff] [blame] | 161 | } |
| 162 | byte[] params = null; |
| 163 | if (v != null) { |
| 164 | params = v.getData(); |
| 165 | } |
Jason Sams | e07694b | 2012-04-03 15:36:36 -0700 | [diff] [blame] | 166 | mRS.nScriptForEach(getID(mRS), slot, in_id, out_id, params); |
Jason Sams | 6e494d3 | 2011-04-27 16:33:11 -0700 | [diff] [blame] | 167 | } |
| 168 | |
Jason Sams | 4d33993 | 2010-05-11 14:03:58 -0700 | [diff] [blame] | 169 | |
Jason Sams | 69f0d31 | 2009-08-03 18:11:17 -0700 | [diff] [blame] | 170 | Script(int id, RenderScript rs) { |
Alex Sakhartchouk | 0de9444 | 2010-08-11 14:41:28 -0700 | [diff] [blame] | 171 | super(id, rs); |
Jason Sams | 69f0d31 | 2009-08-03 18:11:17 -0700 | [diff] [blame] | 172 | } |
| 173 | |
Jason Sams | 67e3d20 | 2011-01-09 13:49:01 -0800 | [diff] [blame] | 174 | |
Stephen Hines | 9c9ad3f8c2 | 2012-05-07 15:34:29 -0700 | [diff] [blame] | 175 | /** |
Jason Sams | 67e3d20 | 2011-01-09 13:49:01 -0800 | [diff] [blame] | 176 | * Only intended for use by generated reflected code. |
| 177 | * |
| 178 | * @param va |
| 179 | * @param slot |
| 180 | */ |
Jason Sams | 69f0d31 | 2009-08-03 18:11:17 -0700 | [diff] [blame] | 181 | public void bindAllocation(Allocation va, int slot) { |
Jason Sams | 771bebb | 2009-12-07 12:40:12 -0800 | [diff] [blame] | 182 | mRS.validate(); |
Jason Sams | 4d33993 | 2010-05-11 14:03:58 -0700 | [diff] [blame] | 183 | if (va != null) { |
Jason Sams | e07694b | 2012-04-03 15:36:36 -0700 | [diff] [blame] | 184 | mRS.nScriptBindAllocation(getID(mRS), va.getID(mRS), slot); |
Jason Sams | 4d33993 | 2010-05-11 14:03:58 -0700 | [diff] [blame] | 185 | } else { |
Jason Sams | e07694b | 2012-04-03 15:36:36 -0700 | [diff] [blame] | 186 | mRS.nScriptBindAllocation(getID(mRS), 0, slot); |
Jason Sams | 4d33993 | 2010-05-11 14:03:58 -0700 | [diff] [blame] | 187 | } |
| 188 | } |
| 189 | |
Stephen Hines | 9c9ad3f8c2 | 2012-05-07 15:34:29 -0700 | [diff] [blame] | 190 | /** |
Jason Sams | 67e3d20 | 2011-01-09 13:49:01 -0800 | [diff] [blame] | 191 | * Only intended for use by generated reflected code. |
| 192 | * |
| 193 | * @param index |
| 194 | * @param v |
| 195 | */ |
Jason Sams | 4d33993 | 2010-05-11 14:03:58 -0700 | [diff] [blame] | 196 | public void setVar(int index, float v) { |
Jason Sams | e07694b | 2012-04-03 15:36:36 -0700 | [diff] [blame] | 197 | mRS.nScriptSetVarF(getID(mRS), index, v); |
Jason Sams | 4d33993 | 2010-05-11 14:03:58 -0700 | [diff] [blame] | 198 | } |
| 199 | |
Stephen Hines | 9c9ad3f8c2 | 2012-05-07 15:34:29 -0700 | [diff] [blame] | 200 | /** |
Jason Sams | 67e3d20 | 2011-01-09 13:49:01 -0800 | [diff] [blame] | 201 | * Only intended for use by generated reflected code. |
| 202 | * |
| 203 | * @param index |
| 204 | * @param v |
| 205 | */ |
Stephen Hines | ca54ec3 | 2010-09-20 17:20:30 -0700 | [diff] [blame] | 206 | public void setVar(int index, double v) { |
Jason Sams | e07694b | 2012-04-03 15:36:36 -0700 | [diff] [blame] | 207 | mRS.nScriptSetVarD(getID(mRS), index, v); |
Stephen Hines | ca54ec3 | 2010-09-20 17:20:30 -0700 | [diff] [blame] | 208 | } |
| 209 | |
Stephen Hines | 9c9ad3f8c2 | 2012-05-07 15:34:29 -0700 | [diff] [blame] | 210 | /** |
Jason Sams | 67e3d20 | 2011-01-09 13:49:01 -0800 | [diff] [blame] | 211 | * Only intended for use by generated reflected code. |
| 212 | * |
| 213 | * @param index |
| 214 | * @param v |
| 215 | */ |
Jason Sams | 4d33993 | 2010-05-11 14:03:58 -0700 | [diff] [blame] | 216 | public void setVar(int index, int v) { |
Jason Sams | e07694b | 2012-04-03 15:36:36 -0700 | [diff] [blame] | 217 | mRS.nScriptSetVarI(getID(mRS), index, v); |
Jason Sams | 4d33993 | 2010-05-11 14:03:58 -0700 | [diff] [blame] | 218 | } |
| 219 | |
Stephen Hines | 9c9ad3f8c2 | 2012-05-07 15:34:29 -0700 | [diff] [blame] | 220 | /** |
Jason Sams | 67e3d20 | 2011-01-09 13:49:01 -0800 | [diff] [blame] | 221 | * Only intended for use by generated reflected code. |
| 222 | * |
| 223 | * @param index |
| 224 | * @param v |
| 225 | */ |
Stephen Hines | 031ec58c | 2010-10-11 10:54:21 -0700 | [diff] [blame] | 226 | public void setVar(int index, long v) { |
Jason Sams | e07694b | 2012-04-03 15:36:36 -0700 | [diff] [blame] | 227 | mRS.nScriptSetVarJ(getID(mRS), index, v); |
Stephen Hines | 031ec58c | 2010-10-11 10:54:21 -0700 | [diff] [blame] | 228 | } |
| 229 | |
Stephen Hines | 9c9ad3f8c2 | 2012-05-07 15:34:29 -0700 | [diff] [blame] | 230 | /** |
Jason Sams | 67e3d20 | 2011-01-09 13:49:01 -0800 | [diff] [blame] | 231 | * Only intended for use by generated reflected code. |
| 232 | * |
| 233 | * @param index |
| 234 | * @param v |
| 235 | */ |
Jason Sams | 0b9a22c | 2010-07-02 15:35:19 -0700 | [diff] [blame] | 236 | public void setVar(int index, boolean v) { |
Jason Sams | e07694b | 2012-04-03 15:36:36 -0700 | [diff] [blame] | 237 | mRS.nScriptSetVarI(getID(mRS), index, v ? 1 : 0); |
Jason Sams | 0b9a22c | 2010-07-02 15:35:19 -0700 | [diff] [blame] | 238 | } |
| 239 | |
Stephen Hines | 9c9ad3f8c2 | 2012-05-07 15:34:29 -0700 | [diff] [blame] | 240 | /** |
Jason Sams | 67e3d20 | 2011-01-09 13:49:01 -0800 | [diff] [blame] | 241 | * Only intended for use by generated reflected code. |
| 242 | * |
| 243 | * @param index |
| 244 | * @param o |
| 245 | */ |
Jason Sams | 6f4cf0b | 2010-11-16 17:37:02 -0800 | [diff] [blame] | 246 | public void setVar(int index, BaseObj o) { |
Jason Sams | e07694b | 2012-04-03 15:36:36 -0700 | [diff] [blame] | 247 | mRS.nScriptSetVarObj(getID(mRS), index, (o == null) ? 0 : o.getID(mRS)); |
Jason Sams | 6f4cf0b | 2010-11-16 17:37:02 -0800 | [diff] [blame] | 248 | } |
| 249 | |
Stephen Hines | 9c9ad3f8c2 | 2012-05-07 15:34:29 -0700 | [diff] [blame] | 250 | /** |
Jason Sams | 67e3d20 | 2011-01-09 13:49:01 -0800 | [diff] [blame] | 251 | * Only intended for use by generated reflected code. |
| 252 | * |
| 253 | * @param index |
| 254 | * @param v |
| 255 | */ |
Jason Sams | 4d33993 | 2010-05-11 14:03:58 -0700 | [diff] [blame] | 256 | public void setVar(int index, FieldPacker v) { |
Jason Sams | e07694b | 2012-04-03 15:36:36 -0700 | [diff] [blame] | 257 | mRS.nScriptSetVarV(getID(mRS), index, v.getData()); |
Jason Sams | 69f0d31 | 2009-08-03 18:11:17 -0700 | [diff] [blame] | 258 | } |
| 259 | |
Stephen Hines | 9c9ad3f8c2 | 2012-05-07 15:34:29 -0700 | [diff] [blame] | 260 | /** |
Stephen Hines | adeb809 | 2012-04-20 14:26:06 -0700 | [diff] [blame] | 261 | * Only intended for use by generated reflected code. |
| 262 | * |
| 263 | * @param index |
| 264 | * @param v |
| 265 | * @param e |
| 266 | * @param dims |
| 267 | */ |
| 268 | public void setVar(int index, FieldPacker v, Element e, int[] dims) { |
| 269 | mRS.nScriptSetVarVE(getID(mRS), index, v.getData(), e.getID(mRS), dims); |
| 270 | } |
| 271 | |
Jason Sams | 2253417 | 2009-08-04 16:58:20 -0700 | [diff] [blame] | 272 | public void setTimeZone(String timeZone) { |
Jason Sams | 771bebb | 2009-12-07 12:40:12 -0800 | [diff] [blame] | 273 | mRS.validate(); |
Jason Sams | 2253417 | 2009-08-04 16:58:20 -0700 | [diff] [blame] | 274 | try { |
Jason Sams | e07694b | 2012-04-03 15:36:36 -0700 | [diff] [blame] | 275 | mRS.nScriptSetTimeZone(getID(mRS), timeZone.getBytes("UTF-8")); |
Jason Sams | 2253417 | 2009-08-04 16:58:20 -0700 | [diff] [blame] | 276 | } catch (java.io.UnsupportedEncodingException e) { |
| 277 | throw new RuntimeException(e); |
| 278 | } |
| 279 | } |
Jason Sams | 69f0d31 | 2009-08-03 18:11:17 -0700 | [diff] [blame] | 280 | |
| 281 | public static class Builder { |
| 282 | RenderScript mRS; |
Jason Sams | 69f0d31 | 2009-08-03 18:11:17 -0700 | [diff] [blame] | 283 | |
| 284 | Builder(RenderScript rs) { |
| 285 | mRS = rs; |
| 286 | } |
Jason Sams | 69f0d31 | 2009-08-03 18:11:17 -0700 | [diff] [blame] | 287 | } |
| 288 | |
Jason Sams | 2d71bc7 | 2010-03-26 16:06:43 -0700 | [diff] [blame] | 289 | |
| 290 | public static class FieldBase { |
| 291 | protected Element mElement; |
Jason Sams | 2d71bc7 | 2010-03-26 16:06:43 -0700 | [diff] [blame] | 292 | protected Allocation mAllocation; |
| 293 | |
| 294 | protected void init(RenderScript rs, int dimx) { |
Jason Sams | 5476b45 | 2010-12-08 16:14:36 -0800 | [diff] [blame] | 295 | mAllocation = Allocation.createSized(rs, mElement, dimx, Allocation.USAGE_SCRIPT); |
| 296 | } |
| 297 | |
| 298 | protected void init(RenderScript rs, int dimx, int usages) { |
| 299 | mAllocation = Allocation.createSized(rs, mElement, dimx, Allocation.USAGE_SCRIPT | usages); |
Jason Sams | 2d71bc7 | 2010-03-26 16:06:43 -0700 | [diff] [blame] | 300 | } |
| 301 | |
| 302 | protected FieldBase() { |
| 303 | } |
| 304 | |
| 305 | public Element getElement() { |
| 306 | return mElement; |
| 307 | } |
| 308 | |
| 309 | public Type getType() { |
Jason Sams | 31a7e42 | 2010-10-26 13:09:17 -0700 | [diff] [blame] | 310 | return mAllocation.getType(); |
Jason Sams | 2d71bc7 | 2010-03-26 16:06:43 -0700 | [diff] [blame] | 311 | } |
| 312 | |
| 313 | public Allocation getAllocation() { |
| 314 | return mAllocation; |
| 315 | } |
| 316 | |
| 317 | //@Override |
| 318 | public void updateAllocation() { |
| 319 | } |
Jason Sams | 2d71bc7 | 2010-03-26 16:06:43 -0700 | [diff] [blame] | 320 | } |
Tim Murray | fbfaa85 | 2012-12-14 16:01:58 -0800 | [diff] [blame^] | 321 | |
| 322 | public static final class LaunchOptions { |
| 323 | protected int xstart = -1; |
| 324 | protected int ystart = -1; |
| 325 | protected int xend = -1 ; |
| 326 | protected int yend = -1; |
| 327 | |
| 328 | protected int strategy; |
| 329 | |
| 330 | public void setX(int xstartArg, int xendArg) { |
| 331 | if (xstartArg < 0 || xendArg <= xstartArg) { |
| 332 | throw new RSIllegalArgumentException("Invalid dimensions"); |
| 333 | } |
| 334 | xstart = xstartArg; |
| 335 | xend = xendArg; |
| 336 | } |
| 337 | |
| 338 | public void setY(int ystartArg, int yendArg) { |
| 339 | if (ystartArg < 0 || yendArg <= ystartArg) { |
| 340 | throw new RSIllegalArgumentException("Invalid dimensions"); |
| 341 | } |
| 342 | ystart = ystartArg; |
| 343 | yend = yendArg; |
| 344 | } |
| 345 | |
| 346 | public int getXStart() { |
| 347 | return xstart; |
| 348 | } |
| 349 | public int getXEnd() { |
| 350 | return xend; |
| 351 | } |
| 352 | public int getYStart() { |
| 353 | return ystart; |
| 354 | } |
| 355 | public int getYEnd() { |
| 356 | return yend; |
| 357 | } |
| 358 | |
| 359 | } |
Jason Sams | 69f0d31 | 2009-08-03 18:11:17 -0700 | [diff] [blame] | 360 | } |
| 361 | |