StdhepWriter.java

package hep.io.stdhep;

import hep.io.mcfio.MCFIOBlock;
import hep.io.mcfio.MCFIOEvent;
import hep.io.mcfio.MCFIOWriter;
import java.io.IOException;


/**
 * A class for writing stdhep files.
 * @author Tony Johnson (tonyj@slac.stanford.edu)
 * @version $Id: StdhepWriter.java 9132 2006-10-13 05:39:06Z tonyj $
 */
public class StdhepWriter extends MCFIOWriter implements StdhepConstants
{
   private static int[] blockIds = 
   {
      MCFIO_STDHEP, MCFIO_STDHEPM, MCFIO_STDHEPBEG, MCFIO_STDHEPEND
   };
   private static String[] blockNames = 
   {
      "Stdhep Event", "Stdhep Multi", "Stdhep Begin Run", "Stdhep End Run"
   };
   private MCFIOEventWrapper event = new MCFIOEventWrapper();

   public StdhepWriter(String file, String title, String comment, int numevts) throws IOException
   {
      super(file, title, comment, numevts, blockIds, blockNames);
   }

   public void writeRecord(StdhepRecord record) throws IOException
   {
      if (compatibilityMode && record instanceof StdhepEvent)
      {
         if (((StdhepEvent) record).getNHEP() > 4000)
         {
            throw new IOException("Too many particles");
         }
      }
      event.setRecord(record);
      write(event);
   }

   private class MCFIOEventWrapper implements MCFIOEvent
   {
      private StdhepRecord record;

      public MCFIOBlock getBlock(int index)
      {
         return record;
      }

      public int getBlockID(int index)
      {
         return record.getID();
      }

      public int getEventNumber()
      {
         if (record instanceof StdhepEvent)
            return ((StdhepEvent) record).getNEVHEP();
         return 0;
      }

      public int getNBlocks()
      {
         return 1;
      }

      public int getRunNumber()
      {
         return 0;
      }

      public int getStoreNumber()
      {
         return 0;
      }

      public int getTrigMask()
      {
         return 0;
      }

      void setRecord(StdhepRecord record) throws IOException
      {
         this.record = record;
      }
   }
}