Added more TSParser API implementations

Author: itsaky

android-tree-sitter/src/main/java/com/itsaky/androidide/treesitter/TSParser.java

@@ -97,18 +97,6 @@ public TSTree parseString(TSTree oldTree, String source, TSInputEncoding encodin
             pointer, oldTree.getPointer(), bytes, bytes.length, encoding.getFlag()));
   }
 
-  /**
-   * Instruct the parser to start the next parse from the beginning.
-   *
-   * <p>If the parser previously failed because of a timeout or a cancellation, then by default, it
-   * will resume where it left off on the next call to any of the parsing functions. If you don't
-   * want to resume, and instead intend to use this parser to parse some other document, you must
-   * call this function first.
-   */
-  public void reset() {
-    Native.reset(this.pointer);
-  }
-
   /**
    * Set the maximum duration in microseconds that parsing should be allowed to take before halting.
    *
@@ -127,6 +115,46 @@ public long getTimeout() {
     return Native.getTimeout(this.pointer);
   }
 
+  /**
+   * Set the ranges of text that the parser should include when parsing.
+   *
+   * <p>By default, the parser will always include entire documents. This function allows you to
+   * parse only a *portion* of a document but still return a syntax tree whose ranges match up with
+   * the document as a whole. You can also pass multiple disjoint ranges.
+   *
+   * <p>The second and third parameters specify the location and length of an array of ranges. The
+   * parser does *not* take ownership of these ranges; it copies the data, so it doesn't matter how
+   * these ranges are allocated.
+   *
+   * <p>If the ranges parameter is an empty array, then the entire document will be parsed.
+   * Otherwise, the given ranges must be ordered from earliest to latest in the document, and they
+   * must not overlap. That is, the following must hold for all `i` < `length - 1`:
+   * ranges[i].end_byte <= ranges[i + 1].start_byte where `length` is the length of the ranges
+   * array.
+   *
+   * <p>If this requirement is not satisfied, the operation will fail, the ranges will not be
+   * assigned, and this function will return `false`. On success, this function returns `true`
+   */
+  public boolean setIncludedRanges(TSRange[] ranges) {
+    return Native.setIncludedRanges(this.pointer, ranges);
+  }
+
+  public TSRange[] getIncludedRanges() {
+    return Native.getIncludedRanges(this.pointer);
+  }
+
+  /**
+   * Instruct the parser to start the next parse from the beginning.
+   *
+   * <p>If the parser previously failed because of a timeout or a cancellation, then by default, it
+   * will resume where it left off on the next call to any of the parsing functions. If you don't
+   * want to resume, and instead intend to use this parser to parse some other document, you must
+   * call this function first.
+   */
+  public void reset() {
+    Native.reset(this.pointer);
+  }
+
   /** Closes and deletes the current parser. */
   @Override
   public void close() {
@@ -152,5 +180,9 @@ public static native long incrementalParseBytes(
     public static native void setTimeout(long parser, long timeout);
 
     public static native long getTimeout(long parser);
+
+    public static native boolean setIncludedRanges(long parser, TSRange[] ranges);
+
+    public static native TSRange[] getIncludedRanges(long parser);
   }
 }

android-tree-sitter/src/main/jniLibs/arm64-v8a/libts.so

@@ -1,6 +1,8 @@
 #include "com_itsaky_androidide_treesitter_TSParser_Native.h"
 #include "ts_utils.h"
 
+#include <iostream>
+
 JNIEXPORT jlong JNICALL
 Java_com_itsaky_androidide_treesitter_TSParser_00024Native_newParser(
     JNIEnv* env, jclass self) {
@@ -69,5 +71,33 @@ Java_com_itsaky_androidide_treesitter_TSParser_00024Native_setTimeout(
 JNIEXPORT jlong JNICALL
 Java_com_itsaky_androidide_treesitter_TSParser_00024Native_getTimeout(
     JNIEnv* env, jclass self, jlong parser) {
-  return (jlong) ts_parser_timeout_micros((TSParser*)parser);
+  return (jlong)ts_parser_timeout_micros((TSParser*)parser);
+}
+
+JNIEXPORT jboolean JNICALL
+Java_com_itsaky_androidide_treesitter_TSParser_00024Native_setIncludedRanges(
+    JNIEnv* env, jclass self, jlong parser, jobjectArray ranges) {
+  int count = env->GetArrayLength(ranges);
+  TSRange tsRanges[count];
+  for (int i = 0; i < count; i++) {
+    tsRanges[i] = _unmarshalRange(env, env->GetObjectArrayElement(ranges, i));
+  }
+
+  const TSRange* r = tsRanges;
+  return (jboolean) ts_parser_set_included_ranges((TSParser*)parser, r, count);
+}
+
+JNIEXPORT jobjectArray JNICALL
+Java_com_itsaky_androidide_treesitter_TSParser_00024Native_getIncludedRanges(
+    JNIEnv* env, jclass self, jlong parser) {
+  uint32_t count;
+  const TSRange* ranges = ts_parser_included_ranges((TSParser*)parser, &count);
+  std::cout << "Number of included ranges is " << count << std::endl;
+  jclass klass = env->FindClass("com/itsaky/androidide/treesitter/TSRange");
+  jobjectArray result = env->NewObjectArray(count, klass, NULL);
+  for (uint32_t i = 0; i < count; i++) {
+    const TSRange *r = (ranges + i);
+    env->SetObjectArrayElement(result, i, _marshalRange(env, *r));
+  }
+  return result;
 }

android-tree-sitter/src/main/jniLibs/armeabi-v7a/libts.so

@@ -18,7 +18,7 @@ struct TreeCursorNode {
 
 #define _loadField(VARIABLE, CLASS, NAME, TYPE) \
   { VARIABLE = env->GetFieldID(CLASS, NAME, TYPE); }
-
+  
 
 void onLoad(JNIEnv* env);
 

android-tree-sitter/src/main/jniLibs/x86/libts.so

@@ -24,6 +24,7 @@ static jclass _pointClass;
 static jfieldID _pointRowField;
 static jfieldID _pointColumnField;
 
+// TSRange
 static jclass _rangeClass;
 static jfieldID _rangeClassStartByteField;
 static jfieldID _rangeClassEndByteField;