Draft: optional callback for write_output

The current implementation ignores output from ocean, because init_logger is called before splitting of ocean and atmoshphere.

Instead of calling init_logger a second time I prefer to pass another callback-function instead.

This callback is only taken into account if present (using a logical OR).