import { BlochVector } from '../math/bloch-vector'
import { Wedge } from './wedge'

/**
 * A wedge which is a quarter of a circle
 */
export class QubitProjWedge extends Wedge {
  constructor() {
    super()
  }

  follow(v: BlochVector) {
    const { theta, phi } = v
    if (theta > Math.PI / 2) {
      this.rotation.set(0, Math.PI, Math.PI - phi)
    } else {
      this.rotation.set(0, 0, phi)
    }
  }
}