import json
with open("osakaalud_taga.txt") as f1:
  osakaalud=json.load(f1)

def sobiv(kolmik):
  return "D" in kolmik and "$" not in kolmik and "^" not in kolmik


kogused={}

def lisaKolmik(plokk, keeletase):
   if plokk not in kogused[keeletase]:
       kogused[keeletase][plokk]=1
   else: kogused[keeletase][plokk]+=1

for sliik in ["A2", "B1", "B2", "C1"]:
 kogused[sliik]={}
 sonaliik = open("../"+sliik+"sonaliigid.txt").readlines()
 sonaliik=[rida.strip() for rida in sonaliik]
 for reanr in range(len(sonaliik)):
   teated=[]
   srida=sonaliik[reanr].replace("Z", "")
   nelikud=[srida[koht:koht+4] for koht in range(len(srida)-3)]
   for nelik in nelikud:
     kolmik=nelik[1:4]
     lisa=nelik[0]
     if sobiv(kolmik):
      if kolmik not in osakaalud:
        lisaKolmik(kolmik, sliik)
      else:
        if lisa not in osakaalud[kolmik]:
           lisaKolmik(lisa+">"+kolmik, sliik)
        if osakaalud[kolmik][lisa][1]<5:
           lisaKolmik(lisa+">"+kolmik, sliik)

for sliik in kogused:
  print(sliik)
  paarid=list(kogused[sliik].items())
  paarid.sort(key=lambda paar: -paar[1])
  for paar in paarid: 
    print(paar[0], "-", str(paar[1]))