1 | use std::mem; |
2 | use std::ops::Deref; |
3 | |
4 | use super::Chapter; |
5 | use ffi::*; |
6 | use format::context::common::Context; |
7 | use {Dictionary, DictionaryMut, Rational}; |
8 | |
9 | // WARNING: index refers to the offset in the chapters array (starting from 0) |
10 | // it is not necessarly equal to the id (which may start at 1) |
11 | pub struct ChapterMut<'a> { |
12 | context: &'a mut Context, |
13 | index: usize, |
14 | |
15 | immutable: Chapter<'a>, |
16 | } |
17 | |
18 | impl<'a> ChapterMut<'a> { |
19 | pub unsafe fn wrap(context: &mut Context, index: usize) -> ChapterMut { |
20 | ChapterMut { |
21 | context: mem::transmute_copy(&context), |
22 | index, |
23 | |
24 | immutable: Chapter::wrap(context:mem::transmute_copy(&context), index), |
25 | } |
26 | } |
27 | |
28 | pub unsafe fn as_mut_ptr(&mut self) -> *mut AVChapter { |
29 | *(*self.context.as_mut_ptr()).chapters.add(self.index) |
30 | } |
31 | } |
32 | |
33 | impl<'a> ChapterMut<'a> { |
34 | pub fn set_id(&mut self, value: i64) { |
35 | unsafe { |
36 | (*self.as_mut_ptr()).id = value as _; |
37 | } |
38 | } |
39 | |
40 | pub fn set_time_base<R: Into<Rational>>(&mut self, value: R) { |
41 | unsafe { |
42 | (*self.as_mut_ptr()).time_base = value.into().into(); |
43 | } |
44 | } |
45 | |
46 | pub fn set_start(&mut self, value: i64) { |
47 | unsafe { |
48 | (*self.as_mut_ptr()).start = value; |
49 | } |
50 | } |
51 | |
52 | pub fn set_end(&mut self, value: i64) { |
53 | unsafe { |
54 | (*self.as_mut_ptr()).end = value; |
55 | } |
56 | } |
57 | |
58 | pub fn set_metadata<K: AsRef<str>, V: AsRef<str>>(&mut self, key: K, value: V) { |
59 | // dictionary.set() allocates the AVDictionary the first time a key/value is inserted |
60 | // so we want to update the metadata dictionary afterwards |
61 | unsafe { |
62 | let mut dictionary = Dictionary::own(self.metadata().as_mut_ptr()); |
63 | dictionary.set(key.as_ref(), value.as_ref()); |
64 | (*self.as_mut_ptr()).metadata = dictionary.disown(); |
65 | } |
66 | } |
67 | |
68 | pub fn metadata(&mut self) -> DictionaryMut { |
69 | unsafe { DictionaryMut::wrap((*self.as_mut_ptr()).metadata) } |
70 | } |
71 | } |
72 | |
73 | impl<'a> Deref for ChapterMut<'a> { |
74 | type Target = Chapter<'a>; |
75 | |
76 | fn deref(&self) -> &Self::Target { |
77 | &self.immutable |
78 | } |
79 | } |
80 | |