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 java.util.ArrayList;

import java.util.List;

import javax.swing.ButtonGroup;

import javax.swing.DefaultListModel;

import javax.swing.ImageIcon;

import javax.swing.JComponent;

import javax.swing.ListCellRenderer;

import javax.swing.ListModel;

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;

import static org.gvsig.fmap.dal.swing.impl.DefaultDALSwingLibrary.LIBRARY_NAME;

import org.gvsig.fmap.dal.swing.impl.expressionevaluator.FeatureStoreElementFactory;

import org.gvsig.tools.locator.ReferenceNotRegisteredException;

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

/**

 *

 * @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() {

        int indexOrder = this.lstOrderColumns.getSelectedIndex();

        ListModel modelOrder = this.lstOrderColumns.getModel();

        this.btnColumnUp.setEnabled(indexOrder >= 1);

        this.btnColumnDown.setEnabled(indexOrder >= 0 && indexOrder < modelOrder.getSize() - 1);

        if (indexOrder >= 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;

    }

    int selectedIndex = this.lstOrderColumns.getSelectedIndex();

    if(selectedIndex<1){

        return;

    }

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

    this.lstOrderColumns.setSelectedIndex(selectedIndex-1);

    this.updateControls();

  }

  private void doColumnDown() {

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

    if (attr == null) {

      return;

    }

    int selectedIndex = this.lstOrderColumns.getSelectedIndex();

    if(selectedIndex>=this.lstOrderColumns.getModel().getSize()-1){

        return;

    }

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

    this.lstOrderColumns.setSelectedIndex(selectedIndex+1);

    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);

      }

    }

    List<FeatureQueryOrderMember> membersToRemove = new ArrayList<>();

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

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

      if (attr==null) {

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

      }

        if(attr == null){

            membersToRemove.add(member);

        } else {

            modelOrder.addElement(attr);

        }

    }

    for (FeatureQueryOrderMember member : membersToRemove) {

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

    }

    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);

      this.btnColumnUp.setEnabled(indexOrder >= 1);

      this.btnColumnDown.setEnabled(indexOrder < modelOrder.getSize()-1);

      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);

      this.btnColumnUp.setEnabled(false);

      this.btnColumnDown.setEnabled(false);

    }

  }

  @Override

  public ImageIcon loadImage(String imageName) {

        return ToolsSwingUtils.loadImage(this, imageName);

  }

  public static void selfRegister() {

//    boolean n = ToolsSwingUtils.registerIcons(

//            DefaultFeatureQueryOrderPanel.class,

//            null,

//            LIBRARY_NAME,

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

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

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

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

//    );

  }

}