io.github.ericmedvet.mrsim2d.viewer.drawers.LinePlotter Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of mrsim2d.viewer Show documentation
Visual components (video maker and GUI) for 2D-MR-Sim.
The newest version!
/*-
 * ========================LICENSE_START=================================
 * mrsim2d-viewer
 * %%
 * Copyright (C) 2020 - 2024 Eric Medvet
 * %%
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 * =========================LICENSE_END==================================
 */
package io.github.ericmedvet.mrsim2d.viewer.drawers;

import io.github.ericmedvet.jnb.datastructure.DoubleRange;
import io.github.ericmedvet.mrsim2d.core.Snapshot;
import io.github.ericmedvet.mrsim2d.core.geometry.BoundingBox;
import io.github.ericmedvet.mrsim2d.core.geometry.Point;
import io.github.ericmedvet.mrsim2d.viewer.AbstractMemoryDrawer;
import io.github.ericmedvet.mrsim2d.viewer.DrawingUtils;
import java.awt.Graphics2D;
import java.util.SortedMap;
import java.util.function.Function;

public class LinePlotter extends AbstractMemoryDrawer {
  private final String format;

  public LinePlotter(Function extractor, double windowT, String format) {
    super(extractor, windowT, WindowType.SNAPSHOT_TIME);
    this.format = format;
  }

  @Override
  protected boolean innerDraw(SortedMap memory, Graphics2D g) {
    // find ranges
    DoubleRange tRange = new DoubleRange(memory.lastKey() - getWindowT(), memory.lastKey());
    DoubleRange vRange = new DoubleRange(
        Math.min(0, memory.values().stream().mapToDouble(d -> d).min().orElse(0d)),
        Math.max(0, memory.values().stream().mapToDouble(d -> d).max().orElse(0d)));
    BoundingBox gBB = DrawingUtils.getBoundingBox(g);
    // prepare points
    Point[] drawPoints = memory.entrySet().stream()
        .map(e -> new Point(
            gBB.xRange().denormalize(tRange.normalize(e.getKey())),
            gBB.yRange().denormalize(1 - vRange.normalize(e.getValue()))))
        .toArray(Point[]::new);
    Point[] fillPoints = new Point[drawPoints.length + 2];
    fillPoints[0] = new Point(drawPoints[0].x(), gBB.max().y());
    fillPoints[fillPoints.length - 1] = new Point(gBB.max().x(), gBB.max().y());
    System.arraycopy(drawPoints, 0, fillPoints, 1, drawPoints.length);
    // draw
    g.setColor(DrawingUtils.Colors.DATA);
    g.draw(DrawingUtils.toPath(drawPoints));
    g.setColor(DrawingUtils.alphaed(DrawingUtils.Colors.DATA, 0.5f));
    g.fill(DrawingUtils.toPath(fillPoints));
    if (format != null) {
      g.setFont(DrawingUtils.FONT);
      String s = format.formatted(memory.get(memory.lastKey()));
      double sW = g.getFontMetrics().stringWidth(s);
      g.setColor(DrawingUtils.Colors.TEXT);
      g.drawString(s, (float) (gBB.center().x() - sW / 2d), (float)
          (gBB.center().y() + g.getFontMetrics().getHeight() / 2d));
    }
    return true;
  }
}