#!/usr/bin/env python3
"""
Parallel Processing with Dask
Author: Molla Samser
Designer & Tester: Rima Khatun
Website: https://rskworld.in
Email: help@rskworld.in, support@rskworld.in
Phone: +91 93305 39277
"""

import dask.array as da
import numpy as np
import time
from dask import delayed, compute


def parallel_array_operations():
    """Demonstrate parallel array operations"""
    print("=" * 60)
    print("Parallel Array Operations")
    print("=" * 60)
    
    # Create large arrays
    print("\nCreating large arrays...")
    array1 = da.random.random((5000, 5000), chunks=(1000, 1000))
    array2 = da.random.random((5000, 5000), chunks=(1000, 1000))
    
    # Perform parallel operations
    print("Performing parallel operations...")
    start_time = time.time()
    
    result = (array1 + array2) * 2
    sum_result = result.sum()
    final = sum_result.compute()
    
    end_time = time.time()
    
    print(f"Result: {final}")
    print(f"Computation time: {end_time - start_time:.2f} seconds")
    print()


def parallel_data_processing():
    """Demonstrate parallel data processing with delayed functions"""
    print("=" * 60)
    print("Parallel Data Processing")
    print("=" * 60)
    
    @delayed
    def process_chunk(data):
        """Process a chunk of data"""
        time.sleep(0.1)  # Simulate processing
        return data.sum()
    
    # Create multiple data chunks
    print("\nCreating data chunks...")
    chunks = [np.random.rand(1000, 1000) for _ in range(10)]
    
    # Process in parallel
    print("Processing chunks in parallel...")
    start_time = time.time()
    
    tasks = [process_chunk(chunk) for chunk in chunks]
    results = compute(*tasks)
    
    end_time = time.time()
    
    print(f"Processed {len(results)} chunks")
    print(f"Total time: {end_time - start_time:.2f} seconds")
    print(f"Results: {[r for r in results]}")
    print()


def main():
    """Main function"""
    print("\nDask Parallel Processing Examples")
    print("=" * 60)
    
    parallel_array_operations()
    parallel_data_processing()
    
    print("=" * 60)
    print("All examples completed!")
    print("=" * 60)


if __name__ == "__main__":
    main()