source: fedd/abac-src/atnvis/src/com/algomagic/atn/Visualization.java @ 2c9fff0

package com.algomagic.atn;

import att.grappa.*;

import java.util.*;
import java.io.*;

import java.awt.*;
import java.awt.event.*;
import java.awt.geom.*;
import javax.swing.*;
import javax.swing.border.*;
import javax.swing.event.*;


public class Visualization extends JFrame
    implements VisualizationConstants {

    private Properties _properties;
    private VisualizationPanel[] _visPanels;

    private ControlPanel _controls;
    private ProgressPanel _progress;

    private String[] _names = new String[2];
    private Collection[] _policies = new Collection[2];

    public void setName( String name, int side ) {
        _names[side] = name;
    }

    public void setPolicies( Collection pol, int side ) {
        _policies[side] = pol;
    }



    public void setAutoRaise( boolean state ) {
        _visPanels[LEFT_SIDE].setAutoRaise( state );
        if( _visPanels[RIGHT_SIDE] != null ) {
            _visPanels[RIGHT_SIDE].setAutoRaise( state );
        }
        
    }


    public Visualization( String log1, String log2 ) {



        addWindowListener( new WindowAdapter( ) {
                public void windowClosing( WindowEvent we ) {
                    Window w = we.getWindow( );
                    w.setVisible( false );
                    w.dispose( );
                    System.exit( 0 );
                }
            });


        loadProperties( );
        initializeFrame( log2 != null );

        LogParser parser = null;

        try {
            parser = 
                (LogParser)Class.forName( _properties.getProperty( PARSER_PROP, 
                                                                   PARSER_DEFAULT ) ).newInstance( );
        } catch( Exception e ) {
            e.printStackTrace( );
            System.exit( 1 );
        }

        
        java.util.List updates = parser.parse( this, log1, log2 );

        updates = _visPanels[LEFT_SIDE].initialize( updates );
        _visPanels[LEFT_SIDE].addCredentials( _policies[LEFT_SIDE], 2 );
        _visPanels[LEFT_SIDE].setName( _names[LEFT_SIDE] );


//      System.out.println( _policies[LEFT_SIDE] );
//      System.out.println( _policies[RIGHT_SIDE] );


        if( log2 != null ) {
            updates = _visPanels[RIGHT_SIDE].initialize(  updates );
            _visPanels[RIGHT_SIDE].addCredentials( _policies[RIGHT_SIDE], 2 );
            _visPanels[RIGHT_SIDE].setName( _names[RIGHT_SIDE] );

        }
        


        java.util.List l = new ArrayList( );

        for( Iterator i = updates.iterator( ); i.hasNext( ); ) {
            Update u = (Update)i.next( );
            PanelUpdate pu = null;;

            if( u.getSide( ) == LEFT_SIDE ) {
                pu = _visPanels[LEFT_SIDE].apply( u );
            } else if( log2 != null ) {
                pu = _visPanels[RIGHT_SIDE].apply( u );
            }

            if( !pu.isEmpty( ) ) {
                l.add( pu );
            }

        }


//      _visPanels[LEFT_SIDE].reset( updates );
//      if( log2 != null ) {
//          _visPanels[RIGHT_SIDE].reset( updates );
//      }
        

        PlayBackController pbc = new PlayBackController( l, _properties );
        pbc.addObserver( _progress );
        pbc.addObserver( _visPanels[LEFT_SIDE] );
        _visPanels[LEFT_SIDE].hideGraph( );

        if( log2 != null ) {
            pbc.addObserver( _visPanels[RIGHT_SIDE] );
            _visPanels[RIGHT_SIDE].hideGraph( );
        }


        _progress.setMax( l.size( ) );
        _controls.addObserver( pbc );


// //   Object[] o = getKeyListeners( );
//      for( int i = 0; o != null && i<o.length; i++ ) {
//          removeKeyListener( (KeyListener)o[i] );
//      }



//      addKeyListener( new KeyListener( ) {
//              public void keyTyped( KeyEvent e ) {
//                  System.out.println( "typed: " + e );
//              }

//              public void keyPressed( KeyEvent e ) {
//                  System.out.println( "pressed: " + e );
//              }

//              public void keyReleased( KeyEvent e ) {
//                  System.out.println( "released: " + e );
//              }


//          });


//      addFocusListener( new FocusListener( ) {
//              public void focusGained( FocusEvent e ) {
//                  System.out.println( "focus gained: " + e );
//              }

//              public void focusLost( FocusEvent e ) {
//                  System.out.println( "focus lost: " + e ); 
// //               requestFocus( );
//              }


//          });
                       


        show( );
//      requestFocus( true );

    }

    private void initializeFrame( boolean twoSided ) {
        JPanel mainPanel = new JPanel( );
        mainPanel.setLayout( new BoxLayout( mainPanel, BoxLayout.X_AXIS ) );

        _visPanels = new VisualizationPanel[2];
        _visPanels[LEFT_SIDE] = new VisualizationPanel( LEFT_SIDE, _properties );
        mainPanel.add( _visPanels[LEFT_SIDE] );

        if( twoSided ) {
            _visPanels[RIGHT_SIDE] = new VisualizationPanel( RIGHT_SIDE, _properties );
            mainPanel.add( Box.createHorizontalStrut( 10 ) );
            mainPanel.add( _visPanels[RIGHT_SIDE] );
        }
        
        _controls = new ControlPanel( _properties, this );
        _progress = new ProgressPanel( );

        JPanel toolPanel = new JPanel( );

        toolPanel.setMaximumSize( new Dimension( 2000, 50 ) );
        toolPanel.setPreferredSize( new Dimension( 2000, 50 ) );


        toolPanel.setLayout( new BoxLayout( toolPanel, BoxLayout.X_AXIS ) );
        toolPanel.add( _controls );
        toolPanel.add( _progress );

        Container contentPane = getContentPane( );
        contentPane.setLayout( new BoxLayout( contentPane, BoxLayout.Y_AXIS ) );
        contentPane.add( mainPanel );
        contentPane.add( toolPanel );

        setTitle( _properties.getProperty( WINDOW_TITLE_PROP, WINDOW_TITLE_DEFAULT ) );
        setSize( Integer.parseInt( _properties.getProperty( WINDOW_WIDTH_PROP, WINDOW_WIDTH_DEFAULT ) ),
                 Integer.parseInt( _properties.getProperty( WINDOW_HEIGHT_PROP, WINDOW_HEIGHT_DEFAULT ) ) );

    }



    private void loadProperties( ) {
        String propsFileName = System.getProperty( CONFIG_FILE_PROP, CONFIG_FILE_DEFAULT );
        File file = new File( propsFileName );

        _properties = new Properties( );

        if( file.exists( ) ) {
            try {
                _properties.load( new FileInputStream( file ) );
            } catch( Exception e ) { }
        }

    }


    public static void main( String[] args ) {
        Visualization vis = null;

        if( args.length == 1 ) {
            vis = new Visualization( args[0], null );
        } else if ( args.length == 2 ) {
            vis = new Visualization( args[0], args[1] );
        } else {
            System.err.println( "usage: Visualization <log file> [<log file>]" );
            System.exit( 1 );
        }

    }


}