1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
|
// Copyright (C) 2022-2022 Fuwn <[email protected]>
// SPDX-License-Identifier: MIT
#![feature(decl_macro, generators, generator_trait)]
#![deny(
warnings,
nonstandard_style,
unused,
future_incompatible,
rust_2018_idioms,
unsafe_code
)]
#![deny(clippy::all, clippy::nursery, clippy::pedantic)]
#![recursion_limit = "128"]
mod ppm;
use std::process::exit;
use image::DynamicImage;
use crate::ppm::PPMParser;
#[allow(unused)]
fn get_image(parser: &mut PPMParser, index: usize) -> DynamicImage {
let frame = parser.get_frame_pixels(index);
let colours = parser.get_frame_palette(index);
let mut img = Vec::new();
let mut img_encoder = image::codecs::bmp::BmpEncoder::new(&mut img);
img_encoder.encode_with_palette(
&frame.into_iter().flatten().collect::<Vec<u8>>(),
256,
192,
image::ColorType::L8,
Some(&[
[colours[0].0, colours[0].1, colours[0].2],
[colours[1].0, colours[1].1, colours[1].2],
[colours[2].0, colours[2].1, colours[2].2],
]),
);
image::load_from_memory(&img).unwrap()
}
fn main() {
human_panic::setup_panic!(Metadata {
version: env!("CARGO_PKG_VERSION").into(),
name: env!("CARGO_PKG_NAME").into(),
authors: env!("CARGO_PKG_AUTHORS").into(),
homepage: env!("CARGO_PKG_HOMEPAGE").into(),
});
let args = std::env::args().collect::<Vec<_>>();
if args.len() < 4 {
println!(
"{}, version {}(1)-{}-({})-{}\n\
usage: {} <in> <index option> <out>\n\
index options:\n\
\tgif\n\
\tthumb\n\
\tdump\n\
\tinteger(u16)\n\n\
{0} home page: <https://github.com/Usugata/{0}>",
env!("CARGO_PKG_NAME"),
env!("CARGO_PKG_VERSION"),
env!("PROFILE"),
env!("TARGET"),
env!("GIT_COMMIT_HASH"),
args[0],
);
exit(1);
}
let path = &args[1];
let index = &args[2];
let out_path = &args[3];
let mut parser = PPMParser::new_from_file(path);
parser.load();
let frame_count = usize::from(parser.get_frame_count());
match index.as_str() {
"gif" => {
let frame_duration = (1.0 / parser.get_framerate()) * 100.0;
let frames = (0..parser.get_frame_count())
.map(|i| get_image(&mut parser, i as usize))
.collect::<Vec<DynamicImage>>();
let mut file_out = std::fs::File::create(out_path).unwrap();
let mut gif_encoder = image::codecs::gif::GifEncoder::new_with_speed(
&mut file_out,
#[allow(clippy::cast_possible_truncation)]
{
frame_duration as i32
},
);
for frame in frames {
gif_encoder
.encode_frame(image::Frame::new(frame.into_rgba8()))
.unwrap();
}
gif_encoder
.set_repeat(image::codecs::gif::Repeat::Infinite)
.unwrap();
}
"thumb" => {
let thumb_index = parser.get_thumb_index() as usize;
get_image(&mut parser, thumb_index).save(out_path).unwrap();
}
"dump" => parser.dump_to_json(out_path),
_ => {
if !(0..frame_count).contains(&index.parse::<usize>().unwrap()) {
println!(
"invalid frame index({}), image has {}(0..{}) frames",
index,
frame_count,
frame_count - 1,
);
exit(1);
}
get_image(&mut parser, index.parse::<usize>().unwrap())
.save(out_path)
.unwrap();
}
}
println!("converted {}({}) to {}", path, index, out_path);
}
|