scalismo.vtk.initialize.scala Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of scalismo-vtk_3 Show documentation
scalismo.vtk
The newest version!
/*
 * Copyright 2015 University of Basel, Graphics and Vision Research Group
 *
 * 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.
 */
package scalismo.vtk

import ch.unibas.cs.gravis.vtkjavanativelibs.VtkNativeLibraries
import vtk.vtkObjectBase

import javax.swing.SwingUtilities

object initialize {
  // this is a hacky way to get an object that can be synchronized on, with a mutable value.
  private val initialized = Array.fill(1)(false)

  /**
   * Initialize and load the required native libraries
   *
   * @param ignoreErrors
   *   ignore failures when trying to load libraries. Only set this if you know what you are doing!
   * @param gcInterval
   *   time interval (in milliseconds) for running the vtk garbage collection. A value <= 0 means that garbage
   *   collection is not run automatically.
   */
  def apply(ignoreErrors: Boolean = false, gcInterval: Long = 60 * 1000) = initialized.synchronized {
    import java.io.File
    val nativeDir = new File(System.getProperty("user.home") + File.separator + ".scalismo")
    if (!initialized(0)) {
      VtkNativeLibraries.initialize(nativeDir)

      if (gcInterval > 0) {
        setupVTKGCThread(gcInterval)
      }
      initialized(0) = true
    }
  }

  private def setupVTKGCThread(gcInterval: Long): Unit = {

    val runGC = new Runnable() {
      override def run(): Unit = {
        vtkObjectBase.JAVA_OBJECT_MANAGER.gc(false)
      }
    }

    val gcThread = new Thread {
      override def run(): Unit = {
        while (true) {
          Thread.sleep(gcInterval)

          // As vtk is very sensitive to threading issues, we run the gc on the EDT thread.
          SwingUtilities.invokeLater(runGC);

        }
      }
    }

    gcThread.setDaemon(true)
    gcThread.start()
  }
}