github_mirrors/vectorscan
3
0
Fork 1
mirror of https://github.com/VectorCamp/vectorscan.git synced 2025-06-28 16:41:01 +03:00
Code Issues Packages Projects Releases Wiki Activity

ng_netflow: use small_color_map

Browse Source
This commit is contained in:
Justin Viiret 2017-06-29 11:20:17 +10:00 committed by Matthew Barr
parent b97fa8c808
commit 33141e64b6
1 changed files with 11 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
src/nfagraph/ng_netflow.cpp
View File

@ -1,5 +1,5 @@
/* /*
* Copyright (c) 2015-2016, Intel Corporation * Copyright (c) 2015-2017, Intel Corporation
* *
* Redistribution and use in source and binary forms, with or without * Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions are met: * modification, are permitted provided that the following conditions are met:
@ -37,6 +37,7 @@
#include "ue2common.h" #include "ue2common.h"
#include "util/container.h" #include "util/container.h"
#include "util/graph_range.h" #include "util/graph_range.h"
#include "util/graph_small_color_map.h"
#include <algorithm> #include <algorithm>
#include <boost/graph/boykov_kolmogorov_max_flow.hpp> #include <boost/graph/boykov_kolmogorov_max_flow.hpp>
@ -118,7 +119,7 @@ void removeEdgesFromIndex(NGHolder &g, vector<u64a> &capacityMap, u32 idx) {
* colour map (from which we can find the min cut). */ * colour map (from which we can find the min cut). */
static static
u64a getMaxFlow(NGHolder &h, const vector<u64a> &capacityMap_in, u64a getMaxFlow(NGHolder &h, const vector<u64a> &capacityMap_in,
vector<default_color_type> &colorMap) { decltype(make_small_color_map(NGHolder())) &colorMap) {
vector<u64a> capacityMap = capacityMap_in; vector<u64a> capacityMap = capacityMap_in;
NFAVertex src = h.start; NFAVertex src = h.start;
NFAVertex sink = h.acceptEod; NFAVertex sink = h.acceptEod;
@ -141,7 +142,6 @@ u64a getMaxFlow(NGHolder &h, const vector<u64a> &capacityMap_in,
vector<u64a> edgeResiduals(numTotalEdges); vector<u64a> edgeResiduals(numTotalEdges);
vector<NFAEdge> predecessors(numVertices); vector<NFAEdge> predecessors(numVertices);
vector<s32> distances(numVertices); vector<s32> distances(numVertices);
assert(colorMap.size() == numVertices);
auto v_index_map = get(vertex_index, h); auto v_index_map = get(vertex_index, h);
auto e_index_map = get(edge_index, h); auto e_index_map = get(edge_index, h);
@ -151,7 +151,7 @@ u64a getMaxFlow(NGHolder &h, const vector<u64a> &capacityMap_in,
make_iterator_property_map(edgeResiduals.begin(), e_index_map), make_iterator_property_map(edgeResiduals.begin(), e_index_map),
make_iterator_property_map(reverseEdges.begin(), e_index_map), make_iterator_property_map(reverseEdges.begin(), e_index_map),
make_iterator_property_map(predecessors.begin(), v_index_map), make_iterator_property_map(predecessors.begin(), v_index_map),
make_iterator_property_map(colorMap.begin(), v_index_map), colorMap,
make_iterator_property_map(distances.begin(), v_index_map), make_iterator_property_map(distances.begin(), v_index_map),
v_index_map, v_index_map,
src, sink); src, sink);
@ -169,8 +169,8 @@ vector<NFAEdge> findMinCut(NGHolder &h, const vector<u64a> &scores) {
assert(hasCorrectlyNumberedEdges(h)); assert(hasCorrectlyNumberedEdges(h));
assert(hasCorrectlyNumberedVertices(h)); assert(hasCorrectlyNumberedVertices(h));
vector<default_color_type> colorMap(num_vertices(h)); auto colors = make_small_color_map(h);
u64a flow = getMaxFlow(h, scores, colorMap); u64a flow = getMaxFlow(h, scores, colors);
vector<NFAEdge> picked_white; vector<NFAEdge> picked_white;
vector<NFAEdge> picked_black; vector<NFAEdge> picked_black;
@ -185,17 +185,17 @@ vector<NFAEdge> findMinCut(NGHolder &h, const vector<u64a> &scores) {
continue; // skips, among other things, reverse edges continue; // skips, among other things, reverse edges
} }
default_color_type fromColor = colorMap[h[from].index]; auto fromColor = get(colors, from);
default_color_type toColor = colorMap[h[to].index]; auto toColor = get(colors, to);
if (fromColor != boost::white_color && toColor == boost::white_color) { if (fromColor != small_color::white && toColor == small_color::white) {
assert(ec <= INVALID_EDGE_CAP); assert(ec <= INVALID_EDGE_CAP);
DEBUG_PRINTF("found white cut edge %zu->%zu cap %llu\n", DEBUG_PRINTF("found white cut edge %zu->%zu cap %llu\n",
h[from].index, h[to].index, ec); h[from].index, h[to].index, ec);
observed_white_flow += ec; observed_white_flow += ec;
picked_white.push_back(e); picked_white.push_back(e);
} }
if (fromColor == boost::black_color && toColor != boost::black_color) { if (fromColor == small_color::black && toColor != small_color::black) {
assert(ec <= INVALID_EDGE_CAP); assert(ec <= INVALID_EDGE_CAP);
DEBUG_PRINTF("found black cut edge %zu->%zu cap %llu\n", DEBUG_PRINTF("found black cut edge %zu->%zu cap %llu\n",
h[from].index, h[to].index, ec); h[from].index, h[to].index, ec);
@ -206,7 +206,7 @@ vector<NFAEdge> findMinCut(NGHolder &h, const vector<u64a> &scores) {
DEBUG_PRINTF("min flow = %llu b flow = %llu w flow %llu\n", flow, DEBUG_PRINTF("min flow = %llu b flow = %llu w flow %llu\n", flow,
observed_black_flow, observed_white_flow); observed_black_flow, observed_white_flow);
if (MIN(observed_white_flow, observed_black_flow) != flow) { if (min(observed_white_flow, observed_black_flow) != flow) {
DEBUG_PRINTF("bad cut\n"); DEBUG_PRINTF("bad cut\n");
} }