Support DST timezones conversion for ORC

src/main/cpp/src/GpuTimeZoneDBJni.cpp

@@ -1,4 +1,4 @@
-/* Copyright (c) 2023-2025, NVIDIA CORPORATION.
+/* Copyright (c) 2023-2026, NVIDIA CORPORATION.
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -97,14 +97,56 @@ Java_com_nvidia_spark_rapids_jni_GpuTimeZoneDB_convertTimestampColumnToUTCWithTz
   JNI_CATCH(env, 0);
 }
 
+static spark_rapids_jni::dst_rule parse_dst_rule(JNIEnv* env, jintArray jrule)
+{
+  spark_rapids_jni::dst_rule rule{};
+  if (jrule == nullptr) {
+    rule.has_dst = false;
+    return rule;
+  }
+  constexpr jsize expected_dst_rule_length = 13;
+  auto const rule_length                   = env->GetArrayLength(jrule);
+  JNI_ARG_CHECK(
+    env, rule_length == expected_dst_rule_length, "dst rule array must have 13 ints", rule);
+  jint* arr = env->GetIntArrayElements(jrule, nullptr);
+  // GetIntArrayElements returns nullptr and sets a pending OutOfMemoryError
+  // on failure; return early so the pending exception is raised on JNI exit.
+  if (arr == nullptr) { return rule; }
+  rule.has_dst = true;
+  // Order must match GpuTimeZoneDB.dstRuleToArray():
+  // dstSavings, startMonth, startDay, startDayOfWeek, startTime,
+  // startTimeMode, startMode, endMonth, endDay, endDayOfWeek,
+  // endTime, endTimeMode, endMode
+  rule.dst_savings     = arr[0];
+  rule.start_month     = arr[1];
+  rule.start_day       = arr[2];
+  rule.start_dow       = arr[3];
+  rule.start_time      = arr[4];
+  rule.start_time_mode = arr[5];
+  rule.start_mode      = arr[6];
+  rule.end_month       = arr[7];
+  rule.end_day         = arr[8];
+  rule.end_dow         = arr[9];
+  rule.end_time        = arr[10];
+  rule.end_time_mode   = arr[11];
+  rule.end_mode        = arr[12];
+  env->ReleaseIntArrayElements(jrule, arr, JNI_ABORT);
+  return rule;
+}
+
 JNIEXPORT jlong JNICALL
 Java_com_nvidia_spark_rapids_jni_GpuTimeZoneDB_convertOrcTimezones(JNIEnv* env,
                                                                    jclass,
                                                                    jlong input_handle,
+                                                                   jlong base_offset_us,
                                                                    jlong writer_tz_info_table,
+                                                                   jint writer_tz_initial_offset,
                                                                    jint writer_tz_raw_offset,
+                                                                   jintArray writer_dst_rule,
                                                                    jlong reader_tz_info_table,
-                                                                   jint reader_tz_raw_offset)
+                                                                   jint reader_tz_initial_offset,
+                                                                   jint reader_tz_raw_offset,
+                                                                   jintArray reader_dst_rule)
 {
   JNI_NULL_CHECK(env, input_handle, "input column is null", 0);
 
@@ -114,9 +156,21 @@ Java_com_nvidia_spark_rapids_jni_GpuTimeZoneDB_convertOrcTimezones(JNIEnv* env,
     auto const input              = reinterpret_cast<cudf::column_view const*>(input_handle);
     auto const writer_tz_info_tab = reinterpret_cast<cudf::table_view const*>(writer_tz_info_table);
     auto const reader_tz_info_tab = reinterpret_cast<cudf::table_view const*>(reader_tz_info_table);
+    auto const writer_dst         = parse_dst_rule(env, writer_dst_rule);
+    cudf::jni::check_java_exception(env);
+    auto const reader_dst = parse_dst_rule(env, reader_dst_rule);
+    cudf::jni::check_java_exception(env);
     return cudf::jni::ptr_as_jlong(
-      spark_rapids_jni::convert_orc_writer_reader_timezones(
-        *input, writer_tz_info_tab, writer_tz_raw_offset, reader_tz_info_tab, reader_tz_raw_offset)
+      spark_rapids_jni::convert_orc_writer_reader_timezones(*input,
+                                                            static_cast<int64_t>(base_offset_us),
+                                                            writer_tz_info_tab,
+                                                            writer_tz_initial_offset,
+                                                            writer_tz_raw_offset,
+                                                            writer_dst,
+                                                            reader_tz_info_tab,
+                                                            reader_tz_initial_offset,
+                                                            reader_tz_raw_offset,
+                                                            reader_dst)
         .release());
   }
   JNI_CATCH(env, 0);