scalismo.ui.rendering.Caches.scala Maven / Gradle / Ivy

Go to download
/*
 * Copyright (C) 2016  University of Basel, Graphics and Vision Research Group
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see .
 */

package scalismo.ui.rendering

import scalismo.common.DiscreteField.{ScalarMeshField, ScalarVolumeMeshField}
import scalismo.common.{DiscreteDomain, DiscreteField, UnstructuredPointsDomain}
import scalismo.geometry._3D
import scalismo.image.DiscreteImage
import scalismo.mesh._
import scalismo.ui.util.Cache
import vtk.{vtkPolyData, vtkStructuredPoints, vtkUnstructuredGrid}

object Caches {

  /*
  It turns out that the triangleMesh spends quite a lot of time computing the hash code. As we know here
  that we are only interested in the geometry, we can speed up computation a lot by caching ourselves
   */
  case class FastCachingTriangleMesh(tm: TriangleMesh3D) {

    override lazy val hashCode: Int = (31 + tm.pointSet.hashCode()) * (31 + tm.triangulation.hashCode())
  }

  case class FastCachingVertexColorMesh(mesh: VertexColorMesh3D) {
    override lazy val hashCode: Int =
      (31 + mesh.shape.pointSet.hashCode()) * (31 + mesh.shape.triangulation.hashCode() * 31 + mesh.color.hashCode())
  }

  case class FastCachingTetrahedralMesh(mesh: TetrahedralMesh3D) {
    override lazy val hashCode: Int =
      (31 + mesh.pointSet.hashCode()) * (31 + mesh.tetrahedralization.hashCode())
  }

  final val TriangleMeshCache = new Cache[FastCachingTriangleMesh, vtkPolyData]
  final val ImageCache = new Cache[DiscreteImage[_3D, _], vtkStructuredPoints]
  final val ScalarMeshFieldCache = new Cache[ScalarMeshField[Float], vtkPolyData]
  final val ScalarFieldCache = new Cache[DiscreteField[_3D, UnstructuredPointsDomain, Float], vtkPolyData]
  final val VertexColorMeshCache = new Cache[FastCachingVertexColorMesh, vtkPolyData]
  final val TetrahedralMeshCache = new Cache[FastCachingTetrahedralMesh, vtkUnstructuredGrid]
  final val ScalarTetrahedralMeshFieldCache = new Cache[ScalarVolumeMeshField[Float], vtkUnstructuredGrid]

}