From 89a9700222e4e16a9b53af3a1189c17fcdc4bca3 Mon Sep 17 00:00:00 2001
From: Jonathan Rampersad <rampersad.jonathan@gmail.com>
Date: Wed, 18 Jun 2025 09:10:37 -0400
Subject: [PATCH] FEAT: Added support for float coefficients

---
 README.md                 |  3 ++-
 pyproject.toml            |  2 +-
 src/polysolve/__init__.py | 25 +++++++++++++++----------
 3 files changed, 18 insertions(+), 12 deletions(-)

diff --git a/README.md b/README.md
index 738ea19..367d852 100644
--- a/README.md
+++ b/README.md
@@ -9,7 +9,7 @@ A Python library for representing, manipulating, and solving polynomial equation
 
 ## Key Features
 
-* **Create and Manipulate Polynomials**: Easily define polynomials of any degree and perform arithmetic operations like addition, subtraction, multiplication, and scaling.
+* **Create and Manipulate Polynomials**: Easily define polynomials of any degree using integer or float coefficients, and perform arithmetic operations like addition, subtraction, multiplication, and scaling.
 * **Genetic Algorithm Solver**: Find approximate real roots for complex polynomials where analytical solutions are difficult or impossible.
 * **CUDA Accelerated**: Leverage NVIDIA GPUs for a massive performance boost when finding roots in large solution spaces.
 * **Analytical Solvers**: Includes standard, exact solvers for simple cases (e.g., `quadratic_solve`).
@@ -44,6 +44,7 @@ Here is a simple example of how to define a quadratic function, find its propert
 from polysolve import Function, GA_Options, quadratic_solve
 
 # 1. Define the function f(x) = 2x^2 - 3x - 5
+#    Coefficients can be integers or floats.
 f1 = Function(largest_exponent=2)
 f1.set_constants([2, -3, -5])
 
diff --git a/pyproject.toml b/pyproject.toml
index f2b6f3c..a7bbb54 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -5,7 +5,7 @@ build-backend = "setuptools.build_meta"
 [project]
 # --- Core Metadata ---
 name = "polysolve"
-version = "0.2.2"
+version = "0.3.0"
 authors = [
   { name="Jonathan Rampersad", email="jonathan@jono-rams.work" },
 ]
diff --git a/src/polysolve/__init__.py b/src/polysolve/__init__.py
index 260acc9..931b2b1 100644
--- a/src/polysolve/__init__.py
+++ b/src/polysolve/__init__.py
@@ -15,7 +15,7 @@ except ImportError:
 # The CUDA kernel for the fitness function
 _FITNESS_KERNEL = """
 extern "C" __global__ void fitness_kernel(
-    const long long* coefficients, 
+    const double* coefficients, 
     int num_coefficients, 
     const double* x_vals, 
     double* ranks, 
@@ -76,13 +76,14 @@ class Function:
         self.coefficients: Optional[np.ndarray] = None
         self._initialized = False
 
-    def set_coeffs(self, coefficients: List[int]):
+    def set_coeffs(self, coefficients: List[Union[int, float]]):
         """
         Sets the coefficients of the polynomial.
 
         Args:
-            coefficients (List[int]): A list of integer coefficients. The list size
-                                   must be largest_exponent + 1.
+            coefficients (List[Union[int, float]]): A list of integer or float
+                                                   coefficients. The list size
+                                                   must be largest_exponent + 1.
 
         Raises:
             ValueError: If the input is invalid.
@@ -96,7 +97,7 @@ class Function:
         if coefficients[0] == 0 and self._largest_exponent > 0:
             raise ValueError("The first constant (for the largest exponent) cannot be 0.")
 
-        self.coefficients = np.array(coefficients, dtype=np.int64)
+        self.coefficients = np.array(coefficients, dtype=np.float64)
         self._initialized = True
 
     def _check_initialized(self):
@@ -279,7 +280,7 @@ class Function:
         fitness_gpu = cupy.RawKernel(_FITNESS_KERNEL, 'fitness_kernel')
         
         # Move coefficients to GPU
-        d_coefficients = cupy.array(self.coefficients, dtype=cupy.int64)
+        d_coefficients = cupy.array(self.coefficients, dtype=cupy.float64)
         
         # Create initial random solutions on the GPU
         d_solutions = cupy.random.uniform(
@@ -337,12 +338,16 @@ class Function:
             power = self._largest_exponent - i
             
             # Coefficient part
-            if c == 1 and power != 0:
+            coeff_val = c
+            if c == int(c):
+                coeff_val = int(c)
+
+            if coeff_val == 1 and power != 0:
                 coeff = ""
-            elif c == -1 and power != 0:
+            elif coeff_val == -1 and power != 0:
                 coeff = "-"
             else:
-                coeff = str(c)
+                coeff = str(coeff_val)
 
             # Variable part
             if power == 0:
@@ -356,7 +361,7 @@ class Function:
             sign = ""
             if i > 0:
                 sign = " + " if c > 0 else " - "
-                coeff = str(abs(c))
+                coeff = str(abs(coeff_val))
                 if abs(c) == 1 and power != 0:
                     coeff = "" # Don't show 1 for non-constant terms