Customized RepeatedHeaderTableDrawer with configurable amount of rows to repeat

EasytableIssue70.java

package org.vandeseer.integrationtest.custom;

import lombok.experimental.SuperBuilder;
import org.apache.pdfbox.pdmodel.PDDocument;
import org.apache.pdfbox.pdmodel.PDPage;
import org.vandeseer.easytable.TableDrawer;

import java.awt.geom.Point2D;
import java.io.IOException;
import java.util.function.Supplier;

@SuperBuilder
public class CustomRepeatedHeaderTableDrawer extends TableDrawer {

    @Override
    protected void drawPage(PageData pageData) {
        if (pageData.firstRowOnPage != 0) {
            float adaption = 0;
            for (int i = 0; i < 3; i++) {
                adaption += table.getRows().get(i).getHeight();
                drawRow(new Point2D.Float(this.startX, this.startY + calculateHeightForFirstRows(3) - adaption), table.getRows().get(i), i, (drawer, drawingContext) -> {
                    drawer.drawBackground(drawingContext);
                    drawer.drawContent(drawingContext);
                    drawer.drawBorders(drawingContext);
                });
            }
        }

        drawerList.forEach(drawer ->
                drawWithFunction(pageData, new Point2D.Float(this.startX, this.startY), drawer)
        );
    }

    @Override
    public void draw(Supplier<PDDocument> documentSupplier, Supplier<PDPage> pageSupplier, float yOffset) throws IOException {
        super.draw(documentSupplier, pageSupplier, yOffset + calculateHeightForFirstRows(3));
    }

    private float calculateHeightForFirstRows(int x) {
        float height = 0;
        for (int i = 0; i < x; i++) {
            height += table.getRows().get(i).getHeight();
        }
        return height;
    }

}

Thanks for solution.