import javax.swing.JComponent;
import java.util.*;
import java.awt.*;
import java.awt.event.MouseEvent;
import javax.swing.event.MouseInputAdapter;

@SuppressWarnings("serial")
public class SketcherView extends JComponent implements Observer {
  public SketcherView(Sketcher theApp) {
    this.theApp = theApp;
    MouseHandler handler = new MouseHandler();                         // create the mouse listener
    addMouseListener(handler);                                         // Listen for button events
    addMouseMotionListener(handler);                                   // Listen for motion events
  }

  // Method called by Observable object when it changes
  public void update(Observable o, Object rectangle) {
    if(rectangle != null) {
      repaint((java.awt.Rectangle)rectangle);
    } else {
      repaint();
    }
  }

  // Method to draw on the view
  @Override
  public void paint(Graphics g) {
    Graphics2D g2D = (Graphics2D)g;                                    // Get a 2D device context
    for(Element element: theApp.getModel()) {                          // For each element in the model
      element.draw(g2D);                                               // ...draw the element
    }
  }

  class MouseHandler extends MouseInputAdapter {
    @Override
    public void mousePressed(MouseEvent e)  {
      start = e.getPoint();                                            // Save the cursor position in start
      buttonState = e.getButton();                                     // Record which button was pressed
      if(buttonState == MouseEvent.BUTTON1) {
        g2D = (Graphics2D)getGraphics();                               // Get graphics context
        g2D.setXORMode(getBackground());                               // Set XOR mode
      }
    }

    @Override
    public void mouseDragged(MouseEvent e) {
      last = e.getPoint();                                             // Save cursor position

      if(buttonState == MouseEvent.BUTTON1) {
        if(tempElement == null) {                                      // Is there an element?
          tempElement = Element.createElement(                         // No, so create one
                                 theApp.getWindow().getElementType(),
                                 theApp.getWindow().getElementColor(),
                                 start, last);
        } else {
          tempElement.draw(g2D);                                       // Yes – draw to erase it
          tempElement.modify(start, last);                             // Now modify it
        }
        tempElement.draw(g2D);                                         // and draw it
     }
    }

    @Override
    public void mouseReleased(MouseEvent e) {
      if(e.getButton() == MouseEvent.BUTTON1) {
        buttonState = MouseEvent.NOBUTTON;                             // Reset the button state

        if(tempElement != null) {                                      // If there is an element...
          theApp.getModel().add(tempElement);                          // ...add it to the model...
          tempElement = null;                                          // ...and reset the field
        }
        if(g2D != null) {                                              // If there’s a graphics context
          g2D.dispose();                                               // ...release the resource...
          g2D = null;                                                  // ...and reset field to null
        }
        start = last = null;                                           // Remove any points
      }
    }

    private Point start;                                               // Stores cursor position on press
    private Point last;                                                // Stores cursor position on drag
    private Element tempElement = null;                                // Stores a temporary element
    private int buttonState = MouseEvent.NOBUTTON;                     // Records button state
    private Graphics2D g2D = null;                                     // Temporary graphics context
   }

  private Sketcher theApp;                                             // The application object
}