validate.go
author Mikael Berthe <mikael@lilotux.net>
Sun, 16 Oct 2016 10:08:11 +0200
changeset 8 955d3add9426
parent 7 takuzu.go@e284e3ad2800
child 9 4b3436c03726
permissions -rw-r--r--
More code refactoring; split takuzu.go and create validate.go
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
0
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
     1
package takuzu
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
     2
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
     3
import (
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
     4
	"fmt"
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
     5
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
     6
	"github.com/pkg/errors"
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
     7
)
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
     8
8
955d3add9426 More code refactoring; split takuzu.go and create validate.go
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
     9
// checkRange returns true if the range is completely defined, and an error
955d3add9426 More code refactoring; split takuzu.go and create validate.go
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    10
// if it doesn't follow the rules for a takuzu line or column
955d3add9426 More code refactoring; split takuzu.go and create validate.go
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    11
// Note that the boolean might be invalid if the error is not nil.
955d3add9426 More code refactoring; split takuzu.go and create validate.go
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    12
func checkRange(cells []Cell) (bool, error) {
955d3add9426 More code refactoring; split takuzu.go and create validate.go
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    13
	full := true
955d3add9426 More code refactoring; split takuzu.go and create validate.go
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    14
	size := len(cells)
955d3add9426 More code refactoring; split takuzu.go and create validate.go
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    15
	counters := []int{0, 0}
0
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    16
8
955d3add9426 More code refactoring; split takuzu.go and create validate.go
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    17
	var prevCell Cell
955d3add9426 More code refactoring; split takuzu.go and create validate.go
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    18
	var prevCellCount int
0
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    19
8
955d3add9426 More code refactoring; split takuzu.go and create validate.go
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    20
	for _, c := range cells {
955d3add9426 More code refactoring; split takuzu.go and create validate.go
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    21
		if !c.Defined {
955d3add9426 More code refactoring; split takuzu.go and create validate.go
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    22
			full = false
955d3add9426 More code refactoring; split takuzu.go and create validate.go
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    23
			prevCell.Defined = false
955d3add9426 More code refactoring; split takuzu.go and create validate.go
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    24
			prevCellCount = 0
955d3add9426 More code refactoring; split takuzu.go and create validate.go
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    25
			continue
0
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    26
		}
8
955d3add9426 More code refactoring; split takuzu.go and create validate.go
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    27
		counters[c.Value]++
955d3add9426 More code refactoring; split takuzu.go and create validate.go
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    28
		if prevCellCount == 0 {
955d3add9426 More code refactoring; split takuzu.go and create validate.go
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    29
			prevCellCount = 1
955d3add9426 More code refactoring; split takuzu.go and create validate.go
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    30
		} else {
955d3add9426 More code refactoring; split takuzu.go and create validate.go
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    31
			if c.Value == prevCell.Value {
955d3add9426 More code refactoring; split takuzu.go and create validate.go
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    32
				prevCellCount++
955d3add9426 More code refactoring; split takuzu.go and create validate.go
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    33
				if prevCellCount > 2 {
955d3add9426 More code refactoring; split takuzu.go and create validate.go
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    34
					return full, errors.Errorf("3+ same values %d", c.Value)
955d3add9426 More code refactoring; split takuzu.go and create validate.go
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    35
				}
955d3add9426 More code refactoring; split takuzu.go and create validate.go
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    36
			} else {
955d3add9426 More code refactoring; split takuzu.go and create validate.go
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    37
				prevCellCount = 1
955d3add9426 More code refactoring; split takuzu.go and create validate.go
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    38
			}
0
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    39
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    40
		}
8
955d3add9426 More code refactoring; split takuzu.go and create validate.go
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    41
		prevCell = c
0
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    42
	}
8
955d3add9426 More code refactoring; split takuzu.go and create validate.go
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    43
	if counters[0] > size/2 {
955d3add9426 More code refactoring; split takuzu.go and create validate.go
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    44
		return full, errors.Errorf("too many zeroes")
0
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    45
	}
8
955d3add9426 More code refactoring; split takuzu.go and create validate.go
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    46
	if counters[1] > size/2 {
955d3add9426 More code refactoring; split takuzu.go and create validate.go
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    47
		return full, errors.Errorf("too many ones")
0
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    48
	}
8
955d3add9426 More code refactoring; split takuzu.go and create validate.go
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    49
	return full, nil
0
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    50
}
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    51
8
955d3add9426 More code refactoring; split takuzu.go and create validate.go
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    52
// CheckRangeCounts returns true if all cells of the provided range are defined,
955d3add9426 More code refactoring; split takuzu.go and create validate.go
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    53
// as well as the number of 0s and the number of 1s in the range.
955d3add9426 More code refactoring; split takuzu.go and create validate.go
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    54
func CheckRangeCounts(cells []Cell) (full bool, n0, n1 int) {
955d3add9426 More code refactoring; split takuzu.go and create validate.go
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    55
	counters := []int{0, 0}
955d3add9426 More code refactoring; split takuzu.go and create validate.go
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    56
	full = true
2
d41c4f8fe066 Add some checks
Mikael Berthe <mikael@lilotux.net>
parents: 0
diff changeset
    57
8
955d3add9426 More code refactoring; split takuzu.go and create validate.go
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    58
	for _, c := range cells {
955d3add9426 More code refactoring; split takuzu.go and create validate.go
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    59
		if c.Defined {
955d3add9426 More code refactoring; split takuzu.go and create validate.go
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    60
			counters[c.Value]++
955d3add9426 More code refactoring; split takuzu.go and create validate.go
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    61
		} else {
955d3add9426 More code refactoring; split takuzu.go and create validate.go
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    62
			full = false
0
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    63
		}
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    64
	}
8
955d3add9426 More code refactoring; split takuzu.go and create validate.go
Mikael Berthe <mikael@lilotux.net>
parents: 7
diff changeset
    65
	return full, counters[0], counters[1]
0
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    66
}
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    67
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    68
// CheckLine returns an error if the line i fails validation
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    69
func (b Takuzu) CheckLine(i int) error {
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    70
	_, err := checkRange(b.GetLine(i))
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    71
	return err
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    72
}
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    73
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    74
// CheckColumn returns an error if the column i fails validation
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    75
func (b Takuzu) CheckColumn(i int) error {
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    76
	_, err := checkRange(b.GetColumn(i))
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    77
	return err
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    78
}
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    79
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    80
// Validate checks a whole board for errors (not completeness)
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    81
// Returns true if all cells are defined.
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    82
func (b Takuzu) Validate() (bool, error) {
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    83
	finished := true
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    84
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    85
	computeVal := func(cells []Cell) (val int) {
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    86
		for i := 0; i < len(cells); i++ {
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    87
			val += cells[i].Value * 1 << uint(i)
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    88
		}
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    89
		return
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    90
	}
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    91
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    92
	lineVals := make(map[int]bool)
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    93
	colVals := make(map[int]bool)
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    94
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    95
	for i := 0; i < b.Size; i++ {
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    96
		var d []Cell
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    97
		var full bool
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    98
		var err error
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    99
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
   100
		// Let's check line i
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
   101
		d = b.GetLine(i)
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
   102
		full, err = checkRange(d)
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
   103
		if err != nil {
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
   104
			return false, errors.Wrapf(err, "line %d", i)
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
   105
		}
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
   106
		if full {
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
   107
			hv := computeVal(d)
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
   108
			if lineVals[hv] {
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
   109
				return false, fmt.Errorf("duplicate lines (%d)", i)
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
   110
			}
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
   111
			lineVals[hv] = true
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
   112
		} else {
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
   113
			finished = false
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
   114
		}
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
   115
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
   116
		// Let's check column i
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
   117
		d = b.GetColumn(i)
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
   118
		full, err = checkRange(d)
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
   119
		if err != nil {
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
   120
			return false, errors.Wrapf(err, "column %d", i)
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
   121
		}
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
   122
		if full {
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
   123
			hv := computeVal(d)
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
   124
			if colVals[hv] {
3
8415b6bd06e9 Fix typo
Mikael Berthe <mikael@lilotux.net>
parents: 2
diff changeset
   125
				return false, fmt.Errorf("duplicate columns (%d)", i)
0
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
   126
			}
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
   127
			colVals[hv] = true
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
   128
		} else {
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
   129
			finished = false
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
   130
		}
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
   131
	}
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
   132
	return finished, nil
00371339bbcc Import refactored version
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
   133
}