Exemplo n.º 1
0
 public void drawCluster(Cluster cluster) {
   int index = clusters.indexOf(cluster);
   if (index != -1) {
     cluster.y = maxY;
     cluster.x = minX + index * factor;
     if (cluster.size() > 1) g.setColor(Color.RED);
     else g.setColor(Color.BLACK);
     g.drawRect(cluster.x - 1, cluster.y - cluster.size(), 2, 1 + cluster.size());
   } else {
     Cluster left = cluster.getLeft();
     Cluster right = cluster.getRight();
     drawCluster(left);
     drawCluster(right);
     int yBar = minY + (int) ((maxY - minY) * (cluster.getSimilarity() / threshold));
     g.setColor(Color.DARK_GRAY);
     if (left.y > yBar) {
       g.drawLine(left.x, left.y - 1, left.x, yBar);
       writeMap(left, yBar);
     }
     if (right.y > yBar) {
       g.drawLine(right.x, right.y - 1, right.x, yBar);
       writeMap(right, yBar);
     }
     g.setColor(Color.BLACK);
     g.drawLine(left.x, yBar, right.x, yBar);
     cluster.x = (right.x + left.x) / 2;
     cluster.y = yBar;
   }
 }
  @Override
  public void mouseDragged(MouseEvent e) {
    mouseX = e.getX();
    mouseY = e.getY();

    if (dragMoveButton.isChecked()) {
      if (structureBeingMoved) {
        int x = Math.min(Main.canvasWidth - 11, Math.max(10, e.getX()));
        int y = Math.min(Main.canvasHeight - 45, Math.max(45, e.getY()));

        if (selectedNeuron != null) {
          selectedNeuron.x = x;
          selectedNeuron.y = y;
        } else if (selectedCluster != null) {
          selectedCluster.x = x;
          selectedCluster.y = y;
        }
        redraw = true;
      }
    } else if (dragDeleteButton.isChecked()) {
      for (int i = 0; i < neurons.size(); i++)
        if (neurons.get(i).collides(mouseX, mouseY)) {
          neurons.get(i).delete();
          i--;
        }

      for (int i = 0; i < inputs.size(); i++)
        if (inputs.get(i).collides(mouseX, mouseY)) {
          inputs.get(i).delete();
          i--;
        }

      for (int i = 0; i < outputs.size(); i++)
        if (outputs.get(i).collides(mouseX, mouseY)) {
          outputs.get(i).delete();
          i--;
        }

      for (int i = 0; i < clusters.size(); i++)
        if (clusters.get(i).collides(mouseX, mouseY)) {
          clusters.get(i).delete();
          i--;
        }
    }
  }