intermediate changes

ref:cde9a383711a11544ce7e107a78147fb96cc4029

1 files changed, 132 insertions, 0 deletions

diff --git a/util/string/vector.h b/util/string/vector.h
new file mode 100644
index 00000000000..e36c348bbef
--- /dev/null
+++ b/util/string/vector.h
@@ -0,0 +1,132 @@
+#pragma once
+
+#include "cast.h"
+#include "split.h"
+
+#include <util/generic/map.h>
+#include <util/generic/strbuf.h>
+#include <util/generic/string.h>
+#include <util/generic/vector.h>
+#include <util/string/cast.h>
+#include <util/system/yassert.h>
+
+#define KEEP_EMPTY_TOKENS 0x01
+
+//
+// NOTE: Check StringSplitter below to get more convenient split string interface.
+
+namespace NPrivate {
+
+    void SplitStringImpl(TVector<TString>* res, const char* ptr,
+                         const char* delimiter, size_t maxFields, int options);
+    void SplitStringImpl(TVector<TString>* res, const char* ptr, size_t len,
+                         const char* delimiter, size_t maxFields, int options);
+
+    void SplitStringImpl(TVector<TUtf16String>* res, const wchar16* ptr,
+                         const wchar16* delimiter, size_t maxFields, int options);
+    void SplitStringImpl(TVector<TUtf16String>* res, const wchar16* ptr, size_t len,
+                         const wchar16* delimiter, size_t maxFields, int options);
+
+    template <typename C>
+    struct TStringDeducer;
+
+    template <>
+    struct TStringDeducer<char> {
+        using type = TString;
+    };
+
+    template <>
+    struct TStringDeducer<wchar16> {
+        using type = TUtf16String;
+    };
+}
+
+template <typename C>
+TVector<typename ::NPrivate::TStringDeducer<C>::type>
+SplitString(const C* ptr, const C* delimiter,
+            size_t maxFields = 0, int options = 0) {
+    TVector<typename ::NPrivate::TStringDeducer<C>::type> res;
+    ::NPrivate::SplitStringImpl(&res, ptr, delimiter, maxFields, options);
+    return res;
+}
+
+template <typename C>
+TVector<typename ::NPrivate::TStringDeducer<C>::type>
+SplitString(const C* ptr, size_t len, const C* delimiter,
+            size_t maxFields = 0, int options = 0) {
+    TVector<typename ::NPrivate::TStringDeducer<C>::type> res;
+    ::NPrivate::SplitStringImpl(&res, ptr, len, delimiter, maxFields, options);
+    return res;
+}
+
+template <typename C>
+TVector<typename ::NPrivate::TStringDeducer<C>::type>
+SplitString(const typename ::NPrivate::TStringDeducer<C>::type& str, const C* delimiter,
+            size_t maxFields = 0, int options = 0) {
+    return SplitString(str.data(), str.size(), delimiter, maxFields, options);
+}
+
+template <class TIter>
+inline TString JoinStrings(TIter begin, TIter end, const TStringBuf delim) {
+    if (begin == end)
+        return TString();
+
+    TString result = ToString(*begin);
+
+    for (++begin; begin != end; ++begin) {
+        result.append(delim);
+        result.append(ToString(*begin));
+    }
+
+    return result;
+}
+
+template <class TIter>
+inline TUtf16String JoinStrings(TIter begin, TIter end, const TWtringBuf delim) {
+    if (begin == end)
+        return TUtf16String();
+
+    TUtf16String result = ToWtring(*begin);
+
+    for (++begin; begin != end; ++begin) {
+        result.append(delim);
+        result.append(ToWtring(*begin));
+    }
+
+    return result;
+}
+
+/// Concatenates elements of given TVector<TString>.
+inline TString JoinStrings(const TVector<TString>& v, const TStringBuf delim) {
+    return JoinStrings(v.begin(), v.end(), delim);
+}
+
+inline TString JoinStrings(const TVector<TString>& v, size_t index, size_t count, const TStringBuf delim) {
+    Y_ASSERT(index + count <= v.size() && "JoinStrings(): index or count out of range");
+    return JoinStrings(v.begin() + index, v.begin() + index + count, delim);
+}
+
+template <typename T>
+inline TString JoinVectorIntoString(const TVector<T>& v, const TStringBuf delim) {
+    return JoinStrings(v.begin(), v.end(), delim);
+}
+
+template <typename T>
+inline TString JoinVectorIntoString(const TVector<T>& v, size_t index, size_t count, const TStringBuf delim) {
+    Y_ASSERT(index + count <= v.size() && "JoinVectorIntoString(): index or count out of range");
+    return JoinStrings(v.begin() + index, v.begin() + index + count, delim);
+}
+
+TUtf16String JoinStrings(const TVector<TUtf16String>& v, const TWtringBuf delim);
+TUtf16String JoinStrings(const TVector<TUtf16String>& v, size_t index, size_t count, const TWtringBuf delim);
+
+//! Converts vector of strings to vector of type T variables
+template <typename T, typename TStringType>
+TVector<T> Scan(const TVector<TStringType>& input) {
+    TVector<T> output;
+    output.reserve(input.size());
+    for (int i = 0; i < input.ysize(); ++i) {
+        output.push_back(FromString<T>(input[i]));
+    }
+    return output;
+}