TmMolStar/src/mol-geo/primitive/prism.ts

/**
 * Copyright (c) 2018-2019 mol* contributors, licensed under MIT, See LICENSE file for more info.
 *
 * @author Alexander Rose <alexander.rose@weirdbyte.de>
 */

import { Vec3 } from '../../mol-math/linear-algebra'
import { Primitive, PrimitiveBuilder } from './primitive';
import { polygon } from './polygon'
import { Cage } from './cage';

const on = Vec3.create(0, 0, -0.5), op = Vec3.create(0, 0, 0.5)
const a = Vec3.zero(), b = Vec3.zero(), c = Vec3.zero(), d = Vec3.zero()

/**
 * Create a prism with a base of 4 or more points
 */
export function Prism(points: ArrayLike<number>): Primitive {
    const sideCount = points.length / 3
    if (sideCount < 4) throw new Error('need at least 4 points to build a prism')

    const count = 4 * sideCount
    const builder = PrimitiveBuilder(count)

    // create sides
    for (let i = 0; i < sideCount; ++i) {
        const ni = (i + 1) % sideCount
        Vec3.set(a, points[i * 3], points[i * 3 + 1], -0.5)
        Vec3.set(b, points[ni * 3], points[ni * 3 + 1], -0.5)
        Vec3.set(c, points[ni * 3], points[ni * 3 + 1], 0.5)
        Vec3.set(d, points[i * 3], points[i * 3 + 1], 0.5)
        builder.add(a, b, c)
        builder.add(c, d, a)
    }

    // create bases
    for (let i = 0; i < sideCount; ++i) {
        const ni = (i + 1) % sideCount
        Vec3.set(a, points[i * 3], points[i * 3 + 1], -0.5)
        Vec3.set(b, points[ni * 3], points[ni * 3 + 1], -0.5)
        builder.add(on, b, a)
        Vec3.set(a, points[i * 3], points[i * 3 + 1], 0.5)
        Vec3.set(b, points[ni * 3], points[ni * 3 + 1], 0.5)
        builder.add(a, b, op)
    }

    return builder.getPrimitive()
}

let diamond: Primitive
export function DiamondPrism() {
    if (!diamond) diamond = Prism(polygon(4, false))
    return diamond
}

let pentagonalPrism: Primitive
export function PentagonalPrism() {
    if (!pentagonalPrism) pentagonalPrism = Prism(polygon(5, false))
    return pentagonalPrism
}

let hexagonalPrism: Primitive
export function HexagonalPrism() {
    if (!hexagonalPrism) hexagonalPrism = Prism(polygon(6, true))
    return hexagonalPrism
}

//

/**
 * Create a prism cage
 */
export function PrismCage(points: ArrayLike<number>): Cage {
    const sideCount = points.length / 3

    // const count = 4 * sideCount
    const vertices: number[] = []
    const edges: number[] = []

    let offset = 0

    // vertices and side edges
    for (let i = 0; i < sideCount; ++i) {
        vertices.push(
            points[i * 3], points[i * 3 + 1], -0.5,
            points[i * 3], points[i * 3 + 1], 0.5
        )
        edges.push(offset, offset + 1)
        offset += 2
    }

    // bases edges
    for (let i = 0; i < sideCount; ++i) {
        const ni = (i + 1) % sideCount
        edges.push(
            i * 2, ni * 2,
            i * 2 + 1, ni * 2 + 1
        )
    }

    return { vertices, edges }
}

let diamondCage: Cage
export function DiamondPrismCage() {
    if (!diamondCage) diamondCage = PrismCage(polygon(4, false))
    return diamondCage
}

let pentagonalPrismCage: Cage
export function PentagonalPrismCage() {
    if (!pentagonalPrismCage) pentagonalPrismCage = PrismCage(polygon(5, false))
    return pentagonalPrismCage
}

let hexagonalPrismCage: Cage
export function HexagonalPrismCage() {
    if (!hexagonalPrismCage) hexagonalPrismCage = PrismCage(polygon(6, true))
    return hexagonalPrismCage
}