Merge pull request #221 from pacher/nested-rows-limit

Add nestedRowsLimit parameter to DisplayConfiguration

Author: Jolanrensen

core/src/main/kotlin/org/jetbrains/kotlinx/dataframe/io/html.kt

@@ -120,8 +120,8 @@ internal fun AnyFrame.toHtmlData(
     val scripts = mutableListOf<String>()
     val queue = LinkedList<Pair<AnyFrame, Int>>()
 
-    fun AnyFrame.columnToJs(col: AnyCol, rowsLimit: Int): ColumnDataForJs {
-        val values = rows().take(rowsLimit)
+    fun AnyFrame.columnToJs(col: AnyCol, rowsLimit: Int?): ColumnDataForJs {
+        val values = if (rowsLimit != null) rows().take(rowsLimit) else rows()
         val scale = if (col.isNumber()) col.asNumbers().scale() else 1
         val format = if (scale > 0) {
             RendererDecimalFormat.fromPrecision(scale)
@@ -158,7 +158,7 @@ internal fun AnyFrame.toHtmlData(
     queue.add(this to rootId)
     while (!queue.isEmpty()) {
         val (nextDf, nextId) = queue.pop()
-        val rowsLimit = if (nextId == rootId) configuration.rowsLimit else 5
+        val rowsLimit = if (nextId == rootId) configuration.rowsLimit else configuration.nestedRowsLimit
         val preparedColumns = nextDf.columns().map { nextDf.columnToJs(it, rowsLimit) }
         val js = tableJs(preparedColumns, nextId, rootId, nextDf.nrow)
         scripts.add(js)
@@ -189,7 +189,7 @@ public fun <T> DataFrame<T>.toHTML(
     cellRenderer: CellRenderer = org.jetbrains.kotlinx.dataframe.jupyter.DefaultCellRenderer,
     getFooter: (DataFrame<T>) -> String = { "DataFrame [${it.size}]" },
 ): HtmlData {
-    val limit = configuration.rowsLimit
+    val limit = configuration.rowsLimit ?: Int.MAX_VALUE
 
     val footer = getFooter(this)
     val bodyFooter = buildString {
@@ -210,7 +210,8 @@ public fun <T> DataFrame<T>.toHTML(
 }
 
 public data class DisplayConfiguration(
-    var rowsLimit: Int = 20,
+    var rowsLimit: Int? = 20,
+    var nestedRowsLimit: Int? = 5,
     var cellContentLimit: Int = 40,
     var cellFormatter: RowColFormatter<*, *>? = null,
     var decimalFormat: RendererDecimalFormat = RendererDecimalFormat.DEFAULT,