EvioTestHelper.java

import java.util.Random;
import java.nio.ByteOrder;

import org.jlab.coda.hipo.CompressionType;
// Import evio
import org.jlab.coda.jevio.*;

public class EvioTestHelper {
    private final Random random;

    // Parameters (simplified here for brevity)
    public static String baseNameV4 = "testEventsV4_javaAPI.evio"; // base name of file to be created. If split > 1, this is the base name of all files created. If split < 1, this is the name of the only file created.
    public static String baseNameV6 = "testEventsV6_javaAPI.evio"; // base name of file to be created. If split > 1, this is the base name of all files created. If split < 1, this is the name of the only file created.
    public static String baseNameHIPO = "testEventsHIPO_javaAPI.hipo"; // base name of file to be created. If split > 1, this is the base name of all files created. If split < 1, this is the name of the only file created.
    public static String directory  = "etc/test_files"; // "/mnt/tmpfs" or null=relative path (or directory in which file is to be placed)
    final static String runType  = "";  // name of run type configuration to be used in naming files
    private final static int runNumber  = 1;    // arbitrary, usually experiment-specific
    private final static long split  = 0; // if < 1, do not split file, write to only one file of unlimited size. Else this is max size in bytes to make a file before closing it and starting writing another. 
    private final static int maxRecordSize = 33554432; // (32 MiB) max number of uncompressed data bytes each record can hold. Value of < 8MB results in default of 8MB. The size of the record will not be larger than this size unless a single event itself is larger
    private final static int maxEventCount = 10000;  // max number of events each RECORD can hold. Value <= O means use default (1M).
    private final static ByteOrder byteOrder = ByteOrder.nativeOrder(); // options: BIG_ENDIAN, LITTLE_ENDIAN, nativeOrder()
    private final static String XMLdictionary = 
            "<xmlDict>\n" +
            "  <bank name=\"floatBank\" tag=\"10\" num=\"1\" type=\"float32\">\n" +
            "    <leaf name=\"X\"/>\n" +
            "    <leaf name=\"Y\"/>\n" +
            "    <leaf name=\"Z\"/>\n" +
            "    <leaf name=\"time\"/>\n" +
            "    <leaf/>\n" +
            "  </bank>\n" +
            // "  <dictEntry name=\"jzint\" tag=\"11\" num=\"2\" type=\"int32\" />\n" +
            // "  <dictEntry name=\"example\" tag=\"12\" num=\"3\" type=\"charstar8\" />\n" +
            "</xmlDict>\n";
        private final static boolean overWriteOK = true;
        private final static boolean append = false;
        private final static EvioBank firstEvent = null;  // The first event written into each file (after any dictionary) including all split files; may be null. Useful for adding common, static info into each split file
        private final static int streamId = 1;  // streamId number (100 > id > -1) for file name
        private final static int splitNumber = 0;  // number at which to start the split numbers
        private final static int splitIncrement = 1; // amount to increment split number each time another file is created
        private final static int streamCount = 1; // total number of streams in DAQ
        private final static CompressionType compressionType = CompressionType.RECORD_UNCOMPRESSED;
        private final static CompressionType compressionTypeHIPO = CompressionType.RECORD_COMPRESSION_LZ4;
        private final static int compressionThreads = 1; // number of threads doing compression simultaneously
        private final static int ringSize = 0;  // number of records in supply ring. If set to < compressionThreads, it is forced to equal that value and is also forced to be a multiple of 2, rounded up.
        private final static int   bufferSize = 33554432; // (32 MiB) number of bytes to make each internal buffer which will be storing events before writing them to a file. 9MB = default if bufferSize = 0.

    // Constructor
    public EvioTestHelper() {
        this.random = new Random();
    }

    // Generate a float array similar to C++ vector<float>
    public float[] genXYZT(int i) {
        float[] x4 = new float[4];
        x4[0] = (float) random.nextGaussian() * 0.1f;
        x4[1] = (float) random.nextGaussian() * 0.1f;
        x4[2] = 0.0f;
        x4[3] = i * 2.008f;
        return x4;
    }

    public static EventWriterV4 defaultEventWriterV4() {

        // EventWriterV4 w = null;

        EventWriterV4 w = null;

        try {
                        
            w = new EventWriterV4(
                baseNameV4, 
                directory, 
                runType, 
                runNumber,
                split, 
                maxRecordSize, 
                maxEventCount, 
                10000, 
                byteOrder, 
                XMLdictionary,
                null, 
                overWriteOK,
                append,
                firstEvent,
                streamId,
                splitNumber,
                splitIncrement,
                streamCount
);
        } 
        catch (EvioException e) {
            System.err.println("Error creating writer. Stack trace:\n ");
            e.printStackTrace();
        }
        return w;
    }

    // Default writer
    public static EventWriter defaultEventWriter() {

        EventWriter w = null;

        try {
            w = new EventWriter(
                baseNameV6, 
                directory, 
                runType, 
                runNumber,
                split, 
                maxRecordSize, 
                maxEventCount, 
                byteOrder, 
                XMLdictionary,
                overWriteOK,
                append,
                firstEvent,
                streamId,
                splitNumber,
                splitIncrement,
                streamCount,
                compressionType,
                compressionThreads,
                ringSize,
                bufferSize);
        } 
        catch (EvioException e) {
            System.err.println("Error creating writer. Stack trace:\n ");
            e.printStackTrace();
        }
        return w;
    }

    
}