Lines Matching refs:sw
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()
67 PetscCall(DMSwarmRegisterPetscDatatypeField(*sw, "initVelocity", dim, PETSC_REAL)); in CreateSwarm()
68 PetscCall(DMSwarmFinalizeFieldRegister(*sw)); in CreateSwarm()
69 PetscCall(DMSwarmComputeLocalSizeFromOptions(*sw)); in CreateSwarm()
70 PetscCall(DMSwarmInitializeCoordinates(*sw)); in CreateSwarm()
71 PetscCall(DMSwarmInitializeVelocitiesFromOptions(*sw, v0)); in CreateSwarm()
72 PetscCall(DMSetFromOptions(*sw)); in CreateSwarm()
73 PetscCall(DMSetApplicationContext(*sw, user)); in CreateSwarm()
74 PetscCall(PetscObjectSetName((PetscObject)*sw, "Particles")); in CreateSwarm()
75 PetscCall(DMViewFromOptions(*sw, NULL, "-sw_view")); in CreateSwarm()
79 PetscCall(DMSwarmCreateGlobalVectorFromField(*sw, DMSwarmPICField_coor, &gc)); in CreateSwarm()
80 PetscCall(DMSwarmCreateGlobalVectorFromField(*sw, "initCoordinates", &gc0)); in CreateSwarm()
82 PetscCall(DMSwarmDestroyGlobalVectorFromField(*sw, DMSwarmPICField_coor, &gc)); in CreateSwarm()
83 PetscCall(DMSwarmDestroyGlobalVectorFromField(*sw, "initCoordinates", &gc0)); in CreateSwarm()
84 PetscCall(DMSwarmCreateGlobalVectorFromField(*sw, "velocity", &gv)); in CreateSwarm()
85 PetscCall(DMSwarmCreateGlobalVectorFromField(*sw, "initVelocity", &gv0)); in CreateSwarm()
87 PetscCall(DMSwarmDestroyGlobalVectorFromField(*sw, "velocity", &gv)); in CreateSwarm()
88 PetscCall(DMSwarmDestroyGlobalVectorFromField(*sw, "initVelocity", &gv0)); in CreateSwarm()
95 DM sw; in RHSFunction() local
102 PetscCall(TSGetDM(ts, &sw)); in RHSFunction()
103 PetscCall(DMGetDimension(sw, &dim)); in RHSFunction()
104 PetscCall(DMSwarmGetField(sw, "initCoordinates", NULL, NULL, (void **)&coords)); in RHSFunction()
105 PetscCall(DMSwarmGetField(sw, "initVelocity", NULL, NULL, (void **)&vel)); in RHSFunction()
120 PetscCall(DMSwarmRestoreField(sw, "initCoordinates", NULL, NULL, (void **)&coords)); in RHSFunction()
121 PetscCall(DMSwarmRestoreField(sw, "initVelocity", NULL, NULL, (void **)&vel)); in RHSFunction()
133 DM sw; in RHSJacobian() local
138 PetscCall(TSGetDM(ts, &sw)); in RHSJacobian()
139 PetscCall(DMGetDimension(sw, &dim)); in RHSJacobian()
142 PetscCall(DMSwarmGetField(sw, "initCoordinates", NULL, NULL, (void **)&coords)); in RHSJacobian()
143 PetscCall(DMSwarmGetField(sw, "initVelocity", NULL, NULL, (void **)&vel)); in RHSJacobian()
156 PetscCall(DMSwarmRestoreField(sw, "initCoordinates", NULL, NULL, (void **)&coords)); in RHSJacobian()
157 PetscCall(DMSwarmRestoreField(sw, "initVelocity", NULL, NULL, (void **)&vel)); in RHSJacobian()
165 DM sw; in RHSFunctionX() local
171 PetscCall(TSGetDM(ts, &sw)); in RHSFunctionX()
172 PetscCall(DMGetDimension(sw, &dim)); in RHSFunctionX()
186 DM sw; in RHSFunctionV() local
193 PetscCall(TSGetDM(ts, &sw)); in RHSFunctionV()
194 PetscCall(DMGetDimension(sw, &dim)); in RHSFunctionV()
195 PetscCall(DMSwarmGetField(sw, "initCoordinates", NULL, NULL, (void **)&coords)); in RHSFunctionV()
196 PetscCall(DMSwarmGetField(sw, "initVelocity", NULL, NULL, (void **)&vel)); in RHSFunctionV()
211 PetscCall(DMSwarmRestoreField(sw, "initCoordinates", NULL, NULL, (void **)&coords)); in RHSFunctionV()
212 PetscCall(DMSwarmRestoreField(sw, "initVelocity", NULL, NULL, (void **)&vel)); in RHSFunctionV()
218 DM sw; in CreateSolution() local
223 PetscCall(TSGetDM(ts, &sw)); in CreateSolution()
224 PetscCall(DMGetDimension(sw, &dim)); in CreateSolution()
225 PetscCall(DMSwarmGetLocalSize(sw, &Np)); in CreateSolution()
238 DM sw; in SetProblem() local
241 PetscCall(TSGetDM(ts, &sw)); in SetProblem()
242 PetscCall(DMGetApplicationContext(sw, &user)); in SetProblem()
248 PetscCall(DMGetDimension(sw, &dim)); in SetProblem()
249 PetscCall(DMSwarmGetLocalSize(sw, &Np)); in SetProblem()
267 PetscCall(DMGetDimension(sw, &dim)); in SetProblem()
268 PetscCall(DMSwarmGetLocalSize(sw, &Np)); in SetProblem()
296 DM sw; in DMSwarmTSRedistribute() local
302 PetscCall(TSGetDM(ts, &sw)); in DMSwarmTSRedistribute()
310 PetscCall(TSSetDM(ts, sw)); in DMSwarmTSRedistribute()
380 DM sw; in InitializeSolveAndSwarm() local
386 PetscCall(TSGetDM(ts, &sw)); in InitializeSolveAndSwarm()
387 PetscCall(DMGetApplicationContext(sw, &user)); in InitializeSolveAndSwarm()
391 PetscCall(DMSwarmInitializeCoordinates(sw)); in InitializeSolveAndSwarm()
392 PetscCall(DMSwarmInitializeVelocitiesFromOptions(sw, v0)); in InitializeSolveAndSwarm()
393 PetscCall(DMSwarmMigrate(sw, PETSC_TRUE)); in InitializeSolveAndSwarm()
399 PetscCall(DMSwarmCreateGlobalVectorFromField(sw, DMSwarmPICField_coor, &gc)); in InitializeSolveAndSwarm()
400 PetscCall(DMSwarmCreateGlobalVectorFromField(sw, "initCoordinates", &gc0)); in InitializeSolveAndSwarm()
403 PetscCall(DMSwarmDestroyGlobalVectorFromField(sw, DMSwarmPICField_coor, &gc)); in InitializeSolveAndSwarm()
404 PetscCall(DMSwarmDestroyGlobalVectorFromField(sw, "initCoordinates", &gc0)); in InitializeSolveAndSwarm()
405 PetscCall(DMSwarmCreateGlobalVectorFromField(sw, "velocity", &gv)); in InitializeSolveAndSwarm()
406 PetscCall(DMSwarmCreateGlobalVectorFromField(sw, "initVelocity", &gv0)); in InitializeSolveAndSwarm()
409 PetscCall(DMSwarmDestroyGlobalVectorFromField(sw, "velocity", &gv)); in InitializeSolveAndSwarm()
410 PetscCall(DMSwarmDestroyGlobalVectorFromField(sw, "initVelocity", &gv0)); in InitializeSolveAndSwarm()
425 DM sw; in ComputeError() local
435 PetscCall(TSGetDM(ts, &sw)); in ComputeError()
436 PetscCall(DMGetApplicationContext(sw, &user)); in ComputeError()
437 PetscCall(DMGetDimension(sw, &dim)); in ComputeError()
442 PetscCall(DMSwarmGetField(sw, "initCoordinates", NULL, NULL, (void **)&coords)); in ComputeError()
443 PetscCall(DMSwarmGetField(sw, "initVelocity", NULL, NULL, (void **)&vel)); in ComputeError()
469 PetscCall(DMSwarmRestoreField(sw, "initCoordinates", NULL, NULL, (void **)&coords)); in ComputeError()
470 PetscCall(DMSwarmRestoreField(sw, "initVelocity", NULL, NULL, (void **)&vel)); in ComputeError()
479 DM sw; in EnergyMonitor() local
485 PetscCall(TSGetDM(ts, &sw)); in EnergyMonitor()
486 PetscCall(DMGetDimension(sw, &dim)); in EnergyMonitor()
504 DM sw; in MigrateParticles() local
507 PetscCall(TSGetDM(ts, &sw)); in MigrateParticles()
508 PetscCall(DMViewFromOptions(sw, NULL, "-migrate_view_pre")); in MigrateParticles()
516 PetscCall(DMSwarmCreateGlobalVectorFromField(sw, DMSwarmPICField_coor, &gc)); in MigrateParticles()
518 PetscCall(DMSwarmDestroyGlobalVectorFromField(sw, DMSwarmPICField_coor, &gc)); in MigrateParticles()
519 PetscCall(DMSwarmCreateGlobalVectorFromField(sw, "velocity", &gv)); in MigrateParticles()
521 PetscCall(DMSwarmDestroyGlobalVectorFromField(sw, "velocity", &gv)); in MigrateParticles()
523 PetscCall(DMSwarmMigrate(sw, PETSC_TRUE)); in MigrateParticles()
531 DM dm, sw; in main() local
540 PetscCall(CreateSwarm(dm, &user, &sw)); in main()
541 PetscCall(DMSetApplicationContext(sw, &user)); in main()
545 PetscCall(TSSetDM(ts, sw)); in main()
562 PetscCall(DMDestroy(&sw)); in main()