GraphvizFormatting.h

Go to the documentation of this file.

/* ----------------------------------------------------------------------------

 * GTSAM Copyright 2010-2021, Georgia Tech Research Corporation,
 * Atlanta, Georgia 30332-0415
 * All Rights Reserved
 * Authors: Frank Dellaert, et al. (see THANKS for the full author list)

 * See LICENSE for the license information

 * -------------------------------------------------------------------------- */

#pragma once

#include <gtsam/inference/DotWriter.h>

namespace gtsam {

class Values;
class Value;

struct GTSAM_EXPORT GraphvizFormatting : public DotWriter {
  enum Axis { X, Y, Z, NEGX, NEGY, NEGZ };

  Axis paperHorizontalAxis;  
  Axis paperVerticalAxis;    
  double scale;  
  bool mergeSimilarFactors;  

  GraphvizFormatting()
      : paperHorizontalAxis(Y),
        paperVerticalAxis(X),
        scale(1),
        mergeSimilarFactors(false) {}

  // Find bounds
  Vector2 findBounds(const Values& values, const KeySet& keys) const;

  std::optional<Vector2> extractPosition(const Value& value) const;

  std::optional<Vector2> variablePos(const Values& values, const Vector2& min,
                                       Key key) const;

  std::optional<Vector2> factorPos(const Vector2& min, size_t i) const;
};

}  // namespace gtsam