library(igraph)
setwd("~/Documents/Violence Cascades/")
start_date = as.Date("2005-12-31")

# plot cascade sizes
data = read.csv('Results/components_dist-91515.csv',header=F)
data = data[order(data$V1),]
sizes = data$V1
counts = data$V2
plot(sizes,counts,log='xy',type='o',lwd=3,
     xlab='Size of Cascade', ylab='Number of Cascades', main='Distribution of Cascade Sizes')


# plot cascades
edges = read.csv('Results/edges-91515.csv',header=F,
                 col.names=c('v1','t1','v2','t2','dist'))
edges$v1 = edges$v1*10000 + edges$t1
edges$v2 = edges$v2*10000 + edges$t2
dag = graph.data.frame(edges[,c(1,3)], directed=TRUE)
table(components(dag,mode='weak')$csize)

clusters = clusters(dag)
membership = clusters$membership
csize = clusters$csize
order = rev(order(csize))

i = 95
V = which(clusters(dag)$membership==order[i]) # get all nodes in cluster
cc = induced.subgraph(dag,V)
times = as.numeric(V(cc)$name) %% 10000
start_date + range(times)

### plot cascade ###
cols = rep('lightblue',vcount(cc))
seed = which(degree(cc,mode='in')==0)
cols[seed] = 'red'
plot(cc,vertex.size=5,edge.arrow.size=0.2,vertex.color=cols,vertex.label=NA,
     layout=layout.reingold.tilford(cc,root=seed))

par(mfrow = c(2,3))