1
0
Fork 0
mirror of https://github.com/Luzifer/gpxhydrant.git synced 2024-12-22 20:21:17 +00:00

Refactored to reduce complexity in single functions

This commit is contained in:
Knut Ahlers 2016-05-07 16:22:26 +02:00
parent 6691b4a053
commit bdbd2857d2
Signed by: luzifer
GPG key ID: DC2729FDD34BE99E
2 changed files with 89 additions and 54 deletions

View file

@ -130,3 +130,7 @@ func (h hydrant) ToNode() *osm.Node {
return out return out
} }
func (h hydrant) NeedsUpdate(in *hydrant) bool {
return h.Diameter != in.Diameter || h.Position != in.Position || h.Pressure != in.Pressure || h.Type != in.Type
}

117
main.go
View file

@ -33,6 +33,23 @@ var (
errWrongGPXComment = errors.New("GPX comment does not match expected format") errWrongGPXComment = errors.New("GPX comment does not match expected format")
) )
type bounds struct{ MinLat, MinLon, MaxLat, MaxLon float64 }
func (b *bounds) Update(lat, lon float64) {
if b.MinLat > lat {
b.MinLat = lat
}
if b.MaxLat < lat {
b.MaxLat = lat
}
if b.MinLon > lon {
b.MinLon = lon
}
if b.MaxLon < lon {
b.MaxLon = lon
}
}
func init() { func init() {
rconfig.Parse(&cfg) rconfig.Parse(&cfg)
@ -56,7 +73,8 @@ func logDebugf(format string, values ...interface{}) {
} }
} }
func main() { func hydrantsFromGPXFile() ([]*hydrant, bounds) {
// Read and parse GPX file
gpsFile, err := os.Open(cfg.GPXFile) gpsFile, err := os.Open(cfg.GPXFile)
if err != nil { if err != nil {
log.Fatalf("Unable to open your GPX file: %s", err) log.Fatalf("Unable to open your GPX file: %s", err)
@ -68,12 +86,9 @@ func main() {
log.Fatalf("Unable to parse your GPX file: %s", err) log.Fatalf("Unable to parse your GPX file: %s", err)
} }
bds := bounds{MinLat: 9999, MinLon: 9999}
hydrants := []*hydrant{} hydrants := []*hydrant{}
var (
minLat = 9999.0
minLon = 9999.0
maxLat, maxLon float64
)
for _, wp := range gpxData.Waypoints { for _, wp := range gpxData.Waypoints {
h, e := parseWaypoint(wp) h, e := parseWaypoint(wp)
if e != nil { if e != nil {
@ -85,39 +100,17 @@ func main() {
logDebugf("Found a hydrant from waypoint %s: %#v", wp.Name, h) logDebugf("Found a hydrant from waypoint %s: %#v", wp.Name, h)
hydrants = append(hydrants, h) hydrants = append(hydrants, h)
if minLat > h.Latitude { bds.Update(h.Latitude, h.Longitude)
minLat = h.Latitude
}
if maxLat < h.Latitude {
maxLat = h.Latitude
}
if minLon > h.Longitude {
minLon = h.Longitude
}
if maxLon < h.Longitude {
maxLon = h.Longitude
}
} }
osmClient, err := osm.New(cfg.OSM.Username, cfg.OSM.Password, cfg.OSM.UseDev) return hydrants, bds
if err != nil { }
log.Fatalf("Unable to log into OSM: %s", err)
}
changeSets, err := osmClient.GetMyChangesets(true) func createChangeset(osmClient *osm.Client) *osm.Changeset {
if err != nil { cs, err := osmClient.CreateChangeset()
log.Fatalf("Unable to get changesets: %s", err)
}
var cs *osm.Changeset
if len(changeSets) > 0 {
cs = changeSets[0]
} else {
cs, err = osmClient.CreateChangeset()
if err != nil { if err != nil {
log.Fatalf("Unable to create changeset: %s", err) log.Fatalf("Unable to create changeset: %s", err)
} }
}
logDebugf("Working on Changeset %d", cs.ID) logDebugf("Working on Changeset %d", cs.ID)
@ -130,8 +123,12 @@ func main() {
log.Fatalf("Unable to save changeset: %s", err) log.Fatalf("Unable to save changeset: %s", err)
} }
return cs
}
func getHydrantsFromOSM(osmClient *osm.Client, bds bounds) []*hydrant {
border := 0.0009 // Equals ~100m using haversine formula border := 0.0009 // Equals ~100m using haversine formula
mapData, err := osmClient.RetrieveMapObjects(minLon-border, minLat-border, maxLon+border, maxLat+border) mapData, err := osmClient.RetrieveMapObjects(bds.MaxLon, bds.MinLat-border, bds.MaxLon+border, bds.MaxLat+border)
if err != nil { if err != nil {
log.Fatalf("Unable to get map data: %s", err) log.Fatalf("Unable to get map data: %s", err)
} }
@ -142,12 +139,34 @@ func main() {
for _, n := range mapData.Nodes { for _, n := range mapData.Nodes {
h, e := fromNode(n) h, e := fromNode(n)
if e != nil { if e != nil {
continue continue // Not a hydrant, ignore that node
} }
availableHydrants = append(availableHydrants, h) availableHydrants = append(availableHydrants, h)
} }
return availableHydrants
}
func main() {
// Convert waypoints from GPX file to hydrants
hydrants, bds := hydrantsFromGPXFile()
osmClient, err := osm.New(cfg.OSM.Username, cfg.OSM.Password, cfg.OSM.UseDev)
if err != nil {
log.Fatalf("Unable to log into OSM: %s", err)
}
// Create a changeset for this import
cs := createChangeset(osmClient)
// Retrieve currently available information from OSM
availableHydrants := getHydrantsFromOSM(osmClient, bds)
updateOrCreateHydrants(hydrants, availableHydrants, cs, osmClient)
}
func updateOrCreateHydrants(hydrants, availableHydrants []*hydrant, cs *osm.Changeset, osmClient *osm.Client) {
for _, h := range hydrants { for _, h := range hydrants {
var found *hydrant var found *hydrant
for _, a := range availableHydrants { for _, a := range availableHydrants {
@ -159,22 +178,24 @@ func main() {
if found == nil { if found == nil {
// No matched hydrant: Lets create one // No matched hydrant: Lets create one
if cfg.NoOp { doNoOp(
log.Printf("[NOOP] Would send a create to OSM (Changeset %d): %#v", cs.ID, h.ToNode()) fmt.Sprintf("[NOOP] Would send a create to OSM (Changeset %d): %#v", cs.ID, h.ToNode()),
} else { func() {
if err := osmClient.SaveNode(h.ToNode(), cs); err != nil { if err := osmClient.SaveNode(h.ToNode(), cs); err != nil {
log.Fatalf("Unable to create node using the OSM API: %s", err) log.Fatalf("Unable to create node using the OSM API: %s", err)
} }
logDebugf("Created a hydrant: %s", h.Name) logDebugf("Created a hydrant: %s", h.Name)
} },
)
continue continue
} }
// Special case: If the diameter of the recorded hydrant is unknown but previously known keep the previous version
if h.Diameter == 0 && found.Diameter > 0 { if h.Diameter == 0 && found.Diameter > 0 {
h.Diameter = found.Diameter h.Diameter = found.Diameter
} }
if h.Diameter == found.Diameter && h.Position == found.Position && h.Pressure == found.Pressure && h.Type == found.Type { if !found.NeedsUpdate(h) {
logDebugf("Found a good looking hydrant which needs no update: %#v", h) logDebugf("Found a good looking hydrant which needs no update: %#v", h)
// Everything matches, we don't care // Everything matches, we don't care
continue continue
@ -182,15 +203,25 @@ func main() {
h.ID = found.ID h.ID = found.ID
h.Version = found.Version h.Version = found.Version
if cfg.NoOp { doNoOp(
log.Printf("[NOOP] Would send a change to OSM (Changeset %d): To=%#v From=%#v", cs.ID, h.ToNode(), found.ToNode()) fmt.Sprintf("[NOOP] Would send a change to OSM (Changeset %d): To=%#v From=%#v", cs.ID, h.ToNode(), found.ToNode()),
} else { func() {
if err := osmClient.SaveNode(h.ToNode(), cs); err != nil { if err := osmClient.SaveNode(h.ToNode(), cs); err != nil {
log.Fatalf("Unable to create node using the OSM API: %s", err) log.Fatalf("Unable to create node using the OSM API: %s", err)
} }
logDebugf("Changed a hydrant: %s", h.Name) logDebugf("Changed a hydrant: %s", h.Name)
},
)
} }
}
func doNoOp(message string, execution func()) {
if cfg.NoOp {
log.Println(message)
return
} }
execution()
} }
func roundPrec(in float64, nd int) float64 { func roundPrec(in float64, nd int) float64 {