Mercurial > gemma
view pkg/octree/contours.go @ 1625:de4e4dcb8f87
staging area: implemented details and zoom to bbox
author | Thomas Junk <thomas.junk@intevation.de> |
---|---|
date | Wed, 19 Dec 2018 12:52:33 +0100 |
parents | 5443f5c9154c |
children | f4dcbe8941a1 |
line wrap: on
line source
// This is Free Software under GNU Affero General Public License v >= 3.0 // without warranty, see README.md and license for details. // // SPDX-License-Identifier: AGPL-3.0-or-later // License-Filename: LICENSES/AGPL-3.0.txt // // Copyright (C) 2018 by via donau // – Österreichische Wasserstraßen-Gesellschaft mbH // Software engineering by Intevation GmbH // // Author(s): // * Sascha L. Teichmann <sascha.teichmann@intevation.de> // * Tom Gottfried <tom.gottfried@intevation.de> package octree import ( "runtime" "sync" ) type ContourResult struct { Height float64 Lines MultiLineStringZ done bool mu sync.Mutex cond *sync.Cond } func NewContourResult(height float64) *ContourResult { cr := ContourResult{Height: height} cr.cond = sync.NewCond(&cr.mu) return &cr } func (cr *ContourResult) wait() { cr.cond.L.Lock() for !cr.done { cr.cond.Wait() } cr.cond.L.Unlock() } func (cr *ContourResult) get() float64 { cr.cond.L.Lock() defer cr.cond.L.Unlock() return cr.Height } func (cr *ContourResult) set(lines MultiLineStringZ) { cr.cond.L.Lock() defer cr.cond.L.Unlock() cr.Lines = lines cr.done = true cr.cond.Signal() } func DoContours(tree *Tree, heights []float64, store func(*ContourResult)) { contours := make([]*ContourResult, len(heights)) for i, h := range heights { contours[i] = NewContourResult(h) } jobs := make(chan *ContourResult) var wg sync.WaitGroup for i, n := 0, runtime.NumCPU(); i < n; i++ { wg.Add(1) go processLevels(tree, jobs, &wg) } done := make(chan struct{}) go func() { defer close(done) for _, cr := range contours { cr.wait() store(cr) } }() for _, cr := range contours { jobs <- cr } close(jobs) wg.Wait() <-done } func processLevels( tree *Tree, jobs <-chan *ContourResult, wg *sync.WaitGroup, ) { defer wg.Done() for cr := range jobs { var lines MultiLineStringZ h := cr.get() tree.Horizontal(h, func(t *Triangle) { line := t.IntersectHorizontal(h) if len(line) > 1 { lines = append(lines, line) } }) cr.set(lines.Merge()) } }