使用Antlr和JfreeChart实现项目源代码行数计算图表

Antlr的功能在我的其他文章里提到了就不多说了，JFreeChart是一个功能强大的Java开源图表生成组件。

是不是经常有人问你，你做的项目一共有多少行代码，你编写了多少行代码？

本文的程序轻松帮你回答这个问题。

<shapetype id="_x0000_t75" coordsize="21600,21600" o:spt="75" o:preferrelative="t" path="m@4@5l@4@11@9@11@9@5xe" filled="f" stroked="f"><stroke joinstyle="miter"></stroke><formulas><f eqn="if lineDrawn pixelLineWidth 0"></f><f eqn="sum @0 1 0"></f><f eqn="sum 0 0 @1"></f><f eqn="prod @2 1 2"></f><f eqn="prod @3 21600 pixelWidth"></f><f eqn="prod @3 21600 pixelHeight"></f><f eqn="sum @0 0 1"></f><f eqn="prod @6 1 2"></f><f eqn="prod @7 21600 pixelWidth"></f><f eqn="sum @8 21600 0"></f><f eqn="prod @7 21600 pixelHeight"></f><f eqn="sum @10 21600 0"></f></formulas><path o:extrusionok="f" gradientshapeok="t" o:connecttype="rect"></path><lock v:ext="edit" aspectratio="t"></lock></shapetype><shape id="_x0000_i1025" style="WIDTH: 384pt; HEIGHT: 254.25pt" type="#_x0000_t75"><imagedata o:title="stat1" src="file:///C:%5CDOCUME~1%5CADMINI~1%5CLOCALS~1%5CTemp%5Cmsohtml1%5C01%5Cclip_image001.png"></imagedata></shape>

<shape id="_x0000_i1026" style="WIDTH: 384pt; HEIGHT: 254.25pt" type="#_x0000_t75"><imagedata o:title="stat2" src="file:///C:%5CDOCUME~1%5CADMINI~1%5CLOCALS~1%5CTemp%5Cmsohtml1%5C01%5Cclip_image003.png"></imagedata></shape>

我越来越喜欢用Antlr来完成分析问题，虽然它在性能和简易性方面不如Lex/Yacc相比，那它能很容易的结合到Java项目里。

1 计算文件行数，和空行行数

//----------------------------------------------------------------------------

// The Colimas source statistics scanner

//----------------------------------------------------------------------------

header{

package org.colimas.src.parser;

}

class SourceStatisticsParser extends Parser;

options {

k = 2; // two token lookahead

codeGenMakeSwitchThreshold = 2; // Some optimizations

codeGenBitsetTestThreshold = 3;

defaultErrorHandler = false;

}

file : //遍历文件

(CODE)+ file //有文字多行

| (EMPTY)+ file //无文字多行

| CODE //文件最后一样

| EOF //文件结束符

;

class SourceStatisticsLexer extends Lexer;

options {

testLiterals=true; // test for literals

k=2; // 2 characters of lookahead

codeGenBitsetTestThreshold=20;

charVocabulary='\u0003'..'\uFFFF';

}

{

private long codeline=0; //有字符行数

private long emptyline=0; //空行行数

private long totalline=0; //文件行数=有字符行数+空行行数

public long getCodeline(){

return this.codeline;

}

public long getEmptyline(){

return this.emptyline;

}

public long getTotalline(){

return this.codeline+this.emptyline;

}

}

CHAR_LITERAL

: ~('\uFFFF' | '\r' |'\n') //结束符，回车符，换行符以外的所有字符

;

CHAR : (CHAR_LITERAL)+; //多个字符

EMPTY

: ('\r')? ('\n') // DOS/Windows

// increment the line count in the scanner

{

newline(); //用于调试

emptyline++; //空行加1

$setType(Token.SKIP);

}

;

CODE : CHAR ( (('\r')? ('\n')) |'\uFFFF')

{

newline();

codeline++; }

;

该程序将被Antlr编译生成Java SourceStatisticsLexer类和SourceStatisticsParser类

2 保存行数

遍历指定一个目录内所有文件。

//递归方法

public void passDirectory(File dir){

File[] files=dir.listFiles();

for(int i=0;i<files.length;i++){

if(files[i].isDirectory()){

System.out.println(files[i].getAbsolutePath());

passDirectory(files[i]);

}else{

//调用Lexer和Parser，得到行数并保存到hash表里。

setLineStat(files[i]);

}

}

}

//实例化Lexer分析器

public void setLineStat(File file){

…

this.lexer=

new SourceStatisticsLexer(in);

if(!FileExtension.checkAscii(type)){ //如果是文本文件则计算行数，否则跳过

files.put(type,stat);

return;

}

this.lexer.setFilename(file.getName());

//实例化Parser分析器

this.parser = new SourceStatisticsParser(lexer);

parser.file(); //遍历分析

//将行数保存到Map里

stat.addCodeLines(lexer.getCodeline()); stat.addEmptyLines(lexer.getEmptyline());

files.put(type,stat);

///////////////////////////////////////////////////////////////////////////

计算所有文件的总行数

directoryCodeLines+=lexer.getCodeline();

directoryEmptyLines+=lexer.getEmptyline();

in.close();

…

////////////////////////////////////////////////////////////////////////////

这样所有的文件行数将根据文件类型保存在以文件类型为Key的HashMap里。

3 生成饼状图与柱状图

饼状图与柱状图都需要先组织数据集，例如饼状图的数据集

/**

*<p>create data set </p>

* @see org.colimas.src.graph.Chart#createDataset(java.util.Map)

*/

public Dataset createDataset(Map files) {

// row keys...

Set keys=files.keySet();

// create the dataset...

DefaultPieDataset dataset = new DefaultPieDataset();

Object[] types=(Object[])keys.toArray();

for(int i=0;i<types.length;i++){

FileStat file=(FileStat)files.get(types[i]);

try {

String desc=FileExtension.getFileDesc((String)types[i]);

if(desc==null)

desc="other files";

String type=desc; //获得文件类型的文件描述

dataset.setValue(type,file.getFiles()); //文件类型的文件数。

} catch (Exception e) {

e.printStackTrace();

}

}

return dataset;

}

然后就能生成图表了

/**

*<p>create chart </p>

* @see org.colimas.src.graph.Chart#createChart(org.jfree.data.general.Dataset)

*/

public JFreeChart createChart(Dataset dataset) {

JFreeChart chart =null;

if (dataset instanceof PieDataset){

chart = ChartFactory.createPieChart(

"File Statistics 1", // chart title

(PieDataset)dataset, // data

true, // include legend

true,

false

);

PiePlot plot = (PiePlot) chart.getPlot();

plot.setSectionOutlinesVisible(false);

plot.setLabelFont(new Font("SansSerif", Font.PLAIN, 12));

plot.setNoDataMessage("No data available");

plot.setCircular(false);

plot.setLabelGap(0.02);

}

return chart;

}

4 显示

在JFrame的类里显示饼状图

/**

* Creates a panel

* @return A panel.

*/

public JPanel createPIEPanel(Map files) {

PIEChart pie=new PIEChart();

JFreeChart chart = pie.createChart(pie.createDataset(files));

return new ChartPanel(chart);

}

之后就像其他JPanel一样可以在JFrame里显示。

5 结果：

源代码可以在Colimas开源项目网站上下载

:pserver:[e-mail]:/cvsroot/colimas

Module为DocBuild