# ***** BEGIN GPL LICENSE BLOCK *****
#
# 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 2
# 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.
#
# If you have Internet access, you can find the license text at
# http://www.gnu.org/licenses/gpl.txt,
# if not, write to the Free Software Foundation,
# Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
#
# ***** END GPL LICENCE BLOCK *****
# --------------------------------------------------------------------------

__bpydoc__ = """\
CreaPrim does what it says. I takes the active object and turns it into an Add Mesh addon.	When you
enable this, your custom object will be added to the Add->Mesh menu.


Documentation

Go to User Preferences->Addons and enable the CreaPrim addon in the Object section.
First select your object or objects.  The addon will show up in the 3dview properties panel.	
The name (in panel) will be set to the active object name.	Select "Apply transform" if you
want transforms to be applied to the selected objects.	Modifiers will taken into account.
You can always change this. Just hit the button and the selected
objects will be saved in your addons folder as an Add Mesh addon with the name 
"add_mesh_XXXX.py" with XXXX being your object name.  The addon will show up in User
Preferences->Addons in the Add Mesh section.  
Enable this addon et voila, your new custom primitive will now show up in the Add Mesh menu.

REMARK - dont need to be admin anymore - saves to user scripts dir
			
ALSO - dont forget to Apply rotation and scale to have your object show up correctly
"""

bl_info = {
    "name": "CreaPrim",
    "author": "Gert De Roost",
    "version": (0, 3, 10),
    "blender": (2, 64, 0),
    "location": "View3D > Object Tools",
    "description": "Create primitive addon",
    "warning": "",
    "wiki_url": "http://wiki.blender.org/index.php/Extensions:2.6/Py/"\
        "Scripts",
    "tracker_url": "https://projects.blender.org/tracker/index.php?"\
        "func=detail&aid=32801",
    "category": "Object"}

if "bpy" in locals():
	import imp


import bpy
import bmesh
import os


started = 0
oldname = ""

bpy.types.Scene.Name = bpy.props.StringProperty(
			name="Name",
			description="name for primitive",
			maxlen= 1024)

bpy.types.Scene.Apply = bpy.props.BoolProperty(
		name = "Apply transform", 
		description = "apply transform to selected objects",
		default = False)


class CreaPrim(bpy.types.Operator):
	bl_idname = "object.creaprim"
	bl_label = "CreaPrim"
	bl_description = "Create primitive addon."
	bl_options = {"REGISTER"}
	
	
	@classmethod
	def poll(cls, context):
		obj = context.active_object
		return (obj and obj.type == 'MESH' and context.mode == 'OBJECT')

	def invoke(self, context, event):
		
		global oldname, groupname, message
		
		scn = bpy.context.scene
				
		objlist = []
		for selobj in bpy.context.scene.objects:
			if selobj.select:
				objlist.append(selobj)
				
		try:
			direc = bpy.utils.script_paths()[1]
			scriptdir = 1
		except:
			direc = bpy.utils.script_paths()[0]
			scriptdir = 0
		if len(objlist) > 1:
			groupname = scn.Name
			groupname = groupname.replace(".", "")
			addondir = direc + os.sep + "addons" + os.sep + "add_mesh_" + groupname + os.sep
			if not os.path.exists(addondir):
				os.makedirs(addondir)
		else:
			groupname = scn.Name
			print (bpy.utils.script_paths())
			addondir = direc + os.sep + "addons" + os.sep
			print (addondir)
			if not os.path.exists(addondir):
				os.makedirs(addondir)
			
		actobj = bpy.context.active_object
		txtlist = []
		namelist = []
		for selobj in objlist:
			if len(objlist) == 1:
				objname = scn.Name
				objname = objname.replace(" ", "_")
			else:
				objname = selobj.name
				objname = objname.replace(".", "")
				objname = objname.replace(" ", "_")
				namelist.append(objname)
			mesh = selobj.to_mesh(scn, True, "PREVIEW")
			oldname = selobj.name
			scn.objects.active = selobj
			if scn.Apply:
				bpy.ops.object.transform_apply(location=False, rotation=True, scale=True)
			txt = do_creaprim(self, mesh, objname, addondir)
			if txt == 0:
				return {'CANCELLED'}
			txtlist.append(txt)
		oldname = actobj.name
		scn.objects.active = actobj
			
		if len(txtlist) > 1:
			makeinit(txtlist, namelist, groupname, addondir)
			bpy.ops.wm.addon_enable(module="add_mesh_" + groupname)
		else:
			bpy.ops.wm.addon_enable(module="add_mesh_" + str.lower(objname))
			
		if scriptdir == 1:
			message = "Add Mesh addon " + groupname + " saved to user scripts directory."
		else:
			message = "Add Mesh addon " + groupname + " saved to main scripts directory."
		bpy.ops.creaprim.message('INVOKE_DEFAULT')
		
		return {'FINISHED'}


class MessageOperator(bpy.types.Operator):
	bl_idname = "creaprim.message"
	bl_label = "Saved"
	 
	def invoke(self, context, event):
		wm = context.window_manager
		return wm.invoke_popup(self, width=500, height=20)
		return {'FINISHED'}
 
	def draw(self, context):
		
		global groupname
		
		layout = self.layout
		row = layout.row()
		row.label(text = '', icon = "PLUGIN")
		row.label(message)
		
def panel_func(self, context):
	
	global started
	
	scn = bpy.context.scene
		
	self.layout.label(text="CreaPrim:")
	self.layout.operator("object.creaprim", text="Create primitive", icon = 'PLUGIN')
	self.layout.prop(scn, "Name")
	self.layout.prop(scn, "Apply")

def register():
	bpy.utils.register_module(__name__)
	bpy.types.VIEW3D_PT_tools_objectmode.append(panel_func)
	bpy.app.handlers.scene_update_post.append(setname)
	
def unregister():
	bpy.utils.unregister_module(__name__)
	bpy.types.VIEW3D_PT_tools_objectmode.remove(panel_func)
	bpy.app.handlers.scene_update_post.remove(setname)

if __name__ == "__main__":
	register()




def do_creaprim(self, mesh, objname, addondir):
	
	global message
	
	objname = objname.replace(".", "")
	objname = objname.replace(" ", "_")
	bm = bmesh.new()
	bm.from_mesh(mesh)	
	
	
	try:
		txt = bpy.data.texts[str.lower("add_mesh_" + objname) + ".py"]
		txt.clear()
	except:
		txt = bpy.data.texts.new("add_mesh_" + str.lower(objname) + ".py")
	
	strlist = []
	strlist.append("bl_info = {\n")
	strlist.append("\"name\": \"" + objname + "\", \n")
	strlist.append("\"author\": \"Gert De Roost\",\n")
	strlist.append("\"version\": (1, 0, 0),\n")
	strlist.append("\"blender\": (2, 65, 0),\n")
	strlist.append("\"location\": \"Add > Mesh\",\n")
	strlist.append("\"description\": \"Create " + objname + " primitive.\",\n")
	strlist.append("\"warning\": \"\",\n")
	strlist.append("\"wiki_url\": \"\",\n")
	strlist.append("\"tracker_url\": \"\",\n")
	strlist.append("\"category\": \"Add Mesh\"}\n")
	strlist.append("\n")
	strlist.append("\n") 
	strlist.append("if \"bpy\" in locals():\n")
	strlist.append("	   import imp\n")
	strlist.append("\n")
	strlist.append("\n")
	strlist.append("import bpy\n")
	strlist.append("import bmesh\n")
	strlist.append("import math\n")
	strlist.append("from mathutils import *\n")
	strlist.append("\n")
	strlist.append("\n")
	strlist.append("\n")
	strlist.append("\n")
	strlist.append("class " + objname + "(bpy.types.Operator):\n")
	strlist.append("	bl_idname = \"mesh." + str.lower(objname) + "\"\n")
	strlist.append("	bl_label = \"" + objname + "\"\n")
	strlist.append("	bl_options = {\'REGISTER\', \'UNDO\'}\n")
	strlist.append("	bl_description = \"add " + objname + " primitive\"\n")
	strlist.append("\n")
	strlist.append("	def invoke(self, context, event):\n")
	strlist.append("\n")
	strlist.append("		mesh = bpy.data.meshes.new(name=\"" + objname + "\")\n")
	strlist.append("		obj = bpy.data.objects.new(name=\"" + objname + "\", object_data=mesh)\n")
	strlist.append("		scene = bpy.context.scene\n")
	strlist.append("		scene.objects.link(obj)\n")
	strlist.append("		obj.location = scene.cursor_location\n")
	strlist.append("		bm = bmesh.new()\n")
	strlist.append("		bm.from_mesh(mesh)\n")
	strlist.append("\n")
	strlist.append("		idxlist = []\n")
	posn = 0
	strlist.append("		vertlist = [")
	for v in bm.verts:
		if posn > 0:
			strlist.append(", ")
		posn += 1
		strlist.append(str(v.co[:]))
	strlist.append("]\n")	
	strlist.append("		for co in vertlist:\n")
	strlist.append("			v = bm.verts.new(co)\n")
	strlist.append("			bm.verts.index_update()\n")
	strlist.append("			idxlist.append(v.index)\n")
	posn = 0
	strlist.append("		edgelist = [")
	for e in bm.edges:
		if posn > 0:
			strlist.append(", ")
		posn += 1
		strlist.append("[" + str(e.verts[0].index) + ", " + str(e.verts[1].index) + "]")
	strlist.append("]\n")	
	strlist.append("		for verts in edgelist:\n")
	strlist.append("			try:\n")
	strlist.append("				bm.edges.new((bm.verts[verts[0]], bm.verts[verts[1]]))\n")
	strlist.append("			except:\n")
	strlist.append("				pass\n")
	posn1 = 0
	strlist.append("		facelist = [(")
	for f in bm.faces:
		if posn1 > 0:
			strlist.append(", (")
		posn1 += 1
		posn2 = 0
		for v in f.verts:
			if posn2 > 0:
				strlist.append(", ")
			strlist.append(str(v.index))
			posn2 += 1
		strlist.append(")")
	strlist.append("]\n")	
	strlist.append("		for verts in facelist:\n")
	strlist.append("			vlist = []\n")	
	strlist.append("			for idx in verts:\n")	
	strlist.append("				vlist.append(bm.verts[idxlist[idx]])\n")	
	strlist.append("			try:\n")
	strlist.append("				bm.faces.new(vlist)\n")
	strlist.append("			except:\n")
	strlist.append("				pass\n")
	strlist.append("\n")
	strlist.append("		bm.to_mesh(mesh)\n")
	strlist.append("		mesh.update()\n")
	strlist.append("		bm.free()\n")	
	strlist.append("		obj.rotation_quaternion = (Matrix.Rotation(math.radians(90), 3, \'X\').to_quaternion())\n")
	strlist.append("\n")
	strlist.append("		return {\'FINISHED\'}\n")
		
	strlist.append("\n")
	strlist.append("\n")
	strlist.append("\n")
	strlist.append("\n") 
	strlist.append("def menu_item(self, context):\n")	
	strlist.append("	   self.layout.operator(" + objname + ".bl_idname, text=\"" + objname + "\", icon=\"PLUGIN\")\n")
	strlist.append("\n") 
	strlist.append("def register():\n")
	strlist.append("	   bpy.utils.register_module(__name__)\n")
	strlist.append("	   bpy.types.INFO_MT_mesh_add.append(menu_item)\n")
	strlist.append("\n") 
	strlist.append("def unregister():\n")
	strlist.append("	   bpy.utils.unregister_module(__name__)\n")
	strlist.append("	   bpy.types.INFO_MT_mesh_add.remove(menu_item)\n")
	strlist.append("\n") 
	strlist.append("if __name__ == \"__main__\":\n") 
	strlist.append("	   register()\n")	
	endstring = ''.join(strlist)
	txt.write(endstring)
	
	try:
		fileobj = open(addondir + "add_mesh_" + str.lower(objname) + ".py", "w")
	except:
		message = "Permission problem - cant write file - run Blender as Administrator!"
		bpy.ops.creaprim.message('INVOKE_DEFAULT')
		return 0

	fileobj.write(endstring)
	fileobj.close()
	
	bm.free()
	
	return txt
	
	
def makeinit(txtlist, namelist, groupname, addondir):
	
	global message
	
	try:
		txt = bpy.data.texts["__init__.py"]
		txt.clear()
	except:
		txt = bpy.data.texts.new("__init__.py")
	
	strlist = []
	strlist.append("bl_info = {\n")
	strlist.append("\"name\": \"" + groupname + "\", \n")
	strlist.append("\"author\": \"Gert De Roost\",\n")
	strlist.append("\"version\": (1, 0, 0),\n")
	strlist.append("\"blender\": (2, 65, 0),\n")
	strlist.append("\"location\": \"Add > Mesh\",\n")
	strlist.append("\"description\": \"Create " + groupname + " primitive group.\",\n")
	strlist.append("\"warning\": \"\",\n")
	strlist.append("\"wiki_url\": \"\",\n")
	strlist.append("\"tracker_url\": \"\",\n")
	strlist.append("\"category\": \"Add Mesh\"}\n")
	strlist.append("\n")
	strlist.append("\n") 
	strlist.append("if \"bpy\" in locals():\n")
	strlist.append("	import imp\n")
	addonlist = []
	for txt in txtlist:
		name = txt.name.replace(".py", "")
		addonlist.append(name)
	for name in addonlist:
		strlist.append("	imp.reload(" + name + ")\n")	
	strlist.append("else:\n")
	for name in addonlist:
		strlist.append("	from . import " + name + "\n")	
	strlist.append("\n")
	strlist.append("\n")
	strlist.append("import bpy\n")
	strlist.append("\n")
	strlist.append("\n")
	strlist.append("\n")
	strlist.append("\n")
	strlist.append("class INFO_MT_mesh_" + str.lower(groupname) + "_add(bpy.types.Menu):\n")
	strlist.append("	bl_idname = \"INFO_MT_mesh_" + str.lower(groupname) + "_add\"\n")
	strlist.append("	bl_label = \"" + groupname + "\"\n")
	strlist.append("\n")
	strlist.append("	def draw(self, context):\n")
	strlist.append("		layout = self.layout\n")
#		 layout.operator_context = 'INVOKE_REGION_WIN'
	for name in namelist:
		strlist.append("		layout.operator(\"mesh." + str.lower(name) + "\", text=\"" + name + "\")\n")
	strlist.append("\n")
	strlist.append("\n")
	strlist.append("\n")
	strlist.append("\n") 
	strlist.append("def menu_item(self, context):\n")	
	strlist.append("	   self.layout.menu(\"INFO_MT_mesh_" + str.lower(groupname) + "_add\", icon=\"PLUGIN\")\n")
	strlist.append("\n") 
	strlist.append("def register():\n")
	strlist.append("	   bpy.utils.register_module(__name__)\n")
	strlist.append("	   bpy.types.INFO_MT_mesh_add.append(menu_item)\n")
	strlist.append("\n") 
	strlist.append("def unregister():\n")
	strlist.append("	   bpy.utils.unregister_module(__name__)\n")
	strlist.append("	   bpy.types.INFO_MT_mesh_add.remove(menu_item)\n")
	strlist.append("\n") 
	strlist.append("if __name__ == \"__main__\":\n") 
	strlist.append("	   register()\n")	
	endstring = ''.join(strlist)
	txt.write(endstring)
	
	try:
		fileobj = open(addondir + "__init__.py", "w")
	except:
		message = "Permission problem - cant write file - run Blender as Administrator!"
		bpy.ops.creaprim.message('INVOKE_DEFAULT')
		return 0
	fileobj.write(endstring)
	fileobj.close()



def setname(dummy):
	
	global oldname
	
	scn = bpy.context.scene
	
	if bpy.context.active_object.name != oldname:
		scn.Name = bpy.context.active_object.name
		oldname = scn.Name