APCS/06 Stacks _ Queues/12 Assembly Line/AssemblyLine_Driver.java

//Name: B6-24
//Date: 1/24/20
   import java.util.*;

    public class AssemblyLine_Driver
   {
      static int NDISKS = 50;
      static int MAXRADIUS = 100;
       public static void main(String[] args)
      {
         AssemblyLine a = new AssemblyLine(NDISKS, MAXRADIUS);
         a.showInput();
         a.process();
         a.showOutput();
      }
   }
   
    class AssemblyLine
   {
      private Queue<Disk> assemblyLineIn;
      private Queue<Pyramid> assemblyLineOut;
      private Pyramid robotArm;
   	/**
   		* initializes this object so the assemblyLineIn contains 
   		* nDisks with random radii;  assemblyLineOut is initialized to * an empty Queue; robotArm is initialized to an empty Pyramid.
   		**/
   	//Part A
       public AssemblyLine( int nDisks, int maxRadius )
      {
         assemblyLineIn = new LinkedList<Disk>();
         assemblyLineOut = new LinkedList<Pyramid>();
         robotArm = new Pyramid();
         
         for (int i = 0; i < nDisks; i++)
           assemblyLineIn.add(new Disk((int) (Math.random() * maxRadius)));   
         
      }
   
   	/**
   		* "flips over" the pyramid in the robotArm and adds it to the
   		* assemblyLineOut queue.
   		* Precondition:  robotArm is not empty and holds an inverted 
   		*				pyramid of disks
   		**/
   	// Part B
       private void unloadRobot()
      { 
         Pyramid p = new Pyramid();
         while (!robotArm.isEmpty())
            p.add(robotArm.pop());
         assemblyLineOut.add(p);
        
      }
   
   	/**
   		* processes all disks from assemblyLineIn; a disk is processed
   		* as follows:  if robotArm is not empty and the next disk does
   		* not fit on top of robotArm (which must be an inverted 
   		* pyramid) then robotArm is unloaded first; the disk from
   		* assemblyLineIn is added to robotArm; when all the disks
   		* have been retrieved from assemblyLineIn, robotArm is unloaded.
   		*  Precondition:  robotArm is empty;
   		*				assemblyLineOut is empty
   		**/
   	//Part C
       public void process()
      {
         while (!assemblyLineIn.isEmpty()) {
            Disk disk = assemblyLineIn.remove();
            if (!robotArm.isEmpty() && disk.compareTo(robotArm.peek()) < 0)
               unloadRobot();
            robotArm.push(disk);
         }
            
         
         unloadRobot();
      }
      
       public void showInput()
      {
         System.out.println(assemblyLineIn);
      }
       public void showOutput()
      {
         System.out.println(assemblyLineOut);
      }
   }
   //Disk is standard and straightforward.
    class Disk implements Comparable<Disk>
   {
      private int num;
      public Disk (int i) {
         num = i;
      }
      
      public int compareTo (Disk other) {
         return num - other.num;
      }
      
      public String toString () {
         return num + "";
      }
   }
   //Pyramid is sly.
    class Pyramid extends Stack<Disk>
   {
      public Pyramid() {
         super();
      }
   }