Lines Matching refs:sw

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()
125 PetscCall(DMSwarmRestoreField(sw, "velocity", NULL, NULL, (void **)&velocity)); in SetInitialCoordinates()
126 PetscCall(DMSwarmRestoreField(sw, "w_q", NULL, NULL, (void **)&vals)); in SetInitialCoordinates()
162 static PetscErrorCode CreateParticles(DM dm, DM *sw, AppCtx *user) in CreateParticles() argument
172 PetscCall(DMCreate(PetscObjectComm((PetscObject)dm), sw)); in CreateParticles()
173 PetscCall(DMSetType(*sw, DMSWARM)); in CreateParticles()
174 PetscCall(DMSetDimension(*sw, dim)); in CreateParticles()
181 PetscCall(DMSwarmSetType(*sw, DMSWARM_PIC)); in CreateParticles()
182 PetscCall(DMSwarmSetCellDM(*sw, dm)); in CreateParticles()
183 PetscCall(DMSwarmRegisterPetscDatatypeField(*sw, "velocity", dim, PETSC_REAL)); in CreateParticles()
184 PetscCall(DMSwarmRegisterPetscDatatypeField(*sw, "w_q", 1, PETSC_SCALAR)); in CreateParticles()
185 PetscCall(DMSwarmFinalizeFieldRegister(*sw)); in CreateParticles()
186 PetscCall(DMSwarmGetCellDMActive(*sw, &celldm)); in CreateParticles()
189 PetscCall(DMSwarmSetLocalSizes(*sw, (cEnd - cStart) * Np, 0)); in CreateParticles()
190 PetscCall(DMSetFromOptions(*sw)); in CreateParticles()
191 PetscCall(DMSwarmGetField(*sw, cellid, NULL, NULL, (void **)&swarm_cellid)); in CreateParticles()
198 PetscCall(DMSwarmRestoreField(*sw, cellid, NULL, NULL, (void **)&swarm_cellid)); in CreateParticles()
199 PetscCall(PetscObjectSetName((PetscObject)*sw, "Particles")); in CreateParticles()
200 PetscCall(DMViewFromOptions(*sw, NULL, "-sw_view")); in CreateParticles()
335 DM sw; /* Particles */ in RHSFunctionParticles() local
344 PetscCall(TSGetDM(ts, &sw)); in RHSFunctionParticles()
345 PetscCall(DMGetDimension(sw, &dim)); in RHSFunctionParticles()
394 DM sw; in UpdateSwarm() local
398 PetscCall(TSGetDM(ts, &sw)); in UpdateSwarm()
399 PetscCall(DMSwarmGetField(sw, "velocity", NULL, NULL, (void **)&velocity)); in UpdateSwarm()
405 PetscCall(DMSwarmRestoreField(sw, "velocity", NULL, NULL, (void **)&velocity)); in UpdateSwarm()
425 DM dm, sw; /* Velocity space mesh and Particle Swarm */ in main() local
436 PetscCall(CreateParticles(dm, &sw, &user)); in main()
437 PetscCall(DMSetApplicationContext(sw, &user)); in main()
438 PetscCall(DMSwarmVectorDefineField(sw, "velocity")); in main()
440 PetscCall(TSSetDM(ts, sw)); in main()
448 PetscCall(DMSwarmCreateGlobalVectorFromField(sw, "velocity", &v)); in main()
450 PetscCall(DMSwarmDestroyGlobalVectorFromField(sw, "velocity", &v)); in main()
456 PetscCall(DMDestroy(&sw)); in main()