/*
 * Copyright (c) 2009, 2011, Oracle  and/or its affiliates. All rights reserved.
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
 *
 * This code is free software; you can redistribute it and/or modify it
 * under the terms of the GNU General Public License version 2 only, as
 * published by the Free Software Foundation.
 *
 * This code 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
 * version 2 for more details (a copy is included in the LICENSE file that
 * accompanied this code).
 *
 * You should have received a copy of the GNU General Public License version
 * 2 along with this work; if not, write to the Free Software Foundation,
 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
 *
 * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
 * CA 95054 USA or visit www.sun.com if you need additional information or
 * have any questions.
 */
package com.sun.javafx.tk.desktop;

import com.sun.javafx.tk.Toolkit;

import java.lang.reflect.Method;
import java.security.AccessController;
import java.security.PrivilegedAction;

/**
 * Class containing implementation for logging, and performance tracking.
 * 
 * @author tdv
 */
public abstract class PerformanceTrackerHelper {
    private static final PerformanceTrackerHelper instance = createInstance();

    public static PerformanceTrackerHelper getInstance() {
        return instance;
    }

    private PerformanceTrackerHelper() {
    }

    private static PerformanceTrackerHelper createInstance() {
        return AccessController.doPrivileged(
                new PrivilegedAction<PerformanceTrackerHelper>() {
                    @Override
                    public PerformanceTrackerHelper run() {
                        try {
                            if (System.getProperty("sun.perflog") != null) {
                                final PerformanceTrackerHelper trackerImpl =
                                        new PerformanceTrackerDefaultImpl();

                                if (System.getProperty(
                                        "sun.perflog.fx.exitflush") != null) {
                                    Runtime.getRuntime().addShutdownHook(
                                            new Thread() {
                                                @Override
                                                public void run() {
                                                    trackerImpl.outputLog();
                                                }
                                            });
                                }

                                return trackerImpl;
                            }
                        } catch (Throwable t) {
                        }

                        return new PerformanceTrackerDummyImpl();
                    }
                });
    }

    public abstract void logEvent(final String s);

    public abstract void outputLog();

    public abstract boolean isPerfLoggingEnabled();

    public final long nanoTime() {
        return Toolkit.getToolkit().getMasterTimer().nanos();
    }

    private static final class PerformanceTrackerDefaultImpl
            extends PerformanceTrackerHelper {
        private long firstTime;
        private long lastTime;

        private final Method logEventMethod;
        private final Method outputLogMethod;
        private final Method getStartTimeMethod;
        private final Method setStartTimeMethod;

        public PerformanceTrackerDefaultImpl() throws ClassNotFoundException,
                                                      NoSuchMethodException {
            final Class perfLoggerClass =
                    Class.forName("sun.misc.PerformanceLogger", true, null);

            logEventMethod =
                    perfLoggerClass.getMethod("setTime", String.class);
            outputLogMethod =
                    perfLoggerClass.getMethod("outputLog");
            getStartTimeMethod =
                    perfLoggerClass.getMethod("getStartTime");
            setStartTimeMethod =
                    perfLoggerClass.getMethod("setStartTime", String.class,
                                              long.class);
        }

        @Override
        public void logEvent(final String s) {
            final long time = System.currentTimeMillis();
            if (firstTime == 0) {
                firstTime = time;
            }
            AccessController.doPrivileged(
                    new PrivilegedAction<Object>() {
                        @Override
                        public Object run() {
                            try {
                                logEventMethod.invoke(
                                        null,
                                        "JavaFX> " + s + " ("
                                            + (time - firstTime) + "ms total, "
                                            + (time - lastTime) + "ms)");
                            } catch (Exception e) {
                            }

                            return null;
                        }
                    });
            
            lastTime = time;
        }

        @Override
        public void outputLog() {
            AccessController.doPrivileged(
                    new PrivilegedAction<Object>() {
                        @Override
                        public Object run() {
                            logLaunchTime();

                            // Output the log
                            try {
                                outputLogMethod.invoke(null);
                            } catch (Exception e) {
                            }

                            return null;
                        }
                    });
        }

        @Override
        public boolean isPerfLoggingEnabled() {
            return true;
        }

        private void logLaunchTime() {
            try {
                // Attempt to log launchTime, if not set already
                if ((Long) getStartTimeMethod.invoke(null) <= 0) {
                    // Standalone apps record launch time as sysprop
                    String launchTimeString = System.getProperty("launchTime");
                    if (launchTimeString != null
                            && !launchTimeString.equals("")) {
                        long launchTime = Long.parseLong(launchTimeString);
                        setStartTimeMethod.invoke(
                                null, "LaunchTime", launchTime);
                    }
                }
            } catch (Throwable t) {
                t.printStackTrace();
            }
        }
    }

    private static final class PerformanceTrackerDummyImpl
            extends PerformanceTrackerHelper {
        @Override
        public void logEvent(final String s) {
        }

        @Override
        public void outputLog() {
        }

        @Override
        public boolean isPerfLoggingEnabled() {
            return false;
        }
    }
}
