OrcLib: OutputSpec: add conversion functions for Text::Encoding

DFIR-ORC · Feb 10, 2021 · 36b951d · 36b951d
1 parent 1803a40
commit 36b951d
Show file tree

Hide file tree

Showing 2 changed files with 34 additions and 0 deletions.
diff --git a/src/OrcLib/OutputSpec.cpp b/src/OrcLib/OutputSpec.cpp
@@ -503,6 +503,20 @@ OutputSpec::Disposition OutputSpec::ToDisposition(Orc::FileDisposition dispositi
     }
 }
 
+OutputSpec::Encoding OutputSpec::ToEncoding(Text::Encoding encoding)
+{
+    switch (encoding)
+    {
+        case Text::Encoding::Utf8:
+            return OutputSpec::Encoding::UTF8;
+        case Text::Encoding::Utf16:
+            return OutputSpec::Encoding::UTF16;
+        default:
+            assert(nullptr);
+            return OutputSpec::Encoding::kUnknown;
+    }
+}
+
 Orc::FileDisposition ToFileDisposition(OutputSpec::Disposition disposition)
 {
     switch (disposition)
@@ -520,4 +534,19 @@ Orc::FileDisposition ToFileDisposition(OutputSpec::Disposition disposition)
     }
 }
 
+Text::Encoding ToEncoding(OutputSpec::Encoding encoding)
+{
+    switch (encoding)
+    {
+        case OutputSpec::Encoding::UTF8:
+            return Text::Encoding::Utf8;
+        case OutputSpec::Encoding::UTF16:
+            return Text::Encoding::Utf16;
+        default: {
+            assert(nullptr);
+            return Text::Encoding::Unknown;
+        }
+    }
+}
+
 }  // namespace Orc
diff --git a/src/OrcLib/OutputSpec.h b/src/OrcLib/OutputSpec.h
@@ -17,6 +17,7 @@
 
 #include "OutputSpecTypes.h"
 #include "FileDisposition.h"
+#include "Text/Encoding.h"
 #include "Text/Fmt/OutputSpecTypes.h"
 
 #pragma managed(push, off)
@@ -111,6 +112,8 @@ class ORCLIB_API OutputSpec
 
     static OutputSpec::Disposition ToDisposition(Orc::FileDisposition disposition);
 
+    static OutputSpec::Encoding ToEncoding(Text::Encoding);
+
     HRESULT Configure(
         OutputSpec::Kind supportedTypes,
         const std::wstring& inputString,
@@ -136,6 +139,8 @@ class ORCLIB_API OutputSpec
 
 Orc::FileDisposition ToFileDisposition(OutputSpec::Disposition disposition);
 
+Text::Encoding ToEncoding(OutputSpec::Encoding encoding);
+
 }  // namespace Orc
 
 #pragma managed(pop)