import { BasePluginConfig, EventHook } from '@embedpdf/core';
import { Task, Rect, Size, PdfPageTextRuns } from '@embedpdf/models';
import { AiErrorReason } from '@embedpdf/ai';
export interface LayoutAnalysisPluginConfig extends BasePluginConfig {
    /** Layout detection confidence threshold (default: 0.35). */
    layoutThreshold?: number;
    /** Table structure confidence threshold (default: 0.8). */
    tableStructureThreshold?: number;
    /** Enable table structure analysis (default: false). */
    tableStructure?: boolean;
    /** Auto-analyze when a page first loads (default: false). */
    autoAnalyze?: boolean;
    /** Scale factor for rendering pages before analysis (default: 2.0). */
    renderScale?: number;
}
/**
 * A single layout block in PDF page coordinates (points).
 */
export interface LayoutBlock {
    /** Globally unique ID for this detection. */
    id: string;
    /** Numeric class ID from the model. */
    classId: number;
    /** Human-readable label (e.g., 'text', 'table', 'image'). */
    label: string;
    /** Confidence score (0 to 1). */
    score: number;
    /** Bounding box in PDF page coordinates (points) [x, y, width, height]. */
    rect: Rect;
    /** Bounding box in source image coordinates [x1, y1, x2, y2]. */
    imageBbox: [number, number, number, number];
    /** Reading order index. */
    readingOrder: number;
}
/**
 * Table structure element in PDF page coordinates.
 */
export interface TableStructureElement {
    classId: number;
    label: string;
    score: number;
    rect: Rect;
}
/**
 * Layout analysis result for a single page.
 */
export interface PageLayout {
    pageIndex: number;
    blocks: LayoutBlock[];
    tableStructures: Map<string, TableStructureElement[]>;
    /** The image dimensions used for analysis. */
    imageSize: Size;
    /** The PDF page size in points. */
    pageSize: Size;
}
/**
 * Layout analysis result for an entire document.
 */
export interface DocumentLayout {
    pages: PageLayout[];
}
export type PageAnalysisProgress = {
    stage: 'rendering';
    pageIndex: number;
} | {
    stage: 'downloading-model';
    pageIndex: number;
    loaded: number;
    total: number;
} | {
    stage: 'creating-session';
    pageIndex: number;
} | {
    stage: 'layout-detection';
    pageIndex: number;
} | {
    stage: 'table-structure';
    pageIndex: number;
    tableIndex: number;
    tableCount: number;
} | {
    stage: 'mapping-coordinates';
    pageIndex: number;
};
export type DocumentAnalysisProgress = {
    stage: 'page-complete';
    pageIndex: number;
    completed: number;
    total: number;
} | PageAnalysisProgress;
export type LayoutAnalysisErrorReason = AiErrorReason | {
    type: 'render-failed';
    pageIndex: number;
    message: string;
} | {
    type: 'no-document';
    message: string;
};
export type LayoutTask<R, P = unknown> = Task<R, LayoutAnalysisErrorReason, P>;
export type PageAnalysisStatus = 'idle' | 'analyzing' | 'complete' | 'error';
export interface PageAnalysisState {
    status: PageAnalysisStatus;
    layout: PageLayout | null;
    error: string | null;
    tableStructureAnalyzed: boolean;
}
export interface LayoutAnalysisState {
    documents: Record<string, {
        pages: Record<number, PageAnalysisState>;
    }>;
    layoutOverlayVisible: boolean;
    tableStructureOverlayVisible: boolean;
    tableStructureEnabled: boolean;
    selectedBlockId: string | null;
    layoutThreshold: number;
    tableStructureThreshold: number;
}
export interface PageLayoutChangeEvent {
    pageIndex: number;
    layout: PageLayout | null;
}
export interface PageLayoutChangeGlobalEvent extends PageLayoutChangeEvent {
    documentId: string;
}
export interface StateChangeEvent {
    documentId: string;
    state: LayoutAnalysisState;
}
export interface AnalyzePageOptions {
    /** Re-analyze even if results already exist. */
    force?: boolean;
}
export interface LayoutAnalysisScope {
    analyzePage(pageIndex: number, options?: AnalyzePageOptions): LayoutTask<PageLayout, PageAnalysisProgress>;
    analyzeAllPages(options?: AnalyzePageOptions): LayoutTask<DocumentLayout, DocumentAnalysisProgress>;
    getPageLayout(pageIndex: number): PageLayout | null;
    getPageTextRuns(pageIndex: number): Task<PdfPageTextRuns, any, any>;
    onPageLayoutChange: EventHook<PageLayoutChangeEvent>;
    getState(): LayoutAnalysisState;
    onStateChange: EventHook<LayoutAnalysisState>;
}
export interface LayoutAnalysisCapability {
    analyzePage(pageIndex: number, options?: AnalyzePageOptions): LayoutTask<PageLayout, PageAnalysisProgress>;
    analyzeAllPages(options?: AnalyzePageOptions): LayoutTask<DocumentLayout, DocumentAnalysisProgress>;
    getPageLayout(pageIndex: number): PageLayout | null;
    getPageTextRuns(pageIndex: number): Task<PdfPageTextRuns, any, any>;
    forDocument(documentId: string): LayoutAnalysisScope;
    onPageLayoutChange: EventHook<PageLayoutChangeGlobalEvent>;
    onStateChange: EventHook<StateChangeEvent>;
    setLayoutOverlayVisible(visible: boolean): void;
    setTableStructureOverlayVisible(visible: boolean): void;
    setTableStructureEnabled(enabled: boolean): void;
    setLayoutThreshold(threshold: number): void;
    setTableStructureThreshold(threshold: number): void;
    selectBlock(blockId: string | null): void;
    clearPageResults(documentId: string, pageIndex: number): void;
    clearAllResults(documentId: string): void;
}