com.simiacryptus.mindseye.art.util.view.RotatedVector.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of deepartist.org Show documentation

DeepArt Studio powered by MindsEye

The newest version!

package com.simiacryptus.mindseye.art.util.view

import com.simiacryptus.mindseye.art.util.{Permutation, ViewMask}
import com.simiacryptus.mindseye.lang.Layer
import com.simiacryptus.mindseye.layers.java.AffineImgViewLayer

case class RotatedVector
(
  rotation: Map[Double, Permutation] = Map.empty,
  symmetry: Boolean = true,
  mask: ViewMask = ViewMask()
) extends ImageView {

  def getView(canvasDims: Array[Int]): Layer = {
    def maybeLayers = for ((rotation, permutation) <- rotation.toArray) yield {
      val layer = new AffineImgViewLayer(canvasDims(0), canvasDims(1), true)
      layer.setChannelSelector(permutation.indices)
      val canvasXY = canvasDims.take(2)
      val r = Math.pow(canvasXY.reduce(_ * _), 1.0 / canvasXY.size)
      layer.setrMin((mask.radius_min * r).toInt)
      layer.setrMax((mask.radius_max * r).toInt)
      layer.setxMin((mask.x_min * canvasDims(0)).toInt)
      layer.setxMax((mask.x_max * canvasDims(0)).toInt)
      layer.setyMin((mask.y_min * canvasDims(1)).toInt)
      layer.setyMax((mask.y_max * canvasDims(1)).toInt)
      if (rotation != 0) {
        layer.setRotationCenterX(canvasDims(0) / 2)
        layer.setRotationCenterY(canvasDims(1) / 2)
        layer.setRotationRadians(rotation % (2 * Math.PI))
      }
      Option(layer)
    }

    if (symmetry) {
      avg(Array(None) ++ maybeLayers)
    } else {
      avg(maybeLayers)
    }
  }

}