hal8174/factorio_blueprint
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Improve station generator

Browse source
This commit is contained in:
hal8174 2025-01-16 21:48:48 +01:00
parent c4a17d1792
commit 2c7373f7d9
3 changed files with 126 additions and 99 deletions
Download patch file Download diff file Expand all files Collapse all files

184
src/blueprint/station.rs
View file

@ -7,16 +7,20 @@ pub fn station_unload(
locomotives: usize,
unloader: &Vec<BlueprintEntity>,
beltspeed: Beltspeed,
double: bool,
outputs: usize,
output_x: f64,
output_y: f64,
) -> Blueprint {
assert!(length.is_power_of_two());
let section_size = length / outputs;
assert!(length % outputs == 0);
assert!(section_size.is_power_of_two());
assert!(outputs <= length);
let mut e = Vec::new();
let global_x_offset = locomotives * 7;
// electric poles
for l in 1..=(length + locomotives) {
e.push(
BlueprintEntity::builder(
@ -28,6 +32,7 @@ pub fn station_unload(
);
}
// unloader
let offset = e.len();
for l in 0..length {
e.extend(unloader.iter().cloned().map(|mut e| {
@ -37,6 +42,7 @@ pub fn station_unload(
}));
}
// train stop
let offset = e.len();
e.push(
BlueprintEntity::builder(
@ -48,6 +54,7 @@ pub fn station_unload(
.build(),
);
// rails
for l in 0..((length * 7 + global_x_offset + 1) / 2) {
let offset = e.len() as u32;
e.push(
@ -61,110 +68,96 @@ pub fn station_unload(
);
}
for i in 0..length {
let depth = (i / 2).count_ones();
// output and merging
for o in 0..outputs {
// stubs
let stubspeed = beltspeed.halvings(section_size.ilog2() as usize);
for i in 0..section_size {
let depth = o + i.count_ones() as usize;
for j in 0..depth {
let offset = e.len() as u32;
e.push(
BlueprintEntity::builder(
stubspeed.string(),
offset,
BlueprintPosition::new(
(7 * (i + o * section_size) + global_x_offset) as f64 + output_x,
output_y - j as f64,
),
)
.build(),
);
}
for j in 0..depth {
let offset = e.len() as u32;
e.push(
BlueprintEntity::builder(
beltspeed.string(),
stubspeed.string(),
offset,
BlueprintPosition::new(
(7 * i + global_x_offset) as f64 + output_x,
output_y - j as f64,
),
)
.build(),
);
}
if i % 2 == 0 {
let offset = e.len() as u32;
e.push(
BlueprintEntity::builder(
beltspeed.string(),
offset,
BlueprintPosition::new(
(7 * i + global_x_offset) as f64 + output_x,
(7 * (i + o * section_size) + global_x_offset) as f64 + output_x,
output_y - depth as f64,
),
)
.direction(12)
.build(),
);
let offset = e.len() as u32;
e.push(
BlueprintEntity::builder(
beltspeed.string_splitter(),
offset,
BlueprintPosition::new(
(7 * i + global_x_offset) as f64 + output_x - 1.0,
output_y - depth as f64 - 0.5,
),
)
.direction(12)
.build(),
);
} else {
let offset = e.len() as u32;
e.extend((0..=7).map(|j| {
BlueprintEntity::builder(
beltspeed.string(),
offset + j as u32,
BlueprintPosition::new(
(7 * i - j + global_x_offset) as f64 + output_x,
output_y - depth as f64 - 1.0,
),
)
.direction(12)
.build()
}));
let offset = e.len() as u32;
e.push(
BlueprintEntity::builder(
beltspeed.string(),
offset,
BlueprintPosition::new(
(7 * i + global_x_offset) as f64 + output_x,
output_y - depth as f64,
),
)
.build(),
);
}
}
for i in 2..(length.ilog2() as usize + 1) {
let p = 1 << i;
for j in 0..(length / p) {
let depth = j.count_ones();
let offset = e.len() as u32;
e.push(
BlueprintEntity::builder(
beltspeed.string_splitter(),
offset,
BlueprintPosition::new(
(7 * j * p + global_x_offset) as f64 - i as f64 + output_x,
output_y - i as f64 + 0.5 - depth as f64,
),
)
.direction(12)
.build(),
);
e.extend((0..(7 * p / 2)).map(|l| {
BlueprintEntity::builder(
beltspeed.string(),
offset + l as u32,
BlueprintPosition::new(
(7 * j * p + global_x_offset) as f64 - i as f64 + output_x + l as f64 + 1.0,
output_y - i as f64 - depth as f64,
),
)
.direction(12)
.build()
}))
// merger
for i in 1..(section_size.ilog2() as usize + 1) {
let p = 1 << i;
let mergespeed = beltspeed.halvings((section_size.ilog2() as usize) - i);
for j in 0..(section_size / p) {
let depth = o + j.count_ones() as usize;
let offset = e.len() as u32;
e.push(
BlueprintEntity::builder(
mergespeed.string_splitter(),
offset,
BlueprintPosition::new(
(7 * (j * p + o * section_size) + global_x_offset) as f64 - i as f64
+ output_x,
output_y - i as f64 + 0.5 - depth as f64,
),
)
.direction(12)
.build(),
);
e.extend((0..(7 * p / 2)).map(|l| {
BlueprintEntity::builder(
mergespeed.halve().string(),
offset + 1 + l as u32,
BlueprintPosition::new(
(7 * (j * p + o * section_size) + global_x_offset) as f64 - i as f64
+ output_x
+ l as f64
+ 1.0,
output_y - i as f64 - depth as f64,
),
)
.direction(12)
.build()
}));
}
}
// connect
let offset = e.len() as u32;
let step = o + section_size.ilog2() as usize;
e.extend((0..(7 * o * section_size)).map(|l| {
BlueprintEntity::builder(
beltspeed.string(),
offset + l as u32,
BlueprintPosition::new(
(l + global_x_offset) as f64 + output_x - section_size.ilog2() as f64,
output_y - step as f64,
),
)
.direction(12)
.build()
}));
}
Blueprint::builder()
@ -178,7 +171,12 @@ pub fn station_unload(
.build()
}
pub fn basic_unload_station() -> Blueprint {
pub fn basic_unload_station(
locomotives: usize,
length: usize,
outputs: usize,
beltspeed: Beltspeed,
) -> Blueprint {
let top = vec![
BlueprintEntity::builder(
"steel-chest".to_owned(),
@ -286,5 +284,5 @@ pub fn basic_unload_station() -> Blueprint {
.build(),
];
station_unload(64, 3, &full, Beltspeed::Normal, false, 3.5, -2.5)
station_unload(length, locomotives, &full, beltspeed, outputs, 3.5, -2.5)
}