polystrips_draw.py

'''
Copyright (C) 2014 CG Cookie
http://cgcookie.com
hello@cgcookie.com

Created by Jonathan Denning, Jonathan Williamson, and Patrick Moore

    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 <http://www.gnu.org/licenses/>.
'''

# System imports
import copy
import math
import itertools
import time
from mathutils import Vector, Quaternion, Matrix
from mathutils.geometry import intersect_point_line, intersect_line_plane

# Blender imports
import bgl
import blf
import bmesh
import bpy
from bpy_extras.view3d_utils import location_3d_to_region_2d, region_2d_to_vector_3d, region_2d_to_location_3d, region_2d_to_origin_3d

# Common imports
from .lib import common_utilities
from .lib import common_drawing
from .lib.common_utilities import iter_running_sum, dprint, get_object_length_scale, profiler, AddonLocator


#Make the addon name and location accessible
AL = AddonLocator()

def draw_gedge_text(gedge,context, text):
    l = len(gedge.cache_igverts)
    if l > 4:
        n_quads = math.floor(l/2) + 1
        mid_vert_ind = math.floor(l/2)
        mid_vert = gedge.cache_igverts[mid_vert_ind]
        position_3d = mid_vert.position + 1.5 * mid_vert.tangent_y * mid_vert.radius
    else:
        position_3d = (gedge.gvert0.position + gedge.gvert3.position)/2
    
    position_2d = location_3d_to_region_2d(context.region, context.space_data.region_3d,position_3d)
    txt_width, txt_height = blf.dimensions(0, text)
    blf.position(0, position_2d[0]-(txt_width/2), position_2d[1]-(txt_height/2), 0)
    blf.draw(0, text)

def draw_gedge_info(gedge,context):
    '''
    helper draw module to display info about the Gedge
    '''
    l = len(gedge.cache_igverts)
    if l > 4:
        n_quads = math.floor(l/2) + 1
    else:
        n_quads = 3
    draw_gedge_text(gedge, context, str(n_quads))

def cubic_bezier_surface_t(v00,v01,v02,v03, v10,v11,v12,v13, v20,v21,v22,v23, v30,v31,v32,v33, t02,t13):
    b00,b01,b02,b03 = cubic_bezier_weights(t02)
    b10,b11,b12,b13 = cubic_bezier_weights(t13)
    v0 = v00*b00*b10 + v01*b01*b10 + v02*b02*b10 + v03*b03*b10
    v1 = v10*b00*b11 + v11*b01*b11 + v12*b02*b11 + v13*b03*b11
    v2 = v20*b00*b12 + v21*b01*b12 + v22*b02*b12 + v23*b03*b12
    v3 = v30*b00*b13 + v31*b01*b13 + v32*b02*b13 + v33*b03*b13
    return v0+v1+v2+v3