private IEnumerable <TerrainTileAsset> GetTerrainTiles()
{
var asset = this.target as TerrainAsset;
Assert.IsNotNull(asset);
var baseName = asset.name.Substring(0, 4);
return(from tile in UnityFileUtils.GetAllAssetsOfType <TerrainTileAsset>()
where tile.name.StartsWith(baseName)
select tile);
}
/// <summary>
/// Import a .htre file.
/// </summary>
/// <param name="ctx"></param>
public override void OnImportAsset(AssetImportContext ctx)
{
// Get the corresponding TerrainAsset.
var baseName = Path.GetFileNameWithoutExtension(ctx.assetPath).Substring(0, 4);
var terrainAssets = (from tile in UnityFileUtils.GetAllAssetsOfType <TerrainAsset>()
where tile.name.StartsWith(baseName)
select tile).ToArray();
if (terrainAssets.Length == 0)
{
ctx.LogImportError("Corresponding TerrainAsset could not be found for " + ctx.assetPath + ". Did you forget to import its .tre2 file?");
return;
}
var terrainAsset = terrainAssets[0];
var heightTiles = new List <float[, ]>(4);
var materialWeightMapTiles = new List <Color[, ]>(4);
var materialIdMapTiles = new List <Color[, ]>(4);
var materialSelectMapTiles = new List <Color[, ]>(4);
const int HalfWidth = HEIGHTMAP_WIDTH / 2;
var asset = ScriptableObject.CreateInstance <TerrainTileAsset>();
asset.name = Path.GetFileNameWithoutExtension(ctx.assetPath);
ctx.AddObjectToAsset("Main", asset);
ctx.SetMainObject(asset);
using (var reader = new BinaryReader(new FileStream(ctx.assetPath, FileMode.Open)))
{
var version = reader.ReadUInt32();
if (version == 3)
{
reader.BaseStream.Seek(HEIGHTMAP_OFFSET_VERSION3, SeekOrigin.Begin);
}
else if (version == 4)
{
reader.BaseStream.Seek(HEIGHTMAP_OFFSET_VERSION4, SeekOrigin.Begin);
}
else
{
Debug.LogError("Unrecognized htre version number: " + version);
}
// Read heightmap
for (var tile = 0; tile < 4; tile++)
{
var heightValues = new float[HEIGHTMAP_WIDTH / 2, HEIGHTMAP_HEIGHT / 2];
heightTiles.Add(heightValues);
for (var i = 0; i < HalfWidth; i++)
{
for (var j = 0; j < HalfWidth; j++)
{
var height = reader.ReadSingle();
heightValues[j, i] = (height - terrainAsset.HeightRangeMin) / (terrainAsset.HeightRangeMax - terrainAsset.HeightRangeMin);
}
}
}
// Read material weight map
for (var tile = 0; tile < 4; tile++)
{
var materialWeightValues = new Color[HEIGHTMAP_WIDTH / 2, HEIGHTMAP_HEIGHT / 2];
materialWeightMapTiles.Add(materialWeightValues);
for (var i = 0; i < HalfWidth; i++)
{
for (var j = 0; j < HalfWidth; j++)
{
var r = reader.ReadByte();
var g = reader.ReadByte();
var b = reader.ReadByte();
var a = reader.ReadByte();
materialWeightValues[j, i] = new Color(r, g, b, a);
}
}
}
// Skip unknown section
reader.BaseStream.Position += 64L;
// Read material ID map
// This can't be only 2x2, can it?
for (var tile = 0; tile < 4; tile++)
{
var materialIdValues = new Color[2, 2];
materialIdMapTiles.Add(materialIdValues);
for (var i = 0; i < 1; i++)
{
for (var j = 0; j < 1; j++)
{
var r = reader.ReadByte();
var g = reader.ReadByte();
var b = reader.ReadByte();
var a = reader.ReadByte();
materialIdValues[j, i] = new Color(r, g, b, a);
}
}
}
// Skip unknown (heightmap LOD?) section
reader.BaseStream.Position += 32L;
// Read material select map
// This can't be only 2x2, can it?
for (var tile = 0; tile < 4; tile++)
{
var materialSelectValues = new Color[2, 2];
materialSelectMapTiles.Add(materialSelectValues);
for (var i = 0; i < 1; i++)
{
for (var j = 0; j < 1; j++)
{
var r = reader.ReadByte();
var g = reader.ReadByte();
var b = reader.ReadByte();
var a = reader.ReadByte();
materialSelectValues[j, i] = new Color(r, g, b, a);
}
}
}
}
var name = Path.GetFileNameWithoutExtension(ctx.assetPath);
// Create material weight map.
var materialWeightMap =
new Texture2D(64, 64, TextureFormat.ARGB32, true)
{
name = name + "_MaterialWeightMap"
};
materialWeightMap.wrapMode = TextureWrapMode.Clamp;
materialWeightMap.SetPixels(0, 0, HalfWidth, HalfWidth, materialWeightMapTiles[0].Cast <Color>().ToArray());
materialWeightMap.SetPixels(HalfWidth, 0, HalfWidth, HalfWidth, materialWeightMapTiles[2].Cast <Color>().ToArray());
materialWeightMap.SetPixels(0, HalfWidth, HalfWidth, HalfWidth, materialWeightMapTiles[1].Cast <Color>().ToArray());
materialWeightMap.SetPixels(HalfWidth, HalfWidth, HalfWidth, HalfWidth, materialWeightMapTiles[3].Cast <Color>().ToArray());
ctx.AddObjectToAsset(name + "MaterialWeightMap", materialWeightMap);
// Create material ID map.
var materialIndicesMap =
new Texture2D(2, 2, TextureFormat.ARGB32, true)
{
name = name + "_MaterialIndicesMap"
};
materialIndicesMap.wrapMode = TextureWrapMode.Clamp;
materialIndicesMap.SetPixels(0, 0, 1, 1, materialIdMapTiles[0].Cast <Color>().ToArray());
materialIndicesMap.SetPixels(1, 0, 1, 1, materialIdMapTiles[2].Cast <Color>().ToArray());
materialIndicesMap.SetPixels(0, 1, 1, 1, materialIdMapTiles[1].Cast <Color>().ToArray());
materialIndicesMap.SetPixels(1, 1, 1, 1, materialIdMapTiles[3].Cast <Color>().ToArray());
ctx.AddObjectToAsset(name + "MaterialIndicesMap", materialIndicesMap);
// Create material select map.
var materialSelectMap =
new Texture2D(2, 2, TextureFormat.ARGB32, true)
{
name = name + "_MaterialSelectMap"
};
materialSelectMap.wrapMode = TextureWrapMode.Clamp;
materialSelectMap.SetPixels(0, 0, 1, 1, materialSelectMapTiles[0].Cast <Color>().ToArray());
materialSelectMap.SetPixels(1, 0, 1, 1, materialSelectMapTiles[2].Cast <Color>().ToArray());
materialSelectMap.SetPixels(0, 1, 1, 1, materialSelectMapTiles[1].Cast <Color>().ToArray());
materialSelectMap.SetPixels(1, 1, 1, 1, materialSelectMapTiles[3].Cast <Color>().ToArray());
ctx.AddObjectToAsset(name + "MaterialSelectMap", materialSelectMap);
// Create heightmap.
var heightMap = new Texture2D(HEIGHTMAP_WIDTH, HEIGHTMAP_HEIGHT, TextureFormat.RGBAFloat, true, true);
heightMap.name = name + "_Heightmap";
heightMap.wrapMode = TextureWrapMode.Clamp;
heightMap.filterMode = FilterMode.Bilinear;
var colors = from height in heightTiles[0].Cast <float>()
select new Color(height, height, height);
heightMap.SetPixels(0, 0, HalfWidth, HalfWidth, colors.ToArray());
colors = from height in heightTiles[2].Cast <float>()
select new Color(height, height, height);
heightMap.SetPixels(HalfWidth, 0, HalfWidth, HalfWidth, colors.ToArray());
colors = from height in heightTiles[1].Cast <float>()
select new Color(height, height, height);
heightMap.SetPixels(0, HalfWidth, HalfWidth, HalfWidth, colors.ToArray());
colors = from height in heightTiles[3].Cast <float>()
select new Color(height, height, height);
heightMap.SetPixels(HalfWidth, HalfWidth, HalfWidth, HalfWidth, colors.ToArray());
ctx.AddObjectToAsset(name + "HeightMap", heightMap);
asset.Heightmap = heightMap;
asset.MaterialSelectMap = materialSelectMap;
asset.MaterialIndicesMap = materialIndicesMap;
asset.MaterialWeightMap = materialWeightMap;
}