mirror of
https://github.com/mozilla/pdf.js.git
synced 2025-04-19 06:38:07 +02:00
This patch is adding some code in order to extract a drawing as curves from an image. The algorithm is basically the following: - reduce the dimensions - make it gray - apply a bilateral filter in order to add some blurryness while keeping the edges - compute the histogram - guess what's the background color which should contain a large majority of the pixels - make a binary image - extract the contours in using the Suzuki algorithm - apply the Douglas-Peucker algorithm in order to reduce the number of points The algorithm is improvable but it should work pretty well if there's a clear difference between the background and the drawing. In a v2 we could use a ML model in order to improve the extraction. There's few changes related to the UI in order to make the tool usable, but they're very basic for the moment.
396 lines
12 KiB
JavaScript
396 lines
12 KiB
JavaScript
/* Copyright 2016 Mozilla Foundation
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
/** @typedef {import("./event_utils.js").EventBus} EventBus */
|
|
|
|
import { AnnotationEditorType, ColorPicker, noContextMenu } from "pdfjs-lib";
|
|
import {
|
|
DEFAULT_SCALE,
|
|
DEFAULT_SCALE_VALUE,
|
|
MAX_SCALE,
|
|
MIN_SCALE,
|
|
toggleExpandedBtn,
|
|
} from "./ui_utils.js";
|
|
|
|
/**
|
|
* @typedef {Object} ToolbarOptions
|
|
* @property {HTMLDivElement} container - Container for the secondary toolbar.
|
|
* @property {HTMLSpanElement} numPages - Label that contains number of pages.
|
|
* @property {HTMLInputElement} pageNumber - Control for display and user input
|
|
* of the current page number.
|
|
* @property {HTMLSelectElement} scaleSelect - Scale selection control.
|
|
* Its width is adjusted, when necessary, on UI localization.
|
|
* @property {HTMLOptionElement} customScaleOption - The item used to display
|
|
* a non-predefined scale.
|
|
* @property {HTMLButtonElement} previous - Button to go to the previous page.
|
|
* @property {HTMLButtonElement} next - Button to go to the next page.
|
|
* @property {HTMLButtonElement} zoomIn - Button to zoom in the pages.
|
|
* @property {HTMLButtonElement} zoomOut - Button to zoom out the pages.
|
|
* @property {HTMLButtonElement} editorFreeTextButton - Button to switch to
|
|
* FreeText editing.
|
|
* @property {HTMLButtonElement} download - Button to download the document.
|
|
*/
|
|
|
|
class Toolbar {
|
|
#opts;
|
|
|
|
/**
|
|
* @param {ToolbarOptions} options
|
|
* @param {EventBus} eventBus
|
|
* @param {number} toolbarDensity - The toolbar density value.
|
|
* The possible values are:
|
|
* - 0 (default) - The regular toolbar size.
|
|
* - 1 (compact) - The small toolbar size.
|
|
* - 2 (touch) - The large toolbar size.
|
|
*/
|
|
constructor(options, eventBus, toolbarDensity = 0) {
|
|
this.#opts = options;
|
|
this.eventBus = eventBus;
|
|
const buttons = [
|
|
{ element: options.previous, eventName: "previouspage" },
|
|
{ element: options.next, eventName: "nextpage" },
|
|
{ element: options.zoomIn, eventName: "zoomin" },
|
|
{ element: options.zoomOut, eventName: "zoomout" },
|
|
{ element: options.print, eventName: "print" },
|
|
{ element: options.download, eventName: "download" },
|
|
{
|
|
element: options.editorFreeTextButton,
|
|
eventName: "switchannotationeditormode",
|
|
eventDetails: {
|
|
get mode() {
|
|
const { classList } = options.editorFreeTextButton;
|
|
return classList.contains("toggled")
|
|
? AnnotationEditorType.NONE
|
|
: AnnotationEditorType.FREETEXT;
|
|
},
|
|
},
|
|
},
|
|
{
|
|
element: options.editorHighlightButton,
|
|
eventName: "switchannotationeditormode",
|
|
eventDetails: {
|
|
get mode() {
|
|
const { classList } = options.editorHighlightButton;
|
|
return classList.contains("toggled")
|
|
? AnnotationEditorType.NONE
|
|
: AnnotationEditorType.HIGHLIGHT;
|
|
},
|
|
},
|
|
},
|
|
{
|
|
element: options.editorInkButton,
|
|
eventName: "switchannotationeditormode",
|
|
eventDetails: {
|
|
get mode() {
|
|
const { classList } = options.editorInkButton;
|
|
return classList.contains("toggled")
|
|
? AnnotationEditorType.NONE
|
|
: AnnotationEditorType.INK;
|
|
},
|
|
},
|
|
},
|
|
{
|
|
element: options.editorStampButton,
|
|
eventName: "switchannotationeditormode",
|
|
eventDetails: {
|
|
get mode() {
|
|
const { classList } = options.editorStampButton;
|
|
return classList.contains("toggled")
|
|
? AnnotationEditorType.NONE
|
|
: AnnotationEditorType.STAMP;
|
|
},
|
|
},
|
|
telemetry: {
|
|
type: "editing",
|
|
data: { action: "pdfjs.image.icon_click" },
|
|
},
|
|
},
|
|
{
|
|
element: options.editorSignatureButton,
|
|
eventName: "switchannotationeditormode",
|
|
eventDetails: {
|
|
get mode() {
|
|
const { classList } = options.editorSignatureButton;
|
|
return classList.contains("toggled")
|
|
? AnnotationEditorType.NONE
|
|
: AnnotationEditorType.SIGNATURE;
|
|
},
|
|
},
|
|
},
|
|
];
|
|
|
|
// Bind the event listeners for click and various other actions.
|
|
this.#bindListeners(buttons);
|
|
|
|
this.#updateToolbarDensity({ value: toolbarDensity });
|
|
this.reset();
|
|
}
|
|
|
|
#updateToolbarDensity({ value }) {
|
|
let name = "normal";
|
|
switch (value) {
|
|
case 1:
|
|
name = "compact";
|
|
break;
|
|
case 2:
|
|
name = "touch";
|
|
break;
|
|
}
|
|
document.documentElement.setAttribute("data-toolbar-density", name);
|
|
}
|
|
|
|
#setAnnotationEditorUIManager(uiManager, parentContainer) {
|
|
const colorPicker = new ColorPicker({ uiManager });
|
|
uiManager.setMainHighlightColorPicker(colorPicker);
|
|
parentContainer.append(colorPicker.renderMainDropdown());
|
|
}
|
|
|
|
setPageNumber(pageNumber, pageLabel) {
|
|
this.pageNumber = pageNumber;
|
|
this.pageLabel = pageLabel;
|
|
this.#updateUIState(false);
|
|
}
|
|
|
|
setPagesCount(pagesCount, hasPageLabels) {
|
|
this.pagesCount = pagesCount;
|
|
this.hasPageLabels = hasPageLabels;
|
|
this.#updateUIState(true);
|
|
}
|
|
|
|
setPageScale(pageScaleValue, pageScale) {
|
|
this.pageScaleValue = (pageScaleValue || pageScale).toString();
|
|
this.pageScale = pageScale;
|
|
this.#updateUIState(false);
|
|
}
|
|
|
|
reset() {
|
|
this.pageNumber = 0;
|
|
this.pageLabel = null;
|
|
this.hasPageLabels = false;
|
|
this.pagesCount = 0;
|
|
this.pageScaleValue = DEFAULT_SCALE_VALUE;
|
|
this.pageScale = DEFAULT_SCALE;
|
|
this.#updateUIState(true);
|
|
this.updateLoadingIndicatorState();
|
|
|
|
// Reset the Editor buttons too, since they're document specific.
|
|
this.#editorModeChanged({ mode: AnnotationEditorType.DISABLE });
|
|
}
|
|
|
|
#bindListeners(buttons) {
|
|
const { eventBus } = this;
|
|
const {
|
|
editorHighlightColorPicker,
|
|
editorHighlightButton,
|
|
pageNumber,
|
|
scaleSelect,
|
|
} = this.#opts;
|
|
const self = this;
|
|
|
|
// The buttons within the toolbar.
|
|
for (const { element, eventName, eventDetails, telemetry } of buttons) {
|
|
element.addEventListener("click", evt => {
|
|
if (eventName !== null) {
|
|
eventBus.dispatch(eventName, {
|
|
source: this,
|
|
...eventDetails,
|
|
// evt.detail is the number of clicks.
|
|
isFromKeyboard: evt.detail === 0,
|
|
});
|
|
}
|
|
if (telemetry) {
|
|
eventBus.dispatch("reporttelemetry", {
|
|
source: this,
|
|
details: telemetry,
|
|
});
|
|
}
|
|
});
|
|
}
|
|
// The non-button elements within the toolbar.
|
|
pageNumber.addEventListener("click", function () {
|
|
this.select();
|
|
});
|
|
pageNumber.addEventListener("change", function () {
|
|
eventBus.dispatch("pagenumberchanged", {
|
|
source: self,
|
|
value: this.value,
|
|
});
|
|
});
|
|
|
|
scaleSelect.addEventListener("change", function () {
|
|
if (this.value === "custom") {
|
|
return;
|
|
}
|
|
eventBus.dispatch("scalechanged", {
|
|
source: self,
|
|
value: this.value,
|
|
});
|
|
});
|
|
// Here we depend on browsers dispatching the "click" event *after* the
|
|
// "change" event, when the <select>-element changes.
|
|
scaleSelect.addEventListener("click", function ({ target }) {
|
|
// Remove focus when an <option>-element was *clicked*, to improve the UX
|
|
// for mouse users (fixes bug 1300525 and issue 4923).
|
|
if (
|
|
this.value === self.pageScaleValue &&
|
|
target.tagName.toUpperCase() === "OPTION"
|
|
) {
|
|
this.blur();
|
|
}
|
|
});
|
|
// Suppress context menus for some controls.
|
|
scaleSelect.oncontextmenu = noContextMenu;
|
|
|
|
eventBus._on(
|
|
"annotationeditormodechanged",
|
|
this.#editorModeChanged.bind(this)
|
|
);
|
|
eventBus._on("showannotationeditorui", ({ mode }) => {
|
|
switch (mode) {
|
|
case AnnotationEditorType.HIGHLIGHT:
|
|
editorHighlightButton.click();
|
|
break;
|
|
}
|
|
});
|
|
eventBus._on("toolbardensity", this.#updateToolbarDensity.bind(this));
|
|
|
|
if (editorHighlightColorPicker) {
|
|
eventBus._on(
|
|
"annotationeditoruimanager",
|
|
({ uiManager }) => {
|
|
this.#setAnnotationEditorUIManager(
|
|
uiManager,
|
|
editorHighlightColorPicker
|
|
);
|
|
},
|
|
// Once the color picker has been added, we don't want to add it again.
|
|
{ once: true }
|
|
);
|
|
}
|
|
}
|
|
|
|
#editorModeChanged({ mode }) {
|
|
const {
|
|
editorFreeTextButton,
|
|
editorFreeTextParamsToolbar,
|
|
editorHighlightButton,
|
|
editorHighlightParamsToolbar,
|
|
editorInkButton,
|
|
editorInkParamsToolbar,
|
|
editorStampButton,
|
|
editorStampParamsToolbar,
|
|
editorSignatureButton,
|
|
editorSignatureParamsToolbar,
|
|
} = this.#opts;
|
|
|
|
toggleExpandedBtn(
|
|
editorFreeTextButton,
|
|
mode === AnnotationEditorType.FREETEXT,
|
|
editorFreeTextParamsToolbar
|
|
);
|
|
toggleExpandedBtn(
|
|
editorHighlightButton,
|
|
mode === AnnotationEditorType.HIGHLIGHT,
|
|
editorHighlightParamsToolbar
|
|
);
|
|
toggleExpandedBtn(
|
|
editorInkButton,
|
|
mode === AnnotationEditorType.INK,
|
|
editorInkParamsToolbar
|
|
);
|
|
toggleExpandedBtn(
|
|
editorStampButton,
|
|
mode === AnnotationEditorType.STAMP,
|
|
editorStampParamsToolbar
|
|
);
|
|
toggleExpandedBtn(
|
|
editorSignatureButton,
|
|
mode === AnnotationEditorType.SIGNATURE,
|
|
editorSignatureParamsToolbar
|
|
);
|
|
|
|
const isDisable = mode === AnnotationEditorType.DISABLE;
|
|
editorFreeTextButton.disabled = isDisable;
|
|
editorHighlightButton.disabled = isDisable;
|
|
editorInkButton.disabled = isDisable;
|
|
editorStampButton.disabled = isDisable;
|
|
editorSignatureButton.disabled = isDisable;
|
|
}
|
|
|
|
#updateUIState(resetNumPages = false) {
|
|
const { pageNumber, pagesCount, pageScaleValue, pageScale } = this;
|
|
const opts = this.#opts;
|
|
|
|
if (resetNumPages) {
|
|
if (this.hasPageLabels) {
|
|
opts.pageNumber.type = "text";
|
|
|
|
opts.numPages.setAttribute("data-l10n-id", "pdfjs-page-of-pages");
|
|
} else {
|
|
opts.pageNumber.type = "number";
|
|
|
|
opts.numPages.setAttribute("data-l10n-id", "pdfjs-of-pages");
|
|
opts.numPages.setAttribute(
|
|
"data-l10n-args",
|
|
JSON.stringify({ pagesCount })
|
|
);
|
|
}
|
|
opts.pageNumber.max = pagesCount;
|
|
}
|
|
|
|
if (this.hasPageLabels) {
|
|
opts.pageNumber.value = this.pageLabel;
|
|
|
|
opts.numPages.setAttribute(
|
|
"data-l10n-args",
|
|
JSON.stringify({ pageNumber, pagesCount })
|
|
);
|
|
} else {
|
|
opts.pageNumber.value = pageNumber;
|
|
}
|
|
|
|
opts.previous.disabled = pageNumber <= 1;
|
|
opts.next.disabled = pageNumber >= pagesCount;
|
|
|
|
opts.zoomOut.disabled = pageScale <= MIN_SCALE;
|
|
opts.zoomIn.disabled = pageScale >= MAX_SCALE;
|
|
|
|
let predefinedValueFound = false;
|
|
for (const option of opts.scaleSelect.options) {
|
|
if (option.value !== pageScaleValue) {
|
|
option.selected = false;
|
|
continue;
|
|
}
|
|
option.selected = true;
|
|
predefinedValueFound = true;
|
|
}
|
|
if (!predefinedValueFound) {
|
|
opts.customScaleOption.selected = true;
|
|
opts.customScaleOption.setAttribute(
|
|
"data-l10n-args",
|
|
JSON.stringify({
|
|
scale: Math.round(pageScale * 10000) / 100,
|
|
})
|
|
);
|
|
}
|
|
}
|
|
|
|
updateLoadingIndicatorState(loading = false) {
|
|
const { pageNumber } = this.#opts;
|
|
pageNumber.classList.toggle("loading", loading);
|
|
}
|
|
}
|
|
|
|
export { Toolbar };
|