Generate graph communities and annotate vertices so that each vertex has its community as a property

generate.scala

package com.xhochy
 
import com.tinkerpop.blueprints.impls.tg.TinkerGraph
import com.tinkerpop.furnace.generators.{ CommunityGenerator, EdgeAnnotator, NormalDistribution, PowerLawDistribution, VertexAnnotator }
 
object Generator {
  def main(args: Array[String]) = {
    val graph = new TinkerGraph
    Range(0, 100).foreach(graph.addVertex(_))
    val generator = new CommunityGenerator("edge", EdgeAnnotator.NONE, VertexAnnotator.COPY_CONTEXT)
    generator.setCommunityDistribution(new NormalDistribution(3));
    generator.setDegreeDistribution(new PowerLawDistribution(2.3));
    generator.setCrossCommunityPercentage(0.1);
    val numEdges = generator.generate(graph, 10, 1000);
  }
}