veloren_world/site/plot/
giant_tree.rs

1use crate::{
2    Land, Sampler, SpriteKind,
3    layer::tree::{ProceduralTree, TreeConfig},
4    site::{Fill, Painter, Site, Structure, namegen::NameGen},
5    util::FastNoise,
6};
7use common::{
8    generation::EntityInfo,
9    terrain::{Block, BlockKind},
10};
11use rand::RngExt;
12use vek::*;
13
14pub struct GiantTree {
15    name: String,
16    wpos: Vec3<i32>,
17    tree: ProceduralTree,
18    seed: u32,
19}
20
21impl GiantTree {
22    pub fn generate(
23        site: &Site,
24        center_tile: Vec2<i32>,
25        land: &Land,
26        rng: &mut impl RngExt,
27    ) -> Self {
28        let wpos = site.tile_center_wpos(center_tile);
29        Self {
30            name: format!("Tree of {}", NameGen::location(rng).generate()),
31            // Get the tree's altitude
32            wpos: wpos.with_z(land.get_alt_approx(wpos) as i32),
33            tree: {
34                let config = TreeConfig::giant(rng, 4.0, true);
35                ProceduralTree::generate(config, rng)
36            },
37            seed: rng.random(),
38        }
39    }
40
41    pub fn name(&self) -> &str { &self.name }
42
43    pub fn radius(&self) -> f32 { 100.0 }
44
45    pub fn tree(&self) -> &ProceduralTree { &self.tree }
46
47    pub fn entity_at(
48        &self,
49        pos: Vec3<i32>,
50        above_block: &Block,
51        dynamic_rng: &mut impl RngExt,
52    ) -> Option<EntityInfo> {
53        if above_block.kind() == BlockKind::Leaves && dynamic_rng.random_bool(0.000055) {
54            let entity = EntityInfo::at(pos.as_());
55            match dynamic_rng.random_range(0..=7) {
56                0 => Some(entity.with_asset_expect(
57                    "common.entity.wild.aggressive.horn_beetle",
58                    dynamic_rng,
59                    None,
60                )),
61                1 => Some(entity.with_asset_expect(
62                    "common.entity.wild.aggressive.stag_beetle",
63                    dynamic_rng,
64                    None,
65                )),
66                2..=3 => Some(entity.with_asset_expect(
67                    "common.entity.wild.aggressive.deadwood",
68                    dynamic_rng,
69                    None,
70                )),
71                4 => Some(entity.with_asset_expect(
72                    "common.entity.wild.aggressive.maneater",
73                    dynamic_rng,
74                    None,
75                )),
76                5..=7 => Some(entity.with_asset_expect(
77                    "common.entity.wild.peaceful.parrot",
78                    dynamic_rng,
79                    None,
80                )),
81                _ => None,
82            }
83        } else {
84            None
85        }
86    }
87
88    pub fn leaf_color(&self) -> Rgb<u8> {
89        let fast_noise = FastNoise::new(self.seed);
90        let dark = Rgb::new(10, 70, 50).map(|e| e as f32);
91        let light = Rgb::new(80, 140, 10).map(|e| e as f32);
92        Lerp::lerp(
93            dark,
94            light,
95            fast_noise.get((self.wpos.map(|e| e as f64) * 0.05) * 0.5 + 0.5),
96        )
97        .map(|e| e as u8)
98    }
99}
100
101impl Structure for GiantTree {
102    #[cfg(feature = "use-dyn-lib")]
103    const UPDATE_FN: &'static [u8] = b"render_gianttree\0";
104
105    #[cfg_attr(feature = "be-dyn-lib", unsafe(export_name = "render_gianttree"))]
106    fn render_inner(&self, _site: &Site, _land: &Land, painter: &Painter) {
107        let leaf_col = self.leaf_color();
108        let mut rng = rand::rng();
109        self.tree.walk(|branch, parent| {
110            let aabr = Aabr {
111                min: self.wpos.xy() + branch.get_aabb().min.xy().as_(),
112                max: self.wpos.xy() + branch.get_aabb().max.xy().as_(),
113            };
114            if aabr.collides_with_aabr(painter.render_aabr().as_()) {
115                painter
116                    .line_two_radius(
117                        self.wpos + branch.get_line().start.as_(),
118                        self.wpos + branch.get_line().end.as_(),
119                        parent.get_wood_radius(),
120                        branch.get_wood_radius(),
121                    )
122                    .fill(Fill::Block(Block::new(
123                        BlockKind::Wood,
124                        Rgb::new(80, 32, 0),
125                    )));
126                if branch.get_leaf_radius() > branch.get_wood_radius() {
127                    painter
128                        .line_two_radius(
129                            self.wpos + branch.get_line().start.as_(),
130                            self.wpos + branch.get_line().end.as_(),
131                            parent.get_leaf_radius(),
132                            branch.get_leaf_radius(),
133                        )
134                        .fill(Fill::Block(Block::new(BlockKind::Leaves, leaf_col)));
135                    // Calculate direction of the branch
136                    let branch_start = branch.get_line().start;
137                    let branch_end = branch.get_line().end;
138                    let branch_direction = (branch_end - branch_start).normalized();
139
140                    // Generate a sprite at the surface of the leafy part of the branch
141                    let displacement =
142                        (branch_direction * branch.get_leaf_radius()).map(|e| e.round() as i32);
143                    let pos = self.wpos + branch_end.as_() + displacement;
144                    if rng.random_bool(0.07) {
145                        let ori = rng.random_range(0..=3) * 2;
146                        painter.resource_sprite(pos, SpriteKind::Ironwood, ori);
147                    }
148                }
149                true
150            } else {
151                false
152            }
153        });
154    }
155}
veloren_world/site/plot/giant_tree.rs

veloren_world/site/plot/
giant_tree.rs
