--- /dev/null
+/*
+ * CINELERRA
+ * Copyright (C) 1997-2014 Adam Williams <broadcast at earthling dot 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., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
+ *
+ */
+
+#include "clip.h"
+#include "filexml.h"
+#include "puzzleobj.h"
+#include "puzzleobjwindow.h"
+#include "language.h"
+#include "mutex.h"
+
+REGISTER_PLUGIN(PuzzleObj)
+
+
+PuzzleObjConfig::PuzzleObjConfig()
+{
+ pixels = 400;
+ iterations = 10;
+}
+
+int PuzzleObjConfig::equivalent(PuzzleObjConfig &that)
+{
+ return 1;
+}
+
+void PuzzleObjConfig::copy_from(PuzzleObjConfig &that)
+{
+}
+
+void PuzzleObjConfig::interpolate( PuzzleObjConfig &prev, PuzzleObjConfig &next,
+ long prev_frame, long next_frame, long current_frame)
+{
+ copy_from(next);
+}
+
+void PuzzleObjConfig::limits()
+{
+}
+
+
+PuzzleObj::PuzzleObj(PluginServer *server)
+ : PluginVClient(server)
+{
+ ss_width = 0; ss_height = 0; ss_channels = 0;
+ ss_pixels = 0; ss_levels = 0; ss_prior = 0;
+ ss_hist_bins = 0;
+}
+
+PuzzleObj::~PuzzleObj()
+{
+}
+
+const char* PuzzleObj::plugin_title() { return N_("PuzzleObj"); }
+int PuzzleObj::is_realtime() { return 1; }
+
+NEW_WINDOW_MACRO(PuzzleObj, PuzzleObjWindow);
+LOAD_CONFIGURATION_MACRO(PuzzleObj, PuzzleObjConfig)
+
+void PuzzleObj::save_data(KeyFrame *keyframe)
+{
+ FileXML output;
+
+// cause data to be stored directly in text
+ output.set_shared_output(keyframe->get_data(), MESSAGESIZE);
+ output.tag.set_title("PUZZLEOBJ");
+ output.tag.set_property("PIXELS", config.pixels);
+ output.tag.set_property("ITERATIONS", config.iterations);
+ output.append_tag();
+ output.append_newline();
+ output.tag.set_title("/PUZZLEOBJ");
+ output.append_tag();
+ output.append_newline();
+ output.terminate_string();
+}
+
+void PuzzleObj::read_data(KeyFrame *keyframe)
+{
+ FileXML input;
+ input.set_shared_input(keyframe->get_data(), strlen(keyframe->get_data()));
+
+ int result = 0;
+ while( !(result = input.read_tag()) ) {
+ if( input.tag.title_is("PUZZLEOBJ") ) {
+ config.pixels = input.tag.get_property("PIXELS", config.pixels);
+ config.iterations = input.tag.get_property("ITERATIONS", config.iterations);
+ config.limits();
+ }
+ else if( input.tag.title_is("/PUZZLEOBJ") )
+ result = 1;
+ }
+}
+
+void PuzzleObj::update_gui()
+{
+ if( !thread ) return;
+ if( !load_configuration() ) return;
+ thread->window->lock_window("PuzzleObj::update_gui");
+ PuzzleObjWindow *window = (PuzzleObjWindow*)thread->window;
+ window->unlock_window();
+}
+
+void PuzzleObj::to_mat(Mat &mat, int mcols, int mrows,
+ VFrame *inp, int ix,int iy, int mcolor_model)
+{
+ int mcomp = BC_CModels::components(mcolor_model);
+ int mbpp = BC_CModels::calculate_pixelsize(mcolor_model);
+ int psz = mbpp / mcomp;
+ int mdepth = psz < 2 ? CV_8U : psz < 4 ? CV_16U : CV_32F;
+ if( mat.dims != 2 || mat.depth() != mdepth || mat.channels() != mcomp ||
+ mat.cols != mcols || mat.rows != mrows ) {
+ mat.release();
+ }
+ if( mat.empty() ) {
+ int type = CV_MAKETYPE(mdepth, mcomp);
+ mat.create(mrows, mcols, type);
+ }
+ uint8_t *mat_rows[mrows];
+ for( int y=0; y<mrows; ++y ) mat_rows[y] = mat.ptr(y);
+ uint8_t **inp_rows = inp->get_rows();
+ int ibpl = inp->get_bytes_per_line(), mbpl = mcols * mbpp;
+ int icolor_model = inp->get_color_model();
+ BC_CModels::transfer(mat_rows, mcolor_model, 0,0, mcols,mrows, mbpl,
+ inp_rows, icolor_model, ix,iy, mcols,mrows, ibpl, 0);
+// VFrame vfrm(mat_rows[0], -1, mcols,mrows, mcolor_model, mat_rows[1]-mat_rows[0]);
+// static int vfrm_no = 0; char vfn[64]; sprintf(vfn,"/tmp/idat/%06d.png", vfrm_no++);
+// vfrm.write_png(vfn);
+}
+
+int PuzzleObj::process_buffer(VFrame *frame, int64_t start_position, double frame_rate)
+{
+
+ //int need_reconfigure =
+ load_configuration();
+ input = get_input(0);
+ output = get_output(0);
+ width = input->get_w();
+ height = input->get_h();
+ color_model = input->get_color_model();
+ VFrame *iframe = input;
+ read_frame(iframe, 0, start_position, frame_rate, 0);
+ int cv_color_model = BC_RGB888;
+ if( color_model != cv_color_model ) {
+ iframe = new_temp(width,height, cv_color_model);
+ iframe->transfer_from(input);
+ }
+ int prior = 2;
+ int levels = 4;
+ int hist_bins = 5;
+ int channels = BC_CModels::components(cv_color_model);
+
+// load next image
+ to_mat(next_img, width,height, iframe, 0,0, cv_color_model);
+ Mat img; //, img1;
+ cvtColor(next_img, img, COLOR_RGB2HSV);
+ if( !sseeds.empty() &&
+ ( ss_width != width || ss_height != height || ss_channels != channels ||
+ ss_pixels != config.pixels || ss_levels != levels || ss_prior != prior ||
+ ss_hist_bins != hist_bins ) )
+ sseeds.release();
+ if( sseeds.empty() )
+ sseeds = ximgproc::createSuperpixelSEEDS(
+ ss_width=width, ss_height=height, ss_channels=channels,
+ ss_pixels=config.pixels, ss_levels=levels, ss_prior=prior,
+ ss_hist_bins=hist_bins, false);
+ sseeds->iterate(img, config.iterations);
+// sseeds->getLabels(img1);
+ sseeds->getLabelContourMask(img, false);
+ uint8_t **irows = input->get_rows();
+ int ibpp = BC_CModels::calculate_pixelsize(color_model);
+ uint8_t **orows = output->get_rows();
+ int obpp = BC_CModels::calculate_pixelsize(color_model);
+ if( BC_CModels::is_float(color_model) ) {
+ for( int y=0; y<height; ++y ) {
+ uint8_t *rp = img.ptr(y);
+ float *ip = (float*)irows[y], *op = (float*)orows[y];
+ for( int x=0; x<width; ++x,++rp ) {
+ if( *rp ) { op[0] = 1.0f; op[1] = 1.0f; op[2] = 1.0f; }
+ else { op[0] = ip[0]; op[1] = ip[1]; op[2] = ip[2]; }
+ ip = (float*)((char*)ip + ibpp);
+ op = (float*)((char*)op + obpp);
+ }
+ }
+ }
+ else if( BC_CModels::is_yuv(color_model) ) {
+ for( int y=0; y<height; ++y ) {
+ uint8_t *rp = img.ptr(y), *ip = irows[y], *op = orows[y];
+ for( int x=0; x<width; ++x,++rp ) {
+ if( *rp ) { op[0] = 0xff; op[1] = 0x80; op[2] = 0x80; }
+ else { op[0] = ip[0]; op[1] = ip[1]; op[2] = ip[2]; }
+ ip += ibpp; op += obpp;
+ }
+ }
+ }
+ else {
+ for( int y=0; y<height; ++y ) {
+ uint8_t *rp = img.ptr(y), *ip = irows[y], *op = orows[y];
+ for( int x=0; x<width; ++x,++rp ) {
+ if( *rp ) { op[0] = 0xff; op[1] = 0xff; op[2] = 0xff; }
+ else { op[0] = ip[0]; op[1] = ip[1]; op[2] = ip[2]; }
+ ip += ibpp; op += obpp;
+ }
+ }
+ }
+ return 0;
+}
+
projects / goodguy / history.git / blobdiff