coding UTF-8;        // Specify character encoding to prevent garbled text
import Math;         // Import the library for mathematical functions
import GUI;          // Import the library for building GUI windows
import tool.Graph3D; // Import the library for 3D graph plotting


// Variables to store wave parameters
double amplitude = 1.0;   // Wave amplitude (A)
double wavelength = 2.0;  // Wavelength of the wave (λ)
double period = 1.0;      // Period of the wave (T)

// Configuration variables for the graph range, number of points, and time increment
const int N = 100;           // Number of grid points per axis (higher = smoother mesh)
const double X_MIN = -5.0;   // Minimum value on the X-axis
const double X_MAX = 5.0;    // Maximum value on the X-axis
const double Y_MIN = -5.0;   // Minimum value on the Y-axis
const double Y_MAX = 5.0;    // Maximum value on the Y-axis
const double Z_MIN = -1.0;   // Minimum value on the Z-axis
const double Z_MAX = 1.0;    // Maximum value on the Z-axis
const double DT = 0.1;       // Time increment per loop iteration (when speed = 1.0)
const double DX = (X_MAX - X_MIN) / (N-1); // Mesh interval along the X-axis
const double DY = (Y_MAX - Y_MIN) / (N-1); // Mesh interval along the Y-axis

// Coordinate arrays to pass wave data to the graph
double waveX[ N ][ N ];  // X-coordinate array
double waveY[ N ][ N ];  // Y-coordinate array
double waveZ[ N ][ N ];  // Z-coordinate array

// Variables for the GUI (setting window)
int window;             // Stores the window ID
int periodSlider;       // Stores the ID of the period slider
int wavelengthSlider;   // Stores the ID of the wavelength slider
int amplitudeSlider;    // Stores the ID of the amplitude slider
int speedSlider;        // Stores the ID of the animation speed slider
int infoLabel;          // Stores the ID of the label showing current settings
int timeLabel;          // Stores the ID of the label showing current time
int resetButton;        // Stores the ID of the reset button for time

// Other control variables
float speed = 0.7;         // Animation speed
int graph;                 // Stores the ID of the graph software
bool continuesLoop = true; // Controls whether the animation keeps running
double t;                  // Time variable


// ======================================================================
// Main procedure - this function runs automatically on startup
// ======================================================================

void main( ){
	
	// Launch the graph window
	createGraphWindow();
	
	// Build the GUI (setting window)
	createSettingWindow();
	
	t = 0.0; // Initialize time variable
	
	// Animation loop
	while( continuesLoop ) {
		
		// Calculate coordinates for the wave and pass them to the graph
		for( int yi=0; yi<N; yi++ ) {
			for( int xi=0; xi<N; xi++ ) {
				
				double x = X_MIN + xi * DX; // X coordinate at xi-th grid line
				double y = Y_MIN + yi * DY; // Y coordinate at yi-th grid line
				double r = sqrt(x*x + y*y); // Distance from the origin
				
				waveX[yi][xi] = x;
				waveY[yi][xi] = y;
				waveZ[yi][xi] = amplitude * sin( 2.0 * PI * ( t/period - r/wavelength ) );
			}
		}
		
		// Send the wave data to the 3D graph
		setGraph3DData( graph, waveX, waveY, waveZ );
		
		// Advance time (scaled by animation speed)
		t += DT * speed;
		
		// Wait for 30 milliseconds to control animation speed
		sleep( 30 );
	}
	
	exit(); // Exit the program when animation loop ends
}


// ======================================================================
// グFunction to launch and configure the 3D graph window
// ======================================================================

void createGraphWindow() {
	
	// Launch the 3D graphing software
	graph = newGraph3D( 0, 0, 800, 600, "Graph Window" );
	
	// Load fine-tuned settings (e.g., aspect ratio, scale, view angle) from a config file
	setGraph3DConfigurationFile(graph, "RinearnGraph3DSetting.ini");
	
	// Disable auto range and manually set the graph range
	setGraph3DAutoRange( graph, false, false, false );
	setGraph3DRange( graph, X_MIN, X_MAX, Y_MIN, Y_MAX, Z_MIN, Z_MAX );
	
	// Plot settings: disable point markers, enable surface plot
	setGraph3DOption(graph, "WITH_POINTS", false);
	setGraph3DOption(graph, "WITH_MEMBRANES", true);
}


// ======================================================================
// Function to build the setting window
// ======================================================================

void createSettingWindow() {

	// Create the window
	window = newWindow( 0, 610, 800, 230, "Setting Window" );
	
	// Create and place the time display label
	timeLabel = newTextLabel( 10, 10, 200, 20, "" );
	mountComponent( timeLabel, window );
	
	// Create and place the labels to the left of sliders
	int amplitudeLabel = newTextLabel( 10, 40, 135, 15, "A: Amplitude (0-1)" );
	mountComponent( amplitudeLabel, window );
	int omegaLabel = newTextLabel( 10, 60, 135, 15, "T: Period (0.2-10)" );
	mountComponent( omegaLabel, window );
	int lambdaLabel = newTextLabel( 10, 80, 135, 15, "λ: Wavelength (1-8)" );
	mountComponent( lambdaLabel, window );
	
	// Create and place the amplitude slider (range 0.0-1.0)
	amplitudeSlider = newHorizontalSlider( 150, 40, 600, 20, amplitude, 0.0, 1.0 );
	mountComponent( amplitudeSlider, window );
	
	// Create and place the period slider (range 0.2-10.0)
	periodSlider = newHorizontalSlider( 150, 60, 600, 20, period, 0.2, 10.0 );
	mountComponent( periodSlider, window );
	
	// Create and place the wavelength slider (range 1.0-8.0)
	wavelengthSlider = newHorizontalSlider( 150, 80, 600, 20, wavelength, 1.0, 8.0 );
	mountComponent( wavelengthSlider, window );
	
	// Create and place the label to show current slider values
	infoLabel = newTextLabel( 150, 100, 400, 20, "" );	
	mountComponent( infoLabel, window );
	updateInfoLabel();
	
	// Create and place the label and slider for animation speed
	int speedLabel = newTextLabel( 10, 140, 140, 15, "Animation Speed" );
	mountComponent( speedLabel, window );
	speedSlider = newHorizontalSlider( 150, 140, 400, 20, speed, 0.0, 1.0 );
	mountComponent( speedSlider, window );
	
	// Create and place the reset button for time
	resetButton = newButton(600, 125, 150, 40, "Reset Time");
	mountComponent( resetButton, window );
	
	// Refresh the window after placing components
	paintComponent( window );
}


// ======================================================================
// Function to update the label that shows current slider values
// ======================================================================

void updateInfoLabel() {
	string infoText
		= "( A="
		+ round(amplitude, 2, HALF_UP) // Round amplitude to 2 decimal place
		+ ", λ="
		+ round(wavelength, 2, HALF_UP) // Round wavelength to 2 decimal places
		+ ", T="
		+ round(period, 2, HALF_UP) // Round period to 2 decimal places
		+ ")";
	
	setComponentString(infoLabel, infoText);
	paintComponent(infoLabel);
	paintComponent(window);
}


// ======================================================================
// Event handler: called when a slider is moved
// ======================================================================

void onSliderMove(int id, float value) {
	
	// Set the value to the corresponding global variable
	
	if(id == amplitudeSlider) {
		amplitude = value;
	}
	
	if(id == periodSlider) {
		period = value;
	}
	
	if(id == wavelengthSlider) {
		wavelength = value;
	}
	
	if (id == speedSlider) {
		speed = value;
	}
	
	updateInfoLabel();
}


// ======================================================================
// Event handler: called when a button is clicked
// ======================================================================

void onButtonClick(int id) {
	
	// If the reset button is clicked, reset time variable t
	if (id == resetButton) {
		t = 0.0;
	}
}


// ======================================================================
// Event handler: called when the window is closed
// ======================================================================

void onWindowClose(int id) {
	
	// Exit the animation loop and terminate the program
	continuesLoop = false;
}


// ======================================================================
// Event handler: called when the 3D graph window is closed
// ======================================================================

void onGraph3DClose(int id) {

	// Exit the animation loop and terminate the program
	continuesLoop = false;
}