Skip to content

Commit

Permalink
StateParamIntegrandWithMeasure -> StateParamMap
Browse files Browse the repository at this point in the history
  • Loading branch information
zjwegert committed Dec 17, 2024
1 parent efc9d68 commit fc811cd
Show file tree
Hide file tree
Showing 19 changed files with 80 additions and 77 deletions.
6 changes: 3 additions & 3 deletions docs/src/reference/chainrules.md
Original file line number Diff line number Diff line change
Expand Up @@ -10,11 +10,11 @@ GridapTopOpt.evaluate_derivatives!
GridapTopOpt.get_state
```

## `StateParamIntegrandWithMeasure`
## `StateParamMap`

```@docs
GridapTopOpt.StateParamIntegrandWithMeasure
GridapTopOpt.rrule(u_to_j::GridapTopOpt.StateParamIntegrandWithMeasure,uh,φh)
GridapTopOpt.StateParamMap
GridapTopOpt.rrule(u_to_j::GridapTopOpt.StateParamMap,uh,φh)
```

## Implemented types of `AbstractFEStateMap`
Expand Down
6 changes: 3 additions & 3 deletions scripts/Embedded/Examples/2d_compliance_L.jl
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@ using Gridap,GridapTopOpt, GridapSolvers
using Gridap.Adaptivity, Gridap.Geometry
using GridapEmbedded, GridapEmbedded.LevelSetCutters

using GridapTopOpt: StateParamIntegrandWithMeasure
using GridapTopOpt: StateParamMap

path="./results/UnfittedFEM_elastic_compliance_LShape_ALM/"
rm(path,force=true,recursive=true)
Expand Down Expand Up @@ -108,8 +108,8 @@ state_collection = EmbeddedCollection(model,φh) do _,_
state_map = AffineFEStateMap(a,l,U,V,V_φ,U_reg,φh;ls,adjoint_ls=ls)
(;
:state_map => state_map,
:J => StateParamIntegrandWithMeasure(J,state_map),
:C => map(Ci -> StateParamIntegrandWithMeasure(Ci,state_map),[Vol,])
:J => StateParamMap(J,state_map),
:C => map(Ci -> StateParamMap(Ci,state_map),[Vol,])
)
end
pcfs = EmbeddedPDEConstrainedFunctionals(state_collection;analytic_dC=(dVol,))
Expand Down
2 changes: 1 addition & 1 deletion scripts/Embedded/Examples/2d_fluid_structure_(stokes).jl
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@ using Gridap,GridapTopOpt, GridapSolvers
using Gridap.Adaptivity, Gridap.Geometry
using GridapEmbedded, GridapEmbedded.LevelSetCutters

using GridapTopOpt: StateParamIntegrandWithMeasure
using GridapTopOpt: StateParamMap

path = "./results/UnfittedFEM_fluid_structure_(stokes)_ALM/"
rm(path,force=true,recursive=true)
Expand Down
6 changes: 3 additions & 3 deletions scripts/Embedded/Examples/2d_thermal.jl
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@ using Gridap,GridapTopOpt, GridapSolvers
using Gridap.Adaptivity, Gridap.Geometry
using GridapEmbedded, GridapEmbedded.LevelSetCutters

using GridapTopOpt: StateParamIntegrandWithMeasure
using GridapTopOpt: StateParamMap

path="./results/UnfittedFEM_thermal_compliance_ALM/"
rm(path,force=true,recursive=true)
Expand Down Expand Up @@ -79,8 +79,8 @@ state_collection = EmbeddedCollection(model,φh) do _,_
state_map = AffineFEStateMap(a,l,U,V,V_φ,U_reg,φh;ls,adjoint_ls=ls)
(;
:state_map => state_map,
:J => StateParamIntegrandWithMeasure(J,state_map),
:C => map(Ci -> StateParamIntegrandWithMeasure(Ci,state_map),[Vol,])
:J => StateParamMap(J,state_map),
:C => map(Ci -> StateParamMap(Ci,state_map),[Vol,])
)
end
pcfs = EmbeddedPDEConstrainedFunctionals(state_collection;analytic_dC=(dVol,))
Expand Down
6 changes: 3 additions & 3 deletions scripts/Embedded/Examples/2d_thermal_AgFEM.jl
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@ using Gridap,GridapTopOpt, GridapSolvers
using Gridap.Adaptivity, Gridap.Geometry
using GridapEmbedded, GridapEmbedded.LevelSetCutters

using GridapTopOpt: StateParamIntegrandWithMeasure
using GridapTopOpt: StateParamMap

path="./results/AgFEM_thermal_compliance_ALM/"
rm(path,force=true,recursive=true)
Expand Down Expand Up @@ -82,8 +82,8 @@ state_collection = EmbeddedCollection(model,φh) do cutgeo,_
state_map = AffineFEStateMap(a,l,U,V,V_φ,U_reg,φh;ls,adjoint_ls=ls)
(;
:state_map => state_map,
:J => StateParamIntegrandWithMeasure(J,state_map),
:C => map(Ci -> StateParamIntegrandWithMeasure(Ci,state_map),[Vol,])
:J => StateParamMap(J,state_map),
:C => map(Ci -> StateParamMap(Ci,state_map),[Vol,])
)
end
pcfs = EmbeddedPDEConstrainedFunctionals(state_collection;analytic_dC=(dVol,))
Expand Down
6 changes: 3 additions & 3 deletions scripts/Embedded/Examples/2d_thermal_L.jl
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@ using Gridap,GridapTopOpt, GridapSolvers
using Gridap.Adaptivity, Gridap.Geometry
using GridapEmbedded, GridapEmbedded.LevelSetCutters

using GridapTopOpt: StateParamIntegrandWithMeasure
using GridapTopOpt: StateParamMap

path="./results/UnfittedFEM_thermal_compliance_LShape_ALM/"
rm(path,force=true,recursive=true)
Expand Down Expand Up @@ -87,8 +87,8 @@ state_collection = EmbeddedCollection(model,φh) do _,_
state_map = AffineFEStateMap(a,l,U,V,V_φ,U_reg,φh;ls,adjoint_ls=ls)
(;
:state_map => state_map,
:J => StateParamIntegrandWithMeasure(J,state_map),
:C => map(Ci -> StateParamIntegrandWithMeasure(Ci,state_map),[Vol,])
:J => StateParamMap(J,state_map),
:C => map(Ci -> StateParamMap(Ci,state_map),[Vol,])
)
end
pcfs = EmbeddedPDEConstrainedFunctionals(state_collection;analytic_dC=(dVol,))
Expand Down
6 changes: 3 additions & 3 deletions scripts/Embedded/Examples/2d_thermal_gmsh_example.jl
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@ using Gridap,GridapTopOpt, GridapSolvers
using Gridap.Adaptivity, Gridap.Geometry
using GridapEmbedded, GridapEmbedded.LevelSetCutters

using GridapTopOpt: StateParamIntegrandWithMeasure
using GridapTopOpt: StateParamMap

path="./results/UnfittedFEM_thermal_compliance_gmsh_ALM/"
rm(path,force=true,recursive=true)
Expand Down Expand Up @@ -87,8 +87,8 @@ state_collection = EmbeddedCollection(model,φh) do _,_
state_map = AffineFEStateMap(a,l,U,V,V_φ,U_reg,φh;ls,adjoint_ls=ls)
(;
:state_map => state_map,
:J => StateParamIntegrandWithMeasure(J,state_map),
:C => map(Ci -> StateParamIntegrandWithMeasure(Ci,state_map),[Vol,])
:J => StateParamMap(J,state_map),
:C => map(Ci -> StateParamMap(Ci,state_map),[Vol,])
)
end
pcfs = EmbeddedPDEConstrainedFunctionals(state_collection;analytic_dC=(dVol,))
Expand Down
6 changes: 3 additions & 3 deletions scripts/Embedded/Examples/2d_thermal_weird_background.jl
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@ using Gridap,GridapTopOpt, GridapSolvers
using Gridap.Adaptivity, Gridap.Geometry
using GridapEmbedded, GridapEmbedded.LevelSetCutters

using GridapTopOpt: StateParamIntegrandWithMeasure
using GridapTopOpt: StateParamMap

path="./results/UnfittedFEM_thermal_compliance_weirdBG_ALM/"
rm(path,force=true,recursive=true)
Expand Down Expand Up @@ -87,8 +87,8 @@ state_collection = EmbeddedCollection(model,φh) do _,_
state_map = AffineFEStateMap(a,l,U,V,V_φ,U_reg,φh;ls,adjoint_ls=ls)
(;
:state_map => state_map,
:J => StateParamIntegrandWithMeasure(J,state_map),
:C => map(Ci -> StateParamIntegrandWithMeasure(Ci,state_map),[Vol,])
:J => StateParamMap(J,state_map),
:C => map(Ci -> StateParamMap(Ci,state_map),[Vol,])
)
end
pcfs = EmbeddedPDEConstrainedFunctionals(state_collection;analytic_dC=(dVol,))
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@ using Gridap,GridapTopOpt, GridapSolvers
using Gridap.Adaptivity, Gridap.Geometry
using GridapEmbedded, GridapEmbedded.LevelSetCutters

using GridapTopOpt: StateParamIntegrandWithMeasure
using GridapTopOpt: StateParamMap

path="./results/CutFEM_thermal_compliance_ALM_island_detect/"
rm(path,force=true,recursive=true)
Expand Down Expand Up @@ -79,8 +79,8 @@ state_collection = EmbeddedCollection(model,φh) do _,_
state_map = AffineFEStateMap(a,l,U,V,V_φ,U_reg,φh)
(;
:state_map => state_map,
:J => StateParamIntegrandWithMeasure(J,state_map),
:C => map(Ci -> StateParamIntegrandWithMeasure(Ci,state_map),[Vol,])
:J => StateParamMap(J,state_map),
:C => map(Ci -> StateParamMap(Ci,state_map),[Vol,])
)
end
pcfs = EmbeddedPDEConstrainedFunctionals(state_collection;analytic_dC=(dVol,))
Expand Down
2 changes: 1 addition & 1 deletion scripts/Embedded/Examples/FCM_2d_compliance_L.jl
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@ using Gridap,GridapTopOpt, GridapSolvers
using Gridap.Adaptivity, Gridap.Geometry
using GridapEmbedded, GridapEmbedded.LevelSetCutters

using GridapTopOpt: StateParamIntegrandWithMeasure
using GridapTopOpt: StateParamMap

path="./results/FCM_elastic_compliance_LShape_ALM_n100/"
rm(path,force=true,recursive=true)
Expand Down
2 changes: 1 addition & 1 deletion scripts/Embedded/Examples/FCM_2d_thermal.jl
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@ using Gridap,GridapTopOpt, GridapSolvers
using Gridap.Adaptivity, Gridap.Geometry
using GridapEmbedded, GridapEmbedded.LevelSetCutters

using GridapTopOpt: StateParamIntegrandWithMeasure
using GridapTopOpt: StateParamMap

path="./results/FCM_thermal_compliance_ALM/"
rm(path,force=true,recursive=true)
Expand Down
2 changes: 1 addition & 1 deletion scripts/Embedded/Examples/FCM_2d_thermal_MPI.jl
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@ using Gridap,GridapTopOpt, GridapSolvers
using Gridap.Adaptivity, Gridap.Geometry
using GridapEmbedded, GridapEmbedded.LevelSetCutters

using GridapTopOpt: StateParamIntegrandWithMeasure
using GridapTopOpt: StateParamMap

using GridapDistributed, GridapPETSc, GridapSolvers,
PartitionedArrays, GridapTopOpt, SparseMatricesCSR
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@ using Gridap,GridapTopOpt, GridapSolvers
using Gridap.Adaptivity, Gridap.Geometry
using GridapEmbedded, GridapEmbedded.LevelSetCutters

using GridapTopOpt: StateParamIntegrandWithMeasure
using GridapTopOpt: StateParamMap

path="./results/FCM_thermal_compliance_ALM/"
rm(path,force=true,recursive=true)
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@ using Gridap,GridapTopOpt, GridapSolvers
using Gridap.Adaptivity, Gridap.Geometry
using GridapEmbedded, GridapEmbedded.LevelSetCutters

using GridapTopOpt: StateParamIntegrandWithMeasure
using GridapTopOpt: StateParamMap

using GridapPETSc, SparseMatricesCSR

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@ using Gridap,GridapTopOpt, GridapSolvers
using Gridap.Adaptivity, Gridap.Geometry
using GridapEmbedded, GridapEmbedded.LevelSetCutters

using GridapTopOpt: StateParamIntegrandWithMeasure
using GridapTopOpt: StateParamMap

using GridapPETSc, SparseMatricesCSR

Expand Down Expand Up @@ -66,7 +66,7 @@ function main(n;order=1,γg=0.1)
σ(ε) = λ*tr(ε)*one(ε) + 2*μ*ε

g = VectorValue(0,-1)
a(u,v,φ) = (ε(v) ε(u)))dΩin +
a(u,v,φ) = (ε(v) ε(u)))dΩin +
((γg*h)*jump(nΛ_D(v)) jump(nΛ_D(u)))dΛ_D +
((γg*h^3)*jump(n_Γg(v)) jump(n_Γg(u)))dΓg
l(v,φ) = (vg)dΓ_N
Expand Down
6 changes: 3 additions & 3 deletions scripts/Embedded/Examples/fsi/TO-2-CutFEM-stokes_fsi.jl
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@ using Gridap, Gridap.Geometry, Gridap.Adaptivity
using GridapEmbedded, GridapEmbedded.LevelSetCutters
using GridapTopOpt

using GridapTopOpt: StateParamIntegrandWithMeasure
using GridapTopOpt: StateParamMap

path = "./results/TO-2-CutFEM-stokes_fsi/"
mkpath(path)
Expand Down Expand Up @@ -190,8 +190,8 @@ state_collection = EmbeddedCollection(model,φh) do _,_
state_map = AffineFEStateMap(a_coupled,l_coupled,X,Y,V_φ,U_reg,φh)
(;
:state_map => state_map,
:J => StateParamIntegrandWithMeasure(J_comp,state_map),
:C => map(Ci -> StateParamIntegrandWithMeasure(Ci,state_map),[Vol,])
:J => StateParamMap(J_comp,state_map),
:C => map(Ci -> StateParamMap(Ci,state_map),[Vol,])
)
end
pcfs = EmbeddedPDEConstrainedFunctionals(state_collection)
Expand Down
4 changes: 2 additions & 2 deletions scripts/Embedded/MWEs/zygote/thermal_compliance_ALM copy.jl
Original file line number Diff line number Diff line change
Expand Up @@ -62,8 +62,8 @@ dVol(q,u,φ) = ∫(-1/vol_D*q*(DH ∘ φ)*(norm ∘ ∇(φ)))dΩ

state_map = AffineFEStateMap(a,l,U,V,V_φ,V_φ,φh)

# Create StateParamIntegrandWithMeasures
_J = GridapTopOpt.StateParamIntegrandWithMeasure(J,state_map)
# Create StateParamMaps
_J = GridapTopOpt.StateParamMap(J,state_map)

function φ_to_j(φ)
u = state_map(φ)
Expand Down
4 changes: 2 additions & 2 deletions scripts/Embedded/MWEs/zygote/thermal_compliance_ALM copy_2.jl
Original file line number Diff line number Diff line change
Expand Up @@ -67,8 +67,8 @@ dVol(q,u,φ) = ∫(-1/vol_D*q*(DH ∘ φ)*(norm ∘ ∇(φ)))dΩ

state_map = AffineFEStateMap(a,l,U,V,V_φ,V_φ,φh)

# Create StateParamIntegrandWithMeasures
_J = GridapTopOpt.StateParamIntegrandWithMeasure(J,state_map)
# Create StateParamMaps
_J = GridapTopOpt.StateParamMap(J,state_map)

function φ_to_j(φ)
u = state_map(φ)
Expand Down
Loading

0 comments on commit fc811cd

Please sign in to comment.