Android Open Source - spades Column Selector

From Project

Back to project page spades.
License

The source code is released under:
Apache License
If you think the Android project spades listed in this page is inappropriate, such as containing malicious code/tools or violating the copyright, please email info at java2s dot com, thanks.
Java Source Code

/**
 * Copyright (C) 2013 Pau Picas Sans <pau.picas@gmail.com>
 */*  w  w w.  ja  v a2 s  .  c  o m*/
 * 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 cat.picas.spades.query;

import java.util.ArrayList;
import java.util.List;

import cat.picas.spades.Column;
import cat.picas.spades.Table;
import cat.picas.spades.query.SelectedColumn.CountColumn;
import cat.picas.spades.query.SelectedColumn.RowIdColumn;

class ColumnSelector {

  private List<Table> mTables = new ArrayList<Table>();
  private List<Column> mColumns = new ArrayList<Column>();
  private List<SelectedColumn> mSelected = new ArrayList<SelectedColumn>();
  private boolean mAutoEntitiesId = true;
  private boolean mAutoRowsId = true;

  public ColumnSelector() {
  }

  public void addTable(Table table) {
    if (mTables.contains(table)) {
      throw new IllegalArgumentException("Table already selected");
    }
    mTables.add(table);
  }

  public boolean hasTable(Table table) {
    return mTables.contains(table);
  }

  public boolean hasTables() {
    return !mTables.isEmpty();
  }

  public void selectColumn(Column column) {
    checkColumn(column);
    mColumns.add(column);
    mSelected.add(new SelectedColumn(column));
  }

  public void selectColumn(String expression, Column... columns) {
    for (Column column : columns) {
      checkColumn(column);
    }
    mSelected.add(new SelectedColumn(expression, columns));
  }

  public List<SelectedColumn> getSelectedColumns() {
    List<SelectedColumn> selected = new ArrayList<SelectedColumn>();

    if (mSelected.isEmpty()) {
      for (Table table : mTables) {
        for (Column column : table.getColumns()) {
          selected.add(new SelectedColumn(column));
        }
      }
    } else {
      selected.addAll(mSelected);

      if (mAutoEntitiesId) {
        addAutoTableIds(selected);
      }
    }

    if (mAutoRowsId && !mTables.isEmpty()) {
      selected.add(new RowIdColumn(mTables.get(0).getColumnId()));
    }

    return selected;
  }

  public List<SelectedColumn> getSelectedColumnsForCount() {
    List<SelectedColumn> selected = new ArrayList<SelectedColumn>();

    selected.add(new CountColumn());
    for (SelectedColumn sel : mSelected) {
      if (sel.isCustom()) {
        selected.add(sel);
      }
    }

    return selected;
  }

  public boolean isAutoEntitiesId() {
    return mAutoEntitiesId;
  }

  public void setAutoEntitiesId(boolean autoEntitiesId) {
    mAutoEntitiesId = autoEntitiesId;
  }

  public boolean isAutoRowsId() {
    return mAutoRowsId;
  }

  public void setAutoRowsId(boolean autoRowsId) {
    mAutoRowsId = autoRowsId;
  }

  private void checkColumn(Column column) {
    if (!mTables.contains(column.getTable())) {
      throw new IllegalArgumentException("The selected Column does not belong to any table "
          + "added before");
    }
  }

  private void addAutoTableIds(List<SelectedColumn> selected) {
    List<SelectedColumn> addList = new ArrayList<SelectedColumn>();
    for (Table table : mTables) {
      if (!mColumns.contains(table.getColumnId())) {
        addList.add(new SelectedColumn(table.getColumnId()));
      }
    }
    selected.addAll(addList);
  }

}
Java Source Code List
