IVisualization.hpp

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file IVisualization.hpp

// This file is part of the Acts project.
//
// Copyright (C) 2019 CERN for the benefit of the Acts project
//
// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/.

#pragma once

#include "Acts/Utilities/Definitions.hpp"

#include <array>
#include <fstream>
#include <string>
#include <vector>

namespace Acts {

class IVisualization {
 public:
  using ColorType = std::array<int, 3>;

  using FaceType = std::vector<size_t>;

  virtual void vertex(const Vector3D& vtx,
                      ColorType color = {120, 120, 120}) = 0;

  virtual void face(const std::vector<Vector3D>& vtxs,
                    ColorType color = {120, 120, 120}) = 0;

  virtual void faces(const std::vector<Vector3D>& vtxs,
                     const std::vector<FaceType>& faces,
                     ColorType color = {120, 120, 120}) = 0;

  virtual void line(const Vector3D& a, const Vector3D& b,
                    ColorType color = {120, 120, 120}) = 0;

  virtual void write(std::ostream& os) const = 0;

  virtual void write(const std::string& path) const = 0;

  virtual void clear() = 0;


  void vertex(const Vector3F& vtx, ColorType color = {120, 120, 120});

  void face(const std::vector<Vector3F>& vtxs,
            ColorType color = {120, 120, 120});

  void line(const Vector3F& a, const Vector3F& b,
            ColorType color = {120, 120, 120});

 protected:
  bool hasExtension(const std::string& path) const;

  void replaceExtension(std::string& path, const std::string& suffix) const;
};

inline std::ostream& operator<<(std::ostream& os, const IVisualization& hlp) {
  hlp.write(os);
  return os;
}
}  // namespace Acts