代码统计工具(支持C#,VB.NET)

小玩意,不多说,想看的一看就会,

界面如下:

 

using System;
using System.Collections.Generic;
using System.ComponentModel;
using System.Data;
using System.Drawing;
using System.Text;
using System.Windows.Forms;
using System.IO;
namespace 代码行数统计工具
{
    public partial class frmCountCodes : Form
    {
        string curDir="";//当前目录
        string wrfile = "";//代码写入的文件
        string filter = "";//要查找的文件类型
        bool chtxt = false;//是否手动更改目录
        //private delegate void desearch(string str);//查找委托
        long count = 0;//行数
        int filecount = 0;//发现文件个数
        StringBuilder strinfo;//返回信息
        bool ischeck = false;//是否为连续多行的注释 如C#中的 /*...*/
        public frmCountCodes()
        {
            InitializeComponent();
        }
        /// <summary>
        /// 设置目录
        /// </summary>
        /// <param name="sender"></param>
        /// <param name="e"></param>
        private void btnDir_Click(object sender, EventArgs e)
        {
            FolderBrowserDialog fbd = new FolderBrowserDialog();
            fbd.Description = "选择要统计的文件所在目录";
            fbd.RootFolder =Environment.SpecialFolder.MyComputer;
            fbd.ShowNewFolderButton = false;
            fbd.ShowDialog();
            if (fbd.SelectedPath.Trim() != "")
            {
                curDir = fbd.SelectedPath;
                txtDir.Text = curDir;
            }
        }

        private void txtDir_TextChanged(object sender, EventArgs e)
        {
            chtxt = true;           
        }

        private void txtDir_Leave(object sender, EventArgs e)
        {
            if (chtxt)
            {
                if (Directory.Exists(txtDir.Text.Trim()))
                {
                    curDir = txtDir.Text.Trim();
                }
                else
                {
                    MessageBox.Show("此并不是一个有效的目录！请重新设置。");
                    txtDir.Text = curDir;
                }
            }
        }

      
        //统计
        private void btnser_Click(object sender, EventArgs e)
        {
            try
            {
                btnser.Enabled = !btnser.Enabled;
                if (!Directory.Exists(curDir))
                {
                    MessageBox.Show("此并不是一个有效的目录！请重新设置。");
                    btnser.Enabled = !btnser.Enabled;
                    return;
                }
                filter = "";
                //desearch searchfile = new desearch(search);
                count = 0;
                filecount = 0;
                strinfo = new StringBuilder();               
                if (chvb.Checked)//选择了VB
                {
                    filter += "*.vb,";
                    //IAsyncResult ir=searchfile.BeginInvoke("*.vb", null, null);
                    //while (!ir.IsCompleted)//异步执行
                    //{
                    //    Application.DoEvents();
                    //}
                    //searchfile.EndInvoke(ir);
                    search("*.vb");
                }
                if (chCSharp.Checked)//选择了C#
                {
                    filter += "*.cs,";
                    //IAsyncResult ir = searchfile.BeginInvoke("*.cs", null, null);
                    //while (!ir.IsCompleted)//异步执行
                    //{
                    //    Application.DoEvents();
                    //}
                    //searchfile.EndInvoke(ir);  
                    search("*.cs");
                }
                if (filter.Trim() == "")
                {
                    MessageBox.Show("请选择要统计的语言种类");
                    btnser.Enabled = !btnser.Enabled;
                    return;
                }
                strinfo.AppendLine("主目录：" + curDir);
                strinfo.AppendLine("所选择的文件类型有:" + filter.Substring(0, filter.Length - 1));
                strinfo.AppendLine("发现文件个数为：" + filecount);
                strinfo.AppendLine("代码总行数为：" + count);
                txtinfo.Text = strinfo.ToString();
            }
            catch
            {
                MessageBox.Show("未知错误，请确保路径正确和其子目录可读！", "", MessageBoxButtons.OK, MessageBoxIcon.Error, MessageBoxDefaultButton.Button1, MessageBoxOptions.DefaultDesktopOnly);
            }           
            btnser.Enabled = !btnser.Enabled;
           
        }
        private void search(string ftype)
        {           
            string[] files;           
            if (chChirld.Checked)
            {
                files = Directory.GetFiles(curDir, ftype, SearchOption.AllDirectories);//遍历所有子目录
            }
            else
            {
                files = Directory.GetFiles(curDir, ftype, SearchOption.TopDirectoryOnly);
            }
            StreamWriter sw =null;               
            if (wrfile != "")
            {
                try
                {
                    if (!File.Exists(wrfile))
                    {
                        FileStream f = File.Create(wrfile);
                        f.Close();
                    }
                    sw = new StreamWriter(wrfile,false,Encoding.GetEncoding("gb2312"));                   
                }
                catch
                {
                    sw = null;
                }
            }
            foreach (string file in files)
            {
                if (file.EndsWith(ftype.Replace("*", "Designer"))) continue;//资源文件
                filecount ++;
                StreamReader sr = new StreamReader(file,Encoding.GetEncoding("gb2312"));
                lbSearchFile.Text = "当前统计文件：" + Path.GetFileName(file);
                Application.DoEvents();
                string line;//当前读取的行
                while (!sr.EndOfStream)
                {
                    line = sr.ReadLine().Trim();
                    if (sw!=null&&line != "") sw.WriteLine(line);
                    if (!checktype(line))
                    {
                        continue;//如果不是有效代码
                    }
                    count++;
                }
                sr.Close();
            }
            if (sw != null) {sw.Close();}
            //sw.Close();
            lbSearchFile.Text = "";
        }
        /// <summary>
        /// 判断是否为有效代码行
        /// </summary>
        /// <param name="strline">代码行</param>
        /// <returns></returns>
        private bool checktype(string strline)
        {
            bool isok = true;
            //C#代码判断
            if (!ischeck && strline.StartsWith("/*") && strline.IndexOf("*/")<0)
            {
                isok = false;
                ischeck = true;//为连续注释               
            }
            else
            {
                if (ischeck&&strline.IndexOf("*/") >= 0)
                {
                    ischeck = false;
                }
            }
           
            if (ischeck||strline.StartsWith("//") || (strline.IndexOf("/*") <= 0 && strline.EndsWith("*/")) || strline == "{" || strline == "}")
            {
                isok = false;
            }
            else
                if (strline.StartsWith("'")||strline=="")//vb.net代码
                    isok = false;
            return isok;
        }
        private void 统计FToolStripMenuItem_Click(object sender, EventArgs e)
        {
           if(btnser.Enabled) btnser_Click(sender, e);
        }

        private void 设置目录CToolStripMenuItem_Click(object sender, EventArgs e)
        {
            btnDir_Click(sender,e);
        }

        private void 退出EToolStripMenuItem_Click(object sender, EventArgs e)
        {
            Application.Exit();
        }

        private void btnser_EnabledChanged(object sender, EventArgs e)
        {
            统计FToolStripMenuItem.Enabled = btnser.Enabled;
        }
        //选择保存路径
        private void btnwr_Click(object sender, EventArgs e)
        {
            SaveFileDialog sfd = new SaveFileDialog();
            sfd.AddExtension = true;
            //sfd.CheckFileExists = true;
            sfd.DefaultExt = "*.doc";
            sfd.Filter = "Word File|*.doc;*.wps|TXT FILE|*.txt";
            sfd.ShowDialog();
            if (sfd.FileName.Trim() != "") { wrfile = sfd.FileName; txtWrite.Text = wrfile; }
        }

        private void txtWrite_TextChanged(object sender, EventArgs e)
        {
            wrfile = txtWrite.Text;
        }
    }
}