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

Add wagon chest options to multistation

Browse source
This commit is contained in:
hal8174 2025-04-25 21:31:48 +02:00
parent 7ef42e5202
commit 0ced2c3c44
7 changed files with 110 additions and 71 deletions
Download patch file Download diff file Expand all files Collapse all files

17
factorio-blueprint-generator/src/bin/multistation.rs
View file

@ -1,8 +1,9 @@
use clap::Parser;
use factorio_blueprint::abstraction::serde::AbstractBlueprintString;
use factorio_blueprint::abstraction::{InserterType, Quality};
use factorio_blueprint::abstraction::{ChestType, InserterType, Quality};
use factorio_blueprint::encode;
use factorio_blueprint_generator::multistation::{StationSpec, multistation};
use factorio_blueprint_generator::station::StationBasicSpec;
use factorio_core::beltoptions::{Beltspeed, Belttype};
use factorio_core::visualize::Visualize;
@ -17,6 +18,12 @@ struct Args {
#[arg(short, long, default_value = "normal")]
wagon_inserter_quality: Quality,
#[arg(short, long, default_value = "steel")]
chest_type: ChestType,
#[arg(short, long, default_value = "normal")]
chest_quality: Quality,
stacker_size: usize,
/// format: <locomotives>-<wagons>[lu][nfet]<lanes>[lr]?
stations: Vec<String>,
@ -72,8 +79,12 @@ fn main() {
let mut b = multistation(
&stations,
args.stacker_size,
args.wagon_inserter_type,
args.wagon_inserter_quality,
&StationBasicSpec {
inserter_type: args.wagon_inserter_type,
inserter_quality: args.wagon_inserter_quality,
chest_type: args.chest_type,
chest_quality: args.chest_quality,
},
)
.0;

8
factorio-blueprint-generator/src/bin/station.rs
View file

@ -1,6 +1,6 @@
use clap::{Parser, Subcommand};
use factorio_blueprint::{BlueprintBook, BlueprintBookEntry, BlueprintString, encode};
use factorio_blueprint_generator::station::basic_station;
use factorio_blueprint_generator::station::{StationBasicSpec, basic_station};
use factorio_core::beltoptions::{Beltspeed, Belttype};
#[derive(Parser)]
@ -70,8 +70,7 @@ fn main() {
beltspeed,
Belttype::Full,
false,
factorio_blueprint::abstraction::InserterType::Bulk,
factorio_blueprint::abstraction::Quality::Normal,
&factorio_blueprint_generator::station::StationBasicSpec::default(),
);
inner_inner_b.push(BlueprintBookEntry::new(
@ -140,8 +139,7 @@ fn main() {
beltspeed,
belttype,
false,
factorio_blueprint::abstraction::InserterType::Bulk,
factorio_blueprint::abstraction::Quality::Normal,
&StationBasicSpec::default(),
)
.to_blueprint(),
);

3
factorio-blueprint-generator/src/factory.rs
View file

@ -783,8 +783,7 @@ pub fn generate_factory<L: Layouter, P: Pathfinder + Sync, R: Rng + SeedableRng
let (multistation_blueprint, multistation_x_offset, mut multistation_y_offsets) = multistation(
&station_spec,
8,
factorio_blueprint::abstraction::InserterType::Bulk,
factorio_blueprint::abstraction::Quality::Normal,
&crate::station::StationBasicSpec::default(),
);
let multistation_y_offset = multistation_y_offsets[0];

11
factorio-blueprint-generator/src/multistation.rs
View file

@ -1,4 +1,7 @@
use crate::{balancer::binary_balancer, station::basic_station};
use crate::{
balancer::binary_balancer,
station::{StationBasicSpec, basic_station},
};
use factorio_blueprint::abstraction::{
Blueprint, ElectricPoleType, Entity, Quality, RailType, UndergroundType,
};
@ -50,8 +53,7 @@ fn calculate_station_height(
pub fn multistation(
stations: &[StationSpec],
stacker_size: usize,
wagon_inserter_type: factorio_blueprint::abstraction::InserterType,
wagon_inserter_quality: Quality,
basic_spec: &StationBasicSpec,
) -> (Blueprint, PositionType, Vec<PositionType>) {
let longest_train = stations
.iter()
@ -272,8 +274,7 @@ pub fn multistation(
station.beltspeed,
station.belttype,
station.stacked,
wagon_inserter_type,
wagon_inserter_quality,
basic_spec,
);
let output_height = -b.bounding_box().min().y;

108
factorio-blueprint-generator/src/station.rs
View file

@ -1,4 +1,6 @@
use factorio_blueprint::abstraction::{Blueprint, ElectricPoleType, Entity, InserterType, Quality};
use factorio_blueprint::abstraction::{
Blueprint, ChestType, ElectricPoleType, Entity, InserterType, Quality,
};
use factorio_core::{
beltoptions::{Beltspeed, Belttype},
prelude::*,
@ -6,12 +8,29 @@ use factorio_core::{
use crate::binary_merger::merger;
pub struct StationBasicSpec {
pub inserter_type: InserterType,
pub inserter_quality: Quality,
pub chest_type: ChestType,
pub chest_quality: Quality,
}
impl Default for StationBasicSpec {
fn default() -> Self {
Self {
inserter_type: InserterType::Bulk,
inserter_quality: Quality::Normal,
chest_type: ChestType::Steel,
chest_quality: Quality::Normal,
}
}
}
pub fn unloader(
beltspeed: Beltspeed,
belttype: Belttype,
stacked: bool,
wagon_inserter_type: InserterType,
wagon_inserter_quality: Quality,
basic_spec: &StationBasicSpec,
) -> (Blueprint, PositionType) {
let mut b = Blueprint::new();
@ -33,20 +52,21 @@ pub fn unloader(
));
if belttype.contains_left() {
b.add_entity(Entity::new_unknown(
"steel-chest",
Position::new(1, 1) + 2 * Position::new(5, -2),
Direction::Up,
Position::new(2, 2),
));
b.add_entity(
Entity::new_chest(
basic_spec.chest_type,
Position::new(1, 1) + 2 * Position::new(5, -2),
)
.quality(basic_spec.chest_quality),
);
b.add_entity(
Entity::new_inserter(
wagon_inserter_type,
basic_spec.inserter_type,
None,
Position::new(1, 1) + 2 * Position::new(5, -1),
Direction::Down,
)
.quality(wagon_inserter_quality),
.quality(basic_spec.inserter_quality),
);
b.add_entity(
@ -61,20 +81,21 @@ pub fn unloader(
}
if belttype.contains_right() {
b.add_entity(Entity::new_unknown(
"steel-chest",
Position::new(1, 1) + 2 * Position::new(1, -2),
Direction::Up,
Position::new(2, 2),
));
b.add_entity(
Entity::new_chest(
basic_spec.chest_type,
Position::new(1, 1) + 2 * Position::new(1, -2),
)
.quality(basic_spec.chest_quality),
);
b.add_entity(
Entity::new_inserter(
wagon_inserter_type,
basic_spec.inserter_type,
None,
Position::new(1, 1) + 2 * Position::new(1, -1),
Direction::Down,
)
.quality(wagon_inserter_quality),
.quality(basic_spec.inserter_quality),
);
b.add_entity(
@ -111,20 +132,21 @@ pub fn unloader(
)
.quality(quality),
);
b.add_entity(Entity::new_unknown(
"steel-chest",
Position::new(1, 1) + 2 * Position::new(x, -2),
Direction::Up,
Position::new(2, 2),
));
b.add_entity(
Entity::new_chest(
basic_spec.chest_type,
Position::new(1, 1) + 2 * Position::new(x, -2),
)
.quality(basic_spec.chest_quality),
);
b.add_entity(
Entity::new_inserter(
wagon_inserter_type,
basic_spec.inserter_type,
None,
Position::new(1, 1) + 2 * Position::new(x, -1),
Direction::Down,
)
.quality(wagon_inserter_quality),
.quality(basic_spec.inserter_quality),
);
};
@ -145,8 +167,7 @@ pub fn unloader(
pub fn one_loader(
beltspeed: Beltspeed,
stacked: bool,
wagon_inserter_type: InserterType,
wagon_inserter_quality: Quality,
basic_spec: &StationBasicSpec,
) -> (Blueprint, PositionType) {
let (belt_inserter, quality) = match (beltspeed, stacked) {
(Beltspeed::Normal, false) => (InserterType::Fast, Quality::Normal),
@ -184,20 +205,18 @@ pub fn one_loader(
Entity::new_inserter(belt_inserter, None, Position::new(x, -5), Direction::Up)
.quality(quality),
);
b.add_entity(Entity::new_unknown(
"steel-chest",
Position::new(x, -3),
Direction::Up,
Position::new(2, 2),
));
b.add_entity(
Entity::new_chest(basic_spec.chest_type, Position::new(x, -3))
.quality(basic_spec.chest_quality),
);
b.add_entity(
Entity::new_inserter(
wagon_inserter_type,
basic_spec.inserter_type,
None,
Position::new(x, -1),
Direction::Up,
)
.quality(wagon_inserter_quality),
.quality(basic_spec.inserter_quality),
);
}
@ -212,8 +231,7 @@ pub fn basic_station(
beltspeed: Beltspeed,
belttype: Belttype,
stacked: bool,
wagon_inserter_type: InserterType,
wagon_inserter_quality: Quality,
basic_spec: &StationBasicSpec,
) -> Blueprint {
let section_size = length / outputs;
assert!(length % outputs == 0);
@ -244,14 +262,12 @@ pub fn basic_station(
beltspeed.halvings((length / outputs).ilog2() as usize),
belttype,
stacked,
wagon_inserter_type,
wagon_inserter_quality,
basic_spec,
),
true => one_loader(
beltspeed.halvings((length / outputs).ilog2() as usize),
stacked,
wagon_inserter_type,
wagon_inserter_quality,
basic_spec,
),
};
for l in 0..length {
@ -260,14 +276,12 @@ pub fn basic_station(
beltspeed.halvings((length / outputs).ilog2() as usize),
belttype,
stacked,
wagon_inserter_type,
wagon_inserter_quality,
basic_spec,
),
true => one_loader(
beltspeed.halvings((length / outputs).ilog2() as usize),
stacked,
wagon_inserter_type,
wagon_inserter_quality,
basic_spec,
),
};