root/trunk/Moonlight/src/eu/moonlight3d/ml3d/modelling/objects/DuplicateObjectsCommand.java

/*
 * Moonlight|3D
 * Copyright (C) 2009 The Moonlight|3D team
 * 
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 * 
 * This library 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
 * Lesser General Public License for more details.
 * 
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
 *
 * Created on Nov 29, 2009
 */

package eu.moonlight3d.ml3d.modelling.objects;

import java.util.ArrayList;

import eu.moonlight3d.common.ui.command.Command;
import eu.moonlight3d.framework.State;
import eu.moonlight3d.framework.localization.L10N;
import eu.moonlight3d.framework.properties.BooleanValue;
import eu.moonlight3d.framework.properties.IntegerValue;
import eu.moonlight3d.framework.properties.StringArrayValue;
import eu.moonlight3d.framework.ui.ExceptionDialog;
import eu.moonlight3d.ml3d.backend.selection.Selection;
import eu.moonlight3d.ml3d.backend.threedim.document.Document;
import eu.moonlight3d.ml3d.backend.threedim.helper.OperatorGraphHelper;
import eu.moonlight3d.ml3d.backend.threedim.og.Manager;
import eu.moonlight3d.ml3d.backend.threedim.og.Node;
import eu.moonlight3d.ml3d.backend.threedim.og.Slot;

public class DuplicateObjectsCommand implements Command {
        private int numCopies;
        private boolean copyChildren;
        private String[] names;
        private String previousOGNodeName;
        private String previousSlotName;
        private String duplicateObjectsNodeName;
        
        public DuplicateObjectsCommand(int numCopies, boolean copyChildren) {
                this.numCopies=numCopies;
                this.copyChildren=copyChildren;
        }

        public boolean execute() {
        try {
            State state=State.getInstance();
            
            // don't do anything if selected object is not a SG node
            Selection selection = ((Document) state.getDocumentManager().getMainDocument()).getSelectionManager().getSelection("viewSelection");
            if(selection.getType()!=eu.moonlight3d.ml3d.backend.selection.Manager.getSelectionType("SGNode")) {
                state.logger().warning(L10N.translate("current selection must be a scene graph node"));
                return false;
            }
            
            String[] path=selection.getActiveObject().split("/");
            previousOGNodeName=path[1];
            previousSlotName=path[2];
            
            ArrayList<Object> objects = selection.getSelectionObjects();
            names=new String[objects.size()];
            for(int i=0;i<objects.size();i++) {
                names[i]=((eu.moonlight3d.ml3d.backend.threedim.sg.Node) objects.get(i)).getName();
            }
            
            redo();

            // clear selection to prevent rogue selection items
            selection.clearSelection(eu.moonlight3d.ml3d.backend.selection.Manager.getSelectionType("SGNode"));
            
            return true;
                } catch (Exception e) {
                        ExceptionDialog.run(e);
                }
                return false;
        }

        public String getBriefDescription() {
                return L10N.translate("Duplicate Objects");
        }

        public boolean isUndoable() {
                return true;
        }

        public void redo() {
        try {
            State state=State.getInstance();
            
            Manager ogManager=((Document) state.getDocumentManager().getMainDocument()).getOGManager();
            Node ogNode=ogManager.getNodeByName(previousOGNodeName);
            Slot slot=ogNode.getSlot(previousSlotName);
            ogNode=ogManager.createNodeByName("DuplicateObjects");
            duplicateObjectsNodeName=ogNode.getName();
            
            ogNode.getProperty("name").setValue(new StringArrayValue(names));
            ogNode.getProperty("copies").setValue(new IntegerValue(numCopies));
            ogNode.getProperty("copyChildren").setValue(new BooleanValue(copyChildren));
            
            // set links
            ogManager.insertNodeAfter(slot,ogNode.getSlot("input"),ogNode.getSlot("output"));

                        ogManager.update();
                } catch (Exception e) {
                        ExceptionDialog.run(e);
                }

        }

        public void undo() {
        try {
            State state=State.getInstance();
            Manager ogManager=((Document) state.getDocumentManager().getMainDocument()).getOGManager();
            Node operation=ogManager.getNodeByName(duplicateObjectsNodeName);
            if(operation.isDrain()) {
                OperatorGraphHelper.remove(duplicateObjectsNodeName);
            } else {
                OperatorGraphHelper.removeAndBridge(duplicateObjectsNodeName);
            }
            ogManager.update();
                } catch (Exception e) {
                        ExceptionDialog.run(e);
                }
        }

}