From a74be78c513ad2fcf0531398ca74049902a33f63 Mon Sep 17 00:00:00 2001
From: Misode <misoloo64@gmail.com>
Date: Mon, 11 Dec 2023 22:59:31 +0100
Subject: [PATCH] Add bamboo feature

---
 src/app/components/previews/Feature.ts        | 23 +++++++++++++++++--
 .../components/previews/FeaturePreview.tsx    | 14 +++++++----
 2 files changed, 30 insertions(+), 7 deletions(-)

diff --git a/src/app/components/previews/Feature.ts b/src/app/components/previews/Feature.ts
index 84e15d8c..48a074c2 100644
--- a/src/app/components/previews/Feature.ts
+++ b/src/app/components/previews/Feature.ts
@@ -1,5 +1,5 @@
-import type { BlockState, Random } from 'deepslate'
-import { BlockPos } from 'deepslate'
+import type { Random } from 'deepslate'
+import { BlockPos, BlockState } from 'deepslate'
 import type { VersionId } from '../../services/index.js'
 import { sampleBlockState, sampleInt } from './WorldgenUtils.jsx'
 
@@ -20,6 +20,25 @@ export function placeFeature(data: any, ctx: FeatureContext) {
 const Features: {
 	[key: string]: (config: any, ctx: FeatureContext) => void,
 } = {
+	bamboo: (config, ctx) => {
+		const n = ctx.nextInt(12) + 5
+		if (ctx.nextFloat() < config?.probability ?? 0) {
+			const s = ctx.nextInt(4) + 1
+			for (let x = -s; x <= s; x += 1) {
+				for (let z = -s; z <= s; z += 1) {
+					if (x * x + z * z <= s * s) {
+						ctx.place([x, -1, z], new BlockState('podzol', { snowy: 'false' }))
+					}
+				}
+			}
+		}
+		for (let i = 0; i < n; i += 1) {
+			ctx.place([0, i, 0], new BlockState('bamboo', { age: '1', leaves: 'none', stage: '0' }))
+		}
+		ctx.place([0, n, 0], new BlockState('bamboo', { age: '1', leaves: 'large', stage: '1'}))
+		ctx.place([0, n-1, 0], new BlockState('bamboo', { age: '1', leaves: 'large', stage: '0'}))
+		ctx.place([0, n-2, 0], new BlockState('bamboo', { age: '1', leaves: 'small', stage: '0'}))
+	},
 	tree: (config, ctx) => {
 		const trunk = config.trunk_placer
 		const trunkPlacerType = trunk.type.replace(/^minecraft:/, '')
diff --git a/src/app/components/previews/FeaturePreview.tsx b/src/app/components/previews/FeaturePreview.tsx
index bec83a9f..db6b2b46 100644
--- a/src/app/components/previews/FeaturePreview.tsx
+++ b/src/app/components/previews/FeaturePreview.tsx
@@ -15,7 +15,7 @@ import { InteractiveCanvas3D } from './InteractiveCanvas3D.jsx'
 import { nextGaussian } from './WorldgenUtils.jsx'
 import type { PreviewProps } from './index.js'
 
-const MAX_SIZE = 25
+const MAX_SIZE = 45
 
 export const FeaturePreview = ({ data, version, shown }: PreviewProps) => {
 	const { locale } = useLocale()
@@ -35,8 +35,8 @@ export const FeaturePreview = ({ data, version, shown }: PreviewProps) => {
 			version: version,
 			random,
 			place: (pos, block) => {
-				const structurePos = BlockPos.offset(pos, placeOffset, 0, placeOffset)
-				if (structurePos.some(v => v < 0 || v >= MAX_SIZE)) return
+				const structurePos = BlockPos.offset(pos, placeOffset, placeOffset, placeOffset)
+				if (structurePos.some((v, i) => v < 0 || v >= structure.getSize()[i])) return
 				const name = typeof block === 'string' ? block : block.getName()
 				const properties = typeof block === 'string' ? undefined : block.getProperties()
 				structure.addBlock(structurePos, name, properties)
@@ -52,10 +52,14 @@ export const FeaturePreview = ({ data, version, shown }: PreviewProps) => {
 	const renderer = useRef<StructureRenderer | undefined>(undefined)
 
 	const onSetup = useCallback((canvas: HTMLCanvasElement) => {
+		if (renderer.current) {
+			renderer.current.setStructure(structure)
+			return
+		}
 		if (!resources || !shown) return
 		const gl = canvas.getContext('webgl')
 		if (!gl) return
-		renderer.current = new StructureRenderer(gl, structure, resources)
+		renderer.current = new StructureRenderer(gl, structure, resources, { useInvisibleBlockBuffer: false })
 	}, [resources, shown, structure])
 	const onResize = useCallback((width: number, height: number) => {
 		renderer.current?.setViewport(0, 0, width, height)
@@ -69,7 +73,7 @@ export const FeaturePreview = ({ data, version, shown }: PreviewProps) => {
 			<Btn icon="sync" tooltip={locale('generate_new_seed')} onClick={() => setSeed(randomSeed())} />
 		</div>
 		<div class="full-preview">
-			<InteractiveCanvas3D onSetup={onSetup} onDraw={onDraw} onResize={onResize} startDistance={10} startPosition={[MAX_SIZE/2, 0, MAX_SIZE/2]} startYRotation={2.6} />
+			<InteractiveCanvas3D onSetup={onSetup} onDraw={onDraw} onResize={onResize} startDistance={10} startPosition={[MAX_SIZE/2, MAX_SIZE/2, MAX_SIZE/2]} startYRotation={2.6} />
 		</div>
 	</>
 }