com.simiacryptus.mindseye.pyramid.ViewJSTest Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of mindseye-art Show documentation
Visual Neural Network Applications
There is a newer version: 2.1.0
/*
 * Copyright (c) 2019 by Andrew Charneski.
 *
 * The author licenses this file to you 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.
 */

package com.simiacryptus.mindseye.pyramid;

import com.simiacryptus.mindseye.ImageScript;
import com.simiacryptus.mindseye.applications.ArtistryUtil;
import com.simiacryptus.mindseye.test.TestUtil;
import com.simiacryptus.notebook.MarkdownNotebookOutput;
import com.simiacryptus.notebook.NotebookOutput;
import com.simiacryptus.util.JsonUtil;

import javax.annotation.Nonnull;
import java.awt.image.BufferedImage;
import java.io.File;
import java.util.Arrays;
import java.util.function.Function;
import java.util.stream.IntStream;

/**
 * The type Style survey.
 */
public class ViewJSTest extends ImageScript {

  public CharSequence[] styleSources;

  public ViewJSTest(final CharSequence[] styleSources) {
    this.styleSources = styleSources;
    this.verbose = true;
  }

  public void accept(@Nonnull NotebookOutput log) {
    TestUtil.addGlobalHandlers(log.getHttpd());
    ((MarkdownNotebookOutput) log).setMaxImageSize(10000);
    log.eval(() -> {
      return JsonUtil.toJson(ViewJSTest.this);
    });
    try {
      PyramidUtil.initJS(log);
      int tileSize = 512;
      PyramidUtil.initImagePyramids(
          log,
          "Image",
          tileSize,
          Arrays.stream(styleSources).map(styleSource -> ArtistryUtil.load(styleSource, -1)).toArray(i -> new BufferedImage[i])
      );
      int startLevel = 3;
      int magLevels = 2;
      int padding = 20;
      PyramidUtil.writeViewer(log, "resized_all", IntStream.range(0, styleSources.length).mapToObj(i -> {
        String localPrefix = "tile_" + i + "_";
        String globalPrefix = "file:///" + new File(log.getResourceDir(), localPrefix).getAbsolutePath();
        double mag = Math.pow(2, magLevels);
        Function imageFunction = image -> TestUtil.resize(image, (int) (image.getWidth() * mag));
        final double aspect = PyramidUtil.getAspect(styleSources[i]);
        new ImagePyramid(tileSize, startLevel, aspect, globalPrefix).buildNewImagePyramidLayer(magLevels, padding,
            imageFunction,
            true
        );
        ImagePyramid imagePyramid = new ImagePyramid(tileSize, startLevel + magLevels, aspect, localPrefix);
        PyramidUtil.writeViewer(log, localPrefix + String.format("%d_%d", startLevel, magLevels + startLevel), imagePyramid);
        return imagePyramid;
      }).toArray(i -> new ImagePyramid[i]));
    } catch (Throwable throwable) {
      log.eval(() -> {
        return throwable;
      });
    }
  }


}