Method Render

tensor/tensorcore/tensorgrid.go:305–413

()

}

func (tg *TensorGrid) Render() {
	if tg.Tensor == nil || tg.Tensor.Len() == 0 {
		return
	}
	tg.EnsureColorMap()
	tg.UpdateRange()

	pc := &tg.Scene.PaintContext

	pos := tg.Geom.Pos.Content
	sz := tg.Geom.Size.Actual.Content
	// sz.SetSubScalar(tg.Disp.BotRtSpace.Dots)

	pc.FillBox(pos, sz, tg.Styles.Background)

	tsr := tg.Tensor

	if tg.Display.Image {
		ysz := tsr.DimSize(0)
		xsz := tsr.DimSize(1)
		nclr := 1
		outclr := false // outer dimension is color
		if tsr.NumDims() == 3 {
			if tsr.DimSize(0) == 3 || tsr.DimSize(0) == 4 {
				outclr = true
				ysz = tsr.DimSize(1)
				xsz = tsr.DimSize(2)
				nclr = tsr.DimSize(0)
			} else {
				nclr = tsr.DimSize(2)
			}
		}
		tsz := math32.Vec2(float32(xsz), float32(ysz))
		gsz := sz.Div(tsz)
		for y := 0; y < ysz; y++ {
			for x := 0; x < xsz; x++ {
				ey := y
				if !tg.Display.TopZero {
					ey = (ysz - 1) - y
				}
				switch {
				case outclr:
					var r, g, b, a float64
					a = 1
					r = tg.Display.Range.ClipNormValue(tsr.Float([]int{0, y, x}))
					g = tg.Display.Range.ClipNormValue(tsr.Float([]int{1, y, x}))
					b = tg.Display.Range.ClipNormValue(tsr.Float([]int{2, y, x}))
					if nclr > 3 {
						a = tg.Display.Range.ClipNormValue(tsr.Float([]int{3, y, x}))
					}
					cr := math32.Vec2(float32(x), float32(ey))
					pr := pos.Add(cr.Mul(gsz))
					pc.StrokeStyle.Color = colors.Uniform(colors.FromFloat64(r, g, b, a))
					pc.FillBox(pr, gsz, pc.StrokeStyle.Color)
				case nclr > 1:
					var r, g, b, a float64
					a = 1
					r = tg.Display.Range.ClipNormValue(tsr.Float([]int{y, x, 0}))