gvSIG projects pool

/*

 * gvSIG. Desktop Geographic Information System.

 * 

 * Copyright (C) 2007-2020 gvSIG Association.

 * 

 * This program is free software; you can redistribute it and/or

 * modify it under the terms of the GNU General Public License

 * as published by the Free Software Foundation; either version 3

 * of the License, or (at your option) any later version.

 * 

 * This program is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 * GNU General Public License for more details.

 * 

 * You should have received a copy of the GNU General Public License 

 * along with this program. If not, see <https://www.gnu.org/licenses/>. 

 * 

 * For any additional information, do not hesitate to contact us

 * at info AT gvsig.com, or visit our website www.gvsig.com.

 */

package org.gvsig.vcsgis.swing.impl.changes;

import java.awt.Dimension;

import java.awt.event.ActionEvent;

import java.awt.event.ActionListener;

import java.awt.event.KeyAdapter;

import java.awt.event.KeyEvent;

import java.sql.Timestamp;

import java.util.Date;

import java.util.HashSet;

import java.util.Iterator;

import java.util.Set;

import java.util.logging.Level;

import java.util.logging.Logger;

import javax.swing.JButton;

import javax.swing.JLabel;

import javax.swing.JTable;

import javax.swing.JTextField;

import javax.swing.ListSelectionModel;

import javax.swing.event.ChangeEvent;

import javax.swing.event.ListSelectionEvent;

import javax.swing.event.TableModelEvent;

import javax.swing.event.TableModelListener;

import javax.swing.table.TableModel;

import org.gvsig.featureform.swing.CreateJFeatureFormException;

import org.gvsig.featureform.swing.JFeatureForm;

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

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

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

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

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

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

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

import org.gvsig.tools.swing.api.windowmanager.WindowManager;

import static org.gvsig.vcsgis.lib.VCSGisManager.OP_ADD_ENTITY;

import static org.gvsig.vcsgis.lib.VCSGisManager.OP_DELETE;

import org.gvsig.vcsgis.lib.workspace.VCSGisWorkspace;

import org.gvsig.vcsgis.lib.workspace.VCSGisWorkspaceChange;

import org.gvsig.vcsgis.lib.workspace.VCSGisWorkspaceChange;

import org.gvsig.vcsgis.lib.workspace.VCSGisWorkspaceChanges;

/**

 *

 * @author gvSIG Team

 */

public class LocalChangesController {

    private final JTable tblLocalChanges;

    private final JButton btnLocalCheckAll;

    private final JButton btnLocalUnCheckAll;

    private final JButton btnLocalShowForm;

    private final JButton btnLocalRefresh;

    private final JButton btnLocalCommit;

    private final JButton btnLocalRevert;

    private final JLabel lblLocalEffectiveDate;

    private final JTextField txtLocalEffectiveDate;

    private final JTextField txtLocalComment;

    private final JButton btnLocalEffectiveDate;

    private final JLabel lblLocalComment;

    private final PickerController<VCSGisWorkspace> workspacePicker;    

    private VCSGisWorkspaceChanges<VCSGisWorkspaceChange> changes;

    private ChangesTableModel localChangesTableModel;

    private DatePickerController effectiveDatePicker;

    private final Set<String> entitiesToUnCheckAll;

    private TableModelListener tableModelListener;

    public LocalChangesController(

        PickerController<VCSGisWorkspace> workspacePicker,

        JTable tblLocalChanges,

        JButton btnLocalCheckAll,

        JButton btnLocalUnCheckAll,

        JButton btnLocalShowForm,

        JButton btnLocalRefresh,

        JButton btnLocalCommit,

        JButton btnLocalRevert,

        JLabel lblLocalEffectiveDate,

        JTextField txtLocalEffectiveDate,

        JTextField txtLocalComment,

        JButton btnLocalEffectiveDate,

        JLabel lblLocalComment

        ) {

        this.workspacePicker = workspacePicker;

        this.tblLocalChanges = tblLocalChanges;

        this.btnLocalCheckAll = btnLocalCheckAll;

        this.btnLocalCommit = btnLocalCommit;

        this.btnLocalEffectiveDate = btnLocalEffectiveDate;

        this.btnLocalRefresh = btnLocalRefresh;

        this.btnLocalRevert = btnLocalRevert;

        this.btnLocalShowForm = btnLocalShowForm;

        this.btnLocalUnCheckAll = btnLocalUnCheckAll;

        this.lblLocalComment = lblLocalComment;

        this.lblLocalEffectiveDate = lblLocalEffectiveDate;

        this.txtLocalComment = txtLocalComment;

        this.txtLocalEffectiveDate = txtLocalEffectiveDate;

        this.entitiesToUnCheckAll = new HashSet<>();

        initComponents();

    }

    private void initComponents() {

        ToolsSwingManager toolsSwingManager = ToolsSwingLocator.getToolsSwingManager();

        translate();

        this.workspacePicker.addChangeListener((ChangeEvent e) -> {

            doUpdateTableLocalChanges();

        });

        this.tblLocalChanges.getSelectionModel().setSelectionMode(ListSelectionModel.MULTIPLE_INTERVAL_SELECTION);

        this.tblLocalChanges.addKeyListener(new KeyAdapter() {

            @Override

            public void keyPressed(KeyEvent e) {

                if (e.getKeyCode() == KeyEvent.VK_SPACE) {

                    doToggleSelection();

                }

            }

        });

        this.tblLocalChanges.getSelectionModel().addListSelectionListener((ListSelectionEvent e) -> {

            doUpdateEnabledComponents();

        });

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

            if (changes != null) {

                doCheckAll();

                doUpdateTableLocalChanges();

            }

        });

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

            if (changes != null) {

                doUnCheckAll();

                doUpdateTableLocalChanges();

            }

        });

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

            doShowForm();

        });

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

            doCommit();

        });

        this.effectiveDatePicker = toolsSwingManager.createDatePickerController(

                this.txtLocalEffectiveDate,

                this.btnLocalEffectiveDate

        );

        this.effectiveDatePicker.set(new Date());

        this.btnLocalRefresh.addActionListener(new ActionListener() {

            @Override

            public void actionPerformed(ActionEvent e) {

                doUpdateTableLocalChanges();

            }

        });

        toolsSwingManager.addClearButton(this.txtLocalComment);

        toolsSwingManager.setDefaultPopupMenu(txtLocalComment);

        doUpdateEnabledComponents();

    }

    private void translate() {

        ToolsSwingManager swingManager = ToolsSwingLocator.getToolsSwingManager();

        swingManager.translate(this.btnLocalCheckAll);

        swingManager.translate(this.btnLocalCommit);

        swingManager.translate(this.btnLocalRefresh);

        swingManager.translate(this.btnLocalRevert);

        swingManager.translate(this.btnLocalShowForm);

        swingManager.translate(this.btnLocalUnCheckAll);

        swingManager.translate(this.lblLocalEffectiveDate);

        swingManager.translate(this.lblLocalComment);

    }

    private Iterator<Long> getSelectionIterator() {

        ListSelectionModel selection = this.tblLocalChanges.getSelectionModel();

        return new Iterator<Long>() {

            long n = selection.getMinSelectionIndex();

            @Override

            public boolean hasNext() {

                while (n <= selection.getMaxSelectionIndex()){

                    if(selection.isSelectedIndex((int) n)){

                       return true; 

                    }

                    n++;

                }

                return false;

            }

            @Override

            public Long next() {

                if(n > selection.getMaxSelectionIndex()){

                    throw new IllegalStateException();

                }

                return n++;

            }

        };

    }

    private void doToggleSelection() {

        Iterator<Long> rows = getSelectionIterator();

        this.changes.process(rows, (VCSGisWorkspaceChange change) -> {

            if(change.getOperation() == OP_ADD_ENTITY) {

                if(change.isSelected()){

                    entitiesToUnCheckAll.add(change.getEntityCode());

                }

            }

            change.setSelected(!change.isSelected());

            return true;

        });

        if(!entitiesToUnCheckAll.isEmpty()){

            doUnCheckAllByEntities();

        }

    }

    private Iterator<Long> getRowsByEntitiesToUnCheckIterator() {

        long size = changes.size64();

        TableModel model = this.tblLocalChanges.getModel();

        return new Iterator<Long>() {

            long n = 0;

            @Override

            public boolean hasNext() {

                while (n < size){

                    VCSGisWorkspaceChange row = changes.get64(n);

                    if(entitiesToUnCheckAll.contains(row.getEntityCode())){

                       return true; 

                    }

                    n++;

                }

                return false;

            }

            @Override

            public Long next() {

                return n++;

            }

        };

    }

    private void doUnCheckAllByEntities(){

        if(!entitiesToUnCheckAll.isEmpty()) {

            Iterator<Long> rows = getRowsByEntitiesToUnCheckIterator();

            this.changes.process(rows, (VCSGisWorkspaceChange change) -> {

                 if(change.getOperation() != OP_ADD_ENTITY) {

                    change.setSelected(false);

                 }

                return true;

            });

            entitiesToUnCheckAll.clear();

        }

    }

    private Iterator<Long> getRowsIterator() {

        long size = changes.size64();

        return new Iterator<Long>() {

            long n = 0;

            @Override

            public boolean hasNext() {

                return (n < size);

            }

            @Override

            public Long next() {

                return n++;

            }

        };

    }

    private void doCheckAll() {

        Iterator<Long> rows = getRowsIterator();

        this.changes.process(rows, (VCSGisWorkspaceChange change) -> {

            change.setSelected(true);

            return true;

        });

    }

    private void doUnCheckAll() {

        Iterator<Long> rows = getRowsIterator();

        this.changes.process(rows, (VCSGisWorkspaceChange change) -> {

             change.setSelected(false);

            return true;

        });

    }

    public VCSGisWorkspace getWorkspace() {

        return this.workspacePicker.get();

    }

    private void doUpdateEnabledComponents() {

        int selectedRowCount = this.tblLocalChanges.getSelectedRowCount();

        btnLocalCheckAll.setEnabled(changes!=null && changes.size64()>0);

        btnLocalUnCheckAll.setEnabled(changes!=null && changes.size64()>0);

        btnLocalRefresh.setEnabled(changes!=null);

        btnLocalRevert.setEnabled(false);      

        btnLocalRevert.setVisible(false);      

        btnLocalCommit.setEnabled(changes!=null && !this.changes.isSelectionEmpty());

        //Show form

        int row = this.tblLocalChanges.getSelectedRow();

        if(row>=0){

            VCSGisWorkspaceChange change = this.changes.get64(row);

            if(change.getOperation() != OP_ADD_ENTITY){

                btnLocalShowForm.setEnabled(true);

                return;

            }

        }

        btnLocalShowForm.setEnabled(false);

    }

    private void doUpdateTableLocalChanges() {

        VCSGisWorkspace ws = this.getWorkspace();

        if(ws != null){

            changes = ws.getLocalChanges();

            localChangesTableModel = new ChangesTableModel(changes, ws);

            this.tblLocalChanges.setModel(localChangesTableModel);

            this.tableModelListener = null;

            localChangesTableModel.addTableModelListener(getTableModelListener());

            this.btnLocalCheckAll.setEnabled(true);

            this.btnLocalUnCheckAll.setEnabled(true);

            this.btnLocalCommit.setEnabled(true);

            this.btnLocalRefresh.setEnabled(true);

            this.btnLocalShowForm.setEnabled(true);

        } else {

            this.btnLocalCheckAll.setEnabled(false);

            this.btnLocalUnCheckAll.setEnabled(false);

            this.btnLocalCommit.setEnabled(false);

            this.btnLocalRefresh.setEnabled(false);

            this.btnLocalShowForm.setEnabled(false);

        }

        doUpdateEnabledComponents();

    }

    private TableModelListener getTableModelListener() {

        if(tableModelListener == null){

            tableModelListener =  (TableModelEvent e) -> {

                if (e.getType() == TableModelEvent.UPDATE) {

                    TableModel model = ((TableModel) (e.getSource()));

                    VCSGisWorkspaceChange change = changes.get64(e.getFirstRow());

                    if(change.getOperation() == OP_ADD_ENTITY) {

                        if(!change.isSelected()) {

                            if (entitiesToUnCheckAll.add(change.getEntityCode())) {

                                doUnCheckAllByEntities();

                            }

                        }

                    }

                }

            };

        }

        return tableModelListener;

    }

    private void doShowForm() {

        int row = this.tblLocalChanges.getSelectedRow();

        VCSGisWorkspaceChange change = this.changes.get64(row);

        if(change.getOperation() == OP_DELETE || change.getOperation() == OP_ADD_ENTITY){

            return;

        }

        Feature feature = change.getRelatedFeature();

        if(feature == null){

            return;

        }

        try {

            DataSwingManager dataSwingManager = DALSwingLocator.getDataSwingManager();

            JFeatureForm form;

            form = dataSwingManager.createJFeatureForm(feature);

            WindowManager winManager = ToolsSwingLocator.getWindowManager();

            form.asJComponent().setPreferredSize(new Dimension(400,200));

            winManager.showWindow(form.asJComponent(), "Local change: "+change.getLabel(), WindowManager.MODE.WINDOW);

        } catch (CreateJFeatureFormException ex) {

            Logger.getLogger(VCSGisJChangesImpl.class.getName()).log(Level.SEVERE, null, ex);

        }

    }

    private void doCommit() {

        Timestamp effectiveDate = new Timestamp(effectiveDatePicker.get().getTime());

        getWorkspace().commit(effectiveDate, this.txtLocalComment.getText(), null);

        doUpdateTableLocalChanges();

    }

}