# -*- coding: utf-8 -*-
"""
Created on Tue Nov 27 16:46:58 2018

@author: noordam
"""
import readAndWriteFiles
import helpFunctions

class SurfaceLine:
    
    def __init__(self):
        self.xCoordsOriginalSurfaceLine=[] 
        self.yCoordsOriginalSurfaceLine=[]
        self.xCoordsNewSurfaceLine=[]
        self.yCoordsNewSurfaceLine=[]
        
        self.info=[]
        self.surfaceLineDefinitions =[]
        
    def ReadSurfaceLine(self,surfaceLineFileName,lineInOriginalFile) :
        self.xCoordsOriginalSurfaceLine, self.yCoordsOriginalSurfaceLine, self.info, self.surfaceLineDefinitions = readAndWriteFiles.ReadSurfaceLine(surfaceLineFileName,lineInOriginalFile)


    def WriteSurfaceLine(self,jj,safeFileNameNewSurfaceLine):
        readAndWriteFiles.WriteSurfaceLine(jj,self.surfaceLineDefinitions,self.xCoordsNewSurfaceLine,self.yCoordsNewSurfaceLine,self.info,safeFileNameNewSurfaceLine)
    #readAndWriteFiles.WriteCharacteristicPoints(jj,list(reversed(newCharaterisitcDefinitions)),list(reversed(xCoordsNewCharacteristicPoints)),list(reversed(yCoordsNewCharacteristicPoints)),profileName,safeFileNameCharacteristicPoints)
    

    def CreateNewSurfaceLine(self,nwo):
        
        x1Glob = nwo.globalXCoords[0]
        x2Glob = nwo.globalXCoords[1]
        x3Glob = nwo.globalXCoords[2]
        x4Glob = nwo.globalXCoords[3]
        
        y1Glob = nwo.globalYCoords[0]
        y2Glob = nwo.globalYCoords[1]
        y3Glob = nwo.globalYCoords[2]
        y4Glob = nwo.globalYCoords[3]
        
        #==============================================================================
        # Create new surfaceLine
        #==============================================================================
        self.xCoordsNewSurfaceLine = self.xCoordsOriginalSurfaceLine[:]
        self.yCoordsNewSurfaceLine = self.yCoordsOriginalSurfaceLine[:]
        
        # replace the y coordinates of the existing points
        indexesWithinNWO = helpFunctions.FindIndexesInRange(x1Glob,x4Glob,self.xCoordsOriginalSurfaceLine,"surfaceLine")
        for ii in range(len(indexesWithinNWO)):
            if self.xCoordsNewSurfaceLine[indexesWithinNWO[ii]]<x2Glob:
                self.yCoordsNewSurfaceLine[indexesWithinNWO[ii]] = helpFunctions.GetYatX(self.xCoordsNewSurfaceLine[indexesWithinNWO[ii]],[x1Glob,x2Glob],[y1Glob,y2Glob])
            elif self.xCoordsNewSurfaceLine[indexesWithinNWO[ii]]<x3Glob: 
                self.yCoordsNewSurfaceLine[indexesWithinNWO[ii]] = helpFunctions.GetYatX(self.xCoordsNewSurfaceLine[indexesWithinNWO[ii]],[x2Glob,x3Glob],[y2Glob,y3Glob])
            else:
                self.yCoordsNewSurfaceLine[indexesWithinNWO[ii]] = helpFunctions.GetYatX(self.xCoordsNewSurfaceLine[indexesWithinNWO[ii]],[x3Glob,x4Glob],[y3Glob,y4Glob])    
        
        
        # add the new coordinates to the existing surfaceline
        self.xCoordsNewSurfaceLine.extend(nwo.globalXCoords)
        self.yCoordsNewSurfaceLine.extend(nwo.globalYCoords)
        
        sortedIndexes = sorted(range(len(self.xCoordsNewSurfaceLine)),key=lambda k: self.xCoordsNewSurfaceLine[k])
        self.xCoordsNewSurfaceLine = [self.xCoordsNewSurfaceLine[i] for i in sortedIndexes]
        self.yCoordsNewSurfaceLine = [self.yCoordsNewSurfaceLine[i] for i in sortedIndexes]


        
class CharacteristicPoints:
    def __init__(self):
        self.xCoordsOriginalCharacteristicPoints=[] 
        self.yCoordsOriginalCharacteristicPoints=[] 
        
        self.xCoordsNewCharacteristicPoints=[] 
        self.yCoordsNewCharacteristicPoints=[] 
        
        self.characteristicDefinitions = []
        self.newCharacteristicDefinitions = []
       
    def ReadCharacteristicPoints(self,charPointsFileName,lineInOriginalFile) :
        self.xCoordsOriginalCharacteristicPoints, self.yCoordsOriginalCharacteristicPoints, self.characteristicDefinitions = readAndWriteFiles.ReadCharacteristicPoints(charPointsFileName,lineInOriginalFile)

    def WriteCharacteristicPoints(self,jj,profileName,safeFileNameCharacteristicPoints):
        readAndWriteFiles.WriteCharacteristicPoints(jj,list(reversed(self.newCharacteristicDefinitions)),list(reversed(self.xCoordsNewCharacteristicPoints)),list(reversed(self.yCoordsNewCharacteristicPoints)),profileName,safeFileNameCharacteristicPoints)
    

    def CreateNewCharacteristicPoints(self,nwo):
        
        x1Glob = nwo.globalXCoords[0]
        x2Glob = nwo.globalXCoords[1]
        x3Glob = nwo.globalXCoords[2]
        x4Glob = nwo.globalXCoords[3]
        
        y1Glob = nwo.globalYCoords[0]
        y2Glob = nwo.globalYCoords[1]
        y3Glob = nwo.globalYCoords[2]
        y4Glob = nwo.globalYCoords[3]
        #==============================================================================
        # Create new characteristic points 
        #==============================================================================
        newNWODefinitions = ["X_Niet waterkerend object punt 4","Y_Niet waterkerend object punt 4","Z_Niet waterkerend object punt 4",
                             "X_Niet waterkerend object punt 3","Y_Niet waterkerend object punt 3","Z_Niet waterkerend object punt 3",
                             "X_Niet waterkerend object punt 2","Y_Niet waterkerend object punt 2","Z_Niet waterkerend object punt 2",
                             "X_Niet waterkerend object punt 1","Y_Niet waterkerend object punt 1","Z_Niet waterkerend object punt 1"];
        self.newCharacteristicDefinitions = self.characteristicDefinitions + newNWODefinitions;
        
        self.newCharacteristicDefinitions = list(reversed(self.newCharacteristicDefinitions));
        self.xCoordsNewCharacteristicPoints = list(reversed(self.xCoordsOriginalCharacteristicPoints));
        self.yCoordsNewCharacteristicPoints = list(reversed(self.yCoordsOriginalCharacteristicPoints));
        
        
        dikeDitchIndex = self.newCharacteristicDefinitions.index('Z_Niet waterkerend object punt 1')
        dikeDitchIndex = (dikeDitchIndex + 2)/3 
        
        NWOXPOints =   [x1Glob ,x2Glob,x3Glob, x4Glob];
        NWOYPOints =   [y1Glob ,y2Glob,y3Glob, y4Glob];
        
        
        self.xCoordsNewCharacteristicPoints =NWOXPOints + self.xCoordsNewCharacteristicPoints;
        self.yCoordsNewCharacteristicPoints =NWOYPOints + self.yCoordsNewCharacteristicPoints;

        # replace the characteristic points which fall within the NWO
        
        
        # replace the y coordinates of the existing points
        areDikeTopPointsMoved =False   
        dikeTopOutsideWithinNWO = False
        dikeTopInsideWithinNWO = False
        
        indexesCharWithinNWO = helpFunctions.FindIndexesInRange(x1Glob,x4Glob,self.xCoordsNewCharacteristicPoints,"characteristicPoints")
        for ii in range(len(indexesCharWithinNWO)):
            if self.xCoordsNewCharacteristicPoints[indexesCharWithinNWO[ii]]== self.xCoordsNewCharacteristicPoints[(self.newCharacteristicDefinitions.index('Z_Kruin buitentalud')+2)/3]: # move kruin buitentalud to top NWO
                dikeTopOutsideWithinNWO=True
            if self.xCoordsNewCharacteristicPoints[indexesCharWithinNWO[ii]]== self.xCoordsNewCharacteristicPoints[(self.newCharacteristicDefinitions.index('Z_Kruin binnentalud')+2)/3]: # move kruin binnentalud to top NWO
                dikeTopInsideWithinNWO=True 
            if dikeTopOutsideWithinNWO and dikeTopInsideWithinNWO and not areDikeTopPointsMoved:
                areDikeTopPointsMoved = True
            elif dikeTopOutsideWithinNWO and not areDikeTopPointsMoved:
                self.xCoordsNewCharacteristicPoints[indexesCharWithinNWO[ii]]=x4Glob
                self.yCoordsNewCharacteristicPoints[indexesCharWithinNWO[ii]]=y4Glob
            elif dikeTopInsideWithinNWO and not areDikeTopPointsMoved:    
                self.xCoordsNewCharacteristicPoints[indexesCharWithinNWO[ii]]=x1Glob
                self.yCoordsNewCharacteristicPoints[indexesCharWithinNWO[ii]]=y1Glob
                
                
            if (self.xCoordsNewCharacteristicPoints[indexesCharWithinNWO[ii]]== self.xCoordsNewCharacteristicPoints[(self.newCharacteristicDefinitions.index('Z_Kruin buitentalud')+2)/3] and
                    self.xCoordsNewCharacteristicPoints[indexesCharWithinNWO[ii]]!= x1Glob):
               self.xCoordsNewCharacteristicPoints[indexesCharWithinNWO[ii]]=x4Glob
               self.yCoordsNewCharacteristicPoints[indexesCharWithinNWO[ii]]=y4Glob
            elif self.xCoordsNewCharacteristicPoints[indexesCharWithinNWO[ii]]<x2Glob:
                self.yCoordsNewCharacteristicPoints[indexesCharWithinNWO[ii]] = helpFunctions.GetYatX(self.xCoordsNewCharacteristicPoints[indexesCharWithinNWO[ii]],[x1Glob,x2Glob],[y1Glob,y2Glob])
            elif self.xCoordsNewCharacteristicPoints[indexesCharWithinNWO[ii]]<x3Glob: 
                self.yCoordsNewCharacteristicPoints[indexesCharWithinNWO[ii]] = helpFunctions.GetYatX(self.xCoordsNewCharacteristicPoints[indexesCharWithinNWO[ii]],[x2Glob,x3Glob],[y2Glob,y3Glob])
            else:
                self.yCoordsNewCharacteristicPoints[indexesCharWithinNWO[ii]] = helpFunctions.GetYatX(self.xCoordsNewCharacteristicPoints[indexesCharWithinNWO[ii]],[x3Glob,x4Glob],[y3Glob,y4Glob])