ThreeTEdgeTriadCounter Class Reference

#include <temporalmotifs.h>

Inherits StarTriad3TEdgeCounter< TriadEdgeData >.

Public Member Functions
	ThreeTEdgeTriadCounter (int max_nodes, int node_u, int node_v)
int	Counts (int dir1, int dir2, int dir3)
Public Member Functions inherited from StarTriad3TEdgeCounter< TriadEdgeData >
	StarTriad3TEdgeCounter ()
void	Count (const TVec< TriadEdgeData > &events, const TIntV &timestamps, double delta)

Protected Member Functions
void	InitializeCounters ()
void	PopPre (const TriadEdgeData &event)
void	PopPos (const TriadEdgeData &event)
void	PushPre (const TriadEdgeData &event)
void	PushPos (const TriadEdgeData &event)
void	ProcessCurrent (const TriadEdgeData &event)
bool	IsEdgeNode (int nbr)

Private Attributes
int	max_nodes_
Counter3D	pre_sum_
Counter3D	pos_sum_
Counter3D	mid_sum_
Counter3D	triad_counts_
Counter3D	pre_nodes_
Counter3D	pos_nodes_
int	node_u_
int	node_v_

Detailed Description

Definition at line 269 of file temporalmotifs.h.

Constructor & Destructor Documentation

ThreeTEdgeTriadCounter::ThreeTEdgeTriadCounter ( int  max_nodes, int  node_u, int  node_v  )

inline

Definition at line 275 of file temporalmotifs.h.

                                                                :
      max_nodes_(max_nodes), node_u_(node_u), node_v_(node_v) {}

Member Function Documentation

int ThreeTEdgeTriadCounter::Counts ( int  dir1, int  dir2, int  dir3  )

inline

Definition at line 279 of file temporalmotifs.h.

{ return triad_counts_(dir1, dir2, dir3); }

void ThreeTEdgeTriadCounter::InitializeCounters ( )

protectedvirtual

Implements StarTriad3TEdgeCounter< TriadEdgeData >.

Definition at line 691 of file temporalmotifs.cpp.

                                                {
  pre_nodes_ = Counter3D(2, 2, max_nodes_);
  pos_nodes_ = Counter3D(2, 2, max_nodes_);
  pre_sum_ = Counter3D(2, 2, 2);
  pos_sum_ = Counter3D(2, 2, 2);
  mid_sum_ = Counter3D(2, 2, 2);
  triad_counts_ = Counter3D(2, 2, 2);
}

bool ThreeTEdgeTriadCounter::IsEdgeNode ( int  nbr )

inlineprotected

Definition at line 288 of file temporalmotifs.h.

{ return nbr == node_u_ || nbr == node_v_; }

void ThreeTEdgeTriadCounter::PopPos ( const TriadEdgeData &  event )

protectedvirtual

Implements StarTriad3TEdgeCounter< TriadEdgeData >.

Definition at line 712 of file temporalmotifs.cpp.

                                                              {
  int nbr = event.nbr;
  int dir = event.dir;
  int u_or_v = event.u_or_v;  
  if (!IsEdgeNode(nbr)) {  
    pos_nodes_(dir, u_or_v, nbr) -= 1;
    for (int i = 0; i < 2; i++) {
      pos_sum_(u_or_v, dir, i) -= pos_nodes_(i, 1 - u_or_v, nbr);
    }
  }
}

void ThreeTEdgeTriadCounter::PopPre ( const TriadEdgeData &  event )

protectedvirtual

Implements StarTriad3TEdgeCounter< TriadEdgeData >.

Definition at line 700 of file temporalmotifs.cpp.

                                                              {
  int nbr = event.nbr;
  int dir = event.dir;
  int u_or_v = event.u_or_v;
  if (!IsEdgeNode(nbr)) {
    pre_nodes_(dir, u_or_v, nbr) -= 1;
    for (int i = 0; i < 2; i++) {
      pre_sum_(u_or_v, dir, i) -= pre_nodes_(i, 1 - u_or_v, nbr);
    }
  }
}

void ThreeTEdgeTriadCounter::ProcessCurrent ( const TriadEdgeData &  event )

protectedvirtual

Implements StarTriad3TEdgeCounter< TriadEdgeData >.

Definition at line 748 of file temporalmotifs.cpp.

                                                                      {
  int nbr = event.nbr;
  int dir = event.dir;
  int u_or_v = event.u_or_v;  
  // Adjust middle sums
  if (!IsEdgeNode(nbr)) {
    for (int i = 0; i < 2; i++) {
      mid_sum_(1 - u_or_v, i, dir) -= pre_nodes_(i, 1 - u_or_v, nbr);
      mid_sum_(u_or_v, dir, i) += pos_nodes_(i, 1 - u_or_v, nbr);
    }
  }
  // Update counts
  if (IsEdgeNode(nbr)) {
    // Determine if the event edge is u --> v or v --> u
    int u_to_v = 0;
    if (((nbr == node_u_) && dir == 0) || ((nbr == node_v_) && dir == 1)) {
      u_to_v = 1;
    }
    // i --> j, k --> j, i --> k    
    triad_counts_(0, 0, 0) += mid_sum_(u_to_v,     0, 0)
                           +  pos_sum_(u_to_v,     0, 1)
                           +  pre_sum_(1 - u_to_v, 1, 1);
    // i --> j, k --> i, j --> k
    triad_counts_(1, 0, 0) += mid_sum_(u_to_v,     1, 0)
                           +  pos_sum_(1 - u_to_v, 0, 1)
                           +  pre_sum_(1 - u_to_v, 0, 1);
    // i --> j, j --> k, i --> k
    triad_counts_(0, 1, 0) += mid_sum_(1 - u_to_v, 0, 0)
                           +  pos_sum_(u_to_v,     1, 1)
                           +  pre_sum_(1 - u_to_v, 1, 0);
    // i --> j, i --> k, j --> k
    triad_counts_(1, 1, 0) += mid_sum_(1 - u_to_v, 1, 0)
                           +  pos_sum_(1 - u_to_v, 1, 1)
                           +  pre_sum_(1 - u_to_v, 0, 0);
    // i --> j, k --> j, k --> i
    triad_counts_(0, 0, 1) += mid_sum_(u_to_v,     0, 1)
                           +  pos_sum_(u_to_v,     0, 0)
                           +  pre_sum_(u_to_v,     1, 1);
    // i --> j, k --> i, k --> j
    triad_counts_(1, 0, 1) += mid_sum_(u_to_v,     1, 1)
                           +  pos_sum_(1 - u_to_v, 0, 0)
                           +  pre_sum_(u_to_v,     0, 1);
    // i --> j, j --> k, k --> i
    triad_counts_(0, 1, 1) += mid_sum_(1 - u_to_v, 0, 1)
                           +  pos_sum_(u_to_v,     1, 0)
                           +  pre_sum_(u_to_v,     1, 0);
    // i --> j, i --> k, k --> j
    triad_counts_(1, 1, 1) += mid_sum_(1 - u_to_v, 1, 1)
                           +  pos_sum_(1 - u_to_v, 1, 0)
                           +  pre_sum_(u_to_v,     0, 0);
  }
}

void ThreeTEdgeTriadCounter::PushPos ( const TriadEdgeData &  event )

protectedvirtual

Implements StarTriad3TEdgeCounter< TriadEdgeData >.

Definition at line 736 of file temporalmotifs.cpp.

                                                               {
  int nbr = event.nbr;
  int dir = event.dir;
  int u_or_v = event.u_or_v;
  if (!IsEdgeNode(nbr)) {
    for (int i = 0; i < 2; i++) {
      pos_sum_(1 - u_or_v, i, dir) += pos_nodes_(i, 1 - u_or_v, nbr);
    }
    pos_nodes_(dir, u_or_v, nbr) += 1;
  }
}

void ThreeTEdgeTriadCounter::PushPre ( const TriadEdgeData &  event )

protectedvirtual

Implements StarTriad3TEdgeCounter< TriadEdgeData >.

Definition at line 724 of file temporalmotifs.cpp.

                                                               {
  int nbr = event.nbr;
  int dir = event.dir;
  int u_or_v = event.u_or_v;  
  if (!IsEdgeNode(nbr)) {  
    for (int i = 0; i < 2; i++) {
      pre_sum_(1 - u_or_v, i, dir) += pre_nodes_(i, 1 - u_or_v, nbr);
    }
    pre_nodes_(dir, u_or_v, nbr) += 1;
  }
}

Member Data Documentation

int ThreeTEdgeTriadCounter::max_nodes_

private

Definition at line 291 of file temporalmotifs.h.

Counter3D ThreeTEdgeTriadCounter::mid_sum_

private

Definition at line 294 of file temporalmotifs.h.

int ThreeTEdgeTriadCounter::node_u_

private

Definition at line 299 of file temporalmotifs.h.

int ThreeTEdgeTriadCounter::node_v_

private

Definition at line 300 of file temporalmotifs.h.

Counter3D ThreeTEdgeTriadCounter::pos_nodes_

private

Definition at line 297 of file temporalmotifs.h.

Counter3D ThreeTEdgeTriadCounter::pos_sum_

private

Definition at line 293 of file temporalmotifs.h.

Counter3D ThreeTEdgeTriadCounter::pre_nodes_

private

Definition at line 296 of file temporalmotifs.h.

Counter3D ThreeTEdgeTriadCounter::pre_sum_

private

Definition at line 292 of file temporalmotifs.h.

Counter3D ThreeTEdgeTriadCounter::triad_counts_

private

Definition at line 295 of file temporalmotifs.h.

---

The documentation for this class was generated from the following files:

• snap-adv/temporalmotifs.h
• snap-adv/temporalmotifs.cpp