plugins/ui-designer-core/src/com/intellij/designer/designSurface/tools/AbstractCreationTool.java

/*
 * Copyright 2000-2012 JetBrains s.r.o.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.intellij.designer.designSurface.tools;

import com.intellij.designer.designSurface.EditableArea;
import com.intellij.designer.model.RadComponent;
import com.intellij.designer.utils.Cursors;

import java.awt.*;
import java.awt.event.KeyEvent;
import java.awt.event.MouseEvent;

/**
 * @author Alexander Lobas
 */
public abstract class AbstractCreationTool extends TargetingTool {
  private boolean myCanUnload;

  protected AbstractCreationTool(boolean canUnload) {
    myCanUnload = canUnload;
    setDefaultCursor(Cursors.getCopyCursor());
    setDisabledCursor(Cursors.getNoCursor());
  }

  @Override
  protected void handleButtonDown(int button) {
    if (button == MouseEvent.BUTTON1) {
      if (myState == STATE_INIT) {
        myState = STATE_DRAG;
      }
    }
    else {
      myState = STATE_INVALID;
      handleInvalidInput();
    }
  }

  @Override
  protected void handleButtonUp(int button) {
    if (myState == STATE_DRAG || myState == STATE_DRAG_IN_PROGRESS) {
      eraseFeedback();
      executeCommand();
      selectAddedObjects();
    }

    myState = STATE_NONE;
    handleFinished();
  }

  @Override
  protected void handleMove() {
    updateContext();
    updateTargetUnderMouse();
    showFeedback();
    updateCommand();
  }

  @Override
  protected void handleDragInProgress() {
    if (myState == STATE_DRAG_IN_PROGRESS) {
      updateContext();
      showFeedback();
      updateCommand();
    }
  }

  private void selectAddedObjects() {
    if (myExecuteEnabled) {
      myArea.setSelection(myContext.getComponents());
    }
  }

  @Override
  protected void resetState() {
    // hack: update cursor
    if (myCanUnload || myArea == null) {
      super.resetState();
    }
  }

  private void handleFinished() {
    if (myCanUnload) {
      myToolProvider.loadDefaultTool();
    }
    else {
      deactivate();
      activate();

      // hack: update cursor
      if (myArea != null) {
        handleMove();
      }
    }
  }

  private void updateTargetUnderMouse() {
    ContainerTargetFilter filter = new ContainerTargetFilter();
    RadComponent target = myArea.findTarget(myCurrentScreenX, myCurrentScreenY, filter);
    setTarget(target, filter);

    if (target != null) {
      updateTarget();
    }
  }

  protected abstract void updateTarget();

  @Override
  protected void updateContext() {
    super.updateContext();

    if (myState == STATE_DRAG_IN_PROGRESS) {
      myContext.setSizeDelta(new Dimension(moveDeltaWidth(), moveDeltaHeight()));
    }

    myContext.setLocation(getLocation());
  }

  @Override
  public void keyPressed(KeyEvent event, EditableArea area) throws Exception {
    super.keyPressed(event, area);

    if (event.getKeyCode() == KeyEvent.VK_ESCAPE) {
      myToolProvider.loadDefaultTool();
    }
  }

  @Override
  protected void handleKeyEvent() {
  }
}