Lines Matching refs:sw
6 PetscErrorCode DMSwarmPrint(DM sw) in DMSwarmPrint() argument
16 PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)sw), &rank)); in DMSwarmPrint()
17 PetscCall(DMSwarmGetLocalSize(sw, &Np)); in DMSwarmPrint()
18 PetscCall(DMSwarmGetCellDMActive(sw, &celldm)); in DMSwarmPrint()
21 …PetscCheck(Nfc == 1, PetscObjectComm((PetscObject)sw), PETSC_ERR_SUP, "We only support a single co… in DMSwarmPrint()
22 PetscCall(DMSwarmGetField(sw, coordFields[0], &bs, NULL, (void **)&array)); in DMSwarmPrint()
23 PetscCall(DMSwarmGetField(sw, DMSwarmField_pid, &bs, NULL, (void **)&pidArray)); in DMSwarmPrint()
24 PetscCall(DMSwarmGetField(sw, cellid, &bs, NULL, (void **)&cidArray)); in DMSwarmPrint()
30 PetscCall(DMSwarmRestoreField(sw, coordFields[0], &bs, NULL, (void **)&array)); in DMSwarmPrint()
31 PetscCall(DMSwarmRestoreField(sw, DMSwarmField_pid, &bs, NULL, (void **)&pidArray)); in DMSwarmPrint()
32 PetscCall(DMSwarmRestoreField(sw, cellid, &bs, NULL, (void **)&pidArray)); in DMSwarmPrint()
38 DM dm, sw; in main() local
54 PetscCall(DMCreate(PETSC_COMM_WORLD, &sw)); in main()
55 PetscCall(PetscObjectSetName((PetscObject)sw, "Particle Grid")); in main()
56 PetscCall(DMSetType(sw, DMSWARM)); in main()
57 PetscCall(DMSetDimension(sw, 2)); in main()
58 PetscCall(DMSwarmSetType(sw, DMSWARM_PIC)); in main()
59 PetscCall(DMSetFromOptions(sw)); in main()
60 PetscCall(DMSwarmSetCellDM(sw, dm)); in main()
61 PetscCall(DMSwarmInitializeFieldRegister(sw)); in main()
62 PetscCall(DMSwarmRegisterPetscDatatypeField(sw, "u", 1, PETSC_SCALAR)); in main()
63 PetscCall(DMSwarmFinalizeFieldRegister(sw)); in main()
64 PetscCall(DMSwarmSetLocalSizes(sw, Np, 2)); in main()
65 PetscCall(DMSwarmGetField(sw, "DMSwarmPIC_coor", &bs, &dtype, (void **)&coords)); in main()
74 PetscCall(DMSwarmRestoreField(sw, "DMSwarmPIC_coor", &bs, &dtype, (void **)&coords)); in main()
75 PetscCall(DMViewFromOptions(sw, NULL, "-swarm_view")); in main()
76 PetscCall(DMSwarmPrint(sw)); in main()
79 PetscCall(DMSwarmMigrate(sw, PETSC_TRUE)); in main()
80 PetscCall(DMViewFromOptions(sw, NULL, "-swarm_view")); in main()
81 PetscCall(DMSwarmPrint(sw)); in main()
83 PetscCall(DMDestroy(&sw)); in main()