DiffEditorsGroup.java

package jetbrains.mps.vcs.diff.ui.common;

/*Generated by MPS */

import java.util.List;
import jetbrains.mps.internal.collections.runtime.ListSequence;
import java.util.ArrayList;
import jetbrains.mps.nodeEditor.EditorComponent;
import jetbrains.mps.smodel.ModelAccess;
import org.jetbrains.mps.openapi.model.SNode;
import org.jetbrains.mps.openapi.model.SModel;
import org.jetbrains.mps.openapi.model.SNodeId;
import jetbrains.mps.openapi.editor.cells.EditorCell;
import java.awt.Point;
import java.awt.Rectangle;
import jetbrains.mps.lang.smodel.generator.smodelAdapter.SNodeOperations;
import jetbrains.mps.nodeEditor.selection.SingularSelectionListenerAdapter;
import jetbrains.mps.openapi.editor.selection.SingularSelection;
import javax.swing.event.ChangeListener;
import javax.swing.event.ChangeEvent;
import jetbrains.mps.internal.collections.runtime.IVisitor;
import org.jetbrains.annotations.Nullable;
import jetbrains.mps.internal.collections.runtime.IWhereFilter;

public class DiffEditorsGroup {
  private List<DiffEditor> myDiffEditors = ListSequence.fromList(new ArrayList<DiffEditor>());
  private DiffEditorsGroup.MyCellSelectionListener myCellSelectionListener = new DiffEditorsGroup.MyCellSelectionListener();
  private boolean myViewportSetInProgress = false;
  public DiffEditorsGroup() {
  }
  public void add(DiffEditor diffEditor) {
    ListSequence.fromList(myDiffEditors).addElement(diffEditor);
    diffEditor.getMainEditor().getSelectionManager().addSelectionListener(myCellSelectionListener);
    diffEditor.getMainEditor().getViewport().addChangeListener(new DiffEditorsGroup.MyViewportChangeListener(diffEditor));
  }
  public void synchronizeViewWithOther(final DiffEditor thisDiffEditor, final DiffEditor otherDiffEditor) {
    if (thisDiffEditor == otherDiffEditor) {
      return;
    }
    final EditorComponent thisEditor = thisDiffEditor.getMainEditor();
    final EditorComponent otherEditor = otherDiffEditor.getMainEditor();

    ModelAccess.instance().runReadAction(new Runnable() {
      public void run() {
        int viewY = thisEditor.getViewport().getViewPosition().y;
        SNode visibleNode = thisEditor.getEditedNode();
        if (viewY > thisEditor.getRootCell().getY()) {
          visibleNode = check_s6qw4f_a0a0c0a0a4a5(thisEditor.findCellWeak(1, viewY));
        }
        SModel otherModel = check_s6qw4f_a0d0a0a4a5(otherEditor.getEditedNode());
        if (otherModel == null) {
          return;
        }

        while (visibleNode != null) {
          SNodeId id = visibleNode.getNodeId();
          EditorCell thisCell = thisEditor.findNodeCell(visibleNode);
          if (thisCell != null) {
            int newRelativePos = viewY - thisCell.getY();
            SNodeId nodeId = mapID(thisDiffEditor, id, otherDiffEditor);
            EditorCell otherCell = (nodeId != null ? otherEditor.findNodeCell(otherModel.getNode(nodeId)) : null);
            Point position = thisEditor.getViewport().getViewPosition();
            if (otherCell != null) {
              Rectangle viewRect = otherEditor.getViewport().getViewRect();
              int newX = Math.min((int) position.getX(), otherEditor.getWidth() - viewRect.width);
              int newY = Math.min(newRelativePos + otherCell.getY(), otherEditor.getHeight() - viewRect.height);
              otherEditor.getViewport().setViewPosition(new Point(newX, newY));
              return;
            }
          }

          SNode prevSibling = SNodeOperations.getPrevSibling(visibleNode);
          if (check_s6qw4f_a5a6a0a0e0f(visibleNode.getRoleInParent(), prevSibling)) {
            visibleNode = prevSibling;
          } else {
            visibleNode = visibleNode.getParent();
          }
        }
      }
    });
  }
  private class MyCellSelectionListener extends SingularSelectionListenerAdapter {
    public MyCellSelectionListener() {
    }
    @Override
    protected void selectionChangedTo(final jetbrains.mps.openapi.editor.EditorComponent component, final SingularSelection newSelection) {
      ModelAccess.instance().runReadAction(new Runnable() {
        public void run() {
          SNodeId selectionId = check_s6qw4f_a0a0a0a0a1g(check_s6qw4f_a0a0a0a0a0b6(newSelection.getEditorCell()));
          if (selectionId != null) {
            DiffEditor diffEditor0 = getDiffEditor(component);
            for (DiffEditor diffEditor : ListSequence.fromList(myDiffEditors)) {
              jetbrains.mps.openapi.editor.EditorComponent mainEditor = diffEditor.getMainEditor();
              SNodeId nodeId = mapID(diffEditor0, selectionId, diffEditor);
              SNode node = (nodeId != null ? check_s6qw4f_a0a2a1a1a0a0a0b6(check_s6qw4f_a0a0c0b0b0a0a0a1g(mainEditor.getEditedNode()), nodeId) : null);
              diffEditor.inspect(node);
            }
          }
        }
      });
    }
  }
  private class MyViewportChangeListener implements ChangeListener {
    private DiffEditor myDiffEditor;
    private MyViewportChangeListener(DiffEditor diffEditor) {
      myDiffEditor = diffEditor;
    }
    @Override
    public void stateChanged(ChangeEvent event) {
      if (myViewportSetInProgress) {
        return;
      }
      myViewportSetInProgress = true;
      ListSequence.fromList(myDiffEditors).visitAll(new IVisitor<DiffEditor>() {
        public void visit(DiffEditor other) {
          synchronizeViewWithOther(myDiffEditor, other);
        }
      });
      myViewportSetInProgress = false;
    }
  }

  @Nullable
  protected SNodeId mapID(DiffEditor myEditor, SNodeId myNodeId, DiffEditor otherEditor) {
    return myNodeId;
  }

  private DiffEditor getDiffEditor(final jetbrains.mps.openapi.editor.EditorComponent component) {
    return ListSequence.fromList(myDiffEditors).findFirst(new IWhereFilter<DiffEditor>() {
      public boolean accept(DiffEditor it) {
        return it.getMainEditor() == component;
      }
    });
  }
  private static SNode check_s6qw4f_a0a0c0a0a4a5(EditorCell checkedDotOperand) {
    if (null != checkedDotOperand) {
      return checkedDotOperand.getSNode();
    }
    return null;
  }
  private static SModel check_s6qw4f_a0d0a0a4a5(SNode checkedDotOperand) {
    if (null != checkedDotOperand) {
      return checkedDotOperand.getModel();
    }
    return null;
  }
  private static boolean check_s6qw4f_a5a6a0a0e0f(String checkedDotOperand, SNode prevSibling) {
    if (null != checkedDotOperand) {
      return checkedDotOperand.equals(check_s6qw4f_a0a5a6a0a0e0f(prevSibling));
    }
    return false;
  }
  private static String check_s6qw4f_a0a5a6a0a0e0f(SNode checkedDotOperand) {
    if (null != checkedDotOperand) {
      return checkedDotOperand.getRoleInParent();
    }
    return null;
  }
  private static SNodeId check_s6qw4f_a0a0a0a0a1g(SNode checkedDotOperand) {
    if (null != checkedDotOperand) {
      return checkedDotOperand.getNodeId();
    }
    return null;
  }
  private static SNode check_s6qw4f_a0a0a0a0a0b6(EditorCell checkedDotOperand) {
    if (null != checkedDotOperand) {
      return checkedDotOperand.getSNode();
    }
    return null;
  }
  private static SNode check_s6qw4f_a0a2a1a1a0a0a0b6(SModel checkedDotOperand, SNodeId nodeId) {
    if (null != checkedDotOperand) {
      return checkedDotOperand.getNode(nodeId);
    }
    return null;
  }
  private static SModel check_s6qw4f_a0a0c0b0b0a0a0a1g(SNode checkedDotOperand) {
    if (null != checkedDotOperand) {
      return checkedDotOperand.getModel();
    }
    return null;
  }
}