---

nci/suif/suif2b/basepasses/usefulpasses/suif_counter.h

Go to the documentation of this file.

#ifndef _SUIFLINK_SUIFCOUNTER_H_
#define _SUIFLINK_SUIFCOUNTER_H_

#include "common/suif_map.h"
#include "common/suif_vector.h"
#include "iokernel/meta_class.h"
#include "suifkernel/suif_object.h"
#include "suifkernel/suif_env.h"
#include "suifkernel/all_walker.h"
#include <iostream.h>

class NodeCounter;

class NodeCounter {
private:
  const MetaClass * const _meta_class;
  NodeCounter *_parent;
  suif_vector<NodeCounter*> _children;
  unsigned _direct_count;
  void set_parent(NodeCounter*);
public:
  NodeCounter(const MetaClass*);
  ~NodeCounter(void);
  MetaClass* get_meta_class(void) const;
  unsigned get_direct_count(void) const;
  unsigned get_indirect_count(void) const;
  void add_child(NodeCounter*);
  unsigned get_child_count(void) const;
  NodeCounter* get_nth_child(unsigned) const;
  unsigned add_direct_count(void);
};



class SuifCounter : public AllWalker {
 private:
  NodeCounter *_root_counter;
  suif_map<const MetaClass*, NodeCounter*> _NodeCounter_map;
  bool make_ancestors(NodeCounter*);
  NodeCounter *add_node_counter(const MetaClass*);

 public:
  SuifCounter(SuifEnv*);

  Walker::ApplyStatus operator()(SuifObject*);

  void count(SuifObject* obj);

  NodeCounter* get_node_counter(const MetaClass* mc);

  ostream& print_result(ostream&) const;

};


#endif // _SUIFLINK_SUIFCOUNTER_H_

---