hughai/mapping/SlopeMap.java

// Copyright Spring project, Hugh Perkins 2006, 2009
// hughperkins@gmail.com http://manageddreams.com
//
// This program is free software; you can redistribute it and/or modify it
// under the terms of the GNU General Public License as published by the
// Free Software Foundation; either version 2 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful, but
// WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
// or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
//  more details.
//
// You should have received a copy of the GNU General Public License along
// with this program in the file licence.txt; if not, write to the
// Free Software Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-
// 1307 USA
// You can find the licence also on the web at:
// http://www.opensource.org/licenses/gpl-license.php
//
// ======================================================================================
//

package hughai.mapping;

import java.io.*;

import com.springrts.ai.*;
import com.springrts.ai.oo.*;
import com.springrts.ai.oo.clb.*;

import hughai.basictypes.*;
import hughai.*;
import hughai.mapping.HeightMap.HeightMapPos;
import hughai.utils.*;
import hughai.ui.*;

public class SlopeMap
{
   public static final int granularity = 2;

   public static class SlopeMapPos extends Int2 {
      public SlopeMapPos() {

      }
      public SlopeMapPos( Int2 int2 ) {
         x = int2.x;
         y = int2.y;
      }
      public SlopeMapPos( int x, int y ) {
         super( x, y );
      }
      public TerrainPos toTerrainPos() {
         return new TerrainPos( x * 8 * granularity, 0, y * 8 * granularity );
      }
      public static SlopeMapPos fromTerrainPos( TerrainPos terrainPos ) {
         return new SlopeMapPos( (int)terrainPos.x / 8 / granularity,
               (int)terrainPos.z / 8 / granularity );
      }
      public static SlopeMapPos fromHeightMapPos( HeightMapPos heightMapPos ) {
         return new SlopeMapPos( (int)heightMapPos.x / granularity,
               (int)heightMapPos.y / granularity );
      }
   }

   CSAI csai;
   OOAICallback aicallback;
   LogFile logfile;
   //	MovementMaps movementMaps;
   Map gameMap;
   //	HeightMap heightMap;
   //	Maps playerObjects.getMaps();
   PlayerObjects playerObjects;

   private float[][] SlopeMap;

   public SlopeMap( PlayerObjects playerObjects )
   {
      csai = playerObjects.getCSAI();
      aicallback = csai.aicallback;
      logfile = playerObjects.getLogFile();
      gameMap = aicallback.getMap();
      this.playerObjects = playerObjects;
      //		movementMaps = playerObjects.getMovementMaps();
      //		heightMap = playerObjects.getHeightMap();
      //		playerObjects.getMaps() = playerObjects.getMaps();

      playerObjects.getMainUI().registerButton( "Dump slopemap distribution",
            new ButtonSlopeDistribution() );
   }

   class ButtonSlopeDistribution implements MainUI.ButtonHandler {
      @Override
      public void go() {
         float[][]slopemap = GetSlopeMap();
         logfile.WriteLine( "Slopemap distribution:" );
         new MapHelper( playerObjects ).DumpMapDistribution( slopemap );
      }
   }

   public float getSlopeAt( SlopeMapPos slopeMapPos ) {
      GetSlopeMap();
      return SlopeMap[slopeMapPos.x][slopeMapPos.y];
   }

   // ported from Spring's ReadMap.cpp by Hugh Perkins
   private float[][]GetSlopeMap()
   {
      if( SlopeMap != null ) {
         return SlopeMap;
      }
      int mapwidth = gameMap.getWidth();
      int mapheight = gameMap.getHeight();

      int slopemapwidth = mapwidth / granularity;
      int slopemapheight = mapheight / granularity;

      int squaresize = playerObjects.getMaps().getMovementMaps().SQUARE_SIZE; // jsut to save typing...

//      float[][] heightmap = playerObjects.getMaps().getHeightMap().GetHeightMap();
      HeightMap heightMap = playerObjects.getMaps().getHeightMap();
      logfile.WriteLine( "Getting heightmap, this could take a while... " );

      // ArrayIndexer heightmapindexer = new ArrayIndexer( mapwidth + 1, mapheight + 1 );
      logfile.WriteLine("calculating slopes..." );
      logfile.WriteLine("mapwidth: " + slopemapwidth + " " + slopemapheight);

      SlopeMap = new float[ slopemapwidth][ slopemapheight ];
      for(int y = 2; y < mapheight - 2; y+= 2)
      {
         for(int x = 2; x < mapwidth - 2; x+= 2)
         {
//            HeightMapPos heightMapPos = new HeightMapPos( x, y );
            Float3 e1 = new Float3(
                  -squaresize * 4,
                  heightMap.getElevationAt( new HeightMapPos( x - 1, y - 1 ) )
                      - heightMap.getElevationAt( new HeightMapPos( x + 3, y - 1) ),
//                  heightmap[x - 1][ y - 1] - heightmap[x + 3][ y - 1]
                  0);
            Float3 e2 = new Float3(
                  0,
                  heightMap.getElevationAt( new HeightMapPos( x - 1, y - 1 ) )
                  - heightMap.getElevationAt( new HeightMapPos( x - 1, y + 3) ),
//                  heightmap[x - 1][ y - 1] - heightmap[x - 1][ y + 3],
                  -squaresize * 4);

            Float3 n=e2.Cross( e1 );

            n.Normalize();

            e1 = new Float3(
                  squaresize * 4,
                  heightMap.getElevationAt( new HeightMapPos( x + 3, y + 3 ) )
                  - heightMap.getElevationAt( new HeightMapPos( x - 1, y + 3) ),
//                  heightmap[x + 3][ y + 3] - heightmap[x - 1][ y + 3],
                  0);
            e2 = new Float3(
                  0,
                  heightMap.getElevationAt( new HeightMapPos( x + 3, y + 3 ) )
                  - heightMap.getElevationAt( new HeightMapPos( x + 3, y - 1 ) ),
//                  heightmap[x + 3][ y + 3] - heightmap[x + 3][ y - 1],
                  squaresize * 4);

            Float3 n2 = e2.Cross(e1);
            n2.Normalize();

            SlopeMap[ x / granularity][ y / granularity ]= 1 - ( n.y + n2.y ) * 0.5f;
         }
      }
      logfile.WriteLine("... slopes calculated" );
      return SlopeMap;
   }
}