added datamatrix support

datamatrix/codelayout.go

@@ -0,0 +1,213 @@
+package datamatrix
+
+import (
+	"github.com/boombuler/barcode"
+)
+
+type setValFunc func(byte)
+
+type codeLayout struct {
+	matrix *barcode.BitList
+	occupy *barcode.BitList
+	size   *dmCodeSize
+}
+
+func newCodeLayout(size *dmCodeSize) *codeLayout {
+	result := new(codeLayout)
+	result.matrix = barcode.NewBitList(size.MatrixColumns() * size.MatrixRows())
+	result.occupy = barcode.NewBitList(size.MatrixColumns() * size.MatrixRows())
+	result.size = size
+	return result
+}
+
+func (l *codeLayout) Occupied(row, col int) bool {
+	return l.occupy.GetBit(col + row*l.size.MatrixColumns())
+}
+
+func (l *codeLayout) Set(row, col int, value, bitNum byte, occupy bool) {
+	val := ((value >> (7 - bitNum)) & 1) == 1
+	if row < 0 {
+		row += l.size.MatrixRows()
+		col += 4 - ((l.size.MatrixRows() + 4) % 8)
+	}
+	if col < 0 {
+		col += l.size.MatrixColumns()
+		row += 4 - ((l.size.MatrixColumns() + 4) % 8)
+	}
+	l.matrix.SetBit(col+row*l.size.MatrixColumns(), val)
+	if occupy {
+		l.occupy.SetBit(col+row*l.size.MatrixColumns(), true)
+	}
+}
+
+func (l *codeLayout) SetSimple(row, col int, value byte) {
+	l.Set(row-2, col-2, value, 0, true)
+	l.Set(row-2, col-1, value, 1, true)
+	l.Set(row-1, col-2, value, 2, true)
+	l.Set(row-1, col-1, value, 3, true)
+	l.Set(row-1, col-0, value, 4, true)
+	l.Set(row-0, col-2, value, 5, true)
+	l.Set(row-0, col-1, value, 6, true)
+	l.Set(row-0, col-0, value, 7, true)
+}
+
+func (l *codeLayout) Corner1(value byte) {
+	l.Set(l.size.MatrixRows()-1, 0, value, 0, true)
+	l.Set(l.size.MatrixRows()-1, 1, value, 1, true)
+	l.Set(l.size.MatrixRows()-1, 2, value, 2, true)
+	l.Set(0, l.size.MatrixColumns()-2, value, 3, true)
+	l.Set(0, l.size.MatrixColumns()-1, value, 4, true)
+	l.Set(1, l.size.MatrixColumns()-1, value, 5, true)
+	l.Set(2, l.size.MatrixColumns()-1, value, 6, true)
+	l.Set(3, l.size.MatrixColumns()-1, value, 7, true)
+}
+
+func (l *codeLayout) Corner2(value byte) {
+	l.Set(l.size.MatrixRows()-3, 0, value, 0, true)
+	l.Set(l.size.MatrixRows()-2, 0, value, 1, true)
+	l.Set(l.size.MatrixRows()-1, 0, value, 2, true)
+	l.Set(0, l.size.MatrixColumns()-4, value, 3, true)
+	l.Set(0, l.size.MatrixColumns()-3, value, 4, true)
+	l.Set(0, l.size.MatrixColumns()-2, value, 5, true)
+	l.Set(0, l.size.MatrixColumns()-1, value, 6, true)
+	l.Set(1, l.size.MatrixColumns()-1, value, 7, true)
+}
+
+func (l *codeLayout) Corner3(value byte) {
+	l.Set(l.size.MatrixRows()-3, 0, value, 0, true)
+	l.Set(l.size.MatrixRows()-2, 0, value, 1, true)
+	l.Set(l.size.MatrixRows()-1, 0, value, 2, true)
+	l.Set(0, l.size.MatrixColumns()-2, value, 3, true)
+	l.Set(0, l.size.MatrixColumns()-1, value, 4, true)
+	l.Set(1, l.size.MatrixColumns()-1, value, 5, true)
+	l.Set(2, l.size.MatrixColumns()-1, value, 6, true)
+	l.Set(3, l.size.MatrixColumns()-1, value, 7, true)
+}
+
+func (l *codeLayout) Corner4(value byte) {
+	l.Set(l.size.MatrixRows()-1, 0, value, 0, true)
+	l.Set(l.size.MatrixRows()-1, l.size.MatrixColumns()-1, value, 1, true)
+	l.Set(0, l.size.MatrixColumns()-3, value, 2, true)
+	l.Set(0, l.size.MatrixColumns()-2, value, 3, true)
+	l.Set(0, l.size.MatrixColumns()-1, value, 4, true)
+	l.Set(1, l.size.MatrixColumns()-3, value, 5, true)
+	l.Set(1, l.size.MatrixColumns()-2, value, 6, true)
+	l.Set(1, l.size.MatrixColumns()-1, value, 7, true)
+}
+
+func (l *codeLayout) ItterateSetter() <-chan setValFunc {
+	result := make(chan setValFunc)
+	go func() {
+		row := 4
+		col := 0
+
+		for (row < l.size.MatrixRows()) || (col < l.size.MatrixColumns()) {
+			if (row == l.size.MatrixRows()) && (col == 0) {
+				result <- l.Corner1
+			}
+			if (row == l.size.MatrixRows()-2) && (col == 0) && (l.size.MatrixColumns()%4 != 0) {
+				result <- l.Corner2
+			}
+			if (row == l.size.MatrixRows()-2) && (col == 0) && (l.size.MatrixColumns()%8 == 4) {
+				result <- l.Corner3
+			}
+
+			if (row == l.size.MatrixRows()+4) && (col == 2) && (l.size.MatrixColumns()%8 == 0) {
+				result <- l.Corner4
+			}
+
+			for true {
+				if (row < l.size.MatrixRows()) && (col >= 0) && !l.Occupied(row, col) {
+					r := row
+					c := col
+					result <- func(b byte) {
+						l.SetSimple(r, c, b)
+					}
+				}
+				row -= 2
+				col += 2
+				if (row < 0) || (col >= l.size.MatrixColumns()) {
+					break
+				}
+			}
+
+			row += 1
+			col += 3
+
+			for true {
+				if (row >= 0) && (col < l.size.MatrixColumns()) && !l.Occupied(row, col) {
+					r := row
+					c := col
+					result <- func(b byte) {
+						l.SetSimple(r, c, b)
+					}
+				}
+				row += 2
+				col -= 2
+				if (row >= l.size.MatrixRows()) || (col < 0) {
+					break
+				}
+			}
+			row += 3
+			col += 1
+
+		}
+		close(result)
+	}()
+
+	return result
+}
+
+func (l *codeLayout) Merge() *datamatrixCode {
+	result := newDataMatrixCode(l.size)
+
+	//dotted horizontal lines
+	for r := 0; r < l.size.Rows; r += (l.size.RegionRows() + 2) {
+		for c := 0; c < l.size.Columns; c += 2 {
+			result.set(c, r, true)
+		}
+	}
+
+	//solid horizontal line
+	for r := l.size.RegionRows() + 1; r < l.size.Rows; r += (l.size.RegionRows() + 2) {
+		for c := 0; c < l.size.Columns; c++ {
+			result.set(c, r, true)
+		}
+	}
+
+	//dotted vertical lines
+	for c := l.size.RegionColumns() + 1; c < l.size.Columns; c += (l.size.RegionColumns() + 2) {
+		for r := 1; r < l.size.Rows; r += 2 {
+			result.set(c, r, true)
+		}
+	}
+
+	//solid vertical line
+	for c := 0; c < l.size.Columns; c += (l.size.RegionColumns() + 2) {
+		for r := 0; r < l.size.Rows; r++ {
+			result.set(c, r, true)
+		}
+	}
+	count := 0
+	for hRegion := 0; hRegion < l.size.RegionCountHorizontal; hRegion++ {
+		for vRegion := 0; vRegion < l.size.RegionCountVertical; vRegion++ {
+			for x := 0; x < l.size.RegionColumns(); x++ {
+				colMatrix := (l.size.RegionColumns() * hRegion) + x
+				colResult := ((2 + l.size.RegionColumns()) * hRegion) + x + 1
+
+				for y := 0; y < l.size.RegionRows(); y++ {
+					rowMatrix := (l.size.RegionRows() * vRegion) + y
+					rowResult := ((2 + l.size.RegionRows()) * vRegion) + y + 1
+					val := l.matrix.GetBit(colMatrix + rowMatrix*l.size.MatrixColumns())
+					if val {
+						count++
+					}
+
+					result.set(colResult, rowResult, val)
+				}
+			}
+		}
+	}
+
+	return result
+}

datamatrix/codesize.go

@@ -0,0 +1,65 @@
+package datamatrix
+
+type dmCodeSize struct {
+	Rows                  int
+	Columns               int
+	RegionCountHorizontal int
+	RegionCountVertical   int
+	ECCCount              int
+	BlockCount            int
+}
+
+func (s *dmCodeSize) RegionRows() int {
+	return (s.Rows - (s.RegionCountHorizontal * 2)) / s.RegionCountHorizontal
+}
+
+func (s *dmCodeSize) RegionColumns() int {
+	return (s.Columns - (s.RegionCountVertical * 2)) / s.RegionCountVertical
+}
+
+func (s *dmCodeSize) MatrixRows() int {
+	return s.RegionRows() * s.RegionCountHorizontal
+}
+
+func (s *dmCodeSize) MatrixColumns() int {
+	return s.RegionColumns() * s.RegionCountVertical
+}
+
+func (s *dmCodeSize) DataCodewords() int {
+	return ((s.MatrixColumns() * s.MatrixRows()) / 8) - s.ECCCount
+}
+
+func (s *dmCodeSize) DataCodewordsPerBlock() int {
+	return s.DataCodewords() / s.BlockCount
+}
+
+func (s *dmCodeSize) ErrorCorrectionCodewordsPerBlock() int {
+	return s.ECCCount / s.BlockCount
+}
+
+var codeSizes []*dmCodeSize = []*dmCodeSize{
+	&dmCodeSize{10, 10, 1, 1, 5, 1},
+	&dmCodeSize{12, 12, 1, 1, 7, 1},
+	&dmCodeSize{14, 14, 1, 1, 10, 1},
+	&dmCodeSize{16, 16, 1, 1, 12, 1},
+	&dmCodeSize{18, 18, 1, 1, 14, 1},
+	&dmCodeSize{20, 20, 1, 1, 18, 1},
+	&dmCodeSize{22, 22, 1, 1, 20, 1},
+	&dmCodeSize{24, 24, 1, 1, 24, 1},
+	&dmCodeSize{26, 26, 1, 1, 28, 1},
+	&dmCodeSize{32, 32, 2, 2, 36, 1},
+	&dmCodeSize{36, 36, 2, 2, 42, 1},
+	&dmCodeSize{40, 40, 2, 2, 48, 1},
+	&dmCodeSize{44, 44, 2, 2, 56, 1},
+	&dmCodeSize{48, 48, 2, 2, 68, 1},
+	&dmCodeSize{52, 52, 2, 2, 84, 2},
+	&dmCodeSize{64, 64, 4, 4, 112, 2},
+	&dmCodeSize{72, 72, 4, 4, 144, 4},
+	&dmCodeSize{80, 80, 4, 4, 192, 4},
+	&dmCodeSize{88, 88, 4, 4, 224, 4},
+	&dmCodeSize{96, 96, 4, 4, 272, 4},
+	&dmCodeSize{104, 104, 4, 4, 336, 6},
+	&dmCodeSize{120, 120, 6, 6, 408, 6},
+	&dmCodeSize{132, 132, 6, 6, 496, 8},
+	&dmCodeSize{144, 144, 6, 6, 620, 10},
+}

datamatrix/datamatrixcode.go

@@ -0,0 +1,48 @@
+package datamatrix
+
+import (
+	"github.com/boombuler/barcode"
+	"image"
+	"image/color"
+)
+
+type datamatrixCode struct {
+	*barcode.BitList
+	*dmCodeSize
+	content string
+}
+
+func newDataMatrixCode(size *dmCodeSize) *datamatrixCode {
+	return &datamatrixCode{barcode.NewBitList(size.Rows * size.Columns), size, ""}
+}
+
+func (c *datamatrixCode) Content() string {
+	return c.content
+}
+
+func (c *datamatrixCode) Metadata() barcode.Metadata {
+	return barcode.Metadata{"DataMatrix", 2}
+}
+
+func (c *datamatrixCode) ColorModel() color.Model {
+	return color.Gray16Model
+}
+
+func (c *datamatrixCode) Bounds() image.Rectangle {
+	return image.Rect(0, 0, c.Columns, c.Rows)
+}
+
+func (c *datamatrixCode) At(x, y int) color.Color {
+	if c.get(x, y) {
+		return color.Black
+	}
+	return color.White
+}
+
+func (c *datamatrixCode) get(x, y int) bool {
+	return c.GetBit(x*c.Rows + y)
+}
+
+func (c *datamatrixCode) set(x, y int, value bool) {
+	c.SetBit(x*c.Rows+y, value)
+}

datamatrix/encoder.go

@@ -0,0 +1,75 @@
+package datamatrix
+
+import (
+	"errors"
+	"github.com/boombuler/barcode"
+)
+
+// Encodes the given content as a DataMatrix code
+func Encode(content string) (barcode.Barcode, error) {
+	data := encodeText(content)
+
+	var size *dmCodeSize = nil
+	for _, s := range codeSizes {
+		if s.DataCodewords() >= len(data) {
+			size = s
+			break
+		}
+	}
+	if size == nil {
+		return nil, errors.New("to much data to encode")
+	}
+	data = addPadding(data, size.DataCodewords())
+	data = generateECC(data, size)
+	code := render(data, size)
+	if code != nil {
+		code.content = content
+		return code, nil
+	}
+	return nil, errors.New("unable to render barcode")
+}
+
+func render(data []byte, size *dmCodeSize) *datamatrixCode {
+	cl := newCodeLayout(size)
+
+	setters := cl.ItterateSetter()
+	for _, b := range data {
+		(<-setters)(b)
+	}
+	return cl.Merge()
+}
+
+func encodeText(content string) []byte {
+	result := make([]byte, 0)
+	input := []byte(content)
+
+	for i := 0; i < len(input); {
+		c := input[i]
+		i++
+
+		if c >= '0' && c <= '9' && i < len(input) && input[i] >= '0' && input[i] <= '9' {
+			// two numbers...
+			c2 := input[i]
+			i++
+			cw := byte(((c-'0')*10 + (c2 - '0')) + 130)
+			result = append(result, cw)
+		} else if c > 127 {
+			// not correct... needs to be redone later...
+			result = append(result, 235, c-127)
+		} else {
+			result = append(result, c+1)
+		}
+	}
+	return result
+}
+
+func addPadding(data []byte, toCount int) []byte {
+	if len(data) < toCount {
+		data = append(data, 129)
+	}
+	for len(data) < toCount {
+		R := ((149 * (len(data) + 1)) % 253) + 1
+		data = append(data, byte((129+R)%254))
+	}
+	return data
+}

datamatrix/errorcorrection.go

@@ -0,0 +1,105 @@
+package datamatrix
+
+// a more generic way would be great!
+// like qr.galoisField with 301
+
+var gf_log []int = []int{
+	0, 255, 1, 240, 2, 225, 241, 53, 3, 38, 226, 133, 242, 43, 54, 210,
+	4, 195, 39, 114, 227, 106, 134, 28, 243, 140, 44, 23, 55, 118, 211, 234,
+	5, 219, 196, 96, 40, 222, 115, 103, 228, 78, 107, 125, 135, 8, 29, 162,
+	244, 186, 141, 180, 45, 99, 24, 49, 56, 13, 119, 153, 212, 199, 235, 91,
+	6, 76, 220, 217, 197, 11, 97, 184, 41, 36, 223, 253, 116, 138, 104, 193,
+	229, 86, 79, 171, 108, 165, 126, 145, 136, 34, 9, 74, 30, 32, 163, 84,
+	245, 173, 187, 204, 142, 81, 181, 190, 46, 88, 100, 159, 25, 231, 50, 207,
+	57, 147, 14, 67, 120, 128, 154, 248, 213, 167, 200, 63, 236, 110, 92, 176,
+	7, 161, 77, 124, 221, 102, 218, 95, 198, 90, 12, 152, 98, 48, 185, 179,
+	42, 209, 37, 132, 224, 52, 254, 239, 117, 233, 139, 22, 105, 27, 194, 113,
+	230, 206, 87, 158, 80, 189, 172, 203, 109, 175, 166, 62, 127, 247, 146, 66,
+	137, 192, 35, 252, 10, 183, 75, 216, 31, 83, 33, 73, 164, 144, 85, 170,
+	246, 65, 174, 61, 188, 202, 205, 157, 143, 169, 82, 72, 182, 215, 191, 251,
+	47, 178, 89, 151, 101, 94, 160, 123, 26, 112, 232, 21, 51, 238, 208, 131,
+	58, 69, 148, 18, 15, 16, 68, 17, 121, 149, 129, 19, 155, 59, 249, 70,
+	214, 250, 168, 71, 201, 156, 64, 60, 237, 130, 111, 20, 93, 122, 177, 150,
+}
+
+var gf_alog []int = []int{
+	1, 2, 4, 8, 16, 32, 64, 128, 45, 90, 180, 69, 138, 57, 114, 228,
+	229, 231, 227, 235, 251, 219, 155, 27, 54, 108, 216, 157, 23, 46, 92, 184,
+	93, 186, 89, 178, 73, 146, 9, 18, 36, 72, 144, 13, 26, 52, 104, 208,
+	141, 55, 110, 220, 149, 7, 14, 28, 56, 112, 224, 237, 247, 195, 171, 123,
+	246, 193, 175, 115, 230, 225, 239, 243, 203, 187, 91, 182, 65, 130, 41, 82,
+	164, 101, 202, 185, 95, 190, 81, 162, 105, 210, 137, 63, 126, 252, 213, 135,
+	35, 70, 140, 53, 106, 212, 133, 39, 78, 156, 21, 42, 84, 168, 125, 250,
+	217, 159, 19, 38, 76, 152, 29, 58, 116, 232, 253, 215, 131, 43, 86, 172,
+	117, 234, 249, 223, 147, 11, 22, 44, 88, 176, 77, 154, 25, 50, 100, 200,
+	189, 87, 174, 113, 226, 233, 255, 211, 139, 59, 118, 236, 245, 199, 163, 107,
+	214, 129, 47, 94, 188, 85, 170, 121, 242, 201, 191, 83, 166, 97, 194, 169,
+	127, 254, 209, 143, 51, 102, 204, 181, 71, 142, 49, 98, 196, 165, 103, 206,
+	177, 79, 158, 17, 34, 68, 136, 61, 122, 244, 197, 167, 99, 198, 161, 111,
+	222, 145, 15, 30, 60, 120, 240, 205, 183, 67, 134, 33, 66, 132, 37, 74,
+	148, 5, 10, 20, 40, 80, 160, 109, 218, 153, 31, 62, 124, 248, 221, 151,
+	3, 6, 12, 24, 48, 96, 192, 173, 119, 238, 241, 207, 179, 75, 150, 1,
+}
+
+var gf_polys map[int][]int = map[int][]int{
+	5:  []int{228, 48, 15, 111, 62},
+	7:  []int{23, 68, 144, 134, 240, 92, 254},
+	10: []int{28, 24, 185, 166, 223, 248, 116, 255, 110, 61},
+	11: []int{175, 138, 205, 12, 194, 168, 39, 245, 60, 97, 120},
+	12: []int{41, 153, 158, 91, 61, 42, 142, 213, 97, 178, 100, 242},
+	14: []int{156, 97, 192, 252, 95, 9, 157, 119, 138, 45, 18, 186, 83, 185},
+	18: []int{83, 195, 100, 39, 188, 75, 66, 61, 241, 213, 109, 129, 94, 254, 225, 48, 90, 188},
+	20: []int{15, 195, 244, 9, 233, 71, 168, 2, 188, 160, 153, 145, 253, 79, 108, 82, 27, 174, 186, 172},
+	24: []int{52, 190, 88, 205, 109, 39, 176, 21, 155, 197, 251, 223, 155, 21, 5, 172, 254, 124, 12, 181, 184, 96, 50, 193},
+	28: []int{211, 231, 43, 97, 71, 96, 103, 174, 37, 151, 170, 53, 75, 34, 249, 121, 17, 138, 110, 213, 141, 136, 120, 151, 233, 168, 93, 255},
+	36: []int{245, 127, 242, 218, 130, 250, 162, 181, 102, 120, 84, 179, 220, 251, 80, 182, 229, 18, 2, 4, 68, 33, 101, 137, 95, 119, 115, 44, 175, 184, 59, 25, 225, 98, 81, 112},
+	42: []int{77, 193, 137, 31, 19, 38, 22, 153, 247, 105, 122, 2, 245, 133, 242, 8, 175, 95, 100, 9, 167, 105, 214, 111, 57, 121, 21, 1, 253, 57, 54, 101, 248, 202, 69, 50, 150, 177, 226, 5, 9, 5},
+	48: []int{245, 132, 172, 223, 96, 32, 117, 22, 238, 133, 238, 231, 205, 188, 237, 87, 191, 106, 16, 147, 118, 23, 37, 90, 170, 205, 131, 88, 120, 100, 66, 138, 186, 240, 82, 44, 176, 87, 187, 147, 160, 175, 69, 213, 92, 253, 225, 19},
+	56: []int{175, 9, 223, 238, 12, 17, 220, 208, 100, 29, 175, 170, 230, 192, 215, 235, 150, 159, 36, 223, 38, 200, 132, 54, 228, 146, 218, 234, 117, 203, 29, 232, 144, 238, 22, 150, 201, 117, 62, 207, 164, 13, 137, 245, 127, 67, 247, 28, 155, 43, 203, 107, 233, 53, 143, 46},
+	62: []int{242, 93, 169, 50, 144, 210, 39, 118, 202, 188, 201, 189, 143, 108, 196, 37, 185, 112, 134, 230, 245, 63, 197, 190, 250, 106, 185, 221, 175, 64, 114, 71, 161, 44, 147, 6, 27, 218, 51, 63, 87, 10, 40, 130, 188, 17, 163, 31, 176, 170, 4, 107, 232, 7, 94, 166, 224, 124, 86, 47, 11, 204},
+	68: []int{220, 228, 173, 89, 251, 149, 159, 56, 89, 33, 147, 244, 154, 36, 73, 127, 213, 136, 248, 180, 234, 197, 158, 177, 68, 122, 93, 213, 15, 160, 227, 236, 66, 139, 153, 185, 202, 167, 179, 25, 220, 232, 96, 210, 231, 136, 223, 239, 181, 241, 59, 52, 172, 25, 49, 232, 211, 189, 64, 54, 108, 153, 132, 63, 96, 103, 82, 186},
+}
+
+func rsBlock(data []byte, ncout []byte, poly []int) {
+	for i := 0; i < len(ncout); i++ {
+		ncout[i] = 0
+	}
+
+	for i := 0; i < len(data); i++ {
+		k := (ncout[0] ^ data[i]) & 0xff
+		for j := 0; j < len(ncout)-1; j++ {
+			k1 := 0
+			if k != 0 {
+				k1 = gf_alog[(gf_log[k]+gf_log[poly[len(ncout)-j-2]])%255]
+			}
+			ncout[j] = ncout[j+1] ^ byte(k1)
+		}
+	}
+}
+
+func generateECC(data []byte, size *dmCodeSize) []byte {
+	buff := make([]byte, size.DataCodewordsPerBlock())
+	ecc := make([]byte, size.ErrorCorrectionCodewordsPerBlock()+1)
+	poly := gf_polys[size.ErrorCorrectionCodewordsPerBlock()]
+
+	dataSize := len(data)
+	// make some space...
+	space := make([]byte, size.ECCCount)
+	data = append(data, space...)
+
+	for b := 0; b < size.BlockCount; b++ {
+		p := 0
+		for n := b; n < dataSize; n += size.BlockCount {
+			buff[p] = data[n]
+			p++
+		}
+		rsBlock(buff, ecc, poly)
+		p = 0
+		for n := b; n < size.ErrorCorrectionCodewordsPerBlock()*size.BlockCount; n += size.BlockCount {
+			data[dataSize+n] = ecc[p]
+			p++
+		}
+	}
+
+	return data
+}