/*
    * Copyright (c) 2004, RV Test Team
    * All rights reserved.
    *
    * Redistribution and use in source and binary forms, with or without
    * modification, are permitted provided that the following conditions are met:
    *
    * Redistributions of source code must retain the above copyright notice, this
    * list of conditions and the following disclaimer.
   *
   * Redistributions in binary form must reproduce the above copyright notice,
   * this list of conditions and the following disclaimer in the documentation
   * and/or other materials provided with the distribution.
   *
   * Neither the name of the "RV Test Team" nor the names of its contributors may
   * be used to endorse or promote products derived from this software without
   * specific prior written permission.
   *
   * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
   * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
   * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
   * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
   * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
   * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
   * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
   * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
   * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
   * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
   * THE POSSIBILITY OF SUCH DAMAGE.
   *
   */
  /*
   * Copyright (c) 2002, Reuters
   * All rights reserved.
   *
   * Redistribution and use in source and binary forms, with or without
   * modification, are permitted provided that the following conditions are met:
   *
   * Redistributions of source code must retain the above copyright notice, this
   * list of conditions and the following disclaimer.
   * Redistributions in binary form must reproduce the above copyright notice,
   * this list of conditions and the following disclaimer in the documentation
   * and/or other materials provided with the distribution.
   * Neither the name of the Reuters nor the names of its contributors may be
   * used to endorse or promote products derived from this software without
   * specific prior written permission.
   *
   * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
   * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
   * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
   * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
   * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
   * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
   * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
   * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
   * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
   * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
   * THE POSSIBILITY OF SUCH DAMAGE.
   *
   */
  package com.reuters.msgtest.load.output;
  
  import com.reuters.msgtest.load.config.LoadTestConfig;
  import org.apache.commons.logging.Log;
  import org.apache.commons.logging.LogFactory;
  import java.io.BufferedOutputStream;
  import java.io.FileOutputStream;
  import java.io.PrintWriter;
  import java.util.Date;
  import java.util.Hashtable;
  import java.util.Iterator;
  import java.util.Vector;
  
  
  // TODO: remove the compilation unit after splitting a logic into different classes.
  public class LoadTestPrinter {
      private Log log = LogFactory.getLog(LoadTestPrinter.class);
      private LoadTestConfig loadTestConfig;
      private String outputFileName;
      private String timeKeyName;
      private Vector messageNames;
      private Vector timeKeys;
      private Hashtable timeMatrix;
      private PrintWriter printWriter = null;
  
      public LoadTestPrinter() {
      }
  
      private final void createFileWithDir(String fileName) {
          int index = fileName.lastIndexOf('/');
  
          if (index != -1) {
              String dirName = fileName.substring(0, index);
              java.io.File dir = new java.io.File(dirName);
  
              if (!dir.exists()) {
                  dir.mkdirs();
              }
          }
 
         java.io.File file = new java.io.File(fileName);
 
         try {
             file.createNewFile();
         } catch (java.io.IOException ioe) {
             ioe.printStackTrace();
         }
     }
 
     public void genOutput() {
         try {
             // first make sure that the full path to the output file is created
             createFileWithDir(getOutputFileName());
 
             // TODO: provide more fine grained interface that would accept printwriter 
             // and data to process. 
             FileOutputStream fout = new FileOutputStream(getOutputFileName(),
                     true);
             printWriter = new PrintWriter(new BufferedOutputStream(fout));
 
             //Write Test Header
             printWriter.println("# BEGIN ------------------");
             printWriter.println("# Date: " + new Date());
             printWriter.println("# TestName = " +
                 getLoadTestConfig().getName());
             printWriter.println("# Total # of Messages: " + timeMatrix.size());
             printWriter.print("#  " + getTimeKeyName() + ", ");
 
             for (Iterator iter = getMessageNames().iterator(); iter.hasNext();) {
                 printWriter.print((String) iter.next() + ", ");
             }
 
             //Output the values.
             Vector mintime = new Vector();
             Vector maxtime = new Vector();
             Vector avgtime = new Vector();
             int rowsize = 0;
 
             for (int k = 0; k < timeKeys.size(); k++) {
                 String aKey = (String) timeKeys.elementAt(timeKeys.size() - k -
                         1);
 
                 Vector timeVec = (Vector) timeMatrix.get(aKey);
                 printWriter.print(aKey);
                 rowsize = timeVec.size();
 
                 for (int i = 0; i < timeVec.size(); i++) {
                     Long recordedTime = (Long) timeVec.elementAt(i);
                     Long tStart = (Long) timeVec.elementAt(0);
                     long tInterval = recordedTime.longValue() -
                         tStart.longValue();
 
                     //use absolute time.
                     printWriter.print(", " + recordedTime.longValue());
 
                     //m_PrintWriter.print(" ,  "+tInterval);
                     if ((k == 0) || (mintime.size() <= i)) {
                         mintime.addElement(new Long(tInterval));
                         maxtime.addElement(new Long(tInterval));
                         avgtime.addElement(new Long(tInterval));
                     } else {
                         Long oldmintime = (Long) mintime.elementAt(i);
                         Long oldmaxtime = (Long) maxtime.elementAt(i);
                         Long oldavgtime = (Long) avgtime.elementAt(i);
 
                         if (tInterval < oldmintime.longValue()) {
                             mintime.setElementAt(new Long(tInterval), i);
                         }
 
                         if (tInterval > oldmaxtime.longValue()) {
                             maxtime.setElementAt(new Long(tInterval), i);
                         }
 
                         avgtime.setElementAt(new Long(tInterval +
                                 oldavgtime.longValue()), i);
                     }
                 }
 
                 printWriter.println();
 
                 //end
             }
 
             //Print out some summary statistics.
             printWriter.print("# min ");
 
             for (int i = 0; i < rowsize; i++) {
                 if (i == (rowsize - 1)) {
                     printWriter.println(mintime.elementAt(i));
                 } else {
                     printWriter.print(mintime.elementAt(i) + ", ");
                 }
             }
 
             printWriter.print("# max ");
 
             for (int i = 0; i < rowsize; i++) {
                 if (i == (rowsize - 1)) {
                     printWriter.println(maxtime.elementAt(i));
                 } else {
                     printWriter.print(maxtime.elementAt(i) + ", ");
                 }
             }
 
             printWriter.print("# avg ");
 
             for (int i = 0; i < rowsize; i++) {
                 Long curtotal = (Long) avgtime.elementAt(i);
 
                 //       double curavg = (curtotal.longValue()*1.0)/timeKeyList.size();
                 double curavg = (curtotal.longValue() * 1.0) / timeMatrix.size();
 
                 if (i == (rowsize - 1)) {
                     printWriter.println(curavg);
                 } else {
                     printWriter.print(curavg + ", ");
                 }
             }
 
             printWriter.println("# END   ------------------");
             printWriter.close();
             fout.close();
         } catch (java.io.FileNotFoundException e) {
             log.error("could not open outputfile: " + outputFileName, e);
         } catch (java.io.IOException e2) {
             log.error("could not close outputfile", e2);
         }
     }
 
     public void genUnitTestOutput() {
         try {
             // first make sure that the full path to the output file is created
             createFileWithDir(outputFileName);
 
             FileOutputStream fout = new FileOutputStream(outputFileName, true);
             printWriter = new PrintWriter(fout);
 
             //mlp Loadtestconfig config = getLoadTestInfo().getLoadTestConfig();
             //Write Test Header
             printWriter.println("# BEGIN ------------------");
             printWriter.println("# Date: " + new Date());
 
             String loadTestName = loadTestConfig.getName();
             printWriter.println("# TestName = " + loadTestName);
             printWriter.println("# Total # of Messages: " + timeKeys.size());
 
             printWriter.print("#  " + timeKeyName + ", ");
 
             int namelistsize = messageNames.size();
 
             for (int i = 0; i < namelistsize; i++) {
                 if (i > 0) {
                     printWriter.print(", ");
                 }
 
                 printWriter.print(messageNames.elementAt(i));
             }
 
             printWriter.println();
 
             Vector mintime = new Vector();
             Vector maxtime = new Vector();
             Vector sumtime = new Vector();
 
             //Used for actual throughput
             Vector minLegTimes = new Vector();
             Vector maxLegTimes = new Vector();
 
             // initialize the statistics
             int vecSize = ((Vector) timeMatrix.get(timeKeys.elementAt(0))).size();
 
             for (int i = 0; i < vecSize; i++) {
                 mintime.addElement(new Long(Long.MAX_VALUE));
                 maxtime.addElement(new Long(0));
                 sumtime.addElement(new Long(0));
                 minLegTimes.addElement(new Long(Long.MAX_VALUE));
                 maxLegTimes.addElement(new Long(0));
             }
 
             for (int k = 0; k < timeKeys.size(); k++) {
                 String aKey = (String) timeKeys.elementAt(timeKeys.size() - k -
                         1);
                 Vector timeVec = (Vector) timeMatrix.get(aKey);
 
                 Long tStart = (Long) timeVec.elementAt(0);
 
                 for (int i = 0; i < vecSize; i++) {
                     Long recordedTime = (Long) timeVec.elementAt(i);
 
                     long tInterval = recordedTime.longValue() -
                         tStart.longValue();
 
                     if (i > 1) {
                         printWriter.print(" ,  ");
                     }
 
                     if (i > 0) {
                         printWriter.print(tInterval);
                     }
 
                     long min = Math.min(((Long) mintime.elementAt(i)).longValue(),
                             tInterval);
                     long max = Math.max(((Long) maxtime.elementAt(i)).longValue(),
                             tInterval);
                     long sum = ((Long) sumtime.elementAt(i)).longValue() +
                         tInterval;
 
                     long maxLegTime = Math.max(((Long) maxLegTimes.elementAt(i)).longValue(),
                             recordedTime.longValue());
                     long minLegTime = Math.min(((Long) minLegTimes.elementAt(i)).longValue(),
                             recordedTime.longValue());
 
                     mintime.setElementAt(new Long(min), i);
                     maxtime.setElementAt(new Long(max), i);
                     sumtime.setElementAt(new Long(sum), i);
                     maxLegTimes.setElementAt(new Long(maxLegTime), i);
                     minLegTimes.setElementAt(new Long(minLegTime), i);
                 }
 
                 printWriter.println();
             }
 
             printWriter.println("\nTotal # of Messages: " + timeKeys.size());
 
             long burstTime = ((Long) maxLegTimes.elementAt(0)).longValue() -
                 ((Long) minLegTimes.elementAt(0)).longValue();
             printWriter.print("Stimulus Burst Time:" + burstTime);
 
             printWriter.print("\nTotal Processing Time: ");
 
             for (int i = 1; i < vecSize; i++) {
                 if (i > 1) {
                     printWriter.print(" ,  ");
                 }
 
                 long time = ((Long) maxLegTimes.elementAt(i)).longValue() -
                     ((Long) minLegTimes.elementAt(0)).longValue();
                 printWriter.print(time);
             }
 
             printWriter.print("\nAverage Processing Time: ");
 
             for (int i = 1; i < vecSize; i++) {
                 if (i > 1) {
                     printWriter.print(" ,  ");
                 }
 
                 long time = ((Long) maxLegTimes.elementAt(i)).longValue() -
                     ((Long) minLegTimes.elementAt(0)).longValue();
                 printWriter.print((time * 1.0) / timeKeys.size());
             }
 
             //Print out some summary statistics.
             printWriter.print("\n\nMin (perceived): ");
 
             for (int i = 1; i < vecSize; i++) {
                 if (i > 1) {
                     printWriter.print(" ,  ");
                 }
 
                 printWriter.print(mintime.elementAt(i));
             }
 
             printWriter.print("\nMax (perceived): ");
 
             for (int i = 1; i < vecSize; i++) {
                 if (i > 1) {
                     printWriter.print(" ,  ");
                 }
 
                 printWriter.print(maxtime.elementAt(i));
             }
 
             printWriter.print("\nAvg (perceived): ");
 
             for (int i = 1; i < vecSize; i++) {
                 if (i > 1) {
                     printWriter.print(" ,  ");
                 }
 
                 Long curtotal = (Long) sumtime.elementAt(i);
                 double curavg = (curtotal.longValue() * 1.0) / timeKeys.size();
                 printWriter.print(curavg);
             }
 
             printWriter.println("\n# END   ------------------");
             printWriter.close();
             fout.close();
         } catch (java.io.FileNotFoundException e) {
             log.error("could not open outputfile: " + outputFileName, e);
         } catch (java.io.IOException e2) {
             log.error("could not close outputfile", e2);
         }
     }
 
     public String getOutputFileName() {
         return outputFileName;
     }
 
     public void setOutputFileName(String outputFileName) {
         this.outputFileName = outputFileName;
     }
 
     public String getTimeKeyName() {
         return timeKeyName;
     }
 
     public void setTimeKeyName(String timeKeyName) {
         this.timeKeyName = timeKeyName;
     }
 
     public Vector getTimeKeys() {
         return timeKeys;
     }
 
     public void setTimeKeys(Vector timeKeys) {
         this.timeKeys = timeKeys;
     }
 
     public Hashtable getTimeMatrix() {
         return timeMatrix;
     }
 
     public void setTimeMatrix(Hashtable timeMatrix) {
         this.timeMatrix = timeMatrix;
     }
 
     public LoadTestConfig getLoadTestConfig() {
         return loadTestConfig;
     }
 
     public void setLoadTestConfig(LoadTestConfig loadTestConfig) {
         this.loadTestConfig = loadTestConfig;
     }
 
     public Vector getMessageNames() {
         return messageNames;
     }
 
     public void setMessageNames(Vector messageNames) {
         this.messageNames = messageNames;
     }
 }