Method ModifyFloat2AttributeParallelWithPoolSize

modeling/mesh.go:715–755

(atr string, size int, f func(i int, v vector2.Float64) vector2.Float64)

}

func (m Mesh) ModifyFloat2AttributeParallelWithPoolSize(atr string, size int, f func(i int, v vector2.Float64) vector2.Float64) Mesh {
	m.requireV2Attribute(atr)

	if size < 1 {
		panic(fmt.Errorf("unable to modify float2, invalid worker pool size: %d", size))
	}

	if size == 1 {
		return m.ModifyFloat2Attribute(atr, f)
	}

	oldData := m.v2Data[atr]
	modified := make([]vector2.Float64, len(oldData))

	var wg sync.WaitGroup

	workSize := int(math.Floor(float64(len(oldData)) / float64(size)))
	for i := 0; i < size; i++ {
		wg.Add(1)

		jobSize := workSize

		// Make sure to clean up potential last cell due to rounding error of
		// division of number of CPUs
		if i == size-1 {
			jobSize = len(oldData) - (workSize * i)
		}

		go func(start, size int) {
			defer wg.Done()
			end := start + size
			for i := start; i < end; i++ {
				modified[i] = f(i, oldData[i])
			}
		}(workSize*i, jobSize)
	}

	wg.Wait()

	return m.SetFloat2Attribute(atr, modified)
}

func (m Mesh) ModifyFloat1Attribute(atr string, f func(i int, v float64) float64) Mesh {
	m.requireV1Attribute(atr)