summaryrefslogtreecommitdiff
path: root/src/lib/imagemagick_encoder.cc
blob: 8d42b3dbea1bc9728bab9838ff4ad0d08f48eba4 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
/*
    Copyright (C) 2012 Carl Hetherington <cth@carlh.net>

    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.

*/

/** @file src/imagemagick_encoder.cc
 *  @brief An encoder that writes image files using ImageMagick (and does nothing with audio).
 */

#include <stdexcept>
#include <vector>
#include <sstream>
#include <iomanip>
#include <iostream>
#include <fstream>
#include <boost/filesystem.hpp>
#include <Magick++/Image.h>
#include "imagemagick_encoder.h"
#include "film.h"
#include "options.h"
#include "exceptions.h"
#include "image.h"
#include "subtitle.h"

using std::string;
using std::ofstream;
using boost::shared_ptr;

/** @param f Film that we are encoding.
 *  @param o Options.
 */
ImageMagickEncoder::ImageMagickEncoder (shared_ptr<const Film> f, shared_ptr<const Options> o)
	: Encoder (f, o)
{
	
}

void
ImageMagickEncoder::process_video (shared_ptr<Image> image, int frame, shared_ptr<Subtitle> sub)
{
	shared_ptr<Image> scaled = image->scale_and_convert_to_rgb (_opt->out_size, _opt->padding, _film->scaler());
	shared_ptr<Image> compact (new CompactImage (scaled));

	string tmp_file = _opt->frame_out_path (frame, true);
	Magick::Image thumb (compact->size().width, compact->size().height, "RGB", MagickCore::CharPixel, compact->data()[0]);
	thumb.magick ("PNG");
	thumb.write (tmp_file);
	boost::filesystem::rename (tmp_file, _opt->frame_out_path (frame, false));

	if (sub) {
		float const x_scale = float (_opt->out_size.width) / _film->size().width;
		float const y_scale = float (_opt->out_size.height) / _film->size().height;

		string tmp_metadata_file = _opt->frame_out_path (frame, false, ".sub");
		ofstream metadata (tmp_metadata_file.c_str ());
		
		Size new_size = sub->image()->size ();
		new_size.width *= x_scale;
		new_size.height *= y_scale;
		shared_ptr<Image> scaled = sub->image()->scale (new_size, _film->scaler());
		shared_ptr<Image> compact (new CompactImage (scaled));
		
		string tmp_sub_file = _opt->frame_out_path (frame, true, ".sub.png");
		Magick::Image sub_thumb (compact->size().width, compact->size().height, "RGBA", MagickCore::CharPixel, compact->data()[0]);
		sub_thumb.magick ("PNG");
		sub_thumb.write (tmp_sub_file);
		boost::filesystem::rename (tmp_sub_file, _opt->frame_out_path (frame, false, ".sub.png"));

		metadata << "x " << sub->position().x << "\n"
			 << "y " << sub->position().y << "\n";

		metadata.close ();
		boost::filesystem::rename (tmp_metadata_file, _opt->frame_out_path (frame, false, ".sub"));
	}
	
	frame_done (frame);
}