.vimrc

Erobaby · Sep 15, 2014 · 802327d · 802327d
1 parent 7278383
commit 802327d
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/.vimrc b/.vimrc
@@ -139,7 +139,7 @@ func SetTitle()
 		call append(line(".")+4, " ************************************************************************/") 
 		call append(line(".")+5, "")
 	endif
-	if &filetype == 'cpp'
+	if expand("%:e") == 'cpp'
 		call append(line(".")+6, "#include<iostream>")
 		call append(line(".")+7, "using namespace std;")
 		call append(line(".")+8, "")
@@ -148,6 +148,11 @@ func SetTitle()
 		call append(line(".")+6, "#include<stdio.h>")
 		call append(line(".")+7, "")
 	endif
+    if expand("%:e")== 'h'
+        call append(line(".")+6, "#ifndef _".toupper(expand("%:r"))."_H")
+        call append(line(".")+7, "#define _".toupper(expand("%:r"))."_H")
+		call append(line(".")+8, "#endif")
+	endif
 	if &filetype == 'java'
 		call append(line(".")+6,"public class ".expand("%:r"))
 		call append(line(".")+7,"")