:reset
import jp.kobe_u.scarab._ , dsl._

var board = Array.ofDim[Int](11,11);
var numOfLights = Array.ofDim[Int](11,11);
// 黒マスの設定 //
board(2)(2) = 1;
board(2)(3) = 1;
board(2)(4) = 1;
board(2)(5) = 1;
board(2)(8) = 1;
board(2)(9) = 1;
board(3)(2) = 1;
board(4)(4) = 1;
board(4)(5) = 1;
board(4)(7) = 1;
board(5)(2) = 1;
board(5)(4) = 1;
board(5)(7) = 1;
board(5)(9) = 1;
board(6)(2) = 1;
board(6)(4) = 1;
board(6)(7) = 1;
board(6)(9) = 1;
board(7)(4) = 1;
board(7)(6) = 1;
board(7)(7) = 1;
board(8)(9) = 1;
board(9)(2) = 1;
board(9)(3) = 1;
board(9)(6) = 1;
board(9)(7) = 1;
board(9)(8) = 1;
board(9)(9) = 1;

// 照明の数の設定 //
numOfLights(2)(2) = 2;
numOfLights(2)(3) = -1;
numOfLights(2)(4) = -1;
numOfLights(2)(5) = 1;
numOfLights(2)(8) = 1;
numOfLights(2)(9) = -1;
numOfLights(3)(2) = -1;
numOfLights(4)(4) = -1;
numOfLights(4)(5) = 1;
numOfLights(4)(7) = -1;
numOfLights(5)(2) = -1;
numOfLights(5)(4) = 1;
numOfLights(5)(7) = -1;
numOfLights(5)(9) = 0;
numOfLights(6)(2) = 0;
numOfLights(6)(4) = -1;
numOfLights(6)(7) = -1;
numOfLights(6)(9) = -1;
numOfLights(7)(4) = 1;
numOfLights(7)(6) = 1;
numOfLights(7)(7) = 1;
numOfLights(8)(9) = -1;
numOfLights(9)(2) = -1;
numOfLights(9)(3) = -1;
numOfLights(9)(6) = 1;
numOfLights(9)(7) = 0;
numOfLights(9)(8) = -1;
numOfLights(9)(9) = -1;

def printSolution(brd : Array[Array[Int]], lig : Array[Array[Int]], sol : Assignment = null) {
    var image = Array.fill[Char](10*3, 10*3) { ' ' };
    for (i <- 1 to 10; j <- 1 to 10) {
       val x = (i-1) * 3 + 1;
       val y = (j-1) * 3 + 1;
       if (brd(i)(j) != 0) image(x)(y) = (brd(i)(j) + '0').toChar; else image(x)(y) = '.';
       if (sol != null) {
       	  if (brd(i)(j) == 1) image(x)(y) = '@'; //黒マス
       	  if (sol('x(i,j)) == 1) image(x)(y) = 'O'; //照明
       }
       else if (brd(i)(j) == 1) image(x)(y) = '@'; //黒マス
    }
    for (i <- 1 to 10 * 3) {
        for (j <- 1 to 10 * 3)
	    print(s"${image(i-1)(j-1)} ");
	println();
    }
}

// 変数の定義
for (i <- 1 to 10; j <- 1 to 10){
    int('x(i,j), 0, 1); // 照明が置かれているかどうか
}

// 制約の追加
for (i <- 1 to 10; j <- 1 to 10) {

    var lights : Seq[Var] = Seq();
    if (board(i)(j) == 1) add('x(i,j) === 0) // 黒マスには照明を置かない


    // 黒マスの周りに置く照明の制約
    if (board(i)(j) == 1){ // 注目マスが黒マスである
       if(i - 1 >= 1) lights = lights :+ 'x(i-1,j);
       if(i + 1 <= 10) lights = lights :+ 'x(i+1,j);
       if(j - 1 >= 1) lights = lights :+ 'x(i,j-1);
       if(j + 1 <= 10) lights = lights :+ 'x(i,j+1);
       if(numOfLights(i)(j) == 0) add(Sum(lights) === 0);
       if(numOfLights(i)(j) == 1) add(Sum(lights) === 1);
       if(numOfLights(i)(j) == 2) add(Sum(lights) === 2);
       if(numOfLights(i)(j) == 3) add(Sum(lights) === 3);
       if(numOfLights(i)(j) == 4) add(Sum(lights) === 4);
   }

   // 照明があるマスは他の照明に照らされてはいけない制約 and 全ての白マスが照らされているという制約
   if (board(i)(j) == 0){ //注目マスが黒マスではない
      var check : Seq[Var] = Seq();
      check = check :+ 'x(i,j); //注目マスは最初に入れる

      // 上方向の走査(iは定数,jが変数)
      var loop: Boolean = true;
      var k: Int = j - 1;
      while(loop){
      	  if(k == 0) loop = false;
	  else if(board(i)(k) == 1) loop = false;
	  else if (board(i)(k) == 0) check = check :+ 'x(i,k);
   	  k = k - 1;
      }

      // 下方向の走査(iは定数,jが変数)
      loop = true;
      k = j + 1;
      while(loop){
   	if((k == 11)) loop = false;
	else if(board(i)(k) == 1) loop = false;
	else if (board(i)(k) == 0) check = check :+ 'x(i,k);
      	k = k + 1;
      }

      // 左方向の走査(iは変数,jが定数)
      loop = true;
      k = i - 1;
      while(loop){
      	  if(k == 0) loop = false;
	  else if(board(k)(j) == 1) loop = false;
	  else if (board(k)(j) == 0) check = check :+ 'x(k,j);
      	  k = k - 1;
      }

      // 右方向の走査(iは定数,jが変数)
      loop = true;
      k = i + 1;
      while(loop){
      	  if(k == 11) loop = false;
      	  else if(board(k)(j) == 1) loop = false;
      	  else if (board(k)(j) == 0) check = check :+ 'x(k,j);
      	  k = k + 1;
      }
      add(!('x(i,j) === 1) || (Sum(check) <= 1)); // 照明から黒マスまたは外枠になるまでに照らされるマスの中に含まれる照明の数は1つであればいい
      if (board(i)(j) == 0) add(Sum(check) >= 1); // 全マスが照らされていなければならない
   }
}

println("求解前")
// 求解前の盤面を表示
printSolution(board, numOfLights, solution)

find;

println("求解後")
// 求解後の盤面を表示
printSolution(board, numOfLights, solution);