private void RemoveTrees(Terrain terrain, IOnPaint editContext, bool clearSelectedOnly) { PaintTreesDetailsContext ctx = PaintTreesDetailsContext.Create(terrain, editContext.uv); for (int i = 0; i < ctx.neighborTerrains.Length; ++i) { Terrain ctxTerrain = ctx.neighborTerrains[i]; if (ctxTerrain != null) { Vector2 ctxUV = ctx.neighborUvs[i]; float radius = 0.5f * brushSize / ctxTerrain.terrainData.size.x; int treePrototype = kInvalidTree; if (clearSelectedOnly && selectedTree != kInvalidTree) { treePrototype = PaintTreesUtils.FindTreePrototype(ctxTerrain, m_TargetTerrain, selectedTree); } if (!clearSelectedOnly || treePrototype != kInvalidTree) { TerrainPaintUtilityEditor.UpdateTerrainDataUndo(ctxTerrain.terrainData, "Terrain - Remove Trees"); ctxTerrain.RemoveTrees(ctxUV, radius, treePrototype); } } } }
public override bool OnPaint(Terrain terrain, IOnPaint editContext) { if (m_TargetTerrain == null || selectedDetail == kInvalidDetail || selectedDetail >= m_TargetTerrain.terrainData.detailPrototypes.Length) { return(false); } Texture2D brush = editContext.brushTexture as Texture2D; if (brush == null) { Debug.LogError("Brush texture is not a Texture2D."); return(false); } if (m_BrushRep == null) { m_BrushRep = new BrushRep(); } PaintTreesDetailsContext ctx = PaintTreesDetailsContext.Create(terrain, editContext.uv); for (int t = 0; t < ctx.terrains.Length; ++t) { Terrain ctxTerrain = ctx.terrains[t]; if (ctxTerrain != null) { int detailPrototype = PaintDetailsUtils.FindDetailPrototype(ctxTerrain, m_TargetTerrain, selectedDetail); if (detailPrototype == kInvalidDetail) { detailPrototype = PaintDetailsUtils.CopyDetailPrototype(ctxTerrain, m_TargetTerrain, selectedDetail); } TerrainData terrainData = ctxTerrain.terrainData; TerrainPaintUtilityEditor.UpdateTerrainDataUndo(terrainData, "Terrain - Detail Edit"); int size = (int)Mathf.Max(1.0f, editContext.brushSize * ((float)terrainData.detailResolution / terrainData.size.x)); m_BrushRep.CreateFromBrush(brush, size); Vector2 ctxUV = ctx.uvs[t]; int xCenter = Mathf.FloorToInt(ctxUV.x * terrainData.detailWidth); int yCenter = Mathf.FloorToInt(ctxUV.y * terrainData.detailHeight); int intRadius = Mathf.RoundToInt(size) / 2; int intFraction = Mathf.RoundToInt(size) % 2; int xmin = xCenter - intRadius; int ymin = yCenter - intRadius; int xmax = xCenter + intRadius + intFraction; int ymax = yCenter + intRadius + intFraction; if (xmin >= terrainData.detailWidth || ymin >= terrainData.detailHeight || xmax <= 0 || ymax <= 0) { continue; } xmin = Mathf.Clamp(xmin, 0, terrainData.detailWidth - 1); ymin = Mathf.Clamp(ymin, 0, terrainData.detailHeight - 1); xmax = Mathf.Clamp(xmax, 0, terrainData.detailWidth); ymax = Mathf.Clamp(ymax, 0, terrainData.detailHeight); int width = xmax - xmin; int height = ymax - ymin; float targetStrength = m_DetailsStrength; if (Event.current.shift || Event.current.control) { targetStrength = -targetStrength; } int[] layers = { detailPrototype }; if (targetStrength < 0.0F && !Event.current.control) { layers = terrainData.GetSupportedLayers(xmin, ymin, width, height); } for (int i = 0; i < layers.Length; i++) { int[,] alphamap = terrainData.GetDetailLayer(xmin, ymin, width, height, layers[i]); for (int y = 0; y < height; y++) { for (int x = 0; x < width; x++) { int xBrushOffset = (xmin + x) - (xCenter - intRadius + intFraction); int yBrushOffset = (ymin + y) - (yCenter - intRadius + intFraction); float opa = detailOpacity * m_BrushRep.GetStrengthInt(xBrushOffset, yBrushOffset); float targetValue = Mathf.Lerp(alphamap[y, x], targetStrength * terrainData.maxDetailScatterPerRes, opa); alphamap[y, x] = Mathf.Min(Mathf.RoundToInt(targetValue - .5f + Random.value), terrainData.maxDetailScatterPerRes); } } terrainData.SetDetailLayer(xmin, ymin, layers[i], alphamap); } } } return(false); }
public override bool OnPaint(Terrain terrain, IOnPaint editContext) { if (m_TargetTerrain == null || selectedDetail == kInvalidDetail || selectedDetail >= m_TargetTerrain.terrainData.detailPrototypes.Length) { return(false); } Texture2D brush = editContext.brushTexture as Texture2D; if (brush == null) { Debug.LogError("Brush texture is not a Texture2D."); return(false); } if (m_BrushRep == null) { m_BrushRep = new DetailBrushRepresentation(); } PaintTreesDetailsContext ctx = PaintTreesDetailsContext.Create(terrain, editContext.uv); for (int t = 0; t < ctx.neighborTerrains.Length; ++t) { Terrain ctxTerrain = ctx.neighborTerrains[t]; if (ctxTerrain != null) { int detailPrototype = PaintDetailsToolUtility.FindDetailPrototype(ctxTerrain, m_TargetTerrain, selectedDetail); if (detailPrototype == kInvalidDetail) { detailPrototype = PaintDetailsToolUtility.CopyDetailPrototype(ctxTerrain, m_TargetTerrain, selectedDetail); } TerrainData terrainData = ctxTerrain.terrainData; TerrainPaintUtilityEditor.UpdateTerrainDataUndo(terrainData, "Terrain - Detail Edit"); int size = (int)Mathf.Max(1.0f, editContext.brushSize * ((float)terrainData.detailResolution / terrainData.size.x)); m_BrushRep.Update(brush, size); float targetStrength = m_DetailsStrength; if (Event.current.shift || Event.current.control) { targetStrength = -targetStrength; } DetailBrushBounds brushBounds = new DetailBrushBounds(terrainData, ctx, size, t); int[] layers = { detailPrototype }; if (targetStrength < 0.0F && !Event.current.control) { layers = terrainData.GetSupportedLayers(brushBounds.min, brushBounds.bounds.size); } for (int i = 0; i < layers.Length; i++) { int[,] alphamap = terrainData.GetDetailLayer(brushBounds.min, brushBounds.bounds.size, layers[i]); for (int y = 0; y < brushBounds.bounds.height; y++) { for (int x = 0; x < brushBounds.bounds.width; x++) { Vector2Int brushOffset = brushBounds.GetBrushOffset(x, y); float opa = detailOpacity * m_BrushRep.GetStrength(brushOffset.x, brushOffset.y); float targetValue = Mathf.Lerp(alphamap[y, x], targetStrength * terrainData.maxDetailScatterPerRes, opa); alphamap[y, x] = Mathf.Min(Mathf.RoundToInt(targetValue - .5f + Random.value), terrainData.maxDetailScatterPerRes); } } terrainData.SetDetailLayer(brushBounds.min, layers[i], alphamap); } } } return(false); }
private void PlaceTrees(Terrain terrain, IOnPaint editContext) { if (m_TargetTerrain == null || selectedTree == kInvalidTree || selectedTree >= m_TargetTerrain.terrainData.treePrototypes.Length) { return; } PaintTreesDetailsContext ctx = PaintTreesDetailsContext.Create(terrain, editContext.uv); int placedTreeCount = 0; int treePrototype = PaintTreesUtils.FindTreePrototype(terrain, m_TargetTerrain, selectedTree); if (treePrototype == kInvalidTree) { treePrototype = PaintTreesUtils.CopyTreePrototype(terrain, m_TargetTerrain, selectedTree); } if (PaintTreesUtils.ValidateTreePrototype(terrain, treePrototype)) { // When painting single tree // And just clicking we always place it, so you can do overlapping trees Vector3 position = new Vector3(editContext.uv.x, 0, editContext.uv.y); bool checkTreeDistance = Event.current.type == EventType.MouseDrag || brushSize > 1; if (!checkTreeDistance || TerrainInspectorUtil.CheckTreeDistance(terrain.terrainData, position, treePrototype, spacing)) { TerrainPaintUtilityEditor.UpdateTerrainDataUndo(terrain.terrainData, "Terrain - Place Trees"); var instanceHeight = GetTreeHeight(); PaintTreesUtils.PlaceTree(terrain, treePrototype, position, GetTreeColor(), instanceHeight, GetTreeWidth(instanceHeight), GetTreeRotation()); ++placedTreeCount; } } for (int i = 0; i < ctx.neighborTerrains.Length; ++i) { Terrain ctxTerrain = ctx.neighborTerrains[i]; if (ctxTerrain != null) { Vector2 ctxUV = ctx.neighborUvs[i]; treePrototype = PaintTreesUtils.FindTreePrototype(ctxTerrain, m_TargetTerrain, selectedTree); if (treePrototype == kInvalidTree) { treePrototype = PaintTreesUtils.CopyTreePrototype(ctxTerrain, m_TargetTerrain, selectedTree); } if (PaintTreesUtils.ValidateTreePrototype(ctxTerrain, treePrototype)) { Vector3 size = TerrainInspectorUtil.GetPrototypeExtent(ctxTerrain.terrainData, treePrototype); size.y = 0; float treeCountOneAxis = brushSize / (size.magnitude * spacing * .5f); int treeCount = (int)((treeCountOneAxis * treeCountOneAxis) * .5f); treeCount = Mathf.Clamp(treeCount, 0, 100); // Plant a bunch of trees for (int j = ctxTerrain == terrain ? 1 : 0; j < treeCount && placedTreeCount < treeCount; ++j) { Vector2 randomOffset = 0.5f * Random.insideUnitCircle; randomOffset.x *= brushSize / ctxTerrain.terrainData.size.x; randomOffset.y *= brushSize / ctxTerrain.terrainData.size.z; Vector3 position = new Vector3(ctxUV.x + randomOffset.x, 0, ctxUV.y + randomOffset.y); if (position.x >= 0 && position.x <= 1 && position.z >= 0 && position.z <= 1 && TerrainInspectorUtil.CheckTreeDistance(ctxTerrain.terrainData, position, treePrototype, spacing * .5f)) { TerrainPaintUtilityEditor.UpdateTerrainDataUndo(ctxTerrain.terrainData, "Terrain - Place Trees"); var instanceHeight = GetTreeHeight(); PaintTreesUtils.PlaceTree(ctxTerrain, treePrototype, position, GetTreeColor(), instanceHeight, GetTreeWidth(instanceHeight), GetTreeRotation()); ++placedTreeCount; } } } } } }