Calculate aggregate stats

cobertura.go

@@ -38,7 +38,7 @@ type Class struct {
 	BranchRate float32   `xml:"branch-rate,attr"`
 	Complexity float32   `xml:"complexity,attr"`
 	Methods    []*Method `xml:"methods>method"`
-	Lines      []*Line   `xml:"lines>line"`
+	Lines      Lines     `xml:"lines>line"`
 }
 
 type Method struct {
@@ -47,10 +47,116 @@ type Method struct {
 	LineRate   float32 `xml:"line-rate,attr"`
 	BranchRate float32 `xml:"branch-rate,attr"`
 	Complexity float32 `xml:"complexity,attr"`
-	Lines      []*Line `xml:"lines>line"`
+	Lines      Lines   `xml:"lines>line"`
 }
 
 type Line struct {
 	Number int   `xml:"number,attr"`
 	Hits   int64 `xml:"hits,attr"`
 }
+
+// Lines is a slice of Line pointers, with some convenience methods
+type Lines []*Line
+
+// HitRate returns a float32 from 0.0 to 1.0 representing what fraction of lines
+// have hits
+func (lines Lines) HitRate() (hitRate float32) {
+	return float32(lines.NumLinesWithHits()) / float32(len(lines))
+}
+
+// NumLines returns the number of lines
+func (lines Lines) NumLines() int64 {
+	return int64(len(lines))
+}
+
+// NumLinesWithHits returns the number of lines with a hit count > 0
+func (lines Lines) NumLinesWithHits() (numLinesWithHits int64) {
+	for _, line := range lines {
+		if line.Hits > 0 {
+			numLinesWithHits++
+		}
+	}
+	return numLinesWithHits
+}
+
+// HitRate returns a float32 from 0.0 to 1.0 representing what fraction of lines
+// have hits
+func (method Method) HitRate() float32 {
+	return method.Lines.HitRate()
+}
+
+// NumLines returns the number of lines
+func (method Method) NumLines() int64 {
+	return method.Lines.NumLines()
+}
+
+// NumLinesWithHits returns the number of lines with a hit count > 0
+func (method Method) NumLinesWithHits() int64 {
+	return method.Lines.NumLinesWithHits()
+}
+
+// HitRate returns a float32 from 0.0 to 1.0 representing what fraction of lines
+// have hits
+func (class Class) HitRate() float32 {
+	return float32(class.NumLinesWithHits()) / float32(class.NumLines())
+}
+
+// NumLines returns the number of lines
+func (class Class) NumLines() (numLines int64) {
+	for _, method := range class.Methods {
+		numLines += method.NumLines()
+	}
+	return numLines
+}
+
+// NumLinesWithHits returns the number of lines with a hit count > 0
+func (class Class) NumLinesWithHits() (numLinesWithHits int64) {
+	for _, method := range class.Methods {
+		numLinesWithHits += method.NumLinesWithHits()
+	}
+	return numLinesWithHits
+}
+
+// HitRate returns a float32 from 0.0 to 1.0 representing what fraction of lines
+// have hits
+func (pkg Package) HitRate() float32 {
+	return float32(pkg.NumLinesWithHits()) / float32(pkg.NumLines())
+}
+
+// NumLines returns the number of lines
+func (pkg Package) NumLines() (numLines int64) {
+	for _, class := range pkg.Classes {
+		numLines += class.NumLines()
+	}
+	return numLines
+}
+
+// NumLinesWithHits returns the number of lines with a hit count > 0
+func (pkg Package) NumLinesWithHits() (numLinesWithHits int64) {
+	for _, class := range pkg.Classes {
+		numLinesWithHits += class.NumLinesWithHits()
+	}
+	return numLinesWithHits
+}
+
+// HitRate returns a float32 from 0.0 to 1.0 representing what fraction of lines
+// have hits
+func (cov Coverage) HitRate() float32 {
+	return float32(cov.NumLinesWithHits()) / float32(cov.NumLines())
+}
+
+// NumLines returns the number of lines
+func (cov Coverage) NumLines() (numLines int64) {
+	for _, pkg := range cov.Packages {
+		numLines += pkg.NumLines()
+	}
+	return numLines
+}
+
+// NumLinesWithHits returns the number of lines with a hit count > 0
+func (cov Coverage) NumLinesWithHits() (numLinesWithHits int64) {
+	for _, pkg := range cov.Packages {
+		numLinesWithHits += pkg.NumLinesWithHits()
+	}
+	return numLinesWithHits
+}

gocover-cobertura.go

@@ -54,6 +54,9 @@ func (cov *Coverage) parseProfiles(profiles []*Profile) error {
 	for _, profile := range profiles {
 		cov.parseProfile(profile)
 	}
+	cov.LinesValid = cov.NumLines()
+	cov.LinesCovered = cov.NumLinesWithHits()
+	cov.LineRate = cov.HitRate()
 	return nil
 }
 
@@ -95,6 +98,7 @@ func (cov *Coverage) parseProfile(profile *Profile) error {
 		profile:  profile,
 	}
 	ast.Walk(visitor, parsed)
+	pkg.LineRate = pkg.HitRate()
 	return nil
 }
 
@@ -112,10 +116,12 @@ func (v *fileVisitor) Visit(node ast.Node) ast.Visitor {
 	case *ast.FuncDecl:
 		class := v.class(n)
 		method := v.method(n)
+		method.LineRate = method.Lines.HitRate()
 		class.Methods = append(class.Methods, method)
 		for _, line := range method.Lines {
 			class.Lines = append(class.Lines, line)
 		}
+		class.LineRate = class.Lines.HitRate()
 	}
 	return v
 }