android/src/com/intellij/android/designer/model/morphing/GridLayout.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.model.morphing;

import com.android.SdkConstants;
import com.intellij.android.designer.model.*;
import com.intellij.android.designer.model.grid.GridInfo;
import com.intellij.android.designer.model.layout.grid.RadGridLayoutComponent;
import com.intellij.android.designer.model.layout.table.RadTableLayout;
import com.intellij.android.designer.model.layout.table.RadTableLayoutComponent;
import com.intellij.designer.model.MetaModel;
import com.intellij.designer.model.RadComponent;
import com.intellij.psi.xml.XmlAttribute;
import com.intellij.psi.xml.XmlTag;

import java.awt.*;
import java.util.ArrayList;
import java.util.List;

/**
 * @author Alexander Lobas
 */
public class GridLayout {
  public static RadViewComponent TableLayout(RadViewComponent component, MetaModel target) throws Exception {
    final List<RadComponent>[][] components = getGridComponents((RadGridLayoutComponent)component);
    final MetaModel tableRowModel = ViewsMetaManager.getInstance(component.getTag().getProject()).getModelByTag("TableRow");

    return new ComponentMorphingTool(component, new RadTableLayoutComponent(), target, new RadTableLayout()) {
      @Override
      protected void convertChildren() throws Exception {
        for (List<RadComponent>[] rowComponents : components) {
          RadViewComponent newRowComponent = RadComponentOperations.createComponent(null, tableRowModel);
          RadComponentOperations.addComponent(myNewComponent, newRowComponent, null);

          for (List<RadComponent> cellComponents : rowComponents) {
            if (cellComponents != null) {
              for (RadComponent cellComponent : cellComponents) {
                RadComponentOperations.moveComponent(newRowComponent, (RadViewComponent)cellComponent, null);
              }
            }
          }
        }
      }

      @Override
      protected void convertTag() {
        XmlTag tag = myNewComponent.getTag();
        RadComponentOperations.deleteAttribute(tag, "rowCount");
        RadComponentOperations.deleteAttribute(tag, "columnCount");

        for (RadComponent rowComponent : myNewComponent.getChildren()) {
          for (RadComponent cellComponent : rowComponent.getChildren()) {
            XmlTag cellTag = ((RadViewComponent)cellComponent).getTag();
            RadComponentOperations.deleteAttribute(cellTag, "layout_row");
            RadComponentOperations.deleteAttribute(cellTag, "layout_rowSpan");

            XmlAttribute attribute = cellTag.getAttribute("layout_columnSpan", SdkConstants.NS_RESOURCES);
            if (attribute != null) {
              attribute.setName(attribute.getNamespacePrefix() + ":layout_span");
            }
          }
        }
      }

      @Override
      protected void loadChildProperties(PropertyParser propertyParser) throws Exception {
      }
    }.result();
  }

  private static List<RadComponent>[][] getGridComponents(RadGridLayoutComponent parent) {
    GridInfo gridInfo = parent.getGridInfo();
    List<RadComponent>[][] components = new List[gridInfo.rowCount][gridInfo.columnCount];

    for (RadComponent child : parent.getChildren()) {
      Rectangle cellInfo = RadGridLayoutComponent.getCellInfo(child);
      List<RadComponent> cellComponents = components[cellInfo.y][cellInfo.x];
      if (cellComponents == null) {
        cellComponents = new ArrayList<RadComponent>();
        components[cellInfo.y][cellInfo.x] = cellComponents;
      }
      cellComponents.add(child);
    }

    return components;
  }
}