ORC-1489: Assign a writer id to CUDF

c++/include/orc/Common.hh

@@ -70,6 +70,7 @@ namespace orc {
     PRESTO_WRITER = 2,
     SCRITCHLEY_GO = 3,
     TRINO_WRITER = 4,
+    CUDF_WRITER = 5,
     UNKNOWN_WRITER = INT32_MAX
   };
 

c++/src/Common.cc

@@ -82,6 +82,8 @@ namespace orc {
         return "Scritchley Go";
       case TRINO_WRITER:
         return "Trino";
+      case CUDF_WRITER:
+        return "CUDF";
       default: {
         std::ostringstream buffer;
         buffer << "Unknown(" << id << ")";

c++/src/Reader.cc

@@ -622,7 +622,7 @@ namespace orc {
   WriterId ReaderImpl::getWriterId() const {
     if (footer->has_writer()) {
       uint32_t id = footer->writer();
-      if (id > WriterId::TRINO_WRITER) {
+      if (id > WriterId::CUDF_WRITER) {
         return WriterId::UNKNOWN_WRITER;
       } else {
         return static_cast<WriterId>(id);

java/core/src/java/org/apache/orc/OrcFile.java

@@ -132,6 +132,7 @@ public enum WriterImplementation {
     PRESTO(2),   // Presto writer
     SCRITCHLEY_GO(3), // Go writer from https://github.com/scritchley/orc
     TRINO(4),   // Trino writer
+    CUDF(5),    // CUDF writer
     UNKNOWN(Integer.MAX_VALUE);
 
     private final int id;
@@ -189,6 +190,9 @@ public enum WriterVersion {
     // Trino Writer
     TRINO_ORIGINAL(WriterImplementation.TRINO, 6),
 
+    // CUDF Writer
+    CUDF_ORIGINAL(WriterImplementation.CUDF, 6),
+
     // Don't use any magic numbers here except for the below:
     FUTURE(WriterImplementation.UNKNOWN, Integer.MAX_VALUE); // a version from a future writer
 

java/core/src/java/org/apache/orc/OrcUtils.java

@@ -414,6 +414,9 @@ public static String getSoftwareVersion(int writer,
       case 4:
         base = "Trino";
         break;
+      case 5:
+        base = "CUDF";
+        break;
       default:
         base = String.format("Unknown(%d)", writer);
         break;

java/core/src/test/org/apache/orc/TestVectorOrcFile.java

@@ -3597,6 +3597,8 @@ public void testWriterVersion(Version fileFormat) throws Exception {
         OrcFile.WriterImplementation.from(2));
     assertEquals(OrcFile.WriterImplementation.TRINO,
             OrcFile.WriterImplementation.from(4));
+    assertEquals(OrcFile.WriterImplementation.CUDF,
+        OrcFile.WriterImplementation.from(5));
     assertEquals(OrcFile.WriterImplementation.UNKNOWN,
         OrcFile.WriterImplementation.from(99));
 
@@ -3615,6 +3617,8 @@ public void testWriterVersion(Version fileFormat) throws Exception {
         OrcFile.WriterVersion.from(OrcFile.WriterImplementation.PRESTO, 6));
     assertEquals(OrcFile.WriterVersion.TRINO_ORIGINAL,
             OrcFile.WriterVersion.from(OrcFile.WriterImplementation.TRINO, 6));
+    assertEquals(OrcFile.WriterVersion.CUDF_ORIGINAL,
+        OrcFile.WriterVersion.from(OrcFile.WriterImplementation.CUDF, 6));
     assertEquals(OrcFile.WriterVersion.FUTURE,
         OrcFile.WriterVersion.from(OrcFile.WriterImplementation.UNKNOWN, 0));
 
@@ -3633,6 +3637,8 @@ public void testWriterVersion(Version fileFormat) throws Exception {
         OrcFile.WriterVersion.PRESTO_ORIGINAL));
     assertTrue(OrcFile.WriterVersion.HIVE_12055.includes(
         OrcFile.WriterVersion.TRINO_ORIGINAL));
+    assertTrue(OrcFile.WriterVersion.HIVE_12055.includes(
+        OrcFile.WriterVersion.CUDF_ORIGINAL));
   }
 
   @ParameterizedTest

proto/orc_proto.proto

@@ -367,6 +367,7 @@ message Footer {
   // 2 = Presto
   // 3 = Scritchley Go from https://github.com/scritchley/orc
   // 4 = Trino
+  // 5 = CUDF
   optional uint32 writer = 9;
 
   // information about the encryption in this file