<转>统计源代码行数的一些实现方法

这个问题的思考其实对于某一种语言而言，基本都能实现，只是简单和复杂而已。而此次我讨论就是只是在linux下面使用了shell和c对源代码进行行数的讨论。本打算是实现一个python版本的，由于python这块还不是太熟，所以就等以后熟了把这块补上。

shell版

shell的强大快捷之处就在此体现出来了。我们使用find命令就直接能将目标的文件进行检索，然后我们就直接对检索出来的对象进行统计。统计我们知道使用wc这个命令，但是我们观察一下wc的输出:

206 ./2014-03-09-jekyll-blog.md

前面是我们想要的内容，而后面的文件名却不是我们要得，怎么解决？一个思路就是将这个结果保存到一个文件中，然后使用cut等命令对文件进行改造，接着再去统计，这个方法个人感觉比较麻烦。那么第二个思路就是用awk，直接对我们的结果进行分割。按照思路二，那我们就对此进行编码了

使用find，对于find，参数较多，考虑到有时候源代码中目录层次不一，我们需要加上参数maxdepth，再考虑到所统计的语言项目不一样，我们同样要指定name这个参数，而且采用一个简单的正则表达式。所以基本的find语句如下：

find $1 -type f -maxdepth $2 -name "*.$3"

这边的$1是路径名 $2 指定的最大深度 $3 文件的后缀名或是一些正则的表达式，$3的这个设计我原先的打算是做一个test，根据用户输入，然后最后自己生成正则，感觉这样有点多此一举了。一个项目中使用的文件后缀一般不会很多，统计的是核心代码，所以这一块就可以简便一点了。

统计行数的核心就是如下的这个：

w =(wc -l $file | awk "{print $1}")

这边的file就是之前find的返回值，我们将上面的这个代码放在一个循环中。这边是awk的最简单使用了，将一个记录的第一个字段打印出来。但是不要混淆了这边的$1和shell脚本的中的$1是不一样的，这在awk的命令中，所以不要担心会出错。

加一些修饰

我们为了让输出变得好看一点,对echo的输出加一些修饰，主要是颜色的修饰：

echo -e "$s: \e[1;32m $t \e[0m"

echo -e "\e[1,31m total: \e[1;32m $w \e[0m"

好了，至此一个基本上能够满足要求的代码统计的脚本就完成了，一个完整的版本可以到我的gist看。

效果截图

C版本

之所以想到使用c来统计，是因为这是在linux平台上，可以使用系统编程方面的一些东西来解决这个问题。因为这个不在标准c中提供对目录的操作对于windows下面，我没有去研究，所以不好多加叙述。

我们来看看如何实现C版本的统计程序？首先等解决的就是一个文件遍历的问题，这个解决不了接下去的工作就不用考虑。好在在linux中提供了一个 dirent.h的头文件，相关的函数都在这里面。我们可以用的是opendir readdirtelldir seekdir closedir着几个函数，这写函数基本就是差不多了，还缺的一个就是更改目录的函数chdir这个函数的实现在unistd.h中，所以这些基本上就齐了。

readdir返回的是一个结构体，结构体中有一个参数d_name,这是保存的文件的名字，所以有了这个，我们就可以对其进行属性的判断了。因为我们要判断这是文件还是目录。

如何判断？没有shell中if [ -d filename ]这样快捷的判断，但是我们要想到的一点就是，这是在liunx中，一切皆文件，对于文件的属性的判断，我们可以利用lstat()

这个函数是在sys/stat.h中，我们使用lstat(filename,&statbuf) 然后就可以用宏还进行判断了S_ISDIR(statbuf.st_mode). 这样就解决了文件类型的判断。

接着我们来考虑如何来遍历一个文件夹呢？首先的着肯定是在一个循环下。因为一个目录下有时不止一个文件，但是对于不同的目录的时候呢？例如一个目录下的一个子目录，如何去实现遍历？我们考虑到的是使用递归，递归调用我们写的那个函数。因为每次读取的目录只有一个。而且当readdir 读到文件尾的时候回返回一个NULL。也许你会问这个递归是如何实现的。我们在循环结束后，调用了chidr("..") 回到上层，这样就继续遍历。这就是递归终止的条件。

是时候统计代码了，我们剔除了目录的文件，剩下的就是真正的文件了。对于统计使用fgets来实现遍历一个文件，然后加个循环就解决问题。这边一个不好的就是没有办法去判断文件的类型，统计是对目录下所以的文件进行的统计，这一点不如shell的灵活。

下面是源代码，基本的思路就是这样，还要注意的一个细节是有两个特殊的目录就是.和..我们要忽略掉，不然会产生一个死循环。

#include <stdio.h>

#include <stdlib.h>

#include <string.h>

#include <fcntl.h>

#include <unistd.h>

#include <errno.h>

#include <dirent.h>  /*some functions option on dir*/

#include <sys/stat.h>

#define MAXLINE 200

    int linecount=0; /*for save the countline*/

    int getLine(char *fname)

    {

        FILE *fp;

        char line[MAXLINE];

        int  total = 0;

        fp = fopen(fname,"r");

        while(fgets(line,200,fp) != NULL)

            total++;

        return total;

    }

    /*Recurse a the target dirent*/

    void Recur_dir(const char *dir, int depth)

    {

        DIR *dp;  /*dir pointer*/

        struct dirent *entry;  /*structure to save dir info*/

        struct stat statbuf; /*use to charge which is dir and which is file*/

        /*get the decriptor*/

        if((dp = opendir(dir)) == NULL)

        {

            printf("cannot open directory:%s\n",dir);

            fprintf(stderr,"opendir error:%s\n",strerror(errno));

            exit(1);

        }

        chdir(dir); /*enter into target dir*/

        while((entry = readdir(dp)) != NULL)

        {

            lstat(entry->d_name,&statbuf);/*get the entry status*/

            if(S_ISDIR(statbuf.st_mode))

            {

                /*is a dirctory and ignore the .. and .*/

                if(strcmp(".",entry->d_name)== 0 || strcmp("..",entry->d_name) == 0)

                    continue;

                //printf("%*s%s\n",depth,"",entry->d_name);

                Recur_dir(entry->d_name,depth+4); /*continue to open */

            }

            else /*if not a dirctroy and we count the line*/

            {

                linecount += getLine(entry->d_name); /*count everyfile line*/

            }

        }

        chdir("..");

        closedir(dp);

    }

    int main(int argc, const char *argv[])

    {

        char *topdir = "."

        if(argc > 2)

        {

            printf("Usage:%s dir\n",argv[0]);

            exit(1);

        }

        if(argc == 2)

        {

            topdir = argv[1];

        }

        Recur_dir(topdir,0);

        printf("Total:%d\n",linecount);

        return 0;

    }

如果不提供参数，就遍历当前目录，提供了就统计给出的目录。

效果截图