outline changes

src/map.cpp

@@ -19,11 +19,24 @@ Map::Map(const Image& mask, const floatvector& node_spacing)
   calculate_node_locs();
   calculate_basis();
   // TODO: mask basis
+  apply_mask_to_basis();
   // initialize displacement storage
   alloc_displacements();
   calculate_laplacian();
 }
 
+void Map::apply_mask_to_basis()
+{
+  // Need to apply mask to both image and map "side" of basis
+  // Mask applies directly to image side of basis
+  apply_image_mask();
+
+  // Work out from basis what nodes are relevant, mask out others
+
+  calculate_map_mask();
+  apply_map_mask();
+}
+
 void Map::update(const Vec& delta_vec)
 {
   PetscErrorCode perr = VecAXPY(*m_displacements, 1, delta_vec);

src/map.hpp

@@ -84,6 +84,7 @@ class Map {
   Vec_unique m_displacements;
   mutable DM_unique map_dmda;
 
+  void apply_mask_to_basis();
   void alloc_displacements();
   void initialize_dmda() const;
   void calculate_node_locs();