/* A representative of a ptolemy model

 Copyright (c) 1998-2014 The Regents of the University of California.
 All rights reserved.
 Permission is hereby granted, without written agreement and without
 license or royalty fees, to use, copy, modify, and distribute this
 software and its documentation for any purpose, provided that the above
 copyright notice and the following two paragraphs appear in all copies
 of this software.

 IN NO EVENT SHALL THE UNIVERSITY OF CALIFORNIA BE LIABLE TO ANY PARTY
 FOR DIRECT, INDIRECT, SPECIAL, INCIDENTAL, OR CONSEQUENTIAL DAMAGES
 ARISING OUT OF THE USE OF THIS SOFTWARE AND ITS DOCUMENTATION, EVEN IF
 THE UNIVERSITY OF CALIFORNIA HAS BEEN ADVISED OF THE POSSIBILITY OF
 SUCH DAMAGE.

 THE UNIVERSITY OF CALIFORNIA SPECIFICALLY DISCLAIMS ANY WARRANTIES,
 INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
 MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE. THE SOFTWARE
 PROVIDED HEREUNDER IS ON AN "AS IS" BASIS, AND THE UNIVERSITY OF
 CALIFORNIA HAS NO OBLIGATION TO PROVIDE MAINTENANCE, SUPPORT, UPDATES,
 ENHANCEMENTS, OR MODIFICATIONS.

 PT_COPYRIGHT_VERSION_2
 COPYRIGHTENDKEY
 */
package ptolemy.actor.gui;

import java.io.File;
import java.io.IOException;
import java.net.URI;
import java.net.URL;
import java.util.List;
import java.util.Locale;

import ptolemy.actor.Manager;
import ptolemy.actor.TypedCompositeActor;
import ptolemy.kernel.CompositeEntity;
import ptolemy.kernel.attributes.URIAttribute;
import ptolemy.kernel.util.Attribute;
import ptolemy.kernel.util.ChangeListener;
import ptolemy.kernel.util.ChangeRequest;
import ptolemy.kernel.util.IllegalActionException;
import ptolemy.kernel.util.NameDuplicationException;
import ptolemy.kernel.util.NamedObj;
import ptolemy.kernel.util.Workspace;
import ptolemy.moml.MoMLParser;
import ptolemy.moml.ParserAttribute;
import ptolemy.util.ClassUtilities;
import ptolemy.util.MessageHandler;
import ptolemy.util.StringUtilities;

///////////////////////////////////////////////////////////////////
//// PtolemyEffigy

/**
 An effigy for a Ptolemy II model.
 An effigy represents model metadata, and is contained by the
 model directory or by another effigy.  This class adds to the base
 class an association with a Ptolemy II model. The model, strictly
 speaking, is any Ptolemy II object (an instance of NamedObj).
 The Effigy class extends CompositeEntity, so an instance of Effigy
 can contain entities.  By convention, an effigy contains all
 open instances of Tableau associated with the model.

 @author Steve Neuendorffer and Edward A. Lee
 @version $Id$
 @since Ptolemy II 1.0
 @Pt.ProposedRating Green (eal)
 @Pt.AcceptedRating Yellow (janneck)
 */
public class PtolemyEffigy extends Effigy implements ChangeListener {
    /** Create a new effigy in the specified workspace with an empty string
     *  for its name.
     *  @param workspace The workspace for this effigy.
     */
    public PtolemyEffigy(Workspace workspace) {
        super(workspace);
    }

    /** Create a new effigy in the given container with the given name.
     *  @param container The container that contains this effigy.
     *  @param name The name of this effigy.
     *  @exception IllegalActionException If the entity cannot be contained
     *   by the proposed container.
     *  @exception NameDuplicationException If the name coincides with
     *   an entity already in the container.
     */
    public PtolemyEffigy(CompositeEntity container, String name)
            throws IllegalActionException, NameDuplicationException {
        super(container, name);
    }

    ///////////////////////////////////////////////////////////////////
    ////                         public methods                    ////

    /** React to the fact that a change has been successfully executed.
     *  This method does nothing.
     *  @param change The change that has been executed.
     */
    @Override
    public void changeExecuted(ChangeRequest change) {
        // Initially, the undo facility wanted us to call
        // setModified(true) here.  However, if we do, then running an
        // SDF Model that sets the bufferSize of a relation would
        // result in the model being marked as modified and the user
        // being queried about saving the model upon exit.
        // Needed for undo.
        //setModified(true);
    }

    /** React to the fact that a change has triggered an error by
     *  reporting the error in a top-level dialog.
     *  @param change The change that was attempted.
     *  @param exception The exception that resulted.
     */
    @Override
    public void changeFailed(ChangeRequest change, Exception exception) {
        // Do not report if it has already been reported.
        // NOTE: This method assumes that the context of the error handler
        // has been set so that there is an owner for the error window.
        if (change == null) {
            MessageHandler.error("Change failed: ", exception);
        } else if (!change.isErrorReported()) {
            change.setErrorReported(true);
            MessageHandler.error(
                    "Change failed: " + StringUtilities
                            .truncateString(change.getDescription(), 80, 1),
                    exception);
        }
    }

    /** Clone the object into the specified workspace. This calls the
     *  base class and then clones the associated model into a new
     *  workspace, if there is one.
     *  @param workspace The workspace for the new effigy.
     *  @return A new effigy.
     *  @exception CloneNotSupportedException If a derived class contains
     *   an attribute that cannot be cloned.
     */
    @Override
    public Object clone(Workspace workspace) throws CloneNotSupportedException {
        PtolemyEffigy newObject = (PtolemyEffigy) super.clone(workspace);

        if (_model != null && !(_model instanceof Configuration)) {
            newObject._model = (NamedObj) _model.clone(new Workspace());
        }

        return newObject;
    }

    /** Return the ptolemy model that this is an effigy of.
     *  @return The model, or null if none has been set.
     *  @see #setModel(NamedObj)
     */
    public NamedObj getModel() {
        return _model;
    }

    /** Return the effigy that is "in charge" of this effigy.
     *  In this base class, this returns the effigy associated
     *  with the top-level of the associated model. If there is
     *  no model, or it has no effigy, then delegate to the base
     *  class.
     *  @return The effigy associated with the top-level of the
     *   model.
     */
    @Override
    public Effigy masterEffigy() {
        if (_model != null) {
            NamedObj toplevel = _model.toplevel();

            if (toplevel == _model) {
                return this;
            }

            Effigy effigyForToplevel = Configuration.findEffigy(toplevel);

            if (effigyForToplevel != null) {
                return effigyForToplevel;
            }
        }

        return super.masterEffigy();
    }

    /** Set the ptolemy model that this is an effigy of.
     *  Register with that model as a change listener.
     *  @param model The model.
     *  @see #getModel()
     */
    public void setModel(NamedObj model) {
        if (_model != null) {
            _model.toplevel().removeChangeListener(this);
        }

        _model = model;

        if (model != null) {
            _model.toplevel().addChangeListener(this);
        }
    }

    /** Write the model associated with this effigy
     *  to the specified file in MoML format.
     *  Change the name of the model to match the
     *  file name, up to its first period.
     *  @param file The file to write to.
     *  @exception IOException If the write fails.
     */
    @Override
    public void writeFile(File file) throws IOException {
        java.io.FileWriter fileWriter = null;

        try {
            fileWriter = new java.io.FileWriter(file);

            String name = getModel().getName();

            String filename = file.getName();
            int period = filename.indexOf(".");

            if (period > 0) {
                name = filename.substring(0, period);
            } else {
                name = filename;
            }
            // If the user has a & in the file name . . .
            // See http://bugzilla.ecoinformatics.org/show_bug.cgi?id=3901
            name = StringUtilities.escapeForXML(name);

            // If the model is not at the top level,
            // then we have to force the writer to export
            // the DTD, because the exportMoML() method
            // will not do it.
            NamedObj model = getModel();
            if (model.getContainer() != null) {
                fileWriter.write("<?xml version=\"1.0\" standalone=\"no\"?>\n"
                        + "<!DOCTYPE " + _elementName + " PUBLIC "
                        + "\"-//UC Berkeley//DTD MoML 1//EN\"\n"
                        + "    \"http://ptolemy.eecs.berkeley.edu"
                        + "/xml/dtd/MoML_1.dtd\">\n");
            }
            model.exportMoML(fileWriter, 0, name);
        } finally {
            if (fileWriter != null) {
                fileWriter.close();
            }
        }
    }

    ///////////////////////////////////////////////////////////////////
    ////                         protected methods                 ////

    /** Check that the specified container is of a suitable class for
     *  this entity, i.e., ModelDirectory or PtolemyEffigy.
     *  @param container The proposed container.
     *  @exception IllegalActionException If the container is not of
     *   an acceptable class.
     */
    @Override
    protected void _checkContainer(CompositeEntity container)
            throws IllegalActionException {
        if (container != null && !(container instanceof ModelDirectory)
                && !(container instanceof PtolemyEffigy)) {
            throw new IllegalActionException(this, container,
                    "The container can only be set to an "
                            + "instance of ModelDirectory or PtolemyEffigy.");
        }
    }

    ///////////////////////////////////////////////////////////////////
    ////                         private members                   ////
    // The model associated with this effigy.
    private NamedObj _model;

    ///////////////////////////////////////////////////////////////////
    ////                         inner classes                     ////

    /** A factory for creating new Ptolemy effigies.
     */
    public static class Factory extends EffigyFactory {
        /** Create a factory with the given name and container.
         *  @param container The container.
         *  @param name The name.
         *  @exception IllegalActionException If the container is incompatible
         *   with this entity.
         *  @exception NameDuplicationException If the name coincides with
         *   an entity already in the container.
         */
        public Factory(CompositeEntity container, String name)
                throws IllegalActionException, NameDuplicationException {
            super(container, name);
        }

        ///////////////////////////////////////////////////////////////
        ////                     public methods                    ////

        /** Return true, indicating that this effigy factory is
         *  capable of creating an effigy without a URL being specified.
         *  @return True.
         */
        @Override
        public boolean canCreateBlankEffigy() {
            return true;
        }

        /** Create a new effigy in the given container by reading the
         *  <i>input</i> URL. If the <i>input</i> URL is null, then
         *  create a blank effigy.
         *  The blank effigy will have a new model associated with it.
         *  If this effigy factory contains an entity or an attribute
         *  named "blank", then
         *  the new model will be a clone of that object.  Otherwise,
         *  it will be an instance of TypedCompositeActor.
         *  If the URL does not end with extension ".xml" or ".moml"
         *  (case insensitive), then return null.  If the URL points
         *  to an XML file that is not
         *  a MoML file, then also return null. A MoML file is required
         *  to have the MoML DTD designation in the first five lines.
         *  That is, it must contain a line beginning with the string
         *  "&lt;!DOCTYPE" and ending with the string
         *  'PUBLIC \"-//UC Berkeley//DTD MoML"'.
         *  The specified base is used to expand any relative file references
         *  within the URL.
         *  If the input URL contains a "#", then the fragment after
         *  the "#" assumed to be a dot separated path to an inner model
         *  and the inner model is opened.
         *  @param container The container for the effigy.
         *  @param base The base for relative file references, or null if
         *   there are no relative file references.
         *  @param input The input URL.
         *  @return A new instance of PtolemyEffigy, or null if the URL
         *   does not specify a Ptolemy II model.
         *  @exception Exception If the URL cannot be read, or if the data
         *   is malformed in some way.
         */
        @Override
        public Effigy createEffigy(CompositeEntity container, URL base,
                URL input) throws Exception {
            if (input == null) {
                // Create a blank effigy.
                // Use the strategy pattern so derived classes can
                // override this.
                PtolemyEffigy effigy = _newEffigy(container,
                        container.uniqueName("effigy"));

                // If this factory contains an entity called "blank", then
                // clone that.
                NamedObj entity = getEntity("blank");
                Attribute attribute = getAttribute("blank");
                NamedObj newModel;

                if (entity != null) {
                    newModel = (NamedObj) entity.clone(new Workspace());

                    // The cloning process results an object that defers change
                    // requests.  By default, we do not want to defer change
                    // requests, but more importantly, we need to execute
                    // any change requests that may have been queued
                    // during cloning. The following call does that.
                    newModel.setDeferringChangeRequests(false);
                } else if (attribute != null) {
                    newModel = (NamedObj) attribute.clone(new Workspace());

                    // The cloning process results an object that defers change
                    // requests.  By default, we do not want to defer change
                    // requests, but more importantly, we need to execute
                    // any change requests that may have been queued
                    // during cloning. The following call does that.
                    newModel.setDeferringChangeRequests(false);
                } else {
                    newModel = new TypedCompositeActor(new Workspace());
                }

                // The model should have a parser associated with it
                // so that undo works. The following method will create
                // a parser, if there isn't one already.
                // We don't need the parser, so we ignore the return value.
                ParserAttribute.getParser(newModel);

                // The name might be "blank" which is confusing.
                // Set it to an empty string.  On Save As, this will
                // be changed to match the file name.
                newModel.setName("");
                effigy.setModel(newModel);
                return effigy;
            } else {
                String extension = getExtension(input)
                        .toLowerCase(Locale.getDefault());

                if (!extension.equals("xml") && !extension.equals("moml")) {
                    return null;
                }

                if (!checkForDTD(input, "<!DOCTYPE",
                        ".*PUBLIC \"-//UC Berkeley//DTD MoML.*")) {
                    return null;
                }

                // Create a blank effigy.
                PtolemyEffigy effigy = _newEffigy(container,
                        container.uniqueName("effigy"));

                MoMLParser parser = new MoMLParser();

                // Make sure that the MoMLParser._modified flag is reset
                // If we don't call reset here, then the second time
                // the code generator is run, we will be prompted to
                // save the model because the first time we ran
                // the code generator the model was marked as modified.
                parser.reset();

                NamedObj toplevel = null;

                try {
                    try {
                        long startTime = 0;
                        long endTime = 0;
                        // If the following fails, we should remove the effigy.
                        try {
                            //                          Report on the time it takes to open the model.
                            startTime = System.currentTimeMillis();
                            toplevel = parser.parse(base, input);
                            endTime = System.currentTimeMillis();
                        } catch (IOException io) {
                            // If we are running under Web Start, we
                            // might have a URL that refers to another
                            // jar file.
                            URL anotherURL = ClassUtilities
                                    .jarURLEntryResource(input.toString());

                            if (anotherURL != null) {
                                startTime = System.currentTimeMillis();
                                toplevel = parser.parse(base, anotherURL);
                                endTime = System.currentTimeMillis();
                            } else {
                                throw io;
                            }
                        }

                        if (toplevel != null) {
                            NamedObj model = toplevel;
                            int index = -1;
                            if ((index = input.toString().indexOf("#")) != -1) {
                                String fullName = input.toString().substring(
                                        index + 1, input.toString().length());
                                if (toplevel instanceof CompositeEntity) {
                                    model = ((CompositeEntity) toplevel)
                                            .getEntity(fullName);
                                }
                            }
                            try {
                                String entityClassName = StringUtilities
                                        .getProperty("entityClassName");
                                if ((entityClassName.length() > 0
                                        || endTime > startTime
                                                + Manager.minimumStatisticsTime)
                                        && model instanceof CompositeEntity) {
                                    System.out.println("Opened " + input
                                            + " in "
                                            + (System.currentTimeMillis()
                                                    - startTime)
                                            + " ms.");

                                    long statisticsStartTime = System
                                            .currentTimeMillis();
                                    System.out.println(((CompositeEntity) model)
                                            .statistics(entityClassName));
                                    long statisticsEndTime = System
                                            .currentTimeMillis();
                                    if (statisticsEndTime
                                            - statisticsStartTime > Manager.minimumStatisticsTime) {
                                        System.out.println(
                                                "Generating statistics took"
                                                        + (statisticsEndTime
                                                                - statisticsStartTime)
                                                        + " ms. ");
                                    }
                                }
                            } catch (SecurityException ex) {
                                System.err.println(
                                        "Warning, while trying to print timing statistics,"
                                                + " failed to read the entityClassName"
                                                + " property (-sandbox always causes this)");
                            }
                            effigy.setModel(model);

                            // A MoMLFilter may have modified the model
                            // as it was being parsed.
                            effigy.setModified(MoMLParser.isModified());

                            // The effigy will handle saving the modified
                            // moml for us, so MoMLParser need
                            // not care anymore.
                            MoMLParser.setModified(false);

                            // Identify the URI from which the model was read
                            // by inserting an attribute into both the model
                            // and the effigy.
                            URIAttribute uriAttribute = new URIAttribute(
                                    toplevel, "_uri");
                            URI inputURI = null;

                            try {
                                inputURI = new URI(input.toExternalForm());
                            } catch (java.net.URISyntaxException ex) {
                                // This is annoying, if the input has a space
                                // in it, then we cannot create a URI,
                                // but we could create a URL.
                                // If, under Windows, we call
                                // File.createTempFile(), then we are likely
                                // to get a pathname that has space.
                                // FIXME: Note that jar urls will barf if there
                                // is a %20 instead of a space.  This could
                                // cause problems in Web Start
                                String inputExternalFormFixed = StringUtilities
                                        .substitute(input.toExternalForm(), " ",
                                                "%20");

                                try {
                                    inputURI = new URI(inputExternalFormFixed);
                                } catch (Exception ex2) {
                                    throw new Exception("Failed to generate "
                                            + "a URI from '"
                                            + input.toExternalForm()
                                            + "' and from '"
                                            + inputExternalFormFixed + "'", ex);
                                }
                            }

                            uriAttribute.setURI(inputURI);

                            // This is used by TableauFrame in its
                            //_save() method.
                            effigy.uri.setURI(inputURI);

                            return effigy;
                        } else {
                            effigy.setContainer(null);
                        }
                    } catch (Throwable throwable) {
                        if (throwable instanceof StackOverflowError) {
                            Throwable newThrowable = new StackOverflowError(
                                    "StackOverflowError: "
                                            + "Which often indicates that a class "
                                            + "could not be found, but there was "
                                            + "possibly a moml file with that same "
                                            + "name in the directory that referred "
                                            + "to the class, so we got into a loop."
                                            + "For example: We had "
                                            + "actor/lib/joystick/Joystick.java "
                                            + "and "
                                            + "actor/lib/joystick/joystick.xml, "
                                            + "but "
                                            + "the .class file would not load "
                                            + "because of a classpath problem, "
                                            + "so we kept "
                                            + "loading joystick.xml which "
                                            + "referred to Joystick and because "
                                            + "of Windows "
                                            + "filename case insensitivity, "
                                            + "we found joystick.xml, which put "
                                            + "us in a loop.");
                            newThrowable.initCause(throwable);
                            throwable = newThrowable;
                        }

                        throwable.printStackTrace();

                        // The finally clause below can result in the
                        // application exiting if there are no other
                        // effigies open.  We check for that condition,
                        // and report the error here.  Otherwise, we
                        // pass the error to the caller.
                        ModelDirectory dir = (ModelDirectory) effigy.topEffigy()
                                .getContainer();
                        List effigies = dir.entityList(Effigy.class);

                        // We might get to here if we are running a
                        // vergil with a model specified as a command
                        // line argument and the model has an invalid
                        // parameter.
                        // We might have three effigies here:
                        // 1) .configuration.directory.configuration
                        // 2) .configuration.directory.UserLibrary
                        // 3) .configuration.directory.effigy
                        // Note that one of the effigies is the configuration
                        // itself, which does not prevent exiting the app.
                        // Hence, we handle the error if there are 3 or fewer.
                        if (effigies.size() <= 3) {
                            // FIXME: This could cause problems with
                            // systems that do not load the user
                            // library.  Currently, VergilApplication
                            // loads the user library, but other
                            // applications like PtolemyApplication do
                            // not.  We could check to see if the
                            // names of two of the three Effigies were
                            // .configuration.directory.configuration
                            // and.configuration.directory.user
                            // library, but this seems like overkill.
                            String errorMessage = "Failed to read " + input;
                            System.err.println(errorMessage);
                            throwable.printStackTrace();
                            MessageHandler.error(errorMessage, throwable);
                        } else {
                            if (throwable instanceof Exception) {
                                // Let the caller handle the error.
                                throw (Exception) throwable;
                            } else {
                                // If we have a parameter that has a backslash
                                // then we might get a data.expr.TokenMgrError
                                // which is an error, so we rethrow this
                                // FIXME: createEffigy() should be
                                // declared to throw Throwable, but that
                                // results in lots of changes elsewhere.
                                throw new Exception(throwable);
                            }
                        }
                    }
                } finally {
                    // If we failed to populate the effigy with a model,
                    // then we remove the effigy from its container.
                    if (toplevel == null) {
                        effigy.setContainer(null);
                    }
                }

                return null;
            }
        }

        /** Create a new effigy.  We use the strategy pattern here
         *  so that derived classes can easily override the exact class
         *  that is created.
         *  @param container The container for the effigy.
         *  @param name The name.
         *  @return A new effigy.
         *  @exception IllegalActionException If the entity cannot be contained
         *   by the proposed container.
         *  @exception NameDuplicationException If the name coincides with
         *   an entity already in the container.
         */
        protected PtolemyEffigy _newEffigy(CompositeEntity container,
                String name)
                throws IllegalActionException, NameDuplicationException {
            return new PtolemyEffigy(container, name);
        }
    }

    /** A factory for creating new Ptolemy effigies, but without the
     *  capability of creating a new blank effigy.  Use this factory
     *  in a configuration if you do not want the factory to appear
     *  in the File-&gt;New menu.
     */
    public static class FactoryWithoutNew extends Factory {
        /** Create a factory with the given name and container.
         *  @param container The container.
         *  @param name The name.
         *  @exception IllegalActionException If the container is incompatible
         *   with this entity.
         *  @exception NameDuplicationException If the name coincides with
         *   an entity already in the container.
         */
        public FactoryWithoutNew(CompositeEntity container, String name)
                throws IllegalActionException, NameDuplicationException {
            super(container, name);
        }

        ///////////////////////////////////////////////////////////////
        ////                     public methods                    ////

        /** Return false, indicating that this effigy factory is not
         *  capable of creating an effigy without a URL being specified.
         *  @return False.
         */
        @Override
        public boolean canCreateBlankEffigy() {
            return false;
        }
    }
}