lttng/org.eclipse.tracecompass.lttng2.kernel.core.tests/src/org/eclipse/tracecompass/lttng2/kernel/core/tests/analysis/kernel/statesystem/GenerateTestValues.java

   1 /*******************************************************************************
   2  * Copyright (c) 2013, 2015 Ericsson
   3  *
   4  * All rights reserved. This program and the accompanying materials are
   5  * made available under the terms of the Eclipse Public License v1.0 which
   6  * accompanies this distribution, and is available at
   7  * http://www.eclipse.org/legal/epl-v10.html
   8  *
   9  * Contributors:
  10  *   Alexandre Montplaisir - Initial API and implementation
  11  ******************************************************************************/
  12
  13 package org.eclipse.tracecompass.lttng2.kernel.core.tests.analysis.kernel.statesystem;
  14
  15 import java.io.File;
  16 import java.io.FileWriter;
  17 import java.io.PrintWriter;
  18 import java.util.List;
  19
  20 import org.eclipse.jdt.annotation.NonNull;
  21 import org.eclipse.tracecompass.analysis.os.linux.core.kernelanalysis.KernelAnalysisModule;
  22 import org.eclipse.tracecompass.statesystem.core.ITmfStateSystem;
  23 import org.eclipse.tracecompass.statesystem.core.interval.ITmfStateInterval;
  24 import org.eclipse.tracecompass.statesystem.core.statevalue.ITmfStateValue;
  25 import org.eclipse.tracecompass.testtraces.ctf.CtfTestTrace;
  26 import org.eclipse.tracecompass.tmf.core.statesystem.TmfStateSystemAnalysisModule;
  27 import org.eclipse.tracecompass.tmf.ctf.core.tests.shared.CtfTmfTestTraceUtils;
  28 import org.eclipse.tracecompass.tmf.ctf.core.trace.CtfTmfTrace;
  29 import org.junit.Test;
  30
  31 /**
  32  * Small program to regenerate the values used in "TestValues.java" from the
  33  * current LTTng-kernel state provider.
  34  *
  35  * It will write its output the a file called 'TestValues<something>.java' in
  36  * your temporary files directory.
  37  *
  38  * @author Alexandre Montplaisir
  39  */
  40 public class GenerateTestValues {
  41
  42     private static @NonNull CtfTestTrace testTrace = CtfTestTrace.TRACE2;
  43     private static final long targetTimestamp = 18670067372290L + 1331649577946812237L;
  44     private static final String INDENT = "    ";
  45
  46     /**
  47      * Test wrapper to run main properly
  48      *
  49      * @throws Exception
  50      *             we're messing with exception
  51      */
  52     @Test
  53     public void test() throws Exception {
  54         main(null);
  55     }
  56
  57     /**
  58      * Run the program
  59      *
  60      * @param args
  61      *            Command-line arguments, unused.
  62      * @throws Exception
  63      *             I'm messing with Exception. Come at me bro!
  64      */
  65     public static void main(String[] args) throws Exception {
  66         /* Prepare the files */
  67         File logFile = File.createTempFile("TestValues", ".java");
  68         try (PrintWriter writer = new PrintWriter(new FileWriter(logFile), true);) {
  69
  70             /* Build and query the state system */
  71             final CtfTmfTrace trace = CtfTmfTestTraceUtils.getTrace(testTrace);
  72             TmfStateSystemAnalysisModule module = new KernelAnalysisModule() {
  73                 @Override
  74                 protected String getSsFileName() {
  75                     return "test-values";
  76                 }
  77             };
  78
  79             if (!module.setTrace(trace)) {
  80                 throw new IllegalStateException();
  81             }
  82             module.setId("test-values");
  83             module.schedule();
  84             module.waitForCompletion();
  85             ITmfStateSystem ssq = module.getStateSystem();
  86             if (ssq == null) {
  87                 throw new IllegalStateException();
  88             }
  89
  90             List<ITmfStateInterval> fullState = ssq.queryFullState(targetTimestamp);
  91
  92             /* Start printing the java file's contents */
  93             writer.println("interface TestValues {");
  94             writer.println();
  95             writer.println(INDENT + "int size = " + fullState.size() + ";");
  96             writer.println();
  97
  98             /* Print the array contents */
  99             writer.println(INDENT + "long[] startTimes = {");
 100             for (ITmfStateInterval interval : fullState) {
 101                 writer.println(INDENT + INDENT + String.valueOf(interval.getStartTime()) + "L,");
 102             }
 103             writer.println(INDENT + "};");
 104             writer.println();
 105
 106             writer.println(INDENT + "long[] endTimes = {");
 107             for (ITmfStateInterval interval : fullState) {
 108                 writer.println(INDENT + INDENT + String.valueOf(interval.getEndTime()) + "L,");
 109             }
 110             writer.println(INDENT + "};");
 111             writer.println();
 112
 113             writer.println(INDENT + "ITmfStateValue[] values = {");
 114             for (ITmfStateInterval interval : fullState) {
 115                 ITmfStateValue val = interval.getStateValue();
 116                 writer.print(INDENT + INDENT);
 117
 118                 switch (val.getType()) {
 119                 case NULL:
 120                     writer.println("TmfStateValue.nullValue(),");
 121                     break;
 122                 case INTEGER:
 123                     writer.println("TmfStateValue.newValueInt(" + val.unboxInt() + "),");
 124                     break;
 125                 case LONG:
 126                     writer.println("TmfStateValue.newValueLong(" + val.unboxLong() + "),");
 127                     break;
 128                 case DOUBLE:
 129                     writer.println("TmfStateValue.newValueDouble(" + val.unboxDouble() + "),");
 130                     break;
 131                 case STRING:
 132                     writer.println("TmfStateValue.newValueString(\"" + val.unboxStr() + "\"),");
 133                     break;
 134                 default:
 135                     writer.println(val.toString());
 136                     break;
 137                 }
 138             }
 139             writer.println(INDENT + "};");
 140
 141             writer.println("}");
 142             writer.println();
 143
 144             module.dispose();
 145             trace.dispose();
 146         }
 147         System.exit(0);
 148     }
 149
 150 }