LogFile.java

Go to the documentation of this file.

package zephyr.plugin.core.api.internal.logfiles;

import java.io.BufferedReader;
import java.io.File;
import java.io.IOException;
import java.util.LinkedHashMap;
import java.util.Map;
import zephyr.plugin.core.api.labels.Labeled;
import zephyr.plugin.core.api.monitoring.annotations.LabelProvider;
import zephyr.plugin.core.api.monitoring.annotations.Monitor;

@Monitor
public abstract class LogFile implements Labeled {
  protected BufferedReader reader;
  final public String filepath;
  private final double[] current;
  private final String[] labels;

  public LogFile(String filepath) throws IOException {
    this.filepath = filepath;
    reader = getReader(filepath);
    labels = readLabels();
    current = new double[labels.length];
  }

  @LabelProvider(ids = { "current" })
  protected String labelOf(int i) {
    return labels[i];
  }

  public boolean eof() {
    if (reader == null)
      return true;
    boolean isReady = false;
    synchronized (reader) {
      try {
        if (reader != null)
          isReady = reader.ready();
      } catch (IOException e) {
        e.printStackTrace();
      }
    }
    return !isReady;
  }

  abstract protected BufferedReader getReader(String filepath) throws IOException;

  protected String[] readLabels() {
    String line = readLine();
    while (line.startsWith("#") || line.startsWith(" "))
      line = line.substring(1);
    String[] labels = line.split(" ");
    makeLabelsUnique(labels);
    return labels;
  }

  private static void makeLabelsUnique(String[] labels) {
    Map<String, Integer> nbInstances = new LinkedHashMap<String, Integer>();
    for (int i = 0; i < labels.length; i++) {
      Integer knownInstance = nbInstances.get(labels[i]);
      if (knownInstance != null)
        labels[i] = labels[i] + " (" + String.valueOf(knownInstance + 1) + ")";
      else
        knownInstance = 0;
      knownInstance++;
      nbInstances.put(labels[i], knownInstance);
    }
  }

  private String readDataLine() {
    if (reader == null)
      return null;
    String line = readLine();
    if (line == null || line.startsWith("#"))
      return null;
    if (line.isEmpty())
      return null;
    return line;
  }


  private String readLine() {
    String line = null;
    synchronized (reader) {
      try {
        line = reader.readLine();
      } catch (IOException e) {
        e.printStackTrace();
        close();
      }
    }
    return line;
  }

  public void step() {
    String line = null;
    while (!eof() && line == null)
      line = readDataLine();
    if (line != null)
      lineToData(line);
  }

  private void lineToData(String line) {
    assert line != null;
    String[] sarray = line.split(" ");
    for (int i = 0; i < Math.min(sarray.length, current.length); i++)
      try {
        current[i] = Double.valueOf(sarray[i]);
      } catch (Exception e) {
      }
  }

  public static LogFile load(String filename) {
    try {
      if (filename.endsWith(".gz"))
        return new GZippedLogFile(filename);
      if (filename.endsWith(".bz2"))
        return new BZippedLogFile(filename);
      if (filename.endsWith(".zip"))
        return new ZippedLogFile(filename);
      return new TextLogFile(filename);
    } catch (IOException e) {
      e.printStackTrace();
      return null;
    }
  }

  public void close() {
    if (reader == null)
      return;
    synchronized (reader) {
      try {
        reader.close();
        reader = null;
      } catch (IOException e) {
        e.printStackTrace();
      }
    }
  }

  @Override
  public String label() {
    String fileName = new File(filepath).getName();
    int extensionIndex = fileName.indexOf(".");
    if (extensionIndex == -1)
      return fileName;
    return fileName.substring(0, extensionIndex);
  }

  public double[] currentLine() {
    return current.clone();
  }

  public String[] labels() {
    return labels.clone();
  }
}