ObjectAmountPostProcessor.java

Go to the documentation of this file.

package instruction.postprocessor;

import instruction.opencyc.OpenCyc;
import instruction.semanticObjects.Instruction;
import instruction.semanticObjects.ObjectX;
import instruction.semanticObjects.Preposition;
import instruction.semanticObjects.Quantifier;
import instruction.semanticObjects.SemanticObject;
import instruction.semanticObjects.Word;

import java.util.ArrayList;
import java.util.List;

public class ObjectAmountPostProcessor implements PostProcessor {

        public void run( List<Instruction> instructions ) {

                for ( int i = 0; i < instructions.size(); i++ )
                        findObjectAmount( instructions.get( i ), instructions.get( i ) );

        }

        private void findObjectAmount( SemanticObject so, SemanticObject parent ) {

                // ==================================================================
                // Instructions
                // ==================================================================
                if ( so instanceof Instruction ) {
                        Instruction in = (Instruction) so;

                        // Objects
                        for ( int i = 0; i < in.getObjects().size(); i++ )
                                findObjectAmount( in.getObjects().get( i ), in );

                        // Prepositions
                        for ( int i = 0; i < in.getPrepositions().size(); i++ )
                                findObjectAmount( in.getPrepositions().get( i ), in );
                }

                // ==================================================================
                // Objects
                // ==================================================================
                else if ( so instanceof ObjectX ) {
                        ObjectX o = (ObjectX) so;

                        for ( int i = 0; i < o.getName().size(); i++ ) {

                                Word n = o.getName().get( i );

                                for ( int j = 0; j < n.getCycConcepts().size(); j++ ) {

                                        String concept = n.getCycConcepts().get( j );

                                        List<ObjectX> newObjects = transformQuantifier( o, concept );

                                        if ( ! newObjects.isEmpty() ) {
                                                if ( parent instanceof Instruction ) {
                                                        ( (Instruction) parent ).getObjects().remove( o );
                                                        ( (Instruction) parent ).getObjects().addAll( newObjects );
                                                }
                                                else if ( parent instanceof Preposition ) {
                                                        ( (Preposition) parent ).getObjects().remove( o );
                                                        ( (Preposition) parent ).getObjects().addAll( newObjects );
                                                }

                                                findObjectAmount( parent, parent );
                                                break;
                                        }
                                }

                        }

                        // Prepositions
                        for ( int i = 0; i < o.getPrepositions().size(); i++ ) {
                                findObjectAmount( o.getPrepositions().get( i ), o );
                        }

                }

                // ==================================================================
                // Prepositions
                // ==================================================================
                else if ( so instanceof Preposition ) {
                        Preposition pp = (Preposition) so;

                        // Objects
                        for ( int i = 0; i < pp.getObjects().size(); i++ )
                                findObjectAmount( pp.getObjects().get( i ), pp );
                }
        }

        private List<ObjectX> transformQuantifier( ObjectX object, String concept ) {

                List<ObjectX> newObjects = new ArrayList<ObjectX>();

                for ( int i = 0; i < object.getPrepositions().size(); i++ ) {
                        Preposition pp = object.getPrepositions().get( i );

                        for ( int j = 0; j < pp.getPrepositions().size(); j++ ) {
                                Word p = pp.getPrepositions().get( j );

                                if ( p.getLabel().equalsIgnoreCase( "of" ) ) {

                                        // parent object becomes quantifier of child objects

                                        Quantifier q = new Quantifier();

                                        try {
                                                if ( OpenCyc.getInstance().isaInUnivVocMt( concept, "IntegerExtent" )
                                                                || OpenCyc.getInstance().genlsInUniversalVocabularyMt( concept, "IntegerExtent" ) ) {
                                                        if ( object.getName().size() > 0 ) {
                                                                object.getName().get( 0 ).getCycConcepts().clear();
                                                                object.getName().get( 0 ).getCycConcepts().add( concept );
                                                                q.getAlternatives().add( object.getName().get( 0 ) );
                                                        }
                                                }
                                                else if ( OpenCyc.getInstance().isaInUnivVocMt( concept, "UnitOfMeasure" )
                                                                || OpenCyc.getInstance().isaInUnivVocMt( concept, "PhysicalAmountSlot" )
                                                                || OpenCyc.getInstance().genlsInUniversalVocabularyMt( concept, "PhysicalAmountSlot" ) ) {
                                                        if ( object.getName().size() > 0 ) {
                                                                q.setMeasure( object.getName().get( 0 ) );
                                                                object.getName().get( 0 ).getCycConcepts().clear();
                                                                object.getName().get( 0 ).getCycConcepts().add( concept );
                                                        }
                                                        q.setAlternatives( object.getQuantifier().getAlternatives() );
                                                }
                                                else
                                                        return newObjects;

                                                for ( int k = 0; k < pp.getObjects().size(); k++ ) {

                                                        pp.getObjects().get( k ).setQuantifier( q );
                                                        newObjects.add( pp.getObjects().get( k ) );
                                                }

                                                return newObjects;
                                        }
                                        catch ( Exception e ) {
                                                System.out.println( e.getMessage() );
                                        }
                                }
                        }
                }
                return newObjects;
        }

}