Application: gvSIG desktop » Library: org.gvsig.tools

package org.gvsig.tools.swing.impl.pickercontroller;

import java.awt.Color;

import static java.awt.Component.RIGHT_ALIGNMENT;

import java.awt.event.ActionEvent;

import java.awt.event.ActionListener;

import java.net.URL;

import java.text.DateFormat;

import java.text.ParseException;

import java.text.SimpleDateFormat;

import java.util.Date;

import java.util.Locale;

import javax.swing.ImageIcon;

import javax.swing.JButton;

import javax.swing.JFormattedTextField;

import javax.swing.JPopupMenu;

import javax.swing.event.DocumentEvent;

import javax.swing.event.DocumentListener;

import javax.swing.text.Document;

import javax.swing.text.JTextComponent;

import javax.swing.text.MaskFormatter;

import org.apache.commons.lang3.StringUtils;

import org.freixas.jcalendar.DateEvent;

import org.freixas.jcalendar.JCalendar;

import org.gvsig.tools.dataTypes.DataTypes;

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

import org.gvsig.tools.swing.api.pickercontroller.AbstractPickerController;

import org.gvsig.tools.swing.api.pickercontroller.DatePickerController;

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

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

/**

 *

 * @author jjdelcerro

 */

@SuppressWarnings("UseSpecificCatch")

public class DatePickerControllerImpl

        extends AbstractPickerController<Date>

        implements DatePickerController {

  private final JButton btnDate;

  private final JTextComponent txtDate;

  private JCalendar jcalendar;

  private DateFormat df;

  public static void selfRegister() {

    URL imageResource = DatePickerControllerImpl.class.getClassLoader().getResource("org/gvsig/tools/swing/picker/picker-date.png");

    if (imageResource != null) {

      IconThemeManager iconThemeManager = ToolsSwingLocator.getIconThemeManager();

      IconTheme theme = iconThemeManager.getCurrent();

      ImageIcon icon = new ImageIcon(imageResource);

      theme.registerDefault("tools", "picker", "picker-date", icon, imageResource);

    }

  }

  private final Document document;

  private final DocumentListener documentListener;

  private final Color colorOk;

  private final Color colorErr;

  private String emptyText;

  private MaskFormatter mask;

  private final ActionListener buttonListener;

  public DatePickerControllerImpl(JTextComponent txtDate, JButton btnDate) {

    this.txtDate = txtDate;

    if (btnDate == null) {

      btnDate = new JButton();

    }

    ToolsSwingLocator.getToolsSwingManager().addClearButton(txtDate, (ActionEvent e) -> {

      set(null);

    });

    this.df = DateFormat.getDateInstance(DateFormat.SHORT, Locale.getDefault());

    this.df.setLenient(false);

    if (this.txtDate instanceof JFormattedTextField) {

      JFormattedTextField ftf = (JFormattedTextField) this.txtDate;

      if (df instanceof SimpleDateFormat) {

        try {

          SimpleDateFormat sdf = (SimpleDateFormat) this.df;

          String s = sdf.toPattern(); 

          if( !s.contains("yyyy") ) {

            if( s.contains("yy") ) {

              s = s.replace("yy", "yyyy");

            } else if( s.contains("y") ) {

              s = s.replace("y", "yyyy");

            }

          }

          if( !s.contains("dd") ) {

            if( s.contains("d") ) {

              s = s.replace("d", "dd");

            }

          }

          if( !s.contains("MM") ) {

            if( s.contains("M") ) {

              s = s.replace("M", "MM");

            }

          }

          sdf.applyPattern(s);

          s = s.replace("y", "#");

          s = s.replace("d", "#");

          s = s.replace("M", "#");

          this.mask = new MaskFormatter(s);

          mask.setPlaceholderCharacter('_');

          this.emptyText = s.replace("#","_");

          mask.setAllowsInvalid(false);

          mask.install(ftf);

        } catch (Exception ex) {

          LOG.warn("Not been able to create mask", ex);

        }

      }

    }

    this.colorOk = this.txtDate.getBackground();

    this.colorErr = new Color(255,225,232);

    this.btnDate = btnDate;

    this.txtDate.setAlignmentX(RIGHT_ALIGNMENT);

    this.txtDate.setEditable(true);

    this.document = this.txtDate.getDocument();

    this.documentListener = new DocumentListener() {

      @Override

      public void insertUpdate(DocumentEvent e) {

        checkContents();

      }

      @Override

      public void removeUpdate(DocumentEvent e) {

        checkContents();

      }

      @Override

      public void changedUpdate(DocumentEvent e) {

        checkContents();

      }

    };

    document.addDocumentListener(this.documentListener);

    this.btnDate.setIcon(this.getIcon("picker-date"));

    if (this.btnDate.getText().equals("...")) {

      this.btnDate.setText("");

    }

    this.buttonListener = (ActionEvent e) -> {

      doShowCalendar();

    };

    this.btnDate.addActionListener(this.buttonListener);

  }

  @Override

  public void uninstall() {

    this.document.removeDocumentListener(documentListener);

    if( this.mask!=null ) {

      this.mask.uninstall();

    }

    this.txtDate.setBackground(colorOk);

    this.btnDate.removeActionListener(buttonListener);    

  }

  protected String getText() {

    String s = this.txtDate.getText();

    if( this.emptyText!=null && this.emptyText.equals(s) ) {

      return "";

    }

    s = s.replace("_", "");

    return s;

  }

  protected void checkContents() {

    if( this.isValid() ) {

      this.txtDate.setBackground(colorOk);

    } else {

      this.txtDate.setBackground(colorErr);

    }

  }

  @Override

  public boolean isValid() {

    if (StringUtils.isBlank(this.getText())) {

      return true;

    }

    try {

      Date date = this.df.parse(this.getText());

      return true;

    } catch (ParseException ex) {

      return false;

    }

  }

  @Override

  public boolean isEmpty() {

    if (StringUtils.isBlank(this.getText())) {

      return true;

    }

    try {

      Date date = this.df.parse(this.getText());

      return false;

    } catch (ParseException ex) {

      return true;

    }

  }

  @Override

  public Date get() {

    if (StringUtils.isBlank(this.getText())) {

      return null;

    }

    try {

      Date date = this.df.parse(this.getText());

      return date;

    } catch (ParseException ex) {

      return null;

    }

  }

  @Override

  public void set(Date date) {

    if( date == null ) {

      this.txtDate.setText("");

      return;

    }

    this.txtDate.setText(this.df.format(date));

  }

  @Override

  public void setEnabled(boolean enabled) {

    this.btnDate.setEnabled(enabled);

    this.txtDate.setEnabled(enabled);

  }

  @Override

  public boolean isEnabled() {

    return this.btnDate.isEnabled();

  }

  private void doShowCalendar() {

    if (!this.isEditable()) {

      return;

    }

    JPopupMenu menu = new JPopupMenu();

    menu.add(getJCalendar());

    Date v = new Date();

    if( !isEmpty() ) {

      v = this.get();

    }

    if (v != null) {

      getJCalendar().setDate(v);

    }

    menu.show(this.txtDate, 0, this.txtDate.getY() + this.txtDate.getPreferredSize().height);

  }

  protected JCalendar getJCalendar() {

    if (this.jcalendar == null) {

      this.jcalendar = new JCalendar();

      this.jcalendar.addDateListener((DateEvent arg0) -> {

        set(getJCalendar().getDate());

        fireChangeEvent();

      });

    }

    return (JCalendar) this.jcalendar;

  }

  @Override

  public void coerceAndSet(Object value) {

    this.set((Date) this.coerce(DataTypes.DATE, value, new Date()));

  }

  @Override

  public DateFormat getDateFormat() {

    return this.df;

  }

  @Override

  public MaskFormatter  getMask() {

      return this.mask;

  }

  @Override

  public void setEmptyText(String emptyText) {

    this.emptyText = emptyText;

  }

}