| /petsc/src/dm/impls/da/tests/ |
| H A D | ex1.c | 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()
[all …]
|
| /petsc/src/dm/impls/swarm/tests/ |
| H A D | ex8.c | 50 static PetscErrorCode CreateSwarm(DM dm, AppCtx *user, DM *sw) in CreateSwarm() argument
56 PetscCall(DMCreate(PetscObjectComm((PetscObject)dm), sw)); in CreateSwarm()
57 PetscCall(DMSetType(*sw, DMSWARM)); in CreateSwarm()
58 PetscCall(DMSetDimension(*sw, dim)); in CreateSwarm()
59 PetscCall(DMSwarmSetType(*sw, DMSWARM_PIC)); in CreateSwarm()
60 PetscCall(DMSwarmSetCellDM(*sw, dm)); in CreateSwarm()
61 PetscCall(DMSwarmRegisterPetscDatatypeField(*sw, "w_q", 1, PETSC_SCALAR)); in CreateSwarm()
62 PetscCall(DMSwarmRegisterPetscDatatypeField(*sw, "velocity", dim, PETSC_REAL)); in CreateSwarm()
63 PetscCall(DMSwarmRegisterPetscDatatypeField(*sw, "species", 1, PETSC_INT)); in CreateSwarm()
64 PetscCall(DMSwarmFinalizeFieldRegister(*sw)); in CreateSwarm()
[all …]
|
| H A D | ex1.c | 50 static PetscErrorCode CreateSwarm(DM dm, DM *sw, AppCtx *user) in CreateSwarm() argument
65 PetscCall(DMCreate(PETSC_COMM_WORLD, sw)); in CreateSwarm()
67 PetscCall(DMSetType(*sw, DMSWARM)); in CreateSwarm()
68 PetscCall(DMSetDimension(*sw, dim)); in CreateSwarm()
69 PetscCall(DMSwarmSetType(*sw, DMSWARM_PIC)); in CreateSwarm()
70 PetscCall(DMSwarmSetCellDM(*sw, dm)); in CreateSwarm()
71 PetscCall(DMSwarmFinalizeFieldRegister(*sw)); in CreateSwarm()
72 PetscCall(DMSwarmSetLocalSizes(*sw, rank == 0 ? particleInitSize : 0, 0)); in CreateSwarm()
73 PetscCall(DMSetFromOptions(*sw)); in CreateSwarm()
74 PetscCall(DMSwarmGetCellDMActive(*sw, &celldm)); in CreateSwarm()
[all …]
|
| H A D | ex5.c | 51 static PetscErrorCode CreateSwarm(DM dm, AppCtx *user, DM *sw) in CreateSwarm() argument
58 PetscCall(DMCreate(PetscObjectComm((PetscObject)dm), sw)); in CreateSwarm()
59 PetscCall(DMSetType(*sw, DMSWARM)); in CreateSwarm()
60 PetscCall(DMSetDimension(*sw, dim)); in CreateSwarm()
61 PetscCall(DMSwarmSetType(*sw, DMSWARM_PIC)); in CreateSwarm()
62 PetscCall(DMSwarmSetCellDM(*sw, dm)); in CreateSwarm()
63 PetscCall(DMSwarmRegisterPetscDatatypeField(*sw, "w_q", 1, PETSC_SCALAR)); in CreateSwarm()
64 PetscCall(DMSwarmRegisterPetscDatatypeField(*sw, "velocity", dim, PETSC_REAL)); in CreateSwarm()
65 PetscCall(DMSwarmRegisterPetscDatatypeField(*sw, "species", 1, PETSC_INT)); in CreateSwarm()
66 PetscCall(DMSwarmRegisterPetscDatatypeField(*sw, "initCoordinates", dim, PETSC_REAL)); in CreateSwarm()
[all …]
|
| H A D | ex3.c | 32 static PetscErrorCode CreateParticles(DM dm, DM *sw, AppCtx *user) in CreateParticles() argument
41 PetscCall(DMCreate(PetscObjectComm((PetscObject)dm), sw)); in CreateParticles()
42 PetscCall(DMSetType(*sw, DMSWARM)); in CreateParticles()
43 PetscCall(DMSetDimension(*sw, dim)); in CreateParticles()
44 PetscCall(DMSwarmSetType(*sw, DMSWARM_PIC)); in CreateParticles()
45 PetscCall(DMSwarmSetCellDM(*sw, dm)); in CreateParticles()
46 PetscCall(DMSwarmRegisterPetscDatatypeField(*sw, "kinematics", 2, PETSC_REAL)); in CreateParticles()
47 PetscCall(DMSwarmFinalizeFieldRegister(*sw)); in CreateParticles()
48 PetscCall(DMSwarmGetCellDMActive(*sw, &celldm)); in CreateParticles()
51 PetscCall(DMSwarmSetLocalSizes(*sw, (cEnd - cStart) * Np, 0)); in CreateParticles()
[all …]
|
| H A D | ex6.c | 184 static PetscErrorCode CreateSwarm(DM dm, AppCtx *user, DM *sw) in CreateSwarm() argument
191 PetscCall(DMCreate(PetscObjectComm((PetscObject)dm), sw)); in CreateSwarm()
192 PetscCall(DMSetType(*sw, DMSWARM)); in CreateSwarm()
193 PetscCall(DMSetDimension(*sw, dim)); in CreateSwarm()
194 PetscCall(DMSwarmSetType(*sw, DMSWARM_PIC)); in CreateSwarm()
195 PetscCall(DMSwarmSetCellDM(*sw, dm)); in CreateSwarm()
196 PetscCall(DMSwarmRegisterPetscDatatypeField(*sw, "w_q", 1, PETSC_SCALAR)); in CreateSwarm()
197 PetscCall(DMSwarmRegisterPetscDatatypeField(*sw, "velocity", dim, PETSC_REAL)); in CreateSwarm()
198 PetscCall(DMSwarmRegisterPetscDatatypeField(*sw, "species", 1, PETSC_INT)); in CreateSwarm()
199 PetscCall(DMSwarmRegisterPetscDatatypeField(*sw, "initCoordinates", dim, PETSC_REAL)); in CreateSwarm()
[all …]
|
| H A D | ex4.c | 89 static PetscErrorCode CreateSwarm(DM dm, AppCtx *user, DM *sw) in CreateSwarm() argument
96 PetscCall(DMCreate(PetscObjectComm((PetscObject)dm), sw)); in CreateSwarm()
97 PetscCall(DMSetType(*sw, DMSWARM)); in CreateSwarm()
98 PetscCall(DMSetDimension(*sw, dim)); in CreateSwarm()
99 PetscCall(DMSwarmSetType(*sw, DMSWARM_PIC)); in CreateSwarm()
100 PetscCall(DMSwarmSetCellDM(*sw, dm)); in CreateSwarm()
101 PetscCall(DMSwarmRegisterPetscDatatypeField(*sw, "w_q", 1, PETSC_SCALAR)); in CreateSwarm()
102 PetscCall(DMSwarmRegisterPetscDatatypeField(*sw, "velocity", dim, PETSC_REAL)); in CreateSwarm()
103 PetscCall(DMSwarmRegisterPetscDatatypeField(*sw, "species", 1, PETSC_INT)); in CreateSwarm()
104 PetscCall(DMSwarmRegisterPetscDatatypeField(*sw, "initCoordinates", dim, PETSC_REAL)); in CreateSwarm()
[all …]
|
| H A D | ex11.c | 183 static PetscErrorCode CreateSwarm(DM dm, DM *sw, AppCtx *user) in CreateSwarm() argument
191 PetscCall(DMCreate(PetscObjectComm((PetscObject)dm), sw)); in CreateSwarm()
192 PetscCall(PetscObjectSetName((PetscObject)*sw, "Particles")); in CreateSwarm()
193 PetscCall(DMSetType(*sw, DMSWARM)); in CreateSwarm()
194 PetscCall(DMSetDimension(*sw, dim)); in CreateSwarm()
195 PetscCall(DMSwarmSetType(*sw, DMSWARM_PIC)); in CreateSwarm()
196 PetscCall(DMSwarmSetCellDM(*sw, dm)); in CreateSwarm()
197 PetscCall(DMSwarmRegisterPetscDatatypeField(*sw, "w_q", 1, PETSC_SCALAR)); in CreateSwarm()
198 PetscCall(DMSwarmRegisterPetscDatatypeField(*sw, "x_q", 2, PETSC_SCALAR)); in CreateSwarm()
199 PetscCall(DMSwarmFinalizeFieldRegister(*sw)); in CreateSwarm()
[all …]
|
| H A D | ex2.c | 170 static PetscErrorCode CreateParticles(DM dm, DM *sw, AppCtx *user) in CreateParticles() argument
188 PetscCall(DMCreate(PetscObjectComm((PetscObject)dm), sw)); in CreateParticles()
189 PetscCall(DMSetType(*sw, DMSWARM)); in CreateParticles()
190 PetscCall(DMSetDimension(*sw, dim)); in CreateParticles()
199 PetscCall(DMSwarmSetType(*sw, DMSWARM_PIC)); in CreateParticles()
200 PetscCall(DMSwarmSetCellDM(*sw, dm)); in CreateParticles()
201 PetscCall(DMSwarmRegisterPetscDatatypeField(*sw, "w_q", 1, PETSC_SCALAR)); in CreateParticles()
202 PetscCall(DMSwarmFinalizeFieldRegister(*sw)); in CreateParticles()
204 PetscCall(DMSwarmSetLocalSizes(*sw, Ncell * Np, 0)); in CreateParticles()
205 PetscCall(DMSetFromOptions(*sw)); in CreateParticles()
[all …]
|
| H A D | ex7.c | 43 PetscErrorCode createSwarm(const DM dm, DM *sw) in createSwarm() argument
48 PetscCall(DMCreate(PETSC_COMM_SELF, sw)); in createSwarm()
49 PetscCall(DMSetType(*sw, DMSWARM)); in createSwarm()
50 PetscCall(DMSetDimension(*sw, dim)); in createSwarm()
51 PetscCall(DMSwarmSetType(*sw, DMSWARM_PIC)); in createSwarm()
52 PetscCall(DMSwarmSetCellDM(*sw, dm)); in createSwarm()
53 PetscCall(DMSwarmRegisterPetscDatatypeField(*sw, "w_q", Nc, PETSC_SCALAR)); in createSwarm()
54 PetscCall(DMSwarmFinalizeFieldRegister(*sw)); in createSwarm()
55 PetscCall(DMSetFromOptions(*sw)); in createSwarm()
59 PetscErrorCode gridToParticles(const DM dm, DM sw, PetscReal *moments, Vec rhs, Mat M_p) in gridToParticles() argument
[all …]
|
| H A D | ex12.c | 68 static PetscErrorCode InitializeParticles(DM sw, UserContext *user) in InitializeParticles() argument
75 PetscCall(DMSwarmGetCellDM(sw, &da)); in InitializeParticles()
80 PetscCall(DMSwarmSetPointsUniformCoordinates(sw, gmin, gmax, ndir, INSERT_VALUES)); in InitializeParticles()
86 DM dm, sw; in main() local
93 PetscCall(CreateSwarm(dm, &sw, &user)); in main()
95 PetscCall(InitializeParticles(sw, &user)); in main()
96 PetscCall(DMSwarmMigrate(sw, PETSC_TRUE)); in main()
97 PetscCall(DMViewFromOptions(sw, NULL, "-sw_view")); in main()
100 PetscCall(DMDestroy(&sw)); in main()
|
| /petsc/src/dm/impls/swarm/tutorials/ |
| H A D | ex1.c | 10 DM dm, sw; in main() local
50 PetscCall(DMCreate(PETSC_COMM_SELF, &sw)); in main()
51 PetscCall(DMSetType(sw, DMSWARM)); in main()
52 PetscCall(DMSetDimension(sw, dim)); in main()
53 PetscCall(DMSwarmSetType(sw, DMSWARM_PIC)); in main()
54 PetscCall(DMSwarmSetCellDM(sw, dm)); in main()
55 PetscCall(DMSwarmRegisterPetscDatatypeField(sw, "w_q", Nc, PETSC_SCALAR)); in main()
56 PetscCall(DMSwarmFinalizeFieldRegister(sw)); in main()
57 PetscCall(DMSwarmSetLocalSizes(sw, Np, zero)); in main()
58 PetscCall(DMSetFromOptions(sw)); in main()
[all …]
|
| H A D | ex1f90.F90 | 11 DM :: dm, sw
39 PetscCallA(DMCreate(PETSC_COMM_WORLD, sw, ierr))
40 PetscCallA(DMSetType(sw, DMSWARM, ierr))
41 PetscCallA(DMSetDimension(sw, dim, ierr))
42 PetscCallA(DMSwarmSetType(sw, DMSWARM_PIC, ierr))
43 PetscCallA(DMSwarmSetCellDM(sw, dm, ierr))
44 PetscCallA(DMSwarmRegisterPetscDatatypeField(sw, 'w_q', Nc, PETSC_SCALAR, ierr))
45 PetscCallA(DMSwarmFinalizeFieldRegister(sw, ierr))
46 PetscCallA(DMSwarmSetLocalSizes(sw, Np, zero, ierr))
47 PetscCallA(DMSetFromOptions(sw, ierr))
[all …]
|
| /petsc/src/ts/tutorials/hamiltonian/ |
| H A D | ex3.c | 148 static PetscErrorCode SetupContext(DM dm, DM sw, AppCtx *user) in SetupContext() argument
180 static PetscErrorCode CheckNonNegativeWeights(DM sw, AppCtx *user) in CheckNonNegativeWeights() argument
187 PetscCall(DMSwarmGetField(sw, "w_q", NULL, NULL, (void **)&w)); in CheckNonNegativeWeights()
188 PetscCall(DMSwarmGetLocalSize(sw, &Np)); in CheckNonNegativeWeights()
190 PetscCall(DMSwarmRestoreField(sw, "w_q", NULL, NULL, (void **)&w)); in CheckNonNegativeWeights()
225 DM sw; in MonitorEField() local
236 PetscCall(TSGetDM(ts, &sw)); in MonitorEField()
237 PetscCall(DMGetDimension(sw, &dim)); in MonitorEField()
238 PetscCall(DMSwarmGetLocalSize(sw, &Np)); in MonitorEField()
239 PetscCall(DMSwarmGetSize(sw, &gNp)); in MonitorEField()
[all …]
|
| H A D | ex2.c | 244 static PetscErrorCode SetupContext(DM dm, DM sw, AppCtx *ctx) in SetupContext() argument
270 PetscCall(DMGetDimension(sw, &dim)); in SetupContext()
306 PetscCall(DMSwarmGetCellDMByName(sw, "velocity", &celldm)); in SetupContext()
417 static PetscErrorCode CheckNonNegativeWeights(DM sw, AppCtx *ctx) in CheckNonNegativeWeights() argument
424 PetscCall(DMSwarmGetField(sw, "w_q", NULL, NULL, (void **)&w)); in CheckNonNegativeWeights()
425 PetscCall(DMSwarmGetLocalSize(sw, &Np)); in CheckNonNegativeWeights()
427 PetscCall(DMSwarmRestoreField(sw, "w_q", NULL, NULL, (void **)&w)); in CheckNonNegativeWeights()
453 static PetscErrorCode computeVelocityFEMMoments(DM sw, PetscReal moments[], AppCtx *user) in computeVelocityFEMMoments() argument
461 PetscCall(DMSwarmSetCellDMActive(sw, "velocity")); in computeVelocityFEMMoments()
462 PetscCall(DMSwarmGetCellDMActive(sw, &celldm)); in computeVelocityFEMMoments()
[all …]
|
| H A D | ex4.c | 325 static PetscErrorCode SetupContext(DM dm, DM sw, AppCtx *ctx) in SetupContext() argument
355 PetscCall(DMGetDimension(sw, &dim)); in SetupContext()
391 PetscCall(DMSwarmGetCellDMByName(sw, "velocity", &celldm)); in SetupContext()
579 static PetscErrorCode CheckNonNegativeWeights(DM sw, AppCtx *ctx) in CheckNonNegativeWeights() argument
586 PetscCall(DMSwarmGetField(sw, "w_q", NULL, NULL, (void **)&w)); in CheckNonNegativeWeights()
587 PetscCall(DMSwarmGetLocalSize(sw, &Np)); in CheckNonNegativeWeights()
589 PetscCall(DMSwarmRestoreField(sw, "w_q", NULL, NULL, (void **)&w)); in CheckNonNegativeWeights()
615 static PetscErrorCode computeVelocityFEMMoments(DM sw, PetscReal moments[], AppCtx *ctx) in computeVelocityFEMMoments() argument
623 PetscCall(DMSwarmSetCellDMActive(sw, "velocity")); in computeVelocityFEMMoments()
624 PetscCall(DMSwarmGetCellDMActive(sw, &celldm)); in computeVelocityFEMMoments()
[all …]
|
| /petsc/src/dm/impls/swarm/ |
| H A D | swarmpic.c | 258 PetscErrorCode DMSwarmCellDMGetBlockSize(DMSwarmCellDM celldm, DM sw, PetscInt *bs) in DMSwarmCellDMGetBlockSize() argument
262 PetscValidHeaderSpecific(sw, DM_CLASSID, 2); in DMSwarmCellDMGetBlockSize()
268 PetscCall(DMSwarmGetFieldInfo(sw, celldm->dmFields[f], &fbs, NULL)); in DMSwarmCellDMGetBlockSize()
344 PetscErrorCode DMSwarmSetPointsUniformCoordinates(DM sw, PetscReal min[], PetscReal max[], PetscInt… in DMSwarmSetPointsUniformCoordinates() argument
363 DMSWARMPICVALID(sw); in DMSwarmSetPointsUniformCoordinates()
364 PetscCall(DMSwarmGetCellDMActive(sw, &celldm)); in DMSwarmSetPointsUniformCoordinates()
366 …PetscCheck(Nfc == 1, PetscObjectComm((PetscObject)sw), PETSC_ERR_SUP, "We only support a single co… in DMSwarmSetPointsUniformCoordinates()
445 PetscCall(DMSwarmGetLocalSize(sw, &n_curr)); in DMSwarmSetPointsUniformCoordinates()
447 PetscCall(DMSwarmSetLocalSizes(sw, n_new_est, -1)); in DMSwarmSetPointsUniformCoordinates()
452 PetscCall(DMSwarmSetLocalSizes(sw, n_new_est, -1)); in DMSwarmSetPointsUniformCoordinates()
[all …]
|
| H A D | swarm.c | 117 PetscErrorCode DMSwarmVectorGetField(DM sw, PetscInt *Nf, const char **fieldnames[]) in DMSwarmVectorGetField() argument
122 PetscValidHeaderSpecificType(sw, DM_CLASSID, 1, DMSWARM); in DMSwarmVectorGetField()
123 PetscCall(DMSwarmGetCellDMActive(sw, &celldm)); in DMSwarmVectorGetField()
176 PetscErrorCode DMSwarmVectorDefineFields(DM sw, PetscInt Nf, const char *fieldnames[]) in DMSwarmVectorDefineFields() argument
178 DM_Swarm *swarm = (DM_Swarm *)sw->data; in DMSwarmVectorDefineFields()
182 PetscValidHeaderSpecificType(sw, DM_CLASSID, 1, DMSWARM); in DMSwarmVectorDefineFields()
184 if (!swarm->issetup) PetscCall(DMSetUp(sw)); in DMSwarmVectorDefineFields()
185 …PetscCheck(Nf >= 0, PetscObjectComm((PetscObject)sw), PETSC_ERR_ARG_OUTOFRANGE, "Number of fields … in DMSwarmVectorDefineFields()
191 PetscCall(DMCreate(PetscObjectComm((PetscObject)sw), &dm)); in DMSwarmVectorDefineFields()
196 PetscCall(DMSwarmAddCellDM(sw, celldm)); in DMSwarmVectorDefineFields()
[all …]
|
| H A D | swarmpic_sort.c | 130 PetscErrorCode DMSwarmSortGetNumberOfPointsPerCell(DM sw, PetscInt cell, PetscInt *npoints) in DMSwarmSortGetNumberOfPointsPerCell() argument
136 PetscCall(DMSwarmGetCellDMActive(sw, &celldm)); in DMSwarmSortGetNumberOfPointsPerCell()
138 …PetscCheck(ctx, PetscObjectComm((PetscObject)sw), PETSC_ERR_USER, "The DMSwarmSort context has not… in DMSwarmSortGetNumberOfPointsPerCell()
164 PetscErrorCode DMSwarmSortGetPointsPerCell(DM sw, PetscInt cell, PetscInt *npoints, PetscInt **pidl… in DMSwarmSortGetPointsPerCell() argument
171 PetscCall(DMSwarmGetCellDMActive(sw, &celldm)); in DMSwarmSortGetPointsPerCell()
173 …PetscCheck(ctx, PetscObjectComm((PetscObject)sw), PETSC_ERR_USER, "The DMSwarmSort context has not… in DMSwarmSortGetPointsPerCell()
174 PetscCall(DMSwarmSortGetNumberOfPointsPerCell(sw, cell, npoints)); in DMSwarmSortGetPointsPerCell()
175 PetscCall(DMGetWorkArray(sw, *npoints, MPIU_SCALAR, pidlist)); in DMSwarmSortGetPointsPerCell()
248 PetscErrorCode DMSwarmSortGetAccess(DM sw) in DMSwarmSortGetAccess() argument
257 PetscCall(DMSwarmGetCellDMActive(sw, &celldm)); in DMSwarmSortGetAccess()
[all …]
|
| /petsc/src/ksp/ksp/utils/dm/ |
| H A D | dmproject.c | 425 static PetscErrorCode DMSwarmProjectField_Conservative_PLEX(DM sw, DM dm, Vec u_p, Vec u_f) in DMSwarmProjectField_Conservative_PLEX() argument
434 PetscCall(DMCreateMassMatrix(sw, dm, &M_p)); in DMSwarmProjectField_Conservative_PLEX()
438 PetscCall(KSPCreate(PetscObjectComm((PetscObject)sw), &ksp)); in DMSwarmProjectField_Conservative_PLEX()
439 PetscCall(PetscObjectGetOptionsPrefix((PetscObject)sw, &prefix)); in DMSwarmProjectField_Conservative_PLEX()
457 static PetscErrorCode DMSwarmProjectParticles_Conservative_PLEX(DM sw, DM dm, Vec u_p, Vec u_f) in DMSwarmProjectParticles_Conservative_PLEX() argument
468 PetscCall(DMCreateMassMatrix(sw, dm, &M_p)); in DMSwarmProjectParticles_Conservative_PLEX()
472 PetscCall(KSPCreate(PetscObjectComm((PetscObject)sw), &ksp)); in DMSwarmProjectParticles_Conservative_PLEX()
473 PetscCall(PetscObjectGetOptionsPrefix((PetscObject)sw, &prefix)); in DMSwarmProjectParticles_Conservative_PLEX()
481 PetscCall(DMSwarmCreateMassMatrixSquare(sw, dm, &PM_p)); in DMSwarmProjectParticles_Conservative_PLEX()
497 static PetscErrorCode DMSwarmProjectFields_Plex_Internal(DM sw, DM dm, PetscInt Nf, const char *fie… in DMSwarmProjectFields_Plex_Internal() argument
[all …]
|
| /petsc/src/ts/tests/ |
| H A D | ex27.c | 60 static PetscErrorCode SetInitialCoordinates(DM sw) in SetInitialCoordinates() argument
72 PetscCall(PetscRandomCreate(PetscObjectComm((PetscObject)sw), &rnd)); in SetInitialCoordinates()
75 PetscCall(PetscRandomCreate(PetscObjectComm((PetscObject)sw), &rndv)); in SetInitialCoordinates()
78 PetscCall(DMGetApplicationContext(sw, &user)); in SetInitialCoordinates()
80 PetscCall(DMGetDimension(sw, &dim)); in SetInitialCoordinates()
81 PetscCall(DMSwarmGetCellDM(sw, &dm)); in SetInitialCoordinates()
88 PetscCall(DMSwarmGetField(sw, DMSwarmPICField_coor, NULL, NULL, (void **)&coords)); in SetInitialCoordinates()
89 PetscCall(DMSwarmGetField(sw, "velocity", NULL, NULL, (void **)&velocity)); in SetInitialCoordinates()
90 PetscCall(DMSwarmGetField(sw, "w_q", NULL, NULL, (void **)&vals)); in SetInitialCoordinates()
124 PetscCall(DMSwarmRestoreField(sw, DMSwarmPICField_coor, NULL, NULL, (void **)&coords)); in SetInitialCoordinates()
[all …]
|
| H A D | ex28.c | 57 static PetscErrorCode SetInitialCoordinates(DM sw) in SetInitialCoordinates() argument
68 PetscCall(PetscRandomCreate(PetscObjectComm((PetscObject)sw), &rnd)); in SetInitialCoordinates()
72 PetscCall(DMGetApplicationContext(sw, &user)); in SetInitialCoordinates()
74 PetscCall(DMGetDimension(sw, &dim)); in SetInitialCoordinates()
75 PetscCall(DMSwarmGetCellDM(sw, &dm)); in SetInitialCoordinates()
82 PetscCall(DMSwarmGetField(sw, DMSwarmPICField_coor, NULL, NULL, (void **)&coords)); in SetInitialCoordinates()
83 PetscCall(DMSwarmGetField(sw, "w_q", NULL, NULL, (void **)&vals)); in SetInitialCoordinates()
112 PetscCall(DMSwarmRestoreField(sw, DMSwarmPICField_coor, NULL, NULL, (void **)&coords)); in SetInitialCoordinates()
113 PetscCall(DMSwarmRestoreField(sw, "w_q", NULL, NULL, (void **)&vals)); in SetInitialCoordinates()
149 static PetscErrorCode CreateParticles(DM dm, DM *sw, AppCtx *user) in CreateParticles() argument
[all …]
|
| H A D | ex30.c | 69 PetscErrorCode createSwarm(const DM dm, PetscInt dim, DM *sw) in createSwarm() argument
74 PetscCall(DMCreate(PETSC_COMM_SELF, sw)); in createSwarm()
75 PetscCall(DMSetType(*sw, DMSWARM)); in createSwarm()
76 PetscCall(DMSetDimension(*sw, dim)); in createSwarm()
77 PetscCall(DMSwarmSetType(*sw, DMSWARM_PIC)); in createSwarm()
78 PetscCall(DMSwarmSetCellDM(*sw, dm)); in createSwarm()
79 PetscCall(DMSwarmRegisterPetscDatatypeField(*sw, "w_q", Nc, PETSC_REAL)); in createSwarm()
80 PetscCall(DMSwarmFinalizeFieldRegister(*sw)); in createSwarm()
81 PetscCall(DMSetFromOptions(*sw)); in createSwarm()
82 PetscCall(PetscObjectSetName((PetscObject)*sw, "Particle Grid")); in createSwarm()
[all …]
|
| /petsc/src/snes/tests/ |
| H A D | ex15.c | 319 static PetscErrorCode InitializeWeights(DM sw, AppCtx *user) in InitializeWeights() argument
326 PetscCall(DMSwarmGetLocalSize(sw, &Np)); in InitializeWeights()
327 PetscCall(DMSwarmGetField(sw, "w_q", NULL, NULL, (void **)&weight)); in InitializeWeights()
328 PetscCall(DMSwarmSortGetAccess(sw)); in InitializeWeights()
335 PetscCall(DMSwarmSortRestoreAccess(sw)); in InitializeWeights()
336 PetscCall(DMSwarmRestoreField(sw, "w_q", NULL, NULL, (void **)&weight)); in InitializeWeights()
340 static PetscErrorCode CreateSwarm(DM dm, AppCtx *user, DM *sw) in CreateSwarm() argument
346 PetscCall(DMCreate(PetscObjectComm((PetscObject)dm), sw)); in CreateSwarm()
347 PetscCall(DMSetType(*sw, DMSWARM)); in CreateSwarm()
348 PetscCall(DMSetDimension(*sw, dim)); in CreateSwarm()
[all …]
|
| /petsc/src/ksp/pc/impls/bddc/ |
| H A D | bddcnullspace.c | 13 PetscCall(MatMultTranspose(corr_ctx->basis_mat, y, corr_ctx->sw[0])); in PCBDDCNullSpaceCorrPreSolve()
15 PetscCall(MatMult(corr_ctx->inv_smat, corr_ctx->sw[0], corr_ctx->sw[1])); in PCBDDCNullSpaceCorrPreSolve()
17 PetscCall(MatMultTranspose(corr_ctx->inv_smat, corr_ctx->sw[0], corr_ctx->sw[1])); in PCBDDCNullSpaceCorrPreSolve()
19 PetscCall(VecScale(corr_ctx->sw[1], -1.0)); in PCBDDCNullSpaceCorrPreSolve()
20 PetscCall(MatMult(corr_ctx->basis_mat, corr_ctx->sw[1], corr_ctx->fw[0])); in PCBDDCNullSpaceCorrPreSolve()
21 PetscCall(VecScale(corr_ctx->sw[1], -1.0)); in PCBDDCNullSpaceCorrPreSolve()
42 PetscCall(MatMultTranspose(corr_ctx->basis_mat, corr_ctx->fw[0], corr_ctx->sw[0])); in PCBDDCNullSpaceCorrPostSolve()
43 PetscCall(VecScale(corr_ctx->sw[0], -1.0)); in PCBDDCNullSpaceCorrPostSolve()
44 PetscCall(MatMult(corr_ctx->inv_smat, corr_ctx->sw[0], corr_ctx->sw[2])); in PCBDDCNullSpaceCorrPostSolve()
45 PetscCall(MatMultAdd(corr_ctx->basis_mat, corr_ctx->sw[2], x, corr_ctx->fw[0])); in PCBDDCNullSpaceCorrPostSolve()
[all …]
|