Application: gvSIG desktop

package org.gvsig.fmap.dal.swing.impl.featurequery;

import java.awt.Dimension;

import java.awt.event.ActionEvent;

import java.net.URL;

import javax.swing.ButtonGroup;

import javax.swing.DefaultListModel;

import javax.swing.ImageIcon;

import javax.swing.JComponent;

import javax.swing.ListCellRenderer;

import javax.swing.event.ListSelectionEvent;

import org.apache.commons.io.FilenameUtils;

import org.gvsig.fmap.dal.exception.DataException;

import org.gvsig.fmap.dal.feature.FeatureAttributeDescriptor;

import org.gvsig.fmap.dal.feature.FeatureQuery;

import org.gvsig.fmap.dal.feature.FeatureQueryOrder.FeatureQueryOrderMember;

import org.gvsig.fmap.dal.feature.FeatureStore;

import org.gvsig.fmap.dal.feature.FeatureType;

import org.gvsig.fmap.dal.swing.DALSwingLocator;

import org.gvsig.tools.swing.api.ActionListenerSupport;

import org.gvsig.tools.swing.api.FilteredListController;

import org.gvsig.tools.swing.api.FilteredListModel;

import org.gvsig.tools.swing.api.ToolsSwingLocator;

import org.gvsig.tools.swing.api.ToolsSwingManager;

import org.gvsig.tools.swing.icontheme.IconTheme;

import org.gvsig.fmap.dal.swing.featurequery.FeatureQueryOrderPanel;

/**

 *

 * @author jjdelcerro

 */

public class DefaultFeatureQueryOrderPanel

        extends DefaultFeatureQueryOrderPanelView

        implements FeatureQueryOrderPanel {

  private FeatureQuery query;

  private FeatureType featureType;

  private final ActionListenerSupport actionListenerSupport;

  private final ButtonGroup ascendingGroup;

  private FilteredListController availableFieldsController;

  public DefaultFeatureQueryOrderPanel() {

    this.actionListenerSupport = ToolsSwingLocator.getToolsSwingManager().createActionListenerSupport();

    this.ascendingGroup = new ButtonGroup();

    this.initComponents();

  }

  @Override

  public FeatureQuery fetch(FeatureQuery query) {

    if (query == null) {

      return this.query.getCopy();

    }

    query.getOrder().copyFrom(this.query.getOrder());

    return query;

  }

  @Override

  public FeatureQuery fetch() {

    return this.fetch(null);

  }

  @Override

  public void put(FeatureQuery query) {

    this.query.copyFrom(query);

    this.updateControls();

  }

  private void initComponents() {

    ToolsSwingManager toolsSwingManager = ToolsSwingLocator.getToolsSwingManager();

    toolsSwingManager.translate(this.btnColumnAdd);

    toolsSwingManager.translate(this.btnColumnDown);

    toolsSwingManager.translate(this.btnColumnRemove);

    toolsSwingManager.translate(this.btnColumnSortAscending);

    toolsSwingManager.translate(this.btnColumnSortDescending);

    toolsSwingManager.translate(this.btnColumnUp);

    toolsSwingManager.translate(this.btnColumnsFilter);

    ListCellRenderer renderer = DALSwingLocator.getDataSwingManager().createDefaultFeatureAttributeListCellRenderer();

    this.lstAvailableColumns.setCellRenderer(renderer);

    this.lstOrderColumns.setCellRenderer(renderer);

    this.ascendingGroup.add(this.btnColumnSortAscending);

    this.ascendingGroup.add(this.btnColumnSortDescending);

    this.lstAvailableColumns.addListSelectionListener((ListSelectionEvent e) -> {

      if (e.getValueIsAdjusting()) {

        return;

      }

      doSelectAvailableColumn();

    });

    this.lstOrderColumns.addListSelectionListener((ListSelectionEvent e) -> {

      if (e.getValueIsAdjusting()) {

        return;

      }

      doSelectOrderColumn();

    });

    this.btnColumnAdd.addActionListener((ActionEvent e) -> {

      doColumnAdd();

    });

    this.btnColumnRemove.addActionListener((ActionEvent e) -> {

      doColumnRemove();

    });

    this.btnColumnDown.addActionListener((ActionEvent e) -> {

      doColumnDown();

    });

    this.btnColumnUp.addActionListener((ActionEvent e) -> {

      doColumnUp();

    });

    this.btnColumnSortAscending.addActionListener((ActionEvent e) -> {

      doColumnSortAscending();

    });

    this.btnColumnSortDescending.addActionListener((ActionEvent e) -> {

      doColumnSortDescending();

    });

    this.availableFieldsController = toolsSwingManager.createFilteredListController(

            lstAvailableColumns,

            txtColumnsFilter,

            btnColumnsFilter

    );

    Dimension sz = this.getPreferredSize();

    if (sz.width < DEFAULT_WIDTH) {

      sz.width = DEFAULT_WIDTH;

    }

    if (sz.height < DEFAULT_HEIGHT) {

      sz.height = DEFAULT_HEIGHT;

    }

    this.setPreferredSize(sz);

  }

  private void doSelectAvailableColumn() {

    if (this.lstAvailableColumns.getSelectedIndex() >= 0) {

      this.btnColumnAdd.setEnabled(true);

      return;

    }

    this.btnColumnAdd.setEnabled(false);

  }

  private void doSelectOrderColumn() {

    if (this.lstOrderColumns.getSelectedIndex() >= 0) {

      FeatureAttributeDescriptor attr = (FeatureAttributeDescriptor) this.lstOrderColumns.getSelectedValue();

      if (attr != null) {

        FeatureQueryOrderMember member = this.query.getOrder().get(attr.getName());

        if (member != null) {

          this.btnColumnRemove.setEnabled(true);

          this.btnColumnDown.setEnabled(true);

          this.btnColumnUp.setEnabled(true);

          this.btnColumnSortAscending.setEnabled(true);

          this.btnColumnSortDescending.setEnabled(true);

          if (member.getAscending()) {

            this.btnColumnSortAscending.setSelected(true);

            this.btnColumnSortDescending.setSelected(false);

          } else {

            this.btnColumnSortAscending.setSelected(false);

            this.btnColumnSortDescending.setSelected(true);

          }

          return;

        }

      }

    }

    this.btnColumnRemove.setEnabled(false);

    this.btnColumnDown.setEnabled(false);

    this.btnColumnUp.setEnabled(false);

    this.btnColumnSortAscending.setEnabled(false);

    this.btnColumnSortDescending.setEnabled(false);

    this.btnColumnSortAscending.setSelected(false);

    this.btnColumnSortDescending.setSelected(false);

  }

  private void doColumnSortAscending() {

    FeatureAttributeDescriptor attr = (FeatureAttributeDescriptor) this.lstOrderColumns.getSelectedValue();

    if (attr == null) {

      return;

    }

    FeatureQueryOrderMember member = this.query.getOrder().get(attr.getName());

    member.setAscending(true);

  }

  private void doColumnSortDescending() {

    FeatureAttributeDescriptor attr = (FeatureAttributeDescriptor) this.lstOrderColumns.getSelectedValue();

    if (attr == null) {

      return;

    }

    FeatureQueryOrderMember member = this.query.getOrder().get(attr.getName());

    member.setAscending(false);

  }

  private void doColumnUp() {

    FeatureAttributeDescriptor attr = (FeatureAttributeDescriptor) this.lstOrderColumns.getSelectedValue();

    if (attr == null) {

      return;

    }

    this.query.getOrder().movePrevious(attr.getName());

    this.updateControls();

  }

  private void doColumnDown() {

    FeatureAttributeDescriptor attr = (FeatureAttributeDescriptor) this.lstOrderColumns.getSelectedValue();

    if (attr == null) {

      return;

    }

    this.query.getOrder().moveNext(attr.getName());

    this.updateControls();

  }

  private void doColumnRemove() {

    FeatureAttributeDescriptor attr = (FeatureAttributeDescriptor) this.lstOrderColumns.getSelectedValue();

    if (attr == null) {

      return;

    }

    FeatureQueryOrderMember member = this.query.getOrder().get(attr.getName());

    this.query.getOrder().remove(member);

    this.updateControls();

  }

  private void doColumnAdd() {

    FeatureAttributeDescriptor attr = (FeatureAttributeDescriptor) this.lstAvailableColumns.getSelectedValue();

    if (attr == null) {

      return;

    }

    this.query.getOrder().add(attr.getName(), true);

    this.updateControls();

  }

  @Override

  public void setStore(FeatureStore store) {

    try {

      this.featureType = store.getDefaultFeatureType();

      this.query = store.createFeatureQuery();

    } catch (DataException ex) {

      throw new RuntimeException("Can't assign store", ex);

    }

  }

  @Override

  public JComponent asJComponent() {

    return this;

  }

  private void updateControls() {

    int indexAvailables = this.lstAvailableColumns.getSelectedIndex();

    int indexOrder = this.lstOrderColumns.getSelectedIndex();

    FilteredListModel modelAvailables = ToolsSwingLocator.getToolsSwingManager().createFilteredListModel();

    DefaultListModel<FeatureAttributeDescriptor> modelOrder = new DefaultListModel<>();

    for (FeatureAttributeDescriptor attr : featureType) {

      if (this.query.getOrder() == null || !this.query.getOrder().contains(attr.getName())) {

        modelAvailables.addElement(attr);

      }

    }

    for (FeatureAttributeDescriptor attr : this.query.getExtraColumn().getColumns()) {

      if (this.query.getOrder() == null || !this.query.getOrder().contains(attr.getName())) {

        modelAvailables.addElement(attr);

      }

    }

    for (FeatureQueryOrderMember member : this.query.getOrder().members()) {

      FeatureAttributeDescriptor attr = this.featureType.getAttributeDescriptor(member.getAttributeName());

      if (attr==null) {

          attr = this.query.getExtraColumn().get(member.getAttributeName());

      }

      modelOrder.addElement(attr);

    }

    modelAvailables.setFilter(this.txtColumnsFilter.getText());

    modelAvailables.sort(true);

    this.lstAvailableColumns.setModel(modelAvailables);

    this.lstOrderColumns.setModel(modelOrder);

    if (indexAvailables >= 0 && modelAvailables.getSize() > 0) {

      if (indexAvailables >= modelAvailables.getSize()) {

        indexAvailables = modelAvailables.getSize() - 1;

      }

      this.lstAvailableColumns.setSelectedIndex(indexAvailables);

      this.btnColumnAdd.setEnabled(true);

    } else {

      this.btnColumnAdd.setEnabled(false);

    }

    if (indexOrder >= 0 && modelOrder.getSize() > 0) {

      if (indexOrder >= modelOrder.getSize()) {

        indexOrder = modelOrder.getSize() - 1;

      }

      this.lstOrderColumns.setSelectedIndex(indexOrder);

      FeatureAttributeDescriptor attr = modelOrder.get(indexOrder);

      FeatureQueryOrderMember member = this.query.getOrder().get(attr.getName());

      if (member == null) {

        this.btnColumnSortAscending.setSelected(true);

        this.btnColumnSortDescending.setSelected(false);

        this.btnColumnSortAscending.setEnabled(false);

        this.btnColumnSortDescending.setEnabled(false);

      } else {

        this.btnColumnSortAscending.setEnabled(true);

        this.btnColumnSortDescending.setEnabled(true);

        if (member.getAscending()) {

          this.btnColumnSortAscending.setSelected(true);

          this.btnColumnSortDescending.setSelected(false);

        } else {

          this.btnColumnSortAscending.setSelected(false);

          this.btnColumnSortDescending.setSelected(true);

        }

      }

      this.btnColumnRemove.setEnabled(true);

    } else {

      this.btnColumnSortAscending.setSelected(true);

      this.btnColumnSortDescending.setSelected(false);

      this.btnColumnSortAscending.setEnabled(false);

      this.btnColumnSortDescending.setEnabled(false);

      this.btnColumnRemove.setEnabled(false);

    }

  }

  @Override

  public ImageIcon loadImage(String imageName) {

    String name = FilenameUtils.getBaseName(imageName);

    IconTheme theme = ToolsSwingLocator.getIconThemeManager().getDefault();

    if (theme.exists(name)) {

      return theme.get(name);

    }

    URL url = this.getClass().getResource(name + ".png");

    if (url == null) {

      return null;

    }

    return new ImageIcon(url);

  }

  public static void selfRegister() {

    String[][] iconNames = new String[][]{

      new String[]{"dalswing", "featurequery-column-add-arrow"},

      new String[]{"dalswing", "featurequery-column-remove-arrow"},

      new String[]{"dalswing", "featurequery-column-down"},

      new String[]{"dalswing", "featurequery-column-up"}

    };

    IconTheme theme = ToolsSwingLocator.getIconThemeManager().getCurrent();

    for (String[] icon : iconNames) {

      URL url = DefaultFeatureQueryOrderPanel.class.getResource(icon[1] + ".png");

      theme.registerDefault("DALSwing", icon[0], icon[1], null, url);

    }

  }

}