up

kbengine · kbengine · commit be1e63a29800 · 2014-08-15T16:12:07.000+08:00
diff --git a/kbe/src/lib/pyscript/pyprofile.cpp b/kbe/src/lib/pyscript/pyprofile.cpp
@@ -144,10 +144,16 @@ bool PyProfile::remove(std::string profile)
 }
 
 //-------------------------------------------------------------------------------------
-void PyProfile::print_stats(const std::string& sort = "time")
+void PyProfile::print_stats(const std::string& sort, const std::string& profileName)
 {
+	PyProfile::PROFILES::iterator iter = profiles_.find(profileName.c_str());
+	if(iter == profiles_.end())
+	{
+		return;
+	}
+
 	PyObject* pyRet = PyObject_CallMethod(iter->second.get(), const_cast<char*>("print_stats"),
-		const_cast<char*>("s"), const_cast<char*>("time"));
+		const_cast<char*>("s"), const_cast<char*>(sort.c_str()));
 	
 	if(pyRet)
 		Py_DECREF(pyRet);
diff --git a/kbe/src/lib/pyscript/pyprofile.hpp b/kbe/src/lib/pyscript/pyprofile.hpp
@@ -44,7 +44,7 @@ class PyProfile
 	static void addToStream(std::string profile, MemoryStream* s);
 	static bool remove(std::string profile);
 
-	static void print_stats(const std::string& sort = "time");
+	static void print_stats(const std::string& sort = "time", const std::string& profileName = "kbengine");
 
 	/** 
 		��ʼ��pickler 
