io.github.ericmedvet.mrsim2d.viewer.Framer 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;

import io.github.ericmedvet.mrsim2d.core.geometry.BoundingBox;
import io.github.ericmedvet.mrsim2d.core.geometry.Point;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

@FunctionalInterface
public interface Framer {

  BoundingBox getFrame(double t, K k, double ratio);

  default Framer averaged(double windowT) {
    Framer thisFramer = this;
    Map memory = new HashMap<>();
    return (t, k, ratio) -> {
      BoundingBox currentBB = thisFramer.getFrame(t, k, ratio);
      // update memory
      memory.put(t, currentBB);
      List toRemoveKeys =
          memory.keySet().stream().filter(ot -> ot < t - windowT).toList();
      toRemoveKeys.forEach(memory::remove);
      // average bounding box
      double x = memory.values().stream()
          .mapToDouble(bb -> bb.center().x())
          .average()
          .orElse(currentBB.center().x());
      double y = memory.values().stream()
          .mapToDouble(bb -> bb.center().y())
          .average()
          .orElse(currentBB.center().y());
      double w = memory.values().stream()
          .mapToDouble(BoundingBox::width)
          .average()
          .orElse(currentBB.width());
      double h = memory.values().stream()
          .mapToDouble(BoundingBox::height)
          .average()
          .orElse(currentBB.height());
      return new BoundingBox(new Point(x - w / 2d, y - h / 2d), new Point(x + w / 2d, y + h / 2d));
    };
  }

  default Framer largest(double windowT) {
    Framer thisFramer = this;
    Map memory = new HashMap<>();
    return (t, k, ratio) -> {
      BoundingBox currentBB = thisFramer.getFrame(t, k, ratio);
      // update memory
      memory.put(t, currentBB);
      List toRemoveKeys =
          memory.keySet().stream().filter(ot -> ot < t - windowT).toList();
      toRemoveKeys.forEach(memory::remove);
      // take largest bounding box
      return memory.values().stream().reduce(BoundingBox::enclosing).orElse(currentBB);
    };
  }
}