Merge pull request #3 from tataratat/collocation-solve

Collocation solve

README.md

@@ -13,8 +13,15 @@ pip install git+https://github.com/tataratat/feigen.git@main
 ```
 
 ## Quick start
+### iganet
 Current version supports iganet's BSplineSurface.
 Assuming that you have a server running,
 ```
 python3 -c "import feigen; feigen.BSpline2D('ws://localhost:9001').start()"
 ```
+
+### IGA examples
+#### Poisson problem 2D
+```
+python3 -c "import feigen; feigen.Poisson2D().start()"
+```

examples/poisson2d.py

@@ -0,0 +1,7 @@
+import feigen
+
+if __name__ == "__main__":
+    # feigen.log.configure(debug=True)
+
+    b = feigen.Poisson2D()
+    b.start()

feigen/__init__.py

@@ -1,11 +1,14 @@
-from feigen import bspline, comm, log
+from feigen import bspline, comm, log, poisson2d
 from feigen._version import __version__
 from feigen.bspline import BSpline2D
+from feigen.poisson2d import Poisson2D
 
 __all__ = [
     "__version__",
     "bspline",
     "comm",
     "log",
+    "poisson2d",
     "BSpline2D",
+    "Poisson2D",
 ]

feigen/bspline.py

@@ -62,8 +62,8 @@ def setup(self, boundary_splines):
         self.end_ids = (
             int(len(boundary_splines[0].cps) - 1),
             int(len(boundary_splines[0].cps) - 1),
-            int(len(boundary_splines[1].cps) - 1),
-            int(len(boundary_splines[1].cps) - 1),
+            int(len(boundary_splines[2].cps) - 1),
+            int(len(boundary_splines[2].cps) - 1),
         )
 
         # this is a bit verbose and can be also done with mod but