src/color/entityindex-colormaker.js

/**
 * @file Entityindex Colormaker
 * @author Alexander Rose <alexander.rose@weirdbyte.de>
 * @private
 */
 
import { ColormakerRegistry } from '../globals.js'
import Colormaker from './colormaker.js'
 
/**
 * Color by entiry index
 */
class EntityindexColormaker extends Colormaker {
  constructor (params) {
    super(params)
 
    if (!params.scale) {
      this.scale = 'Spectral'
    }
    if (!params.domain) {
      this.domain = [ 0, this.structure.entityList.length - 1 ]
    }
    var entityindexScale = this.getScale()
 
    this.atomColor = function (a) {
      return entityindexScale(a.entityIndex)
    }
  }
}
 
ColormakerRegistry.add('entityindex', EntityindexColormaker)
 
export default EntityindexColormaker