func main() {
// Let GC run in its own thread
runtime.GOMAXPROCS(2)
flag.Parse()
if *graphFilename == "" {
fmt.Fprintf(os.Stderr, "No graph file specified\n")
os.Exit(ERR_USAGE)
}
f, err := os.Open(*graphFilename)
if err != nil {
fmt.Fprintf(os.Stderr, "Error opening graph file: %v\n", err)
os.Exit(ERR_IO)
}
fi, err := os.Stat(*outputDir)
if err != nil {
fmt.Fprintf(os.Stderr, "Bad output directory: %v\n", err)
os.Exit(ERR_IO)
}
if !fi.IsDir() {
fmt.Fprintf(os.Stderr, "Bad output directory: not a directory\n", err)
os.Exit(ERR_IO)
}
gd, err := ioutil.ReadAll(f)
def, err := encoding.Unmarshal(gd)
if err != nil {
fmt.Fprintf(os.Stderr, "Error parsing graph file: %v\n", err)
os.Exit(ERR_PARSE)
}
g := graph.NewGraph(def, graph.MaxCost)
var t0, tprev time.Time
round := 0
var mergeLog *bytes.Buffer
roundFunc := func() {
if round == 0 {
// This is the first time we're called
t0 = time.Now()
tprev = t0
mergeLog = &bytes.Buffer{}
if err := writeLogfile(g, round); err != nil {
fmt.Fprintf(os.Stderr, "%v\n", err)
}
fmt.Printf("ROUND %v...\n", round)
round++
return
}
t := time.Now()
diff := t.Sub(tprev)
tprev = t
fmt.Printf("Round %v took %v\n", round-1, diff)
data := mergeLog.Bytes()
if err := writeMergeLogfile(data, round); err != nil {
fmt.Fprintf(os.Stderr, "%v\n", err)
}
if err := writeLogfile(g, round); err != nil {
fmt.Fprintf(os.Stderr, "%v\n", err)
}
fmt.Println()
fmt.Printf("ROUND %v...\n", round)
round++
}
merge := func(c, d graph.ClusterID) {
fmt.Fprintf(mergeLog, "%v\t%v\n", c, d)
}
g.MergeCallback(roundFunc, merge)
t := time.Now()
diff := t.Sub(tprev)
fmt.Printf("Round %v took %v\n", round-1, diff)
data := mergeLog.Bytes()
if err := writeMergeLogfile(data, round); err != nil {
fmt.Fprintf(os.Stderr, "%v\n", err)
}
if err := writeLogfile(g, round); err != nil {
fmt.Fprintf(os.Stderr, "%v", err)
}
fmt.Println()
diff = t.Sub(t0)
fmt.Println("Graph stabilized.")
fmt.Printf("%v rounds completed in %v\n", round-1, diff)
if round != 1 {
fmt.Printf("Average time per round: %v\n", diff/time.Duration(round-1))
}
}
func main() {
flag.Parse()
f, err := os.Open(*graphFilename)
if err != nil {
fmt.Fprintf(os.Stderr, "Error opening graph file: %v\n", err)
os.Exit(ERR_IO)
}
gd, err := ioutil.ReadAll(f)
def, err := encoding.Unmarshal(gd)
if err != nil {
fmt.Fprintf(os.Stderr, "Error parsing graph file: %v\n", err)
os.Exit(ERR_PARSE)
}
g := graph.NewGraph(def, graph.MaxCost)
var clusterList map[graph.ClusterID]*graph.Cluster
if *clusters == "all" {
clusterList = g.Clusters()
} else {
clusterList = make(map[graph.ClusterID]*graph.Cluster)
arr := strings.Split(*clusters, ",")
for _, str := range arr {
// Allow for "--clusters=all,"
// (in case there's a cluster
// called "all")
if str == "" {
continue
}
cid := graph.ClusterID(str)
c := g.Cluster(cid)
if c == nil {
fmt.Fprintf(os.Stderr, "Nonexistant cluster ID: %v\n", str)
continue
}
clusterList[cid] = c
}
}
printedYet := false
if *basic {
fmt.Println("BASIC STATISTICS")
runBasic(g)
printedYet = true
}
if *advanced {
if printedYet {
fmt.Println()
}
printedYet = true
fmt.Println("ADVANCED STATISTICS")
for _, f := range overallAnalyzers {
f(g)
}
}
if len(clusterList) > 0 {
if printedYet {
fmt.Println()
}
fmt.Println("PER-CLUSTER ANALYSIS")
for cid, c := range clusterList {
fmt.Printf(" %v:\n", cid)
for _, f := range clusterAnalyzers {
f(g, c)
}
}
}
}