class Sortovanie
  def usporiadaj
    riadky = []
    name = "e:\\skola\\bakalarska_praca\\dataset\\illinois_dataset_twitter\\zoznam.txt"
    file=File.open(name)
    file.each_line do |riadok|
      riadky << riadok[/^\d+/].to_i
    end
    riadky.sort!

    pole = []
    pole[riadky.last] = 1
    riadky.each do |riadok|
      pole[riadok] = 1
    end

    file2 = File.open("e:\\skola\\bakalarska_praca\\dataset\\illinois_dataset_twitter\\zoznam3.txt", "w")
    file3 = File.open("e:\\skola\\bakalarska_praca\\dataset\\illinois_dataset_twitter\\network.txt")

    counter=0
    counter2=0
    counter3=0

    i=0

    stopky = Time.now

    file3.each_line do |riadok|
      counter2 +=1
      counter3 +=1

      follower = riadok[/^\d+/].to_i
      friend = riadok[/\d+$/].to_i

      #if follower == riadky.last then
      #  puts "Pocet prehladanych " + counter2.to_s + "  Pocet pridanych " + counter.to_s
      #  puts "Koniec"
      #  break
      #end

      if !pole[follower].nil? and !pole[friend].nil? then
        #file2 << riadok
        counter +=1
      end

      if counter3 == 1000000 then
        counter3 = 0
        stopky_zmnena = Time.now - stopky
        puts "Pocet prehladanych " + counter2.to_s + "  Pocet pridanych " + counter.to_s + " za " + stopky_zmnena.to_s
      end
    end

    file2.close
  end
end

s = Sortovanie.new
s.usporiadaj