mirror of
https://github.com/visioncortex/vtracer.git
synced 2026-09-12 15:16:08 -07:00
Compare commits
22 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| fd9cdb08e6 | |||
| 3b2991be84 | |||
| f24ea56a52 | |||
| 592bee6c5c | |||
| 3ebc358b48 | |||
| 3a24c2f755 | |||
| 8acb6bd911 | |||
| efa4351b2c | |||
| a46292b5ed | |||
| 8889cbc7ea | |||
| 6b379a02ef | |||
| 2635d5b874 | |||
| f6cf3e8705 | |||
| 36b16de17a | |||
| 7887c1ebf8 | |||
| 6fdfec8610 | |||
| 1aff9a300a | |||
| ac0a89e08a | |||
| b09f71a2b5 | |||
| e4897dfe99 | |||
| 4544ca740d | |||
| 3d92586e33 |
+108
-46
@@ -1,4 +1,4 @@
|
|||||||
# This file is autogenerated by maturin v1.2.3
|
# This file is autogenerated by maturin v1.11.5
|
||||||
# To update, run
|
# To update, run
|
||||||
#
|
#
|
||||||
# maturin generate-ci github
|
# maturin generate-ci github
|
||||||
@@ -17,86 +17,142 @@ permissions:
|
|||||||
|
|
||||||
jobs:
|
jobs:
|
||||||
linux:
|
linux:
|
||||||
runs-on: ubuntu-latest
|
runs-on: ${{ matrix.platform.runner }}
|
||||||
strategy:
|
strategy:
|
||||||
matrix:
|
matrix:
|
||||||
target: [x86_64, x86, aarch64, armv7, s390x, ppc64le]
|
platform:
|
||||||
|
- runner: ubuntu-22.04
|
||||||
|
target: x86_64
|
||||||
|
- runner: ubuntu-22.04
|
||||||
|
target: x86
|
||||||
|
- runner: ubuntu-22.04
|
||||||
|
target: aarch64
|
||||||
|
- runner: ubuntu-22.04
|
||||||
|
target: armv7
|
||||||
|
- runner: ubuntu-22.04
|
||||||
|
target: s390x
|
||||||
|
- runner: ubuntu-22.04
|
||||||
|
target: ppc64le
|
||||||
steps:
|
steps:
|
||||||
- uses: actions/checkout@v3
|
- uses: actions/checkout@v6
|
||||||
- uses: actions/setup-python@v4
|
- uses: actions/setup-python@v6
|
||||||
with:
|
with:
|
||||||
python-version: '3.10'
|
python-version: 3.x
|
||||||
- name: Build wheels
|
- name: Build wheels
|
||||||
uses: PyO3/maturin-action@v1
|
uses: PyO3/maturin-action@v1
|
||||||
with:
|
with:
|
||||||
target: ${{ matrix.target }}
|
target: ${{ matrix.platform.target }}
|
||||||
args: -m cmdapp/Cargo.toml --release --out dist --find-interpreter
|
args: --release --out dist --find-interpreter --manifest-path cmdapp/Cargo.toml
|
||||||
sccache: 'true'
|
sccache: ${{ !startsWith(github.ref, 'refs/tags/') }}
|
||||||
manylinux: auto
|
manylinux: auto
|
||||||
- name: Upload wheels
|
- name: Upload wheels
|
||||||
uses: actions/upload-artifact@v3
|
uses: actions/upload-artifact@v5
|
||||||
with:
|
with:
|
||||||
name: wheels
|
name: wheels-linux-${{ matrix.platform.target }}
|
||||||
|
path: dist
|
||||||
|
|
||||||
|
musllinux:
|
||||||
|
runs-on: ${{ matrix.platform.runner }}
|
||||||
|
strategy:
|
||||||
|
matrix:
|
||||||
|
platform:
|
||||||
|
- runner: ubuntu-22.04
|
||||||
|
target: x86_64
|
||||||
|
- runner: ubuntu-22.04
|
||||||
|
target: x86
|
||||||
|
- runner: ubuntu-22.04
|
||||||
|
target: aarch64
|
||||||
|
- runner: ubuntu-22.04
|
||||||
|
target: armv7
|
||||||
|
steps:
|
||||||
|
- uses: actions/checkout@v6
|
||||||
|
- uses: actions/setup-python@v6
|
||||||
|
with:
|
||||||
|
python-version: 3.x
|
||||||
|
- name: Build wheels
|
||||||
|
uses: PyO3/maturin-action@v1
|
||||||
|
with:
|
||||||
|
target: ${{ matrix.platform.target }}
|
||||||
|
args: --release --out dist --find-interpreter --manifest-path cmdapp/Cargo.toml
|
||||||
|
sccache: ${{ !startsWith(github.ref, 'refs/tags/') }}
|
||||||
|
manylinux: musllinux_1_2
|
||||||
|
- name: Upload wheels
|
||||||
|
uses: actions/upload-artifact@v5
|
||||||
|
with:
|
||||||
|
name: wheels-musllinux-${{ matrix.platform.target }}
|
||||||
path: dist
|
path: dist
|
||||||
|
|
||||||
windows:
|
windows:
|
||||||
runs-on: windows-latest
|
runs-on: ${{ matrix.platform.runner }}
|
||||||
strategy:
|
strategy:
|
||||||
matrix:
|
matrix:
|
||||||
target: [x64, x86]
|
platform:
|
||||||
|
- runner: windows-latest
|
||||||
|
target: x64
|
||||||
|
python_arch: x64
|
||||||
|
- runner: windows-latest
|
||||||
|
target: x86
|
||||||
|
python_arch: x86
|
||||||
|
- runner: windows-11-arm
|
||||||
|
target: aarch64
|
||||||
|
python_arch: arm64
|
||||||
steps:
|
steps:
|
||||||
- uses: actions/checkout@v3
|
- uses: actions/checkout@v6
|
||||||
- uses: actions/setup-python@v4
|
- uses: actions/setup-python@v6
|
||||||
with:
|
with:
|
||||||
python-version: '3.10'
|
python-version: 3.13
|
||||||
architecture: ${{ matrix.target }}
|
architecture: ${{ matrix.platform.python_arch }}
|
||||||
- name: Build wheels
|
- name: Build wheels
|
||||||
uses: PyO3/maturin-action@v1
|
uses: PyO3/maturin-action@v1
|
||||||
with:
|
with:
|
||||||
target: ${{ matrix.target }}
|
target: ${{ matrix.platform.target }}
|
||||||
args: -m cmdapp/Cargo.toml --release --out dist --find-interpreter
|
args: --release --out dist --find-interpreter --manifest-path cmdapp/Cargo.toml
|
||||||
sccache: 'true'
|
sccache: ${{ !startsWith(github.ref, 'refs/tags/') }}
|
||||||
- name: Upload wheels
|
- name: Upload wheels
|
||||||
uses: actions/upload-artifact@v3
|
uses: actions/upload-artifact@v5
|
||||||
with:
|
with:
|
||||||
name: wheels
|
name: wheels-windows-${{ matrix.platform.target }}
|
||||||
path: dist
|
path: dist
|
||||||
|
|
||||||
macos:
|
macos:
|
||||||
runs-on: macos-latest
|
runs-on: ${{ matrix.platform.runner }}
|
||||||
strategy:
|
strategy:
|
||||||
matrix:
|
matrix:
|
||||||
target: [x86_64, aarch64]
|
platform:
|
||||||
|
- runner: macos-15-intel
|
||||||
|
target: x86_64
|
||||||
|
- runner: macos-latest
|
||||||
|
target: aarch64
|
||||||
steps:
|
steps:
|
||||||
- uses: actions/checkout@v3
|
- uses: actions/checkout@v6
|
||||||
- uses: actions/setup-python@v4
|
- uses: actions/setup-python@v6
|
||||||
with:
|
with:
|
||||||
python-version: '3.10'
|
python-version: 3.x
|
||||||
- name: Build wheels
|
- name: Build wheels
|
||||||
uses: PyO3/maturin-action@v1
|
uses: PyO3/maturin-action@v1
|
||||||
with:
|
with:
|
||||||
target: ${{ matrix.target }}
|
target: ${{ matrix.platform.target }}
|
||||||
args: -m cmdapp/Cargo.toml --release --out dist --find-interpreter
|
args: --release --out dist --find-interpreter --manifest-path cmdapp/Cargo.toml
|
||||||
sccache: 'true'
|
sccache: ${{ !startsWith(github.ref, 'refs/tags/') }}
|
||||||
- name: Upload wheels
|
- name: Upload wheels
|
||||||
uses: actions/upload-artifact@v3
|
uses: actions/upload-artifact@v5
|
||||||
with:
|
with:
|
||||||
name: wheels
|
name: wheels-macos-${{ matrix.platform.target }}
|
||||||
path: dist
|
path: dist
|
||||||
|
|
||||||
sdist:
|
sdist:
|
||||||
runs-on: ubuntu-latest
|
runs-on: ubuntu-latest
|
||||||
steps:
|
steps:
|
||||||
- uses: actions/checkout@v3
|
- uses: actions/checkout@v6
|
||||||
- name: Build sdist
|
- name: Build sdist
|
||||||
uses: PyO3/maturin-action@v1
|
uses: PyO3/maturin-action@v1
|
||||||
with:
|
with:
|
||||||
command: sdist
|
command: sdist
|
||||||
args: -m cmdapp/Cargo.toml --out dist
|
args: --out dist --manifest-path cmdapp/Cargo.toml
|
||||||
- name: Upload sdist
|
- name: Upload sdist
|
||||||
uses: actions/upload-artifact@v3
|
uses: actions/upload-artifact@v5
|
||||||
with:
|
with:
|
||||||
name: wheels
|
name: wheels-sdist
|
||||||
path: dist
|
path: dist
|
||||||
|
|
||||||
release:
|
release:
|
||||||
@@ -104,18 +160,24 @@ jobs:
|
|||||||
runs-on: ubuntu-latest
|
runs-on: ubuntu-latest
|
||||||
# Specifying a GitHub environment is optional, but strongly encouraged
|
# Specifying a GitHub environment is optional, but strongly encouraged
|
||||||
environment: python
|
environment: python
|
||||||
|
if: ${{ startsWith(github.ref, 'refs/tags/') || github.event_name == 'workflow_dispatch' }}
|
||||||
|
needs: [linux, musllinux, windows, macos, sdist]
|
||||||
permissions:
|
permissions:
|
||||||
# IMPORTANT: this permission is mandatory for trusted publishing
|
# Use to sign the release artifacts
|
||||||
id-token: write
|
id-token: write
|
||||||
if: "startsWith(github.ref, 'refs/tags/')"
|
# Used to upload release artifacts
|
||||||
needs: [windows, macos, sdist, linux]
|
contents: write
|
||||||
|
# Used to generate artifact attestation
|
||||||
|
attestations: write
|
||||||
steps:
|
steps:
|
||||||
- uses: actions/download-artifact@v3
|
- uses: actions/download-artifact@v6
|
||||||
|
- name: Generate artifact attestation
|
||||||
|
uses: actions/attest-build-provenance@v3
|
||||||
with:
|
with:
|
||||||
name: wheels
|
subject-path: 'wheels-*/*'
|
||||||
|
- name: Install uv
|
||||||
|
if: ${{ startsWith(github.ref, 'refs/tags/') }}
|
||||||
|
uses: astral-sh/setup-uv@v7
|
||||||
- name: Publish to PyPI
|
- name: Publish to PyPI
|
||||||
uses: PyO3/maturin-action@v1
|
if: ${{ startsWith(github.ref, 'refs/tags/') }}
|
||||||
with:
|
run: uv publish 'wheels-*/*'
|
||||||
command: upload
|
|
||||||
args: --non-interactive --skip-existing *
|
|
||||||
# manylinux: auto
|
|
||||||
|
|||||||
@@ -1,10 +1,23 @@
|
|||||||
name: Rust
|
name: Rust
|
||||||
|
|
||||||
on:
|
on:
|
||||||
push:
|
|
||||||
branches: [ master ]
|
|
||||||
pull_request:
|
pull_request:
|
||||||
branches: [ master ]
|
paths-ignore:
|
||||||
|
- '**.md'
|
||||||
|
- '.github/ISSUE_TEMPLATE/**'
|
||||||
|
push:
|
||||||
|
paths-ignore:
|
||||||
|
- '**.md'
|
||||||
|
- '.github/ISSUE_TEMPLATE/**'
|
||||||
|
branches:
|
||||||
|
- master
|
||||||
|
- 0.*.x
|
||||||
|
- pr/**/ci
|
||||||
|
- ci-*
|
||||||
|
|
||||||
|
concurrency:
|
||||||
|
group: ${{ github.workflow }}-${{ github.head_ref || github.ref || github.run_id }}
|
||||||
|
cancel-in-progress: true
|
||||||
|
|
||||||
env:
|
env:
|
||||||
CARGO_TERM_COLOR: always
|
CARGO_TERM_COLOR: always
|
||||||
|
|||||||
@@ -5,6 +5,14 @@ All notable changes to this project will be documented in this file.
|
|||||||
The format is based on [Keep a Changelog](http://keepachangelog.com/)
|
The format is based on [Keep a Changelog](http://keepachangelog.com/)
|
||||||
and this project adheres to [Semantic Versioning](http://semver.org/).
|
and this project adheres to [Semantic Versioning](http://semver.org/).
|
||||||
|
|
||||||
|
## 0.6.12 - 2026-02-04
|
||||||
|
|
||||||
|
* Python Binding
|
||||||
|
|
||||||
|
## 0.6.5 - 2025-10-17
|
||||||
|
|
||||||
|
* Update `fastrand` to `2.3`
|
||||||
|
|
||||||
## 0.6.4 - 2024-03-29
|
## 0.6.4 - 2024-03-29
|
||||||
|
|
||||||
* Update `visioncortex` version to `0.8.8`
|
* Update `visioncortex` version to `0.8.8`
|
||||||
|
|||||||
@@ -1,4 +1,4 @@
|
|||||||
Copyright (c) 2023 Tsang Hao Fung
|
Copyright (c) 2024 TSANG, Hao Fung
|
||||||
|
|
||||||
Permission is hereby granted, free of charge, to any
|
Permission is hereby granted, free of charge, to any
|
||||||
person obtaining a copy of this software and associated
|
person obtaining a copy of this software and associated
|
||||||
|
|||||||
@@ -4,15 +4,15 @@
|
|||||||
<h1>VTracer</h1>
|
<h1>VTracer</h1>
|
||||||
|
|
||||||
<p>
|
<p>
|
||||||
<strong>Raster to Vector Graphics Converter built on top of visioncortex</strong>
|
<strong>Raster to Vector Graphics Converter</strong>
|
||||||
</p>
|
</p>
|
||||||
|
|
||||||
<h3>
|
<h3>
|
||||||
<a href="https://www.visioncortex.org/vtracer-docs">Article</a>
|
<a href="https://www.visioncortex.org/vtracer-docs">Article</a>
|
||||||
<span> | </span>
|
<span> | </span>
|
||||||
<a href="https://www.visioncortex.org/vtracer/">Demo</a>
|
<a href="https://www.visioncortex.org/vtracer/">Web App</a>
|
||||||
<span> | </span>
|
<span> | </span>
|
||||||
<a href="https://github.com/visioncortex/vtracer/releases/tag/0.6.0">Download</a>
|
<a href="https://github.com/visioncortex/vtracer/releases">Download</a>
|
||||||
</h3>
|
</h3>
|
||||||
|
|
||||||
<sub>Built with 🦀 by <a href="https://www.visioncortex.org/">The Vision Cortex Research Group</a></sub>
|
<sub>Built with 🦀 by <a href="https://www.visioncortex.org/">The Vision Cortex Research Group</a></sub>
|
||||||
@@ -28,7 +28,7 @@ Comparing to Adobe Illustrator's [Image Trace](https://helpx.adobe.com/illustrat
|
|||||||
|
|
||||||
VTracer is originally designed for processing high resolution scans of historic blueprints up to gigapixels. At the same time, VTracer can also handle low resolution pixel art, simulating `image-rendering: pixelated` for retro game artworks.
|
VTracer is originally designed for processing high resolution scans of historic blueprints up to gigapixels. At the same time, VTracer can also handle low resolution pixel art, simulating `image-rendering: pixelated` for retro game artworks.
|
||||||
|
|
||||||
A technical description of the algorithm is on [visioncortex.org/vtracer-docs](https://www.visioncortex.org/vtracer-docs).
|
Technical descriptions of the [tracing algorithm](https://www.visioncortex.org/vtracer-docs) and [clustering algorithm](https://www.visioncortex.org/impression-docs).
|
||||||
|
|
||||||
## Web App
|
## Web App
|
||||||
|
|
||||||
@@ -71,9 +71,9 @@ OPTIONS:
|
|||||||
-s, --splice_threshold <splice_threshold> Minimum angle displacement (degree) to splice a spline
|
-s, --splice_threshold <splice_threshold> Minimum angle displacement (degree) to splice a spline
|
||||||
```
|
```
|
||||||
|
|
||||||
### Install
|
## Downloads
|
||||||
|
|
||||||
You can download pre-built binaries from [Releases](https://github.com/visioncortex/vtracer/releases/tag/0.6.0).
|
You can download pre-built binaries from [Releases](https://github.com/visioncortex/vtracer/releases).
|
||||||
|
|
||||||
You can also install the program from source from [crates.io/vtracer](https://crates.io/crates/vtracer):
|
You can also install the program from source from [crates.io/vtracer](https://crates.io/crates/vtracer):
|
||||||
|
|
||||||
@@ -81,13 +81,15 @@ You can also install the program from source from [crates.io/vtracer](https://cr
|
|||||||
cargo install vtracer
|
cargo install vtracer
|
||||||
```
|
```
|
||||||
|
|
||||||
|
> You are strongly advised to not download from any other third-party sources
|
||||||
|
|
||||||
### Usage
|
### Usage
|
||||||
|
|
||||||
```sh
|
```sh
|
||||||
./vtracer --input input.jpg --output output.svg
|
./vtracer --input input.jpg --output output.svg
|
||||||
```
|
```
|
||||||
|
|
||||||
## Rust Library
|
### Rust Library
|
||||||
|
|
||||||
You can install [`vtracer`](https://crates.io/crates/vtracer) as a Rust library.
|
You can install [`vtracer`](https://crates.io/crates/vtracer) as a Rust library.
|
||||||
|
|
||||||
@@ -95,7 +97,7 @@ You can install [`vtracer`](https://crates.io/crates/vtracer) as a Rust library.
|
|||||||
cargo add vtracer
|
cargo add vtracer
|
||||||
```
|
```
|
||||||
|
|
||||||
## Python Library
|
### Python Library
|
||||||
|
|
||||||
Since `0.6`, [`vtracer`](https://pypi.org/project/vtracer/) is also packaged as Python native extensions, thanks to the awesome [pyo3](https://github.com/PyO3/pyo3) project.
|
Since `0.6`, [`vtracer`](https://pypi.org/project/vtracer/) is also packaged as Python native extensions, thanks to the awesome [pyo3](https://github.com/PyO3/pyo3) project.
|
||||||
|
|
||||||
@@ -105,7 +107,7 @@ pip install vtracer
|
|||||||
|
|
||||||
## In the wild
|
## In the wild
|
||||||
|
|
||||||
VTracer is used by the following projects (feel free to add yours!):
|
VTracer is used by the following products (open a PR to add yours):
|
||||||
|
|
||||||
<table>
|
<table>
|
||||||
<tbody>
|
<tbody>
|
||||||
@@ -118,12 +120,20 @@ VTracer is used by the following projects (feel free to add yours!):
|
|||||||
</tbody>
|
</tbody>
|
||||||
</table>
|
</table>
|
||||||
|
|
||||||
## Anecdotes
|
## Citations
|
||||||
|
|
||||||
|
VTracer has since been cited by a few academic papers in computer graphics / vision research. Please kindly let us know if you have cited our work:
|
||||||
|
|
||||||
|
+ SKILL 2023 [Framework to Vectorize Digital Artworks for Physical Fabrication based on Geometric Stylization Techniques](https://www.researchgate.net/publication/374448489_Framework_to_Vectorize_Digital_Artworks_for_Physical_Fabrication_based_on_Geometric_Stylization_Techniques)
|
||||||
|
+ arXiv 2023 [Image Vectorization: a Review](https://arxiv.org/abs/2306.06441)
|
||||||
|
+ arXiv 2023 [StarVector: Generating Scalable Vector Graphics Code from Images](https://arxiv.org/abs/2312.11556)
|
||||||
|
+ arXiv 2024 [Text-Based Reasoning About Vector Graphics](https://arxiv.org/abs/2404.06479)
|
||||||
|
+ arXiv 2024 [Delving into LLMs' visual understanding ability using SVG to bridge image and text](https://openreview.net/pdf?id=pwlm6Po61I)
|
||||||
|
|
||||||
|
## How did VTracer come about?
|
||||||
|
|
||||||
> The following content is an excerpt from my [unpublished memoir](https://github.com/visioncortex/memoir).
|
> The following content is an excerpt from my [unpublished memoir](https://github.com/visioncortex/memoir).
|
||||||
|
|
||||||
### How / when / where did VTracer come about?
|
|
||||||
|
|
||||||
At my teenage, two open source projects in the vector graphics space inspired me the most: Potrace and Anti-Grain Geometry (AGG).
|
At my teenage, two open source projects in the vector graphics space inspired me the most: Potrace and Anti-Grain Geometry (AGG).
|
||||||
|
|
||||||
Many years later, in 2020, I was developing a video processing engine. And it became evident that it requires way more investment to be commercially viable. So before abandoning the project, I wanted to publish *something* as open-source for posterity. At that time, I already developed a prototype vector graphics tracer. It can convert high-resolution scans of hand-drawn blueprints into vectors. But it can only process black and white images, and can only output polygons, not splines.
|
Many years later, in 2020, I was developing a video processing engine. And it became evident that it requires way more investment to be commercially viable. So before abandoning the project, I wanted to publish *something* as open-source for posterity. At that time, I already developed a prototype vector graphics tracer. It can convert high-resolution scans of hand-drawn blueprints into vectors. But it can only process black and white images, and can only output polygons, not splines.
|
||||||
|
|||||||
+4
-3
@@ -1,10 +1,10 @@
|
|||||||
[package]
|
[package]
|
||||||
name = "vtracer"
|
name = "vtracer"
|
||||||
version = "0.6.4"
|
version = "0.6.12"
|
||||||
authors = ["Chris Tsang <chris.2y3@outlook.com>"]
|
authors = ["Chris Tsang <chris.2y3@outlook.com>"]
|
||||||
edition = "2021"
|
edition = "2021"
|
||||||
description = "A cmd app to convert images into vector graphics."
|
description = "A cmd app to convert images into vector graphics."
|
||||||
license = "MIT OR Apache-2.0"
|
license = "MIT"
|
||||||
homepage = "http://www.visioncortex.org/vtracer"
|
homepage = "http://www.visioncortex.org/vtracer"
|
||||||
repository = "https://github.com/visioncortex/vtracer/"
|
repository = "https://github.com/visioncortex/vtracer/"
|
||||||
categories = ["graphics"]
|
categories = ["graphics"]
|
||||||
@@ -14,11 +14,12 @@ keywords = ["svg", "computer-graphics"]
|
|||||||
clap = "2.33.3"
|
clap = "2.33.3"
|
||||||
image = "0.23.10"
|
image = "0.23.10"
|
||||||
visioncortex = { version = "0.8.8" }
|
visioncortex = { version = "0.8.8" }
|
||||||
fastrand = "1.8"
|
fastrand = { version = "2.3" }
|
||||||
pyo3 = { version = "0.19.0", optional = true }
|
pyo3 = { version = "0.19.0", optional = true }
|
||||||
|
|
||||||
[features]
|
[features]
|
||||||
python-binding = ["pyo3"]
|
python-binding = ["pyo3"]
|
||||||
|
wasm = ["fastrand/js"]
|
||||||
|
|
||||||
[lib]
|
[lib]
|
||||||
name = "vtracer"
|
name = "vtracer"
|
||||||
|
|||||||
@@ -1,4 +1,4 @@
|
|||||||
Copyright (c) 2023 Tsang Hao Fung
|
Copyright (c) 2024 TSANG, Hao Fung
|
||||||
|
|
||||||
Permission is hereby granted, free of charge, to any
|
Permission is hereby granted, free of charge, to any
|
||||||
person obtaining a copy of this software and associated
|
person obtaining a copy of this software and associated
|
||||||
@@ -1,201 +0,0 @@
|
|||||||
Apache License
|
|
||||||
Version 2.0, January 2004
|
|
||||||
http://www.apache.org/licenses/
|
|
||||||
|
|
||||||
TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
|
|
||||||
|
|
||||||
1. Definitions.
|
|
||||||
|
|
||||||
"License" shall mean the terms and conditions for use, reproduction,
|
|
||||||
and distribution as defined by Sections 1 through 9 of this document.
|
|
||||||
|
|
||||||
"Licensor" shall mean the copyright owner or entity authorized by
|
|
||||||
the copyright owner that is granting the License.
|
|
||||||
|
|
||||||
"Legal Entity" shall mean the union of the acting entity and all
|
|
||||||
other entities that control, are controlled by, or are under common
|
|
||||||
control with that entity. For the purposes of this definition,
|
|
||||||
"control" means (i) the power, direct or indirect, to cause the
|
|
||||||
direction or management of such entity, whether by contract or
|
|
||||||
otherwise, or (ii) ownership of fifty percent (50%) or more of the
|
|
||||||
outstanding shares, or (iii) beneficial ownership of such entity.
|
|
||||||
|
|
||||||
"You" (or "Your") shall mean an individual or Legal Entity
|
|
||||||
exercising permissions granted by this License.
|
|
||||||
|
|
||||||
"Source" form shall mean the preferred form for making modifications,
|
|
||||||
including but not limited to software source code, documentation
|
|
||||||
source, and configuration files.
|
|
||||||
|
|
||||||
"Object" form shall mean any form resulting from mechanical
|
|
||||||
transformation or translation of a Source form, including but
|
|
||||||
not limited to compiled object code, generated documentation,
|
|
||||||
and conversions to other media types.
|
|
||||||
|
|
||||||
"Work" shall mean the work of authorship, whether in Source or
|
|
||||||
Object form, made available under the License, as indicated by a
|
|
||||||
copyright notice that is included in or attached to the work
|
|
||||||
(an example is provided in the Appendix below).
|
|
||||||
|
|
||||||
"Derivative Works" shall mean any work, whether in Source or Object
|
|
||||||
form, that is based on (or derived from) the Work and for which the
|
|
||||||
editorial revisions, annotations, elaborations, or other modifications
|
|
||||||
represent, as a whole, an original work of authorship. For the purposes
|
|
||||||
of this License, Derivative Works shall not include works that remain
|
|
||||||
separable from, or merely link (or bind by name) to the interfaces of,
|
|
||||||
the Work and Derivative Works thereof.
|
|
||||||
|
|
||||||
"Contribution" shall mean any work of authorship, including
|
|
||||||
the original version of the Work and any modifications or additions
|
|
||||||
to that Work or Derivative Works thereof, that is intentionally
|
|
||||||
submitted to Licensor for inclusion in the Work by the copyright owner
|
|
||||||
or by an individual or Legal Entity authorized to submit on behalf of
|
|
||||||
the copyright owner. For the purposes of this definition, "submitted"
|
|
||||||
means any form of electronic, verbal, or written communication sent
|
|
||||||
to the Licensor or its representatives, including but not limited to
|
|
||||||
communication on electronic mailing lists, source code control systems,
|
|
||||||
and issue tracking systems that are managed by, or on behalf of, the
|
|
||||||
Licensor for the purpose of discussing and improving the Work, but
|
|
||||||
excluding communication that is conspicuously marked or otherwise
|
|
||||||
designated in writing by the copyright owner as "Not a Contribution."
|
|
||||||
|
|
||||||
"Contributor" shall mean Licensor and any individual or Legal Entity
|
|
||||||
on behalf of whom a Contribution has been received by Licensor and
|
|
||||||
subsequently incorporated within the Work.
|
|
||||||
|
|
||||||
2. Grant of Copyright License. Subject to the terms and conditions of
|
|
||||||
this License, each Contributor hereby grants to You a perpetual,
|
|
||||||
worldwide, non-exclusive, no-charge, royalty-free, irrevocable
|
|
||||||
copyright license to reproduce, prepare Derivative Works of,
|
|
||||||
publicly display, publicly perform, sublicense, and distribute the
|
|
||||||
Work and such Derivative Works in Source or Object form.
|
|
||||||
|
|
||||||
3. Grant of Patent License. Subject to the terms and conditions of
|
|
||||||
this License, each Contributor hereby grants to You a perpetual,
|
|
||||||
worldwide, non-exclusive, no-charge, royalty-free, irrevocable
|
|
||||||
(except as stated in this section) patent license to make, have made,
|
|
||||||
use, offer to sell, sell, import, and otherwise transfer the Work,
|
|
||||||
where such license applies only to those patent claims licensable
|
|
||||||
by such Contributor that are necessarily infringed by their
|
|
||||||
Contribution(s) alone or by combination of their Contribution(s)
|
|
||||||
with the Work to which such Contribution(s) was submitted. If You
|
|
||||||
institute patent litigation against any entity (including a
|
|
||||||
cross-claim or counterclaim in a lawsuit) alleging that the Work
|
|
||||||
or a Contribution incorporated within the Work constitutes direct
|
|
||||||
or contributory patent infringement, then any patent licenses
|
|
||||||
granted to You under this License for that Work shall terminate
|
|
||||||
as of the date such litigation is filed.
|
|
||||||
|
|
||||||
4. Redistribution. You may reproduce and distribute copies of the
|
|
||||||
Work or Derivative Works thereof in any medium, with or without
|
|
||||||
modifications, and in Source or Object form, provided that You
|
|
||||||
meet the following conditions:
|
|
||||||
|
|
||||||
(a) You must give any other recipients of the Work or
|
|
||||||
Derivative Works a copy of this License; and
|
|
||||||
|
|
||||||
(b) You must cause any modified files to carry prominent notices
|
|
||||||
stating that You changed the files; and
|
|
||||||
|
|
||||||
(c) You must retain, in the Source form of any Derivative Works
|
|
||||||
that You distribute, all copyright, patent, trademark, and
|
|
||||||
attribution notices from the Source form of the Work,
|
|
||||||
excluding those notices that do not pertain to any part of
|
|
||||||
the Derivative Works; and
|
|
||||||
|
|
||||||
(d) If the Work includes a "NOTICE" text file as part of its
|
|
||||||
distribution, then any Derivative Works that You distribute must
|
|
||||||
include a readable copy of the attribution notices contained
|
|
||||||
within such NOTICE file, excluding those notices that do not
|
|
||||||
pertain to any part of the Derivative Works, in at least one
|
|
||||||
of the following places: within a NOTICE text file distributed
|
|
||||||
as part of the Derivative Works; within the Source form or
|
|
||||||
documentation, if provided along with the Derivative Works; or,
|
|
||||||
within a display generated by the Derivative Works, if and
|
|
||||||
wherever such third-party notices normally appear. The contents
|
|
||||||
of the NOTICE file are for informational purposes only and
|
|
||||||
do not modify the License. You may add Your own attribution
|
|
||||||
notices within Derivative Works that You distribute, alongside
|
|
||||||
or as an addendum to the NOTICE text from the Work, provided
|
|
||||||
that such additional attribution notices cannot be construed
|
|
||||||
as modifying the License.
|
|
||||||
|
|
||||||
You may add Your own copyright statement to Your modifications and
|
|
||||||
may provide additional or different license terms and conditions
|
|
||||||
for use, reproduction, or distribution of Your modifications, or
|
|
||||||
for any such Derivative Works as a whole, provided Your use,
|
|
||||||
reproduction, and distribution of the Work otherwise complies with
|
|
||||||
the conditions stated in this License.
|
|
||||||
|
|
||||||
5. Submission of Contributions. Unless You explicitly state otherwise,
|
|
||||||
any Contribution intentionally submitted for inclusion in the Work
|
|
||||||
by You to the Licensor shall be under the terms and conditions of
|
|
||||||
this License, without any additional terms or conditions.
|
|
||||||
Notwithstanding the above, nothing herein shall supersede or modify
|
|
||||||
the terms of any separate license agreement you may have executed
|
|
||||||
with Licensor regarding such Contributions.
|
|
||||||
|
|
||||||
6. Trademarks. This License does not grant permission to use the trade
|
|
||||||
names, trademarks, service marks, or product names of the Licensor,
|
|
||||||
except as required for reasonable and customary use in describing the
|
|
||||||
origin of the Work and reproducing the content of the NOTICE file.
|
|
||||||
|
|
||||||
7. Disclaimer of Warranty. Unless required by applicable law or
|
|
||||||
agreed to in writing, Licensor provides the Work (and each
|
|
||||||
Contributor provides its Contributions) on an "AS IS" BASIS,
|
|
||||||
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
|
|
||||||
implied, including, without limitation, any warranties or conditions
|
|
||||||
of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
|
|
||||||
PARTICULAR PURPOSE. You are solely responsible for determining the
|
|
||||||
appropriateness of using or redistributing the Work and assume any
|
|
||||||
risks associated with Your exercise of permissions under this License.
|
|
||||||
|
|
||||||
8. Limitation of Liability. In no event and under no legal theory,
|
|
||||||
whether in tort (including negligence), contract, or otherwise,
|
|
||||||
unless required by applicable law (such as deliberate and grossly
|
|
||||||
negligent acts) or agreed to in writing, shall any Contributor be
|
|
||||||
liable to You for damages, including any direct, indirect, special,
|
|
||||||
incidental, or consequential damages of any character arising as a
|
|
||||||
result of this License or out of the use or inability to use the
|
|
||||||
Work (including but not limited to damages for loss of goodwill,
|
|
||||||
work stoppage, computer failure or malfunction, or any and all
|
|
||||||
other commercial damages or losses), even if such Contributor
|
|
||||||
has been advised of the possibility of such damages.
|
|
||||||
|
|
||||||
9. Accepting Warranty or Additional Liability. While redistributing
|
|
||||||
the Work or Derivative Works thereof, You may choose to offer,
|
|
||||||
and charge a fee for, acceptance of support, warranty, indemnity,
|
|
||||||
or other liability obligations and/or rights consistent with this
|
|
||||||
License. However, in accepting such obligations, You may act only
|
|
||||||
on Your own behalf and on Your sole responsibility, not on behalf
|
|
||||||
of any other Contributor, and only if You agree to indemnify,
|
|
||||||
defend, and hold each Contributor harmless for any liability
|
|
||||||
incurred by, or claims asserted against, such Contributor by reason
|
|
||||||
of your accepting any such warranty or additional liability.
|
|
||||||
|
|
||||||
END OF TERMS AND CONDITIONS
|
|
||||||
|
|
||||||
APPENDIX: How to apply the Apache License to your work.
|
|
||||||
|
|
||||||
To apply the Apache License to your work, attach the following
|
|
||||||
boilerplate notice, with the fields enclosed by brackets "[]"
|
|
||||||
replaced with your own identifying information. (Don't include
|
|
||||||
the brackets!) The text should be enclosed in the appropriate
|
|
||||||
comment syntax for the file format. We also recommend that a
|
|
||||||
file or class name and description of purpose be included on the
|
|
||||||
same "printed page" as the copyright notice for easier
|
|
||||||
identification within third-party archives.
|
|
||||||
|
|
||||||
Copyright [yyyy] [name of copyright owner]
|
|
||||||
|
|
||||||
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.
|
|
||||||
@@ -1,11 +1,12 @@
|
|||||||
[project]
|
[project]
|
||||||
name = "vtracer"
|
name = "vtracer"
|
||||||
version = "0.6.10"
|
version = "0.6.15"
|
||||||
description = "Python bindings for the Rust Vtracer raster-to-vector library"
|
description = "Python bindings for the Rust Vtracer raster-to-vector library"
|
||||||
authors = [ { name = "Chris Tsang", email = "chris.2y3@outlook.com" } ]
|
authors = [ { name = "Chris Tsang", email = "chris.2y3@outlook.com" } ]
|
||||||
readme = "vtracer/README.md"
|
readme = "vtracer/README.md"
|
||||||
requires-python = ">=3.7"
|
requires-python = ">=3.7"
|
||||||
license = "MIT"
|
license = "MIT"
|
||||||
|
license-files = ["LICENSE"]
|
||||||
classifiers = [
|
classifiers = [
|
||||||
"Programming Language :: Rust",
|
"Programming Language :: Rust",
|
||||||
"Programming Language :: Python :: Implementation :: CPython",
|
"Programming Language :: Python :: Implementation :: CPython",
|
||||||
@@ -25,4 +26,4 @@ build-backend = "maturin"
|
|||||||
[tool.maturin]
|
[tool.maturin]
|
||||||
features = ["pyo3/extension-module", "python-binding"]
|
features = ["pyo3/extension-module", "python-binding"]
|
||||||
compatibility = "manylinux2014"
|
compatibility = "manylinux2014"
|
||||||
sdist-include = ["LICENSE-MIT", "vtracer/README.md"]
|
include = ["LICENSE"]
|
||||||
|
|||||||
@@ -1,23 +1,27 @@
|
|||||||
use std::str::FromStr;
|
use std::str::FromStr;
|
||||||
use visioncortex::PathSimplifyMode;
|
use visioncortex::PathSimplifyMode;
|
||||||
|
|
||||||
|
#[derive(Debug, Clone)]
|
||||||
pub enum Preset {
|
pub enum Preset {
|
||||||
Bw,
|
Bw,
|
||||||
Poster,
|
Poster,
|
||||||
Photo,
|
Photo,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[derive(Debug, Clone)]
|
||||||
pub enum ColorMode {
|
pub enum ColorMode {
|
||||||
Color,
|
Color,
|
||||||
Binary,
|
Binary,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[derive(Debug, Clone)]
|
||||||
pub enum Hierarchical {
|
pub enum Hierarchical {
|
||||||
Stacked,
|
Stacked,
|
||||||
Cutout,
|
Cutout,
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Converter config
|
/// Converter config
|
||||||
|
#[derive(Debug, Clone)]
|
||||||
pub struct Config {
|
pub struct Config {
|
||||||
pub color_mode: ColorMode,
|
pub color_mode: ColorMode,
|
||||||
pub hierarchical: Hierarchical,
|
pub hierarchical: Hierarchical,
|
||||||
@@ -32,6 +36,7 @@ pub struct Config {
|
|||||||
pub path_precision: Option<u32>,
|
pub path_precision: Option<u32>,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[derive(Debug, Clone)]
|
||||||
pub(crate) struct ConverterConfig {
|
pub(crate) struct ConverterConfig {
|
||||||
pub color_mode: ColorMode,
|
pub color_mode: ColorMode,
|
||||||
pub hierarchical: Hierarchical,
|
pub hierarchical: Hierarchical,
|
||||||
|
|||||||
@@ -53,7 +53,7 @@ fn find_unused_color_in_image(img: &ColorImage) -> Result<Color, String> {
|
|||||||
Color::new(0, 255, 255),
|
Color::new(0, 255, 255),
|
||||||
Color::new(255, 0, 255),
|
Color::new(255, 0, 255),
|
||||||
]);
|
]);
|
||||||
let rng = Rng::new();
|
let mut rng = Rng::new();
|
||||||
let random_colors =
|
let random_colors =
|
||||||
(0..NUM_UNUSED_COLOR_ITERATIONS).map(|_| Color::new(rng.u8(..), rng.u8(..), rng.u8(..)));
|
(0..NUM_UNUSED_COLOR_ITERATIONS).map(|_| Color::new(rng.u8(..), rng.u8(..), rng.u8(..)));
|
||||||
for color in special_colors.chain(random_colors) {
|
for color in special_colors.chain(random_colors) {
|
||||||
@@ -73,7 +73,7 @@ fn should_key_image(img: &ColorImage) -> bool {
|
|||||||
|
|
||||||
// Check for transparency at several scanlines
|
// Check for transparency at several scanlines
|
||||||
let threshold = ((img.width * 2) as f32 * KEYING_THRESHOLD) as usize;
|
let threshold = ((img.width * 2) as f32 * KEYING_THRESHOLD) as usize;
|
||||||
let mut num_transparent_boundary_pixels = 0;
|
let mut num_transparent_pixels = 0;
|
||||||
let y_positions = [
|
let y_positions = [
|
||||||
0,
|
0,
|
||||||
img.height / 4,
|
img.height / 4,
|
||||||
@@ -84,9 +84,9 @@ fn should_key_image(img: &ColorImage) -> bool {
|
|||||||
for y in y_positions {
|
for y in y_positions {
|
||||||
for x in 0..img.width {
|
for x in 0..img.width {
|
||||||
if img.get_pixel(x, y).a == 0 {
|
if img.get_pixel(x, y).a == 0 {
|
||||||
num_transparent_boundary_pixels += 1;
|
num_transparent_pixels += 1;
|
||||||
}
|
}
|
||||||
if num_transparent_boundary_pixels >= threshold {
|
if num_transparent_pixels >= threshold {
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
+149
-6
@@ -1,5 +1,7 @@
|
|||||||
use crate::*;
|
use crate::*;
|
||||||
use pyo3::prelude::*;
|
use image::{io::Reader, ImageFormat};
|
||||||
|
use pyo3::{exceptions::PyException, prelude::*};
|
||||||
|
use std::io::{BufReader, Cursor};
|
||||||
use std::path::PathBuf;
|
use std::path::PathBuf;
|
||||||
use visioncortex::PathSimplifyMode;
|
use visioncortex::PathSimplifyMode;
|
||||||
|
|
||||||
@@ -23,6 +25,148 @@ fn convert_image_to_svg_py(
|
|||||||
let input_path = PathBuf::from(image_path);
|
let input_path = PathBuf::from(image_path);
|
||||||
let output_path = PathBuf::from(out_path);
|
let output_path = PathBuf::from(out_path);
|
||||||
|
|
||||||
|
let config = construct_config(
|
||||||
|
colormode,
|
||||||
|
hierarchical,
|
||||||
|
mode,
|
||||||
|
filter_speckle,
|
||||||
|
color_precision,
|
||||||
|
layer_difference,
|
||||||
|
corner_threshold,
|
||||||
|
length_threshold,
|
||||||
|
max_iterations,
|
||||||
|
splice_threshold,
|
||||||
|
path_precision,
|
||||||
|
);
|
||||||
|
|
||||||
|
convert_image_to_svg(&input_path, &output_path, config).unwrap();
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
#[pyfunction]
|
||||||
|
fn convert_raw_image_to_svg(
|
||||||
|
img_bytes: Vec<u8>,
|
||||||
|
img_format: Option<&str>, // Format of the image (e.g. 'jpg', 'png'... A full list of supported formats can be found [here](https://docs.rs/image/latest/image/enum.ImageFormat.html)). If not provided, the image format will be guessed based on its contents.
|
||||||
|
colormode: Option<&str>, // "color" or "binary"
|
||||||
|
hierarchical: Option<&str>, // "stacked" or "cutout"
|
||||||
|
mode: Option<&str>, // "polygon", "spline", "none"
|
||||||
|
filter_speckle: Option<usize>, // default: 4
|
||||||
|
color_precision: Option<i32>, // default: 6
|
||||||
|
layer_difference: Option<i32>, // default: 16
|
||||||
|
corner_threshold: Option<i32>, // default: 60
|
||||||
|
length_threshold: Option<f64>, // in [3.5, 10] default: 4.0
|
||||||
|
max_iterations: Option<usize>, // default: 10
|
||||||
|
splice_threshold: Option<i32>, // default: 45
|
||||||
|
path_precision: Option<u32>, // default: 8
|
||||||
|
) -> PyResult<String> {
|
||||||
|
let config = construct_config(
|
||||||
|
colormode,
|
||||||
|
hierarchical,
|
||||||
|
mode,
|
||||||
|
filter_speckle,
|
||||||
|
color_precision,
|
||||||
|
layer_difference,
|
||||||
|
corner_threshold,
|
||||||
|
length_threshold,
|
||||||
|
max_iterations,
|
||||||
|
splice_threshold,
|
||||||
|
path_precision,
|
||||||
|
);
|
||||||
|
let mut img_reader = Reader::new(BufReader::new(Cursor::new(img_bytes)));
|
||||||
|
let img_format = img_format.and_then(|ext_name| ImageFormat::from_extension(ext_name));
|
||||||
|
let img = match img_format {
|
||||||
|
Some(img_format) => {
|
||||||
|
img_reader.set_format(img_format);
|
||||||
|
img_reader.decode()
|
||||||
|
}
|
||||||
|
None => img_reader
|
||||||
|
.with_guessed_format()
|
||||||
|
.map_err(|_| PyException::new_err("Unrecognized image format. "))?
|
||||||
|
.decode(),
|
||||||
|
};
|
||||||
|
let img = match img {
|
||||||
|
Ok(img) => img.to_rgba8(),
|
||||||
|
Err(_) => return Err(PyException::new_err("Failed to decode img_bytes. ")),
|
||||||
|
};
|
||||||
|
let (width, height) = (img.width() as usize, img.height() as usize);
|
||||||
|
let img = ColorImage {
|
||||||
|
pixels: img.as_raw().to_vec(),
|
||||||
|
width,
|
||||||
|
height,
|
||||||
|
};
|
||||||
|
let svg =
|
||||||
|
convert(img, config).map_err(|_| PyException::new_err("Failed to convert the image. "))?;
|
||||||
|
Ok(format!("{}", svg))
|
||||||
|
}
|
||||||
|
|
||||||
|
#[pyfunction]
|
||||||
|
fn convert_pixels_to_svg(
|
||||||
|
rgba_pixels: Vec<(u8, u8, u8, u8)>,
|
||||||
|
size: (usize, usize),
|
||||||
|
colormode: Option<&str>, // "color" or "binary"
|
||||||
|
hierarchical: Option<&str>, // "stacked" or "cutout"
|
||||||
|
mode: Option<&str>, // "polygon", "spline", "none"
|
||||||
|
filter_speckle: Option<usize>, // default: 4
|
||||||
|
color_precision: Option<i32>, // default: 6
|
||||||
|
layer_difference: Option<i32>, // default: 16
|
||||||
|
corner_threshold: Option<i32>, // default: 60
|
||||||
|
length_threshold: Option<f64>, // in [3.5, 10] default: 4.0
|
||||||
|
max_iterations: Option<usize>, // default: 10
|
||||||
|
splice_threshold: Option<i32>, // default: 45
|
||||||
|
path_precision: Option<u32>, // default: 8
|
||||||
|
) -> PyResult<String> {
|
||||||
|
let expected_pixel_count = size.0 * size.1;
|
||||||
|
if rgba_pixels.len() != expected_pixel_count {
|
||||||
|
return Err(PyException::new_err(format!(
|
||||||
|
"Length of rgba_pixels does not match given image size. Expected {} ({} * {}), got {}. ",
|
||||||
|
expected_pixel_count,
|
||||||
|
size.0,
|
||||||
|
size.1,
|
||||||
|
rgba_pixels.len()
|
||||||
|
)));
|
||||||
|
}
|
||||||
|
let config = construct_config(
|
||||||
|
colormode,
|
||||||
|
hierarchical,
|
||||||
|
mode,
|
||||||
|
filter_speckle,
|
||||||
|
color_precision,
|
||||||
|
layer_difference,
|
||||||
|
corner_threshold,
|
||||||
|
length_threshold,
|
||||||
|
max_iterations,
|
||||||
|
splice_threshold,
|
||||||
|
path_precision,
|
||||||
|
);
|
||||||
|
let mut flat_pixels: Vec<u8> = vec![];
|
||||||
|
for (r, g, b, a) in rgba_pixels {
|
||||||
|
flat_pixels.push(r);
|
||||||
|
flat_pixels.push(g);
|
||||||
|
flat_pixels.push(b);
|
||||||
|
flat_pixels.push(a);
|
||||||
|
}
|
||||||
|
let mut img = ColorImage::new();
|
||||||
|
img.pixels = flat_pixels;
|
||||||
|
(img.width, img.height) = size;
|
||||||
|
|
||||||
|
let svg =
|
||||||
|
convert(img, config).map_err(|_| PyException::new_err("Failed to convert the image. "))?;
|
||||||
|
Ok(format!("{}", svg))
|
||||||
|
}
|
||||||
|
|
||||||
|
fn construct_config(
|
||||||
|
colormode: Option<&str>,
|
||||||
|
hierarchical: Option<&str>,
|
||||||
|
mode: Option<&str>,
|
||||||
|
filter_speckle: Option<usize>,
|
||||||
|
color_precision: Option<i32>,
|
||||||
|
layer_difference: Option<i32>,
|
||||||
|
corner_threshold: Option<i32>,
|
||||||
|
length_threshold: Option<f64>,
|
||||||
|
max_iterations: Option<usize>,
|
||||||
|
splice_threshold: Option<i32>,
|
||||||
|
path_precision: Option<u32>,
|
||||||
|
) -> Config {
|
||||||
// TODO: enforce color mode with an enum so that we only
|
// TODO: enforce color mode with an enum so that we only
|
||||||
// accept the strings 'color' or 'binary'
|
// accept the strings 'color' or 'binary'
|
||||||
let color_mode = match colormode.unwrap_or("color") {
|
let color_mode = match colormode.unwrap_or("color") {
|
||||||
@@ -52,7 +196,7 @@ fn convert_image_to_svg_py(
|
|||||||
let splice_threshold = splice_threshold.unwrap_or(45);
|
let splice_threshold = splice_threshold.unwrap_or(45);
|
||||||
let max_iterations = max_iterations.unwrap_or(10);
|
let max_iterations = max_iterations.unwrap_or(10);
|
||||||
|
|
||||||
let config = Config {
|
Config {
|
||||||
color_mode,
|
color_mode,
|
||||||
hierarchical,
|
hierarchical,
|
||||||
filter_speckle,
|
filter_speckle,
|
||||||
@@ -65,15 +209,14 @@ fn convert_image_to_svg_py(
|
|||||||
splice_threshold,
|
splice_threshold,
|
||||||
path_precision,
|
path_precision,
|
||||||
..Default::default()
|
..Default::default()
|
||||||
};
|
}
|
||||||
|
|
||||||
convert_image_to_svg(&input_path, &output_path, config).unwrap();
|
|
||||||
Ok(())
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/// A Python module implemented in Rust.
|
/// A Python module implemented in Rust.
|
||||||
#[pymodule]
|
#[pymodule]
|
||||||
fn vtracer(_py: Python, m: &PyModule) -> PyResult<()> {
|
fn vtracer(_py: Python, m: &PyModule) -> PyResult<()> {
|
||||||
m.add_function(wrap_pyfunction!(convert_image_to_svg_py, m)?)?;
|
m.add_function(wrap_pyfunction!(convert_image_to_svg_py, m)?)?;
|
||||||
|
m.add_function(wrap_pyfunction!(convert_raw_image_to_svg, m)?)?;
|
||||||
|
m.add_function(wrap_pyfunction!(convert_pixels_to_svg, m)?)?;
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,6 +1,7 @@
|
|||||||
use std::fmt;
|
use std::fmt;
|
||||||
use visioncortex::{Color, CompoundPath, PointF64};
|
use visioncortex::{Color, CompoundPath, PointF64};
|
||||||
|
|
||||||
|
#[derive(Debug, Clone)]
|
||||||
pub struct SvgFile {
|
pub struct SvgFile {
|
||||||
pub paths: Vec<SvgPath>,
|
pub paths: Vec<SvgPath>,
|
||||||
pub width: usize,
|
pub width: usize,
|
||||||
@@ -8,6 +9,7 @@ pub struct SvgFile {
|
|||||||
pub path_precision: Option<u32>,
|
pub path_precision: Option<u32>,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[derive(Debug, Clone)]
|
||||||
pub struct SvgPath {
|
pub struct SvgPath {
|
||||||
pub path: CompoundPath,
|
pub path: CompoundPath,
|
||||||
pub color: Color,
|
pub color: Color,
|
||||||
|
|||||||
@@ -52,7 +52,17 @@ vtracer.convert_image_to_svg_py(inp, out)
|
|||||||
# Single-color example. Good for line art, and much faster than full color:
|
# Single-color example. Good for line art, and much faster than full color:
|
||||||
vtracer.convert_image_to_svg_py(inp, out, colormode='binary')
|
vtracer.convert_image_to_svg_py(inp, out, colormode='binary')
|
||||||
|
|
||||||
# All the bells & whistles
|
# Convert from raw image bytes
|
||||||
|
input_img_bytes: bytes = get_bytes() # e.g. reading bytes from a file or a HTTP request body
|
||||||
|
svg_str: str = vtracer.convert_raw_image_to_svg(input_img_bytes, img_format='jpg')
|
||||||
|
|
||||||
|
# Convert from RGBA image pixels
|
||||||
|
from PIL import Image
|
||||||
|
img = Image.open(input_path).convert('RGBA')
|
||||||
|
pixels: list[tuple[int, int, int, int]] = list(img.getdata())
|
||||||
|
svg_str: str = vtracer.convert_pixels_to_svg(pixels, img.size)
|
||||||
|
|
||||||
|
# All the bells & whistles, also applicable to convert_raw_image_to_svg and convert_pixels_to_svg.
|
||||||
vtracer.convert_image_to_svg_py(inp,
|
vtracer.convert_image_to_svg_py(inp,
|
||||||
out,
|
out,
|
||||||
colormode = 'color', # ["color"] or "binary"
|
colormode = 'color', # ["color"] or "binary"
|
||||||
|
|||||||
@@ -1 +1,2 @@
|
|||||||
from .vtracer import convert_image_to_svg_py
|
from .vtracer import (convert_image_to_svg_py, convert_pixels_to_svg,
|
||||||
|
convert_raw_image_to_svg)
|
||||||
|
|||||||
@@ -15,3 +15,35 @@ def convert_image_to_svg_py(image_path: str,
|
|||||||
path_precision: Optional[int] = None, # default: 8
|
path_precision: Optional[int] = None, # default: 8
|
||||||
) -> None:
|
) -> None:
|
||||||
...
|
...
|
||||||
|
|
||||||
|
def convert_raw_image_to_svg(img_bytes: bytes,
|
||||||
|
img_format: Optional[str] = None, # Format of the image (e.g. 'jpg', 'png'... A full list of supported formats can be found [here](https://docs.rs/image/latest/image/enum.ImageFormat.html)). If not provided, the image format will be guessed based on its contents.
|
||||||
|
colormode: Optional[str] = None, # ["color"] or "binary"
|
||||||
|
hierarchical: Optional[str] = None, # ["stacked"] or "cutout"
|
||||||
|
mode: Optional[str] = None, # ["spline"], "polygon", "none"
|
||||||
|
filter_speckle: Optional[int] = None, # default: 4
|
||||||
|
color_precision: Optional[int] = None, # default: 6
|
||||||
|
layer_difference: Optional[int] = None, # default: 16
|
||||||
|
corner_threshold: Optional[int] = None, # default: 60
|
||||||
|
length_threshold: Optional[float] = None, # in [3.5, 10] default: 4.0
|
||||||
|
max_iterations: Optional[int] = None, # default: 10
|
||||||
|
splice_threshold: Optional[int] = None, # default: 45
|
||||||
|
path_precision: Optional[int] = None, # default: 8
|
||||||
|
) -> str:
|
||||||
|
...
|
||||||
|
|
||||||
|
def convert_pixels_to_svg(rgba_pixels: list[tuple[int, int, int, int]],
|
||||||
|
size: tuple[int, int],
|
||||||
|
colormode: Optional[str] = None, # ["color"] or "binary"
|
||||||
|
hierarchical: Optional[str] = None, # ["stacked"] or "cutout"
|
||||||
|
mode: Optional[str] = None, # ["spline"], "polygon", "none"
|
||||||
|
filter_speckle: Optional[int] = None, # default: 4
|
||||||
|
color_precision: Optional[int] = None, # default: 6
|
||||||
|
layer_difference: Optional[int] = None, # default: 16
|
||||||
|
corner_threshold: Optional[int] = None, # default: 60
|
||||||
|
length_threshold: Optional[float] = None, # in [3.5, 10] default: 4.0
|
||||||
|
max_iterations: Optional[int] = None, # default: 10
|
||||||
|
splice_threshold: Optional[int] = None, # default: 45
|
||||||
|
path_precision: Optional[int] = None, # default: 8
|
||||||
|
) -> str:
|
||||||
|
...
|
||||||
+1
-1
@@ -22,7 +22,7 @@ console_log = { version = "0.2", features = ["color"] }
|
|||||||
wasm-bindgen = { version = "0.2", features = ["serde-serialize"] }
|
wasm-bindgen = { version = "0.2", features = ["serde-serialize"] }
|
||||||
serde = { version = "1.0", features = ["derive"] }
|
serde = { version = "1.0", features = ["derive"] }
|
||||||
serde_json = "1.0"
|
serde_json = "1.0"
|
||||||
visioncortex = "0.6.0"
|
visioncortex = "0.8.1"
|
||||||
|
|
||||||
# The `console_error_panic_hook` crate provides better debugging of panics by
|
# The `console_error_panic_hook` crate provides better debugging of panics by
|
||||||
# logging them with `console.error`. This is great for development, but requires
|
# logging them with `console.error`. This is great for development, but requires
|
||||||
|
|||||||
+6
-2
@@ -35,7 +35,11 @@ document.addEventListener('paste', function (e) {
|
|||||||
|
|
||||||
// Download as SVG
|
// Download as SVG
|
||||||
document.getElementById('export').addEventListener('click', function (e) {
|
document.getElementById('export').addEventListener('click', function (e) {
|
||||||
const blob = new Blob([new XMLSerializer().serializeToString(svg)], {type: 'octet/stream'}),
|
const blob = new Blob([
|
||||||
|
`<?xml version="1.0" encoding="UTF-8"?>\n`,
|
||||||
|
`<!-- Generator: visioncortex VTracer -->\n`,
|
||||||
|
new XMLSerializer().serializeToString(svg)
|
||||||
|
], {type: 'octet/stream'}),
|
||||||
url = window.URL.createObjectURL(blob);
|
url = window.URL.createObjectURL(blob);
|
||||||
|
|
||||||
this.href = url;
|
this.href = url;
|
||||||
@@ -444,7 +448,7 @@ class ConverterRunner {
|
|||||||
this.converter.init();
|
this.converter.init();
|
||||||
this.stopped = false;
|
this.stopped = false;
|
||||||
if (clustering_mode == 'binary') {
|
if (clustering_mode == 'binary') {
|
||||||
svg.style.background = '#000';
|
svg.style.background = '#fff';
|
||||||
canvas.style.display = 'none';
|
canvas.style.display = 'none';
|
||||||
} else {
|
} else {
|
||||||
svg.style.background = '';
|
svg.style.background = '';
|
||||||
|
|||||||
@@ -77,7 +77,7 @@ impl BinaryImageConverter {
|
|||||||
self.params.max_iterations,
|
self.params.max_iterations,
|
||||||
self.params.splice_threshold
|
self.params.splice_threshold
|
||||||
);
|
);
|
||||||
let color = Color::color(&ColorName::White);
|
let color = Color::color(&ColorName::Black);
|
||||||
self.svg.prepend_path(
|
self.svg.prepend_path(
|
||||||
&paths,
|
&paths,
|
||||||
&color,
|
&color,
|
||||||
|
|||||||
@@ -1,6 +1,6 @@
|
|||||||
use wasm_bindgen::prelude::*;
|
use wasm_bindgen::prelude::*;
|
||||||
use visioncortex::PathSimplifyMode;
|
use visioncortex::{Color, ColorImage, PathSimplifyMode};
|
||||||
use visioncortex::color_clusters::{IncrementalBuilder, Clusters, Runner, RunnerConfig, HIERARCHICAL_MAX};
|
use visioncortex::color_clusters::{Clusters, Runner, RunnerConfig, HIERARCHICAL_MAX, IncrementalBuilder, KeyingAction};
|
||||||
|
|
||||||
use crate::canvas::*;
|
use crate::canvas::*;
|
||||||
use crate::svg::*;
|
use crate::svg::*;
|
||||||
@@ -8,6 +8,8 @@ use crate::svg::*;
|
|||||||
use serde::Deserialize;
|
use serde::Deserialize;
|
||||||
use super::util;
|
use super::util;
|
||||||
|
|
||||||
|
const KEYING_THRESHOLD: f32 = 0.2;
|
||||||
|
|
||||||
#[derive(Debug, Deserialize)]
|
#[derive(Debug, Deserialize)]
|
||||||
pub struct ColorImageConverterParams {
|
pub struct ColorImageConverterParams {
|
||||||
pub canvas_id: String,
|
pub canvas_id: String,
|
||||||
@@ -67,7 +69,26 @@ impl ColorImageConverter {
|
|||||||
pub fn init(&mut self) {
|
pub fn init(&mut self) {
|
||||||
let width = self.canvas.width() as u32;
|
let width = self.canvas.width() as u32;
|
||||||
let height = self.canvas.height() as u32;
|
let height = self.canvas.height() as u32;
|
||||||
let image = self.canvas.get_image_data_as_color_image(0, 0, width, height);
|
let mut image = self.canvas.get_image_data_as_color_image(0, 0, width, height);
|
||||||
|
|
||||||
|
let key_color = if Self::should_key_image(&image) {
|
||||||
|
if let Ok(key_color) = Self::find_unused_color_in_image(&image) {
|
||||||
|
for y in 0..height as usize {
|
||||||
|
for x in 0..width as usize {
|
||||||
|
if image.get_pixel(x, y).a == 0 {
|
||||||
|
image.set_pixel(x, y, &key_color);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
key_color
|
||||||
|
} else {
|
||||||
|
Color::default()
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
// The default color is all zeroes, which is treated by visioncortex as a special value meaning no keying will be applied.
|
||||||
|
Color::default()
|
||||||
|
};
|
||||||
|
|
||||||
let runner = Runner::new(RunnerConfig {
|
let runner = Runner::new(RunnerConfig {
|
||||||
diagonal: self.params.layer_difference == 0,
|
diagonal: self.params.layer_difference == 0,
|
||||||
hierarchical: HIERARCHICAL_MAX,
|
hierarchical: HIERARCHICAL_MAX,
|
||||||
@@ -78,6 +99,12 @@ impl ColorImageConverter {
|
|||||||
is_same_color_b: 1,
|
is_same_color_b: 1,
|
||||||
deepen_diff: self.params.layer_difference,
|
deepen_diff: self.params.layer_difference,
|
||||||
hollow_neighbours: 1,
|
hollow_neighbours: 1,
|
||||||
|
key_color,
|
||||||
|
keying_action: if self.params.hierarchical == "cutout" {
|
||||||
|
KeyingAction::Keep
|
||||||
|
} else {
|
||||||
|
KeyingAction::Discard
|
||||||
|
},
|
||||||
}, image);
|
}, image);
|
||||||
self.stage = Stage::Clustering(runner.start());
|
self.stage = Stage::Clustering(runner.start());
|
||||||
}
|
}
|
||||||
@@ -108,6 +135,8 @@ impl ColorImageConverter {
|
|||||||
is_same_color_b: 1,
|
is_same_color_b: 1,
|
||||||
deepen_diff: 0,
|
deepen_diff: 0,
|
||||||
hollow_neighbours: 0,
|
hollow_neighbours: 0,
|
||||||
|
key_color: Default::default(),
|
||||||
|
keying_action: KeyingAction::Discard,
|
||||||
}, image);
|
}, image);
|
||||||
self.stage = Stage::Reclustering(runner.start());
|
self.stage = Stage::Reclustering(runner.start());
|
||||||
},
|
},
|
||||||
@@ -167,4 +196,56 @@ impl ColorImageConverter {
|
|||||||
}) as i32
|
}) as i32
|
||||||
}
|
}
|
||||||
|
|
||||||
|
fn color_exists_in_image(img: &ColorImage, color: Color) -> bool {
|
||||||
|
for y in 0..img.height {
|
||||||
|
for x in 0..img.width {
|
||||||
|
let pixel_color = img.get_pixel(x, y);
|
||||||
|
if pixel_color.r == color.r && pixel_color.g == color.g && pixel_color.b == color.b {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
false
|
||||||
|
}
|
||||||
|
|
||||||
|
fn find_unused_color_in_image(img: &ColorImage) -> Result<Color, String> {
|
||||||
|
let special_colors = IntoIterator::into_iter([
|
||||||
|
Color::new(255, 0, 0),
|
||||||
|
Color::new(0, 255, 0),
|
||||||
|
Color::new(0, 0, 255),
|
||||||
|
Color::new(255, 255, 0),
|
||||||
|
Color::new(0, 255, 255),
|
||||||
|
Color::new(255, 0, 255),
|
||||||
|
Color::new(128, 128, 128),
|
||||||
|
]);
|
||||||
|
for color in special_colors {
|
||||||
|
if !Self::color_exists_in_image(img, color) {
|
||||||
|
return Ok(color);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Err(String::from("unable to find unused color in image to use as key"))
|
||||||
|
}
|
||||||
|
|
||||||
|
fn should_key_image(img: &ColorImage) -> bool {
|
||||||
|
if img.width == 0 || img.height == 0 {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Check for transparency at several scanlines
|
||||||
|
let threshold = ((img.width * 2) as f32 * KEYING_THRESHOLD) as usize;
|
||||||
|
let mut num_transparent_pixels = 0;
|
||||||
|
let y_positions = [0, img.height / 4, img.height / 2, 3 * img.height / 4, img.height - 1];
|
||||||
|
for y in y_positions {
|
||||||
|
for x in 0..img.width {
|
||||||
|
if img.get_pixel(x, y).a == 0 {
|
||||||
|
num_transparent_pixels += 1;
|
||||||
|
}
|
||||||
|
if num_transparent_pixels >= threshold {
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
false
|
||||||
|
}
|
||||||
}
|
}
|
||||||
@@ -1,6 +1,3 @@
|
|||||||
mod binary_image;
|
mod binary_image;
|
||||||
mod color_image;
|
mod color_image;
|
||||||
mod util;
|
mod util;
|
||||||
|
|
||||||
pub use binary_image::*;
|
|
||||||
pub use color_image::*;
|
|
||||||
Reference in New Issue
Block a user