Function printTypeAndEncodingTree

bolt/vector/VectorPrinter.cpp:308–365

}

std::string printTypeAndEncodingTree(
    const BaseVector& vector,
    const std::string& indent) {
  std::ostringstream out;

  const auto newIndent = addIndent(indent);
  switch (vector.encoding()) {
    case VectorEncoding::Simple::CONSTANT:
    case VectorEncoding::Simple::DICTIONARY: {
      out << indent << VectorEncoding::mapSimpleToName(vector.encoding())
          << " ";
      printSizeAndNullCount(vector, out);
      out << std::endl;
      // Constant vector of primitive type doesn't have valueVector.
      if (vector.valueVector()) {
        out << printTypeAndEncodingTree(*vector.valueVector(), newIndent);
      } else {
        out << newIndent << vector.type()->toString() << std::endl;
      }
      break;
    }
    case VectorEncoding::Simple::FLAT:
      printEncodingAndType(vector, indent, out);
      break;
    case VectorEncoding::Simple::ARRAY: {
      auto* arrayVector = vector.as<ArrayVector>();
      printEncodingAndType(vector, indent, out);
      out << indent << "Elements: " << std::endl;
      out << printTypeAndEncodingTree(*arrayVector->elements(), newIndent);
      break;
    }
    case VectorEncoding::Simple::MAP: {
      auto* mapVector = vector.as<MapVector>();
      printEncodingAndType(vector, indent, out);
      out << indent << "Keys: " << std::endl;
      out << printTypeAndEncodingTree(*mapVector->mapKeys(), newIndent);
      out << indent << "Values: " << std::endl;
      out << printTypeAndEncodingTree(*mapVector->mapValues(), newIndent);
      break;
    }
    case VectorEncoding::Simple::ROW: {
      printEncodingAndType(vector, indent, out);
      const auto* rowVector = vector.as<RowVector>();
      const auto& rowType = vector.type()->asRow();
      for (auto i = 0; i < rowType.size(); ++i) {
        out << indent << "Field " << rowType.nameOf(i) << ":" << std::endl;
        out << printTypeAndEncodingTree(*rowVector->childAt(i), newIndent);
      }
      break;
    }
    default:
      BOLT_UNSUPPORTED(
          "Unsupported encoding: {}",
          VectorEncoding::mapSimpleToName(vector.encoding()));
  }

  return out.str();
}