Как найти количество строк и различных элементов в файле и их записи на заголовок, Scala

Question

У меня есть что-то вроде:

object Example_01_IO { 

    val s = Source.fromFile("example_01.txt") 
    val source = s.getLines() 
    val destination = new PrintWriter(new File("des_example_01.txt")) 

    var nrVariables: Int = 0 
    var nrLines: Int = 0 

    // here are the extracted lines from example_01 that fulfills some conditions. 
    val linesToWrite: Iterator[String] = ... 

    def main(args: Array[String]): Unit = { 

    //Here is the header that I want to write in a destination file 
    destination.write("des_example_01.txt \n") 
    destination.write("Nr. of Variables and Lines: " + nrVariables + " " + nrLines + "\n") 

    for(line <- linesToWrite) { 
     println(line) 
     destination.write(line) 
     destination.write("\n") 
     nrLines += 1 
    } 

    s.close() 
    destination.close() 

    } 

мне нужно иметь значения для nrVariables и nrLines писать в заголовке файла назначения (например, во второй строке). Есть ли возможность вычислить эти два значения, прежде чем начинать писать другие строки?

Любая помощь или ссылка действительно приветствуется. Спасибо.

Answer 1

Ну, Source.fromFile не могут быть повторно использованы, один ниже работает отлично:

package example 

import java.io.PrintWriter 
import scala.io.Source 
import java.io.File 

object Example_01_IO { 

    def s = Source.fromFile("/tmp/example_01.txt") // notice def everywhere, looks like Source.fromFile could not be reused :(
    def source = s.getLines() 
    val destination = new PrintWriter(new File("/tmp/des_example_01.txt")) 

    var nrVariables: Int = 0 
    var nrLines: Int = 0 

    // here are the extracted lines from example_01 that fulfills some conditions. 
    def linesToWrite: Iterator[String] = source.filter { s => s.contains("a") } 

    def main(args: Array[String]): Unit = { 

    linesToWrite.foreach { s => 
     nrLines += 1 
     if (s contains "variable") { 
     nrVariables += 1 
     } 
    } 


    //Here is the header that I want to write in a destination file 
    destination.write("des_example_01.txt \n") 
    destination.write("Nr. of Variables and Lines: " + nrVariables + " " + nrLines + "\n") 

    for(line <- linesToWrite) { 
     println(line) 
     destination.write(line) 
     destination.write("\n") 
     /*nrLines += 1*/ 
    } 

    s.close() 
    destination.close() 

    } 
    }