bundles/org.simantics.threadlog/src/org/simantics/threadlog/ThreadLog.java

   1 /*******************************************************************************\r
   2  * Copyright (c) 2007, 2010 Association for Decentralized Information Management\r
   3  * in Industry THTH ry.\r
   4  * All rights reserved. This program and the accompanying materials\r
   5  * are made available under the terms of the Eclipse Public License v1.0\r
   6  * which accompanies this distribution, and is available at\r
   7  * http://www.eclipse.org/legal/epl-v10.html\r
   8  *\r
   9  * Contributors:\r
  10  *     VTT Technical Research Centre of Finland - initial API and implementation\r
  11  *******************************************************************************/\r
  12 package org.simantics.threadlog;\r
  13 \r
  14 import gnu.trove.TDoubleArrayList;\r
  15 import gnu.trove.TLongArrayList;\r
  16 \r
  17 import java.io.BufferedInputStream;\r
  18 import java.io.BufferedOutputStream;\r
  19 import java.io.DataInput;\r
  20 import java.io.DataInputStream;\r
  21 import java.io.DataOutput;\r
  22 import java.io.DataOutputStream;\r
  23 import java.io.File;\r
  24 import java.io.FileInputStream;\r
  25 import java.io.FileOutputStream;\r
  26 import java.io.IOException;\r
  27 import java.io.StreamCorruptedException;\r
  28 import java.util.ArrayList;\r
  29 \r
  30 public class ThreadLog {\r
  31 \r
  32     private static final transient String HDR             = "TLOG";\r
  33     private static final transient int    CURRENT_VERSION = 1;\r
  34 \r
  35     static ThreadLog defaultLog;\r
  36 \r
  37     ArrayList<String> tasks = new ArrayList<String>();\r
  38     TDoubleArrayList times = new TDoubleArrayList();\r
  39     TLongArrayList threads = new TLongArrayList();\r
  40 \r
  41     public double[] getTimes() {\r
  42         return times.toNativeArray();\r
  43     }\r
  44 \r
  45     public String[] getTasks() {\r
  46         return tasks.toArray(new String[tasks.size()]);\r
  47     }\r
  48 \r
  49     public long[] getThreads() {\r
  50         return threads.toNativeArray();\r
  51     }\r
  52 \r
  53     private class TaskImpl implements Task {\r
  54         public String name;\r
  55         public long thread;\r
  56         public long beginTime;\r
  57 \r
  58         public TaskImpl(String name) {\r
  59             this.name = name;\r
  60             this.thread = Thread.currentThread().getId();\r
  61             this.beginTime = System.nanoTime();\r
  62         }\r
  63 \r
  64         @Override\r
  65         public void end() {\r
  66             long endTime = System.nanoTime();\r
  67             synchronized(tasks) {\r
  68                 tasks.add(name);\r
  69                 times.add(beginTime*1e-9);\r
  70                 times.add(endTime*1e-9);\r
  71                 threads.add(thread);\r
  72             }\r
  73         }\r
  74     }\r
  75 \r
  76     private static enum DummyTask implements Task {\r
  77         INSTANCE;\r
  78 \r
  79         @Override\r
  80         public void end() {\r
  81         }\r
  82     }\r
  83 \r
  84     public Task begin(String name) {\r
  85         return new TaskImpl(name);\r
  86     }\r
  87 \r
  88     public static Task BEGIN(String name) {\r
  89         try {\r
  90             if(defaultLog != null)\r
  91                 return defaultLog.begin(name);\r
  92         } catch(NullPointerException e) {\r
  93         }\r
  94         return DummyTask.INSTANCE;\r
  95     }\r
  96 \r
  97     public static ThreadLog setDefaultThreadLog(ThreadLog log) {\r
  98         ThreadLog currentLog = defaultLog;\r
  99         defaultLog = log;\r
 100         return currentLog;\r
 101     }\r
 102 \r
 103     // ------------------------------------------------------------------------\r
 104     // SERIALIZATION\r
 105     // ------------------------------------------------------------------------\r
 106 \r
 107     public static ThreadLog deserialize(File file) throws IOException {\r
 108         DataInputStream in = null;\r
 109         try {\r
 110             in = new DataInputStream(new BufferedInputStream(new FileInputStream(file)));\r
 111             ThreadLog log = new ThreadLog();\r
 112             log.doDeserialize(in);\r
 113             return log;\r
 114         } finally {\r
 115             if (in != null) {\r
 116                 in.close();\r
 117             }\r
 118         }\r
 119     }\r
 120 \r
 121     public static ThreadLog deserialize(DataInput in) throws IOException {\r
 122         ThreadLog log = new ThreadLog();\r
 123         log.doDeserialize(in);\r
 124         return log;\r
 125     }\r
 126 \r
 127     private void doDeserialize(DataInput in) throws IOException {\r
 128         String hdr = in.readUTF();\r
 129         if (!HDR.equals(hdr)) {\r
 130             throw new StreamCorruptedException("invalid header '" + hdr + "', expected " + HDR);\r
 131         }\r
 132         int ver = in.readInt();\r
 133         if (ver == CURRENT_VERSION) {\r
 134             int taskCount = in.readInt();\r
 135             for (int i = 0; i < taskCount; ++i) {\r
 136                 String task = in.readUTF();\r
 137                 double beginTime = in.readDouble();\r
 138                 double endTime = in.readDouble();\r
 139                 long threadId = in.readLong();\r
 140                 tasks.add(task);\r
 141                 times.add(beginTime);\r
 142                 times.add(endTime);\r
 143                 threads.add(threadId);\r
 144             }\r
 145         } else {\r
 146             throw new StreamCorruptedException("unrecognized log version: " + ver);\r
 147         }\r
 148     }\r
 149 \r
 150     public void serialize(File file) throws IOException {\r
 151         DataOutputStream out = null;\r
 152         try {\r
 153             out = new DataOutputStream(new BufferedOutputStream(new FileOutputStream(file)));\r
 154             serialize(out);\r
 155         } finally {\r
 156             if (out != null) {\r
 157                 out.close();\r
 158             }\r
 159         }\r
 160     }\r
 161 \r
 162     public void serialize(DataOutput out) throws IOException {\r
 163         out.writeUTF(HDR);\r
 164         out.writeInt(CURRENT_VERSION);\r
 165         int len = tasks.size();\r
 166         out.writeInt(len);\r
 167         for (int i = 0; i < len; ++i) {\r
 168             out.writeUTF(tasks.get(i));\r
 169             out.writeDouble(times.getQuick(i*2));\r
 170             out.writeDouble(times.getQuick(i*2+1));\r
 171             out.writeLong(threads.getQuick(i));\r
 172         }\r
 173     }\r
 174 \r
 175 }\r