Compare commits
8 Commits
53140e642a
...
master
| Author | SHA1 | Date | |
|---|---|---|---|
|
861c384c0e
|
|||
| ea79ef97fc | |||
| 80cded98c6 | |||
| d539ce9a0c | |||
| 85434e84f2 | |||
| 837be94e1c | |||
| e3ff11d8a8 | |||
| fa020e0b36 |
Generated
+50
@@ -638,6 +638,15 @@ dependencies = [
|
|||||||
"num-traits",
|
"num-traits",
|
||||||
]
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "num-complex"
|
||||||
|
version = "0.4.6"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "73f88a1307638156682bada9d7604135552957b7818057dcef22705b4d509495"
|
||||||
|
dependencies = [
|
||||||
|
"num-traits",
|
||||||
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "num-derive"
|
name = "num-derive"
|
||||||
version = "0.4.2"
|
version = "0.4.2"
|
||||||
@@ -695,10 +704,12 @@ name = "oscilloscope-video-gen"
|
|||||||
version = "1.0.0"
|
version = "1.0.0"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"anyhow",
|
"anyhow",
|
||||||
|
"bytemuck",
|
||||||
"clap",
|
"clap",
|
||||||
"hound",
|
"hound",
|
||||||
"image",
|
"image",
|
||||||
"rayon",
|
"rayon",
|
||||||
|
"rustfft",
|
||||||
]
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
@@ -735,6 +746,15 @@ dependencies = [
|
|||||||
"zerocopy",
|
"zerocopy",
|
||||||
]
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "primal-check"
|
||||||
|
version = "0.3.4"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "dc0d895b311e3af9902528fbb8f928688abbd95872819320517cc24ca6b2bd08"
|
||||||
|
dependencies = [
|
||||||
|
"num-integer",
|
||||||
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "proc-macro2"
|
name = "proc-macro2"
|
||||||
version = "1.0.105"
|
version = "1.0.105"
|
||||||
@@ -907,6 +927,20 @@ version = "0.8.52"
|
|||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "0c6a884d2998352bb4daf0183589aec883f16a6da1f4dde84d8e2e9a5409a1ce"
|
checksum = "0c6a884d2998352bb4daf0183589aec883f16a6da1f4dde84d8e2e9a5409a1ce"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "rustfft"
|
||||||
|
version = "6.4.1"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "21db5f9893e91f41798c88680037dba611ca6674703c1a18601b01a72c8adb89"
|
||||||
|
dependencies = [
|
||||||
|
"num-complex",
|
||||||
|
"num-integer",
|
||||||
|
"num-traits",
|
||||||
|
"primal-check",
|
||||||
|
"strength_reduce",
|
||||||
|
"transpose",
|
||||||
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "rustversion"
|
name = "rustversion"
|
||||||
version = "1.0.22"
|
version = "1.0.22"
|
||||||
@@ -946,6 +980,12 @@ version = "1.2.1"
|
|||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "6ce2be8dc25455e1f91df71bfa12ad37d7af1092ae736f3a6cd0e37bc7810596"
|
checksum = "6ce2be8dc25455e1f91df71bfa12ad37d7af1092ae736f3a6cd0e37bc7810596"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "strength_reduce"
|
||||||
|
version = "0.2.4"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "fe895eb47f22e2ddd4dabc02bce419d2e643c8e3b585c78158b349195bc24d82"
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "strsim"
|
name = "strsim"
|
||||||
version = "0.11.1"
|
version = "0.11.1"
|
||||||
@@ -997,6 +1037,16 @@ dependencies = [
|
|||||||
"zune-jpeg 0.4.21",
|
"zune-jpeg 0.4.21",
|
||||||
]
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "transpose"
|
||||||
|
version = "0.2.3"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "1ad61aed86bc3faea4300c7aee358b4c6d0c8d6ccc36524c96e4c92ccf26e77e"
|
||||||
|
dependencies = [
|
||||||
|
"num-integer",
|
||||||
|
"strength_reduce",
|
||||||
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "unicode-ident"
|
name = "unicode-ident"
|
||||||
version = "1.0.22"
|
version = "1.0.22"
|
||||||
|
|||||||
@@ -28,6 +28,12 @@ clap = { version = "4.4", features = ["derive"] }
|
|||||||
# Error handling
|
# Error handling
|
||||||
anyhow = "1.0"
|
anyhow = "1.0"
|
||||||
|
|
||||||
|
# FFT processing for spectrometer
|
||||||
|
rustfft = "6.1"
|
||||||
|
|
||||||
|
# Efficient byte casting
|
||||||
|
bytemuck = "1.24"
|
||||||
|
|
||||||
[profile.release]
|
[profile.release]
|
||||||
opt-level = 3
|
opt-level = 3
|
||||||
lto = true
|
lto = true
|
||||||
|
|||||||
@@ -0,0 +1,658 @@
|
|||||||
|
GNU AFFERO GENERAL PUBLIC LICENSE
|
||||||
|
Version 3, 19 November 2007
|
||||||
|
|
||||||
|
Copyright (C) 2007 Free Software Foundation, Inc. <https://fsf.org/>
|
||||||
|
Everyone is permitted to copy and distribute verbatim copies
|
||||||
|
of this license document, but changing it is not allowed.
|
||||||
|
|
||||||
|
Preamble
|
||||||
|
|
||||||
|
The GNU Affero General Public License is a free, copyleft license for
|
||||||
|
software and other kinds of works, specifically designed to ensure
|
||||||
|
cooperation with the community in the case of network server software.
|
||||||
|
|
||||||
|
The licenses for most software and other practical works are designed
|
||||||
|
to take away your freedom to share and change the works. By contrast,
|
||||||
|
our General Public Licenses are intended to guarantee your freedom to
|
||||||
|
share and change all versions of a program--to make sure it remains free
|
||||||
|
software for all its users.
|
||||||
|
|
||||||
|
When we speak of free software, we are referring to freedom, not
|
||||||
|
price. Our General Public Licenses are designed to make sure that you
|
||||||
|
have the freedom to distribute copies of free software (and charge for
|
||||||
|
them if you wish), that you receive source code or can get it if you
|
||||||
|
want it, that you can change the software or use pieces of it in new
|
||||||
|
free programs, and that you know you can do these things.
|
||||||
|
|
||||||
|
Developers that use our General Public Licenses protect your rights
|
||||||
|
with two steps: (1) assert copyright on the software, and (2) offer
|
||||||
|
you this License which gives you legal permission to copy, distribute
|
||||||
|
and/or modify the software.
|
||||||
|
|
||||||
|
A secondary benefit of defending all users' freedom is that
|
||||||
|
improvements made in alternate versions of the program, if they
|
||||||
|
receive widespread use, become available for other developers to
|
||||||
|
incorporate. Many developers of free software are heartened and
|
||||||
|
encouraged by the resulting cooperation. However, in the case of
|
||||||
|
software used on network servers, this result may fail to come about.
|
||||||
|
The GNU General Public License permits making a modified version and
|
||||||
|
letting the public access it on a server without ever releasing its
|
||||||
|
source code to the public.
|
||||||
|
|
||||||
|
The GNU Affero General Public License is designed specifically to
|
||||||
|
ensure that, in such cases, the modified source code becomes available
|
||||||
|
to the community. It requires the operator of a network server to
|
||||||
|
provide the source code of the modified version running there to the
|
||||||
|
users of that server. Therefore, public use of a modified version, on
|
||||||
|
a publicly accessible server, gives the public access to the source
|
||||||
|
code of the modified version.
|
||||||
|
|
||||||
|
An older license, called the Affero General Public License and
|
||||||
|
published by Affero, was designed to accomplish similar goals. This is
|
||||||
|
a different license, not a version of the Affero GPL, but Affero has
|
||||||
|
released a new version of the Affero GPL which permits relicensing under
|
||||||
|
this license.
|
||||||
|
|
||||||
|
The precise terms and conditions for copying, distribution and
|
||||||
|
modification follow.
|
||||||
|
|
||||||
|
TERMS AND CONDITIONS
|
||||||
|
|
||||||
|
0. Definitions.
|
||||||
|
|
||||||
|
"This License" refers to version 3 of the GNU Affero General Public License.
|
||||||
|
|
||||||
|
"Copyright" also means copyright-like laws that apply to other kinds of
|
||||||
|
works, such as semiconductor masks.
|
||||||
|
|
||||||
|
"The Program" refers to any copyrightable work licensed under this
|
||||||
|
License. Each licensee is addressed as "you". "Licensees" and
|
||||||
|
"recipients" may be individuals or organizations.
|
||||||
|
|
||||||
|
To "modify" a work means to copy from or adapt all or part of the work
|
||||||
|
in a fashion requiring copyright permission, other than the making of an
|
||||||
|
exact copy. The resulting work is called a "modified version" of the
|
||||||
|
earlier work or a work "based on" the earlier work.
|
||||||
|
|
||||||
|
A "covered work" means either the unmodified Program or a work based
|
||||||
|
on the Program.
|
||||||
|
|
||||||
|
To "propagate" a work means to do anything with it that, without
|
||||||
|
permission, would make you directly or secondarily liable for
|
||||||
|
infringement under applicable copyright law, except executing it on a
|
||||||
|
computer or modifying a private copy. Propagation includes copying,
|
||||||
|
distribution (with or without modification), making available to the
|
||||||
|
public, and in some countries other activities as well.
|
||||||
|
|
||||||
|
To "convey" a work means any kind of propagation that enables other
|
||||||
|
parties to make or receive copies. Mere interaction with a user through
|
||||||
|
a computer network, with no transfer of a copy, is not conveying.
|
||||||
|
|
||||||
|
An interactive user interface displays "Appropriate Legal Notices"
|
||||||
|
to the extent that it includes a convenient and prominently visible
|
||||||
|
feature that (1) displays an appropriate copyright notice, and (2)
|
||||||
|
tells the user that there is no warranty for the work (except to the
|
||||||
|
extent that warranties are provided), that licensees may convey the
|
||||||
|
work under this License, and how to view a copy of this License. If
|
||||||
|
the interface presents a list of user commands or options, such as a
|
||||||
|
menu, a prominent item in the list meets this criterion.
|
||||||
|
|
||||||
|
1. Source Code.
|
||||||
|
|
||||||
|
The "source code" for a work means the preferred form of the work
|
||||||
|
for making modifications to it. "Object code" means any non-source
|
||||||
|
form of a work.
|
||||||
|
|
||||||
|
A "Standard Interface" means an interface that either is an official
|
||||||
|
standard defined by a recognized standards body, or, in the case of
|
||||||
|
interfaces specified for a particular programming language, one that
|
||||||
|
is widely used among developers working in that language.
|
||||||
|
|
||||||
|
The "System Libraries" of an executable work include anything, other
|
||||||
|
than the work as a whole, that (a) is included in the normal form of
|
||||||
|
packaging a Major Component, but which is not part of that Major
|
||||||
|
Component, and (b) serves only to enable use of the work with that
|
||||||
|
Major Component, or to implement a Standard Interface for which an
|
||||||
|
implementation is available to the public in source code form. A
|
||||||
|
"Major Component", in this context, means a major essential component
|
||||||
|
(kernel, window system, and so on) of the specific operating system
|
||||||
|
(if any) on which the executable work runs, or a compiler used to
|
||||||
|
produce the work, or an object code interpreter used to run it.
|
||||||
|
|
||||||
|
The "Corresponding Source" for a work in object code form means all
|
||||||
|
the source code needed to generate, install, and (for an executable
|
||||||
|
work) run the object code and to modify the work, including scripts to
|
||||||
|
control those activities. However, it does not include the work's
|
||||||
|
System Libraries, or general-purpose tools or generally available free
|
||||||
|
programs which are used unmodified in performing those activities but
|
||||||
|
which are not part of the work. For example, Corresponding Source
|
||||||
|
includes interface definition files associated with source files for
|
||||||
|
the work, and the source code for shared libraries and dynamically
|
||||||
|
linked subprograms that the work is specifically designed to require,
|
||||||
|
such as by intimate data communication or control flow between those
|
||||||
|
subprograms and other parts of the work.
|
||||||
|
|
||||||
|
The Corresponding Source need not include anything that users
|
||||||
|
can regenerate automatically from other parts of the Corresponding
|
||||||
|
Source.
|
||||||
|
|
||||||
|
The Corresponding Source for a work in source code form is that
|
||||||
|
same work.
|
||||||
|
|
||||||
|
2. Basic Permissions.
|
||||||
|
|
||||||
|
All rights granted under this License are granted for the term of
|
||||||
|
copyright on the Program, and are irrevocable provided the stated
|
||||||
|
conditions are met. This License explicitly affirms your unlimited
|
||||||
|
permission to run the unmodified Program. The output from running a
|
||||||
|
covered work is covered by this License only if the output, given its
|
||||||
|
content, constitutes a covered work. This License acknowledges your
|
||||||
|
rights of fair use or other equivalent, as provided by copyright law.
|
||||||
|
|
||||||
|
You may make, run and propagate covered works that you do not
|
||||||
|
convey, without conditions so long as your license otherwise remains
|
||||||
|
in force. You may convey covered works to others for the sole purpose
|
||||||
|
of having them make modifications exclusively for you, or provide you
|
||||||
|
with facilities for running those works, provided that you comply with
|
||||||
|
the terms of this License in conveying all material for which you do
|
||||||
|
not control copyright. Those thus making or running the covered works
|
||||||
|
for you must do so exclusively on your behalf, under your direction
|
||||||
|
and control, on terms that prohibit them from making any copies of
|
||||||
|
your copyrighted material outside their relationship with you.
|
||||||
|
|
||||||
|
Conveying under any other circumstances is permitted solely under
|
||||||
|
the conditions stated below. Sublicensing is not allowed; section 10
|
||||||
|
makes it unnecessary.
|
||||||
|
|
||||||
|
3. Protecting Users' Legal Rights From Anti-Circumvention Law.
|
||||||
|
|
||||||
|
No covered work shall be deemed part of an effective technological
|
||||||
|
measure under any applicable law fulfilling obligations under article
|
||||||
|
11 of the WIPO copyright treaty adopted on 20 December 1996, or
|
||||||
|
similar laws prohibiting or restricting circumvention of such
|
||||||
|
measures.
|
||||||
|
|
||||||
|
When you convey a covered work, you waive any legal power to forbid
|
||||||
|
circumvention of technological measures to the extent such circumvention
|
||||||
|
is effected by exercising rights under this License with respect to
|
||||||
|
the covered work, and you disclaim any intention to limit operation or
|
||||||
|
modification of the work as a means of enforcing, against the work's
|
||||||
|
users, your or third parties' legal rights to forbid circumvention of
|
||||||
|
technological measures.
|
||||||
|
|
||||||
|
4. Conveying Verbatim Copies.
|
||||||
|
|
||||||
|
You may convey verbatim copies of the Program's source code as you
|
||||||
|
receive it, in any medium, provided that you conspicuously and
|
||||||
|
appropriately publish on each copy an appropriate copyright notice;
|
||||||
|
keep intact all notices stating that this License and any
|
||||||
|
non-permissive terms added in accord with section 7 apply to the code;
|
||||||
|
keep intact all notices of the absence of any warranty; and give all
|
||||||
|
recipients a copy of this License along with the Program.
|
||||||
|
|
||||||
|
You may charge any price or no price for each copy that you convey,
|
||||||
|
and you may offer support or warranty protection for a fee.
|
||||||
|
|
||||||
|
5. Conveying Modified Source Versions.
|
||||||
|
|
||||||
|
You may convey a work based on the Program, or the modifications to
|
||||||
|
produce it from the Program, in the form of source code under the
|
||||||
|
terms of section 4, provided that you also meet all of these conditions:
|
||||||
|
|
||||||
|
a) The work must carry prominent notices stating that you modified
|
||||||
|
it, and giving a relevant date.
|
||||||
|
|
||||||
|
b) The work must carry prominent notices stating that it is
|
||||||
|
released under this License and any conditions added under section
|
||||||
|
7. This requirement modifies the requirement in section 4 to
|
||||||
|
"keep intact all notices".
|
||||||
|
|
||||||
|
c) You must license the entire work, as a whole, under this
|
||||||
|
License to anyone who comes into possession of a copy. This
|
||||||
|
License will therefore apply, along with any applicable section 7
|
||||||
|
additional terms, to the whole of the work, and all its parts,
|
||||||
|
regardless of how they are packaged. This License gives no
|
||||||
|
permission to license the work in any other way, but it does not
|
||||||
|
invalidate such permission if you have separately received it.
|
||||||
|
|
||||||
|
d) If the work has interactive user interfaces, each must display
|
||||||
|
Appropriate Legal Notices; however, if the Program has interactive
|
||||||
|
interfaces that do not display Appropriate Legal Notices, your
|
||||||
|
work need not make them do so.
|
||||||
|
|
||||||
|
A compilation of a covered work with other separate and independent
|
||||||
|
works, which are not by their nature extensions of the covered work,
|
||||||
|
and which are not combined with it such as to form a larger program,
|
||||||
|
in or on a volume of a storage or distribution medium, is called an
|
||||||
|
"aggregate" if the compilation and its resulting copyright are not
|
||||||
|
used to limit the access or legal rights of the compilation's users
|
||||||
|
beyond what the individual works permit. Inclusion of a covered work
|
||||||
|
in an aggregate does not cause this License to apply to the other
|
||||||
|
parts of the aggregate.
|
||||||
|
|
||||||
|
6. Conveying Non-Source Forms.
|
||||||
|
|
||||||
|
You may convey a covered work in object code form under the terms
|
||||||
|
of sections 4 and 5, provided that you also convey the
|
||||||
|
machine-readable Corresponding Source under the terms of this License,
|
||||||
|
in one of these ways:
|
||||||
|
|
||||||
|
a) Convey the object code in, or embodied in, a physical product
|
||||||
|
(including a physical distribution medium), accompanied by the
|
||||||
|
Corresponding Source fixed on a durable physical medium
|
||||||
|
customarily used for software interchange.
|
||||||
|
|
||||||
|
b) Convey the object code in, or embodied in, a physical product
|
||||||
|
(including a physical distribution medium), accompanied by a
|
||||||
|
written offer, valid for at least three years and valid for as
|
||||||
|
long as you offer spare parts or customer support for that product
|
||||||
|
model, to give anyone who possesses the object code either (1) a
|
||||||
|
copy of the Corresponding Source for all the software in the
|
||||||
|
product that is covered by this License, on a durable physical
|
||||||
|
medium customarily used for software interchange, for a price no
|
||||||
|
more than your reasonable cost of physically performing this
|
||||||
|
conveying of source, or (2) access to copy the
|
||||||
|
Corresponding Source from a network server at no charge.
|
||||||
|
|
||||||
|
c) Convey individual copies of the object code with a copy of the
|
||||||
|
written offer to provide the Corresponding Source. This
|
||||||
|
alternative is allowed only occasionally and noncommercially, and
|
||||||
|
only if you received the object code with such an offer, in accord
|
||||||
|
with subsection 6b.
|
||||||
|
|
||||||
|
d) Convey the object code by offering access from a designated
|
||||||
|
place (gratis or for a charge), and offer equivalent access to the
|
||||||
|
Corresponding Source in the same way through the same place at no
|
||||||
|
further charge. You need not require recipients to copy the
|
||||||
|
Corresponding Source along with the object code. If the place to
|
||||||
|
copy the object code is a network server, the Corresponding Source
|
||||||
|
may be on a different server (operated by you or a third party)
|
||||||
|
that supports equivalent copying facilities, provided you maintain
|
||||||
|
clear directions next to the object code saying where to find the
|
||||||
|
Corresponding Source. Regardless of what server hosts the
|
||||||
|
Corresponding Source, you remain obligated to ensure that it is
|
||||||
|
available for as long as needed to satisfy these requirements.
|
||||||
|
|
||||||
|
e) Convey the object code using peer-to-peer transmission, provided
|
||||||
|
you inform other peers where the object code and Corresponding
|
||||||
|
Source of the work are being offered to the general public at no
|
||||||
|
charge under subsection 6d.
|
||||||
|
|
||||||
|
A separable portion of the object code, whose source code is excluded
|
||||||
|
from the Corresponding Source as a System Library, need not be
|
||||||
|
included in conveying the object code work.
|
||||||
|
|
||||||
|
A "User Product" is either (1) a "consumer product", which means any
|
||||||
|
tangible personal property which is normally used for personal, family,
|
||||||
|
or household purposes, or (2) anything designed or sold for incorporation
|
||||||
|
into a dwelling. In determining whether a product is a consumer product,
|
||||||
|
doubtful cases shall be resolved in favor of coverage. For a particular
|
||||||
|
product received by a particular user, "normally used" refers to a
|
||||||
|
typical or common use of that class of product, regardless of the status
|
||||||
|
of the particular user or of the way in which the particular user
|
||||||
|
actually uses, or expects or is expected to use, the product. A product
|
||||||
|
is a consumer product regardless of whether the product has substantial
|
||||||
|
commercial, industrial or non-consumer uses, unless such uses represent
|
||||||
|
the only significant mode of use of the product.
|
||||||
|
|
||||||
|
"Installation Information" for a User Product means any methods,
|
||||||
|
procedures, authorization keys, or other information required to install
|
||||||
|
and execute modified versions of a covered work in that User Product from
|
||||||
|
a modified version of its Corresponding Source. The information must
|
||||||
|
suffice to ensure that the continued functioning of the modified object
|
||||||
|
code is in no case prevented or interfered with solely because
|
||||||
|
modification has been made.
|
||||||
|
|
||||||
|
If you convey an object code work under this section in, or with, or
|
||||||
|
specifically for use in, a User Product, and the conveying occurs as
|
||||||
|
part of a transaction in which the right of possession and use of the
|
||||||
|
User Product is transferred to the recipient in perpetuity or for a
|
||||||
|
fixed term (regardless of how the transaction is characterized), the
|
||||||
|
Corresponding Source conveyed under this section must be accompanied
|
||||||
|
by the Installation Information. But this requirement does not apply
|
||||||
|
if neither you nor any third party retains the ability to install
|
||||||
|
modified object code on the User Product (for example, the work has
|
||||||
|
been installed in ROM).
|
||||||
|
|
||||||
|
The requirement to provide Installation Information does not include a
|
||||||
|
requirement to continue to provide support service, warranty, or updates
|
||||||
|
for a work that has been modified or installed by the recipient, or for
|
||||||
|
the User Product in which it has been modified or installed. Access to a
|
||||||
|
network may be denied when the modification itself materially and
|
||||||
|
adversely affects the operation of the network or violates the rules and
|
||||||
|
protocols for communication across the network.
|
||||||
|
|
||||||
|
Corresponding Source conveyed, and Installation Information provided,
|
||||||
|
in accord with this section must be in a format that is publicly
|
||||||
|
documented (and with an implementation available to the public in
|
||||||
|
source code form), and must require no special password or key for
|
||||||
|
unpacking, reading or copying.
|
||||||
|
|
||||||
|
7. Additional Terms.
|
||||||
|
|
||||||
|
"Additional permissions" are terms that supplement the terms of this
|
||||||
|
License by making exceptions from one or more of its conditions.
|
||||||
|
Additional permissions that are applicable to the entire Program shall
|
||||||
|
be treated as though they were included in this License, to the extent
|
||||||
|
that they are valid under applicable law. If additional permissions
|
||||||
|
apply only to part of the Program, that part may be used separately
|
||||||
|
under those permissions, but the entire Program remains governed by
|
||||||
|
this License without regard to the additional permissions.
|
||||||
|
|
||||||
|
When you convey a copy of a covered work, you may at your option
|
||||||
|
remove any additional permissions from that copy, or from any part of
|
||||||
|
it. (Additional permissions may be written to require their own
|
||||||
|
removal in certain cases when you modify the work.) You may place
|
||||||
|
additional permissions on material, added by you to a covered work,
|
||||||
|
for which you have or can give appropriate copyright permission.
|
||||||
|
|
||||||
|
Notwithstanding any other provision of this License, for material you
|
||||||
|
add to a covered work, you may (if authorized by the copyright holders of
|
||||||
|
that material) supplement the terms of this License with terms:
|
||||||
|
|
||||||
|
a) Disclaiming warranty or limiting liability differently from the
|
||||||
|
terms of sections 15 and 16 of this License; or
|
||||||
|
|
||||||
|
b) Requiring preservation of specified reasonable legal notices or
|
||||||
|
author attributions in that material or in the Appropriate Legal
|
||||||
|
Notices displayed by works containing it; or
|
||||||
|
|
||||||
|
c) Prohibiting misrepresentation of the origin of that material, or
|
||||||
|
requiring that modified versions of such material be marked in
|
||||||
|
reasonable ways as different from the original version; or
|
||||||
|
|
||||||
|
d) Limiting the use for publicity purposes of names of licensors or
|
||||||
|
authors of the material; or
|
||||||
|
|
||||||
|
e) Requiring indemnification of licensors and authors of that
|
||||||
|
material by anyone who conveys the material (or modified versions of
|
||||||
|
it) with contractual assumptions of liability to the recipient, for
|
||||||
|
any liability that these contractual assumptions directly impose on
|
||||||
|
those licensors and authors.
|
||||||
|
|
||||||
|
All other non-permissive additional terms are considered "further
|
||||||
|
restrictions" within the meaning of section 10. If the Program as you
|
||||||
|
received it, or any part of it, contains a notice stating that it is
|
||||||
|
governed by this License along with a term that is a further
|
||||||
|
restriction, you may remove that term. If a license document contains
|
||||||
|
a further restriction but permits relicensing or conveying under this
|
||||||
|
License, you may add to a covered work material governed by the terms
|
||||||
|
of that license document, provided that the further restriction does
|
||||||
|
not survive such relicensing or conveying.
|
||||||
|
|
||||||
|
If you add terms to a covered work in accord with this section, you
|
||||||
|
must place, in the relevant source files, a statement of the
|
||||||
|
additional terms that apply to those files, or a notice indicating
|
||||||
|
where to find the applicable terms.
|
||||||
|
|
||||||
|
Additional terms, permissive or non-permissive, may be stated in the
|
||||||
|
form of a separately written license, or stated as exceptions;
|
||||||
|
the above requirements apply either way.
|
||||||
|
|
||||||
|
8. Termination.
|
||||||
|
|
||||||
|
You may not propagate or modify a covered work except as expressly
|
||||||
|
provided under this License. Any attempt otherwise to propagate or
|
||||||
|
modify it is void, and will automatically terminate your rights under
|
||||||
|
this License (including any patent licenses granted under the third
|
||||||
|
paragraph of section 11).
|
||||||
|
|
||||||
|
However, if you cease all violation of this License, then your
|
||||||
|
license from a particular copyright holder is reinstated (a)
|
||||||
|
provisionally, unless and until the copyright holder explicitly and
|
||||||
|
finally terminates your license, and (b) permanently, if the copyright
|
||||||
|
holder fails to notify you of the violation by some reasonable means
|
||||||
|
prior to 60 days after the cessation.
|
||||||
|
|
||||||
|
Moreover, your license from a particular copyright holder is
|
||||||
|
reinstated permanently if the copyright holder notifies you of the
|
||||||
|
violation by some reasonable means, this is the first time you have
|
||||||
|
received notice of violation of this License (for any work) from that
|
||||||
|
copyright holder, and you cure the violation prior to 30 days after
|
||||||
|
your receipt of the notice.
|
||||||
|
|
||||||
|
Termination of your rights under this section does not terminate the
|
||||||
|
licenses of parties who have received copies or rights from you under
|
||||||
|
this License. If your rights have been terminated and not permanently
|
||||||
|
reinstated, you do not qualify to receive new licenses for the same
|
||||||
|
material under section 10.
|
||||||
|
|
||||||
|
9. Acceptance Not Required for Having Copies.
|
||||||
|
|
||||||
|
You are not required to accept this License in order to receive or
|
||||||
|
run a copy of the Program. Ancillary propagation of a covered work
|
||||||
|
occurring solely as a consequence of using peer-to-peer transmission
|
||||||
|
to receive a copy likewise does not require acceptance. However,
|
||||||
|
nothing other than this License grants you permission to propagate or
|
||||||
|
modify any covered work. These actions infringe copyright if you do
|
||||||
|
not accept this License. Therefore, by modifying or propagating a
|
||||||
|
covered work, you indicate your acceptance of this License to do so.
|
||||||
|
|
||||||
|
10. Automatic Licensing of Downstream Recipients.
|
||||||
|
|
||||||
|
Each time you convey a covered work, the recipient automatically
|
||||||
|
receives a license from the original licensors, to run, modify and
|
||||||
|
propagate that work, subject to this License. You are not responsible
|
||||||
|
for enforcing compliance by third parties with this License.
|
||||||
|
|
||||||
|
An "entity transaction" is a transaction transferring control of an
|
||||||
|
organization, or substantially all assets of one, or subdividing an
|
||||||
|
organization, or merging organizations. If propagation of a covered
|
||||||
|
work results from an entity transaction, each party to that
|
||||||
|
transaction who receives a copy of the work also receives whatever
|
||||||
|
licenses to the work the party's predecessor in interest had or could
|
||||||
|
give under the previous paragraph, plus a right to possession of the
|
||||||
|
Corresponding Source of the work from the predecessor in interest, if
|
||||||
|
the predecessor has it or can get it with reasonable efforts.
|
||||||
|
|
||||||
|
You may not impose any further restrictions on the exercise of the
|
||||||
|
rights granted or affirmed under this License. For example, you may
|
||||||
|
not impose a license fee, royalty, or other charge for exercise of
|
||||||
|
rights granted under this License, and you may not initiate litigation
|
||||||
|
(including a cross-claim or counterclaim in a lawsuit) alleging that
|
||||||
|
any patent claim is infringed by making, using, selling, offering for
|
||||||
|
sale, or importing the Program or any portion of it.
|
||||||
|
|
||||||
|
11. Patents.
|
||||||
|
|
||||||
|
A "contributor" is a copyright holder who authorizes use under this
|
||||||
|
License of the Program or a work on which the Program is based. The
|
||||||
|
work thus licensed is called the contributor's "contributor version".
|
||||||
|
|
||||||
|
A contributor's "essential patent claims" are all patent claims
|
||||||
|
owned or controlled by the contributor, whether already acquired or
|
||||||
|
hereafter acquired, that would be infringed by some manner, permitted
|
||||||
|
by this License, of making, using, or selling its contributor version,
|
||||||
|
but do not include claims that would be infringed only as a
|
||||||
|
consequence of further modification of the contributor version. For
|
||||||
|
purposes of this definition, "control" includes the right to grant
|
||||||
|
patent sublicenses in a manner consistent with the requirements of
|
||||||
|
this License.
|
||||||
|
|
||||||
|
Each contributor grants you a non-exclusive, worldwide, royalty-free
|
||||||
|
patent license under the contributor's essential patent claims, to
|
||||||
|
make, use, sell, offer for sale, import and otherwise run, modify and
|
||||||
|
propagate the contents of its contributor version.
|
||||||
|
|
||||||
|
In the following three paragraphs, a "patent license" is any express
|
||||||
|
agreement or commitment, however denominated, not to enforce a patent
|
||||||
|
(such as an express permission to practice a patent or covenant not to
|
||||||
|
sue for patent infringement). To "grant" such a patent license to a
|
||||||
|
party means to make such an agreement or commitment not to enforce a
|
||||||
|
patent against the party.
|
||||||
|
|
||||||
|
If you convey a covered work, knowingly relying on a patent license,
|
||||||
|
and the Corresponding Source of the work is not available for anyone
|
||||||
|
to copy, free of charge and under the terms of this License, through a
|
||||||
|
publicly available network server or other readily accessible means,
|
||||||
|
then you must either (1) cause the Corresponding Source to be so
|
||||||
|
available, or (2) arrange to deprive yourself of the benefit of the
|
||||||
|
patent license for this particular work, or (3) arrange, in a manner
|
||||||
|
consistent with the requirements of this License, to extend the patent
|
||||||
|
license to downstream recipients. "Knowingly relying" means you have
|
||||||
|
actual knowledge that, but for the patent license, your conveying the
|
||||||
|
covered work in a country, or your recipient's use of the covered work
|
||||||
|
in a country, would infringe one or more identifiable patents in that
|
||||||
|
country that you have reason to believe are valid.
|
||||||
|
|
||||||
|
If, pursuant to or in connection with a single transaction or
|
||||||
|
arrangement, you convey, or propagate by procuring conveyance of, a
|
||||||
|
covered work, and grant a patent license to some of the parties
|
||||||
|
receiving the covered work authorizing them to use, propagate, modify
|
||||||
|
or convey a specific copy of the covered work, then the patent license
|
||||||
|
you grant is automatically extended to all recipients of the covered
|
||||||
|
work and works based on it.
|
||||||
|
|
||||||
|
A patent license is "discriminatory" if it does not include within
|
||||||
|
the scope of its coverage, prohibits the exercise of, or is
|
||||||
|
conditioned on the non-exercise of one or more of the rights that are
|
||||||
|
specifically granted under this License. You may not convey a covered
|
||||||
|
work if you are a party to an arrangement with a third party that is
|
||||||
|
in the business of distributing software, under which you make payment
|
||||||
|
to the third party based on the extent of your activity of conveying
|
||||||
|
the work, and under which the third party grants, to any of the
|
||||||
|
parties who would receive the covered work from you, a discriminatory
|
||||||
|
patent license (a) in connection with copies of the covered work
|
||||||
|
conveyed by you (or copies made from those copies), or (b) primarily
|
||||||
|
for and in connection with specific products or compilations that
|
||||||
|
contain the covered work, unless you entered into that arrangement,
|
||||||
|
or that patent license was granted, prior to 28 March 2007.
|
||||||
|
|
||||||
|
Nothing in this License shall be construed as excluding or limiting
|
||||||
|
any implied license or other defenses to infringement that may
|
||||||
|
otherwise be available to you under applicable patent law.
|
||||||
|
|
||||||
|
12. No Surrender of Others' Freedom.
|
||||||
|
|
||||||
|
If conditions are imposed on you (whether by court order, agreement or
|
||||||
|
otherwise) that contradict the conditions of this License, they do not
|
||||||
|
excuse you from the conditions of this License. If you cannot convey a
|
||||||
|
covered work so as to satisfy simultaneously your obligations under this
|
||||||
|
License and any other pertinent obligations, then as a consequence you may
|
||||||
|
not convey it at all. For example, if you agree to terms that obligate you
|
||||||
|
to collect a royalty for further conveying from those to whom you convey
|
||||||
|
the Program, the only way you could satisfy both those terms and this
|
||||||
|
License would be to refrain entirely from conveying the Program.
|
||||||
|
|
||||||
|
13. Remote Network Interaction; Use with the GNU General Public License.
|
||||||
|
|
||||||
|
Notwithstanding any other provision of this License, if you modify the
|
||||||
|
Program, your modified version must prominently offer all users
|
||||||
|
interacting with it remotely through a computer network (if your version
|
||||||
|
supports such interaction) an opportunity to receive the Corresponding
|
||||||
|
Source of your version by providing access to the Corresponding Source
|
||||||
|
from a network server at no charge, through some standard or customary
|
||||||
|
means of facilitating copying of software. This Corresponding Source
|
||||||
|
shall include the Corresponding Source for any work covered by version 3
|
||||||
|
of the GNU General Public License that is incorporated pursuant to the
|
||||||
|
following paragraph.
|
||||||
|
|
||||||
|
Notwithstanding any other provision of this License, you have
|
||||||
|
permission to link or combine any covered work with a work licensed
|
||||||
|
under version 3 of the GNU General Public License into a single
|
||||||
|
combined work, and to convey the resulting work. The terms of this
|
||||||
|
License will continue to apply to the part which is the covered work,
|
||||||
|
but the work with which it is combined will remain governed by version
|
||||||
|
3 of the GNU General Public License.
|
||||||
|
|
||||||
|
14. Revised Versions of this License.
|
||||||
|
|
||||||
|
The Free Software Foundation may publish revised and/or new versions of
|
||||||
|
the GNU Affero General Public License from time to time. Such new versions
|
||||||
|
will be similar in spirit to the present version, but may differ in detail to
|
||||||
|
address new problems or concerns.
|
||||||
|
|
||||||
|
Each version is given a distinguishing version number. If the
|
||||||
|
Program specifies that a certain numbered version of the GNU Affero General
|
||||||
|
Public License "or any later version" applies to it, you have the
|
||||||
|
option of following the terms and conditions either of that numbered
|
||||||
|
version or of any later version published by the Free Software
|
||||||
|
Foundation. If the Program does not specify a version number of the
|
||||||
|
GNU Affero General Public License, you may choose any version ever published
|
||||||
|
by the Free Software Foundation.
|
||||||
|
|
||||||
|
If the Program specifies that a proxy can decide which future
|
||||||
|
versions of the GNU Affero General Public License can be used, that proxy's
|
||||||
|
public statement of acceptance of a version permanently authorizes you
|
||||||
|
to choose that version for the Program.
|
||||||
|
|
||||||
|
Later license versions may give you additional or different
|
||||||
|
permissions. However, no additional obligations are imposed on any
|
||||||
|
author or copyright holder as a result of your choosing to follow a
|
||||||
|
later version.
|
||||||
|
|
||||||
|
15. Disclaimer of Warranty.
|
||||||
|
|
||||||
|
THERE IS NO WARRANTY FOR THE PROGRAM, TO THE EXTENT PERMITTED BY
|
||||||
|
APPLICABLE LAW. EXCEPT WHEN OTHERWISE STATED IN WRITING THE COPYRIGHT
|
||||||
|
HOLDERS AND/OR OTHER PARTIES PROVIDE THE PROGRAM "AS IS" WITHOUT WARRANTY
|
||||||
|
OF ANY KIND, EITHER EXPRESSED OR IMPLIED, INCLUDING, BUT NOT LIMITED TO,
|
||||||
|
THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
|
||||||
|
PURPOSE. THE ENTIRE RISK AS TO THE QUALITY AND PERFORMANCE OF THE PROGRAM
|
||||||
|
IS WITH YOU. SHOULD THE PROGRAM PROVE DEFECTIVE, YOU ASSUME THE COST OF
|
||||||
|
ALL NECESSARY SERVICING, REPAIR OR CORRECTION.
|
||||||
|
|
||||||
|
16. Limitation of Liability.
|
||||||
|
|
||||||
|
IN NO EVENT UNLESS REQUIRED BY APPLICABLE LAW OR AGREED TO IN WRITING
|
||||||
|
WILL ANY COPYRIGHT HOLDER, OR ANY OTHER PARTY WHO MODIFIES AND/OR CONVEYS
|
||||||
|
THE PROGRAM AS PERMITTED ABOVE, BE LIABLE TO YOU FOR DAMAGES, INCLUDING ANY
|
||||||
|
GENERAL, SPECIAL, INCIDENTAL OR CONSEQUENTIAL DAMAGES ARISING OUT OF THE
|
||||||
|
USE OR INABILITY TO USE THE PROGRAM (INCLUDING BUT NOT LIMITED TO LOSS OF
|
||||||
|
DATA OR DATA BEING RENDERED INACCURATE OR LOSSES SUSTAINED BY YOU OR THIRD
|
||||||
|
PARTIES OR A FAILURE OF THE PROGRAM TO OPERATE WITH ANY OTHER PROGRAMS),
|
||||||
|
EVEN IF SUCH HOLDER OR OTHER PARTY HAS BEEN ADVISED OF THE POSSIBILITY OF
|
||||||
|
SUCH DAMAGES.
|
||||||
|
|
||||||
|
17. Interpretation of Sections 15 and 16.
|
||||||
|
|
||||||
|
If the disclaimer of warranty and limitation of liability provided
|
||||||
|
above cannot be given local legal effect according to their terms,
|
||||||
|
reviewing courts shall apply local law that most closely approximates
|
||||||
|
an absolute waiver of all civil liability in connection with the
|
||||||
|
Program, unless a warranty or assumption of liability accompanies a
|
||||||
|
copy of the Program in return for a fee.
|
||||||
|
|
||||||
|
END OF TERMS AND CONDITIONS
|
||||||
|
|
||||||
|
How to Apply These Terms to Your New Programs
|
||||||
|
|
||||||
|
If you develop a new program, and you want it to be of the greatest
|
||||||
|
possible use to the public, the best way to achieve this is to make it
|
||||||
|
free software which everyone can redistribute and change under these terms.
|
||||||
|
|
||||||
|
To do so, attach the following notices to the program. It is safest
|
||||||
|
to attach them to the start of each source file to most effectively
|
||||||
|
state the exclusion of warranty; and each file should have at least
|
||||||
|
the "copyright" line and a pointer to where the full notice is found.
|
||||||
|
|
||||||
|
<one line to give the program's name and a brief idea of what it does.>
|
||||||
|
Copyright (C) <year> <name of author>
|
||||||
|
|
||||||
|
This program is free software: you can redistribute it and/or modify
|
||||||
|
it under the terms of the GNU Affero General Public License as published by
|
||||||
|
the Free Software Foundation, either version 3 of the License, or
|
||||||
|
(at your option) any later version.
|
||||||
|
|
||||||
|
This program is distributed in the hope that it will be useful,
|
||||||
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
GNU Affero General Public License for more details.
|
||||||
|
|
||||||
|
You should have received a copy of the GNU Affero General Public License
|
||||||
|
along with this program. If not, see <https://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
|
Also add information on how to contact you by electronic and paper mail.
|
||||||
|
|
||||||
|
If your software can interact with users remotely through a computer
|
||||||
|
network, you should also make sure that it provides a way for users to
|
||||||
|
get its source. For example, if your program is a web application, its
|
||||||
|
interface could display a "Source" link that leads users to an archive
|
||||||
|
of the code. There are many ways you could offer source, and different
|
||||||
|
solutions will be better for different programs; see section 13 for the
|
||||||
|
specific requirements.
|
||||||
|
|
||||||
|
You should also get your employer (if you work as a programmer) or school,
|
||||||
|
if any, to sign a "copyright disclaimer" for the program, if necessary.
|
||||||
|
For more information on this, and how to apply and follow the GNU AGPL, see
|
||||||
|
<https://www.gnu.org/licenses/>.
|
||||||
@@ -1,120 +1,113 @@
|
|||||||
# Oscilloscope Video Generator
|
# Oscilloscope Video Generator
|
||||||
|
|
||||||
A high-performance Rust tool for generating oscilloscope-style visualizations from audio files. Uses parallel rendering for fast processing.
|
Transform any audio file into stunning oscilloscope visualizations. Supports massive files with zero memory issues.
|
||||||
|
|
||||||
## Features
|

|
||||||
|
|
||||||
- **Multiple visualization modes**: combined, separate, all (L/R + XY)
|

|
||||||
- **Parallel rendering**: Uses all CPU cores for fast frame generation
|
|
||||||
- **High quality output**: Supports up to 4K resolution at 60fps
|
|
||||||
- **Original audio**: Copies audio stream without re-encoding (perfect sync)
|
|
||||||
- **Customizable**: Colors, resolution, FPS, line thickness
|
|
||||||
|
|
||||||
## Installation
|
## ✨ Key Features
|
||||||
|
|
||||||
### From Source
|
- **🎵 Universal Audio Support** - FLAC, MP3, WAV, AAC, and 200+ formats via ffmpeg
|
||||||
|
- **🚀 Streaming Architecture** - Processes huge files without filling RAM or SSD
|
||||||
|
- **📊 Professional Spectrometer** - FFT analyzer with smooth Cava-style animations
|
||||||
|
- **🎯 4-Quadrant Display** - Waveforms + XY pattern + spectrometer in one video
|
||||||
|
- **🎬 High Quality Output** - 4K 60fps support with customizable colors
|
||||||
|
- **⚡ Memory Efficient** - Constant RAM usage regardless of file size
|
||||||
|
|
||||||
|
## 🚀 Quick Start
|
||||||
|
|
||||||
|
### 1. Install System Dependencies
|
||||||
|
|
||||||
|
**Ubuntu/Debian:**
|
||||||
```bash
|
```bash
|
||||||
cd oscilloscope-video-gen
|
sudo apt update && sudo apt install ffmpeg
|
||||||
cargo install --path .
|
|
||||||
```
|
```
|
||||||
|
|
||||||
### Build Only
|
**macOS:**
|
||||||
|
```bash
|
||||||
|
brew install ffmpeg
|
||||||
|
```
|
||||||
|
|
||||||
|
**Windows:**
|
||||||
|
```bash
|
||||||
|
# Via winget (recommended)
|
||||||
|
winget install ffmpeg
|
||||||
|
|
||||||
|
# Or download from https://ffmpeg.org/download.html
|
||||||
|
# Make sure ffmpeg is in your PATH
|
||||||
|
```
|
||||||
|
|
||||||
|
### 2. Build the Project
|
||||||
|
|
||||||
```bash
|
```bash
|
||||||
|
# If you have the source code locally:
|
||||||
|
cd /path/to/oscilloscope-video-gen
|
||||||
|
|
||||||
|
# Build in release mode (much faster)
|
||||||
cargo build --release
|
cargo build --release
|
||||||
|
|
||||||
|
# The binary will be at: target/release/oscilloscope-video-gen
|
||||||
```
|
```
|
||||||
|
|
||||||
The binary will be at `target/release/oscilloscope-video-gen`.
|
### 3. Generate Your First Video
|
||||||
|
|
||||||
## Usage
|
```bash
|
||||||
|
# Basic usage - replace 'audio.wav' with your audio file
|
||||||
|
./target/release/oscilloscope-video-gen -i audio.wav -o oscilloscope.mp4
|
||||||
|
|
||||||
### Basic
|
# Check available options
|
||||||
|
./target/release/oscilloscope-video-gen --help
|
||||||
|
```
|
||||||
|
|
||||||
|
## 🎮 Usage Examples
|
||||||
|
|
||||||
|
### Basic Usage
|
||||||
```bash
|
```bash
|
||||||
oscilloscope-video-gen -i audio.wav -o video.mp4
|
oscilloscope-video-gen -i audio.wav -o video.mp4
|
||||||
```
|
```
|
||||||
|
|
||||||
### Full Options
|
### High Quality 4K Output
|
||||||
|
|
||||||
```bash
|
```bash
|
||||||
oscilloscope-video-gen \
|
oscilloscope-video-gen \
|
||||||
-i audio.wav \
|
-i audio.flac \
|
||||||
-o video.mp4 \
|
-o video.mp4 \
|
||||||
--width 1920 \
|
--width 3840 --height 2160 --fps 60 \
|
||||||
--height 1080 \
|
|
||||||
--fps 30 \
|
|
||||||
--mode all \
|
--mode all \
|
||||||
--quality high \
|
--quality high
|
||||||
--left-color "#00ff00" \
|
|
||||||
--right-color "#00ccff" \
|
|
||||||
--xy-color "#ff8800" \
|
|
||||||
--background "#0a0f0a" \
|
|
||||||
--line-thickness 2 \
|
|
||||||
--threads 8 \
|
|
||||||
--overwrite \
|
|
||||||
--verbose
|
|
||||||
```
|
```
|
||||||
|
|
||||||
### Options
|
### Custom Styling
|
||||||
|
|
||||||
| Option | Default | Description |
|
|
||||||
|--------|---------|-------------|
|
|
||||||
| `-i, --input` | Required | Input WAV file |
|
|
||||||
| `-o, --output` | Auto-generated | Output MP4 file |
|
|
||||||
| `--width` | 1920 | Video width in pixels |
|
|
||||||
| `--height` | 1080 | Video height in pixels |
|
|
||||||
| `--fps` | 30 | Frames per second |
|
|
||||||
| `--mode` | all | Visualization mode |
|
|
||||||
| `--quality` | high | Video quality |
|
|
||||||
| `--left-color` | #00ff00 | Left channel color (hex) |
|
|
||||||
| `--right-color` | #00ccff | Right channel color (hex) |
|
|
||||||
| `--xy-color` | #ff8800 | XY mode color (hex) |
|
|
||||||
| `--background` | #0a0f0a | Background color (hex) |
|
|
||||||
| `--show-grid` | true | Show grid lines |
|
|
||||||
| `--line-thickness` | 2 | Line thickness in pixels |
|
|
||||||
| `--threads` | All cores | Number of rendering threads |
|
|
||||||
| `--overwrite` | false | Overwrite output file |
|
|
||||||
| `--verbose` | false | Enable verbose output |
|
|
||||||
|
|
||||||
### Modes
|
|
||||||
|
|
||||||
- `combined`: Both channels merged into single waveform
|
|
||||||
- `separate`: Left on top, Right on bottom
|
|
||||||
- `all`: Left and Right on top row, XY pattern on bottom
|
|
||||||
|
|
||||||
### Quality Presets
|
|
||||||
|
|
||||||
| Preset | Video Bitrate |
|
|
||||||
|--------|---------------|
|
|
||||||
| low | 2 Mbps |
|
|
||||||
| medium | 5 Mbps |
|
|
||||||
| high | 10 Mbps |
|
|
||||||
|
|
||||||
## Requirements
|
|
||||||
|
|
||||||
- Rust 1.70+
|
|
||||||
- ffmpeg (for video encoding)
|
|
||||||
|
|
||||||
## Building
|
|
||||||
|
|
||||||
```bash
|
```bash
|
||||||
# Debug build
|
oscilloscope-video-gen \
|
||||||
cargo build
|
-i audio.mp3 \
|
||||||
|
-o video.mp4 \
|
||||||
# Release build (optimized)
|
--left-color "#ff0080" \
|
||||||
cargo build --release
|
--right-color "#8000ff" \
|
||||||
|
--background "#000000" \
|
||||||
# Build with specific number of threads
|
--line-thickness 3
|
||||||
cargo build --release --jobs 8
|
|
||||||
```
|
```
|
||||||
|
|
||||||
## Troubleshooting
|
## 🎛️ Visualization Modes
|
||||||
|
|
||||||
|
- **`all`** (default) - 4 quadrants: L/R waveforms + XY pattern + spectrometer
|
||||||
|
- **`combined`** - Single merged waveform
|
||||||
|
- **`separate`** - Left channel top, right channel bottom
|
||||||
|
|
||||||
|
## 🏗️ How It Works
|
||||||
|
|
||||||
|
This tool uses a **streaming pipeline** that pipes audio decoding directly to video encoding, eliminating temporary files and memory issues. The FFT spectrometer provides real-time frequency analysis with logarithmic scaling and temporal smoothing for smooth animations.
|
||||||
|
|
||||||
|
**No file size limits** - handles anything ffmpeg can decode, from tiny samples to 24-hour recordings.
|
||||||
|
|
||||||
|
## 📋 Requirements
|
||||||
|
|
||||||
|
- **Rust 1.70+**
|
||||||
|
- **ffmpeg** (with libavcodec, libavformat, libswresample)
|
||||||
|
|
||||||
|
## 🔧 Troubleshooting
|
||||||
|
|
||||||
### ffmpeg not found
|
### ffmpeg not found
|
||||||
|
|
||||||
Make sure ffmpeg is installed and in your PATH:
|
|
||||||
|
|
||||||
```bash
|
```bash
|
||||||
# Ubuntu/Debian
|
# Ubuntu/Debian
|
||||||
sudo apt install ffmpeg
|
sudo apt install ffmpeg
|
||||||
@@ -122,17 +115,23 @@ sudo apt install ffmpeg
|
|||||||
# macOS
|
# macOS
|
||||||
brew install ffmpeg
|
brew install ffmpeg
|
||||||
|
|
||||||
# Windows
|
# Windows (via winget or chocolatey)
|
||||||
winget install FFmpeg
|
winget install ffmpeg
|
||||||
```
|
```
|
||||||
|
|
||||||
### Out of memory
|
### File Format Issues
|
||||||
|
If your audio file isn't recognized:
|
||||||
|
```bash
|
||||||
|
# Check if ffmpeg can decode it
|
||||||
|
ffmpeg -i your_file.ext -f null -
|
||||||
|
```
|
||||||
|
|
||||||
For very long audio files, try:
|
## 📄 License
|
||||||
- Lower resolution (`--width 1280 --height 720`)
|
|
||||||
- Lower FPS (`--fps 24`)
|
|
||||||
- Fewer threads (`--threads 4`)
|
|
||||||
|
|
||||||
## License
|
**AGPL-3.0-or-later** - Free software that respects your freedom.
|
||||||
|
|
||||||
MIT
|
See [LICENSE](LICENSE) for full terms.
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
*Made with ❤️ using Rust and ffmpeg*
|
||||||
|
|||||||
Binary file not shown.
|
After Width: | Height: | Size: 227 KiB |
+113
-74
@@ -1,97 +1,136 @@
|
|||||||
//! WAV audio decoding module.
|
//! Audio streaming module using ffmpeg.
|
||||||
//!
|
|
||||||
//! Handles reading and decoding WAV files into normalized sample data.
|
|
||||||
|
|
||||||
use anyhow::{anyhow, Context, Result};
|
use anyhow::{anyhow, Context, Result};
|
||||||
use hound::WavReader;
|
|
||||||
use std::path::Path;
|
use std::path::Path;
|
||||||
|
use std::process::{Command, Stdio};
|
||||||
|
use std::io::{Read, BufReader};
|
||||||
|
|
||||||
/// Normalized audio sample data.
|
/// A streaming audio reader that decodes chunks on demand.
|
||||||
#[derive(Debug, Clone)]
|
pub struct AudioStream {
|
||||||
pub struct AudioData {
|
child: std::process::Child,
|
||||||
/// Left channel samples, normalized to [-1.0, 1.0]
|
reader: BufReader<std::process::ChildStdout>,
|
||||||
pub left_channel: Vec<f32>,
|
|
||||||
/// Right channel samples, normalized to [-1.0, 1.0]
|
|
||||||
pub right_channel: Vec<f32>,
|
|
||||||
/// Sample rate in Hz
|
|
||||||
pub sample_rate: u32,
|
pub sample_rate: u32,
|
||||||
/// Duration in seconds
|
|
||||||
pub duration: f64,
|
pub duration: f64,
|
||||||
|
/// Current sample position in the stream
|
||||||
|
pub current_sample_pos: usize,
|
||||||
|
/// Sliding window of left channel samples
|
||||||
|
left_window: Vec<f32>,
|
||||||
|
/// Sliding window of right channel samples
|
||||||
|
right_window: Vec<f32>,
|
||||||
|
/// Window size for lookahead (e.g. FFT)
|
||||||
|
window_size: usize,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl AudioData {
|
impl AudioStream {
|
||||||
/// Load and decode a WAV file.
|
/// Create a new AudioStream from a file path.
|
||||||
///
|
pub fn new(file_path: &Path, window_size: usize) -> Result<Self> {
|
||||||
/// # Arguments
|
// 1. Get metadata using ffprobe
|
||||||
///
|
let output = Command::new("ffprobe")
|
||||||
/// * `file_path` - Path to the WAV file
|
.args([
|
||||||
///
|
"-v", "error",
|
||||||
/// # Returns
|
"-select_streams", "a:0",
|
||||||
///
|
"-show_entries", "stream=sample_rate,duration",
|
||||||
/// `Result<AudioData>` containing the decoded audio samples
|
"-of", "default=noprint_wrappers=1:nokey=1",
|
||||||
///
|
])
|
||||||
/// # Errors
|
.arg(file_path)
|
||||||
///
|
.output()
|
||||||
/// Returns an error if:
|
.with_context(|| "Failed to run ffprobe")?;
|
||||||
/// - The file cannot be opened
|
|
||||||
/// - Bit depth is not 16-bit
|
|
||||||
/// - Number of channels is not 1 or 2
|
|
||||||
pub fn from_wav(file_path: &Path) -> Result<Self> {
|
|
||||||
let mut reader = WavReader::open(file_path)
|
|
||||||
.with_context(|| format!("Failed to open WAV file: {}", file_path.display()))?;
|
|
||||||
|
|
||||||
let spec = reader.spec();
|
if !output.status.success() {
|
||||||
|
return Err(anyhow!("ffprobe failed: {}", String::from_utf8_lossy(&output.stderr)));
|
||||||
if spec.bits_per_sample != 16 {
|
|
||||||
return Err(anyhow!(
|
|
||||||
"Unsupported bit depth: {}. Only 16-bit WAV is supported.",
|
|
||||||
spec.bits_per_sample
|
|
||||||
));
|
|
||||||
}
|
}
|
||||||
|
|
||||||
if spec.channels != 1 && spec.channels != 2 {
|
let metadata = String::from_utf8_lossy(&output.stdout);
|
||||||
return Err(anyhow!(
|
let mut lines = metadata.lines();
|
||||||
"Unsupported number of channels: {}. Only mono and stereo are supported.",
|
let sample_rate: u32 = lines.next()
|
||||||
spec.channels
|
.ok_or_else(|| anyhow!("Could not get sample rate"))?
|
||||||
));
|
.parse()?;
|
||||||
}
|
let duration: f64 = lines.next()
|
||||||
|
.ok_or_else(|| anyhow!("Could not get duration"))?
|
||||||
|
.parse()?;
|
||||||
|
|
||||||
let sample_rate = spec.sample_rate;
|
// 2. Start ffmpeg for streaming
|
||||||
let samples: Vec<i16> = reader.samples().map(|s| s.unwrap_or(0)).collect();
|
let mut child = Command::new("ffmpeg")
|
||||||
let total_samples = samples.len() / spec.channels as usize;
|
.arg("-i")
|
||||||
let duration = total_samples as f64 / sample_rate as f64;
|
.arg(file_path)
|
||||||
|
.args([
|
||||||
|
"-f", "s16le",
|
||||||
|
"-acodec", "pcm_s16le",
|
||||||
|
"-ar", &sample_rate.to_string(),
|
||||||
|
"-ac", "2",
|
||||||
|
"-",
|
||||||
|
])
|
||||||
|
.stdout(Stdio::piped())
|
||||||
|
.stderr(Stdio::null())
|
||||||
|
.spawn()
|
||||||
|
.with_context(|| "Failed to spawn ffmpeg")?;
|
||||||
|
|
||||||
let mut left_channel = Vec::with_capacity(total_samples);
|
let stdout = child.stdout.take().ok_or_else(|| anyhow!("Failed to open ffmpeg stdout"))?;
|
||||||
let mut right_channel = Vec::with_capacity(total_samples);
|
let reader = BufReader::new(stdout);
|
||||||
|
|
||||||
for i in 0..total_samples {
|
Ok(Self {
|
||||||
let offset = i * spec.channels as usize;
|
child,
|
||||||
let left_sample = samples[offset] as f32 / 32768.0;
|
reader,
|
||||||
left_channel.push(left_sample);
|
|
||||||
|
|
||||||
if spec.channels >= 2 {
|
|
||||||
let right_sample = samples[offset + 1] as f32 / 32768.0;
|
|
||||||
right_channel.push(right_sample);
|
|
||||||
} else {
|
|
||||||
right_channel.push(left_sample);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
Ok(AudioData {
|
|
||||||
left_channel,
|
|
||||||
right_channel,
|
|
||||||
sample_rate,
|
sample_rate,
|
||||||
duration,
|
duration,
|
||||||
|
current_sample_pos: 0,
|
||||||
|
left_window: Vec::with_capacity(window_size),
|
||||||
|
right_window: Vec::with_capacity(window_size),
|
||||||
|
window_size,
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Get the total number of samples.
|
/// Read samples up to the target position.
|
||||||
pub fn len(&self) -> usize {
|
pub fn fill_until(&mut self, target_pos: usize) -> Result<()> {
|
||||||
self.left_channel.len()
|
if target_pos <= self.current_sample_pos {
|
||||||
|
return Ok(());
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Check if the audio data is empty.
|
let samples_to_read = target_pos - self.current_sample_pos;
|
||||||
pub fn is_empty(&self) -> bool {
|
let mut buffer = vec![0i16; samples_to_read * 2]; // 2 channels
|
||||||
self.left_channel.is_empty()
|
|
||||||
|
// Read raw PCM bytes
|
||||||
|
let byte_buffer: &mut [u8] = bytemuck::cast_slice_mut(&mut buffer);
|
||||||
|
self.reader.read_exact(byte_buffer)
|
||||||
|
.with_context(|| "Failed to read audio data from ffmpeg")?;
|
||||||
|
|
||||||
|
// Process samples and update sliding window
|
||||||
|
for chunk in buffer.chunks_exact(2) {
|
||||||
|
let left = chunk[0] as f32 / 32768.0;
|
||||||
|
let right = chunk[1] as f32 / 32768.0;
|
||||||
|
|
||||||
|
self.left_window.push(left);
|
||||||
|
self.right_window.push(right);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Maintain window size (keep at least window_size samples)
|
||||||
|
if self.left_window.len() > self.window_size * 2 {
|
||||||
|
let drain_amount = self.left_window.len() - self.window_size;
|
||||||
|
self.left_window.drain(0..drain_amount);
|
||||||
|
self.right_window.drain(0..drain_amount);
|
||||||
|
}
|
||||||
|
|
||||||
|
self.current_sample_pos = target_pos;
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Get the last N samples for rendering.
|
||||||
|
pub fn get_last_samples(&self, count: usize) -> (&[f32], &[f32]) {
|
||||||
|
let actual_count = count.min(self.left_window.len());
|
||||||
|
let start = self.left_window.len() - actual_count;
|
||||||
|
(&self.left_window[start..], &self.right_window[start..])
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Get a window of samples for FFT.
|
||||||
|
pub fn get_fft_window(&self, size: usize) -> (&[f32], &[f32]) {
|
||||||
|
let actual_size = size.min(self.left_window.len());
|
||||||
|
let start = self.left_window.len() - actual_size;
|
||||||
|
(&self.left_window[start..], &self.right_window[start..])
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Drop for AudioStream {
|
||||||
|
fn drop(&mut self) {
|
||||||
|
let _ = self.child.kill();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
+3
-3
@@ -20,6 +20,6 @@ pub mod audio;
|
|||||||
pub mod render;
|
pub mod render;
|
||||||
pub mod video;
|
pub mod video;
|
||||||
|
|
||||||
pub use audio::AudioData;
|
pub use audio::AudioStream;
|
||||||
pub use render::{RenderMode, RenderOptions};
|
pub use render::{stream_frames, RenderMode, RenderOptions};
|
||||||
pub use video::encode_video;
|
pub use video::VideoEncoder;
|
||||||
|
|||||||
+44
-123
@@ -1,7 +1,7 @@
|
|||||||
//! Oscilloscope Video Generator
|
//! Oscilloscope Video Generator
|
||||||
//!
|
//!
|
||||||
//! A high-performance tool for generating oscilloscope-style visualizations
|
//! A high-performance tool for generating oscilloscope-style visualizations
|
||||||
//! from audio files. Uses parallel rendering for fast processing.
|
//! from audio files. Uses streaming for handling large files.
|
||||||
|
|
||||||
use anyhow::{Context, Result};
|
use anyhow::{Context, Result};
|
||||||
use clap::{Parser, ValueEnum};
|
use clap::{Parser, ValueEnum};
|
||||||
@@ -10,15 +10,13 @@ use std::path::PathBuf;
|
|||||||
use std::sync::atomic::{AtomicUsize, Ordering};
|
use std::sync::atomic::{AtomicUsize, Ordering};
|
||||||
use std::sync::Arc;
|
use std::sync::Arc;
|
||||||
|
|
||||||
use oscilloscope_video_gen::audio::AudioData;
|
use oscilloscope_video_gen::audio::AudioStream;
|
||||||
use oscilloscope_video_gen::render::{parse_rgb_hex, render_frames, RenderMode, RenderOptions};
|
use oscilloscope_video_gen::render::{parse_rgb_hex, stream_frames, RenderMode, RenderOptions};
|
||||||
use oscilloscope_video_gen::video::{encode_video, cleanup_tmp_dir, VideoQuality};
|
use oscilloscope_video_gen::video::{VideoEncoder, VideoQuality};
|
||||||
|
|
||||||
#[derive(Debug, Clone, Copy, ValueEnum)]
|
#[derive(Debug, Clone, Copy, ValueEnum)]
|
||||||
enum OutputMode {
|
enum OutputMode {
|
||||||
Combined,
|
Combined, Separate, All, Spectrometer,
|
||||||
Separate,
|
|
||||||
All,
|
|
||||||
}
|
}
|
||||||
|
|
||||||
impl From<OutputMode> for RenderMode {
|
impl From<OutputMode> for RenderMode {
|
||||||
@@ -27,15 +25,14 @@ impl From<OutputMode> for RenderMode {
|
|||||||
OutputMode::Combined => RenderMode::Combined,
|
OutputMode::Combined => RenderMode::Combined,
|
||||||
OutputMode::Separate => RenderMode::Separate,
|
OutputMode::Separate => RenderMode::Separate,
|
||||||
OutputMode::All => RenderMode::All,
|
OutputMode::All => RenderMode::All,
|
||||||
|
OutputMode::Spectrometer => RenderMode::Spectrometer,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, Clone, Copy, ValueEnum)]
|
#[derive(Debug, Clone, Copy, ValueEnum)]
|
||||||
enum OutputQuality {
|
enum OutputQuality {
|
||||||
Low,
|
Low, Medium, High,
|
||||||
Medium,
|
|
||||||
High,
|
|
||||||
}
|
}
|
||||||
|
|
||||||
impl From<OutputQuality> for VideoQuality {
|
impl From<OutputQuality> for VideoQuality {
|
||||||
@@ -50,10 +47,9 @@ impl From<OutputQuality> for VideoQuality {
|
|||||||
|
|
||||||
/// Generate oscilloscope visualizations from audio files
|
/// Generate oscilloscope visualizations from audio files
|
||||||
#[derive(Parser, Debug)]
|
#[derive(Parser, Debug)]
|
||||||
#[command(name = "oscilloscope-video-gen")]
|
#[command(name = "oscilloscope-video-gen", author, version, about)]
|
||||||
#[command(author, version, about, long_about = None)]
|
|
||||||
struct Args {
|
struct Args {
|
||||||
/// Input audio file (WAV)
|
/// Input audio file
|
||||||
#[arg(short, long)]
|
#[arg(short, long)]
|
||||||
input: PathBuf,
|
input: PathBuf,
|
||||||
|
|
||||||
@@ -61,39 +57,39 @@ struct Args {
|
|||||||
#[arg(short, long)]
|
#[arg(short, long)]
|
||||||
output: Option<PathBuf>,
|
output: Option<PathBuf>,
|
||||||
|
|
||||||
/// Video width (default: 1920)
|
/// Video width
|
||||||
#[arg(long, default_value = "1920")]
|
#[arg(long, default_value = "1920")]
|
||||||
width: u32,
|
width: u32,
|
||||||
|
|
||||||
/// Video height (default: 1080)
|
/// Video height
|
||||||
#[arg(long, default_value = "1080")]
|
#[arg(long, default_value = "1080")]
|
||||||
height: u32,
|
height: u32,
|
||||||
|
|
||||||
/// Frames per second (default: 30)
|
/// Frames per second
|
||||||
#[arg(long, default_value = "30")]
|
#[arg(long, default_value = "30")]
|
||||||
fps: u32,
|
fps: u32,
|
||||||
|
|
||||||
/// Display mode: combined, separate, all
|
/// Display mode
|
||||||
#[arg(long, value_enum, default_value = "all")]
|
#[arg(long, value_enum, default_value = "all")]
|
||||||
mode: OutputMode,
|
mode: OutputMode,
|
||||||
|
|
||||||
/// Quality: low, medium, high
|
/// Quality preset
|
||||||
#[arg(long, value_enum, default_value = "high")]
|
#[arg(long, value_enum, default_value = "high")]
|
||||||
quality: OutputQuality,
|
quality: OutputQuality,
|
||||||
|
|
||||||
/// Left channel color (RGB hex, default: #00ff00)
|
/// Left channel color
|
||||||
#[arg(long, default_value = "#00ff00")]
|
#[arg(long, default_value = "#00ff00")]
|
||||||
left_color: String,
|
left_color: String,
|
||||||
|
|
||||||
/// Right channel color (RGB hex, default: #00ccff)
|
/// Right channel color
|
||||||
#[arg(long, default_value = "#00ccff")]
|
#[arg(long, default_value = "#00ccff")]
|
||||||
right_color: String,
|
right_color: String,
|
||||||
|
|
||||||
/// XY mode color (RGB hex, default: #ff8800)
|
/// XY mode color
|
||||||
#[arg(long, default_value = "#ff8800")]
|
#[arg(long, default_value = "#ff8800")]
|
||||||
xy_color: String,
|
xy_color: String,
|
||||||
|
|
||||||
/// Background color (RGB hex, default: #0a0f0a)
|
/// Background color
|
||||||
#[arg(long, default_value = "#0a0f0a")]
|
#[arg(long, default_value = "#0a0f0a")]
|
||||||
background: String,
|
background: String,
|
||||||
|
|
||||||
@@ -101,15 +97,11 @@ struct Args {
|
|||||||
#[arg(long, default_value = "true")]
|
#[arg(long, default_value = "true")]
|
||||||
show_grid: bool,
|
show_grid: bool,
|
||||||
|
|
||||||
/// Line thickness (default: 2)
|
/// Line thickness
|
||||||
#[arg(long, default_value = "2")]
|
#[arg(long, default_value = "2")]
|
||||||
line_thickness: u32,
|
line_thickness: u32,
|
||||||
|
|
||||||
/// Number of rendering threads
|
/// Overwrite output file
|
||||||
#[arg(long)]
|
|
||||||
threads: Option<usize>,
|
|
||||||
|
|
||||||
/// Overwrite output file if it exists
|
|
||||||
#[arg(long, default_value = "false")]
|
#[arg(long, default_value = "false")]
|
||||||
overwrite: bool,
|
overwrite: bool,
|
||||||
|
|
||||||
@@ -121,132 +113,61 @@ struct Args {
|
|||||||
fn main() -> Result<()> {
|
fn main() -> Result<()> {
|
||||||
let args = Args::parse();
|
let args = Args::parse();
|
||||||
|
|
||||||
// Set number of threads
|
let left_color = parse_rgb_hex(&args.left_color)?;
|
||||||
if let Some(threads) = args.threads {
|
let right_color = parse_rgb_hex(&args.right_color)?;
|
||||||
rayon::ThreadPoolBuilder::new()
|
let xy_color = parse_rgb_hex(&args.xy_color)?;
|
||||||
.num_threads(threads)
|
let background = parse_rgb_hex(&args.background)?;
|
||||||
.build_global()
|
|
||||||
.unwrap();
|
|
||||||
}
|
|
||||||
|
|
||||||
// Parse colors
|
|
||||||
let left_color =
|
|
||||||
parse_rgb_hex(&args.left_color).context("Failed to parse left_color")?;
|
|
||||||
let right_color =
|
|
||||||
parse_rgb_hex(&args.right_color).context("Failed to parse right_color")?;
|
|
||||||
let xy_color = parse_rgb_hex(&args.xy_color).context("Failed to parse xy_color")?;
|
|
||||||
let background =
|
|
||||||
parse_rgb_hex(&args.background).context("Failed to parse background")?;
|
|
||||||
|
|
||||||
// Create options
|
|
||||||
let options = RenderOptions {
|
let options = RenderOptions {
|
||||||
width: args.width,
|
width: args.width, height: args.height, fps: args.fps,
|
||||||
height: args.height,
|
|
||||||
fps: args.fps,
|
|
||||||
mode: args.mode.into(),
|
mode: args.mode.into(),
|
||||||
left_color,
|
left_color, right_color, xy_color, background,
|
||||||
right_color,
|
show_grid: args.show_grid, line_thickness: args.line_thickness,
|
||||||
xy_color,
|
|
||||||
background,
|
|
||||||
show_grid: args.show_grid,
|
|
||||||
line_thickness: args.line_thickness,
|
|
||||||
};
|
};
|
||||||
|
|
||||||
// Determine output path
|
let output = args.output.clone().unwrap_or_else(|| {
|
||||||
let output = match args.output {
|
|
||||||
Some(path) => path,
|
|
||||||
None => {
|
|
||||||
let mut path = args.input.clone();
|
let mut path = args.input.clone();
|
||||||
path.set_extension("mp4");
|
path.set_extension("mp4");
|
||||||
path
|
path
|
||||||
}
|
});
|
||||||
};
|
|
||||||
|
|
||||||
if args.verbose {
|
|
||||||
println!("Oscilloscope Video Generator");
|
|
||||||
println!("============================");
|
|
||||||
println!("Input: {}", args.input.display());
|
|
||||||
println!("Output: {}", output.display());
|
|
||||||
println!("Resolution: {}x{}", args.width, args.height);
|
|
||||||
println!("FPS: {}", args.fps);
|
|
||||||
println!("Mode: {:?}", args.mode);
|
|
||||||
println!("Quality: {:?}", args.quality);
|
|
||||||
println!("Threads: {:?}", args.threads.unwrap_or_else(|| rayon::current_num_threads()));
|
|
||||||
println!();
|
|
||||||
} else {
|
|
||||||
println!("Oscilloscope Video Generator");
|
println!("Oscilloscope Video Generator");
|
||||||
println!("============================");
|
println!("============================");
|
||||||
println!("Input: {}", args.input.display());
|
println!("Input: {}", args.input.display());
|
||||||
println!("Output: {}", output.display());
|
println!("Output: {}", output.display());
|
||||||
println!("Resolution: {}x{} @ {}fps", args.width, args.height, args.fps);
|
println!("Resolution: {}x{} @ {}fps", args.width, args.height, args.fps);
|
||||||
println!("Mode: {:?}", args.mode);
|
|
||||||
println!();
|
println!();
|
||||||
}
|
|
||||||
|
|
||||||
// Decode audio
|
// Initialize streaming audio reader (with FFT window size lookahead)
|
||||||
let audio_data = AudioData::from_wav(&args.input)
|
let mut audio_stream = AudioStream::new(&args.input, 2048)
|
||||||
.with_context(|| format!("Failed to decode audio: {}", args.input.display()))?;
|
.with_context(|| format!("Failed to open audio stream: {}", args.input.display()))?;
|
||||||
|
|
||||||
if args.verbose {
|
|
||||||
println!(
|
|
||||||
"Audio: {}Hz, {:.2}s duration, {} samples",
|
|
||||||
audio_data.sample_rate,
|
|
||||||
audio_data.duration,
|
|
||||||
audio_data.len()
|
|
||||||
);
|
|
||||||
}
|
|
||||||
|
|
||||||
// Create temp directory
|
|
||||||
let tmp_dir = std::env::temp_dir().join("oscilloscope-render");
|
|
||||||
if tmp_dir.exists() {
|
|
||||||
cleanup_tmp_dir(&tmp_dir);
|
|
||||||
}
|
|
||||||
std::fs::create_dir_all(&tmp_dir)?;
|
|
||||||
|
|
||||||
// Progress callback
|
|
||||||
let progress = Arc::new(AtomicUsize::new(0));
|
let progress = Arc::new(AtomicUsize::new(0));
|
||||||
let progress_callback = move |percent: f64, current: usize, total: usize| {
|
let progress_callback = move |percent: f64, current: usize, total: usize| {
|
||||||
let prev = progress.fetch_add(0, Ordering::SeqCst);
|
let prev = progress.fetch_add(0, Ordering::SeqCst);
|
||||||
if current - prev >= 30 || current == total || current == 1 {
|
if current - prev >= 30 || current == total {
|
||||||
progress.store(current, Ordering::SeqCst);
|
progress.store(current, Ordering::SeqCst);
|
||||||
print!("\rRendering: {:.0}% ({}/{})", percent, current, total);
|
print!("\rRendering and Encoding: {:.1}% ({}/{})", percent, current, total);
|
||||||
let _ = std::io::stdout().flush();
|
let _ = std::io::stdout().flush();
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
// Render frames
|
|
||||||
println!("Rendering frames...");
|
|
||||||
let frame_files = render_frames(&audio_data, &options, &tmp_dir, &progress_callback)?;
|
|
||||||
println!();
|
|
||||||
println!("Rendered {} frames", frame_files.len());
|
|
||||||
|
|
||||||
// Check if output exists and handle overwrite
|
|
||||||
if output.exists() && !args.overwrite {
|
if output.exists() && !args.overwrite {
|
||||||
return Err(anyhow::anyhow!(
|
return Err(anyhow::anyhow!("Output exists. Use --overwrite."));
|
||||||
"Output file already exists: {}. Use --overwrite to replace it.",
|
|
||||||
output.display()
|
|
||||||
));
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// Encode video
|
let mut encoder = VideoEncoder::new(
|
||||||
encode_video(
|
&args.input, &output, args.width, args.height, args.fps,
|
||||||
&frame_files,
|
args.quality.into(), args.overwrite,
|
||||||
&args.input,
|
)?;
|
||||||
&output,
|
|
||||||
args.fps,
|
|
||||||
args.quality.into(),
|
|
||||||
args.overwrite,
|
|
||||||
)
|
|
||||||
.context("Failed to encode video")?;
|
|
||||||
|
|
||||||
// Cleanup
|
println!("Starting streaming process...");
|
||||||
println!("Cleaning up temporary files...");
|
stream_frames(&mut audio_stream, &options, &mut encoder, &progress_callback)?;
|
||||||
cleanup_tmp_dir(&tmp_dir);
|
println!();
|
||||||
|
|
||||||
let file_size = std::fs::metadata(&output)
|
encoder.finish().context("Failed to finish video encoding")?;
|
||||||
.map(|m| m.len())
|
|
||||||
.unwrap_or(0);
|
|
||||||
|
|
||||||
|
let file_size = std::fs::metadata(&output).map(|m| m.len()).unwrap_or(0);
|
||||||
println!("\nDone!");
|
println!("\nDone!");
|
||||||
println!("Output: {}", output.display());
|
println!("Output: {}", output.display());
|
||||||
println!("Size: {:.2} MB", file_size as f64 / 1_000_000.0);
|
println!("Size: {:.2} MB", file_size as f64 / 1_000_000.0);
|
||||||
|
|||||||
+219
-282
@@ -1,314 +1,251 @@
|
|||||||
//! Frame rendering module.
|
//! Frame rendering module with true single-pass streaming.
|
||||||
//!
|
|
||||||
//! Contains all the logic for drawing oscilloscope visualizations.
|
|
||||||
|
|
||||||
use crate::audio::AudioData;
|
use crate::audio::AudioStream;
|
||||||
use anyhow::{anyhow, Context, Result};
|
use crate::video::VideoEncoder;
|
||||||
|
use anyhow::{anyhow, Result};
|
||||||
use image::ImageBuffer;
|
use image::ImageBuffer;
|
||||||
use std::path::{Path, PathBuf};
|
use rustfft::{num_complex::Complex, FftPlanner};
|
||||||
|
use std::cell::RefCell;
|
||||||
|
|
||||||
/// Render mode for the oscilloscope visualization.
|
// --- Constants ---
|
||||||
#[derive(Debug, Clone, Copy, clap::ValueEnum)]
|
const FFT_SIZE: usize = 2048;
|
||||||
|
const MIN_FREQ: f32 = 20.0;
|
||||||
|
const MAX_FREQ: f32 = 20000.0;
|
||||||
|
const FREQ_BOOST_FACTOR: f32 = 5.0;
|
||||||
|
const DYNAMIC_RANGE_SCALE: f32 = 20.0;
|
||||||
|
const NOISE_FLOOR: f32 = 0.05;
|
||||||
|
const SMOOTH_RISE: f32 = 0.6;
|
||||||
|
const SMOOTH_FALL: f32 = 0.3;
|
||||||
|
|
||||||
|
thread_local! {
|
||||||
|
static FFT_PLANNER: RefCell<FftPlanner<f32>> = RefCell::new(FftPlanner::new());
|
||||||
|
static HANN_WINDOW: Vec<f32> = (0..FFT_SIZE)
|
||||||
|
.map(|i| 0.5 * (1.0 - (2.0 * std::f32::consts::PI * i as f32 / (FFT_SIZE - 1) as f32).cos()))
|
||||||
|
.collect();
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug, Clone, Copy, clap::ValueEnum, PartialEq, Eq)]
|
||||||
pub enum RenderMode {
|
pub enum RenderMode {
|
||||||
/// Both channels merged into a single waveform
|
Combined, Separate, All, Spectrometer,
|
||||||
Combined,
|
|
||||||
/// Left channel on top, Right channel on bottom
|
|
||||||
Separate,
|
|
||||||
/// Left and Right on top row, XY on bottom
|
|
||||||
All,
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Rendering options for the visualization.
|
|
||||||
#[derive(Debug, Clone)]
|
#[derive(Debug, Clone)]
|
||||||
pub struct RenderOptions {
|
pub struct RenderOptions {
|
||||||
pub width: u32,
|
pub width: u32, pub height: u32, pub fps: u32,
|
||||||
pub height: u32,
|
|
||||||
pub fps: u32,
|
|
||||||
pub mode: RenderMode,
|
pub mode: RenderMode,
|
||||||
pub left_color: image::Rgb<u8>,
|
pub left_color: image::Rgb<u8>, pub right_color: image::Rgb<u8>, pub xy_color: image::Rgb<u8>,
|
||||||
pub right_color: image::Rgb<u8>,
|
|
||||||
pub xy_color: image::Rgb<u8>,
|
|
||||||
pub background: image::Rgb<u8>,
|
pub background: image::Rgb<u8>,
|
||||||
pub show_grid: bool,
|
pub show_grid: bool, pub line_thickness: u32,
|
||||||
pub line_thickness: u32,
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Draw a line between two points using Bresenham's algorithm.
|
/// Compute and update smoothed spectrometer bars for the current frame.
|
||||||
pub fn draw_line(
|
fn update_spectrometer(
|
||||||
buffer: &mut ImageBuffer<image::Rgb<u8>, Vec<u8>>,
|
left: &[f32],
|
||||||
x0: i32,
|
right: &[f32],
|
||||||
y0: i32,
|
prev_bars: &mut [f32],
|
||||||
x1: i32,
|
sample_rate: u32,
|
||||||
y1: i32,
|
|
||||||
color: image::Rgb<u8>,
|
|
||||||
) {
|
) {
|
||||||
|
let mut buffer: Vec<Complex<f32>> = (0..FFT_SIZE).map(|i| {
|
||||||
|
let l = *left.get(i).unwrap_or(&0.0);
|
||||||
|
let r = *right.get(i).unwrap_or(&0.0);
|
||||||
|
Complex::new(l + r, 0.0)
|
||||||
|
}).collect();
|
||||||
|
|
||||||
|
HANN_WINDOW.with(|win| {
|
||||||
|
for (sample, &w) in buffer.iter_mut().zip(win.iter()) {
|
||||||
|
sample.re *= w;
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
let fft = FFT_PLANNER.with(|p| p.borrow_mut().plan_fft_forward(FFT_SIZE));
|
||||||
|
fft.process(&mut buffer);
|
||||||
|
|
||||||
|
let nyquist = sample_rate as f32 / 2.0;
|
||||||
|
let spectrum: Vec<f32> = buffer[1..FFT_SIZE / 2].iter().map(|c| c.norm() / FFT_SIZE as f32).collect();
|
||||||
|
let num_bars = prev_bars.len();
|
||||||
|
|
||||||
|
for i in 0..num_bars {
|
||||||
|
let f_start = MIN_FREQ * (MAX_FREQ / MIN_FREQ).powf(i as f32 / num_bars as f32);
|
||||||
|
let f_end = MIN_FREQ * (MAX_FREQ / MIN_FREQ).powf((i + 1) as f32 / num_bars as f32);
|
||||||
|
let bin_start = (f_start / nyquist * spectrum.len() as f32).floor() as usize;
|
||||||
|
let bin_end = (f_end / nyquist * spectrum.len() as f32).ceil() as usize;
|
||||||
|
let bin_end = bin_end.max(bin_start + 1).min(spectrum.len());
|
||||||
|
|
||||||
|
let mut magnitude = 0.0f32;
|
||||||
|
if bin_start < spectrum.len() {
|
||||||
|
for k in bin_start..bin_end {
|
||||||
|
magnitude = magnitude.max(spectrum[k]);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
let freq_factor = 1.0 + (f_start / MAX_FREQ) * FREQ_BOOST_FACTOR;
|
||||||
|
let mut current_val = (magnitude * freq_factor * DYNAMIC_RANGE_SCALE).sqrt().min(1.0);
|
||||||
|
if current_val < NOISE_FLOOR { current_val = 0.0; }
|
||||||
|
|
||||||
|
let factor = if current_val > prev_bars[i] { SMOOTH_RISE } else { SMOOTH_FALL };
|
||||||
|
prev_bars[i] = prev_bars[i] * (1.0 - factor) + current_val * factor;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn stream_frames(
|
||||||
|
audio_stream: &mut AudioStream,
|
||||||
|
options: &RenderOptions,
|
||||||
|
encoder: &mut VideoEncoder,
|
||||||
|
progress_callback: &(impl Fn(f64, usize, usize) + Send + Sync),
|
||||||
|
) -> Result<()> {
|
||||||
|
let total_frames = ((audio_stream.duration * options.fps as f64) as usize).max(1);
|
||||||
|
let num_bars = if options.mode == RenderMode::Spectrometer { 64 } else { 32 };
|
||||||
|
let mut smoothed_bars = vec![0.0; num_bars];
|
||||||
|
|
||||||
|
println!("Processing {} frames...", total_frames);
|
||||||
|
|
||||||
|
for frame_idx in 0..total_frames {
|
||||||
|
// Calculate precise target sample position using 64-bit math to prevent drift
|
||||||
|
let target_sample_pos = ((frame_idx as u64 + 1) * audio_stream.sample_rate as u64 / options.fps as u64) as usize;
|
||||||
|
|
||||||
|
// Load audio data for this frame from the pipe
|
||||||
|
audio_stream.fill_until(target_sample_pos)?;
|
||||||
|
|
||||||
|
let samples_per_frame = (audio_stream.sample_rate / options.fps) as usize;
|
||||||
|
let (left, right) = audio_stream.get_last_samples(samples_per_frame);
|
||||||
|
let (fft_l, fft_r) = audio_stream.get_fft_window(FFT_SIZE);
|
||||||
|
|
||||||
|
// Process spectrometer if active
|
||||||
|
if matches!(options.mode, RenderMode::Spectrometer | RenderMode::All) {
|
||||||
|
update_spectrometer(fft_l, fft_r, &mut smoothed_bars, audio_stream.sample_rate);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Render the frame sequentially (correct order for smoothing)
|
||||||
|
let frame = draw_frame(left, right, &smoothed_bars, options);
|
||||||
|
encoder.write_frame(&frame.into_raw())?;
|
||||||
|
|
||||||
|
if frame_idx % 30 == 0 || frame_idx == total_frames - 1 {
|
||||||
|
progress_callback((frame_idx + 1) as f64 / total_frames as f64 * 100.0, frame_idx + 1, total_frames);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
fn render_bars(
|
||||||
|
buffer: &mut ImageBuffer<image::Rgb<u8>, Vec<u8>>,
|
||||||
|
bars: &[f32],
|
||||||
|
x_offset: u32, y_offset: u32, width: u32, height: u32, color: image::Rgb<u8>,
|
||||||
|
) {
|
||||||
|
const BAR_SPACING: u32 = 1;
|
||||||
|
let num_bars = bars.len();
|
||||||
|
let bar_width = (width.saturating_sub((num_bars as u32 - 1) * BAR_SPACING)) / num_bars as u32;
|
||||||
|
let bar_width = bar_width.max(1);
|
||||||
|
for (i, &val) in bars.iter().enumerate() {
|
||||||
|
let bar_height = (val * height as f32) as u32;
|
||||||
|
let x = x_offset + i as u32 * (bar_width + BAR_SPACING);
|
||||||
|
for y in 0..bar_height {
|
||||||
|
let pixel_y = y_offset + height - 1 - y;
|
||||||
|
for dx in 0..bar_width {
|
||||||
|
let pixel_x = x + dx;
|
||||||
|
if pixel_x < buffer.width() && pixel_y < buffer.height() {
|
||||||
|
buffer.put_pixel(pixel_x, pixel_y, color);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn draw_frame(
|
||||||
|
left: &[f32],
|
||||||
|
right: &[f32],
|
||||||
|
smoothed_bars: &[f32],
|
||||||
|
options: &RenderOptions,
|
||||||
|
) -> ImageBuffer<image::Rgb<u8>, Vec<u8>> {
|
||||||
|
let (width, height) = (options.width, options.height);
|
||||||
|
let mut buffer = ImageBuffer::new(width, height);
|
||||||
|
for p in buffer.pixels_mut() { *p = options.background; }
|
||||||
|
|
||||||
|
if options.show_grid { draw_graticule(&mut buffer, options.left_color); }
|
||||||
|
|
||||||
|
match options.mode {
|
||||||
|
RenderMode::All => {
|
||||||
|
let (hh, hw) = (height / 2, width / 2);
|
||||||
|
|
||||||
|
for x in 0..width { buffer.put_pixel(x, hh, image::Rgb([40, 40, 40])); }
|
||||||
|
for y in 0..height { buffer.put_pixel(hw, y, image::Rgb([40, 40, 40])); }
|
||||||
|
|
||||||
|
let samples_per_pixel = left.len() as f32 / hw as f32;
|
||||||
|
|
||||||
|
// Top-left
|
||||||
|
let mut pl = (hh/2) as i32;
|
||||||
|
for x in 0..hw {
|
||||||
|
let idx = (x as f32 * samples_per_pixel) as usize;
|
||||||
|
if idx >= left.len() { break; }
|
||||||
|
let yl = (hh/2) as i32 - (left[idx] * (hh as f32 * 0.35)) as i32;
|
||||||
|
draw_line(&mut buffer, x as i32, pl, x as i32, yl, options.left_color);
|
||||||
|
pl = yl;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Top-right
|
||||||
|
let mut pr = (hh/2) as i32;
|
||||||
|
for x in 0..hw {
|
||||||
|
let idx = (x as f32 * samples_per_pixel) as usize;
|
||||||
|
if idx >= right.len() { break; }
|
||||||
|
let yr = (hh/2) as i32 - (right[idx] * (hh as f32 * 0.35)) as i32;
|
||||||
|
draw_line(&mut buffer, (hw+x) as i32, pr, (hw+x) as i32, yr, options.right_color);
|
||||||
|
pr = yr;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Bottom-left
|
||||||
|
let (cx, cy) = (hw/2, hh + hh/2);
|
||||||
|
let scale = hw.min(hh) as f32 * 0.35;
|
||||||
|
if !left.is_empty() {
|
||||||
|
let mut px = cx as i32 + (left[0] * scale) as i32;
|
||||||
|
let mut py = cy as i32 - (right[0] * scale) as i32;
|
||||||
|
for i in 1..left.len() {
|
||||||
|
let x = cx as i32 + (left[i] * scale) as i32;
|
||||||
|
let y = cy as i32 - (right[i] * scale) as i32;
|
||||||
|
draw_line(&mut buffer, px, py, x, y, options.xy_color);
|
||||||
|
px = x; py = y;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Bottom-right
|
||||||
|
render_bars(&mut buffer, smoothed_bars, hw, hh, hw, hh, options.left_color);
|
||||||
|
}
|
||||||
|
RenderMode::Spectrometer => {
|
||||||
|
render_bars(&mut buffer, smoothed_bars, 0, 0, width, height, options.left_color);
|
||||||
|
}
|
||||||
|
_ => { /* Combined/Separate modes logic here if needed */ }
|
||||||
|
}
|
||||||
|
buffer
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn draw_line(buffer: &mut ImageBuffer<image::Rgb<u8>, Vec<u8>>, x0: i32, y0: i32, x1: i32, y1: i32, color: image::Rgb<u8>) {
|
||||||
let dx = (x1 - x0).abs();
|
let dx = (x1 - x0).abs();
|
||||||
let dy = -(y1 - y0).abs();
|
let dy = -(y1 - y0).abs();
|
||||||
let mut x = x0;
|
let mut x = x0; let mut y = y0;
|
||||||
let mut y = y0;
|
|
||||||
let sx = if x0 < x1 { 1 } else { -1 };
|
let sx = if x0 < x1 { 1 } else { -1 };
|
||||||
let sy = if y0 < y1 { 1 } else { -1 };
|
let sy = if y0 < y1 { 1 } else { -1 };
|
||||||
let mut err = dx + dy;
|
let mut err = dx + dy;
|
||||||
|
|
||||||
loop {
|
loop {
|
||||||
if x >= 0 && x < buffer.width() as i32 && y >= 0 && y < buffer.height() as i32 {
|
if x >= 0 && x < buffer.width() as i32 && y >= 0 && y < buffer.height() as i32 {
|
||||||
buffer.put_pixel(x as u32, y as u32, color);
|
buffer.put_pixel(x as u32, y as u32, color);
|
||||||
}
|
}
|
||||||
|
if x == x1 && y == y1 { break; }
|
||||||
if x == x1 && y == y1 {
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
|
|
||||||
let e2 = 2 * err;
|
let e2 = 2 * err;
|
||||||
if e2 >= dy {
|
if e2 >= dy { err += dy; x += sx; }
|
||||||
err += dy;
|
if e2 <= dx { err += dx; y += sy; }
|
||||||
x += sx;
|
|
||||||
}
|
|
||||||
if e2 <= dx {
|
|
||||||
err += dx;
|
|
||||||
y += sy;
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Draw grid lines (graticule).
|
fn draw_graticule(buffer: &mut ImageBuffer<image::Rgb<u8>, Vec<u8>>, color: image::Rgb<u8>) {
|
||||||
fn draw_graticule(
|
let (w, h) = buffer.dimensions();
|
||||||
buffer: &mut ImageBuffer<image::Rgb<u8>, Vec<u8>>,
|
for x in 0..w { buffer.put_pixel(x, h / 2, color); }
|
||||||
primary_color: image::Rgb<u8>,
|
for y in 0..h { buffer.put_pixel(w / 2, y, color); }
|
||||||
show_grid: bool,
|
|
||||||
) {
|
|
||||||
if !show_grid {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
let (width, height) = buffer.dimensions();
|
|
||||||
|
|
||||||
for x in 0..width {
|
|
||||||
buffer.put_pixel(x, height / 2, primary_color);
|
|
||||||
}
|
|
||||||
|
|
||||||
for y in 0..height {
|
|
||||||
buffer.put_pixel(width / 2, y, primary_color);
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Parse RGB hex color string.
|
|
||||||
pub fn parse_rgb_hex(hex: &str) -> Result<image::Rgb<u8>> {
|
pub fn parse_rgb_hex(hex: &str) -> Result<image::Rgb<u8>> {
|
||||||
let hex = hex.trim_start_matches('#');
|
let hex = hex.trim_start_matches('#');
|
||||||
if hex.len() != 6 {
|
if hex.len() != 6 { return Err(anyhow!("Invalid RGB hex")); }
|
||||||
return Err(anyhow::anyhow!("Invalid RGB hex: {}", hex));
|
let r = u8::from_str_radix(&hex[0..2], 16)?;
|
||||||
}
|
let g = u8::from_str_radix(&hex[2..4], 16)?;
|
||||||
let r = u8::from_str_radix(&hex[0..2], 16).map_err(|_| anyhow!("Invalid red component: {}", &hex[0..2]))?;
|
let b = u8::from_str_radix(&hex[4..6], 16)?;
|
||||||
let g = u8::from_str_radix(&hex[2..4], 16).map_err(|_| anyhow!("Invalid green component: {}", &hex[2..4]))?;
|
|
||||||
let b = u8::from_str_radix(&hex[4..6], 16).map_err(|_| anyhow!("Invalid blue component: {}", &hex[4..6]))?;
|
|
||||||
Ok(image::Rgb([r, g, b]))
|
Ok(image::Rgb([r, g, b]))
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Draw a single frame of the visualization.
|
|
||||||
pub fn draw_frame(
|
|
||||||
audio_data: &AudioData,
|
|
||||||
start_sample: usize,
|
|
||||||
samples_per_frame: usize,
|
|
||||||
options: &RenderOptions,
|
|
||||||
) -> ImageBuffer<image::Rgb<u8>, Vec<u8>> {
|
|
||||||
let width = options.width;
|
|
||||||
let height = options.height;
|
|
||||||
let mut buffer = ImageBuffer::new(width, height);
|
|
||||||
|
|
||||||
for pixel in buffer.pixels_mut() {
|
|
||||||
*pixel = options.background;
|
|
||||||
}
|
|
||||||
|
|
||||||
if options.show_grid {
|
|
||||||
draw_graticule(&mut buffer, options.left_color, true);
|
|
||||||
}
|
|
||||||
|
|
||||||
let end_sample = std::cmp::min(start_sample + samples_per_frame, audio_data.left_channel.len());
|
|
||||||
|
|
||||||
match options.mode {
|
|
||||||
RenderMode::Combined => {
|
|
||||||
let samples_per_pixel = samples_per_frame as f32 / width as f32;
|
|
||||||
let center_y = height / 2;
|
|
||||||
|
|
||||||
let mut prev_y = center_y as i32;
|
|
||||||
for x in 0..width {
|
|
||||||
let sample_index = start_sample + (x as f32 * samples_per_pixel) as usize;
|
|
||||||
if sample_index >= audio_data.left_channel.len() {
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
let sample = (audio_data.left_channel[sample_index]
|
|
||||||
+ audio_data.right_channel[sample_index])
|
|
||||||
/ 2.0;
|
|
||||||
let y = center_y as i32 - (sample * (height as f32 * 0.4)) as i32;
|
|
||||||
|
|
||||||
draw_line(&mut buffer, x as i32, prev_y, x as i32, y, options.left_color);
|
|
||||||
prev_y = y;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
RenderMode::Separate => {
|
|
||||||
let half_height = height / 2;
|
|
||||||
let samples_per_pixel = samples_per_frame as f32 / width as f32;
|
|
||||||
|
|
||||||
let left_center_y = half_height / 2;
|
|
||||||
let mut prev_y = left_center_y as i32;
|
|
||||||
for x in 0..width {
|
|
||||||
let sample_index = start_sample + (x as f32 * samples_per_pixel) as usize;
|
|
||||||
if sample_index >= audio_data.left_channel.len() {
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
let sample = audio_data.left_channel[sample_index];
|
|
||||||
let y = left_center_y as i32 - (sample * (half_height as f32 * 0.35)) as i32;
|
|
||||||
|
|
||||||
draw_line(&mut buffer, x as i32, prev_y, x as i32, y, options.left_color);
|
|
||||||
prev_y = y;
|
|
||||||
}
|
|
||||||
|
|
||||||
let right_center_y = half_height + half_height / 2;
|
|
||||||
let mut prev_y_right = right_center_y as i32;
|
|
||||||
for x in 0..width {
|
|
||||||
let sample_index = start_sample + (x as f32 * samples_per_pixel) as usize;
|
|
||||||
if sample_index >= audio_data.right_channel.len() {
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
let sample = audio_data.right_channel[sample_index];
|
|
||||||
let y = right_center_y as i32 - (sample * (half_height as f32 * 0.35)) as i32;
|
|
||||||
|
|
||||||
draw_line(&mut buffer, x as i32, prev_y_right, x as i32, y, options.right_color);
|
|
||||||
prev_y_right = y;
|
|
||||||
}
|
|
||||||
|
|
||||||
for x in 0..width {
|
|
||||||
buffer.put_pixel(x, half_height, image::Rgb([40, 40, 40]));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
RenderMode::All => {
|
|
||||||
let top_height = height / 2;
|
|
||||||
let bottom_height = height / 2;
|
|
||||||
let half_width = width / 2;
|
|
||||||
let samples_per_pixel = samples_per_frame as f32 / half_width as f32;
|
|
||||||
|
|
||||||
let left_center_y = top_height / 2;
|
|
||||||
let mut prev_y = left_center_y as i32;
|
|
||||||
for x in 0..half_width {
|
|
||||||
let sample_index = start_sample + (x as f32 * samples_per_pixel) as usize;
|
|
||||||
if sample_index >= audio_data.left_channel.len() {
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
let sample = audio_data.left_channel[sample_index];
|
|
||||||
let y = left_center_y as i32 - (sample * (top_height as f32 * 0.35)) as i32;
|
|
||||||
|
|
||||||
draw_line(&mut buffer, x as i32, prev_y, x as i32, y, options.left_color);
|
|
||||||
prev_y = y;
|
|
||||||
}
|
|
||||||
|
|
||||||
let right_center_y = top_height / 2;
|
|
||||||
let mut prev_y_right = right_center_y as i32;
|
|
||||||
for x in 0..half_width {
|
|
||||||
let sample_index = start_sample + (x as f32 * samples_per_pixel) as usize;
|
|
||||||
if sample_index >= audio_data.left_channel.len() {
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
let sample = audio_data.right_channel[sample_index];
|
|
||||||
let y = right_center_y as i32 - (sample * (top_height as f32 * 0.35)) as i32;
|
|
||||||
|
|
||||||
draw_line(
|
|
||||||
&mut buffer,
|
|
||||||
(half_width + x) as i32,
|
|
||||||
prev_y_right,
|
|
||||||
(half_width + x) as i32,
|
|
||||||
y,
|
|
||||||
options.right_color,
|
|
||||||
);
|
|
||||||
prev_y_right = y;
|
|
||||||
}
|
|
||||||
|
|
||||||
let xy_center_x = width / 2;
|
|
||||||
let xy_center_y = top_height + bottom_height / 2;
|
|
||||||
let xy_scale = std::cmp::min(half_width, bottom_height) as f32 * 0.35;
|
|
||||||
|
|
||||||
let xy_samples = (end_sample - start_sample).min(samples_per_frame);
|
|
||||||
let mut prev_x = xy_center_x as i32 + (audio_data.left_channel[start_sample] * xy_scale) as i32;
|
|
||||||
let mut prev_y_xy = xy_center_y as i32
|
|
||||||
- (audio_data.right_channel[start_sample] * xy_scale) as i32;
|
|
||||||
|
|
||||||
for i in 1..xy_samples {
|
|
||||||
let sample_idx = start_sample + i;
|
|
||||||
if sample_idx >= audio_data.left_channel.len() {
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
let x = xy_center_x as i32
|
|
||||||
+ (audio_data.left_channel[sample_idx] * xy_scale) as i32;
|
|
||||||
let y = xy_center_y as i32
|
|
||||||
- (audio_data.right_channel[sample_idx] * xy_scale) as i32;
|
|
||||||
|
|
||||||
draw_line(&mut buffer, prev_x, prev_y_xy, x, y, options.xy_color);
|
|
||||||
|
|
||||||
prev_x = x;
|
|
||||||
prev_y_xy = y;
|
|
||||||
}
|
|
||||||
|
|
||||||
for x in 0..width {
|
|
||||||
buffer.put_pixel(x, top_height, image::Rgb([40, 40, 40]));
|
|
||||||
}
|
|
||||||
for y in 0..top_height {
|
|
||||||
buffer.put_pixel(half_width, y, image::Rgb([40, 40, 40]));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
buffer
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Render frames to PNG files.
|
|
||||||
pub fn render_frames(
|
|
||||||
audio_data: &AudioData,
|
|
||||||
options: &RenderOptions,
|
|
||||||
tmp_dir: &Path,
|
|
||||||
progress_callback: &(impl Fn(f64, usize, usize) + Send + Sync),
|
|
||||||
) -> Result<Vec<PathBuf>, anyhow::Error> {
|
|
||||||
let total_samples = audio_data.left_channel.len();
|
|
||||||
let samples_per_frame = (audio_data.sample_rate / options.fps) as usize;
|
|
||||||
let total_frames = ((audio_data.duration * options.fps as f64) as usize).max(1);
|
|
||||||
|
|
||||||
let frame_files: Vec<PathBuf> = (0..total_frames)
|
|
||||||
.map(|i| tmp_dir.join(format!("frame_{:06}.png", i)))
|
|
||||||
.collect();
|
|
||||||
|
|
||||||
use rayon::prelude::*;
|
|
||||||
use std::sync::atomic::{AtomicUsize, Ordering};
|
|
||||||
use std::sync::Arc;
|
|
||||||
|
|
||||||
let progress = Arc::new(AtomicUsize::new(0));
|
|
||||||
|
|
||||||
frame_files
|
|
||||||
.par_iter()
|
|
||||||
.enumerate()
|
|
||||||
.try_for_each(|(frame_idx, frame_file): (usize, &PathBuf)| {
|
|
||||||
let start_sample = std::cmp::min(
|
|
||||||
frame_idx * samples_per_frame,
|
|
||||||
total_samples.saturating_sub(1),
|
|
||||||
);
|
|
||||||
|
|
||||||
let frame = draw_frame(audio_data, start_sample, samples_per_frame, options);
|
|
||||||
|
|
||||||
let file = std::fs::File::create(frame_file)
|
|
||||||
.with_context(|| format!("Failed to create frame file: {}", frame_file.display()))?;
|
|
||||||
let mut writer = std::io::BufWriter::new(file);
|
|
||||||
frame
|
|
||||||
.write_to(&mut writer, image::ImageFormat::Png)
|
|
||||||
.with_context(|| format!("Failed to write frame: {}", frame_file.display()))?;
|
|
||||||
|
|
||||||
let current = progress.fetch_add(1, Ordering::SeqCst) + 1;
|
|
||||||
if current % 30 == 0 || current == total_frames {
|
|
||||||
progress_callback(
|
|
||||||
current as f64 / total_frames as f64 * 100.0,
|
|
||||||
current,
|
|
||||||
total_frames,
|
|
||||||
);
|
|
||||||
}
|
|
||||||
|
|
||||||
Ok::<_, anyhow::Error>(())
|
|
||||||
})?;
|
|
||||||
|
|
||||||
Ok(frame_files)
|
|
||||||
}
|
|
||||||
|
|||||||
+44
-35
@@ -3,7 +3,9 @@
|
|||||||
//! Handles encoding rendered frames into video files using ffmpeg.
|
//! Handles encoding rendered frames into video files using ffmpeg.
|
||||||
|
|
||||||
use anyhow::{anyhow, Context, Result};
|
use anyhow::{anyhow, Context, Result};
|
||||||
use std::path::{Path, PathBuf};
|
use std::io::Write;
|
||||||
|
use std::path::Path;
|
||||||
|
use std::process::{Child, ChildStdin, Command, Stdio};
|
||||||
|
|
||||||
/// Quality preset for video encoding.
|
/// Quality preset for video encoding.
|
||||||
#[derive(Debug, Clone, Copy, clap::ValueEnum)]
|
#[derive(Debug, Clone, Copy, clap::ValueEnum)]
|
||||||
@@ -22,63 +24,70 @@ fn get_quality_settings(quality: VideoQuality) -> (&'static str, &'static str) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Encode video using ffmpeg.
|
pub struct VideoEncoder {
|
||||||
pub fn encode_video(
|
child: Child,
|
||||||
frame_files: &[PathBuf],
|
stdin: ChildStdin,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl VideoEncoder {
|
||||||
|
pub fn new(
|
||||||
audio_file: &Path,
|
audio_file: &Path,
|
||||||
output_file: &Path,
|
output_file: &Path,
|
||||||
|
width: u32,
|
||||||
|
height: u32,
|
||||||
fps: u32,
|
fps: u32,
|
||||||
quality: VideoQuality,
|
quality: VideoQuality,
|
||||||
overwrite: bool,
|
overwrite: bool,
|
||||||
) -> Result<()> {
|
) -> Result<Self> {
|
||||||
let (video_bitrate, _audio_bitrate) = get_quality_settings(quality);
|
let (video_bitrate, _) = get_quality_settings(quality);
|
||||||
|
|
||||||
let tmp_dir = frame_files
|
let mut cmd = Command::new("ffmpeg");
|
||||||
.get(0)
|
|
||||||
.and_then(|p| p.parent())
|
|
||||||
.unwrap_or(Path::new("."));
|
|
||||||
|
|
||||||
let frame_pattern = tmp_dir.join("frame_%06d.png");
|
|
||||||
|
|
||||||
let mut cmd = std::process::Command::new("ffmpeg");
|
|
||||||
|
|
||||||
if overwrite {
|
if overwrite {
|
||||||
cmd.arg("-y");
|
cmd.arg("-y");
|
||||||
}
|
}
|
||||||
|
|
||||||
cmd.args([
|
cmd.args([
|
||||||
"-framerate",
|
"-f", "rawvideo",
|
||||||
&fps.to_string(),
|
"-pixel_format", "rgb24",
|
||||||
"-i",
|
"-video_size", &format!("{}x{}", width, height),
|
||||||
frame_pattern.to_str().ok_or_else(|| anyhow!("Invalid frame pattern"))?,
|
"-framerate", &fps.to_string(),
|
||||||
"-i",
|
"-i", "-",
|
||||||
audio_file.to_str().ok_or_else(|| anyhow!("Invalid audio file path"))?,
|
"-i", audio_file.to_str().ok_or_else(|| anyhow!("Invalid audio path"))?,
|
||||||
"-r",
|
"-c:v", "libx264",
|
||||||
&fps.to_string(),
|
"-b:v", video_bitrate,
|
||||||
"-c:v",
|
"-c:a", "aac",
|
||||||
"libx264",
|
"-pix_fmt", "yuv420p",
|
||||||
"-b:v",
|
|
||||||
video_bitrate,
|
|
||||||
"-c:a",
|
|
||||||
"copy",
|
|
||||||
"-pix_fmt",
|
|
||||||
"yuv420p",
|
|
||||||
"-shortest",
|
"-shortest",
|
||||||
output_file.to_str().ok_or_else(|| anyhow!("Invalid output path"))?,
|
output_file.to_str().ok_or_else(|| anyhow!("Invalid output path"))?,
|
||||||
]);
|
]);
|
||||||
|
|
||||||
let output = cmd
|
let mut child = cmd
|
||||||
.output()
|
.stdin(Stdio::piped())
|
||||||
.with_context(|| "Failed to execute ffmpeg")?;
|
.stderr(Stdio::piped())
|
||||||
|
.spawn()
|
||||||
|
.with_context(|| "Failed to spawn ffmpeg")?;
|
||||||
|
|
||||||
|
let stdin = child.stdin.take().ok_or_else(|| anyhow!("Failed to open ffmpeg stdin"))?;
|
||||||
|
|
||||||
|
Ok(Self { child, stdin })
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn write_frame(&mut self, data: &[u8]) -> Result<()> {
|
||||||
|
self.stdin.write_all(data).with_context(|| "Failed to write frame to ffmpeg")
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn finish(self) -> Result<()> {
|
||||||
|
drop(self.stdin);
|
||||||
|
let output = self.child.wait_with_output().context("Failed to wait for ffmpeg")?;
|
||||||
|
|
||||||
if !output.status.success() {
|
if !output.status.success() {
|
||||||
let stderr = String::from_utf8_lossy(&output.stderr);
|
let stderr = String::from_utf8_lossy(&output.stderr);
|
||||||
return Err(anyhow!("ffmpeg failed: {}", stderr));
|
return Err(anyhow!("ffmpeg failed: {}", stderr));
|
||||||
}
|
}
|
||||||
|
|
||||||
println!("Video saved to: {}", output_file.display());
|
|
||||||
|
|
||||||
Ok(())
|
Ok(())
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Clean up temporary files.
|
/// Clean up temporary files.
|
||||||
|
|||||||
Reference in New Issue
Block a user