gstreamer-rs/gstreamer/src/format.rs

// Copyright (C) 2017 Sebastian Dröge <sebastian@centricular.com>
//
// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
// option. This file may not be copied, modified, or distributed
// except according to those terms.

use ClockTime;
use Format;

#[derive(PartialEq, Eq, Hash, Clone, Copy, Debug)]
pub enum FormatValue {
    Undefined(i64),
    Default(Option<u64>),
    Bytes(Option<u64>),
    Time(ClockTime),
    Buffers(Option<u64>),
    Percent(Option<u32>),
    Other(Format, i64),
}

impl FormatValue {
    pub fn new(format: Format, value: i64) -> Self {
        match format {
            Format::Undefined => FormatValue::Undefined(value),
            Format::Default => FormatValue::Default(if value == -1 {
                None
            } else {
                Some(value as u64)
            }),
            Format::Bytes => FormatValue::Bytes(if value == -1 {
                None
            } else {
                Some(value as u64)
            }),
            Format::Time => FormatValue::Time(if value == -1 {
                ClockTime::none()
            } else {
                ClockTime::from(value as u64)
            }),
            Format::Buffers => FormatValue::Buffers(if value == -1 {
                None
            } else {
                Some(value as u64)
            }),
            Format::Percent => FormatValue::Percent(if value == -1 {
                None
            } else {
                Some(value as u32)
            }),
            Format::__Unknown(_) => FormatValue::Other(format, value),
        }
    }

    pub fn from_undefined(v: i64) -> Self {
        FormatValue::Undefined(v)
    }

    pub fn from_default<V: Into<Option<u64>>>(v: V) -> Self {
        FormatValue::Default(v.into())
    }

    pub fn from_bytes<V: Into<Option<u64>>>(v: V) -> Self {
        FormatValue::Bytes(v.into())
    }

    pub fn from_time(v: ClockTime) -> Self {
        FormatValue::Time(v)
    }

    pub fn from_buffers<V: Into<Option<u64>>>(v: V) -> Self {
        FormatValue::Buffers(v.into())
    }

    pub fn from_percent<V: Into<Option<u32>>>(v: V) -> Self {
        FormatValue::Percent(v.into())
    }

    pub fn from_other(format: Format, v: i64) -> Self {
        FormatValue::Other(format, v)
    }

    pub fn to_format(&self) -> Format {
        match *self {
            FormatValue::Undefined(_) => Format::Undefined,
            FormatValue::Default(_) => Format::Default,
            FormatValue::Bytes(_) => Format::Bytes,
            FormatValue::Time(_) => Format::Time,
            FormatValue::Buffers(_) => Format::Buffers,
            FormatValue::Percent(_) => Format::Percent,
            FormatValue::Other(f, _) => f,
        }
    }

    pub fn to_value(&self) -> i64 {
        match *self {
            FormatValue::Undefined(v) => v,
            FormatValue::Default(v) => v.map(|v| v as i64).unwrap_or(-1),
            FormatValue::Bytes(v) => v.map(|v| v as i64).unwrap_or(-1),
            FormatValue::Time(v) => v.map(|v| v as i64).unwrap_or(-1),
            FormatValue::Buffers(v) => v.map(|v| v as i64).unwrap_or(-1),
            FormatValue::Percent(v) => v.map(|v| v as i64).unwrap_or(-1),
            FormatValue::Other(_, v) => v,
        }
    }

    pub fn try_to_undefined(&self) -> Option<i64> {
        if let FormatValue::Undefined(v) = *self {
            Some(v)
        } else {
            None
        }
    }

    pub fn try_to_default(&self) -> Option<Option<u64>> {
        if let FormatValue::Default(v) = *self {
            Some(v)
        } else {
            None
        }
    }

    pub fn try_to_bytes(&self) -> Option<Option<u64>> {
        if let FormatValue::Bytes(v) = *self {
            Some(v)
        } else {
            None
        }
    }

    pub fn try_to_time(&self) -> Option<ClockTime> {
        if let FormatValue::Time(v) = *self {
            Some(v)
        } else {
            None
        }
    }

    pub fn try_to_buffers(&self) -> Option<Option<u64>> {
        if let FormatValue::Buffers(v) = *self {
            Some(v)
        } else {
            None
        }
    }

    pub fn try_to_percent(&self) -> Option<Option<u32>> {
        if let FormatValue::Percent(v) = *self {
            Some(v)
        } else {
            None
        }
    }

    pub fn try_to_other(&self) -> Option<(Format, i64)> {
        if let FormatValue::Other(f, v) = *self {
            Some((f, v))
        } else {
            None
        }
    }
}

impl From<ClockTime> for FormatValue {
    fn from(v: ClockTime) -> FormatValue {
        FormatValue::Time(v)
    }
}