android/src/com/intellij/android/designer/designSurface/layout/caption/GridHorizontalCaptionOperation.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.android.designer.designSurface.layout.caption;

import com.intellij.android.designer.model.layout.grid.RadCaptionGridColumn;
import com.intellij.android.designer.model.layout.grid.RadGridLayoutComponent;
import com.intellij.designer.designSurface.EditableArea;
import com.intellij.designer.designSurface.FeedbackLayer;
import com.intellij.designer.designSurface.OperationContext;
import com.intellij.designer.model.RadComponent;
import org.jetbrains.annotations.Nullable;

import java.util.List;

/**
 * @author Alexander Lobas
 */
public class GridHorizontalCaptionOperation extends HorizontalCaptionFlowBaseOperation<RadGridLayoutComponent> {
  public GridHorizontalCaptionOperation(RadGridLayoutComponent mainContainer,
                                        RadComponent container,
                                        OperationContext context,
                                        EditableArea mainArea) {
    super(mainContainer, container, context, mainArea);
  }

  @Override
  protected int getMainFeedbackHeight(FeedbackLayer layer, int mainYLocation) {
    return myMainContainer.getGridInfo().getSize(layer).height;
  }

  @Override
  protected void execute(@Nullable RadComponent insertBefore) throws Exception {
    RadComponent[][] components = myMainContainer.getGridComponents(false);

    for (RadComponent component : myComponents) {
      component.removeFromParent();
      myContainer.add(component, insertBefore);
    }

    List<RadComponent> columns = myContainer.getChildren();
    int size = columns.size();
    for (int i = 0; i < size; i++) {
      int index = ((RadCaptionGridColumn)columns.get(i)).getIndex();

      for (int j = 0; j < components.length; j++) {
        RadComponent cellComponent = components[j][index];
        if (cellComponent != null) {
          RadGridLayoutComponent.setCellIndex(cellComponent, j, i, false, true);
        }
      }
    }
  }
}