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()
105   PetscCall(DMSwarmFinalizeFieldRegister(*sw));  in CreateSwarm()
106   PetscCall(DMSwarmComputeLocalSizeFromOptions(*sw));  in CreateSwarm()
107   PetscCall(DMSwarmInitializeCoordinates(*sw));  in CreateSwarm()
108   PetscCall(DMSwarmInitializeVelocitiesFromOptions(*sw, v0));  in CreateSwarm()
109   PetscCall(DMSetFromOptions(*sw));  in CreateSwarm()
110   PetscCall(DMSetApplicationContext(*sw, user));  in CreateSwarm()
111   PetscCall(PetscObjectSetName((PetscObject)*sw, "Particles"));  in CreateSwarm()
112   PetscCall(DMViewFromOptions(*sw, NULL, "-sw_view"));  in CreateSwarm()
116     PetscCall(DMSwarmCreateGlobalVectorFromField(*sw, DMSwarmPICField_coor, &gc));  in CreateSwarm()
117     PetscCall(DMSwarmCreateGlobalVectorFromField(*sw, "initCoordinates", &gc0));  in CreateSwarm()
119     PetscCall(DMSwarmDestroyGlobalVectorFromField(*sw, DMSwarmPICField_coor, &gc));  in CreateSwarm()
120     PetscCall(DMSwarmDestroyGlobalVectorFromField(*sw, "initCoordinates", &gc0));  in CreateSwarm()
124     PetscCall(DMSwarmVectorDefineFields(*sw, 2, fieldnames));  in CreateSwarm()
132   DM                 sw;  in RHSFunction()  local
138   PetscCall(TSGetDM(ts, &sw));  in RHSFunction()
139   PetscCall(DMGetDimension(sw, &dim));  in RHSFunction()
162   DM          sw;  in RHSJacobian()  local
166   PetscCall(TSGetDM(ts, &sw));  in RHSJacobian()
167   PetscCall(DMGetDimension(sw, &dim));  in RHSJacobian()
218   DM          sw;  in RHSJacobianS()  local
222   PetscCall(TSGetDM(ts, &sw));  in RHSJacobianS()
223   PetscCall(DMGetDimension(sw, &dim));  in RHSJacobianS()
241   DM                 sw;  in RHSObjectiveF()  local
246   PetscCall(TSGetDM(ts, &sw));  in RHSObjectiveF()
247   PetscCall(DMGetDimension(sw, &dim));  in RHSObjectiveF()
266   DM                 sw;  in RHSFunctionG()  local
272   PetscCall(TSGetDM(ts, &sw));  in RHSFunctionG()
273   PetscCall(DMGetDimension(sw, &dim));  in RHSFunctionG()
291   DM       sw;  in CreateSolution()  local
296   PetscCall(TSGetDM(ts, &sw));  in CreateSolution()
297   PetscCall(DMGetDimension(sw, &dim));  in CreateSolution()
298   PetscCall(DMSwarmGetLocalSize(sw, &Np));  in CreateSolution()
311   DM      sw;  in SetProblem()  local
314   PetscCall(TSGetDM(ts, &sw));  in SetProblem()
315   PetscCall(DMGetApplicationContext(sw, &user));  in SetProblem()
321     PetscCall(DMGetDimension(sw, &dim));  in SetProblem()
322     PetscCall(DMSwarmGetLocalSize(sw, &Np));  in SetProblem()
338     PetscCall(DMGetDimension(sw, &dim));  in SetProblem()
339     PetscCall(DMSwarmGetLocalSize(sw, &Np));  in SetProblem()
366   DM      sw;  in InitializeSolve()  local
372   PetscCall(TSGetDM(ts, &sw));  in InitializeSolve()
373   PetscCall(DMGetApplicationContext(sw, &user));  in InitializeSolve()
377     PetscCall(DMSwarmInitializeCoordinates(sw));  in InitializeSolve()
378     PetscCall(DMSwarmInitializeVelocitiesFromOptions(sw, v0));  in InitializeSolve()
383   PetscCall(DMSwarmCreateGlobalVectorFromField(sw, DMSwarmPICField_coor, &gc));  in InitializeSolve()
384   PetscCall(DMSwarmCreateGlobalVectorFromField(sw, "initCoordinates", &gc0));  in InitializeSolve()
387   PetscCall(DMSwarmDestroyGlobalVectorFromField(sw, DMSwarmPICField_coor, &gc));  in InitializeSolve()
388   PetscCall(DMSwarmDestroyGlobalVectorFromField(sw, "initCoordinates", &gc0));  in InitializeSolve()
396   DM                 sw;  in ComputeError()  local
406   PetscCall(TSGetDM(ts, &sw));  in ComputeError()
407   PetscCall(DMGetApplicationContext(sw, &user));  in ComputeError()
408   PetscCall(DMGetDimension(sw, &dim));  in ComputeError()
413   PetscCall(DMSwarmGetField(sw, "initCoordinates", NULL, NULL, (void **)&coords));  in ComputeError()
435   PetscCall(DMSwarmRestoreField(sw, "initCoordinates", NULL, NULL, (void **)&coords));  in ComputeError()
445   DM                 sw;  in EnergyMonitor()  local
454     PetscCall(TSGetDM(ts, &sw));  in EnergyMonitor()
456     PetscCall(DMGetDimension(sw, &dim));  in EnergyMonitor()
476   DM     dm, sw;  in main()  local
485   PetscCall(CreateSwarm(dm, &user, &sw));  in main()
486   PetscCall(DMSetApplicationContext(sw, &user));  in main()
490   PetscCall(TSSetDM(ts, sw));  in main()
507   PetscCall(DMDestroy(&sw));  in main()