OpenDiablo2/d2common/d2fileformats/d2dat/dat_color.go

package d2dat

// DATColor represents a single color in a DAT file.
type DATColor struct {
	r uint8
	g uint8
	b uint8
	a uint8
}

const (
	colorBits = 8
	mask = 0xff
)

const (
	bitShift0 = iota*colorBits
	bitShift8
	bitShift16
	bitShift24
)

// R gets the red component
func (c *DATColor) R() uint8 {
	return c.r
}

// G gets the green component
func (c *DATColor) G() uint8 {
	return c.g
}

// B gets the blue component
func (c *DATColor) B() uint8 {
	return c.b
}

// A gets the alpha component
func (c *DATColor) A() uint8 {
	return mask
}

// RGBA gets the combination of the color components (0xRRGGBBAA)
func (c *DATColor) RGBA() uint32 {
	return toComposite(c.r, c.g, c.b, c.a)
}

// SetRGBA sets the color components using the given RGBA form
func (c *DATColor) SetRGBA(rgba uint32) {
	c.r, c.g, c.b, c.a = toComponent(rgba)
}

func (c *DATColor) BGRA() uint32 {
	return toComposite(c.b, c.g, c.r, c.a)
}

func (c *DATColor) SetBGRA(bgra uint32) {
	c.b, c.g, c.r, c.a = toComponent(bgra)
}

func toComposite (w,x,y,z uint8) uint32 {
	composite := uint32(w)<<bitShift24
	composite += uint32(x)<<bitShift16
	composite += uint32(y)<<bitShift8
	composite += uint32(z)<<bitShift0

	return composite
}

func toComponent (wxyz uint32) (w,x,y,z uint8){
	w = uint8(wxyz>>bitShift24 & mask)
	x = uint8(wxyz>>bitShift16 & mask)
	y = uint8(wxyz>>bitShift8 & mask)
	z = uint8(wxyz>>bitShift0 & mask)
	return w, x, y, z
}