# knesset22 by knesset size.R
rm(list=ls())
knesset22=data.frame(party=c("likud", "bluewhite", 
                             "meshutefet", "shas", "aguda",
                             "liberman", "yamina", 
                             "avoda", "democrats"),
                     votes=c(1113617, 1151214, 
                             470211, 330199, 268775, 
                             310154, 260655,
                             212782, 192495))

# surplus agreements
# likud + yamina
# bluewhite + liberman
# shas + aguda 
# avoda + democrats

suragg=function(k){
  # k=knesset0
  sagg=k
  sagg$party=as.character(sagg$party)
  
  # likud + yamina
  w1=which(sagg$party=="likud")
  w2=which(sagg$party=="yamina")
  sagg$party[w1]="likud + yamina"
  sagg$mandats[w1]=sagg$mandats[w1]+sagg$mandats[w2]
  sagg$votes[w1]=sagg$votes[w1]+sagg$votes[w2]
  sagg=sagg[-w2,]  

  # bluewhite + liberman 
  w1=which(sagg$party=="bluewhite")
  w2=which(sagg$party=="liberman")
  sagg$party[w1]="bluewhite + liberman"
  sagg$mandats[w1]=sagg$mandats[w1]+sagg$mandats[w2]
  sagg$votes[w1]=sagg$votes[w1]+sagg$votes[w2]
  sagg=sagg[-w2,]  
  
  # shas + aguda 
  w1=which(sagg$party=="shas")
  w2=which(sagg$party=="aguda")
  sagg$party[w1]="shas + aguda"
  sagg$mandats[w1]=sagg$mandats[w1]+sagg$mandats[w2]
  sagg$votes[w1]=sagg$votes[w1]+sagg$votes[w2]
  sagg=sagg[-w2,]
  
  # avoda + democrats 
  w1=which(sagg$party=="avoda")
  w2=which(sagg$party=="democrats")
  sagg$party[w1]="avoda + democrats"
  sagg$mandats[w1]=sagg$mandats[w1]+sagg$mandats[w2]
  sagg$votes[w1]=sagg$votes[w1]+sagg$votes[w2]
  sagg=sagg[-w2,]
  
  return(sagg)
}

baderofer=function(kn){
  # kn=knesset0
  sa=suragg(kn)
  kn$pay=kn$votes/(1+kn$mandats) 
  sa$pay=sa$votes/(1+sa$mandats)
  w=which(sa$pay==max(sa$pay))
  
  # likud + yamina
  if(sa$party[w]=="likud + yamina"){
    w1=which(kn$party=="likud")
    w2=which(kn$party=="yamina")
    if(kn$pay[w1]>kn$pay[w2]){
      kn$mandats[w1]=kn$mandats[w1]+1
    }
    if(kn$pay[w1]<kn$pay[w2]){
      kn$mandats[w2]=kn$mandats[w2]+1
    }
  }
  
  # bluewhite + liberman
  if(sa$party[w]=="bluewhite + liberman"){
    w1=which(kn$party=="bluewhite")
    w2=which(kn$party=="liberman")
    if(kn$pay[w1]>kn$pay[w2]){
      kn$mandats[w1]=kn$mandats[w1]+1
    }
    if(kn$pay[w1]<kn$pay[w2]){
      kn$mandats[w2]=kn$mandats[w2]+1
    }
  }

  # shas + aguda
  if(sa$party[w]=="shas + aguda"){
    w1=which(kn$party=="shas")
    w2=which(kn$party=="aguda")
    if(kn$pay[w1]>kn$pay[w2]){
      kn$mandats[w1]=kn$mandats[w1]+1
    }
    if(kn$pay[w1]<kn$pay[w2]){
      kn$mandats[w2]=kn$mandats[w2]+1
    }
  }
  
  # avoda + democrats
  if(sa$party[w]=="avoda + democrats"){
    w1=which(kn$party=="avoda")
    w2=which(kn$party=="democrats")
    if(kn$pay[w1]>kn$pay[w2]){
      kn$mandats[w1]=kn$mandats[w1]+1
    }
    if(kn$pay[w1]<kn$pay[w2]){
      kn$mandats[w2]=kn$mandats[w2]+1
    }
  }

  allsa=c("likud + yamina", "bluewhite + liberman", 
          "shas + aguda", "avoda + democrats")
  if (!(sa$party[w] %in% allsa)){
      w1=which(kn$party %in% sa$party[w])
      kn$mandats[w1]=kn$mandats[w1]+1
  }
  
  kn=kn[,-4]
  return(kn)
}


theknesset=function(knesset, mandats){
  kname=deparse(substitute(knesset))
  kn=knesset

  # calculate moded
  moded=floor(sum(kn$votes)/mandats)
  
  # assign mandats by moded
  kn$mandats=floor(kn$votes/moded)
  remained=mandats-sum(kn$mandats)
  
  for (j in 1:remained){
    kn=baderofer(kn)
  }
  
  kn=kn[,-2]
  names(kn)[2]=paste(kname, mandats, sep="_")
  return(kn)
}

knesset120=theknesset(knesset22, 120)
knesset180=theknesset(knesset22, 180)
knesset70=theknesset(knesset22, 70)
knesset22results=merge(knesset120, knesset180)
knesset22results=merge(knesset22results, knesset70)

knesset22results=knesset22results[order(-knesset22results$knesset22_120),]
rownames(knesset22results)=1:nrow(knesset22results)

gushim22=data.frame(gush=c("right", "center", "meshutefet", "liberman"))
for(j in 2:4){
  mandats=knesset22results[,j]
  gushim22$newcol=c(sum(mandats[c(2,4,5,9)]),
                    sum(mandats[c(1,6,7)]),
                    mandats[3],
                    mandats[8])
  names(gushim22)[ncol(gushim22)]=names(knesset22results)[j]
}