test_app.C

Go to the documentation of this file.

#include "test_app.H"
#include "gtex/smooth_shade.H"  // used in BALLwidget_anchor
#include "gtex/wireframe.H"  // used in GRIDwidget_anchor
#include "mesh/lmesh.H"

const int HEIGHT = 6;
const double REGULARITY = 20;
const double MIN_DIST = 0.35;

/*****************************************************************
 * BALLwidget_anchor:
 *
 *      Class used for those sample point balls
 *****************************************************************/
MAKE_PTR_SUBC(BALLwidget_anchor, GEOM);
class BALLwidget_anchor : public GEOM {
 public:
   BALLwidget_anchor() : GEOM() {
      // Start with an empty LMESH:
      LMESHptr mesh = new LMESH();

      // Ensure mesh knows its containing GEOM:
      mesh->set_geom(this);

      // Make a ball shape
      mesh->Icosahedron();

      // Regardless of current rendering mode,
      // always use Gouraud shading:
      mesh->set_render_style(SmoothShadeTexture::static_name());

      // Color the Patch blue:
      mesh->patch(0)->set_color(COLOR(0.1, 0.1, 0.9));
//      mesh->patch(0)->set_transp(0.9);

      // Store the mesh:
      _body = mesh;

      set_name("BALLwidget_anchor");
   }

   // We're not in the picking game:
   virtual int draw_vis_ref() { return 0; }

   virtual bool needs_blend() const { return false; }
};

/*****************************************************************
 * GRIDwidget_anchor:
 *
 *      Class used for those octree grid
 *****************************************************************/
MAKE_PTR_SUBC(GRIDwidget_anchor, GEOM);
class GRIDwidget_anchor : public GEOM {
 public:
   GRIDwidget_anchor(Wtransf ff) : GEOM() {
      // Start with an empty LMESH:
      LMESHptr mesh = new LMESH();

      // Ensure mesh knows its containing GEOM:
      mesh->set_geom(this);

      // Make a cube shape
      mesh->Cube();
      set_xform(ff);

      // Regardless of current rendering mode,
      // always use wireframe shading:
      mesh->set_render_style(WireframeTexture::static_name());

      // Color the Patch black:
      mesh->patch(0)->set_color(COLOR::black);
      mesh->patch(0)->set_transp(0.5);

      // Store the mesh:
      _body = mesh;

      set_name("GRIDwidget_anchor");
   }

   // We're not in the picking game:
   virtual int draw_vis_ref() { return 0; }

   virtual bool needs_blend() const { return false; }
};

/*****************************************************************
 * TestSPSapp Methods
 *****************************************************************/
TestSPSapp* TestSPSapp::_instance = 0;

void
TestSPSapp::create_grid(Wtransf ff)
{
      _boxes += new GRIDwidget_anchor(ff);
      GEOMptr _anchor = _boxes.last();
      _anchor->set_pickable(0); // Don't allow picking of anchor
      NETWORK.set(_anchor, 0);// Don't network the anchor
      CONSTRAINT.set(_anchor, GEOM::SCREEN_WIDGET);
      WORLD::create   (_anchor, false); 
      WORLD::undisplay(_anchor, false);
}

void
TestSPSapp::visit(OctreeNode* node)
{
   if (node->get_leaf()) {
      if (node->get_disp()) {
         Wtransf ff = Wtransf(node->min()) * Wtransf::scaling(node->dim());
         create_grid(ff);
      }
   } else {
      for (int i = 0; i < 8; i++)
         visit(node->get_children()[i]);
   }
}

void
TestSPSapp::load_scene()
{
   BaseJOTapp::load_scene();

   for (int i=0; i<EXIST.num(); i++) {
      GEOMptr geom = GEOM::upcast(EXIST[i]);
      if (geom && BMESH::isa(geom->body())) {
         Bvert_list list;
         Bface_list fs;
         ARRAY<Wvec> bcs;
         //generate_samples(BMESH::upcast(geom->body()), 1.0, fs, bcs);
         //generate_samples(BMESH::upcast(geom->body()), fs, bcs, HEIGHT, MIN_DIST);
         _nodes += sps(BMESH::upcast(geom->body()), HEIGHT, REGULARITY, MIN_DIST, fs, bcs);
         for (int i = 0; i < fs.num(); i++) {
            Wpt pt;
            fs[i]->bc2pos(bcs[i], pt);
            list += new Bvert(pt);
         }
         _pts += list;
      }
   }

   for (int i = 0; i < _nodes.num(); i++)
      visit(_nodes[i]);

   for (int i = 0; i < _pts.num(); i++)
      for (int j = 0; j < _pts[i].num(); j++) {        
      _balls += new BALLwidget_anchor;
      GEOMptr _anchor = _balls.last();
      _anchor->set_pickable(0); // Don't allow picking of anchor
      NETWORK.set(_anchor, 0);// Don't network the anchor
      CONSTRAINT.set(_anchor, GEOM::SCREEN_WIDGET);
      WORLD::create   (_anchor, false); 
      WORLD::undisplay(_anchor, false);
   }
}

void
TestSPSapp::init_kbd(WINDOW &base_window)
{
   // Set up keyboard callbacks.

   BaseJOTapp::init_kbd(base_window);

   // Add key commands specific to test app:
   _key_menu->add_menu_item('s', "Toggle Sample",      &toggle_sample_cb);
   _key_menu->add_menu_item('g', "Toggle Grid",      &toggle_grid_cb);
}

int
TestSPSapp::toggle_sample_cb(const Event&, State *&)
{
   // Switch between showing and hiding samples

   assert(_instance);
   _instance->_show_sample = !_instance->_show_sample;
   if (_instance->_show_sample) {
      for (int i = 0, k = 0; i < _instance->_pts.num(); i++)
         for (int j = 0; j < _instance->_pts[i].num(); j++, k++) {
            WORLD::display(_instance->_balls[k], false);  
            Wpt p = _instance->_pts[i][j]->loc();
            Wvec    delt(p-Wpt(XYpt(p)+XYvec(VEXEL(3,3)), p));
            Wtransf ff = Wtransf(p) * Wtransf::scaling(1.5*delt.length()); 
            _instance->_balls[k]->set_xform(ff); 
         }
      err_msg("showing sample");
   }
   else {
      for (int i = 0; i < _instance->_balls.num(); i++)
         WORLD::undisplay(_instance->_balls[i], false);
      err_msg("samples not shown");
   }

   return 0;
}

int
TestSPSapp::toggle_grid_cb(const Event&, State *&)
{
   // Switch between showing and hiding grid

   assert(_instance);
   _instance->_show_grid = !_instance->_show_grid;
   if (_instance->_show_grid) {
      for (int i = 0; i < _instance->_boxes.num(); i++) 
         WORLD::display(_instance->_boxes[i], false);  
      err_msg("showing grid");
   }
   else {
      for (int i = 0; i < _instance->_boxes.num(); i++)
         WORLD::undisplay(_instance->_boxes[i], false);
      err_msg("grid not shown");
   }

   return 0;
}

/**********************************************************************
 * main()
 **********************************************************************/
int
main(int argc, char **argv)
{
   TestSPSapp app(argc, argv);

   // Define the program name displayed in the window title bar:
   Config::set_var_str("JOT_WINDOW_NAME", "Stratified Point Sampling test app");

   app.init();
   app.Run();

   return 0;
}

// end of file test_app.C

---