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

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

for sliik in ["A2", "B1", "B2", "C1"]:
 sonaliik = open("../"+sliik+"sonaliigid.txt").readlines()
 sonaliik=[rida.strip() for rida in sonaliik]
 tekst = open("/mnt/c/jaagup/23/04/keel/tasemetekstid/"+sliik+".txt").readlines()
 vastus=open(sliik+"_harv_osakaal_ees.txt", "w", encoding="utf-8")
 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[0:3]
     lisa=nelik[3]
     if sobiv(kolmik):
      if kolmik not in osakaalud:
        teated.append("puuduv kolmik "+kolmik)
      else:
        if lisa not in osakaalud[kolmik]:
           teated.append("puuduv  tagakontekst "+kolmik+"<"+lisa)
        if osakaalud[kolmik][lisa][1]<5:
           teated.append(str(round(osakaalud[kolmik][lisa][1], 3))+"% "+kolmik+"<"+lisa)

   if teated:
     print(tekst[reanr].strip(), file=vastus)
     print(sonaliik[reanr], file=vastus)
     print("\n".join(teated), file=vastus)
     print(file=vastus)
 vastus.close()