import bpy

class DataButtonsPanel(bpy.types.Panel):
	__space_type__ = 'PROPERTIES'
	__region_type__ = 'WINDOW'
	__context__ = "data"
	
	def poll(self, context):
		return (context.object and context.object.type in ('CURVE', 'SURFACE') and context.curve)
		
class DataButtonsPanelCurve(DataButtonsPanel):
	'''
	Same as above but for curves only
	'''
	def poll(self, context):
		return (context.object and context.object.type == 'CURVE' and context.curve)

class DataButtonsPanelActive(DataButtonsPanel):
	'''
	Same as above but for curves only
	'''
	def poll(self, context):
		curve = context.curve
		return (curve and curve.active_spline)

class DATA_PT_context_curve(DataButtonsPanel):
	__show_header__ = False
	
	def draw(self, context):
		layout = self.layout
		
		ob = context.object
		curve = context.curve
		space = context.space_data

		split = layout.split(percentage=0.65)

		if ob:
			split.template_ID(ob, "data")
			split.itemS()
		elif curve:
			split.template_ID(space, "pin_id")
			split.itemS()

class DATA_PT_shape_curve(DataButtonsPanel):
	__label__ = "Shape"
	
	def draw(self, context):
		layout = self.layout
		
		ob = context.object
		curve = context.curve
		space = context.space_data
		is_surf = (ob.type == 'SURFACE')

		if not is_surf:
			row = layout.row()
			row.itemR(curve, "curve_2d")			
			row.itemR(curve, "use_twist_correction")
		
		split = layout.split()
		
		col = split.column()
		
		if not is_surf:
			sub = col.column()
			sub.active = curve.curve_2d
			sub.itemL(text="Caps:")
			row = sub.row()
			row.itemR(curve, "front")
			row.itemR(curve, "back")
			
		col.itemL(text="Textures:")
#		col.itemR(curve, "uv_orco")
		col.itemR(curve, "auto_texspace")
			
		col = split.column()	
		col.itemL(text="Resolution:")
		sub = col.column(align=True)
		sub.itemR(curve, "resolution_u", text="Preview U")
		sub.itemR(curve, "render_resolution_u", text="Render U")

		if is_surf:
			sub = col.column(align=True)
			sub.itemR(curve, "resolution_v", text="Preview V")
			sub.itemR(curve, "render_resolution_v", text="Render V")
		

#		col.itemL(text="Display:")
#		col.itemL(text="HANDLES")
#		col.itemL(text="NORMALS")
#		col.itemR(curve, "vertex_normal_flip")

class DATA_PT_geometry_curve(DataButtonsPanelCurve):
	__label__ = "Geometry "

	def draw(self, context):
		layout = self.layout
		
		curve = context.curve
		
		split = layout.split()
	
		col = split.column()
		col.itemL(text="Modification:")
		col.itemR(curve, "width")
		col.itemR(curve, "extrude")
		col.itemR(curve, "taper_object", icon='ICON_OUTLINER_OB_CURVE')
		
		col = split.column()
		col.itemL(text="Bevel:")
		col.itemR(curve, "bevel_depth", text="Depth")
		col.itemR(curve, "bevel_resolution", text="Resolution")
		col.itemR(curve, "bevel_object", icon='ICON_OUTLINER_OB_CURVE')

	
class DATA_PT_pathanim(DataButtonsPanelCurve):
	__label__ = "Path Animation"
	
	def draw_header(self, context):
		curve = context.curve

		self.layout.itemR(curve, "use_path", text="")

	def draw(self, context):
		layout = self.layout
		
		curve = context.curve
		
		layout.active = curve.use_path	
		
		split = layout.split()		
		
		col = split.column()
		col.itemR(curve, "path_length", text="Frames")
		col.itemR(curve, "use_path_follow")

		col = split.column()
		col.itemR(curve, "use_stretch")
		col.itemR(curve, "use_time_offset", text="Offset Children")
	
class DATA_PT_active_spline(DataButtonsPanelActive):
	__label__ = "Active Spline"

	def draw(self, context):
		layout = self.layout
		
		ob = context.object
		curve = context.curve
		act_spline = curve.active_spline
		is_surf = (ob.type == 'SURFACE')
		is_poly = (act_spline.type == 'POLY')
		
		split = layout.split()
		
		if is_poly:
			# These settings are below but its easier to have 
			# poly's set aside since they use so few settings
			col = split.column()
			col.itemL(text="Cyclic:")
			col.itemR(act_spline, "smooth")
			col = split.column()
			col.itemR(act_spline, "cyclic_u", text="U")
		
		else:
			col = split.column()
			col.itemL(text="Cyclic:")
			if act_spline.type == 'NURBS':
				col.itemL(text="Bezier:")
				col.itemL(text="Endpoint:")
				col.itemL(text="Order:")
			
			col.itemL(text="Resolution:")
					
			col = split.column()
			col.itemR(act_spline, "cyclic_u", text="U")
			
			if act_spline.type == 'NURBS':
				sub = col.column()
				# sub.active = (not act_spline.cyclic_u)
				sub.itemR(act_spline, "bezier_u", text="U")
				sub.itemR(act_spline, "endpoint_u", text="U")
				
				sub = col.column()
				sub.itemR(act_spline, "order_u", text="U")
			col.itemR(act_spline, "resolution_u", text="U")
			
			if is_surf:
				col = split.column()
				col.itemR(act_spline, "cyclic_v", text="V")
				
				# its a surface, assume its a nurb.
				sub = col.column()
				sub.active = (not act_spline.cyclic_v)
				sub.itemR(act_spline, "bezier_v", text="V")
				sub.itemR(act_spline, "endpoint_v", text="V")
				sub = col.column()
				sub.itemR(act_spline, "order_v", text="V")
				sub.itemR(act_spline, "resolution_v", text="V")

			
			if not is_surf:
				split = layout.split()
				col = split.column()
				col.active = (not curve.curve_2d)
				
				col.itemL(text="Interpolation:")
				col.itemR(act_spline, "tilt_interpolation", text="Tilt")
				col.itemR(act_spline, "radius_interpolation", text="Radius")
			
			split = layout.split()
			col = split.column()
			col.itemR(act_spline, "smooth")


bpy.types.register(DATA_PT_context_curve)
bpy.types.register(DATA_PT_shape_curve)
bpy.types.register(DATA_PT_geometry_curve)
bpy.types.register(DATA_PT_pathanim)
bpy.types.register(DATA_PT_active_spline)