Merge pull request #1196 from BellaLq/master

TriangleMesh增加device

fealpy/experimental/mesh/lagrange_triangle_mesh.py

@@ -314,9 +314,6 @@ def jacobi_matrix(self, bc: TensorLike, p=None, index: Index=_S, return_grad=Fal
         else:
             return J, gphi
 
-    def uniform_refine(self, n=1):
-        pass
-
     # fundamental form
     def first_fundamental_form(self, bc: Union[TensorLike, Tuple[TensorLike]], 
             index: Index=_S, return_jacobi=False, return_grad=False):

fealpy/experimental/mesh/triangle_mesh.py

@@ -12,21 +12,23 @@
 from scipy.sparse import spdiags, eye, tril, triu, bmat
 
 class TriangleMesh(SimplexMesh, Plotable):
-    def __init__(self, node: TensorLike, cell: TensorLike) -> None:
+    def __init__(self, node: TensorLike, cell: TensorLike, device=None) -> None:
         """
         """
         super().__init__(TD=2, itype=cell.dtype, ftype=node.dtype)
         kwargs = bm.context(cell) 
+        self.device = device
+
         self.node = node
         self.cell = cell
-        self.localEdge = bm.tensor([(1, 2), (2, 0), (0, 1)], **kwargs)
-        self.localFace = bm.tensor([(1, 2), (2, 0), (0, 1)], **kwargs)
-        self.ccw = bm.tensor([0, 1, 2], **kwargs)
+        self.localEdge = bm.tensor([(1, 2), (2, 0), (0, 1)], **kwargs, device=self.device)
+        self.localFace = bm.tensor([(1, 2), (2, 0), (0, 1)], **kwargs, device=self.device)
+        self.ccw = bm.tensor([0, 1, 2], **kwargsi, device=self.device)
 
         self.localCell = bm.tensor([
             (0, 1, 2),
             (1, 2, 0),
-            (2, 0, 1)], **kwargs)
+            (2, 0, 1)], **kwargs, device=self.device)
 
         self.construct()
 
@@ -1108,10 +1110,12 @@ def from_box(cls, box=[0, 1, 0, 1], nx=10, ny=10, threshold=None):
         @param threshold Optional function to filter cells based on their barycenter coordinates (default: None)
         @return TriangleMesh instance
         """
+        device = self.device
+
         NN = (nx + 1) * (ny + 1)
         NC = nx * ny
-        x = bm.linspace(box[0], box[1], nx+1, dtype=bm.float64)
-        y = bm.linspace(box[2], box[3], ny+1, dtype=bm.float64)
+        x = bm.linspace(box[0], box[1], nx+1, dtype=bm.float64i, device=device)
+        y = bm.linspace(box[2], box[3], ny+1, dtype=bm.float64, device=device)
         X, Y = bm.meshgrid(x, y, indexing='ij')
 
         node = bm.concatenate((X.reshape(-1, 1), Y.reshape(-1, 1)), axis=1)