1 files changed, 124 insertions, 0 deletions
diff --git a/include/libcamera/internal/media_object.h b/include/libcamera/internal/media_object.h
new file mode 100644
index 00000000..748eafdc
--- /dev/null
+++ b/include/libcamera/internal/media_object.h
@@ -0,0 +1,124 @@
+/* SPDX-License-Identifier: LGPL-2.1-or-later */
+/*
+ * Copyright (C) 2018, Google Inc.
+ *
+ * media_object.h - Media Device objects: entities, pads and links.
+ */
+#ifndef __LIBCAMERA_MEDIA_OBJECT_H__
+#define __LIBCAMERA_MEDIA_OBJECT_H__
+
+#include <string>
+#include <vector>
+
+#include <linux/media.h>
+
+namespace libcamera {
+
+class MediaDevice;
+class MediaEntity;
+class MediaPad;
+
+class MediaObject
+{
+public:
+	MediaDevice *device() { return dev_; }
+	unsigned int id() const { return id_; }
+
+protected:
+	friend class MediaDevice;
+
+	MediaObject(MediaDevice *dev, unsigned int id)
+		: dev_(dev), id_(id)
+	{
+	}
+	virtual ~MediaObject() {}
+
+	MediaDevice *dev_;
+	unsigned int id_;
+};
+
+class MediaLink : public MediaObject
+{
+public:
+	MediaPad *source() const { return source_; }
+	MediaPad *sink() const { return sink_; }
+	unsigned int flags() const { return flags_; }
+	int setEnabled(bool enable);
+
+private:
+	friend class MediaDevice;
+
+	MediaLink(const struct media_v2_link *link,
+		  MediaPad *source, MediaPad *sink);
+	MediaLink(const MediaLink &) = delete;
+	~MediaLink() {}
+
+	MediaPad *source_;
+	MediaPad *sink_;
+	unsigned int flags_;
+};
+
+class MediaPad : public MediaObject
+{
+public:
+	unsigned int index() const { return index_; }
+	MediaEntity *entity() const { return entity_; }
+	unsigned int flags() const { return flags_; }
+	const std::vector<MediaLink *> &links() const { return links_; }
+
+	void addLink(MediaLink *link);
+
+private:
+	friend class MediaDevice;
+
+	MediaPad(const struct media_v2_pad *pad, MediaEntity *entity);
+	MediaPad(const MediaPad &) = delete;
+	~MediaPad();
+
+	unsigned int index_;
+	MediaEntity *entity_;
+	unsigned int flags_;
+
+	std::vector<MediaLink *> links_;
+};
+
+class MediaEntity : public MediaObject
+{
+public:
+	const std::string &name() const { return name_; }
+	unsigned int function() const { return function_; }
+	unsigned int flags() const { return flags_; }
+	const std::string &deviceNode() const { return deviceNode_; }
+	unsigned int deviceMajor() const { return major_; }
+	unsigned int deviceMinor() const { return minor_; }
+
+	const std::vector<MediaPad *> &pads() const { return pads_; }
+
+	const MediaPad *getPadByIndex(unsigned int index) const;
+	const MediaPad *getPadById(unsigned int id) const;
+
+	int setDeviceNode(const std::string &deviceNode);
+
+private:
+	friend class MediaDevice;
+
+	MediaEntity(MediaDevice *dev, const struct media_v2_entity *entity,
+		    unsigned int major = 0, unsigned int minor = 0);
+	MediaEntity(const MediaEntity &) = delete;
+	~MediaEntity();
+
+	std::string name_;
+	unsigned int function_;
+	unsigned int flags_;
+	std::string deviceNode_;
+	unsigned int major_;
+	unsigned int minor_;
+
+	std::vector<MediaPad *> pads_;
+
+	void addPad(MediaPad *pad);
+};
+
+} /* namespace libcamera */
+
+#endif /* __LIBCAMERA_MEDIA_OBJECT_H__ */