jni/filters/kmeans.cc

   1 /*
   2  * Copyright (C) 2012 The Android Open Source Project
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  *      http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16
  17 #include "filters.h"
  18 #include "kmeans.h"
  19
  20 #ifdef __cplusplus
  21 extern "C" {
  22 #endif
  23
  24 /*
  25  * For reasonable speeds:
  26  * k < 30
  27  * small_ds_bitmap width/height < 64 pixels.
  28  * large_ds_bitmap width/height < 512 pixels
  29  *
  30  * bad for high-frequency image noise
  31  */
  32
  33 void JNIFUNCF(ImageFilterKMeans, nativeApplyFilter, jobject bitmap, jint width, jint height,
  34         jobject large_ds_bitmap, jint lwidth, jint lheight, jobject small_ds_bitmap,
  35         jint swidth, jint sheight, jint p, jint seed)
  36 {
  37     char* destination = 0;
  38     char* larger_ds_dst = 0;
  39     char* smaller_ds_dst = 0;
  40     AndroidBitmap_lockPixels(env, bitmap, (void**) &destination);
  41     AndroidBitmap_lockPixels(env, large_ds_bitmap, (void**) &larger_ds_dst);
  42     AndroidBitmap_lockPixels(env, small_ds_bitmap, (void**) &smaller_ds_dst);
  43     unsigned char * dst = (unsigned char *) destination;
  44
  45     unsigned char * small_ds = (unsigned char *) smaller_ds_dst;
  46     unsigned char * large_ds = (unsigned char *) larger_ds_dst;
  47
  48     // setting for small bitmap
  49     int len = swidth * sheight * 4;
  50     int dimension = 3;
  51     int stride = 4;
  52     int iterations = 20;
  53     int k = p;
  54     unsigned int s = seed;
  55     unsigned char finalCentroids[k * stride];
  56
  57     // get initial picks from small downsampled image
  58     runKMeans<unsigned char, int>(k, finalCentroids, small_ds, len, dimension,
  59             stride, iterations, s);
  60
  61
  62     len = lwidth * lheight * 4;
  63     iterations = 8;
  64     unsigned char nextCentroids[k * stride];
  65
  66     // run kmeans on large downsampled image
  67     runKMeansWithPicks<unsigned char, int>(k, nextCentroids, large_ds, len,
  68             dimension, stride, iterations, finalCentroids);
  69
  70     len = width * height * 4;
  71
  72     // apply to final image
  73     applyCentroids<unsigned char, int>(k, nextCentroids, dst, len, dimension, stride);
  74
  75     AndroidBitmap_unlockPixels(env, small_ds_bitmap);
  76     AndroidBitmap_unlockPixels(env, large_ds_bitmap);
  77     AndroidBitmap_unlockPixels(env, bitmap);
  78 }
  79 #ifdef __cplusplus
  80 }
  81 #endif