'''
Created on 17 juin 2021

@author: tleduc

Copyright 2020 Thomas Leduc

This file is part of t4gpd.

t4gpd is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.

t4gpd is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with t4gpd.  If not, see <https://www.gnu.org/licenses/>.
'''
from shapely.geometry import Point
from t4gpd.commons.GeomLib import GeomLib
from t4gpd.morph.geoProcesses.AbstractGeoprocess import AbstractGeoprocess

from t4gpd.commons.crossroads_identification.FWTMethod import FWTMethod


class RectifyByFWT(AbstractGeoprocess):
    '''
    classdocs
    '''

    def __init__(self, nOutputRays=8, wavelet='haar'):
        '''
        Constructor
        '''
        self.nOutputRays = nOutputRays
        self.wavelet = wavelet

    def runWithArgs(self, row):
        geom = row.geometry
        raylens = GeomLib.fromMultiLineStringToLengths(geom)
        origin = Point(geom.geoms[0].coords[0])
        nRays = len(raylens)

        recognEngine = FWTMethod(
            level=None, wavelet=self.wavelet, nRays=nRays,
            nOutputRays=self.nOutputRays)
        shapeSignature = recognEngine.signature(raylens, isAPatternLayer=False)

        return {
            'geometry': GeomLib.fromRayLengthsToPolygon(shapeSignature, origin)
            }
