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
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
|
#![feature(iter_array_chunks)]
use anyhow::Result;
use clap::Parser;
use rand::random;
use std::{
io::Write,
net::{SocketAddr, TcpStream},
path::PathBuf,
thread::{self, sleep},
time::Duration,
};
use weareshared::{
Affine3A, Vec3A,
packets::{Data, Object, Packet, ReadWrite},
resources::{Attribute, AttributeArray, IndexArray, Part, Prefab},
store::ResourceStore,
vec3a,
};
#[derive(Parser)]
struct Args {
address: SocketAddr,
scene: PathBuf,
#[arg(short, long)]
push: bool,
}
fn main() -> Result<()> {
env_logger::init_from_env("LOG");
let args = Args::parse();
let mut sock = TcpStream::connect(args.address)?;
let store = ResourceStore::new_memory();
Packet::Connect(random()).write(&mut sock)?;
let (gltf, buffers, _) = gltf::import(args.scene)?;
let mut prefab = Prefab::default();
for node in gltf.nodes() {
if let Some(mesh) = node.mesh() {
for p in mesh.primitives() {
let reader = p.reader(|buf| Some(&buffers[buf.index()]));
let mut pos_x = vec![];
let mut pos_y = vec![];
let mut pos_z = vec![];
let mut norm_x = vec![];
let mut norm_y = vec![];
let mut norm_z = vec![];
let mut uv_x = vec![];
let mut uv_y = vec![];
for p in reader.read_positions().unwrap() {
pos_x.push(p[0]);
pos_y.push(p[1]);
pos_z.push(p[2]);
}
for p in reader.read_normals().unwrap() {
norm_x.push(p[0]);
norm_y.push(p[1]);
norm_z.push(p[2]);
}
for p in reader.read_tex_coords(0).unwrap().into_f32() {
uv_x.push(p[0]);
uv_y.push(p[1]);
}
let index = reader
.read_indices()
.unwrap()
.into_u32()
.map(|e| e as u16)
.array_chunks::<3>()
.collect::<Vec<_>>();
if let Some(tex) = p.material().pbr_metallic_roughness().base_color_texture() {}
let part = store.set(
&Part {
va_position: Some([
Attribute::Vertex(store.set(&AttributeArray(pos_x).write_alloc())?),
Attribute::Vertex(store.set(&AttributeArray(pos_y).write_alloc())?),
Attribute::Vertex(store.set(&AttributeArray(pos_z).write_alloc())?),
]),
va_normal: Some([
Attribute::Vertex(store.set(&AttributeArray(norm_x).write_alloc())?),
Attribute::Vertex(store.set(&AttributeArray(norm_y).write_alloc())?),
Attribute::Vertex(store.set(&AttributeArray(norm_z).write_alloc())?),
]),
va_texcoord: Some([
Attribute::Vertex(store.set(&AttributeArray(uv_x).write_alloc())?),
Attribute::Vertex(store.set(&AttributeArray(uv_y).write_alloc())?),
]),
va_pbr_albedo: Some([
Attribute::Constant(0.9),
Attribute::Constant(0.1),
Attribute::Constant(0.1),
]),
index: Some(store.set(&IndexArray(index).write_alloc())?),
..Part::default()
}
.write_alloc(),
)?;
let mat = node.transform().matrix();
let aff = Affine3A::from_cols_array_2d(&[
[mat[0][0], mat[0][1], mat[0][2]],
[mat[1][0], mat[1][1], mat[1][2]],
[mat[2][0], mat[2][1], mat[2][2]],
[mat[3][0], mat[3][1], mat[3][2]],
]);
prefab.0.push((aff, part))
}
}
}
let ob = Object::new();
Packet::Add(ob, store.set(&prefab.write_alloc())?).write(&mut sock)?;
let mut sock2 = sock.try_clone().unwrap();
thread::spawn(move || {
let mut x = 0.;
loop {
Packet::Position(ob, Vec3A::ZERO, vec3a(x, x * 0.3, x * 0.1))
.write(&mut sock2)
.unwrap();
sock2.flush().unwrap();
x += 0.1;
sleep(Duration::from_millis(50));
}
});
if args.push {
store.iter(|d| {
Packet::RespondResource(Data(d.to_vec()))
.write(&mut sock)
.unwrap();
})?;
sock.flush()?;
} else {
loop {
let packet = Packet::read(&mut sock)?;
match packet {
Packet::RequestResource(hash) => {
if let Some(d) = store.get(hash)? {
Packet::RespondResource(Data(d)).write(&mut sock)?;
sock.flush()?;
}
}
_ => (),
}
}
}
Ok(())
}
|
