org.eclipse.tracecompass.tmf.core.tests/stubs/org/eclipse/tracecompass/tmf/tests/stubs/CreateTestFiles.java

   1 /*******************************************************************************
   2  * Copyright (c) 2009, 2014 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  *   Francois Chouinard - Initial API and implementation
  11  *******************************************************************************/
  12
  13 package org.eclipse.tracecompass.tmf.tests.stubs;
  14
  15 import java.io.BufferedOutputStream;
  16 import java.io.DataOutputStream;
  17 import java.io.File;
  18 import java.io.FileNotFoundException;
  19 import java.io.FileOutputStream;
  20 import java.io.IOException;
  21 import java.util.Random;
  22
  23 /**
  24  * <b><u>CreateTestFiles</u></b>
  25  * <p>
  26  * Create a number of event test files of various lengths.
  27  * <p>
  28  * Events have the following format:
  29  * <ul>
  30  * <li> [timestamp] [source] [type] [ref] [field]*
  31  * <li> There are NB_SOURCES sources and NB_TYPES types.
  32  * <li> The number of fields (0 .. NB_TYPES-1) depends on the event type.
  33  * </ul>
  34  */
  35 public class CreateTestFiles {
  36
  37     // ========================================================================
  38     // Constants
  39     // ========================================================================
  40
  41     private static final String DIRECTORY = "testfiles";
  42     //    private static final String FILE_NAMES[] = { "Test-10", "Test-1K", "Test-10K", "Test-100K" };
  43     //    private static final int    FILE_SIZES[] = {       10 ,     1000 ,     10000 ,     100000  };
  44     private static final String FILE_NAMES[] = { "Test-10K" };
  45     private static final int    FILE_SIZES[] = {     10000  };
  46
  47     private static final int NB_SOURCES = 15;
  48     private static final int NB_TYPES   =  7;
  49
  50     // ========================================================================
  51     // Constructors
  52     // ========================================================================
  53
  54     /**
  55      * @param args unused
  56      */
  57     public static void main(final String[] args) {
  58
  59         try {
  60             System.out.println("Creating test files in directory: " + new File(".").getCanonicalPath() + File.separator + DIRECTORY);
  61         } catch (final IOException e) {
  62             e.printStackTrace();
  63         }
  64
  65         for (int i = 0; i < FILE_SIZES.length; i++) {
  66             try {
  67                 createTestFile("testfiles" + File.separator + "O-" + FILE_NAMES[i], FILE_SIZES[i], true,  true);
  68                 createTestFile("testfiles" + File.separator + "E-" + FILE_NAMES[i], FILE_SIZES[i], true,  false);
  69                 createTestFile("testfiles" + File.separator + "R-" + FILE_NAMES[i], FILE_SIZES[i], false, false);
  70             } catch (final FileNotFoundException e) {
  71             } catch (final IOException e) {
  72             }
  73         }
  74
  75         System.out.println("Done.");
  76     }
  77
  78     // ========================================================================
  79     // Operators
  80     // ========================================================================
  81
  82     /**
  83      * @param file
  84      * @param size
  85      * @param monotonic
  86      * @throws FileNotFoundException
  87      * @throws IOException
  88      */
  89     private static void createTestFile(final String file, final int size,
  90             final boolean monotonic, final boolean odd)
  91             throws IOException {
  92         System.out.println("Creating " + file);
  93         try (DataOutputStream out = new DataOutputStream(new BufferedOutputStream(new FileOutputStream(file)));) {
  94
  95             final Random generator = new Random(19580427 + size);
  96             long ts = (monotonic && odd) ? -1 : 0;
  97             for (int i = 0; i < size; i++) {
  98                 ts += monotonic ? 2 : generator.nextInt(10);
  99                 final int sourceIndex = i % NB_SOURCES;
 100                 final int typeIndex = i % NB_TYPES;
 101                 out.writeLong(ts); // Timestamp
 102                 out.writeUTF("Source-" + sourceIndex); // Source
 103                 out.writeUTF("Type-" + typeIndex); // Type
 104                 out.writeInt(i + 1); // Reference (event #)
 105                 for (int j = 0; j < typeIndex; j++) {
 106                     out.writeUTF("Field-" + sourceIndex + "-" + j);
 107                 }
 108             }
 109             out.flush();
 110         }
 111     }
 112
 113 }