package org.simantics.interop.update.model;

import java.util.ArrayDeque;
import java.util.ArrayList;
import java.util.Collections;
import java.util.Deque;
import java.util.List;
import java.util.Map.Entry;

import org.eclipse.core.runtime.IProgressMonitor;
import org.eclipse.core.runtime.NullProgressMonitor;
import org.simantics.Simantics;
import org.simantics.db.ReadGraph;
import org.simantics.db.Resource;
import org.simantics.db.Session;
import org.simantics.db.Statement;
import org.simantics.db.VirtualGraph;
import org.simantics.db.WriteGraph;
import org.simantics.db.common.request.ReadRequest;
import org.simantics.db.common.request.WriteRequest;
import org.simantics.db.exception.DatabaseException;
import org.simantics.db.layer0.util.Layer0Utils;
import org.simantics.db.request.Read;
import org.simantics.interop.test.GraphChanges;
import org.simantics.interop.test.GraphChanges.Modification;
import org.simantics.interop.test.GraphComparator;
import org.simantics.utils.datastructures.Pair;

public abstract class ModelUpdate {
	
	private Resource oldModel;      // old model that is going to be updated (User modified model)
	private Resource newModel;      // new model that contains the updates   (New design model)
	private Resource originalModel; // original model (optional) that is used for detecting and retaining user made changes (Old design model)
	
	private GraphChanges changes;   // changes between old /new
	private UpdateTree updateTree;
	private UpdateList updateList;
	
	private GraphChanges changes2;  // changes between original / old
	private UpdateTree updateTree2;
	private UpdateList updateList2;
	
	private GraphChanges changes3;  // changes between original / new
	private UpdateTree updateTree3;
	private UpdateList updateList3;
	
	private UpdateNode3 updateNode3;
	
	private List<ChangeFilter> filters = new ArrayList<ChangeFilter>();
	private List<ChangeFilter2> userFilters = new ArrayList<ChangeFilter2>();
	
	boolean init = false;
	private int chunkSize = -1;
	
	public void setInput(Resource oldModel, Resource newModel) throws DatabaseException {
		setInput(oldModel, newModel, null, false);
	}
	
	/**
	 * Initialises the ModelUpdate with given input
	 * @param oldModel the model that is going to be updated (User modified model)
	 * @param newModel the model containing updates (New design model)
	 * @param originalModel the model that is used for detecting and retaining user made changes (Old design model). Parameter can be null.
	 * @param newDistinct when originalModel is given, additions to the old and the new model (when compared to the original model) are forced to be distinct. 
	 * @throws DatabaseException
	 */
	public void setInput(Resource oldModel, Resource newModel, Resource originalModel, boolean newDistinct) throws DatabaseException{
		this.setInput(oldModel, newModel, originalModel, newDistinct, null);
	}
	public void setInput(Resource oldModel, Resource newModel, Resource originalModel, boolean newDistinct, IProgressMonitor monitor) throws DatabaseException{
		
		this.oldModel = oldModel;
		this.newModel = newModel;
		this.originalModel = originalModel;
		
		if (monitor == null)
			monitor = new NullProgressMonitor();
//		addFilters(filters);
		if (originalModel != null) {
			// tree way comparison
			// compare the original and the old model
			Pair<GraphComparator,String> result2 = getChanges(originalModel, oldModel);
			GraphComparator comparator2  = result2.first;
			if (result2.second != null)
				showWarning(result2.second);
			comparator2.test(getSession(), monitor);
			changes2 = comparator2.getChanges();
			comparator2.dispose();
			if (monitor.isCanceled()) {
				dispose();
				return;
			}
			changes2 = getSession().syncRequest(createFilterRead(changes2, filters));
			Pair<UpdateTree, UpdateList> chg2 = createChangeObjects(changes2, null);
			updateTree2 = chg2.first;
			updateList2 = chg2.second;
			
			// compare the original and the new model
			Pair<GraphComparator,String> result3 = getChanges(originalModel,newModel);
			GraphComparator comparator3  = result3.first;
			if (result3.second != null)
				showWarning(result3.second);
			comparator3.test(getSession(), monitor);
			changes3 = comparator3.getChanges();
			comparator2.dispose();
			changes3 = getSession().syncRequest(createFilterRead(changes3, filters));
		}
		if (monitor.isCanceled()) {
			dispose();
			return;
		}
		
		Pair<GraphComparator,String> result = getChanges(oldModel,newModel);
		GraphComparator comparator  = result.first;
		if (result.second != null)
			showWarning(result.second);
		if (originalModel != null) {
			// three-way comparison: use change information to configure
			// the comparison between the old and the new model.
			
			// 1. map comparable resources
			for (Entry<Resource, Resource> origToOld : changes2.getComparable().getEntries()) {
				Resource oldR = origToOld.getValue();
				Resource newR = changes3.getComparable().getRight(origToOld.getKey());
				if (newR != null) {
					comparator.addComparableResources(oldR, newR);
				}
			}
			// 2. mark removed resources as distinct, so that comparison does not pair them
			for (Statement s : changes2.getDeletions()) {
				if (changes3.getComparable().containsLeft(s.getObject()))
					comparator.addNonMatchedRight(changes3.getComparable().getRight(s.getObject()));
			}
			
			for (Statement s : changes3.getDeletions()) {
				if (changes2.getComparable().containsLeft(s.getObject()))
					comparator.addNonMatchedLeft(changes2.getComparable().getRight(s.getObject()));
			}
			if (newDistinct) {
				// 3. mark added resources as distinct, so that comparison does not pair them
				for (Statement s : changes2.getAdditions()) {
					comparator.addNonMatchedLeft(s.getObject());
				}
				
				for (Statement s : changes3.getAdditions()) {
					comparator.addNonMatchedRight(s.getObject());
				}
			}
		}
		if (monitor.isCanceled()) {
			dispose();
			return;
		}
		comparator.test(getSession(), monitor);
		monitor.setTaskName("Processing changes...");
		monitor.subTask("");
		changes = comparator.getChanges();
		comparator.dispose();
		changes = getSession().syncRequest(createFilterRead(changes, filters));
		Pair<UpdateTree, UpdateList> chg = createChangeObjects(changes, monitor);
		if (chg == null) {
			dispose();
			return;
		}
        updateTree = chg.first;
        updateList = chg.second;
		if (userFilters.size() != 0) {
			refreshUserFilters();
		}
		
		
		if (originalModel != null) {
			defaultSelections();
		}
		init = true;
	}
	
	public void setInput(Resource oldModel, Resource newModel, GraphChanges changes, IProgressMonitor monitor) throws DatabaseException{
		if (!oldModel.equals(changes.getResource1()) ||
			!newModel.equals(changes.getResource2())) {
			throw new DatabaseException("GraphChanges does not match input models");
		}
		this.changes = getSession().syncRequest(createFilterRead(changes, filters));
		Pair<UpdateTree, UpdateList> chg = createChangeObjects(changes, monitor);
		if (chg == null) {
			dispose();
			return;
		}
        updateTree = chg.first;
        updateList = chg.second;
		if (userFilters.size() != 0) {
			refreshUserFilters();
		}
		
		
		if (originalModel != null) {
			defaultSelections();
		}
		init = true;
	}
	
	public void addFilter(ChangeFilter filter) {
		if (init)
			throw new IllegalStateException("ModelUpdate has been initialized, adjusting filters is no longer possible.");
		filters.add(filter);
		
	}
	
	public List<ChangeFilter> getFilters() {
		return Collections.unmodifiableList(filters);
	}
	
	/**
	 * Adds an user filter. Use refreshUserFilters() to apply the changes.
	 * @param filter
	 */
	public void addUserFilter(ChangeFilter2 filter) {
		userFilters.add(filter);
	}
	
	/**
	 * Removes an user filter. Use refreshUserFilters() to apply the changes.
	 * @param filter
	 */
	public void removeUserFilter(ChangeFilter2 filter) {
		userFilters.remove(filter);
	}
	
	/**
	 * Clears user filters.  Use refreshUserFilters() to apply the changes.
	 */
	public void clearUserFilters() {
		userFilters.clear();
	}
	
	public List<ChangeFilter2> getUserFilters() {
		return userFilters;
	}
	
	public int getChunkSize() {
		return chunkSize;
	}
	
	public void setChunkSize(int chunkSize) {
		this.chunkSize = chunkSize;
	}
	
	public void refreshUserFilters() throws DatabaseException{
		// use user filters to set visible flags of changes.
		// First, set all changes visible.
		Deque<UpdateNode> stack = new ArrayDeque<>();
		stack.push(updateTree.getRootNode());
		while (!stack.isEmpty()) {
			UpdateNode n = stack.pop();
			n.setVisible(true);
			stack.addAll(n.getChildren());
		}
		for (PropertyChange pc : updateList.getChanges()) {
			pc.setVisible(true);
		}
		if (userFilters.size() > 0) {
		    if (changes2 != null && changes3 != null) {
		        getUpdateTree3();
		    }
		    getSession().syncRequest(new ReadRequest() {
                
                @Override
                public void run(ReadGraph graph) throws DatabaseException {
                    for (PropertyChange change : updateList.getChanges()) {
                        boolean visible = true;
                        for (ChangeFilter2 filter : userFilters) {
                            if (!filter.accept(graph, change)) {
                                visible = false;
                                break;
                            }
                        }
                        change.setVisible(visible);
                    }
                    if (updateTree3 != null) {
                        Deque<UpdateNode3> stack = new ArrayDeque<>();
                        stack.add(getUpdateNode3());
                        while (!stack.isEmpty()) {
                            UpdateNode3 n = stack.pop();
                            boolean visible = true;
                            for (ChangeFilter2 filter : userFilters) {
                                if (!filter.accept(graph, n)) {
                                    visible = false;
                                    break;
                                }
                            }
                            n.setVisible(visible);
                            for (UpdateNode3 c : n.getChildren())
                                stack.push(c);
                        }
                    } else {
                    
                        Deque<UpdateNode> stack = new ArrayDeque<>();
                        stack.add(updateTree.getRootNode());
                        while (!stack.isEmpty()) {
                            UpdateNode n = stack.pop();
                            boolean visible = true;
                            for (ChangeFilter2 filter : userFilters) {
                                if (!filter.accept(graph, n)) {
                                    visible = false;
                                    break;
                                }
                            }
                            n.setVisible(visible);
                            for (UpdateNode c : n.getChildren())
                                stack.push(c);
                        }
                    }
                }
            });
		}
	}
	
	protected abstract Pair<GraphComparator,String> getChanges(Resource r1, Resource r2)  throws DatabaseException;
	
	protected Pair<UpdateTree, UpdateList> createChangeObjects(GraphChanges changes, IProgressMonitor monitor) throws DatabaseException{
		if (monitor != null) {
			if (monitor.isCanceled())
				return null;
			monitor.subTask("Processing structural changes");
		}
		UpdateTree updateTree = getUpdateTree(changes, monitor);
		if (monitor != null) {
			if (monitor.isCanceled())
				return null;
			monitor.subTask("Processing property changes");
		}
        UpdateList updateList = getUpdateList(changes, monitor);
        if (monitor != null) {
        	if (monitor.isCanceled())
				return null;
        	monitor.subTask("Postprocessing changes");
        }
        postProcess(updateTree, updateList);
        return new Pair<UpdateTree, UpdateList>(updateTree, updateList);
	}
	
	protected abstract UpdateTree getUpdateTree(GraphChanges changes, IProgressMonitor monitor) throws DatabaseException;
	protected UpdateList getUpdateList(GraphChanges changes, IProgressMonitor monitor) throws DatabaseException {
		return new UpdateList(changes, changes.getModifications());
	}
	

	protected void postProcess(UpdateTree updateTree, UpdateList updateList) throws DatabaseException{
	    
	}
	
	public Resource getOldModel() {
		return oldModel;
	}
	
	public Resource getNewModel() {
		return newModel;
	}
	
	public Resource getOriginalModel() {
		return originalModel;
	}
	
	public boolean isInit() {
		return init;
	}
	
	public GraphChanges getChanges() {
		return changes;
	}
	public UpdateTree getUpdateTree() {
		return updateTree;
	}
	public UpdateList getUpdateList() {
		return updateList;
	}
	public GraphChanges getChanges2() {
		return changes2;
	}
	
	public UpdateTree getUpdateTree2() {
		return updateTree2;
	}
	public UpdateList getUpdateList2() {
		return updateList2;
	}
	
	public GraphChanges getChanges3() {
		return changes3;
	}
	
	public UpdateTree getUpdateTree3() throws DatabaseException{
		if (updateTree3 == null && changes3 != null) {
		    Pair<UpdateTree, UpdateList> chg3 = createChangeObjects(changes3, null);
            updateTree3 = chg3.first;
            updateList3 = chg3.second;
		}
		return updateTree3;
	}
	public UpdateList getUpdateList3() throws DatabaseException {
		if (updateList3 == null && changes3 != null) {
			Pair<UpdateTree, UpdateList> chg3 = createChangeObjects(changes3, null);
			updateTree3 = chg3.first;
			updateList3 = chg3.second;
		}
		return updateList3;
	}
	
	public UpdateNode3 getUpdateNode3() throws DatabaseException {
	    if (updateNode3 == null && changes2 != null && changes3 != null) {
	        updateNode3 = UpdateNode3.getCombinedTree(this);
	    }
	    return updateNode3;
	}

	public void applyAll(WriteGraph graph) throws DatabaseException {
		List<PropertyChange> list = updateList.getChanges();
		apply(graph, list, "Apply all model updates");
		
		updateTree.getUpdateOps().applyAll(graph);
	}
	
	public void applySelected(WriteGraph graph) throws DatabaseException {
		List<PropertyChange> list = new ArrayList<PropertyChange>();
		for (PropertyChange mod : updateList.getChanges()) {
			if (mod.selected())
				list.add(mod);
		}
		apply(graph, list, "Apply selected model updates");
		
		updateTree.getUpdateOps().applySelected(graph);
	}
	
	public void applyAll(Session session, VirtualGraph vg) throws DatabaseException {
		List<PropertyChange> list = updateList.getChanges();
		apply(session, list, "Apply all model updates", vg);
		
		updateTree.getUpdateOps().applyAll(session, vg);
	}
	
	public void applySelected(Session session, VirtualGraph vg) throws DatabaseException {
		List<PropertyChange> list = new ArrayList<PropertyChange>();
		for (PropertyChange mod : updateList.getChanges()) {
			if (mod.selected())
				list.add(mod);
		}
		apply(session, list, "Apply selected model updates", vg);
		
		updateTree.getUpdateOps().applySelected(session, vg);
		
	}
	
	protected void apply(WriteGraph graph, List<PropertyChange> list, String message) throws DatabaseException {
		Layer0Utils.addCommentMetadata(graph, message);
		graph.markUndoPoint();
		for (PropertyChange mod : list) {
			mod.apply(graph);
		}
	}
	
	protected void apply(Session session, List<PropertyChange> list, String message, VirtualGraph vg) throws DatabaseException {
		if (getChunkSize() > 0) {
			for (int s = 0; s < list.size(); ) {
				int e = s + getChunkSize();
				if (e > list.size())
					e = list.size();
				List<PropertyChange> subList = updateList.getChanges().subList(e, e);
				if (s == 0) {
					session.syncRequest(new WriteRequest(vg) {
						public void perform(WriteGraph graph) throws DatabaseException {
							Layer0Utils.addCommentMetadata(graph, message);
							graph.markUndoPoint();
						}
					});
				}
				session.syncRequest(new WriteRequest(vg) {
					
					@Override
					public void perform(WriteGraph graph) throws DatabaseException {
						for (PropertyChange mod : subList) {
							mod.apply(graph);
						}
					}
				});
				s = e;
			}
		} else {
			session.syncRequest(new WriteRequest(vg) {
				
				@Override
				public void perform(WriteGraph graph) throws DatabaseException {
					Layer0Utils.addCommentMetadata(graph, message);
					graph.markUndoPoint();
					for (PropertyChange mod : list) {
						mod.apply(graph);
					}
				}
			});
		}
	}
	
	
	protected Session getSession() {
		return Simantics.getSession();
	}
	
	public Read<GraphChanges> createFilterRead(GraphChanges changes, List<ChangeFilter> filters) {
		return new FilterChangesRead(changes, filters);
	}
	
	
	
	public static class FilterChangesRead implements Read<GraphChanges> {
		private GraphChanges changes;
		private List<ChangeFilter> filters;
		
		public FilterChangesRead(GraphChanges changes, List<ChangeFilter> filters) {
			this.changes = changes;
			this.filters = filters;
		}
		
		@Override
		public GraphChanges perform(ReadGraph graph) throws DatabaseException {
			return filterChanges(graph, changes);
		}
		
		/**
		 * Filters changes:
		 * 1. Changes that are not essential for model update (changes that can be found when the models are exactly the same)
		 * 2. Runs custom filters for value changes. 
		 * 
		 * @param g
		 * @param changes
		 * @return
		 * @throws DatabaseException
		 */
		protected GraphChanges filterChanges(ReadGraph g, GraphChanges changes) throws DatabaseException 
	    {
			
			List<Modification> modifications = new ArrayList<Modification>();
			
	    	for (Modification mod : changes.getModifications()) {
	    		
	    		boolean accept = true;
	    		for (ChangeFilter filter : filters) {
	    			if (!filter.accept(g, mod)) {
	    				accept = false;
	    				break;
	    			}	
	    		}
	    		if (accept)
	    			modifications.add(mod);
	    	}
	    	List<Statement> deletions = new ArrayList<Statement>();
	    	for (Statement del : changes.getDeletions()) {
                
                boolean accept = true;
                for (ChangeFilter filter : filters) {
                    if (!filter.acceptDeletion(g, del)) {
                        accept = false;
                        break;
                    }   
                }
                if (accept)
                    deletions.add(del);
            }
	    	List<Statement> additions = new ArrayList<Statement>();
            for (Statement del : changes.getAdditions()) {
                
                boolean accept = true;
                for (ChangeFilter filter : filters) {
                    if (!filter.acceptAddition(g, del)) {
                        accept = false;
                        break;
                    }   
                }
                if (accept)
                    additions.add(del);
            }

	    	GraphChanges newChanges =  new GraphChanges(changes.getResource1(),changes.getResource2(),deletions, additions, modifications, changes.getComparable());
	    	return newChanges;
	    }
	}

	/**
	 * Interface for built-in filters that are used for processing raw change data before forming UpdateTree + UpdateList
	 * @author luukkainen
	 *
	 */
	public interface ChangeFilter {
		public boolean accept(ReadGraph g, Modification change) throws DatabaseException;
		public boolean acceptAddition(ReadGraph g, Statement addition) throws DatabaseException;
		public boolean acceptDeletion(ReadGraph g, Statement deletion) throws DatabaseException;
	}
	
	/**
	 * Interface for user defined filters. 
	 * 
	 * This filter only affects visible flags.
	 *  
	 * @author luukkainen
	 *
	 */
	public interface ChangeFilter2 {
	    public boolean accept(ReadGraph g, PropertyChange change) throws DatabaseException;
	    public boolean accept(ReadGraph g, UpdateNode change) throws DatabaseException;
	    public boolean accept(ReadGraph g, UpdateNode3 change) throws DatabaseException;
	}

	
	/**
	 * 
	 * Filters floating point value changes (default filter is set filter when the change is less than 1%)  
	 *
	 */
	protected class FPValueFilter implements ChangeFilter  {
		
		private double percentage = 0.01;
		
		public FPValueFilter() {
			
		}
		
		public FPValueFilter(double percentage) {
			if (percentage < 0.0 || percentage > 1.0)
				throw new IllegalArgumentException("Percentage must be between 0.0 and 1.0.");
			this.percentage = percentage;
		}
		
		@Override
		public boolean accept(ReadGraph g, Modification change) throws DatabaseException {
			//filter floating point values that have less than 1% difference.
			if (change.getLeftStm() == null || change.getRightStm() == null)
				return true;
			if (!g.hasValue(change.getLeftStm().getObject()) || !g.hasValue(change.getRightStm().getObject()))
				return true;
    		Object v1 = g.getValue(change.getLeftStm().getObject());
    		Object v2 = g.getValue(change.getRightStm().getObject());
    		
    		if (v1 instanceof Double && v2 instanceof Double) {
    			double d1 = (Double)v1;
    			double d2 = (Double)v2;
    			if (Math.abs(d1-d2) / Math.max(Math.abs(d1), Math.abs(d2)) < percentage)
    				return false;
    		} else if (v1 instanceof Float && v2 instanceof Float) {
    			float d1 = (Float)v1;
    			float d2 = (Float)v2;
    			if (Math.abs(d1-d2) / Math.max(Math.abs(d1), Math.abs(d2)) < percentage)
    				return false;
    		}

    		return true;
		}
		
		@Override
		public boolean acceptAddition(ReadGraph g, Statement addition) throws DatabaseException {
		    return true;
		}
		
		@Override
		public boolean acceptDeletion(ReadGraph g, Statement deletion) throws DatabaseException {
		    return true;
		}
	}
	
	public void defaultSelections() {
		if (changes3 == null) {
			return;
		}
		// select all changes
		for (Entry<Resource, UpdateOp> op : updateTree.getUpdateOps().getResourceMap().entrySet()) {
			op.getValue().select(true);
		}
		
		
		for (PropertyChange pair : updateList.getChanges()) {
			pair.select(true);
		}
		
		// preserve user-made changes (by removing selections)
		for (Entry<Resource, UpdateOp> op : updateTree.getUpdateOps().getResourceMap().entrySet()) {
			UpdateOp op2 = updateTree2.getUpdateOps().getUpdateOp(op.getKey());
			if (op2 == null) {
				if (changes3.getComparable().containsRight(op.getKey())){
					op2 = updateTree2.getUpdateOps().getUpdateOp(changes3.getComparable().getLeft(op.getKey()));
				}
			}
			if (op2 != null && op.getValue().getClass() == op2.getClass()) {
				op.getValue().select(false);
			}
		}
		
		for (PropertyChange pair : updateList.getChanges()) {
			if (pair.getFirst() != null) {
				boolean found = false;
				for (PropertyChange pair2 : updateList2.getChanges()) {
					if (pair.getFirst() != null && pair.getFirst().equals(pair2.getSecond())) {
						found = true;
						break;
					}
				}
				if (found) {
					pair.select(false);
				}
			}
		}
		
	}
	
	private void showWarning(String string) {
		for (WarningListener l : warningListeners)
			l.showWarning(this, string);
	}
	
	private List<WarningListener> warningListeners = new ArrayList<>();
	
	public static interface WarningListener {
		void showWarning(ModelUpdate update, String warning);
	}
	
	public void addListener(WarningListener listener) {
		warningListeners.add(listener);
	}
	
	public void removeListener(WarningListener listener) {
		warningListeners.remove(listener);
	}
	
	public void dispose() {
		changes = null;
		changes2 = null;
		changes3 = null;
		filters = null;
		userFilters = null;
		updateList = null;
		updateList2 = null;
		updateList3 = null;
		updateTree = null;
		updateTree2 = null;
		updateTree3 = null;
		updateNode3 = null;
		init = false;
	}
}