diff --git a/src/main/java/mjkf/xinke/main/constant/MaterialType.java b/src/main/java/mjkf/xinke/main/constant/MaterialType.java
new file mode 100644
index 0000000..eee365d
--- /dev/null
+++ b/src/main/java/mjkf/xinke/main/constant/MaterialType.java
@@ -0,0 +1,8 @@
+package mjkf.xinke.main.constant;
+
+
+public class MaterialType {
+    public static final Integer MAIN = 1;  // 主材
+    public static final Integer LOCAL = 2;  // 地材
+}
+
diff --git a/src/main/java/mjkf/xinke/main/controller/MaterialController.java b/src/main/java/mjkf/xinke/main/controller/MaterialController.java
index 558d66e..68dd660 100644
--- a/src/main/java/mjkf/xinke/main/controller/MaterialController.java
+++ b/src/main/java/mjkf/xinke/main/controller/MaterialController.java
@@ -39,9 +39,12 @@ public class MaterialController {
 
     @ApiOperation("获取材料树")
     @GetMapping("/tree")
-    public HttpResponse getTree () {
+    public HttpResponse getTree (
+            @ApiParam("类型") @RequestParam(value="type") Integer type
+    ) {
         LambdaQueryWrapper<Material> query = new LambdaQueryWrapper<>();
         query.eq(Material::getIsTree, 1);
+        query.eq(Material::getType, type);
         List<Map<String, Object>> maps = materialService.listMaps(query);
         maps = maps.stream().map(map -> convertKeysToLowercase(map)).collect(Collectors.toList());
         // todo 列转树改为泛型对象入参
diff --git a/src/main/java/mjkf/xinke/main/model/db/Material.java b/src/main/java/mjkf/xinke/main/model/db/Material.java
index bbe2deb..6eae6be 100644
--- a/src/main/java/mjkf/xinke/main/model/db/Material.java
+++ b/src/main/java/mjkf/xinke/main/model/db/Material.java
@@ -148,6 +148,7 @@ public class Material extends Model<Material> {
         this.unit = params.getUnit();
         this.spec = params.getSpec();
         this.tax = params.getTax();
+        this.type = params.getType();
         this.isTree = params.getIsTree();
         this.updateTime = LocalDateTime.now();
         this.createTime = LocalDateTime.now();
@@ -164,6 +165,7 @@ public class Material extends Model<Material> {
         this.unit = params.getUnit();
         this.spec = params.getSpec();
         this.tax = params.getTax();
+        this.type = params.getType();
         this.updateTime = LocalDateTime.now();
         this.updateUserName = user.getName();
         this.updateUserId = user.getId();
diff --git a/src/main/java/mjkf/xinke/main/model/vo/MaterialCreateRequest.java b/src/main/java/mjkf/xinke/main/model/vo/MaterialCreateRequest.java
index 09623bc..6cbfb0d 100644
--- a/src/main/java/mjkf/xinke/main/model/vo/MaterialCreateRequest.java
+++ b/src/main/java/mjkf/xinke/main/model/vo/MaterialCreateRequest.java
@@ -42,6 +42,9 @@ public class MaterialCreateRequest {
     @ApiModelProperty("税率")
     private Integer tax;
 
+    @ApiModelProperty("类型")
+    private Integer type;
+
     @ApiModelProperty("是否树")
     @JsonProperty(value = "is_tree")
     private Integer isTree;
diff --git a/src/main/java/mjkf/xinke/main/model/vo/MaterialEditRequest.java b/src/main/java/mjkf/xinke/main/model/vo/MaterialEditRequest.java
index 6726cdb..a20de42 100644
--- a/src/main/java/mjkf/xinke/main/model/vo/MaterialEditRequest.java
+++ b/src/main/java/mjkf/xinke/main/model/vo/MaterialEditRequest.java
@@ -30,6 +30,9 @@ public class MaterialEditRequest {
     @ApiModelProperty("税率")
     private Integer tax;
 
+    @ApiModelProperty("类型")
+    private Integer type;
+
 
     public void check() throws Exception{