belly

GainStation13/code/modules/mob/living/belly.dm

@@ -3,16 +3,17 @@
 	desc 					= "You see a belly on their midsection."
 	icon_state 				= "belly"
 	icon 					= 'hyperstation/icons/obj/genitals/belly.dmi'
-	zone 					= "chest"
-	slot 					= "belly"
+	zone 					= BODY_ZONE_CHEST
+	slot 					= ORGAN_SLOT_BELLY
 	w_class 				= 3
 	size 					= 0
 	var/statuscheck			= FALSE
-	shape					= "Pair"
+	shape					= "belly"
 	masturbation_verb 		= "massage"
 	var/sent_full_message	= TRUE //defaults to 1 since they're full to start
 	var/inflatable			= FALSE //For inflation connoisseurs
-
+	var/size_cached			= 0
+	var/prev_size			= 0
 
 /obj/item/organ/genital/belly/on_life()
 	if(QDELETED(src))
@@ -20,6 +21,16 @@
 	if(!owner)
 		return
 
+/obj/item/organ/genital/belly/modify_size(modifier, min = BELLY_SIZE_DEF, max = BELLY_SIZE_MAX)
+	var/new_value = clamp(size_cached + modifier, min, max)
+	if(new_value == size_cached)
+		return
+	prev_size = size_cached
+	size_cached = new_value
+	size = round(size_cached)
+	update()
+	..()
+
 /obj/item/organ/genital/belly/update_appearance()
 	var/string
 	if(owner)
@@ -36,4 +47,13 @@
 
 			icon_state = sanitize_text(string)
 
+/obj/item/organ/genital/belly/get_features(mob/living/carbon/human/H)
+	var/datum/dna/D = H.dna
+	if(D.species.use_skintones && D.features["genitals_use_skintone"])
+		color = SKINTONE2HEX(H.skin_tone)
+	else
+		color = "[D.features["belly_color"]]"
+	size = "[D.features["belly_size"]]"
+	toggle_visibility(D.features["belly_visibility"], FALSE)
+