统计编码量工具

 

利用windows提供的目录操作库和文件操作库，统计指定目录及其子目录中所有.cpp、.c、.h文件的代码量，用来检测自己的编程工作量（）。

#include <stdio.h>
#include <string.h>
#include <direct.h>
#include <malloc.h>
#include <io.h>

long nLen = 0;
int GetFileLength(char *pszPath)
{
    FILE *pRead = fopen(pszPath, "r");
    if (NULL == pRead)
    {
        return -1;
    }
    char szBuf[1024];
    while (!feof(pRead))
    {
        fgets(szBuf, sizeof(szBuf), pRead);
        nLen++;
    }

    fclose(pRead);
    
}
int SearchPath(char *pszPath)
{
    int rv = 0;
    rv = chdir(pszPath);
    if (0 != rv)
    {
        printf("func chdir() error
");
        rv = -1;
        return rv;
    }

    struct _finddata_t data;
    long handle;
    if (-1L == (handle = _findfirst("*.*", &data)))   //成功返回唯一的搜索句柄, 出错返回-1
    {
        return rv;
    }
    do 
    {
        if (data.attrib == _A_SUBDIR )
        {//目录类型
            char szBuf[1024] = {0};
            if (strcmp(data.name, ".") != 0 && strcmp(data.name, "..") != 0)
            {
                sprintf(szBuf, "%s\%s", pszPath, data.name);
                SearchPath(szBuf);
            }
        }
        else
        {//单个文件
            int nLen = strlen(data.name);
            if (data.name[nLen - 1] == 'p' && data.name[nLen - 2] == 'p' &&
                data.name[nLen - 3] == 'c' &&data.name[nLen - 4] == '.' )
            {//过滤出所有cpp的文件
                printf("   [%s]
", data.name );
                char szBuf[1024] = {0};
                sprintf(szBuf, "%s\%s", pszPath, data.name);
                GetFileLength(szBuf);
            }    
        }
    } while(_findnext( handle, &data ) == 0);     //成功返回0 , 出错返回-1
    
    _findclose( handle );     // 关闭当前句柄
    
    return rv;
}

int main()
{
    char *pszPath = "L:\谷歌大赛";
    SearchPath(pszPath);
    printf("总代码量为:%ld
", nLen);
    return 0;
}