2012-12-26 17:07:30 +04:00
|
|
|
package main
|
|
|
|
|
|
|
|
import (
|
2013-05-02 20:45:33 +04:00
|
|
|
"flag"
|
2012-12-26 17:07:30 +04:00
|
|
|
"fmt"
|
2013-04-08 23:45:13 +04:00
|
|
|
"goposm/cache"
|
|
|
|
"goposm/element"
|
2013-02-12 22:45:49 +04:00
|
|
|
"goposm/parser"
|
2013-05-06 13:03:52 +04:00
|
|
|
"goposm/stats"
|
2013-04-16 23:14:19 +04:00
|
|
|
"log"
|
2013-05-02 22:37:31 +04:00
|
|
|
"os"
|
2013-04-08 23:45:13 +04:00
|
|
|
"runtime"
|
2013-05-02 22:37:31 +04:00
|
|
|
"runtime/pprof"
|
2013-04-08 23:45:13 +04:00
|
|
|
"sync"
|
2012-12-26 17:07:30 +04:00
|
|
|
)
|
|
|
|
|
2013-05-06 13:03:52 +04:00
|
|
|
func parse(cache *cache.OSMCache, progress *stats.Statistics, filename string) {
|
2013-04-11 21:50:20 +04:00
|
|
|
nodes := make(chan []element.Node)
|
2013-05-02 20:45:33 +04:00
|
|
|
coords := make(chan []element.Node)
|
2013-04-11 21:50:20 +04:00
|
|
|
ways := make(chan []element.Way)
|
2013-04-20 18:50:23 +04:00
|
|
|
relations := make(chan []element.Relation)
|
2013-04-08 23:45:13 +04:00
|
|
|
|
|
|
|
positions := parser.PBFBlockPositions(filename)
|
|
|
|
|
|
|
|
waitParser := sync.WaitGroup{}
|
2013-04-20 18:50:23 +04:00
|
|
|
for i := 0; i < runtime.NumCPU(); i++ {
|
2013-04-08 23:45:13 +04:00
|
|
|
waitParser.Add(1)
|
|
|
|
go func() {
|
|
|
|
for pos := range positions {
|
2013-05-05 22:14:39 +04:00
|
|
|
parser.ParseBlock(
|
|
|
|
pos,
|
|
|
|
coords,
|
|
|
|
nodes,
|
|
|
|
ways,
|
|
|
|
relations,
|
|
|
|
)
|
2013-04-08 23:45:13 +04:00
|
|
|
}
|
2013-05-05 22:14:39 +04:00
|
|
|
//runtime.GC()
|
2013-04-08 23:45:13 +04:00
|
|
|
waitParser.Done()
|
|
|
|
}()
|
|
|
|
}
|
|
|
|
|
|
|
|
waitCounter := sync.WaitGroup{}
|
2013-05-04 18:27:05 +04:00
|
|
|
|
2013-04-20 18:50:23 +04:00
|
|
|
for i := 0; i < runtime.NumCPU(); i++ {
|
2013-04-11 21:50:20 +04:00
|
|
|
waitCounter.Add(1)
|
|
|
|
go func() {
|
|
|
|
for ws := range ways {
|
2013-05-04 18:27:05 +04:00
|
|
|
cache.Ways.PutWays(ws)
|
2013-05-06 13:03:52 +04:00
|
|
|
progress.AddWays(len(ws))
|
2013-04-11 21:50:20 +04:00
|
|
|
}
|
|
|
|
waitCounter.Done()
|
|
|
|
}()
|
|
|
|
}
|
2013-04-20 18:50:23 +04:00
|
|
|
for i := 0; i < runtime.NumCPU(); i++ {
|
|
|
|
waitCounter.Add(1)
|
|
|
|
go func() {
|
|
|
|
for rels := range relations {
|
2013-05-04 18:27:05 +04:00
|
|
|
cache.Relations.PutRelations(rels)
|
2013-05-06 13:03:52 +04:00
|
|
|
progress.AddRelations(len(rels))
|
2013-04-20 18:50:23 +04:00
|
|
|
}
|
|
|
|
waitCounter.Done()
|
|
|
|
}()
|
|
|
|
}
|
|
|
|
for i := 0; i < runtime.NumCPU(); i++ {
|
2013-04-08 23:45:13 +04:00
|
|
|
waitCounter.Add(1)
|
|
|
|
go func() {
|
2013-05-02 20:45:33 +04:00
|
|
|
for nds := range coords {
|
2013-05-04 18:27:05 +04:00
|
|
|
cache.Coords.PutCoords(nds)
|
2013-05-06 13:03:52 +04:00
|
|
|
progress.AddCoords(len(nds))
|
2013-04-08 23:45:13 +04:00
|
|
|
}
|
2013-05-02 20:45:33 +04:00
|
|
|
waitCounter.Done()
|
|
|
|
}()
|
|
|
|
}
|
|
|
|
for i := 0; i < 2; i++ {
|
|
|
|
waitCounter.Add(1)
|
|
|
|
go func() {
|
|
|
|
for nds := range nodes {
|
2013-05-04 18:27:05 +04:00
|
|
|
n, _ := cache.Nodes.PutNodes(nds)
|
2013-05-06 13:03:52 +04:00
|
|
|
progress.AddNodes(n)
|
2013-05-02 20:45:33 +04:00
|
|
|
}
|
2013-04-08 23:45:13 +04:00
|
|
|
waitCounter.Done()
|
|
|
|
}()
|
|
|
|
}
|
2013-05-02 20:45:33 +04:00
|
|
|
|
2013-04-08 23:45:13 +04:00
|
|
|
waitParser.Wait()
|
2013-05-02 20:45:33 +04:00
|
|
|
close(coords)
|
2013-04-08 23:45:13 +04:00
|
|
|
close(nodes)
|
|
|
|
close(ways)
|
|
|
|
close(relations)
|
|
|
|
waitCounter.Wait()
|
|
|
|
}
|
|
|
|
|
2013-05-06 13:10:37 +04:00
|
|
|
var (
|
2013-05-06 14:51:30 +04:00
|
|
|
cpuprofile = flag.String("cpuprofile", "", "filename of cpu profile output")
|
|
|
|
cachedir = flag.String("cachedir", "/tmp/goposm", "cache directory")
|
|
|
|
overwritecache = flag.Bool("overwritecache", false, "overwritecache")
|
|
|
|
appendcache = flag.Bool("appendcache", false, "append cache")
|
|
|
|
read = flag.String("read", "", "read")
|
|
|
|
write = flag.Bool("write", false, "write")
|
2013-05-06 13:10:37 +04:00
|
|
|
)
|
|
|
|
|
2012-12-26 17:07:30 +04:00
|
|
|
func main() {
|
2013-05-06 16:20:00 +04:00
|
|
|
log.SetFlags(log.LstdFlags | log.Lshortfile)
|
2013-05-06 13:10:37 +04:00
|
|
|
runtime.GOMAXPROCS(runtime.NumCPU())
|
|
|
|
flag.Parse()
|
|
|
|
|
|
|
|
if *cpuprofile != "" {
|
|
|
|
f, err := os.Create(*cpuprofile)
|
2013-05-05 22:14:39 +04:00
|
|
|
if err != nil {
|
|
|
|
log.Fatal(err)
|
|
|
|
}
|
|
|
|
pprof.StartCPUProfile(f)
|
|
|
|
defer pprof.StopCPUProfile()
|
2013-05-02 22:37:31 +04:00
|
|
|
}
|
2013-05-05 22:14:39 +04:00
|
|
|
|
2013-05-06 14:51:30 +04:00
|
|
|
osmCache := cache.NewOSMCache(*cachedir)
|
|
|
|
|
|
|
|
if *read != "" && osmCache.Exists() {
|
|
|
|
if *overwritecache {
|
|
|
|
log.Println("removing existing cache", *cachedir)
|
|
|
|
err := osmCache.Remove()
|
|
|
|
if err != nil {
|
|
|
|
log.Fatal("unable to remove cache:", err)
|
|
|
|
}
|
|
|
|
} else if !*appendcache {
|
|
|
|
log.Fatal("cache already exists use -appendcache or -overwritecache")
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
err := osmCache.Open()
|
2013-05-02 22:37:31 +04:00
|
|
|
if err != nil {
|
|
|
|
log.Fatal(err)
|
|
|
|
}
|
2013-05-05 22:14:39 +04:00
|
|
|
defer osmCache.Close()
|
2013-05-04 18:27:05 +04:00
|
|
|
|
2013-05-06 13:03:52 +04:00
|
|
|
fmt.Println("start")
|
|
|
|
progress := stats.StatsReporter()
|
|
|
|
|
2013-05-06 14:51:30 +04:00
|
|
|
if *read != "" {
|
|
|
|
parse(osmCache, progress, *read)
|
2013-05-06 16:20:00 +04:00
|
|
|
progress.Message("done reading")
|
2013-05-06 14:51:30 +04:00
|
|
|
}
|
2013-05-02 22:37:31 +04:00
|
|
|
|
2013-05-06 14:51:30 +04:00
|
|
|
if *write {
|
2013-05-06 16:20:00 +04:00
|
|
|
progress.Reset()
|
2013-05-06 14:51:30 +04:00
|
|
|
rel := osmCache.Relations.Iter()
|
2013-05-06 16:20:00 +04:00
|
|
|
for _ = range rel {
|
|
|
|
progress.AddRelations(1)
|
|
|
|
// fmt.Println(r)
|
2013-05-06 14:51:30 +04:00
|
|
|
}
|
2013-05-02 22:37:31 +04:00
|
|
|
|
2013-05-06 14:51:30 +04:00
|
|
|
way := osmCache.Ways.Iter()
|
2013-05-06 16:20:00 +04:00
|
|
|
|
|
|
|
diffCache := cache.NewDiffCache(*cachedir)
|
|
|
|
if err = diffCache.Remove(); err != nil {
|
|
|
|
log.Fatal(err)
|
|
|
|
}
|
|
|
|
if err = diffCache.Open(); err != nil {
|
2013-05-06 14:51:30 +04:00
|
|
|
log.Fatal(err)
|
|
|
|
}
|
2013-05-06 12:21:03 +04:00
|
|
|
|
2013-05-06 14:51:30 +04:00
|
|
|
waitFill := sync.WaitGroup{}
|
|
|
|
for i := 0; i < runtime.NumCPU(); i++ {
|
|
|
|
waitFill.Add(1)
|
2013-05-06 12:21:03 +04:00
|
|
|
|
2013-05-06 14:51:30 +04:00
|
|
|
go func() {
|
|
|
|
for w := range way {
|
2013-05-06 16:20:00 +04:00
|
|
|
progress.AddWays(1)
|
2013-05-06 14:51:30 +04:00
|
|
|
ok := osmCache.Coords.FillWay(w)
|
|
|
|
if !ok {
|
|
|
|
continue
|
|
|
|
}
|
|
|
|
if true {
|
|
|
|
for _, node := range w.Nodes {
|
2013-05-06 16:20:00 +04:00
|
|
|
diffCache.Coords.Add(node.Id, w.Id)
|
2013-05-06 14:51:30 +04:00
|
|
|
}
|
2013-05-06 12:21:03 +04:00
|
|
|
}
|
|
|
|
}
|
2013-05-06 14:51:30 +04:00
|
|
|
waitFill.Done()
|
|
|
|
}()
|
|
|
|
}
|
|
|
|
waitFill.Wait()
|
2013-05-02 22:37:31 +04:00
|
|
|
}
|
2013-05-06 14:51:30 +04:00
|
|
|
|
2013-04-08 23:45:13 +04:00
|
|
|
//parser.PBFStats(os.Args[1])
|
2013-05-06 16:20:00 +04:00
|
|
|
fmt.Println("\ndone")
|
2012-12-26 17:07:30 +04:00
|
|
|
}
|