новая папка / лаб 4-5-6 / FractalExplorer

// import javafx.scene.control.ComboBox;
// import javafx.scene.layout.Border;
import java.awt.geom.Rectangle2D;
// import java.awt.geom.Rectangle2D.Double;
import java.awt.event.*;
// import javax.swing.JFileChooser.*;
import javax.swing.filechooser.*;
import java.awt.image.*;
import java.awt.*;
import javax.swing.*;

public class FractalExplorer {

/** Элементы, которые надо блокировать методом enableUI **/
private JComboBox<FractalGenerator> comboBox;
private JButton button_reset;
private JButton button_save;

private int screenSize;
private JImageDisplay img;
private FractalGenerator gen;
private Rectangle2D.Double rectangle;
private int rowsRemaining;

FractalExplorer(int size) {
// сохранение значения размера отображения в переменной screenSize
screenSize = size;
// инициализация объектов диапазона и фрактального генератора
gen = new Mandelbrot();
rectangle = new Rectangle2D.Double();
img = new JImageDisplay(screenSize, screenSize);
//
gen.getInitialRange(rectangle);

}

// создание интерфейса
public void createAndShowGUI() {
// создание рамки
JFrame frame = new JFrame("Fractal");
// закрытие окна по клику на крестик
frame.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
// определение размера окна
frame.setSize(screenSize, screenSize);
img.setLayout(new BorderLayout());
// делаем рамку видимой
frame.setVisible(true);

// инициализация панели с кнопкой ресета
JPanel but_panel = new JPanel();
// but_panel.setBackground(Color.BLACK);

// инициализация кнопки ресета
JButton reset_b = new JButton("reset");
// определение размера кнопки
reset_b.setPreferredSize(new Dimension(200, 100));

ActionListener DisplayReset = new ResetDisplay();
reset_b.addActionListener(DisplayReset);

// добавление кнопки на панель
but_panel.add(reset_b);

MouseHandler click = new MouseHandler();
img.addMouseListener(click);

// добавление панели на рамку и привязка ее к середине снизу рамки
frame.add(but_panel, BorderLayout.SOUTH);
frame.add(img, BorderLayout.CENTER);

JPanel top_panel = new JPanel();
JLabel lable = new JLabel("Fractal: ");
top_panel.add(lable);

FractalGenerator tricorn_frac = new Tricorn();
FractalGenerator ship_frac = new BurningShip();

JComboBox<FractalGenerator> comboBox = new JComboBox<>();
comboBox.addItem(gen);
comboBox.addItem(tricorn_frac);
comboBox.addItem(ship_frac);

ActionListener choose = new ChooseFractal();
comboBox.addActionListener(choose);

top_panel.add(comboBox);
frame.add(top_panel, BorderLayout.NORTH);

JButton save_b = new JButton("save");
save_b.setPreferredSize(new Dimension(200, 100));
ActionListener save = new SavePic();
save_b.addActionListener(save);

but_panel.add(save_b);

frame.pack();
frame.setResizable(false);

}

public void drawFractal() {
// отключение всех элементов пользовательского интерфейса во время рисования
enableUI(false);
// определяем количество строк, которые нужно нарисовать
rowsRemaining = screenSize;
for (int x = 0; x < screenSize; x++) {
FractalWorker drawRow = new FractalWorker(x);
// запуск фонового потока
drawRow.execute();
}
}

// отключение/включение элементов пользовательского интерфейса
private void enableUI(boolean value) {
comboBox.setEnabled(value);
button_reset.setEnabled(value);
button_save.setEnabled(value);
}

private class ResetDisplay implements ActionListener {
public void actionPerformed(ActionEvent e) {
gen.getInitialRange(rectangle);
FractalExplorer.this.drawFractal();
}
}

private class SavePic implements ActionListener {
public void actionPerformed(ActionEvent e) {
// объект, в который будем сохранять картинку
JFileChooser chooser = new JFileChooser();
// указываем фильтр по расширению файла
FileFilter filter = new FileNameExtensionFilter("PNG Images", "png");
chooser.setFileFilter(filter);
chooser.setAcceptAllFileFilterUsed(false);
// вызываем диалоговое окно сохранения файла
int userSelection = chooser.showSaveDialog(img);
// если юзер подтверждает сохранение
if (userSelection == JFileChooser.APPROVE_OPTION) {
// узнаем директорию, выбранную пользователем
java.io.File file = chooser.getSelectedFile();
try {
BufferedImage displayImage = img.getImage();
// записываем картинку в указанный файл
javax.imageio.ImageIO.write(displayImage, "png", file);
} catch (Exception exception) {
// сообщение об ошибке
JOptionPane.showMessageDialog(img, exception.getMessage(), "Cannot Save Image",
JOptionPane.ERROR_MESSAGE);
}
}
}
}

// выбор фрактала через комбобокс
private class ChooseFractal implements ActionListener {
public void actionPerformed(ActionEvent e) {
JComboBox<FractalGenerator> target = (JComboBox<FractalGenerator>) e.getSource();
gen = (FractalGenerator) target.getSelectedItem();
gen.getInitialRange(rectangle);
FractalExplorer.this.drawFractal();
}
}

private class MouseHandler extends MouseAdapter {

@Override
public void mousePressed(MouseEvent e) {
if (rowsRemaining != 0) {
return;
}
int x = e.getX();
int y = e.getY();
double xCoord = FractalGenerator.getCoord(rectangle.x, rectangle.x + rectangle.width, screenSize, x);
double yCoord = FractalGenerator.getCoord(rectangle.y, rectangle.y + rectangle.height, screenSize, y);
gen.recenterAndZoomRange(rectangle, xCoord, yCoord, 0.5);
FractalExplorer.this.drawFractal();
}
}

// вычисляет значения цветов строки фрактала
private class FractalWorker extends SwingWorker<Object, Object> {
// поле строки
int yCoordinate;
// массив с цветами пикселей
int[] pixColor;

private FractalWorker(int yTarget) {
yCoordinate = yTarget;
}

// фоновый поток, сохранение каждого значения RGB в соответствующем элементе
// целочисленного массива
protected Object doInBackground() {
pixColor = new int[screenSize];

double xCoord;
double yCoord;
for (int i = 0; i < pixColor.length; i++) {
xCoord = FractalGenerator.getCoord(rectangle.x, rectangle.x + rectangle.width, screenSize, i);
yCoord = FractalGenerator.getCoord(rectangle.y, rectangle.y + rectangle.height, screenSize,
yCoordinate);
int iterations_number = gen.numIterations(xCoord, yCoord);

if (iterations_number == -1) {
pixColor[i] = 0;
} else {
float hue = 0.7f + (float) iterations_number / 200f;
int rgbColor = Color.HSBtoRGB(hue, 1f, 1f);
pixColor[i] = rgbColor;

}
}

return null;
}

// перебираем массив строк, рисуем пиксели, находящиеся в нем
protected void done() {
for (int i = 0; i < pixColor.length; i++) {
img.drawPixel(i, yCoordinate, pixColor[i]);
}

// указываем область для перерисовки
img.repaint(0, 0, yCoordinate, screenSize, 1);
rowsRemaining--;
if (rowsRemaining == 0) {
enableUI(true);
}
}

}

public static void main(String[] args) {
FractalExplorer explorer = new FractalExplorer(600);
explorer.createAndShowGUI();
explorer.drawFractal();
}

}