Added streaming statistics.

circuit/timing.go

@@ -74,9 +74,17 @@ func (t *Timing) Print(stats p2p.IOStats) {
 			row.Column(col)
 		}
 
-		for _, sub := range sample.Samples {
+		for idx, sub := range sample.Samples {
 			row := tab.Row()
-			row.Column("\u2570\u2574" + sub.Label).SetFormat(tabulate.FmtItalic)
+
+			var prefix string
+			if idx+1 >= len(sample.Samples) {
+				prefix = "\u2570\u2574"
+			} else {
+				prefix = "\u251C\u2574"
+			}
+
+			row.Column(prefix + sub.Label).SetFormat(tabulate.FmtItalic)
 
 			var d time.Duration
 			if sub.Abs > 0 {
@@ -89,7 +97,6 @@ func (t *Timing) Print(stats p2p.IOStats) {
 			row.Column(
 				fmt.Sprintf("%.2f%%", float64(d)/float64(duration)*100)).
 				SetFormat(tabulate.FmtItalic)
-
 		}
 	}
 	row := tab.Row()

compiler/ssa/streamer.go

@@ -145,6 +145,7 @@ func (prog *Program) StreamCircuit(conn *p2p.Conn, oti ot.OT,
 
 	start := time.Now()
 	lastReport := start
+	var circCompileDuration time.Duration
 
 	istats := make(map[string]circuit.Stats)
 
@@ -433,6 +434,7 @@ func (prog *Program) StreamCircuit(conn *p2p.Conn, oti ot.OT,
 			if !ok {
 				var cIn [][]*circuits.Wire
 				var flat []*circuits.Wire
+				startTime := time.Now()
 
 				for _, in := range wires {
 					w := circuits.MakeWires(types.Size(len(in)))
@@ -467,6 +469,7 @@ func (prog *Program) StreamCircuit(conn *p2p.Conn, oti ot.OT,
 					fmt.Printf("%05d: - %s\n", idx, circ)
 				}
 				circ.AssignLevels()
+				circCompileDuration += time.Now().Sub(startTime)
 			}
 			if false {
 				circ.Dump()
@@ -497,7 +500,15 @@ func (prog *Program) StreamCircuit(conn *p2p.Conn, oti ot.OT,
 
 	xfer = conn.Stats.Sum() - ioStats
 	ioStats = conn.Stats.Sum()
-	timing.Sample("Garble", []string{circuit.FileSize(xfer).String()})
+	sample := timing.Sample("Stream", []string{circuit.FileSize(xfer).String()})
+	sample.Samples = append(sample.Samples, &circuit.Sample{
+		Label: "Compile",
+		Abs:   circCompileDuration,
+	})
+	sample.Samples = append(sample.Samples, &circuit.Sample{
+		Label: "Garble",
+		Abs:   prog.garbleDuration,
+	})
 
 	result := new(big.Int)