de.sciss.fscape.stream.ImageFileOut.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of fscape-core_2.11 Show documentation

An audio rendering library

There is a newer version: 2.28.0

/*
 *  ImageFileOut.scala
 *  (FScape)
 *
 *  Copyright (c) 2001-2016 Hanns Holger Rutz. All rights reserved.
 *
 *  This software is published under the GNU General Public License v2+
 *
 *
 *  For further information, please contact Hanns Holger Rutz at
 *  [email protected]
 */

package de.sciss.fscape
package stream

import akka.stream.Attributes
import de.sciss.file._
import de.sciss.fscape.graph.ImageFile.Spec
import de.sciss.fscape.stream.impl.{BlockingGraphStage, ImageFileOutImpl, NodeImpl, UniformSinkShape}

import scala.collection.immutable.{IndexedSeq => Vec, Seq => ISeq}

object ImageFileOut {
  def apply(file: File, spec: Spec, in: ISeq[OutD])(implicit b: Builder): Unit = {
    require (spec.numChannels == in.size, s"Channel mismatch (spec has ${spec.numChannels}, in has ${in.size})")
    val sink = new Stage(file, spec)
    val stage = b.add(sink)
    (in zip stage.inlets).foreach { case (output, input) =>
      b.connect(output, input)
    }
  }

  private final val name = "ImageFileOut"

  private type Shape = UniformSinkShape[BufD]

  private final class Stage(f: File, spec: Spec)(implicit protected val ctrl: Control)
    extends BlockingGraphStage[Shape](s"$name(${f.name})") {

    override val shape = UniformSinkShape[BufD](Vector.tabulate(spec.numChannels)(ch => InD(s"$name.in$ch")))

    def createLogic(attr: Attributes) = new Logic(shape, f, spec)
  }

  private final class Logic(shape: Shape, f: File, protected val spec: Spec)(implicit ctrl: Control)
    extends NodeImpl(s"$name(${f.name})", shape)
    with ImageFileOutImpl[Shape] {

    protected val inlets1: Vec[InD ] = shape.inlets.toIndexedSeq

    shape.inlets.foreach(setHandler(_, this))

    override def preStart(): Unit = {
      super.preStart()
      openImage(f)
    }

    protected def process1(): Unit = {
      val chunk = readIns1()
      if (chunk > 0) {
        processChunk(inOff = 0, chunk = chunk)
      }
      if (framesWritten == numFrames) {
        logStream(s"completeStage() $this")
        completeStage()
      }
    }
  }
}