map-dots/imaging/circles.go

83 lines
2.4 KiB
Go

package imaging
import (
"fmt"
"image"
"math"
"git.fjla.uk/fred.boniface/map-dots/data"
"git.fjla.uk/fred.boniface/map-dots/log"
"github.com/fogleman/gg"
)
func mapCirclesToCanvas(img *image.RGBA, locations []data.LocationData) {
log.Msg.Debug("Mapping locations to canvas")
dc := gg.NewContextForRGBA(img)
dc.SetRGB(1, 1, 1) // Set canvas background color (white in this case)
circleRadius := 4 // Replace with your desired fixed radius
bounds := img.Bounds()
canvasWidth := bounds.Max.X - bounds.Min.X
canvasHeight := bounds.Max.Y - bounds.Min.Y
var minLat, maxLat, minLon, maxLon float64
for _, loc := range locations {
if loc.Latitude < minLat {
minLat = loc.Latitude
}
if loc.Latitude > maxLat {
maxLat = loc.Latitude
}
if loc.Longitude < minLon {
minLon = loc.Longitude
}
if loc.Longitude > maxLon {
maxLon = loc.Longitude
}
}
centerLat := (maxLat + minLat) / 2.0
centerLon := (maxLon + minLon) / 2.0
latRange := maxLat - minLat
lonRange := maxLon - minLon
for _, loc := range locations {
pixelX, pixelY := convertCoordinatesToPixels(loc.Latitude, loc.Longitude, centerLat, centerLon, latRange, lonRange, canvasWidth, canvasHeight)
// Draw the circle on the canvas using gg
dc.DrawCircle(float64(pixelX), float64(pixelY), float64(circleRadius))
dc.SetRGBA(1, 1, 1, 0.5) // Circle fill color (white with 50% opacity)
dc.Fill()
}
dc.Clip() // Apply clipping
// Optional: Save the canvas as an image file
err := dc.SavePNG("output.png")
if err != nil {
log.Msg.Error("Error saving file")
} else {
fmt.Println("Canvas saves to file")
}
}
func convertCoordinatesToPixels(latitude, longitude float64, centerLat, centerLon, latRange, lonRange float64, canvasWidth, canvasHeight int) (int, int) {
// Calculate normalized latitude and longitude distances from the center
latDist := (latitude - centerLat) / (latRange * 0.5)
lonDist := (longitude - centerLon) / (lonRange * 0.5)
// Calculate the maximum distance from the center as a proportion of canvas size
maxDistance := math.Max(math.Abs(latDist), math.Abs(lonDist))
// Adjust the normalized distances to match the canvas size with margins
adjustedLatDist := latDist / maxDistance * 0.45
adjustedLonDist := lonDist / maxDistance * 0.45
// Calculate pixel positions
pixelX := int((adjustedLonDist + 0.5) * float64(canvasWidth))
pixelY := int((0.5 - adjustedLatDist) * float64(canvasHeight))
return pixelX, pixelY
}