godot/scene/resources/bit_mask.cpp
2015-04-18 14:38:54 -03:00

255 lines
6.5 KiB
C++

/*************************************************************************/
/* bit_mask.cpp */
/*************************************************************************/
/* This file is part of: */
/* GODOT ENGINE */
/* http://www.godotengine.org */
/*************************************************************************/
/* Copyright (c) 2007-2015 Juan Linietsky, Ariel Manzur. */
/* */
/* Permission is hereby granted, free of charge, to any person obtaining */
/* a copy of this software and associated documentation files (the */
/* "Software"), to deal in the Software without restriction, including */
/* without limitation the rights to use, copy, modify, merge, publish, */
/* distribute, sublicense, and/or sell copies of the Software, and to */
/* permit persons to whom the Software is furnished to do so, subject to */
/* the following conditions: */
/* */
/* The above copyright notice and this permission notice shall be */
/* included in all copies or substantial portions of the Software. */
/* */
/* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, */
/* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF */
/* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.*/
/* IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY */
/* CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, */
/* TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE */
/* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. */
/*************************************************************************/
#include "bit_mask.h"
#include "io/image_loader.h"
void BitMap::create(const Size2& p_size) {
ERR_FAIL_COND(p_size.width<1);
ERR_FAIL_COND(p_size.height<1);
width=p_size.width;
height=p_size.height;
bitmask.resize(((width*height)/8)+1);
zeromem(bitmask.ptr(),bitmask.size());
}
void BitMap::create_from_image_alpha(const Image& p_image){
ERR_FAIL_COND(p_image.empty());
Image img=p_image;
img.convert(Image::FORMAT_INTENSITY);
ERR_FAIL_COND(img.get_format()!=Image::FORMAT_INTENSITY);
create(Size2(img.get_width(),img.get_height()));
DVector<uint8_t>::Read r = img.get_data().read();
uint8_t *w = bitmask.ptr();
for(int i=0;i<width*height;i++) {
int bbyte = i/8;
int bbit = i % 8;
if (r[i])
w[bbyte]|=(1<<bbit);
}
}
void BitMap::set_bit_rect(const Rect2& p_rect,bool p_value) {
Rect2i current = Rect2i(0,0,width,height).clip(p_rect);
uint8_t *data = bitmask.ptr();
for(int i=current.pos.x;i<current.pos.x+current.size.x;i++) {
for(int j=current.pos.y;j<current.pos.y+current.size.y;j++) {
int ofs = width * j + i;
int bbyte = ofs/8;
int bbit = ofs % 8;
uint8_t b = data[bbyte];
if (p_value)
b|=(1<<bbit);
else
b&=!(1<<bbit);
data[bbyte]=b;
}
}
}
int BitMap::get_true_bit_count() const {
int ds = bitmask.size();
const uint8_t *d = bitmask.ptr();
int c=0;
//fast, almot branchless version
for(int i=0;i<ds;i++) {
c+=(d[i]&(1<<7))>>7;
c+=(d[i]&(1<<6))>>6;
c+=(d[i]&(1<<5))>>5;
c+=(d[i]&(1<<4))>>4;
c+=(d[i]&(1<<3))>>3;
c+=(d[i]&(1<<2))>>2;
c+=d[i]&1;
}
return c;
}
void BitMap::set_bit(const Point2& p_pos,bool p_value){
int x=Math::fast_ftoi(p_pos.x);
int y=Math::fast_ftoi(p_pos.y);
ERR_FAIL_INDEX(x,width);
ERR_FAIL_INDEX(y,height);
int ofs = width * y + x;
int bbyte = ofs/8;
int bbit = ofs % 8;
uint8_t b = bitmask[bbyte];
if (p_value)
b|=(1<<bbit);
else
b&=!(1<<bbit);
bitmask[bbyte]=b;
}
bool BitMap::get_bit(const Point2& p_pos) const{
int x=Math::fast_ftoi(p_pos.x);
int y=Math::fast_ftoi(p_pos.y);
ERR_FAIL_INDEX_V(x,width,false);
ERR_FAIL_INDEX_V(y,height,false);
int ofs = width * y + x;
int bbyte = ofs/8;
int bbit = ofs % 8;
return (bitmask[bbyte]&(1<<bbit))!=0;
}
Size2 BitMap::get_size() const {
return Size2(width,height);
}
void BitMap::_set_data(const Dictionary& p_d) {
ERR_FAIL_COND(!p_d.has("size"));
ERR_FAIL_COND(!p_d.has("data"));
create(p_d["size"]);
bitmask=p_d["data"];
}
Dictionary BitMap::_get_data() const{
Dictionary d;
d["size"]=get_size();
d["data"]=bitmask;
return d;
}
void BitMap::_bind_methods() {
ObjectTypeDB::bind_method(_MD("create","size"),&BitMap::create);
ObjectTypeDB::bind_method(_MD("create_from_image_alpha","image"),&BitMap::create_from_image_alpha);
ObjectTypeDB::bind_method(_MD("set_bit","pos","bit"),&BitMap::set_bit);
ObjectTypeDB::bind_method(_MD("get_bit","pos"),&BitMap::get_bit);
ObjectTypeDB::bind_method(_MD("set_bit_rect","p_rect","bit"),&BitMap::set_bit_rect);
ObjectTypeDB::bind_method(_MD("get_true_bit_count"),&BitMap::get_true_bit_count);
ObjectTypeDB::bind_method(_MD("get_size"),&BitMap::get_size);
ObjectTypeDB::bind_method(_MD("_set_data"),&BitMap::_set_data);
ObjectTypeDB::bind_method(_MD("_get_data"),&BitMap::_get_data);
ADD_PROPERTY(PropertyInfo(Variant::DICTIONARY,"data",PROPERTY_HINT_NONE,"",PROPERTY_USAGE_NOEDITOR),_SCS("_set_data"),_SCS("_get_data"));
}
BitMap::BitMap() {
width=0;
height=0;
}
//////////////////////////////////////
RES ResourceFormatLoaderBitMap::load(const String &p_path,const String& p_original_path) {
BitMap* ptr = memnew(BitMap);
Ref<BitMap> bitmap( ptr );
Image image;
Error err = ImageLoader::load_image(p_path,&image);
ERR_EXPLAIN("Failed loading image for BitMap: "+p_path);
ERR_FAIL_COND_V(err, RES());
bitmap->create_from_image_alpha(image);
return bitmap;
}
bool ResourceFormatLoaderBitMap::handles_type(const String& p_type) const {
return (p_type=="BitMap");
}
void ResourceFormatLoaderBitMap::get_recognized_extensions(List<String> *p_extensions) const {
ImageLoader::get_recognized_extensions(p_extensions);
}
String ResourceFormatLoaderBitMap::get_resource_type(const String &p_path) const {
List<String> extensions;
ImageLoader::get_recognized_extensions(&extensions);
String ext=p_path.extension().to_lower();
for(List<String>::Element *E=extensions.front();E;E=E->next()) {
if (E->get()==ext)
return "BitMap";
}
return "";
}
ResourceFormatLoaderBitMap::ResourceFormatLoaderBitMap() {
}