// 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>

package models

import (
	"context"
	"database/sql"
)

const reprojectSQL = `
SELECT ST_X(p), ST_Y(p)
FROM ST_Transform(ST_SetSRID(ST_MakePoint($1, $2), $3::integer), $4::integer) AS p`

// Reprojector re-projects point data from one coord system
// to another.
type Reprojector struct {
	stmt     *sql.Stmt
	FromEPSG uint32
	ToEPSG   uint32
}

// NewReprojector creates a reprojector.
func NewReprojector(
	ctx context.Context,
	conn *sql.Conn,
	fromEPSG, toEPSG uint32,
) (*Reprojector, error) {
	stmt, err := conn.PrepareContext(ctx, reprojectSQL)
	if err != nil {
		return nil, err
	}
	return &Reprojector{
		stmt:     stmt,
		FromEPSG: fromEPSG,
		ToEPSG:   toEPSG,
	}, nil
}

// Close closes the reprojector.
func (r *Reprojector) Close() error {
	if s := r.stmt; s != nil {
		r.stmt = nil
		return s.Close()
	}
	return nil
}

// Reproject performs the re-projection for a given point.
func (r *Reprojector) Reproject(
	ctx context.Context,
	x, y float64,
) (v, w float64, err error) {
	err = r.stmt.QueryRowContext(ctx, x, y, r.FromEPSG, r.ToEPSG).Scan(&v, &w)
	return
}