Automap/Renderers/StandardRenderer.cs

   1 using System;
   2 using System.Linq;
   3
   4 using Hjg.Pngcs;
   5
   6 using Vintagestory.API.Client;
   7 using Vintagestory.API.Common;
   8 using Vintagestory.API.MathTools;
   9 using Vintagestory.Common;
  10
  11 namespace Automap
  12 {
  13         public class StandardRenderer : AChunkRenderer
  14         {
  15
  16                 /// <summary>
  17                 /// Renders shards similar to the Default VS version, plus P.O.I. markings.
  18                 /// </summary>
  19                 /// <param name="clientAPI">Client API.</param>
  20                 /// <param name="logger">Logger.</param>
  21                 public StandardRenderer(ICoreClientAPI clientAPI, ILogger logger, bool seasonalColor) : base(clientAPI, logger, seasonalColor)
  22                 {
  23
  24                 }
  25
  26                 public override void GenerateChunkPngShard(Vec2i chunkPos, IMapChunk mc, ColumnMeta metaData, out uint pixelCount)
  27                 {
  28                 pixelCount = 0;
  29                 BlockPos tmpPos = new BlockPos( );
  30                 Vec2i localpos = new Vec2i( );
  31
  32                 var chunksColumn = new IWorldChunk[ClientAPI.World.BlockAccessor.MapSizeY / chunkSize];
  33
  34                 //pre-create PNG line slices...
  35                 ImageLine[ ] lines = Enumerable.Repeat(new object( ), chunkSize).Select(l => new ImageLine(this.PngWriter.ImgInfo)).ToArray( );
  36
  37                 int topChunkY = metaData.YMax / chunkSize;
  38
  39                 for (int chunkY = 0; chunkY <= topChunkY; chunkY++)
  40                 {
  41                         chunksColumn[chunkY] = ClientAPI.World.BlockAccessor.GetChunk(chunkPos.X, chunkY, chunkPos.Y);
  42                         WorldChunk ownChunk = chunksColumn[chunkY] as WorldChunk;
  43                         if (ownChunk != null)
  44                                 {
  45                                 if (ownChunk.IsPacked( )) ownChunk.Unpack( );//Gah - probably done already by chunk processor
  46                                 }
  47                                 else
  48                                 {
  49                                 Logger.Warning("CHUNK A.W.O.L. : X{0} Y{1} Z{2} - Missing slice FOR COLUMN", chunkPos.X, chunkY, chunkPos.Y);
  50                                 //return;
  51                                 }
  52                 }
  53
  54                 // Prefetch map chunks, in pattern
  55                 IMapChunk[ ] mapChunks = new IMapChunk[ ]
  56                 {
  57                         ClientAPI.World.BlockAccessor.GetMapChunk(chunkPos.X - 1, chunkPos.Y - 1),
  58                         ClientAPI.World.BlockAccessor.GetMapChunk(chunkPos.X - 1, chunkPos.Y),
  59                         ClientAPI.World.BlockAccessor.GetMapChunk(chunkPos.X, chunkPos.Y - 1)
  60                 };
  61
  62
  63                 for (int pixelIndex = 0; pixelIndex < (chunkSize * chunkSize); pixelIndex++) {
  64                 MapUtil.PosInt2d(pixelIndex, chunkSize, localpos);
  65                 int localX = localpos.X;
  66                 int localZ = localpos.Y;
  67                 ushort localY = metaData.HeightMap[localX, localZ];//mc.RainHeightMap[posIndex];//NOPE NOPE NOPE, RAINMAP IS CRAP. USE IT NOT!
  68
  69                 int localChunkY = localY / chunkSize;
  70                 if (localChunkY >= (chunksColumn.Length)) continue;//Out of range!
  71                 if (chunksColumn[localChunkY] == null) continue;//BIG Gaps!
  72
  73                 float slopeBoost = 1;
  74                 int leftTop, rightTop, leftBot;
  75
  76                 IMapChunk leftTopMapChunk = mc;
  77                 IMapChunk rightTopMapChunk = mc;
  78                 IMapChunk leftBotMapChunk = mc;
  79
  80                 int topX = localX - 1;
  81                 int botX = localX;
  82                 int leftZ = localZ - 1;
  83                 int rightZ = localZ;
  84
  85                 if (topX < 0 && leftZ < 0) {
  86                 leftTopMapChunk = mapChunks[0];
  87                 rightTopMapChunk = mapChunks[1];
  88                 leftBotMapChunk = mapChunks[2];
  89                 }
  90                 else {
  91                 if (topX < 0) {
  92                 leftTopMapChunk = mapChunks[1];
  93                 rightTopMapChunk = mapChunks[1];
  94                 }
  95                 if (leftZ < 0) {
  96                 leftTopMapChunk = mapChunks[2];
  97                 leftBotMapChunk = mapChunks[2];
  98                 }
  99                 }
 100
 101                 topX = GameMath.Mod(topX, chunkSize);
 102                 leftZ = GameMath.Mod(leftZ, chunkSize);
 103
 104                 leftTop = leftTopMapChunk == null ? 0 : Math.Sign(localY - leftTopMapChunk.RainHeightMap[leftZ * chunkSize + topX]);
 105                 rightTop = rightTopMapChunk == null ? 0 : Math.Sign(localY - rightTopMapChunk.RainHeightMap[rightZ * chunkSize + topX]);
 106                 leftBot = leftBotMapChunk == null ? 0 : Math.Sign(localY - leftBotMapChunk.RainHeightMap[leftZ * chunkSize + botX]);
 107
 108                 float slopeness = (leftTop + rightTop + leftBot);
 109
 110                 if (slopeness > 0) slopeBoost = 1.2f;
 111                 if (slopeness < 0) slopeBoost = 0.8f;
 112
 113                 slopeBoost -= 0.15f; //Slope boost value
 114
 115                 int blockId = chunksColumn[localChunkY].Blocks[MapUtil.Index3d(localX, (localY % chunkSize), localZ, chunkSize, chunkSize)];
 116
 117                 Block block = ClientAPI.World.Blocks[blockId];
 118
 119                 tmpPos.Set(chunkSize * chunkPos.X + localpos.X, localY, chunkSize * chunkPos.Y + localpos.Y);
 120
 121                 int red = 0, green = 0, blue = 0;
 122
 123                 ExtractBlockColor(tmpPos, block, slopeBoost, out red, out green, out blue);
 124
 125                 //============ POI Population =================
 126                 if (BlockID_Designators.ContainsKey(blockId)) {
 127                 var desig = BlockID_Designators[blockId];
 128
 129                         if (desig.Enabled)
 130                         {
 131                         red = desig.OverwriteColor.R;
 132                         green = desig.OverwriteColor.G;
 133                         blue = desig.OverwriteColor.B;
 134                         }
 135                 }
 136
 137                 ImageLineHelper.SetPixel(lines[localZ], localX, red, green, blue);
 138                 pixelCount++;
 139                 }
 140
 141                 for (int row = 0; row < this.PngWriter.ImgInfo.Rows; row++) {
 142                 this.PngWriter.WriteRow(lines[row], row);
 143                 }
 144
 145                 this.PngWriter.End( );
 146                 }
 147         }
 148 }