add some saving, some buttons

ai.lua

@@ -20,6 +20,7 @@ end
 function t.load( filename )
   local img, imgd = bmp.load( filename )
   local nodes = { 
+    filename = filename,
     visible = true,
     all = {},
     att = {},
@@ -77,8 +78,8 @@ function t.draw( nodes )
   lg.points( nodes.ptsDef )
 end
 
-function t.save( nodes, filename )
-
+function t.save( nodes )
+  return bmp.savePoints( nodes.all, "512rgb24" )
 end
 
 return t

bmp.lua

@@ -7,6 +7,41 @@ local lfs = love.filesystem
 local ffi = require 'ffi'
 local bit = require 'bit'
 
+local function getHeader( filename )
+  local offset = love.data.unpack( "<I4", assert(love.filesystem.read( "data", filename, 14 )), 11 )
+  local header, size = assert( love.filesystem.read( filename, offset ) )
+  print( "BMP HEADER", filename, size, "\n", header )
+  return header
+end
+
+local formats = {
+  ["512rgb24"] = {
+    header = getHeader( "data/earth/africa.bmp" ),
+    encode = function( r ) return math.floor( r * 255 ) end,
+    bytesPerPixel = 3,
+    w = 512,
+    h = 285,
+  },
+  ["512r4"]     = {
+    header = getHeader( "data/earth/sailable.bmp" ),
+    encode = function( r ) return math.floor( r * 255 / 16 )  end,
+    bytesPerPixel = 0.5,
+    w = 512,
+    h = 285,
+  },
+  ["800r4"]     = {
+    header = getHeader( "data/earth/travel_nodes.bmp" ),
+    encode = function( r ) return math.floor( r * 255 / 16 ) end,
+    bytesPerPixel = 0.5,
+    w = 512,
+    h = 285,
+  }
+}
+
+function t.header( format )
+  return formats[format] and formats[format].header
+end
+
 function t.load( filename )
   local imgd = love.image.newImageData( filename )
   print( "LOADING BITMAP: ", filename, imgd:getSize(), imgd:getFormat(), imgd:getDimensions() )
@@ -15,8 +50,93 @@ function t.load( filename )
   return img, imgd
 end
 
-function t.save( data, filename )
-  local w, h = data:getDimensions()
+--maps an array of 1-byte pixel values  (numbers 0 to 15 inclusive)
+--to an array of half-byte pixel values (which can be concatenated into a string)
+--BUT don't touch the first element (which is assumed to be a header or something)
+local function foldByteArray( bytes )
+  local nybbles = { bytes[1] }
+  for j = 2, #bytes / 2 do
+    local a, b = bytes[ 2 * j - 2 ], bytes[ 2 * j - 1 ]
+    nybbles[j] = string.char( 16 * a + b )
+  end
+  return nybbles
 end
 
+local function packTwentyFour( bytes )
+  local twentyFour = { bytes[1] }
+  for j = 2, #bytes do
+    bytes[j] = string.char( bytes[j], bytes[j], bytes[j] )
+  end
+end
+
+function t.save( data, format )
+  local w, h = data:getDimensions()
+  format = assert( formats[format] )
+  local bytes = { format.header }
+  format.byte = 0
+  local i = 2
+  for x = 0, w - 1 do
+    for y = 0, h - 1 do
+      bytes[i] = format.encode( data:getPixel(x, y) )
+      i = i + 1
+    end  
+  end
+  if format.bytesPerPixel < 1 then bytes = foldByteArray( bytes )
+  else packTwentyFour( bytes ) end
+  return table.concat( bytes )
+end
+
+--takes an array of world-space points in [-180, 180] x [-100, 100] 
+--e.g { { x = 0.5, y = 0.5 }, { x = 0.4, y = 0.6 }, }, etc.
+--and a function which maps points to colours
+function t.savePoints( points, format )
+  format = assert( formats[format] )
+
+  --set up bitmap as an array of pixels
+  local w, h = format.w, format.h
+  local size = 2 + format.w * format.h
+  local bitmap = { format.header }
+  for j = 2, size do bitmap[j] = 0 end
+
+  --this is black-and-white only. easy case
+  if format.bytesPerPixel < 1 then
+    for i, point in ipairs( points ) do
+      local wx, wy = point.x, point.y
+      --get bitmap coordinates
+      local x, y = math.floor(format.w * (wx + 180) / 360), math.floor(format.h * (1.0 - (wy + 100) / 200))
+      --get index into byte array
+      local idx = 2 + x * format.h + y
+      bitmap[idx] = 0x0f --0b00001111
+    end
+    --now map pixels to 4-bit strings, slap on the header, and pass it back up
+    return table.concat( foldByteArray( bitmap ) )
+  end
+
+  --cf. ai.lua: these points can be red or green
+  for i, point in ipairs( points ) do
+    local wx, wy = point.x, point.y
+    --get bitmap coordinates
+    local x, y = math.floor(format.w * (wx + 180) / 360), math.floor(format.h * (1.0 - (wy + 100) / 200))
+    --get index into byte array
+    local idx = 2 + x * format.h + y
+    --in-editor we could have two points in the same place, possibly of the same or different types
+    --in case we miss something upstream, don't corrupt these duplicate points,
+    --just saturate the attack/defense value
+    if bitmap[idx] == 0 then bitmap[idx] = { attack = point.attacking, place = not(point.attacking)}
+    else 
+      bitmap[idx].attack = bitmap[idx].attack or point.attacking
+      bitmap[idx].place  = bitmap[idx].place  or not(point.attacking)
+    end
+  end
+
+  --now map pixels to 3-byte strings
+  for j = 2, size do
+    if   bitmap[j] == 0
+    then bitmap[j] = "\0\0\0"
+    else bitmap[j] = string.char( bitmap[j].attack and 0xff or 0, bitmap[j].place and 0xff or 0, 0 )
+    end
+  end
+
+  return table.concat( bitmap )
+end
 return t

button.lua

@@ -1,32 +1,101 @@
-local t = {}
+local lg = love.graphics
 
-function t.onHover( button )
-  
+local t = {
+  name = "",
+  tooltip = "button",
+  icon = lg.newImage( "icons/eye.bmp" ),
+  x = 8,
+  y = 250,
+  w = 176,
+  h = 24,
+  group = false,
+  visible = true,
+  callback = function( self ) return print( "clicked button: ", self.name, self.x, self.y, self.w, self.h, self.visible ) end
+}
+t.selected, t.next, t.prev = t, t, t
+
+function t.contains( button, x, y )
+  return x < button.x + button.w and x > button.x
+  and    y < button.y + button.h and y > button.y
 end
 
-function t.onClick( button )
-  
+function t.new( b )
+  b = setmetatable( b or {}, t )
+  b.next = t
+  t.prev.next = b
+  b.prev = t.prev
+  t.prev = b
+  return b
 end
 
-function t.newButton( name, tooltip, icon, x, y, w, h, callback )
-  return setmetatable( {
-      name = name,
-      tooltip = tooltip,
-      icon = icon,
-      x = x,
-      y = y,
-      w = w,
-      h = h,
-      callback = callback },
-    t )
+local drawPassOngoing = false
+function t.draw( b )
+  if b == t then
+    drawPassOngoing = not( drawPassOngoing )
+    if not drawPassOngoing then return end
+  elseif b.visible then
+    lg.rectangle( "line", b.x, b.y, b.w, b.h, 6 )
+    lg.printf( b.name,
+      b.x,
+      b.y + 0.5 * ( b.h- lg.getFont():getHeight() ),
+      b.w - 5,
+      "right" )
+    if b.icon then lg.draw( b.icon,
+        b.x, b.y + 0.5 * (b.h - b.icon:getHeight()),
+        0,
+        0.5, 0.5 )
+    end
+    if t.selected == b then
+      lg.rectangle( "fill", b.x, b.y, b.w, b.h, 6 )
+    end
+  end
+  return t.draw( b.next )
 end
 
-function t.draw( button )
-  
+function t.select( b )
+  t.selected = b
 end
 
-t.__index = t
-t.__call = t.newButton
+function t.selectNext()
+  repeat t.selected = t.selected.next until t.selected.visible
+end
+
+function t.selectPrev()
+  repeat t.selected = t.selected.prev until t.selected.visible
+end
+
+function t.selectIn( x, y )
+  t.selected = t
+  repeat t.selected = t.selected.next until (t.selected == t) or (t.selected.visible and t.selected:contains( x, y ))
+end
+
+function t.selectNextInGroup()
+  local group = t.selected and t.selected.group
+  if not group then return t.selectNext() end
+  repeat t.selectNext() until group == t.selected.group
+end
+
+function t.selectPrevInGroup()
+  local group = t.selected and t.selected.group
+  if not group then return t.selectPrev() end
+  repeat t.selectPrev() until group == t.selected.group
+end
+
+function t.displayGroup( group, show )
+  local b = t
+  repeat 
+    b = b.next
+    b.visible = ( b.group == group )
+  until b == t
+  t.visible = true
+end
+
+function t.deselect( b )
+  t.selected = t
+end
 
 setmetatable( t, t )
+t.__index = t
+t.__call = t.new
+
 return t

camera.lua

@@ -47,8 +47,8 @@ function Camera.Set( x, y, w, h )
   tfTerritory:translate( -x * 512 / 360, y * 512 / 360 )  
   
   tfNodes:reset()
-  tfNodes:scale( w / 360, h / 200 )
-  tfNodes:translate( 180 - x , y + 100 )
+  tfNodes:scale( w / 360, -h / 200 )
+  tfNodes:translate( 180 - x , -y - 100 )
   --tfNodes:translate( -x * 800 / 360, y * 400 / 200 )
 end
 

cities.lua

@@ -45,19 +45,19 @@ local citymt = {__index = city}
 function city:formatDisplayInfo()
   return (
     [[
-    NAME:    %s
-    COUNTRY:   %s
-    LONGITUDE: %3.2f
-    LATITUDE:  %3.2f
-    POP:       %d
-    CAPITAL:   %s]]):format( self.name, self.country, self.x, self.y, self.pop, tostring(self.capital) )
+  NAME:    %s
+  COUNTRY:   %s
+  LONGITUDE: %3.2f
+  LATITUDE:  %3.2f
+  POP:       %d
+  CAPITAL:   %s]]):format( self.name, self.country, self.x, self.y, self.pop, tostring(self.capital) )
 end
 
 function t.load( filename )
   
   print( "=== LOADING CITIES. ===" )
 
-  cities = { visible = true, active = false }
+  cities = { visible = true, active = false, filename = filename }
   local n = 1
   local idxPts = 1
   local idxCaps = 1
@@ -97,23 +97,13 @@ function t.load( filename )
   return cities
 end
 
-function t.save( cities, filename )
-  print( "=== SAVING CITIES ===" )
+function t.save( cities )
   local str = {}
   for n, city in ipairs( cities ) do
-    str[n] = ("%-41s%-41s%-14f%-14f%-19d %d"):format( city.name, city.country, city.x, city.y, city.pop, city.capital and 1 or 0 )
+    str[n] = ("%-41s%-41s%-14f%-14f%-19d %d"):format(
+      city.name, city.country, city.x, city.y, city.pop, city.capital and 1 or 0 )
   end
-  str = assert(table.concat( str, "\n" ))
-  local file = assert( io.open( filename, "w+" ) )
-  assert( file:write( str ) )
-  file:close()
-  --[[local tmpname = os.tmpname()
-  local tmpfile = assert( io.open( tmpname, "w+" ) )
-  assert( tmpfile:write( str ) )
-  tmpfile:close()
-  os.remove( filename ) --ooh this is probably pretty bad
-  os.rename( tmpname, filename )]]
-  print( "=== CITIES SAVED ===" )
+  return assert(table.concat( str, "\n" ))
 end
 
 return t

conf.lua

@@ -13,7 +13,7 @@ function love.conf(t)
     t.window.title = "dcEarth"          -- The window title (string)
     t.window.icon = "icons/favicon.png" -- Filepath to an image to use as the window's icon (string)
     t.window.width = 800                -- The window width (number)
-    t.window.height = 600               -- The window height (number)
+    t.window.height = 640               -- The window height (number)
     t.window.borderless = false         -- Remove all border visuals from the window (boolean)
     t.window.resizable = true           -- Let the window be user-resizable (boolean)
     t.window.minwidth = 512             -- Minimum window width if the window is resizable (number)

lines.lua

@@ -3,16 +3,28 @@ local t = {}
 local lfs = love.filesystem
 local lg = love.graphics
 
+local polygon = {  x = 180, X = -180, y = 100, Y = -100  } --default empty bounding box
+local polymt = { __index = polygon }
+
+function polygon:formatDisplayInfo()
+  return ([[
+  x: %f
+  y: %f
+  X: %f
+  Y: %f
+  N: %d]]):format( self.x, self.y, self.X, self.Y, #self )
+end
+
 function t.load( filename )
-  local polys = { visible = true }
+  local polys = { visible = true, filename = filename }
   local poly = {}
-  local n = 0
+  local n = 1
   local k = 1
   for line in assert( lfs.lines( filename ) ) do
     if line:find "b" then
       k = 1
       if #poly > 2 then n = n + 1 end
-      poly = {}
+      poly = setmetatable({}, polymt) --axis-aligned bounding box
       polys[n] = poly
     else
       local _, _, x, y = line:find( "(%g+)%s+(%g+)" )
@@ -20,12 +32,16 @@ function t.load( filename )
       if x and y then 
         poly[k], poly[ k + 1 ] = x, y
         k = k + 2
+        if x < poly.x then poly.x = x end
+        if x > poly.X then poly.X = x end
+        if y < poly.y then poly.y = y end
+        if y > poly.Y then poly.Y = y end
       else
         print( "LINES: malformed line:", filename, line )
       end
     end
   end
-  
+
   if not polys[n] or (#(polys[n]) < 3) then
     polys[n] = nil
     n = n - 1
@@ -35,25 +51,50 @@ function t.load( filename )
   return setmetatable( polys, {__index = t } )
 end
 
-function t.save( lines, filename )
-
+function t.selectNearest( lines, wx, wy )
+  local d = math.huge
+  local nearest
+  for i, poly in ipairs( lines ) do
+    if poly.x - 5 < wx and
+    poly.X + 5 > wx and
+    poly.y - 5 < wy and
+    poly.Y + 5 > wy then
+      for k = 1, #poly, 2 do
+        local x, y = poly[k], poly[k + 1]
+        local r = ( x - wx ) * ( x - wx ) + ( y - wy ) * ( y - wy )
+        if r < d then
+          d = r
+          nearest = poly
+        end
+      end
+    end
+  end
+  return nearest
+end
 
+function t.save( lines )
+  local str = {}
+  for i, poly in ipairs( lines ) do
+    str[i] = table.concat( poly, " " )
+  end
+  str = table.concat( str, "\nb\n" ):gsub("(%S+) (%S+) ", "%1 %2\n")
+  return str
 end
 
 function t.newPolygon( x, y )
-  
+
 end
 
 function t.addToCurrentPolygon( x, y )
-  
+
 end
 
 function t.deletePolygon( poly )
-  
+
 end
 
 function t.drawPolygonEditHandles( poly )
-  
+
 end
 
 function t.draw( lines )

main.lua

@@ -1,10 +1,10 @@
 local love = assert( love, "This tool requires LOVE: love2d.org" )
 --assert( require('mobdebug') ).start() --remote debugger
 local map = require 'map'
+local savemodal = require 'savemodal'
 local button = require 'button'
 local SAVEDIRECTORY = "out/"
 local Camera = require 'camera'
-local wasKeyPressed
 
 function love.load()
 
@@ -13,12 +13,15 @@ function love.load()
   assert( lfs.createDirectory( SAVEDIRECTORY.."data/earth" ))
   assert( lfs.createDirectory( SAVEDIRECTORY.."data/graphics" ))
 
-
-  love.graphics.setNewFont( 12, "mono" )
+  love.keyboard.setKeyRepeat( true )
+  love.graphics.setNewFont( 14 )--, "mono" )
 end
 
 function love.directorydropped( path )
-  if map.path then assert( love.filesystem.unmount( map.path ) ) end
+  if map.path then 
+    assert( love.filesystem.unmount( map.path ) )
+    map.loaded = false
+  end
   love.filesystem.mount( path, "" )
   return map.load( path )
 end
@@ -37,72 +40,35 @@ function love.update( dt )
 
 end
 
-local toolButtons = {
-  "Brush",
-  "Move Nodes",
-  "Add Nodes",
-  "Edit Node",
-  "Draw Polygon",
-  "Erase Polygon"
-}
-
-local layerButtons = {
-  "Africa",
-  "Europe",
-  "North America",
-  "South America",
-  "Asia",
-  "Russia",
-  "Travel Nodes",
-  "AI Markers",
-  "Cities",
-  "Coastlines",
-  "Coastlines Low",
-  "Sailable"
-}
-
-
 function love.draw()
   if not map.loaded then
-    return love.graphics.print( "Drag and drop folder to begin.")
+    local w, h = love.graphics.getDimensions()
+    return love.graphics.printf( "Drag and drop folder to begin.", w / 2 - 200, h / 2 - 128, 400, "center")
   end
 
   love.graphics.push( "all" )
   map.draw()
   love.graphics.pop()
 
-  --Layer buttons.
-  do
-    love.graphics.setColor( 1, 1, 1, 1 )
-    local h = love.graphics.getHeight() - 60
-    for x = 0, 300, 30 do
-      love.graphics.rectangle( "line", x, h, 30, 30 )
-    end
-
-  end
-
   --Status bar.
   local x, y = love.mouse.getPosition()
   local wx, wy = Camera.GetWorldCoordinate( x, y )
   local bx, by = Camera.GetBitmapCoordinate( x, y )
-  local h = love.graphics.getHeight() - 30
-  love.graphics.setColor( 0.2, 0.1, 0.1, 1.0 )
-  love.graphics.rectangle( "fill", 0, h, love.graphics.getWidth() / 2, 30 )
+  local h = love.graphics.getHeight() - 60
+  love.graphics.setColor( 0.1, 0.1, 0.5, 0.8 )
+  love.graphics.rectangle( "fill", 0, 0, 250, love.graphics.getHeight() )
   love.graphics.setColor( 1, 1, 1, 1 )
-  love.graphics.rectangle( "line", 0, h, love.graphics.getWidth() / 2, 30 )
-  love.graphics.print(("SCREEN\t%d\t%d\nWORLD \t%5.2f\t%5.2f"):format(x, y, wx, wy), 0, h)
-  love.graphics.print(("BITMAP\t%5.2f\t%5.2f"):format(bx, by), 200, h )
+  love.graphics.print(([[
+  SCREEN  %-12d %-12d
+  WORLD   %-12.2f%-12.2f
+  BITMAP   %-12.2f%-12.2f
+  %s]]):format(x, y, wx, wy, bx, by, map.editLayer and map.editLayer.filename or ""), 0, 0)
 
-  --Edit box.
-  love.graphics.rectangle( "line", love.graphics.getWidth() / 2, h, love.graphics.getWidth() / 2, 30 )
-  if map.selected then
-    love.graphics.setColor( 0.2, 0.1, 0.1, 1.0 )
-    love.graphics.rectangle( "fill", 0, 0, 150 ,100 )    
-    love.graphics.setColor( 1, 1, 1, 1 )
-    love.graphics.rectangle( "line", 0, 0, 150 ,100 )   
-    love.graphics.setColor( 1.2, 1.1, 1.1, 1.5 )
-    love.graphics.print( map.selected:formatDisplayInfo(), 0, 0 )
-  end
+  if map.selected then love.graphics.print( map.selected:formatDisplayInfo(), 0, 80 ) end
+  if map.selectionLocked then end
+
+  love.graphics.setColor( 1, 1, 1, 0.8 )
+  button:draw()
 end
 
 function love.resize(w, h)
@@ -110,62 +76,185 @@ function love.resize(w, h)
 end
 
 function love.wheelmoved(x, y)
-  Camera.Zoom( (y > 0) and 0.5 or -0.5 )
+  Camera.Zoom( (y > 0) and 0.1 or -0.1 )
 end
 
-function love.mousepressed( x, y, button, istouch, presses )
+function love.mousepressed( x, y, mouseButton, istouch, presses )
   local wx, wy = Camera.GetWorldCoordinate( x, y )
-  if map.loaded then
-    if button == 1 then
-      map.cities.lockSelection()
-    else
-      map.cities.unlockSelection()
-    end
+
+  if button.selected and button.selected:contains( x, y ) then 
+    print( ("MOUSE\tx %f\ty %f\twx %f\twy %f"):format(x, y, wx, wy) )
+    return button.selected:callback()
   end
-  print( ("MOUSE\tx %f\ty %f\twx %f\twy %f"):format(x, y, wx, wy) )
 end
 
 function love.mousemoved( x, y, dx, dy, istouch )
-  local wx, wy = Camera.GetWorldCoordinate( x, y )
-  if map.loaded then 
-    if map.cities.visible then map.selected = map.cities:selectNearest( wx, wy )
-    elseif map.travelnodes.visible then map.selected = map.travelnodes:selectNearest( wx, wy )
-    elseif map.ainodes.visible then map.selected = map.ainodes:selectNearest( wx, wy )
+  if not map.loaded then return end
+  --mouse over menu
+  button.selectIn( x, y )
+
+  --mouse on map
+  if map.selectionLocked then return end
+  if map.editLayer and map.editLayer.selectNearest then 
+    map.selected = map.editLayer:selectNearest( Camera.GetWorldCoordinate( x, y ) )
+  end
+end
+
+function love.keypressed(key, code, isRepeat)
+
+  if code == "left"   then return button.selectPrev() end
+  if code == "right"  then return button.selectNext() end
+  if code == "down"   then return button.selectNextInGroup() end
+  if code == "up"     then return button.selectPrevInGroup() end
+  if code == "return" then return button.selected:callback() end
+
+  if key == "c" then
+    map.selectionLocked = not( map.selectionLocked )
+  end
+end
+
+
+
+do
+
+  local function toolCallback( self )
+    local f = (map.layers[self.layer])[self.name]
+    if f then return f(self) end
+  end
+
+  local tools = {
+    button.new{ name = "SELECT"},
+    button.new{ name = "ERASE",},
+    button.new{ name = "MOVE", },
+    button.new{ name = "ADD",  },
+    button.new{ name = "EDIT", },
+    button.new{ name = "DRAW", },
+  }
+  for i, v in ipairs( tools ) do 
+    v.callback = toolCallback 
+    v.y = 250 + (v.h + 4) * ( i + 1 )
+    v.visible = false
+  end
+
+  local layerButtons = {}
+
+  local layers = {
+    { name = "AF",    layer = "africa"       },
+    { name = "EU",    layer = "europe"       },
+    { name = "NA",    layer = "northamerica" },
+    { name = "SA",    layer = "southamerica" },
+    { name = "AS",    layer = "southasia"    },
+    { name = "RU",    layer = "russia"       },
+    { name = "PATH",  layer = "travelnodes"  },
+    { name = "AI",    layer = "ainodes"      },
+    { name = "CITY",  layer = "cities"       },
+    { name = "COAST", layer = "coastlines"   },
+    { name = "LOW",   layer = "coastlinesLow"},
+    { name = "SAIL",  layer = "sailable"     },
+  }
+
+  local visibilityIcon = love.graphics.newImage( "icons/eye.bmp" )
+  local function toggleVisibleLayer( self )
+    if not (self and self.layer) then return end
+    local ml = map.layers[ self.layer ]
+    ml.visible = not( ml.visible )
+    self.icon = ml.visible and visibilityIcon
+  end
+
+  local soloIcon = love.graphics.newImage( "icons/eye.bmp" )
+  local function soloVisibleLayer( self )
+    --hide icons for disabled invisible layers
+    print( "===SOLO LAYER===", self.layer )
+    for i, button in ipairs( layerButtons ) do
+      if button.layer ~= self.layer then
+        button.icon = false
+      end
     end
+    self.icon = soloIcon
+    
+    for k, layer in pairs( map.layers ) do
+      print( "invisible layer, map:", k, layer)
+      layer.visible = false
+    end
+    map.layers[ self.layer ].visible = true
+    
   end
-end
 
-local function ToggleVisibility( layer )
-  if not layer then return end
-  local ml
-  if map[layer] then ml = map[layer] end 
-  if map.territory[layer] then ml = map.territory[layer] end
-  assert( ml )
-  ml.visible = not( ml.visible )
-  print( layer, ml.visible )
-end
-
-local layerVisibilityKeybinds = {
-  ["1"] = "africa",
-  ["2"] = "europe",
-  ["3"] = "northamerica",
-  ["4"] = "southamerica",
-  ["5"] = "southasia",
-  ["6"] = "russia",
-  ["7"] = "sailable",
-  ["8"] = "coastlines",
-  ["9"] = "coastlinesLow",
-  ["0"] = "international",
-  ["-"] = "cities",
-  ["="] = "travelnodes",
-  ["backspace"] = "ainodes",
-}
-
-function love.keypressed(key)
-  ToggleVisibility( layerVisibilityKeybinds[key] )
-  wasKeyPressed = true
-  
-  if key == "l" then
-    return map.save()
+  local backButton
+  local function editLayer( self )
+    map.editLayer = map.layers[ self.layer ]
+    for k, button in pairs( layerButtons ) do button.visible = false end
+    for k, button in pairs( tools ) do 
+      button.visible = true
+      button.layer = self.layer
+    end
+    backButton.visible = true
+    return soloVisibleLayer( self )
   end
+
+  local function copy( i, target )
+    for k, v in pairs( layers[i] ) do 
+      target[k] = target[k] or v
+    end
+    return target
+  end
+
+
+  local y = 250
+  local soloButtons = {}
+  local showButtons = {}
+  local editButtons = {}
+  for i = 1, #layers do
+    editButtons[i] = button.new( copy( i, { 
+          x = 8,
+          y = y + (button.h + 4) * i,
+          w = 112,
+          callback = editLayer,
+          group = "edit",
+        }))
+    layerButtons[ 3 * i - 2 ] = editButtons[i]
+
+    showButtons[i] = button.new( copy( i, {
+          x = 128,
+          y = y + (button.h + 4) * i,
+          w = 24,
+          name = "V",
+          callback = toggleVisibleLayer,
+          icon = visibilityIcon,
+          group = "show",
+        }))
+    layerButtons[ 3 * i - 1 ] = showButtons[i]
+
+    soloButtons[i] = button.new( copy( i, {
+          x = 160,
+          y = y + (button.h + 4) * i,
+          w = 24,
+          name = "S",
+          callback = soloVisibleLayer,
+          icon = soloIcon,
+          group = "solo",
+        }))
+    layerButtons[ 3 * i    ] = soloButtons[i]
+  end
+
+
+  local function back( self )
+    for k, button in pairs( tools ) do button.visible = false end
+    for k, button in pairs( layerButtons ) do button.visible = true end
+    self.visible = false
+    map.editLayer = false
+  end
+
+  backButton = button.new{
+    name = "UP", 
+    visible = false,
+    y = 250 + button.h + 4,
+    icon = love.graphics.newImage( "icons/up.bmp" ),
+    callback = back,
+  }
+
+  button.new{ name = "UNDO", y = 250, callback = map.undo, icon = love.graphics.newImage( "icons/undo.bmp" ) }
+  button.new{ name = "SAVE", y = 222, callback = savemodal.start, icon = false }
+
 end
+

map.lua

@@ -6,11 +6,31 @@ local Nodes = require 'travelNodes'
 local Camera = require 'camera'
 local Territory = require 'territory'
 
-local map = {  
-  loaded = false,
+--flat list of editable layers for convenience
+local layers = {
   coastlines = false,
   coastlinesLow = false,
   international = false,
+  africa = false,
+  europe = false,
+  northamerica = false,
+  russia = false,
+  southamerica = false,
+  southasia = false,
+  travelnodes = false,
+  sailable = false,
+  ainodes = false,
+  cities = false,
+}
+
+local map = {  
+  layers = layers,
+  path = false,
+  loaded = false,
+  selected = false,
+  selectionLocked = false,
+  editLayer = false,
+  
   territory = {
     africa = false,
     europe = false,
@@ -19,6 +39,11 @@ local map = {
     southamerica = false,
     southasia = false
   },
+  
+  background = false,
+  coastlines = false,
+  coastlinesLow = false,
+  international = false,
   travelnodes = false,
   sailable = false,
   ainodes = false,
@@ -26,6 +51,7 @@ local map = {
 }
 
 function map.load( path )
+  map.background = lg.newImage( "data/graphics/blur.bmp" )
   map.cities = Cities.load( "data/earth/cities.dat" )
   map.coastlines = Lines.load( "data/earth/coastlines.dat" )
   map.coastlinesLow = Lines.load( "data/earth/coastlines-low.dat" )
@@ -38,6 +64,11 @@ function map.load( path )
   end
   map.loaded = true
   map.path = path
+  
+  --update references
+  for k, v in pairs( layers ) do
+    layers[k] = map[k] or map.territory[k]
+  end
 end
 
 function map.draw()
@@ -45,10 +76,13 @@ function map.draw()
   if not map.loaded then return end
 
   do --territory
-    lg.setColor( 1,1,1,1)
     lg.setLineJoin( "none" )
     lg.replaceTransform( Camera.tfTerritory )
     lg.setBlendMode( "add" )
+
+    lg.setColor( 1, 1, 1, 0.2 )
+    lg.draw( map.background )
+    lg.setColor( 1, 1, 1, 0.5 )
     for k, v in pairs(map.territory) do
       if v.visible then 
         v:draw()
@@ -78,12 +112,21 @@ function map.draw()
   do --all this stuff is drawn in world coordinates, ( -180, 180 ) x ( -100, 100 )
 
     lg.replaceTransform( Camera.tf )
-    
+
     if map.selected then
-      lg.setColor( 1.0, 0.5, 0.5, 0.9 )
-      lg.setLineJoin( "miter" )
-      lg.setLineWidth( 1.0 / Camera.zoom )
-      lg.circle( "line", map.selected.x, map.selected.y, 0.2 + 1.0 / Camera.zoom )
+      if map.selected[1] then --lines
+        local p = map.selected
+        lg.setColor( 0.4, 0.5, 0.8, 0.5 )
+        lg.setLineWidth( 0.2 / Camera.zoom )
+        lg.rectangle( "fill", p.x, p.y, p.X - p.x, p.Y - p.y )
+        lg.setColor( 1.0, 0, 0, 1 )
+        lg.line( p )
+      else --points
+        lg.setColor( 1.0, 0.5, 0.5, 0.9 )
+        lg.setLineJoin( "miter" )
+        lg.setLineWidth( 1.0 / Camera.zoom )
+        lg.circle( "line", map.selected.x, map.selected.y, 0.2 + 1.0 / Camera.zoom )
+      end
     end
 
     if map.cities.visible then --points
@@ -94,7 +137,7 @@ function map.draw()
 
       lg.setColor( 1, 1, 0.0, 0.5 )
       map.cities.drawCapitals()
-      
+
     end
 
     if map.ainodes.visible then
@@ -128,22 +171,24 @@ function map.draw()
         map.travelnodes:draw()
       end
     end
-    
+
 
   end
 
 end
 
+local function write( filename, string )
+  print( "Pretending to write", string:len(), "bytes to", filename )
+  --[[  os.rename( filename, filename..".bak" ) --just in case :^)
+  local file = assert( io.open( filename, "w+" ) )
+  assert( file:write( string ) )
+  file:close()]]
+end
+
 function map.save()
-  map.cities:save( map.path.."/data/earth/cities.dat" )
-  map.coastlines.save()
-  map.coastlinesLow.save()
-  map.international.save()
-  map.sailable.save()
-  map.travelnodes.save()
-  map.ainodes.save()
-  for k, v in pairs(map.territory) do
-    map.territory[k].save()
+  for k, layer in pairs( layers ) do
+    print( "SAVING:", k, tostring( layer.filename ) )
+    write( map.path..tostring( layer.filename ), assert( layer:save() ) )
   end
 end
 
@@ -151,4 +196,8 @@ function map.hover(x, y)
 
 end
 
+function map.undo()
+  print( "=== UNDO ===" )
+end
+
 return map

modal.lua

@@ -0,0 +1,51 @@
+local love = assert( love )
+local button = require( "button" )
+local t = {}
+t.__index = t
+
+local i = 0
+function t.start( self )
+  i = i + 1
+  t[i] = t[i] or {}
+  
+  --store callbacks
+  for name in pairs( self ) do
+    if love[name] then
+      t[i][name] = love[name]
+      love[name] = self[name]
+    end
+  end
+  
+  --store menus
+  local b = button.next
+  repeat 
+    t[i][b] = b.visible
+    b = b.next
+  until b == button
+end
+
+function t.stop( self )
+  --restore callbacks
+  for name in pairs( self ) do
+    if love[name] then
+      love[name] = t[i][name]
+    end
+  end
+  
+  --restore menus
+  local b = button
+  repeat 
+    b = b.next
+    b.visible = t[i][b]
+  until b == button
+  
+  t[i] = nil
+  i = i - 1
+  
+end
+
+function t.new( modal )
+  return setmetatable( modal or {}, t )
+end
+
+return t

savemodal.lua

@@ -0,0 +1,52 @@
+local love = assert( love )
+local modal = require( "modal" )
+local button = require( "button" )
+local map = require( "map" )
+local t = {}
+
+local saveLocation = map.path
+local saveButton = button.new{   
+  group = "saveModal",
+  name = "save",
+  callback = function() map.save(); return t:stop() end,
+  visible = false,
+  x = 20,
+  y = 20,
+  w = 400,
+  h = 100,
+}
+
+local cancelButton = button.new{
+  group = "saveModal",
+  name = "cancel",
+  visible = false,
+  callback = function() return t:stop() end,
+  x = 20,
+  y = 150,
+  w = 400,
+  h = 100
+  }
+
+function t.start()
+  modal.start( t )
+  button.selected = saveButton
+  saveButton.name = "save to "..map.path
+  button.displayGroup( "saveModal", true )
+end
+
+function t.draw()
+  love.graphics.clear( 0,0,0,1 )
+  love.graphics.setColor( 1, 1, 1, 0.5 )
+  button:draw()
+  
+end
+
+function t.directorydropped( path )
+  saveLocation = path
+  map.path = path
+  saveButton.name = "save to "..map.path
+  return love.filesystem.mount( path, "" )
+end
+
+
+return modal.new( t )

scratch.lua

@@ -1,2 +0,0 @@
-local jit = require 'jit'
-for k, v in pairs( jit.opt ) do print(k , v ) end

territory.lua

@@ -18,6 +18,7 @@ function t.load( filename, name )
   local territory = { 
     visible = true, 
     name = name,
+    filename = filename,
     colour = colours[name],
     border = {},
     img = img,
@@ -144,12 +145,8 @@ function t.computeBorder( territory, threshold, key )
   end
 end
 
-function t.drawConnections( nodes )
-
-end
-
-function t.save( nodes, filename )
-
+function t.save( territory )
+  return bmp.save( territory.imgd, "512rgb24" )
 end
 
 return t

textinput.lua

@@ -0,0 +1,64 @@
+local love = assert( love )
+local utf8 = require("utf8")
+local modal = require( "modal" )
+local t = modal.new{ }
+
+function t.setCurrentModal( fields )
+  t.currentModal = assert( fields )
+  t.currentIdx = 1
+  return t:start()
+end
+
+function t.draw()
+  if not t.currentModal then return end
+  love.graphics.setColor( 0, 0, 0, 0.3 )
+  love.graphics.rectangle("fill", 0, 0, love.graphics.getDimensions())
+
+
+  -- other fields
+  for i, field in ipairs( t.currentModal ) do
+    love.graphics.setColor( 1, 1, 1, 0.2 )
+    local mode = (i == t.currentIdx) and "fill" or "line"    
+    love.graphics.rectangle( mode, 200, i * 28, 200, 14, 4 )
+    love.graphics.setColor( 1, 1, 1, 1 )
+    love.graphics.print( field.name, 8, i * 28 )
+    love.graphics.print( field.value, 216, i * 28 )
+  end
+end
+
+function t.textinput( char )
+  if t.currentModal then 
+    local field = t.currentModal.currentFieldIdx
+    t.currentModal.currentField.value = t.currentModal.currentField.value .. char
+  end
+end
+
+function t.keypressed(key, code, isRepeat)
+  if code == "down" then
+    t.currentIdx = t.currentIdx + 1
+    if t.currentIdx > #(t.currentModal) then
+      t.currentIdx = 1
+    end
+  elseif code == "up" then
+    t.currentIdx = t.currentIdx - 1
+    if t.currentIdx < 1 then
+      t.currentIdx = #(t.currentModal)
+    end
+  end
+
+  if key == "backspace" then
+    -- get the byte offset to the last UTF-8 character in the string.
+    local byteoffset = utf8.offset(text, -1)
+
+    if byteoffset then
+      -- remove the last UTF-8 character.
+      -- string.sub operates on bytes rather than UTF-8 characters, so we couldn't do string.sub(text, 1, -2).
+      text = string.sub(text, 1, byteoffset - 1)
+    end
+  end
+  if key == "escape" then
+    return t:stop()
+  end
+end
+
+return t

travelNodes.lua

@@ -14,17 +14,17 @@ local function isConnected( startNode, endNode )
   local ix, iy, fx, fy = startNode.x, startNode.y, endNode.x, endNode.y
   if fx < -180 then fx = fx + 180 end
   if fx > 180 then fx = fx - 180 end
-  
+
   local dx, dy = fx - ix, fy - iy
   local mag = math.sqrt( dx * dx + dy * dy )
   local n = 2 * math.floor( mag )
   dx, dy = 0.5 * dx / mag, 0.5 * dy / mag
-  
+
   for i = 1, n do
     ix, iy = ix + dx, iy + dy
     if not( isSailable( ix, -iy ) ) then return nil end
   end
-  
+
   return true
 end
 
@@ -41,39 +41,43 @@ function travelNode:formatDisplayInfo()
   LATITUDE: %3.2f]]):format( self.idx, self.x, self.y )
 end
 
+local function worldToBitmap( x, y )
+
+end
+
+local function bitmapToWorld( x, y )
+  return 360 * ( x - 800 ) / 800 - 360 / 2 + 360, 
+  360 * ( 600 / 800 ) * ( y - 600 ) / 600 + 180
+end
+
 function t.load( filename, sailable )
-  
+
   isSailable = sailable
   local img, imgd = bmp.load( filename )
-  local nodes = { visible = true, nodes = {}, points = {}, connections = {}, img = img }
-  
-  print( "=== Loading Nodes: ===" )
+  local nodes = { filename = filename, visible = true, nodes = {}, points = {}, connections = {}, img = img }
   local n = 1
   for x = 0, 799 do
     for y = 0, 399 do
       if imgd:getPixel( x, 399 - y ) > 0 then
-        local long = 360 * ( x - 800 ) / 800 - 360 / 2 + 360
-        local lat = 360 * ( 600 / 800 ) * ( 600 - y ) / 600 - 180
-        nodes.nodes[n] = setmetatable({x = long, y = -lat, idx = n}, mtTravelNode )
-        print( nodes.nodes[n]:formatDisplayInfo() )
+        local long, lat = bitmapToWorld( x, y )
+        nodes.nodes[n] = setmetatable({x = long, y = lat, idx = n}, mtTravelNode )
         nodes.points[ 2 * n - 1 ] = long
         nodes.points[ 2 * n ] = lat
         n = n + 1
       end
     end
   end
-  
+
   for i, srcNode in ipairs( nodes.nodes ) do
     local adjacent = {}
-    
+
     for j, destNode in ipairs( nodes.nodes ) do
       adjacent[j] = isConnected( srcNode, destNode )
     end
-    
+
     nodes.connections[i] = adjacent
   end
-  
-  print( "=== Nodes Loaded ===" )
+
   nodes.nodes = locationQuery.New( nodes.nodes )
   setmetatable( nodes, {__index = t} )
   return nodes
@@ -81,7 +85,7 @@ end
 
 --Determine if graph has more than one connected component.
 function t.isConnected( nodes )
-  
+
 end
 
 function t.draw( nodes )
@@ -92,18 +96,18 @@ function t.draw( nodes )
 end
 
 function t.drawConnections( nodes )
-  
+
   for i, connection in pairs( nodes.connections ) do
     for j in pairs( connection ) do
       local ix, iy, fx, fy = nodes.nodes[i].x, nodes.nodes[i].y, nodes.nodes[j].x, nodes.nodes[j].y
-      lg.line( ix, -iy, fx, -fy )
+      lg.line( ix, iy, fx, fy )
     end
   end
-  
+
 end
 
-function t.save( nodes, filename )
-  
+function t.save( nodes )
+  return bmp.savePoints( nodes.nodes, "800r4")
 end
 
 return t