parametricCone.py

from math import *
import numpy as np
from matrix import matrix
from parametricObject import parametricObject


class parametricCone(parametricObject):
    """
    These are the inputs to the initializer: cone T matrix, cone Height, cone Radius, cone Color, cone Reflectance,
    (0.0, 1.0), (0.0, 2.0 * pi), (1.0 / 10.0, pi / 18.0)

    So make some default values for the parameters and the pass it through the super constructor
    """

    def __init__(self, T=matrix(np.identity(4)), cone_height=20.0, cone_radius=10.0, color=(255, 255, 255),
                 reflectance=(0.2, 0.4, 0.4, 1.0),
                 uRange=(0.0, 1.0), vRange=(0.0, 2*pi), uvDelta=(1.0 / 10.0, pi / 18.0)):
        super().__init__(T, color, reflectance, uRange, vRange, uvDelta)
        self.__cone_height = cone_height
        self.__cone_radius = cone_radius

    def getPoint(self, u, v):
        """
        This will give the parametric point given the values of u and v
        :param u: the first parameter value between 0 and 1
        :param v: the second parameter value between 0 and 2*pi
        :return: the 4*1 matrix that represents the parametric form of the cone
        """
        P = matrix(np.ones((4, 1)))
        P.set(0, 0, (1-u)*self.__cone_radius*sin(v))
        P.set(1, 0, (1-u)*self.__cone_radius*cos(v))
        P.set(2, 0, self.__cone_height*u)
        return P

    def set_cone_height(self, cone_height):
        self.__cone_height = cone_height

    def get_cone_height(self):
        return self.__cone_height

    def set_cone_radius(self, cone_radius):
        self.__cone_radius = cone_radius

    def get_cone_radius(self):
        return self.__cone_radius