From 96dd51e484a7f4cf49bc06e38ba774fdf6fb8b81 Mon Sep 17 00:00:00 2001
From: Nicola Demo <demo.nicola@gmail.com>
Date: Tue, 22 Mar 2016 00:32:04 +0100
Subject: [PATCH] Added colorize flag (2dim)

---
 source/grid/grid_generator.cc | 31 ++++++++++++++++++++++++++++---
 1 file changed, 28 insertions(+), 3 deletions(-)

diff --git a/source/grid/grid_generator.cc b/source/grid/grid_generator.cc
index 6ad89aa450..e32577f2a6 100644
--- a/source/grid/grid_generator.cc
+++ b/source/grid/grid_generator.cc
@@ -350,6 +350,8 @@ namespace GridGenerator
         p1(i) = std::min(p_1(i), p_2(i));
         p2(i) = std::max(p_1(i), p_2(i));
       }
+      //for (typename Triangulation<1,spacedim>::cell_iterator cell = tria.begin();
+      //     cell != tria.end(); ++cell)
 
     std::vector<Point<spacedim> > vertices (GeometryInfo<dim>::vertices_per_cell);
     switch (dim)
@@ -1846,7 +1848,8 @@ namespace GridGenerator
   template <>
   void hyper_L (Triangulation<1> &,
                 const double,
-                const double)
+                const double,
+                const bool)
   {
     Assert (false, ExcNotImplemented());
   }
@@ -2080,7 +2083,8 @@ namespace GridGenerator
   void
   hyper_L (Triangulation<2> &tria,
            const double a,
-           const double b)
+           const double b,
+           const bool colorize)
   {
     const Point<2> vertices[8] = { Point<2> (a,a),
                                    Point<2> ((a+b)/2,a),
@@ -2109,6 +2113,26 @@ namespace GridGenerator
       std::vector<Point<2> >(&vertices[0], &vertices[8]),
       cells,
       SubCellData());       // no boundary information
+
+    if (colorize)
+      {
+        auto cell = tria.begin();
+        
+        cell->face(0)->set_boundary_id(0);
+        cell->face(2)->set_boundary_id(1);
+        cell++;
+
+        cell->face(1)->set_boundary_id(2);
+        cell->face(2)->set_boundary_id(1);
+        cell->face(3)->set_boundary_id(3);
+        cell++;
+
+        cell->face(0)->set_boundary_id(0);
+        cell->face(1)->set_boundary_id(4);
+        cell->face(3)->set_boundary_id(5);
+
+      }
+
   }
 
 
@@ -2707,7 +2731,8 @@ namespace GridGenerator
   void
   hyper_L (Triangulation<3> &tria,
            const double      a,
-           const double      b)
+           const double      b,
+           const bool)
   {
     // we slice out the top back right
     // part of the cube
-- 
2.39.5