ext/javax.mail.mbox/src/com/sun/mail/mbox/LineCounter.java

   1 /*
   2  * Copyright (c) 1997, 2018 Oracle and/or its affiliates. All rights reserved.
   3  *
   4  * This program and the accompanying materials are made available under the
   5  * terms of the Eclipse Public License v. 2.0, which is available at
   6  * http://www.eclipse.org/legal/epl-2.0.
   7  *
   8  * This Source Code may also be made available under the following Secondary
   9  * Licenses when the conditions for such availability set forth in the
  10  * Eclipse Public License v. 2.0 are satisfied: GNU General Public License,
  11  * version 2 with the GNU Classpath Exception, which is available at
  12  * https://www.gnu.org/software/classpath/license.html.
  13  *
  14  * SPDX-License-Identifier: EPL-2.0 OR GPL-2.0 WITH Classpath-exception-2.0
  15  */
  16
  17 package com.sun.mail.mbox;
  18
  19 import java.io.*;
  20
  21 /**
  22  * Count number of lines output.
  23  */
  24 class LineCounter extends FilterOutputStream {
  25     private int lastb = -1;
  26     protected int lineCount;
  27
  28     public LineCounter(OutputStream os) {
  29         super(os);
  30     }
  31
  32     public void write(int b) throws IOException {
  33         // If we have a full line, count it.
  34         if (b == '\r' || (b == '\n' && lastb != '\r'))
  35             lineCount++;
  36         out.write(b);
  37         lastb = b;
  38     }
  39
  40     public void write(byte[] b) throws IOException {
  41         write(b, 0, b.length);
  42     }
  43
  44     public void write(byte[] b, int off, int len) throws IOException {
  45         for (int i = 0 ; i < len ; i++) {
  46             write(b[off + i]);
  47         }
  48     }
  49
  50     public int getLineCount() {
  51         return lineCount;
  52     }
  53
  54     // for testing
  55     public static void main(String argv[]) throws Exception {
  56         int b;
  57         LineCounter os =
  58             new LineCounter(System.out);
  59         while ((b = System.in.read()) >= 0)
  60             os.write(b);
  61         os.flush();
  62         System.out.println(os.getLineCount());
  63     }
  64 }