Theora (empty) → 1.0
raw patch · 4 files changed
+765/−0 lines, 4 filesdep +basesetup-changed
Dependencies added: base
Files
- LICENSE +347/−0
- Setup.hs +2/−0
- Theora.cabal +13/−0
- Theora/Encoding.hsc +403/−0
+ LICENSE view
@@ -0,0 +1,347 @@+This program is free software; you can redistribute it and/or modify+it under the terms of the GNU General Public License as published by+the Free Software Foundation; either version 2 of the License, or+(at your option) any later version.++++ GNU GENERAL PUBLIC LICENSE+ Version 2, June 1991++ Copyright (C) 1989, 1991 Free Software Foundation, Inc.+ 675 Mass Ave, Cambridge, MA 02139, USA+ Everyone is permitted to copy and distribute verbatim copies+ of this license document, but changing it is not allowed.++ Preamble++ The licenses for most software are designed to take away your+freedom to share and change it. By contrast, the GNU General Public+License is intended to guarantee your freedom to share and change free+software--to make sure the software is free for all its users. This+General Public License applies to most of the Free Software+Foundation's software and to any other program whose authors commit to+using it. (Some other Free Software Foundation software is covered by+the GNU Library General Public License instead.) You can apply it to+your programs, too.++ When we speak of free software, we are referring to freedom, not+price. Our General Public Licenses are designed to make sure that you+have the freedom to distribute copies of free software (and charge for+this service if you wish), that you receive source code or can get it+if you want it, that you can change the software or use pieces of it+in new free programs; and that you know you can do these things.++ To protect your rights, we need to make restrictions that forbid+anyone to deny you these rights or to ask you to surrender the rights.+These restrictions translate to certain responsibilities for you if you+distribute copies of the software, or if you modify it.++ For example, if you distribute copies of such a program, whether+gratis or for a fee, you must give the recipients all the rights that+you have. You must make sure that they, too, receive or can get the+source code. And you must show them these terms so they know their+rights.++ We protect your rights with two steps: (1) copyright the software, and+(2) offer you this license which gives you legal permission to copy,+distribute and/or modify the software.++ Also, for each author's protection and ours, we want to make certain+that everyone understands that there is no warranty for this free+software. If the software is modified by someone else and passed on, we+want its recipients to know that what they have is not the original, so+that any problems introduced by others will not reflect on the original+authors' reputations.++ Finally, any free program is threatened constantly by software+patents. We wish to avoid the danger that redistributors of a free+program will individually obtain patent licenses, in effect making the+program proprietary. To prevent this, we have made it clear that any+patent must be licensed for everyone's free use or not licensed at all.++ The precise terms and conditions for copying, distribution and+modification follow.++ GNU GENERAL PUBLIC LICENSE+ TERMS AND CONDITIONS FOR COPYING, DISTRIBUTION AND MODIFICATION++ 0. This License applies to any program or other work which contains+a notice placed by the copyright holder saying it may be distributed+under the terms of this General Public License. The "Program", below,+refers to any such program or work, and a "work based on the Program"+means either the Program or any derivative work under copyright law:+that is to say, a work containing the Program or a portion of it,+either verbatim or with modifications and/or translated into another+language. (Hereinafter, translation is included without limitation in+the term "modification".) Each licensee is addressed as "you".++Activities other than copying, distribution and modification are not+covered by this License; they are outside its scope. The act of+running the Program is not restricted, and the output from the Program+is covered only if its contents constitute a work based on the+Program (independent of having been made by running the Program).+Whether that is true depends on what the Program does.++ 1. You may copy and distribute verbatim copies of the Program's+source code as you receive it, in any medium, provided that you+conspicuously and appropriately publish on each copy an appropriate+copyright notice and disclaimer of warranty; keep intact all the+notices that refer to this License and to the absence of any warranty;+and give any other recipients of the Program a copy of this License+along with the Program.++You may charge a fee for the physical act of transferring a copy, and+you may at your option offer warranty protection in exchange for a fee.++ 2. You may modify your copy or copies of the Program or any portion+of it, thus forming a work based on the Program, and copy and+distribute such modifications or work under the terms of Section 1+above, provided that you also meet all of these conditions:++ a) You must cause the modified files to carry prominent notices+ stating that you changed the files and the date of any change.++ b) You must cause any work that you distribute or publish, that in+ whole or in part contains or is derived from the Program or any+ part thereof, to be licensed as a whole at no charge to all third+ parties under the terms of this License.++ c) If the modified program normally reads commands interactively+ when run, you must cause it, when started running for such+ interactive use in the most ordinary way, to print or display an+ announcement including an appropriate copyright notice and a+ notice that there is no warranty (or else, saying that you provide+ a warranty) and that users may redistribute the program under+ these conditions, and telling the user how to view a copy of this+ License. (Exception: if the Program itself is interactive but+ does not normally print such an announcement, your work based on+ the Program is not required to print an announcement.)++These requirements apply to the modified work as a whole. If+identifiable sections of that work are not derived from the Program,+and can be reasonably considered independent and separate works in+themselves, then this License, and its terms, do not apply to those+sections when you distribute them as separate works. But when you+distribute the same sections as part of a whole which is a work based+on the Program, the distribution of the whole must be on the terms of+this License, whose permissions for other licensees extend to the+entire whole, and thus to each and every part regardless of who wrote it.++Thus, it is not the intent of this section to claim rights or contest+your rights to work written entirely by you; rather, the intent is to+exercise the right to control the distribution of derivative or+collective works based on the Program.++In addition, mere aggregation of another work not based on the Program+with the Program (or with a work based on the Program) on a volume of+a storage or distribution medium does not bring the other work under+the scope of this License.++ 3. You may copy and distribute the Program (or a work based on it,+under Section 2) in object code or executable form under the terms of+Sections 1 and 2 above provided that you also do one of the following:++ a) Accompany it with the complete corresponding machine-readable+ source code, which must be distributed under the terms of Sections+ 1 and 2 above on a medium customarily used for software interchange; or,++ b) Accompany it with a written offer, valid for at least three+ years, to give any third party, for a charge no more than your+ cost of physically performing source distribution, a complete+ machine-readable copy of the corresponding source code, to be+ distributed under the terms of Sections 1 and 2 above on a medium+ customarily used for software interchange; or,++ c) Accompany it with the information you received as to the offer+ to distribute corresponding source code. (This alternative is+ allowed only for noncommercial distribution and only if you+ received the program in object code or executable form with such+ an offer, in accord with Subsection b above.)++The source code for a work means the preferred form of the work for+making modifications to it. For an executable work, complete source+code means all the source code for all modules it contains, plus any+associated interface definition files, plus the scripts used to+control compilation and installation of the executable. However, as a+special exception, the source code distributed need not include+anything that is normally distributed (in either source or binary+form) with the major components (compiler, kernel, and so on) of the+operating system on which the executable runs, unless that component+itself accompanies the executable.++If distribution of executable or object code is made by offering+access to copy from a designated place, then offering equivalent+access to copy the source code from the same place counts as+distribution of the source code, even though third parties are not+compelled to copy the source along with the object code.++ 4. You may not copy, modify, sublicense, or distribute the Program+except as expressly provided under this License. Any attempt+otherwise to copy, modify, sublicense or distribute the Program is+void, and will automatically terminate your rights under this License.+However, parties who have received copies, or rights, from you under+this License will not have their licenses terminated so long as such+parties remain in full compliance.++ 5. You are not required to accept this License, since you have not+signed it. However, nothing else grants you permission to modify or+distribute the Program or its derivative works. These actions are+prohibited by law if you do not accept this License. Therefore, by+modifying or distributing the Program (or any work based on the+Program), you indicate your acceptance of this License to do so, and+all its terms and conditions for copying, distributing or modifying+the Program or works based on it.++ 6. Each time you redistribute the Program (or any work based on the+Program), the recipient automatically receives a license from the+original licensor to copy, distribute or modify the Program subject to+these terms and conditions. You may not impose any further+restrictions on the recipients' exercise of the rights granted herein.+You are not responsible for enforcing compliance by third parties to+this License.++ 7. If, as a consequence of a court judgment or allegation of patent+infringement or for any other reason (not limited to patent issues),+conditions are imposed on you (whether by court order, agreement or+otherwise) that contradict the conditions of this License, they do not+excuse you from the conditions of this License. If you cannot+distribute so as to satisfy simultaneously your obligations under this+License and any other pertinent obligations, then as a consequence you+may not distribute the Program at all. For example, if a patent+license would not permit royalty-free redistribution of the Program by+all those who receive copies directly or indirectly through you, then+the only way you could satisfy both it and this License would be to+refrain entirely from distribution of the Program.++If any portion of this section is held invalid or unenforceable under+any particular circumstance, the balance of the section is intended to+apply and the section as a whole is intended to apply in other+circumstances.++It is not the purpose of this section to induce you to infringe any+patents or other property right claims or to contest validity of any+such claims; this section has the sole purpose of protecting the+integrity of the free software distribution system, which is+implemented by public license practices. Many people have made+generous contributions to the wide range of software distributed+through that system in reliance on consistent application of that+system; it is up to the author/donor to decide if he or she is willing+to distribute software through any other system and a licensee cannot+impose that choice.++This section is intended to make thoroughly clear what is believed to+be a consequence of the rest of this License.++ 8. If the distribution and/or use of the Program is restricted in+certain countries either by patents or by copyrighted interfaces, the+original copyright holder who places the Program under this License+may add an explicit geographical distribution limitation excluding+those countries, so that distribution is permitted only in or among+countries not thus excluded. In such case, this License incorporates+the limitation as if written in the body of this License.++ 9. The Free Software Foundation may publish revised and/or new versions+of the General Public License from time to time. Such new versions will+be similar in spirit to the present version, but may differ in detail to+address new problems or concerns.++Each version is given a distinguishing version number. If the Program+specifies a version number of this License which applies to it and "any+later version", you have the option of following the terms and conditions+either of that version or of any later version published by the Free+Software Foundation. If the Program does not specify a version number of+this License, you may choose any version ever published by the Free Software+Foundation.++ 10. If you wish to incorporate parts of the Program into other free+programs whose distribution conditions are different, write to the author+to ask for permission. For software which is copyrighted by the Free+Software Foundation, write to the Free Software Foundation; we sometimes+make exceptions for this. Our decision will be guided by the two goals+of preserving the free status of all derivatives of our free software and+of promoting the sharing and reuse of software generally.++ NO WARRANTY++ 11. BECAUSE THE PROGRAM IS LICENSED FREE OF CHARGE, THERE IS NO WARRANTY+FOR THE PROGRAM, TO THE EXTENT PERMITTED BY APPLICABLE LAW. EXCEPT WHEN+OTHERWISE STATED IN WRITING THE COPYRIGHT HOLDERS AND/OR OTHER PARTIES+PROVIDE THE PROGRAM "AS IS" WITHOUT WARRANTY OF ANY KIND, EITHER EXPRESSED+OR IMPLIED, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF+MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE. THE ENTIRE RISK AS+TO THE QUALITY AND PERFORMANCE OF THE PROGRAM IS WITH YOU. SHOULD THE+PROGRAM PROVE DEFECTIVE, YOU ASSUME THE COST OF ALL NECESSARY SERVICING,+REPAIR OR CORRECTION.++ 12. IN NO EVENT UNLESS REQUIRED BY APPLICABLE LAW OR AGREED TO IN WRITING+WILL ANY COPYRIGHT HOLDER, OR ANY OTHER PARTY WHO MAY MODIFY AND/OR+REDISTRIBUTE THE PROGRAM AS PERMITTED ABOVE, BE LIABLE TO YOU FOR DAMAGES,+INCLUDING ANY GENERAL, SPECIAL, INCIDENTAL OR CONSEQUENTIAL DAMAGES ARISING+OUT OF THE USE OR INABILITY TO USE THE PROGRAM (INCLUDING BUT NOT LIMITED+TO LOSS OF DATA OR DATA BEING RENDERED INACCURATE OR LOSSES SUSTAINED BY+YOU OR THIRD PARTIES OR A FAILURE OF THE PROGRAM TO OPERATE WITH ANY OTHER+PROGRAMS), EVEN IF SUCH HOLDER OR OTHER PARTY HAS BEEN ADVISED OF THE+POSSIBILITY OF SUCH DAMAGES.++ END OF TERMS AND CONDITIONS++ Appendix: How to Apply These Terms to Your New Programs++ If you develop a new program, and you want it to be of the greatest+possible use to the public, the best way to achieve this is to make it+free software which everyone can redistribute and change under these terms.++ To do so, attach the following notices to the program. It is safest+to attach them to the start of each source file to most effectively+convey the exclusion of warranty; and each file should have at least+the "copyright" line and a pointer to where the full notice is found.++ <one line to give the program's name and a brief idea of what it does.>+ Copyright (C) 19yy <name of author>++ This program is free software; you can redistribute it and/or modify+ it under the terms of the GNU General Public License as published by+ the Free Software Foundation; either version 2 of the License, or+ (at your option) any later version.++ This program is distributed in the hope that it will be useful,+ but WITHOUT ANY WARRANTY; without even the implied warranty of+ MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the+ GNU General Public License for more details.++ You should have received a copy of the GNU General Public License+ along with this program; if not, write to the Free Software+ Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.++Also add information on how to contact you by electronic and paper mail.++If the program is interactive, make it output a short notice like this+when it starts in an interactive mode:++ Gnomovision version 69, Copyright (C) 19yy name of author+ Gnomovision comes with ABSOLUTELY NO WARRANTY; for details type `show w'.+ This is free software, and you are welcome to redistribute it+ under certain conditions; type `show c' for details.++The hypothetical commands `show w' and `show c' should show the appropriate+parts of the General Public License. Of course, the commands you use may+be called something other than `show w' and `show c'; they could even be+mouse-clicks or menu items--whatever suits your program.++You should also get your employer (if you work as a programmer) or your+school, if any, to sign a "copyright disclaimer" for the program, if+necessary. Here is a sample; alter the names:++ Yoyodyne, Inc., hereby disclaims all copyright interest in the program+ `Gnomovision' (which makes passes at compilers) written by James Hacker.++ <signature of Ty Coon>, 1 April 1989+ Ty Coon, President of Vice++This General Public License does not permit incorporating your program into+proprietary programs. If your program is a subroutine library, you may+consider it more useful to permit linking proprietary applications with the+library. If this is what you want to do, use the GNU Library General+Public License instead of this License.+
+ Setup.hs view
@@ -0,0 +1,2 @@+import Distribution.Simple+main=defaultMain
+ Theora.cabal view
@@ -0,0 +1,13 @@+Name: Theora+Version: 1.0+Description: Video Compression Library (see http://theora.org)+License: GPL+License-file: LICENSE+Author: Pierre-Etienne Meunier+Maintainer: pierreetienne.meunier@gmail.com+Build-Type: Simple+Cabal-Version: >=1.2+Library+ Build-Depends: base+ Exposed-modules: Theora.Encoding+ extra-libraries: ogg theora
+ Theora/Encoding.hsc view
@@ -0,0 +1,403 @@+{-#LANGUAGE ForeignFunctionInterface #-}+{-#OPTIONS -fglasgow-exts -logg -ltheora #-}++-- | This module calls the libtheora C library to generate video ogg files from YCbCr images.+-- As explained on <http://theora.org/doc/libtheora-1.0/group__encfuncs.html>, the basic steps+-- for creating a video are the following :+--+-- * create in an 'Info' record+--+-- * call 'encodeAlloc' on it+--+-- * call 'flushHeader' until it returns 'Nothing'+--+-- * For each uncompressed frame, submit it with 'encodeIn', then retrieve potential packets +-- with 'encodeOut'++module Theora.Encoding (-- * Theora structures and function+ Info(..),Comment (..),+ PixelFmt(Pf420,Pf422,Pf444),ColorSpace(..),+ encodeAlloc,+ flushHeader,+ ImgPlane,+ YCbCrBuffer(..),newYCbCr,+ encodeIn, encodeOut,+ -- * Ogg-related functions+ OggPacket,newOggStreamState,+ OggPage,streamPacketIn, streamPageOut, streamFlush) where++++import Foreign.Storable+import Foreign+import Foreign.ForeignPtr+import Foreign.C.Types+import Foreign.C.String+import Data.Int++#include <ogg/ogg.h>+#include <theora/codec.h>++{-| Theora supports 4 different pixel formats, listed in + <http://theora.org/doc/Theora.pdf>. YCbCr chroma frames (Cb and Cr) may be resized for+ compression (see <http://en.wikipedia.org/Ycbcr>). The resulting pixel formats+ are as follows :++-}++data PixelFmt=+ Pf420 -- ^Chroma decimation by 2 in both directions+ | PfRsvd+ | Pf422 -- ^Chroma decimation by 2 in the /x/ direction+ | Pf444 -- ^No chroma decimation++instance Enum PixelFmt where+ fromEnum Pf420= #const TH_PF_420+ fromEnum PfRsvd= #const TH_PF_RSVD+ fromEnum Pf422= #const TH_PF_422+ fromEnum Pf444= #const TH_PF_444++ toEnum (#const TH_PF_420)=Pf420+ toEnum (#const TH_PF_RSVD)=PfRsvd+ toEnum (#const TH_PF_422)=Pf422+ toEnum (#const TH_PF_444)=Pf444++data ColorSpace=+ Unspecified+ |ItuRec470M -- ^A color space designed for NTSC content+ |ItuRec470BG -- ^A color space designed for PAL/SECAM content+instance Enum ColorSpace where+ fromEnum Unspecified = #const TH_CS_UNSPECIFIED+ fromEnum ItuRec470M = #const TH_CS_ITU_REC_470M+ fromEnum ItuRec470BG = #const TH_CS_ITU_REC_470BG++ toEnum (#const TH_CS_UNSPECIFIED)=Unspecified+ toEnum (#const TH_CS_ITU_REC_470M)=ItuRec470M+ toEnum (#const TH_CS_ITU_REC_470BG)=ItuRec470BG+ +data Info=Info {frameWidth::Int32,+ frameHeight::Int32,+ picWidth::Int32,+ picHeight::Int32,+ picX::Int32,+ picY::Int32,+ colorSpace::ColorSpace,+ pixelFmt::PixelFmt,+ targetBitRate::CInt,+ quality::CInt,+ keyframeGranuleShift::CInt}+efault= #const TH_EFAULT+instance Storable Info where+ sizeOf _ = #size th_info+ alignment _=1+ peek ptr = do+ fw<-(#peek th_info, frame_width) ptr+ fh<-(#peek th_info, frame_height) ptr+ pw<-(#peek th_info, pic_width) ptr+ ph<-(#peek th_info, pic_width) ptr+ px<-(#peek th_info, pic_x) ptr+ py<-(#peek th_info, pic_y) ptr+ cspace<-(#peek th_info, colorspace) ptr+ pfmt<-(#peek th_info, pixel_fmt) ptr+ brate<-(#peek th_info, target_bitrate) ptr+ qual<-(#peek th_info, quality) ptr+ kframe<-(#peek th_info, keyframe_granule_shift) ptr+ return Info {frameWidth=fw,+ frameHeight=fh,+ picWidth=pw,+ picHeight=ph,+ picX=px,+ picY=py,+ colorSpace=toEnum cspace,+ pixelFmt=toEnum pfmt,+ targetBitRate=brate,+ quality=qual,+ keyframeGranuleShift=kframe}++ poke ptr (Info fw fh pw ph px py cspace pfmt brate qual kframe)=do+ (#poke th_info, frame_width) ptr fw+ (#poke th_info, frame_height) ptr fh+ (#poke th_info, pic_width) ptr pw+ (#poke th_info, pic_height) ptr ph+ (#poke th_info, pic_x) ptr px+ (#poke th_info, pic_y) ptr py+ (#poke th_info, colorspace) ptr $ fromEnum cspace+ (#poke th_info, pixel_fmt) ptr $ fromEnum pfmt+ (#poke th_info, target_bitrate) ptr brate+ (#poke th_info, quality) ptr qual+ (#poke th_info, keyframe_granule_shift) ptr kframe++data OggPacket=Packet {packet::Ptr CChar,+ bytes::CLong,+ bos::CLong,+ eos::CLong,+ granulePos::Int64,+ packetNo::Int64}+ +instance Storable OggPacket where+ sizeOf _= #size ogg_packet+ alignment _=1+ peek ptr=do+ packet<-(#peek ogg_packet, packet) ptr+ bytes<-(#peek ogg_packet, bytes) ptr+ bos<-(#peek ogg_packet, b_o_s) ptr+ eos<-(#peek ogg_packet, e_o_s) ptr+ granulePos<-(#peek ogg_packet, granulepos) ptr+ packetNo<-(#peek ogg_packet, packetno) ptr+ return $ Packet {packet=packet,+ bytes=bytes,+ bos=bos,+ eos=eos,+ granulePos=granulePos,+ packetNo=packetNo}+ poke ptr (Packet pack bytes bos eos gpos pno)=do+ (#poke ogg_packet, packet) ptr pack+ (#poke ogg_packet, bytes) ptr bytes+ (#poke ogg_packet, b_o_s) ptr bos+ (#poke ogg_packet, e_o_s) ptr eos+ (#poke ogg_packet, granulepos) ptr gpos+ (#poke ogg_packet, packetno) ptr pno+{- +instance Show OggPacket where+ show (Packet p b bos eos gran no)=+ unsafePerformIO (peekCStringLen (p,fromIntegral b))+-}++data ThEncCtx+type EncCtx=ForeignPtr ThEncCtx+foreign import ccall "theora/theora.h th_encode_alloc" th_encode_alloc::(Ptr Info)->IO (Ptr ThEncCtx)+foreign import ccall "theora/theora.h th_info_init" th_info_init::(Ptr Info)->IO ()+foreign import ccall "theora/theora.h &th_encode_free" th_encode_free::FunPtr ((Ptr ThEncCtx)->IO())++-- | Makes an encoder ('EncCtx') instance out of an 'Info' record+encodeAlloc::Info->IO (Maybe EncCtx)+encodeAlloc inf=do+ ptr<-alloca (\i->do+ th_info_init i+ poke i inf+ th_encode_alloc i)+ if ptr==nullPtr then return Nothing else+ newForeignPtr th_encode_free ptr >>= (return.Just)+++data Comment=Comment {+ user::[String], -- ^Your comments+ vendor::String -- ^Name of the encoder, i.e. your application+ }+instance Storable Comment where+ sizeOf _ = #size th_comment+ alignment _=1+ peek ptr=do+ length<- (#peek th_comment, comments) ptr+ c_userComments<-peekArray length =<< (#peek th_comment, comments) ptr+ userComments<-mapM (peekCString) c_userComments+ vendor<-(#peek th_comment, vendor) ptr >>= peekCString+ return $Comment {user=userComments,+ vendor=vendor}+ poke ptr comment= do+ -- user_comments+ userComments<-mallocArray (length $ user comment)+ cstrings<-mapM (\str->newCString str) $user comment+ pokeArray userComments cstrings+ (#poke th_comment, user_comments) ptr userComments+ -- comments_length+ commentLengths<-mallocArray (length $ user comment)+ pokeArray commentLengths (map length $ user comment)+ (#poke th_comment, comment_lengths) ptr commentLengths+ -- length+ (#poke th_comment, comments) ptr (length $ user comment)+ -- vendor+ (#poke th_comment, vendor) ptr =<< (newCString $ vendor comment)++data HeaderPacket=HeaderPacket+foreign import ccall "theora/theora.h th_comment_init" th_comment_init::Ptr Comment->IO()+foreign import ccall "theora/theora.h th_encode_flushheader" th_encode_flushheader::+ (Ptr ThEncCtx)->(Ptr Comment)->(Ptr OggPacket)->IO CInt++-- | Returns the last header packets. This function should be called before+-- encoding actual video.+flushHeader::EncCtx->Comment->IO [OggPacket]+flushHeader enc comment=+ withForeignPtr enc (\encctx->+ alloca (\ogg->+ alloca (\pComment->+ do+ th_comment_init pComment+ poke pComment comment+ ret<-th_encode_flushheader encctx pComment ogg+ if ret==0 || ret==efault then return [] else+ do+ tl<-flushHeader enc comment+ peek ogg >>= return.(:tl))))+++data ImgPlane=ImgPlane {width::CInt,+ height::CInt,+ stride::CInt,+ img::ForeignPtr CUChar}++instance Storable ImgPlane where+ sizeOf _= #size th_img_plane+ alignment _=1+ peek ptr=do+ width<-(#peek th_img_plane, width) ptr+ height<-(#peek th_img_plane, height) ptr+ stride<-(#peek th_img_plane, stride) ptr+ img<-((#peek th_img_plane, data) ptr) >>= newForeignPtr finalizerFree+ return $ ImgPlane {width=width,+ height=height,+ stride=stride,+ img=img}+ poke ptr (ImgPlane w h s i)=do+ (#poke th_img_plane, width) ptr w+ (#poke th_img_plane, height) ptr h+ (#poke th_img_plane, stride) ptr s+ withForeignPtr i (\img->(#poke th_img_plane, data) ptr img)++foreign import ccall "theora/theora.h th_encode_ycbcr_in" th_encode_ycbcr_in::+ (Ptr ThEncCtx)->(Ptr YCbCrBuffer)->IO CInt++-- | 'YCbCrBuffer' is the type of raw YCbCr frames+data YCbCrBuffer=YCbCr {+ y ::ImgPlane, -- ^ the Y plane+ cb::ImgPlane, -- ^ the Cb plane+ cr::ImgPlane -- ^ the Cr plane+ }++instance Storable YCbCrBuffer where+ sizeOf _= #size th_ycbcr_buffer+ alignment _= sizeOf (undefined :: Ptr ImgPlane)+ peek imgPtr=do+ let ptr=castPtr imgPtr+ y<-peek ptr+ cb<-peek (ptr`plusPtr`(sizeOf (undefined :: ImgPlane)))+ cr<-peek (ptr`plusPtr`(2*(sizeOf (undefined :: ImgPlane))))+ return $ YCbCr {y=y,+ cb=cb,+ cr=cr}+ poke imgPtr (YCbCr y cb cr)=do+ let ptr=castPtr imgPtr+ poke ptr y+ poke (ptr`plusPtr`(sizeOf (undefined::ImgPlane))) cb+ poke (ptr`plusPtr`(2*(sizeOf (undefined::ImgPlane)))) cr+++-- |Allocate a new YCbCrBuffer object+newYCbCr::Int->Int->PixelFmt->IO YCbCrBuffer+newYCbCr wH hH Pf420=do+ yImg<-newForeignPtr finalizerFree =<< (mallocBytes $ wH*hH)+ cbImg<-newForeignPtr finalizerFree =<< (mallocBytes $ (wH*hH)`quot`4)+ crImg<-newForeignPtr finalizerFree =<< (mallocBytes $ (wH*hH)`quot`4)+ let w=(fromIntegral wH)`quot`2+ h=(fromIntegral hH)`quot`2+ y=ImgPlane {width=fromIntegral wH,+ height=fromIntegral hH,+ stride=w,+ img=yImg}+ cb=ImgPlane{width=w,+ height=h,+ stride=w,+ img=cbImg}+ cr=ImgPlane{width=w,+ height=h,+ stride=w,+ img=crImg}+ return $ YCbCr {y=y,cb=cb,cr=cr}++-- |Submits a frame for encoding+encodeIn::EncCtx->YCbCrBuffer->IO()+encodeIn enc buf=+ withForeignPtr enc (\enc->alloca (\pbuf->do+ poke pbuf buf+ ret<-th_encode_ycbcr_in enc pbuf+ return ()))++foreign import ccall "theora/theora.h th_encode_packetout" th_encode_packetout::+ Ptr ThEncCtx->CInt->Ptr OggPacket->IO CInt++-- |Retrieve all the ready encoded packets+encodeOut::EncCtx->Bool->IO [OggPacket]+encodeOut e last=+ withForeignPtr e (\enc->+ alloca (\packet->do+ res<-th_encode_packetout enc (if last then 1 else 0) packet+ if res==0 || res==efault then return [] else do+ tl<-encodeOut e last+ peek packet >>= (return.(:tl))))+ ++data OggPage=OggPage{header::Ptr CChar,+ headerLen::CLong,+ body::Ptr CChar,+ bodyLen::CLong}++instance Storable OggPage where+ sizeOf _= #size ogg_page+ alignment _=1+ peek ptr=do+ header<-(#peek ogg_page, header) ptr+ headerLen<-(#peek ogg_page, header_len) ptr+ body<-(#peek ogg_page, body) ptr+ bodyLen<-(#peek ogg_page, body_len) ptr+ return $ OggPage{header=header, headerLen=headerLen, body=body, bodyLen=bodyLen}+ poke ptr x=do+ (#poke ogg_page, header) ptr (header x)+ (#poke ogg_page, header_len) ptr (headerLen x)+ (#poke ogg_page, body) ptr $ body x+ (#poke ogg_page, body_len) ptr $ bodyLen x+++instance Show OggPage where+ show (OggPage hdr hdrl body bodyl)=+ unsafePerformIO (do+ head<-peekCStringLen (hdr,fromIntegral hdrl)+ body<-peekCStringLen (body,fromIntegral bodyl)+ return $ head++body)+data COggStreamState+type OggStreamState=ForeignPtr COggStreamState+instance Storable COggStreamState where+ sizeOf _= #size ogg_stream_state+ alignment _=1++foreign import ccall "theora/theora.h ogg_stream_init" ogg_stream_init::+ (Ptr COggStreamState)->CInt->IO CInt+foreign import ccall "theora/theora.h ogg_stream_packetin" ogg_stream_packetin::+ (Ptr COggStreamState)->(Ptr OggPacket)->IO CInt+foreign import ccall "theora/theora.h ogg_stream_pageout" ogg_stream_pageout::+ (Ptr COggStreamState)->(Ptr OggPage)->IO CInt+foreign import ccall "theora/theora.h ogg_stream_flush" ogg_stream_flush::+ (Ptr COggStreamState)->(Ptr OggPage)->IO CInt++-- |Initializes an Ogg container stream+newOggStreamState::Int->IO OggStreamState+newOggStreamState initial=do+ state<-malloc+ success<-ogg_stream_init state $ fromIntegral initial+ newForeignPtr finalizerFree state++-- |Submits an encoded packet into the streams+streamPacketIn::OggStreamState->OggPacket->IO ()+streamPacketIn state packet=+ withForeignPtr state (\s->+ alloca(\pack->do+ poke pack packet+ ogg_stream_packetin s pack+ return ()))+-- |Get an outputable Ogg page +streamPageOut::OggStreamState->IO (Maybe OggPage)+streamPageOut state=+ withForeignPtr state (\s->+ alloca(\page->do+ result<-ogg_stream_pageout s page+ if result==0 then return Nothing else+ peek page >>= return.Just))++-- | Retrieve the last pages of the stream+streamFlush::OggStreamState->IO (Maybe OggPage)+streamFlush state=+ withForeignPtr state (\s->+ alloca (\page->do+ res<-ogg_stream_flush s page+ if res==0 then return Nothing else+ peek page >>= (return.Just) ))+